Gabriele Cosmo
116 lines
4.3 KiB
C++
116 lines
4.3 KiB
C++
//
|
|
// ********************************************************************
|
|
// * License and Disclaimer *
|
|
// * *
|
|
// * The Geant4 software is copyright of the Copyright Holders of *
|
|
// * the Geant4 Collaboration. It is provided under the terms and *
|
|
// * conditions of the Geant4 Software License, included in the file *
|
|
// * LICENSE and available at http://cern.ch/geant4/license . These *
|
|
// * include a list of copyright holders. *
|
|
// * *
|
|
// * Neither the authors of this software system, nor their employing *
|
|
// * institutes,nor the agencies providing financial support for this *
|
|
// * work make any representation or warranty, express or implied, *
|
|
// * regarding this software system or assume any liability for its *
|
|
// * use. Please see the license in the file LICENSE and URL above *
|
|
// * for the full disclaimer and the limitation of liability. *
|
|
// * *
|
|
// * This code implementation is the result of the scientific and *
|
|
// * technical work of the GEANT4 collaboration. *
|
|
// * By using, copying, modifying or distributing the software (or *
|
|
// * any work based on the software) you agree to acknowledge its *
|
|
// * use in resulting scientific publications, and indicate your *
|
|
// * acceptance of all terms of the Geant4 Software license. *
|
|
// ********************************************************************
|
|
//
|
|
#ifndef G4MPIRUNMERGER_HH
|
|
#define G4MPIRUNMERGER_HH
|
|
#include "G4Run.hh"
|
|
#include <mpi.h>
|
|
#include "G4MPImanager.hh"
|
|
|
|
class G4VUserMPIrunMerger {
|
|
public:
|
|
G4VUserMPIrunMerger();
|
|
G4VUserMPIrunMerger( const G4Run* aRun ,
|
|
G4int destination = G4MPImanager::kRANK_MASTER ,
|
|
G4int verbosity = 0);
|
|
virtual ~G4VUserMPIrunMerger() { if ( ownsBuffer) DestroyBuffer(); }
|
|
void SetRun( G4Run* r ) { run = r; }
|
|
void SetDestinationRank( G4int i ) { destinationRank = i; }
|
|
void SetVerbosity( G4int ver ) { verbose = ver; }
|
|
|
|
virtual void Merge();
|
|
|
|
protected:
|
|
virtual void Pack() = 0;
|
|
virtual G4Run* UnPack() = 0;
|
|
|
|
void InputUserData( /*const*/ void* input_data ,const MPI::Datatype& dt, int count) {
|
|
input_userdata.push_back( const_registered_data{input_data,dt,count} );
|
|
}
|
|
void OutputUserData( void* input_data ,const MPI::Datatype& dt, int count) {
|
|
output_userdata.push_back( registered_data{input_data,dt,count} );
|
|
}
|
|
|
|
// void GetUserData(void* output_data,const MPI::Datatype& dt, int count);
|
|
|
|
void SetupOutputBuffer(char* buff, G4int size, G4int position) {
|
|
outputBuffer = buff;
|
|
outputBufferSize=size;
|
|
outputBufferPosition=position;
|
|
}
|
|
void DestroyBuffer() {
|
|
delete[] outputBuffer;
|
|
outputBuffer = nullptr;
|
|
outputBufferSize=0;
|
|
outputBufferPosition=0;
|
|
ownsBuffer = false;
|
|
}
|
|
|
|
G4int GetPosition() const { return outputBufferPosition; }
|
|
char* GetBuffer() const { return outputBuffer; }
|
|
G4int GetBufferSize() const { return outputBufferSize; }
|
|
|
|
void Send(const unsigned int destination);
|
|
void Receive(const unsigned int source);
|
|
private:
|
|
char* outputBuffer;
|
|
G4int outputBufferSize;
|
|
G4int outputBufferPosition;
|
|
G4bool ownsBuffer;
|
|
unsigned int destinationRank;
|
|
G4Run* run;
|
|
unsigned int commSize;
|
|
MPI::Intracomm COMM_G4COMMAND_;
|
|
G4int verbose;
|
|
long bytesSent;
|
|
|
|
//Input data to send (read-only)
|
|
struct const_registered_data {
|
|
const_registered_data(const const_registered_data&) = default;
|
|
const_registered_data& operator=(const const_registered_data&) = default;
|
|
//const_registered_data(const_registered_data&&) = default;
|
|
//const_registered_data& operator=(const_registered_data&&) = default;
|
|
/*const*/ void* p_data;
|
|
/*const*/ MPI::Datatype dt;
|
|
/*const*/ int count;
|
|
};
|
|
std::vector<const_registered_data> input_userdata;
|
|
|
|
//Output data
|
|
struct registered_data {
|
|
registered_data(const registered_data&) = default;
|
|
registered_data& operator=(const registered_data&) = default;
|
|
void* p_data;
|
|
/*const*/ MPI::Datatype dt;
|
|
/*const*/ int count;
|
|
};
|
|
std::vector<registered_data> output_userdata;
|
|
|
|
};
|
|
|
|
|
|
#endif //G4MPIRUNMERGER_HH
|
|
|