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的用法，只能在今后的设计中尽量避免出现该情况。