Gabriele Cosmo
251 lines
7.4 KiB
C++
251 lines
7.4 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. *
|
|
// ********************************************************************
|
|
//
|
|
//
|
|
// Author: Mathieu Karamitros (kara (AT) cenbg . in2p3 . fr)
|
|
//
|
|
// WARNING : This class is released as a prototype.
|
|
// It might strongly evolve or even disapear in the next releases.
|
|
//
|
|
// History:
|
|
// -----------
|
|
// 10 Oct 2011 M.Karamitros created
|
|
//
|
|
// -------------------------------------------------------------------
|
|
|
|
#ifndef G4ITStepManager_h
|
|
#define G4ITStepManager_h
|
|
|
|
#include "globals.hh"
|
|
#include "G4TrackList.hh"
|
|
#include "G4ITModelHandler.hh"
|
|
#include <vector>
|
|
#include <map>
|
|
#include <memory>
|
|
|
|
class G4ITTrackingManager;
|
|
class G4ITModelProcessor;
|
|
class G4ITStepProcessor;
|
|
class G4Track ;
|
|
class G4UserReactionAction;
|
|
|
|
/**
|
|
* G4ITStepManager enables to synchronize in time
|
|
* the step of tracks.
|
|
*/
|
|
class G4ITStepManager
|
|
{
|
|
|
|
public :
|
|
static G4ITStepManager* Instance();
|
|
/** DeleteInstance should be used instead
|
|
* of the destructor
|
|
*/
|
|
static void DeleteInstance();
|
|
|
|
~G4ITStepManager();
|
|
|
|
void Initialize() ;
|
|
void Reset();
|
|
void Process() ;
|
|
void PushTrack(G4Track*);
|
|
void ClearList();
|
|
void SetEndTime(const double);
|
|
|
|
inline void SetTimeTolerance(double);
|
|
// Two tracks below the time tolerance are supposed to be
|
|
// in the same time slice
|
|
inline double GetTimeTolerance() const;
|
|
|
|
inline G4ITModelHandler* GetModelHandler();
|
|
inline void SetTimeSteps(std::map<double,double>*);
|
|
inline G4int GetNbSteps() const;
|
|
inline G4double GetEndTime() const;
|
|
inline G4double GetTimeStep() const;
|
|
inline G4double GetPreviousTimeStep() const;
|
|
inline G4double GetGlobalTime() const;
|
|
inline void SetUserAction(G4UserReactionAction*);
|
|
inline G4UserReactionAction* GetUserReactionAction() const;
|
|
|
|
inline void SetVerbose(int);
|
|
// 1 : Reaction information
|
|
// 2 : (1) + step information
|
|
// 3 : (2) + trackList processing info + push track info
|
|
inline int GetVerbose() const;
|
|
|
|
protected:
|
|
|
|
void DoProcess();
|
|
void SynchronizeTracks();
|
|
void Stepping();
|
|
|
|
void FindUserPreDefinedTimeStep();
|
|
void CalculateMinTimeStep() ;
|
|
void ComputeInteractionLength();
|
|
void DoIt();
|
|
void ComputeTrackReaction();
|
|
void MergeSecondariesWithMainList();
|
|
|
|
void PushSecondaries(G4ITStepProcessor*);
|
|
void _PushTrack(G4Track*);
|
|
void PushDelayed(G4Track*, const G4double&);
|
|
|
|
void EndTracking(G4Track*);
|
|
void KillTracks();
|
|
void EndTracking();
|
|
|
|
void ExtractTimeStepperData(G4ITModelProcessor*);
|
|
void ExtractILData(G4ITStepProcessor*);
|
|
void ExtractDoItData(G4ITStepProcessor*);
|
|
|
|
void AddTrackID(G4Track*);
|
|
|
|
private:
|
|
G4ITStepManager();
|
|
|
|
static std::auto_ptr<G4ITStepManager> fgStepManager ;
|
|
int fVerbose;
|
|
bool fInitialized;
|
|
bool fRunning ;
|
|
|
|
int fNbTracks ;
|
|
int fNbSteps ;
|
|
|
|
// Time members
|
|
double fTimeTolerance;
|
|
double fGlobalTime;
|
|
double fTmpGlobalTime ;
|
|
double fEndTime ; // fEndTime : stores the biggest global time steps here (for synchronizing tracks)
|
|
double fTmpEndTime ; // fTmpEndTime : Stores the biggest end time here (for synchronizing tracks)
|
|
double fPreviousStepTime;
|
|
|
|
// Flags
|
|
bool fComputeTimeStep;
|
|
bool fComputeReaction;
|
|
|
|
double fTimeStep ; // The selected minimum time step
|
|
|
|
double fTSTimeStep; // Time calculated by the time stepper in CalculateMinTimeStep()
|
|
double fILTimeStep; // Time calculated by the interaction length methods in ComputeInteractionLength()
|
|
|
|
// User steps
|
|
bool fUsePreDefinedTimeSteps ;
|
|
std::map<double, double>* fpUserTimeSteps ; // One can give time steps in respect to the global time
|
|
double fDefinedMinTimeStep ; // selected user time step in respect to the global time
|
|
bool fReachedUserTimeLimit ; // if fMinTimeStep == the user time step
|
|
|
|
std::map<G4Track*, G4TrackVectorHandle> fReactingTracks ;
|
|
std::vector<G4Track*> fLeadingTracks ;
|
|
|
|
bool fInteractionStep ;
|
|
// Flag : if the step is lead by the interaction with the matter and NOT by tracks' reaction
|
|
|
|
G4TrackList* fpMainList ;
|
|
G4TrackList fSecondaries ; //to merge with fpMainList
|
|
G4TrackList* fpWaitingList ; // Waiting queue of currentList
|
|
std::map<double,G4TrackList* > fDelayedList ;
|
|
G4TrackList fToBeKilledList ;
|
|
|
|
G4ITStepProcessor* fpStepProcessor ;
|
|
G4ITStepProcessor* fpMasterStepProcessor ;
|
|
|
|
G4ITModelProcessor* fpModelProcessor;
|
|
G4ITModelProcessor* fpMasterModelProcessor;
|
|
|
|
G4ITModelHandler* fpModelHandler;
|
|
|
|
G4ITTrackingManager* fpTrackingManager;
|
|
G4UserReactionAction* fpUserReactionAction;
|
|
|
|
};
|
|
|
|
inline G4ITModelHandler* G4ITStepManager::GetModelHandler() {return fpModelHandler;}
|
|
|
|
inline void G4ITStepManager::SetEndTime(const double __endtime) { fEndTime = __endtime ;}
|
|
|
|
inline void G4ITStepManager::SetTimeSteps(std::map<double,double>* steps)
|
|
{
|
|
fUsePreDefinedTimeSteps = true;
|
|
fpUserTimeSteps = steps ;
|
|
}
|
|
|
|
inline G4int G4ITStepManager::GetNbSteps() const
|
|
{
|
|
return fNbSteps;
|
|
}
|
|
|
|
inline G4double G4ITStepManager::GetEndTime() const
|
|
{
|
|
return fEndTime;
|
|
}
|
|
|
|
inline G4double G4ITStepManager::GetTimeStep() const
|
|
{
|
|
return fTimeStep ;
|
|
}
|
|
|
|
|
|
inline G4double G4ITStepManager::GetGlobalTime() const
|
|
{
|
|
return fGlobalTime ;
|
|
}
|
|
|
|
inline void G4ITStepManager::SetUserAction(G4UserReactionAction* userITAction)
|
|
{
|
|
fpUserReactionAction = userITAction;
|
|
}
|
|
|
|
inline G4UserReactionAction* G4ITStepManager::GetUserReactionAction() const
|
|
{
|
|
return fpUserReactionAction;
|
|
}
|
|
|
|
inline void G4ITStepManager::SetVerbose(int verbose)
|
|
{
|
|
fVerbose = verbose;
|
|
}
|
|
|
|
inline int G4ITStepManager::GetVerbose() const
|
|
{
|
|
return fVerbose ;
|
|
}
|
|
|
|
inline void G4ITStepManager::SetTimeTolerance(double time)
|
|
{
|
|
fTimeTolerance = time;
|
|
}
|
|
|
|
inline double G4ITStepManager::GetTimeTolerance() const
|
|
{
|
|
return fTimeTolerance;
|
|
}
|
|
|
|
inline G4double G4ITStepManager::GetPreviousTimeStep() const
|
|
{
|
|
return fPreviousStepTime;
|
|
}
|
|
|
|
#endif
|