Kamil Sedlak
fcd5eea567
This is the first version of the muSR simulation code (musrSim) based on the merged codes of Kamil Sedlak and Toni Shiroka. It should be a running version of the simulation code, however it has not been very well tested, therefore it will probably need some further development.
292 lines
12 KiB
C++
292 lines
12 KiB
C++
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
|
|
|
|
#ifndef musrRootOutput_h
|
|
#define musrRootOutput_h 1
|
|
|
|
//#include "G4UserRunAction.hh"
|
|
#include "globals.hh"
|
|
#include "G4ThreeVector.hh"
|
|
// ROOT
|
|
#include "TFile.h"
|
|
#include "TTree.h"
|
|
#include "TH1.h"
|
|
#include "TH2.h"
|
|
#include "TVectorD.h"
|
|
//
|
|
#include <map>
|
|
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......
|
|
|
|
class musrRootOutput {
|
|
public:
|
|
musrRootOutput();
|
|
~musrRootOutput();
|
|
static musrRootOutput* GetRootInstance();
|
|
|
|
public:
|
|
void BeginOfRunAction();
|
|
void EndOfRunAction();
|
|
void FillEvent();
|
|
void ClearAllRootVariables();
|
|
void SetVolumeIDMapping(std::string logivol, int volumeID);
|
|
G4int ConvertVolumeToID(std::string logivol);
|
|
G4int ConvertProcessToID(std::string processName);
|
|
void SetSpecialSaveVolumeDefined() {boolIsAnySpecialSaveVolumeDefined=true;};
|
|
|
|
// Getting variables (just for debugging)
|
|
G4double GetDecayPositionZ() {return muDecayPosZ_t;};
|
|
G4double GetDecayTime() {return muDecayTime_t*microsecond;};
|
|
G4double GetTimeInTarget() {return muTargetTime_t*microsecond;};
|
|
|
|
// Setting variables common to the whole event:
|
|
void SetRunID (G4int id) {runID_t = id;};
|
|
void SetEventID (G4int id) {eventID_t = id;};
|
|
void SetDecayDetectorID (std::string detectorName) {muDecayDetID_t = SensDetectorMapping[detectorName];};
|
|
void SetBField (G4double F[6]) {B_t[0]=F[0]/tesla; B_t[1]=F[1]/tesla; B_t[2]=F[2]/tesla;
|
|
B_t[3]=F[3]/tesla; B_t[4]=F[4]/tesla; B_t[5]=F[5]/tesla;};
|
|
void SetDecayPolarisation (G4ThreeVector pol) {muDecayPolX_t=pol.x(); muDecayPolY_t=pol.y(); muDecayPolZ_t=pol.z();};
|
|
void SetDecayPosition (G4ThreeVector pos) {muDecayPosX_t=pos.x()/mm; muDecayPosY_t=pos.y()/mm;
|
|
muDecayPosZ_t=pos.z()/mm;};
|
|
void SetEventWeight (G4double w) {weight_t *= w;}
|
|
void SetDetectorInfo (G4int nDetectors, G4int ID, G4int particleID, G4double edep,
|
|
G4double edep_el, G4double edep_pos,
|
|
G4double edep_gam, G4double edep_mup,G4int nsteps, G4double length, G4double t1,
|
|
G4double t2, G4double x, G4double y, G4double z,
|
|
G4double ek, G4double ekVertex, G4double xVertex, G4double yVertex, G4double zVertex,
|
|
G4int idVolVertex, G4int idProcVertex, G4int idTrackVertex) ;
|
|
|
|
void SetDetectorInfoVvv (G4int nDetectors,
|
|
G4double ekVertex, G4double xVertex, G4double yVertex, G4double zVertex,
|
|
G4int idVolVertex, G4int idProcVertex, G4int idTrackVertex, G4int particleID) ;
|
|
|
|
void SetSaveDetectorInfo (G4int ID, G4int particleID, G4double ke, G4double x, G4double y, G4double z,
|
|
G4double px, G4double py, G4double pz) ;
|
|
|
|
void SetInitialMuonParameters(G4double x, G4double y, G4double z, G4double px, G4double py, G4double pz,
|
|
G4double xpolaris, G4double ypolaris, G4double zpolaris) {
|
|
muIniPosX_t=x; muIniPosY_t=y; muIniPosZ_t=z;
|
|
muIniMomX_t=px; muIniMomY_t=py; muIniMomZ_t=pz;
|
|
muIniPolX_t=xpolaris; muIniPolY_t=ypolaris; muIniPolZ_t=zpolaris;
|
|
}
|
|
void PrintInitialMuonParameters() {
|
|
G4cout<<"musrRootOutput.hh: Initial muon parameters: x="<<muIniPosX_t<<", y="<<muIniPosY_t<<", z="<<muIniPosZ_t
|
|
<<", px="<<muIniMomX_t << ", py="<<muIniMomY_t<<", pz="<<muIniMomZ_t<<G4endl;
|
|
G4cout<<" polx="<<muIniPolX_t<<", poly="<<muIniPolY_t<<", polz="<<muIniPolZ_t<<G4endl;
|
|
G4cout<<" numberOfGeneratedEvents = "<<GeantParametersD[7]<<G4endl;
|
|
}
|
|
|
|
void SetPolInTarget(G4ThreeVector pol) {muTargetPolX_t=pol.x(); muTargetPolY_t=pol.y(); muTargetPolZ_t=pol.z();};
|
|
void SetTimeInTarget(G4double time) {muTargetTime_t = time/microsecond;};
|
|
void SetPolInM0(G4ThreeVector pol) {muM0PolX_t=pol.x(); muM0PolY_t=pol.y(); muM0PolZ_t=pol.z();};
|
|
void SetTimeInM0(G4double time) {muM0Time_t = time/microsecond;};
|
|
void SetPolInM1(G4ThreeVector pol) {muM1PolX_t=pol.x(); muM1PolY_t=pol.y(); muM1PolZ_t=pol.z();};
|
|
void SetTimeInM1(G4double time) {muM1Time_t = time/microsecond;};
|
|
void SetPolInM2(G4ThreeVector pol) {muM2PolX_t=pol.x(); muM2PolY_t=pol.y(); muM2PolZ_t=pol.z();};
|
|
void SetTimeInM2(G4double time) {muM2Time_t = time/microsecond;};
|
|
void SetInitialPositronMomentum(G4ThreeVector mom) {posIniMomx_t=mom.x(); posIniMomy_t=mom.y(); posIniMomz_t=mom.z();};
|
|
void SetDecayTime(G4double time) {muDecayTime_t=time/microsecond;};
|
|
void SetNrFieldNomVal(G4int n) {nFieldNomVal = n;}
|
|
void SetFieldNomVal(G4int i, G4double value);
|
|
G4int GetNrOfVolumes() {return det_nMax;}
|
|
void SetBFieldIntegral(G4double BxInt,G4double ByInt,G4double BzInt,G4double BzInt1,G4double BzInt2,G4double BzInt3) {
|
|
BxIntegral_t=BxInt/m/tesla; ByIntegral_t=ByInt/m/tesla; BzIntegral_t=BzInt/m/tesla;
|
|
BzIntegral1_t=BzInt1/m/tesla;BzIntegral2_t=BzInt2/mm;BzIntegral3_t=BzInt3/mm;
|
|
}
|
|
void StoreGeantParameter(Int_t i, Double_t value) {
|
|
if (i<maxNGeantParameters) { GeantParametersD[i]=value; }
|
|
else {G4cout<<"musrRootOutput.hh::StoreGeantParameter: index="<<i<<" out of range"
|
|
<<" (maxNGeantParameters=" <<maxNGeantParameters<<")"<<G4endl;}
|
|
};
|
|
|
|
|
|
TH2F *htest1, *htest2;
|
|
TH1F *htest3, *htest4, *htest5, *htest6, *htest7, *htest8;
|
|
|
|
public:
|
|
static G4bool store_runID;
|
|
static G4bool store_eventID;
|
|
static G4bool store_weight;
|
|
static G4bool store_BFieldAtDecay;
|
|
static G4bool store_muIniPosX;
|
|
static G4bool store_muIniPosY;
|
|
static G4bool store_muIniPosZ;
|
|
static G4bool store_muIniMomX;
|
|
static G4bool store_muIniMomY;
|
|
static G4bool store_muIniMomZ;
|
|
static G4bool store_muIniPolX;
|
|
static G4bool store_muIniPolY;
|
|
static G4bool store_muIniPolZ;
|
|
static G4bool store_muDecayDetID;
|
|
static G4bool store_muDecayPosX;
|
|
static G4bool store_muDecayPosY;
|
|
static G4bool store_muDecayPosZ;
|
|
static G4bool store_muDecayTime;
|
|
static G4bool store_muDecayPolX;
|
|
static G4bool store_muDecayPolY;
|
|
static G4bool store_muDecayPolZ;
|
|
static G4bool store_muTargetTime;
|
|
static G4bool store_muTargetPolX;
|
|
static G4bool store_muTargetPolY;
|
|
static G4bool store_muTargetPolZ;
|
|
static G4bool store_muM0Time;
|
|
static G4bool store_muM0PolX;
|
|
static G4bool store_muM0PolY;
|
|
static G4bool store_muM0PolZ;
|
|
static G4bool store_muM1Time;
|
|
static G4bool store_muM1PolX;
|
|
static G4bool store_muM1PolY;
|
|
static G4bool store_muM1PolZ;
|
|
static G4bool store_muM2Time;
|
|
static G4bool store_muM2PolX;
|
|
static G4bool store_muM2PolY;
|
|
static G4bool store_muM2PolZ;
|
|
static G4bool store_posIniMomX;
|
|
static G4bool store_posIniMomY;
|
|
static G4bool store_posIniMomZ;
|
|
static G4bool store_det_ID;
|
|
static G4bool store_det_edep;
|
|
static G4bool store_det_edep_el;
|
|
static G4bool store_det_edep_pos;
|
|
static G4bool store_det_edep_gam;
|
|
static G4bool store_det_edep_mup;
|
|
static G4bool store_det_nsteps;
|
|
static G4bool store_det_length;
|
|
static G4bool store_det_start;
|
|
static G4bool store_det_end;
|
|
static G4bool store_det_x;
|
|
static G4bool store_det_y;
|
|
static G4bool store_det_z;
|
|
static G4bool store_det_kine;
|
|
static G4bool store_det_VrtxKine;
|
|
static G4bool store_det_VrtxX;
|
|
static G4bool store_det_VrtxY;
|
|
static G4bool store_det_VrtxZ;
|
|
static G4bool store_det_VrtxVolID;
|
|
static G4bool store_det_VrtxProcID;
|
|
static G4bool store_det_VrtxTrackID;
|
|
static G4bool store_det_VrtxParticleID;
|
|
static G4bool store_det_VvvKine;
|
|
static G4bool store_det_VvvX;
|
|
static G4bool store_det_VvvY;
|
|
static G4bool store_det_VvvZ;
|
|
static G4bool store_det_VvvVolID;
|
|
static G4bool store_det_VvvProcID;
|
|
static G4bool store_det_VvvTrackID;
|
|
static G4bool store_det_VvvParticleID;
|
|
static G4bool store_fieldNomVal;
|
|
static G4bool store_fieldIntegralBx;
|
|
static G4bool store_fieldIntegralBy;
|
|
static G4bool store_fieldIntegralBz;
|
|
static G4bool store_fieldIntegralBz1;
|
|
static G4bool store_fieldIntegralBz2;
|
|
static G4bool store_fieldIntegralBz3;
|
|
|
|
|
|
static G4int oldEventNumberInG4EqEMFieldWithSpinFunction;
|
|
|
|
private:
|
|
TFile* rootFile;
|
|
TTree* rootTree;
|
|
static musrRootOutput* pointerToRoot;
|
|
static const Int_t maxNGeantParameters=30;
|
|
Double_t GeantParametersD[maxNGeantParameters]; // parameters transfered from GEANT to Root
|
|
// 0 ... fieldOption: 0 ... no field, 1 ... uniform, 2 ... gaussian, 3 ... from table
|
|
// 1 ... fieldValue: intensity of the magnetic field
|
|
// 2 ... minimum of the generated decay time of the muon (in microsecond)
|
|
// 3 ... maximum of the generated decay time of the muon (in microsecond)
|
|
// 4 ... muon mean life time (in microsecond)
|
|
// 5 ... nr. of the last generated event
|
|
// 6 ... run number
|
|
// 7 ... numberOfGeneratedEvents (i.e. number of the generated events;
|
|
// in case of Turtle nr. of events tried);
|
|
|
|
// Variables common to the whole event:
|
|
Int_t runID_t;
|
|
Int_t eventID_t;
|
|
Double_t weight_t;
|
|
Double_t B_t[6];
|
|
Double_t muIniPosX_t, muIniPosY_t, muIniPosZ_t;
|
|
Double_t muIniMomX_t, muIniMomY_t, muIniMomZ_t;
|
|
Double_t muIniPolX_t, muIniPolY_t, muIniPolZ_t;
|
|
Int_t muDecayDetID_t;
|
|
Double_t muDecayPolX_t, muDecayPolY_t, muDecayPolZ_t;
|
|
Double_t muTargetTime_t, muTargetPolX_t, muTargetPolY_t, muTargetPolZ_t;
|
|
Double_t muM0Time_t, muM0PolX_t, muM0PolY_t, muM0PolZ_t;
|
|
Double_t muM1Time_t, muM1PolX_t, muM1PolY_t, muM1PolZ_t;
|
|
Double_t muM2Time_t, muM2PolX_t, muM2PolY_t, muM2PolZ_t;
|
|
Double_t muDecayPosX_t, muDecayPosY_t, muDecayPosZ_t;
|
|
Double_t muDecayTime_t;
|
|
Double_t posIniMomx_t, posIniMomy_t, posIniMomz_t;
|
|
|
|
public:
|
|
static const Int_t maxNFieldnNominalValues=30;
|
|
private:
|
|
Int_t nFieldNomVal;
|
|
Double_t fieldNomVal[maxNFieldnNominalValues];
|
|
Double_t BxIntegral_t, ByIntegral_t, BzIntegral_t;
|
|
Double_t BzIntegral1_t, BzIntegral2_t, BzIntegral3_t;
|
|
|
|
// Variables for a particle in a given detector within the event
|
|
public:
|
|
static const Int_t maxNSubTracks=30;
|
|
private:
|
|
// Variables for the activity inside a given detector
|
|
public:
|
|
static const Int_t det_nMax=100; // must be by 1 higher than the real number of detector "hits", because
|
|
// else the detector nr. 0 is counted (0 is used if no
|
|
// SensDetectorMapping correspond to a given logical volume).
|
|
private:
|
|
G4int det_n;
|
|
G4int det_ID[det_nMax];
|
|
G4double det_edep[det_nMax];
|
|
G4int det_nsteps[det_nMax];
|
|
G4double det_length[det_nMax];
|
|
G4double det_edep_el[det_nMax];
|
|
G4double det_edep_pos[det_nMax];
|
|
G4double det_edep_gam[det_nMax];
|
|
G4double det_edep_mup[det_nMax];
|
|
G4double det_time_start[det_nMax];
|
|
G4double det_time_end[det_nMax];
|
|
G4double det_x[det_nMax];
|
|
G4double det_y[det_nMax];
|
|
G4double det_z[det_nMax];
|
|
G4double det_kine[det_nMax];
|
|
G4double det_VrtxKine[det_nMax];
|
|
G4double det_VrtxX[det_nMax];
|
|
G4double det_VrtxY[det_nMax];
|
|
G4double det_VrtxZ[det_nMax];
|
|
G4int det_VrtxVolID[det_nMax];
|
|
G4int det_VrtxProcID[det_nMax];
|
|
G4int det_VrtxTrackID[det_nMax];
|
|
G4int det_VrtxParticleID[det_nMax];
|
|
G4double det_VvvKine[det_nMax];
|
|
G4double det_VvvX[det_nMax];
|
|
G4double det_VvvY[det_nMax];
|
|
G4double det_VvvZ[det_nMax];
|
|
G4int det_VvvVolID[det_nMax];
|
|
G4int det_VvvProcID[det_nMax];
|
|
G4int det_VvvTrackID[det_nMax];
|
|
G4int det_VvvParticleID[det_nMax];
|
|
|
|
public:
|
|
static const Int_t save_nMax=2000;
|
|
|
|
private:
|
|
G4int save_n;
|
|
G4int save_detID[save_nMax];
|
|
G4int save_particleID[save_nMax];
|
|
G4double save_ke[save_nMax];
|
|
G4double save_x[save_nMax];
|
|
G4double save_y[save_nMax];
|
|
G4double save_z[save_nMax];
|
|
G4double save_px[save_nMax];
|
|
G4double save_py[save_nMax];
|
|
G4double save_pz[save_nMax];
|
|
|
|
G4bool boolIsAnySpecialSaveVolumeDefined;
|
|
|
|
std::map<std::string,int> SensDetectorMapping;
|
|
std::map<std::string,int> ProcessIDMapping;
|
|
};
|
|
|
|
#endif
|