1 |
// $Id: run_g4.C,v 1.0 2007/06/01 |
2 |
// |
3 |
// Macro for running G4PAM_VMC with Geant4. |
4 |
void run_g4_GP(const TString& configMacro = "../macro/g4Config.C") { |
5 |
|
6 |
// Load basic libraries |
7 |
gROOT->LoadMacro("../macro/basiclibs.C"); |
8 |
basiclibs(); |
9 |
|
10 |
// Load PAMELA libs |
11 |
TString PAMLIB=gSystem->Getenv("PAM_LIB"); |
12 |
gSystem->Load(PAMLIB+"/libyoda.so"); |
13 |
gSystem->Load(PAMLIB+"/libDarthVader.so"); |
14 |
gSystem->Load(PAMLIB+"/libPamLevel2.so"); |
15 |
|
16 |
// Load Geant4 libraries |
17 |
|
18 |
gROOT->LoadMacro("../macro/PamG4Libs.C"); |
19 |
PamG4Libs(); |
20 |
|
21 |
// Load this example libraries |
22 |
TString G4WD=gSystem->Getenv("G4WORKDIR"); |
23 |
TString PLAT=gSystem->Getenv("PLATFORM"); |
24 |
gSystem->Load(G4WD+"/lib/tgt_"+PLAT+"/libPamVMC_fc"); |
25 |
|
26 |
// MC application |
27 |
PamVMCApplication* appl |
28 |
= new PamVMCApplication("PAMG4_VMC", "PAMELA GEANT4 VMC application"); |
29 |
|
30 |
|
31 |
// Initialize MC |
32 |
appl->InitMC(configMacro); |
33 |
((TGeant4*)gMC)->ProcessGeantMacro("../macro/g4config2.in"); |
34 |
|
35 |
//START READING INPUT FILE |
36 |
|
37 |
Int_t Ipa; |
38 |
Float_t X0, Y0, Z0, Theta, Phi, P0; |
39 |
|
40 |
TTree * h20 = 0; |
41 |
|
42 |
TString Filename = "protons_N0_1-2.Beam.root"; |
43 |
|
44 |
TString HOME=gSystem->Getenv("HOME"); |
45 |
|
46 |
TSystemDirectory *readdir=new TSystemDirectory("readdir",HOME+"/RES"); |
47 |
if ((TList*)(readdir->GetListOfFiles())->FindObject(Filename) ){ |
48 |
cout<< Filename<<" found..."<<endl; |
49 |
TFile* inpf = new TFile(HOME+"/RES/"+Filename); |
50 |
h20 = (TTree*)inpf->Get("h20"); |
51 |
if (h20->GetBranch("Ipa")){ |
52 |
h20->SetBranchAddress("Ipa",&Ipa); |
53 |
h20->SetBranchStatus("Ipa",1); |
54 |
} |
55 |
if (h20->GetBranch("X0")){ |
56 |
h20->SetBranchAddress("X0",&X0); |
57 |
h20->SetBranchStatus("X0",1); |
58 |
} |
59 |
if (h20->GetBranch("Y0")){ |
60 |
h20->SetBranchAddress("Y0",&Y0); |
61 |
h20->SetBranchStatus("Y0",1); |
62 |
} |
63 |
if (h20->GetBranch("Z0")){ |
64 |
h20->SetBranchAddress("Z0",&Z0); |
65 |
h20->SetBranchStatus("Z0",1); |
66 |
} |
67 |
if (h20->GetBranch("P0")){ |
68 |
h20->SetBranchAddress("P0",&P0); |
69 |
h20->SetBranchStatus("P0",1); |
70 |
} |
71 |
if (h20->GetBranch("Theta")){ |
72 |
h20->SetBranchAddress("Theta",&Theta); |
73 |
h20->SetBranchStatus("Theta",1); |
74 |
} |
75 |
if (h20->GetBranch("Phi")){ |
76 |
h20->SetBranchAddress("Phi",&Phi); |
77 |
h20->SetBranchStatus("Phi",1); |
78 |
} |
79 |
} |
80 |
|
81 |
Int_t nevents = (Int_t)h20->GetEntries(); |
82 |
|
83 |
nevents = 1000; |
84 |
cout<<"PROCESS "<<nevents<<" EVENTS"<<endl; |
85 |
|
86 |
TParticlePDG* particlePDG = TDatabasePDG::Instance()->GetParticle(kProton); //correct |
87 |
Double_t mass = particlePDG->Mass(); |
88 |
Double_t PX, PY, PZ; |
89 |
for (Int_t i=0; i<nevents; i++){ |
90 |
h20->GetEntry(i); |
91 |
cout<<"Ipa="<<Ipa<<endl; |
92 |
appl->GetPrimaryGenerator()->SetParticle(kProton); |
93 |
appl->GetPrimaryGenerator()->SetMomentum(P0); |
94 |
cout<<"EKIN "<<appl->GetPrimaryGenerator()->GetKinEnergy()<<" GeV"<<endl; |
95 |
cout<<"applP0="<<appl->GetPrimaryGenerator()->GetMomentum()<<" GV"<<endl; |
96 |
cout<<"applRIG "<<appl->GetPrimaryGenerator()->GetRigidity()<<" GV"<<endl; |
97 |
PX=P0*sin(Theta)*cos(Phi); |
98 |
PY=P0*sin(Theta)*sin(Phi); |
99 |
PZ=-P0*cos(Theta); |
100 |
cout<<"(PX0,PY0,PZ0)="<<PX<<","<<PY<<","<<PZ<<endl; |
101 |
appl->GetPrimaryGenerator()->SetDirection(Theta,Phi); |
102 |
cout<<"(X0,Y0,Z0)"<<X0<<","<<Y0<<","<<Z0<<endl; |
103 |
appl->GetPrimaryGenerator()->SetPosition(X0,Y0,Z0); |
104 |
|
105 |
|
106 |
appl->RunMC(1); |
107 |
|
108 |
} |
109 |
|
110 |
|
111 |
// Visualization settings |
112 |
// ((TGeant4*)gMC)->ProcessGeantMacro("../macro/g4vis.in"); |
113 |
// inpf.cd(); |
114 |
// delete inpf; |
115 |
// Run MC |
116 |
// appl->RunMC(1000); |
117 |
appl->FinishRun(); |
118 |
((TGeant4*)gMC)->StartGeantUI(); |
119 |
// delete appl; |
120 |
|
121 |
} |