Gabriele Cosmo
126 lines
4.5 KiB
C++
126 lines
4.5 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. *
|
|
// ********************************************************************
|
|
/// @file G4VUserMPIrunMerger.hh
|
|
/// @brief A base class for user run merger
|
|
|
|
#ifndef G4MPIRUNMERGER_HH
|
|
#define G4MPIRUNMERGER_HH
|
|
#include "G4MPImanager.hh"
|
|
#include "G4Run.hh"
|
|
|
|
#include <mpi.h>
|
|
|
|
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;
|
|
G4int destinationRank;
|
|
G4Run* run;
|
|
unsigned int commSize;
|
|
MPI_Comm 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
|