global.h

MPICH是MPI的重要研究,提供了一系列的接口函数,为并行计算的实现提供了编程环境.

#ifndef GLOBAL_H
#define GLOBAL_H

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#ifdef WSOCK2_BEFORE_WINDOWS
#include <winsock2.h>
#endif
#include <windows.h>
#include <tchar.h>

#define MAX_CMD_LENGTH                8192
#define MAX_HOST_LENGTH	                64
#define MPIRUN_DEFAULT_TIMEOUT          30
#define MPIRUN_SHORT_TIMEOUT            15
#define MPIRUN_CREATE_PROCESS_TIMEOUT   60

struct HostNode
{
    TCHAR host[100];
    TCHAR exe[MAX_CMD_LENGTH];
    long nSMPProcs;
    HostNode *next;
};

struct ForwardHostStruct
{
    char pszHost[MAX_HOST_LENGTH];
    int nPort;
};

struct MapDriveNode
{
    char cDrive;
    char pszShare[MAX_PATH];
    //char pszAccount[40];
    //char pszPassword[40];
    MapDriveNode *pNext;
};

// Global variables
extern MapDriveNode *g_pDriveMapList;
extern bool g_bNoDriveMapping;
extern HANDLE g_hRedirectIOListenThread;
extern SOCKET g_sockStopIOSignalSocket;
extern HANDLE g_hAbortEvent;
extern SOCKET *g_pProcessSocket;
extern int *g_pProcessLaunchId;
extern int *g_pLaunchIdToRank;
extern LONG g_nNumProcessSockets;
#define FORWARD_NPROC_THRESHOLD 8
extern ForwardHostStruct *g_pForwardHost;
extern SOCKET g_sockBreak;
extern HANDLE g_hBreakReadyEvent;
extern HostNode *g_pHosts;
extern long g_nHosts;
extern int g_nNproc;
extern long g_nRootPort;
extern TCHAR g_pszAccount[100], g_pszPassword[100];
extern bool g_bNoMPI;
extern char g_pszExe[MAX_CMD_LENGTH], g_pszArgs[MAX_CMD_LENGTH], g_pszEnv[MAX_CMD_LENGTH];
extern char g_pszDir[MAX_PATH];
extern char g_pszExeOrig[MAX_CMD_LENGTH];
extern char g_pszFirstHost[MAX_HOST_LENGTH];
extern HANDLE g_hFinishedEvent;
extern HANDLE g_hConsoleOutputMutex;
extern char g_pszIOHost[MAX_HOST_LENGTH];
extern int g_nIOPort;
extern bool g_bDoMultiColorOutput;
extern WORD g_ConsoleAttribute;
extern bool g_bUseJobHost;
extern bool g_bOutputExitCodes;
extern char g_pszJobHost[MAX_HOST_LENGTH];
extern bool g_bUseJobMPDPwd;
extern char g_pszJobHostMPDPwd[100];
extern int g_nLaunchTimeout;
extern bool g_bSuppressErrorOutput;
extern HANDLE g_hLaunchThreadsRunning;
extern bool g_bUseMPDUser;
extern int g_nMPIRUN_DEFAULT_TIMEOUT;
extern int g_nMPIRUN_SHORT_TIMEOUT;
extern int g_nMPIRUN_CREATE_PROCESS_TIMEOUT;

#include <wincon.h>
#define NUM_OUTPUT_COLORS 32

// foregrounds
#define frgnd_RGB	FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY
#define frgnd_RG	FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_INTENSITY
#define frgnd_RB	FOREGROUND_RED | FOREGROUND_BLUE | FOREGROUND_INTENSITY
#define frgnd_R		FOREGROUND_RED | FOREGROUND_INTENSITY
#define frgnd_GB	FOREGROUND_GREEN | FOREGROUND_BLUE | FOREGROUND_INTENSITY
#define frgnd_G		FOREGROUND_GREEN | FOREGROUND_INTENSITY
#define frgnd_B		FOREGROUND_BLUE | FOREGROUND_INTENSITY
#define frgnd_rgb	FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE
#define frgnd_rg	FOREGROUND_RED | FOREGROUND_GREEN
#define frgnd_rb	FOREGROUND_RED | FOREGROUND_BLUE
#define frgnd_r		FOREGROUND_RED
#define frgnd_gb	FOREGROUND_GREEN | FOREGROUND_BLUE
#define frgnd_g		FOREGROUND_GREEN
#define frgnd_b		FOREGROUND_BLUE

// backgrounds
#define bkgnd_RGB	BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE | BACKGROUND_INTENSITY
#define bkgnd_RG	BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_INTENSITY
#define bkgnd_RB	BACKGROUND_RED | BACKGROUND_BLUE | BACKGROUND_INTENSITY
#define bkgnd_R		BACKGROUND_RED | BACKGROUND_INTENSITY
#define bkgnd_GB	BACKGROUND_GREEN | BACKGROUND_BLUE | BACKGROUND_INTENSITY
#define bkgnd_G		BACKGROUND_GREEN | BACKGROUND_INTENSITY
#define bkgnd_B		BACKGROUND_BLUE | BACKGROUND_INTENSITY
#define bkgnd_rgb	BACKGROUND_RED | BACKGROUND_GREEN | BACKGROUND_BLUE
#define bkgnd_rg	BACKGROUND_RED | BACKGROUND_GREEN
#define bkgnd_rb	BACKGROUND_RED | BACKGROUND_BLUE
#define bkgnd_r		BACKGROUND_RED
#define bkgnd_gb	BACKGROUND_GREEN | BACKGROUND_BLUE
#define bkgnd_g		BACKGROUND_GREEN
#define bkgnd_b		BACKGROUND_BLUE
extern WORD aConsoleColorAttribute[NUM_OUTPUT_COLORS];

#endif