Send / Recv Usage(2) - Refactoring

觉得每次生成新的MPI类型是一件很麻烦的事情，于是重构了一下代码使得该过程更方便一点。

#include "Class_mpi.h" #include <cmath> class CTestSendRecv { public: CTestSendRecv(){Init();} void Init() { m_d =5.6; m_f =8.2f; m_n =11; m_arr[0]= 1.1; m_arr[1]= 2.2; m_arr[2]= 3.3;m_arr[3]= 4.4; } static MPI::Datatype GenerateMPINewType() { // 与成员变量对应 int blockLength[] = {1,1,4,1}; MPI::Datatype oldTypes[] = {MPI_INT,MPI_DOUBLE,MPI_DOUBLE,MPI_FLOAT}; //直接计算偏移量,每一个数据以double对齐 MPI::Aint addressOffsets[]={0,sizeof(double),2*sizeof(double),6*sizeof(double)}; return MPI::Datatype::Create_struct( sizeof(blockLength)/sizeof(int), /**/ blockLength, addressOffsets, oldTypes);; } // int m_n; double m_d; double m_arr[4]; float m_f; }; int main(int argc, char* argv[]) { Cmpi mpi; //mpi.PrintMachineInfo(); mpi.PrintMachineInfoWithStdIO(); MPI::Datatype newType = CTestSendRecv::GenerateMPINewType();//CTestSendRecv::GenerateMPINewType(); newType.Commit(); const int nMsgTag =999; /*消息标识号 */ int nSendProcessId(1); /*发送数据的进程的编号*/ int nRecvProcessId(0); /*目的进程的标识号*/ const int minProcessorCount =2; if (mpi.GetTotalProcessCount() < minProcessorCount) { std::cout<<"Not enough processor"<<std::endl; return 0; } if (mpi.GetCurrentProcessId()==nSendProcessId) { CTestSendRecv testSend; testSend.m_arr[1]= -1.1; MPI::COMM_WORLD.Send(&testSend, /*发送数据的内存地址*/ sizeof(testSend)/sizeof(CTestSendRecv), /*发送数据的数目*/ newType, /*发送数据的类型*/ nRecvProcessId, /*目的进程的标识号*/ nMsgTag /*本消息标识号 MPI_ANY_TAG*/ ); } else { CTestSendRecv testRecv;; /*接收数据的变量*/ if (nSendProcessId >= mpi.GetTotalProcessCount() ) return 0; MPI::COMM_WORLD.Recv(&testRecv, sizeof(testRecv)/sizeof(CTestSendRecv), newType, nSendProcessId, nMsgTag); std::cout<<"Data="<<testRecv.m_arr[1]<<std::endl; } newType.Free(); return 0; }

 

只使用一个调用语句来生成自定义数据的MPI类型，这样用起来就稍稍简单一些：

MPI::Datatype newType = CTestSendRecv::GenerateMPINewType();

 

 

此外，也尝试了一下嵌套的自定义数据类型的数据交换，可惜没有成功。可能的原因是我还没有深入理解MPI的用法，只能在今后的设计中尽量避免出现该情况。