ESyS-Particle
4.0.1
|
00001 00002 // // 00003 // Copyright (c) 2003-2011 by The University of Queensland // 00004 // Earth Systems Science Computational Centre (ESSCC) // 00005 // http://www.uq.edu.au/esscc // 00006 // // 00007 // Primary Business: Brisbane, Queensland, Australia // 00008 // Licensed under the Open Software License version 3.0 // 00009 // http://www.opensource.org/licenses/osl-3.0.php // 00010 // // 00012 00013 #ifndef __MPIVBUF_H 00014 #define __MPIVBUF_H 00015 00016 #include <mpi.h> 00017 #include <string> 00018 #include "Parallel/mpibuf.h" 00019 00034 class CVarMPIBuffer : public AMPIBufferPP 00035 { 00036 private: 00037 char* m_buffer; 00038 int m_buffersize; 00039 int m_position; 00040 int m_int_increment,m_dbl_increment; 00041 bool m_lock; 00042 MPI_Request m_req[2]; 00043 MPI_Status m_stat[2]; 00044 00045 protected: 00046 void grow(); 00047 void growTo(int); 00048 00049 public: 00050 CVarMPIBuffer(MPI_Comm,int size=16); 00051 virtual ~CVarMPIBuffer(); 00052 00053 virtual void clear(){m_position=0;}; 00054 virtual void sendTo(int,int); 00055 virtual void NBsendTo(int,int); 00056 virtual void initSendTo(int,int); 00057 virtual void wait(); 00058 virtual void receiveFrom(int src=MPI_ANY_SOURCE,int tag=MPI_ANY_TAG); 00059 virtual void append(int); 00060 virtual void append(double); 00061 virtual void append(const char*); 00062 virtual void append(const Vec3 &V) { AMPIBufferPP::append(V); } ; 00063 virtual int pop_int(); 00064 virtual double pop_double(); 00065 virtual void pop_doubles(double*,int); 00066 virtual std::string pop_string(); 00067 virtual void broadcast(int); 00068 virtual void receiveBroadcast(int); 00069 }; 00070 #endif //__MPIVBUF_H