Fork me on GitHub

source: svn/trunk/src/FrogUtil.cc@ 239

Last change on this file since 239 was 219, checked in by Xavier Rouby, 16 years ago

JetUtils.cc

File size: 22.3 KB
RevLine 
[97]1/*
2 * ---- Delphes ----
3 * A Fast Simulator for general purpose LHC detector
4 * S. Ovyn ~~~~ severine.ovyn@uclouvain.be
5 *
6 * Center for Particle Physics and Phenomenology (CP3)
7 * Universite Catholique de Louvain (UCL)
8 * Louvain-la-Neuve, Belgium
9 * */
10
11// \brief Trigger class, and some generic definitions
[219]12#include <iostream>
[97]13
[219]14#include "TROOT.h"
15#include "TApplication.h"
16#include "TStyle.h"
17#include "TChain.h"
18#include "TClonesArray.h"
[97]19
[219]20#include "BlockClasses.h"
21#include "ExRootTreeReader.h"
22#include "FrogUtil.h"
[97]23
[219]24#include "FROG_Coord.h"
25#include "FROG_Coord.cpp"
26#include "FROG_Geometry.h"
27#include "FROG_Geometry.cpp"
28#include "FROG_DetId.h"
29#include "FROG_Events.h"
30#include "FROG_Events.cpp"
31#include "FROG_Element_Tools.h"
32#include "FROG_Element_Tools.cpp"
33#include "FROG_Net_Tools.h"
34#include "FROG_Net_Tools.cpp"
35#include "FROG_Path.h"
36#include "FROG_Path.cpp"
37#include "FROG_ReadCards.h"
38#include "FROG_ReadCards.cpp"
39#include "FROG_ZLib.h"
40#include "FROG_ZLib.cpp"
[97]41
42using namespace std;
43using namespace FROG_COORD;
44
[219]45const float twopi = 6.283185307179586476925286766559;
[97]46
[219]47FrogDisplay::FrogDisplay() {
48 //Smearing information
49 DET = new RESOLution();
50 nEntryFrog = DET->NEvents_Frog;
51}
[97]52
[219]53FrogDisplay::FrogDisplay(const string& DetDatacard) {
54 //Smearing information
55 DET = new RESOLution();
56 DET->ReadDataCard(DetDatacard);
57 nEntryFrog = DET->NEvents_Frog;
58}
[112]59
[219]60FrogDisplay::FrogDisplay(const RESOLution* DetDatacard) {
61 //Smearing information
62 DET = new RESOLution(*DetDatacard);
63 nEntryFrog = DET->NEvents_Frog;
64}
[97]65
[219]66FrogDisplay::FrogDisplay(const FrogDisplay& frog) {
67 DET = new RESOLution(*(frog.DET));
68 nEntryFrog = DET->NEvents_Frog;
[97]69}
70
[219]71FrogDisplay& FrogDisplay::operator=(const FrogDisplay& frog) {
72 if (this==&frog) return *this;
73 DET = new RESOLution(*(frog.DET));
74 nEntryFrog = DET->NEvents_Frog;
75 return *this;
76}
77
78void FrogDisplay::BuildEvents(const string& outputfilename) {
[97]79 const Font_t kExRootFont = 42;
80 const Float_t kExRootFontSize = 0.07;
81 gROOT->SetStyle("Plain");
82 gROOT->cd();
83 gStyle->SetCanvasColor(10);
84 gStyle->SetPadColor(10);
85 gStyle->SetFillColor(-1);
86 gStyle->SetPaperSize(20, 24);
87 gStyle->SetStatFont(kExRootFont);
88 gStyle->SetTextFont(kExRootFont);
89 gStyle->SetTextSize(kExRootFontSize);
90 gStyle->SetLegendBorderSize(0);
91 gStyle->SetOptStat(0);
92
93 TChain* chain = new TChain("Analysis");
94 TChain* chainGen = new TChain("GEN");
95 chain ->Add(outputfilename.c_str());
96 chainGen->Add(outputfilename.c_str());
97
98 ExRootTreeReader *treeReader = new ExRootTreeReader(chain);
99 ExRootTreeReader *treeReaderGen = new ExRootTreeReader(chainGen);
100
101 const TClonesArray* branchElectron = treeReader ->UseBranch("Electron");
102 const TClonesArray* branchMuon = treeReader ->UseBranch("Muon");
103 const TClonesArray* branchTau = treeReader ->UseBranch("TauJet");
104 const TClonesArray* branchPhoton = treeReader ->UseBranch("Photon");
105 const TClonesArray* branchJet = treeReader ->UseBranch("Jet");
106 const TClonesArray* branchMET = treeReader ->UseBranch("ETmis");
107
108 TRootElectron* elec;
109 TRootMuon* muon;
110 TRootTauJet* tau;
111 TRootPhoton* photon;
112 TRootJet* jet;
113 TRootETmis* met;
114
[219]115 unsigned int allEntries=treeReader->GetEntries();
[97]116
117 FROG_Events* frog_events = new FROG_Events();
[219]118 if(nEntryFrog > allEntries) {
119 cout<<"nEntry to display by FROG greater than number of available events... "<<endl;
120 nEntryFrog = allEntries;
121 }
122
123 for(unsigned int entry = 0; entry < nEntryFrog; ++entry){
124 //cout<<"Event n° "<<entry<<endl;
[97]125 treeReader ->ReadEntry(entry);
126 treeReaderGen->ReadEntry(entry);
127
128 //Create a new event
129 FROG_Element_Event* frog_event = new FROG_Element_Event(1,entry);
130 frog_events->AddEvent(frog_event);
131
132 //Create a new branch in the event with name SIM and ID = EVTID_SIM
133 FROG_Element_Base_With_DetId_And_Name* frog_branchSIM = new FROG_Element_Base_With_DetId_And_Name(EVTID_SIM,"SIM");
134 frog_event->addDaughter(frog_branchSIM);
[219]135
136
[97]137 //SIM ELEC
138 //Create a new sub-branch in the SIM branch with name Electrons: this sub branch will contais all the electrons
139 FROG_Element_Base_With_DetId_And_Name* frog_branchElectrons = new FROG_Element_Base_With_DetId_And_Name(EVTID_SIM+1000,"Electrons");
140 frog_branchSIM->addDaughter(frog_branchElectrons);
141
142 for(int p=0;p<branchElectron->GetEntriesFast();p++){
143 elec = (TRootElectron*) branchElectron->At(p);
[219]144 // cout<<"Elec nO "<<p<<" PT: "<<elec->PT<<" Eta "<<elec->Eta<<" Phi"<<elec->Phi<<endl;
[97]145 FROG_Element_Event_Candidate* frog_elec = new FROG_Element_Event_Candidate(11,elec->E,elec->Eta,elec->Phi);
146 frog_branchElectrons->addDaughter(frog_elec);
147 }
[219]148
[97]149
150 //SIM MUONS
151 //Create a new sub-branch in the SIM branch with name Muons: this sub branch will contais all the muons
152 FROG_Element_Base_With_DetId_And_Name* frog_branchMuons = new FROG_Element_Base_With_DetId_And_Name(EVTID_SIM+2000,"Muons");
153 frog_branchSIM->addDaughter(frog_branchMuons);
154 for(int p=0;p<branchMuon->GetEntriesFast();p++){
155 muon = (TRootMuon*) branchMuon->At(p);
[219]156 // cout<<"Muon nO"<<p<<" PT: "<<muon->PT<<"Eta "<<muon->Eta<<" Phi"<<muon->Phi<<endl;
[97]157 FROG_Element_Event_Candidate* frog_muon = new FROG_Element_Event_Candidate(13,muon->E,muon->Eta,muon->Phi);
158 frog_branchMuons->addDaughter(frog_muon);
159 }
[219]160
[97]161
162 //SIM TAUS
163 //Create a new sub-branch in the SIM branch with name Taus: this sub branch will contais all the Taus
164 FROG_Element_Base_With_DetId_And_Name* frog_branchTaus = new FROG_Element_Base_With_DetId_And_Name(EVTID_SIM+3000,"Taus");
165 frog_branchSIM->addDaughter(frog_branchTaus);
166 for(int p=0;p<branchTau->GetEntriesFast();p++){
167 tau = (TRootTauJet*) branchTau->At(p);
[219]168 // cout<<"Tau nO"<<p<<" PT: "<<tau->PT<<"Eta "<<tau->Eta<<" Phi"<<tau->Phi<<endl;
[97]169 FROG_Element_Event_Candidate* frog_tau = new FROG_Element_Event_Candidate(15,tau->E,tau->Eta,tau->Phi);
170 frog_branchTaus->addDaughter(frog_tau);
171 }
[219]172
[97]173
174 //SIM PHOTONS
175 //Create a new sub-branch in the SIM branch with name Photons: this sub branch will contais all the Photons
176 FROG_Element_Base_With_DetId_And_Name* frog_branchPhotons = new FROG_Element_Base_With_DetId_And_Name(EVTID_SIM+4000,"Photons");
177 frog_branchSIM->addDaughter(frog_branchPhotons);
178 for(int p=0;p<branchPhoton->GetEntriesFast();p++){
179 photon = (TRootPhoton*) branchPhoton->At(p);
[219]180 // cout<<"Photon nO"<<p<<" PT: "<<photon->PT<<"Eta "<<photon->Eta<<" Phi"<<photon->Phi<<endl;
[107]181 FROG_Element_Event_Candidate* frog_photon = new FROG_Element_Event_Candidate(22,photon->E,photon->Eta,photon->Phi);
[97]182 frog_branchPhotons->addDaughter(frog_photon);
183 }
[219]184
[97]185
186 //SIM JETS
[112]187 //Create a new sub-branch in the SIM branch with name Jets: this sub branch will contais all the Jets
[97]188 FROG_Element_Base_With_DetId_And_Name* frog_branchJets = new FROG_Element_Base_With_DetId_And_Name(EVTID_SIM+5000,"Jets");
189 frog_branchSIM->addDaughter(frog_branchJets);
190 for(int p=0;p<branchJet->GetEntriesFast();p++){
191 jet = (TRootJet*) branchJet->At(p);
[219]192 // cout<<"Jet nO"<<p<<" PT: "<<jet->PT<<"Eta "<<jet->Eta<<" Phi"<<jet->Phi<<endl;
[97]193 FROG_Element_Event_Jet* frog_jet = new FROG_Element_Event_Jet(jet->E,jet->Eta,jet->Phi);
194 frog_branchJets->addDaughter(frog_jet);
195 }
196
[219]197
[97]198 //SIM MET
199 //Create a new sub-branch in the SIM branch with name MET: this sub branch will contais all the METs
200 FROG_Element_Base_With_DetId_And_Name* frog_branchMET = new FROG_Element_Base_With_DetId_And_Name(EVTID_SIM+6000,"MET");
201 frog_branchSIM->addDaughter(frog_branchMET);
202 for(int p=0;p<branchMET->GetEntriesFast();p++){
203 met = (TRootETmis*) branchMET->At(p);
[219]204 // cout<<"MET nO"<<p<<" ET: "<<met->ET<<"Eta "<<0<<" Phi"<<met->Phi<<endl;
[97]205 FROG_Element_Event_MET* frog_met = new FROG_Element_Event_MET(0,met->ET,0,met->Phi, met->ET);
206 frog_branchMET->addDaughter(frog_met);
207 }
208
209 //Save the event in the .vis file
210 frog_events->SaveInLive("DelphesToFrog.vis",false,false,(unsigned int)-1);
211 }
[219]212
213/* Pointer hierachy
214 frog_events
215 -frog_event
216 -frog_branchSIM
217 -frog_branchElectrons
218 -frog_elec
219 -frog_branchMuons
220 -frog_muon
221 -frog_branchTaus
222 -frog_tau
223 -frog_branchPhotons
224 -frog_photon
225 -frog_photon
226 -frog_jet
227 -frog_branchMET
228 -frog_met
229*/
230
[97]231 delete treeReader;
232 delete treeReaderGen;
233 delete frog_events;
[219]234 delete chain;
235 delete chainGen;
[97]236}
237
[219]238void FrogDisplay::BuildGeom() {
239
[97]240 // This element is the root of the "file" tree. (don't change this line)
241 FROG_Element_Base* prim = new FROG_Element_Base(C_PRIMARY);
[219]242
[97]243 FROG_Element_Base* mygeom = new FROG_Element_Base(C_GEOMETRY);
244 prim->addDaughter(mygeom);
245
246 FROG_Element_Base_With_DetId_And_Name* detector = new FROG_Element_Base_With_DetId_And_Name(900000000,"Delphes");
247 mygeom->addDaughter(detector);
248
[112]249 double Rayon_Tracker=40;
[100]250 double Rayon_Calo = Rayon_Tracker*1.5;
251 double Rayon_Muon = Rayon_Tracker*2;
252 double Lenght_Tracker=100;
253 double Lenght_Calo=Lenght_Tracker+Lenght_Tracker/2.5;
254 double Lenght_Muon=Lenght_Calo+Lenght_Calo/2.5;
[112]255 double Lenght_CaloFwd=Lenght_Muon+Lenght_Muon/2.5;
[97]256
[112]257 int NumPhi=100;
258 float frac=1;
[97]259
260 //************************************************Tracker*************************************************
261 //********************************************************************************************************
262
263 FROG_Element_Base_With_DetId_And_Name* Tracker = new FROG_Element_Base_With_DetId_And_Name(910000000,"Tracker");
264 detector->addDaughter(Tracker);
265 unsigned int DetIdCountTracker = 1;
[112]266
267 double ray=Rayon_Tracker;
268 double rayCone = tan(EtaToTheta(DET->CEN_max_tracker))*Lenght_Tracker/2;
269 if(ray < rayCone)ray=rayCone;
270
[219]271 double dphi = twopi /NumPhi;
[183]272 for(double phi=0; phi<=twopi/frac;phi+=dphi){
[112]273
274 FROG_Element_Primitive_CustomSurface* trackerCone1 = new FROG_Element_Primitive_CustomSurface(9100000+DetIdCountTracker*10,
[183]275 rayCone*sin(phi) ,rayCone*cos(phi) ,-Lenght_Tracker/2,
276 rayCone*sin(phi+dphi) ,rayCone*cos(phi+dphi) ,-Lenght_Tracker/2,
277 0 , 0 , 0,
278 0 , 0 , 0);
[112]279 Tracker->addDaughter(trackerCone1); DetIdCountTracker++;
280
281 FROG_Element_Primitive_CustomSurface* trackerCone2 = new FROG_Element_Primitive_CustomSurface(9100000+DetIdCountTracker*10,
[183]282 0 , 0 ,0,
283 0 , 0 ,0,
284 rayCone*sin(phi) , rayCone*cos(phi) , Lenght_Tracker/2,
285 rayCone*sin(phi+dphi) , rayCone*sin(phi+dphi) , Lenght_Tracker/2);
[112]286 Tracker->addDaughter(trackerCone2); DetIdCountTracker++;
287
288
289 FROG_Element_Primitive_CustomSurface* trackerB = new FROG_Element_Primitive_CustomSurface(9100000+DetIdCountTracker*10,
[183]290 ray*sin(phi) ,ray*cos(phi) ,-Lenght_Tracker*0.5,
291 ray*sin(phi+dphi) ,ray*cos(phi+dphi) ,-Lenght_Tracker*0.5,
292 ray*sin(phi+dphi) ,ray*cos(phi+dphi) , Lenght_Tracker*0.5,
293 ray*sin(phi) ,ray*cos(phi) , Lenght_Tracker*0.5);
[112]294 Tracker->addDaughter(trackerB); DetIdCountTracker++;
295
296 FROG_Element_Primitive_CustomSurface* trackerEndCap1 = new FROG_Element_Primitive_CustomSurface(9100000+DetIdCountTracker*10,
[183]297 rayCone*sin(phi) , rayCone*cos(phi) , -(Lenght_Tracker)/2,
298 rayCone*sin(phi+dphi) , rayCone*cos(phi+dphi) , -(Lenght_Tracker)/2,
299 ray*sin(phi+dphi) , ray*cos(phi+dphi) , -(Lenght_Tracker)/2,
300 ray*sin(phi) , ray*cos(phi) , -(Lenght_Tracker)/2);
[112]301 Tracker->addDaughter(trackerEndCap1); DetIdCountTracker++;
302
303 FROG_Element_Primitive_CustomSurface* trackerEndCap2 = new FROG_Element_Primitive_CustomSurface(9100000+DetIdCountTracker*10,
[183]304 rayCone*sin(phi) , rayCone*cos(phi) , (Lenght_Tracker)/2,
305 rayCone*sin(phi+dphi) , rayCone*cos(phi+dphi) , (Lenght_Tracker)/2,
306 ray*sin(phi+dphi) , ray*cos(phi+dphi) , (Lenght_Tracker)/2,
307 ray*sin(phi) , ray*cos(phi) , (Lenght_Tracker)/2);
[112]308 Tracker->addDaughter(trackerEndCap2); DetIdCountTracker++;
309
[97]310 }
311
[112]312 Rayon_Calo = ray*1.5;
313
[97]314 //******************************************Central calorimeters******************************************
315 //********************************************************************************************************
316
317 FROG_Element_Base_With_DetId_And_Name* CALO = new FROG_Element_Base_With_DetId_And_Name(920000000,"Calo");
318 detector->addDaughter(CALO);
319 unsigned int DetIdCountCalo = 1;
320
[112]321 float rayConeD = tan(EtaToTheta(DET->CEN_max_calo_cen))*Lenght_Tracker/2;
322 float rayConeF = tan(EtaToTheta(DET->CEN_max_calo_cen))*Lenght_Calo/2;
323 ray=Rayon_Calo;
324 if(ray<rayConeF)ray=rayConeF;
325
[183]326 for(double phi=0; phi<=twopi/frac;phi+=dphi){
[112]327
328 //BARREL
329 FROG_Element_Primitive_CustomSurface* caloB = new FROG_Element_Primitive_CustomSurface(9100000+DetIdCountCalo*10,
[183]330 ray*sin(phi) ,ray*cos(phi) ,-Lenght_Calo*0.5,
331 ray*sin(phi+dphi) ,ray*cos(phi+dphi) ,-Lenght_Calo*0.5,
332 ray*sin(phi+dphi) ,ray*cos(phi+dphi) , Lenght_Calo*0.5,
333 ray*sin(phi) ,ray*cos(phi) , Lenght_Calo*0.5);
[112]334 CALO->addDaughter(caloB); DetIdCountCalo++;
335
336 //Partial cone + side
337 FROG_Element_Primitive_CustomSurface* caloEndCap1a = new FROG_Element_Primitive_CustomSurface(9200000+DetIdCountCalo*10,
[183]338 rayConeD*sin(phi) , rayConeD*cos(phi) , (Lenght_Tracker)/2,
339 rayConeD*sin(phi+dphi) , rayConeD*cos(phi+dphi) , (Lenght_Tracker)/2,
340 rayConeF*sin(phi+dphi) , rayConeF*cos(phi+dphi) , (Lenght_Calo)/2,
341 rayConeF*sin(phi) , rayConeF*cos(phi) , (Lenght_Calo)/2);
[112]342 CALO->addDaughter(caloEndCap1a); DetIdCountCalo++;
343
344 //Close Endcap + side
345 FROG_Element_Primitive_CustomSurface* caloEndCap1b = new FROG_Element_Primitive_CustomSurface(9200000+DetIdCountTracker*10,
[183]346 rayConeF*sin(phi) , rayConeF*cos(phi) , (Lenght_Calo)/2,
347 rayConeF*sin(phi+dphi) , rayConeF*cos(phi+dphi) , (Lenght_Calo)/2,
348 ray*sin(phi+dphi) , ray*cos(phi+dphi) , (Lenght_Calo)/2,
349 ray*sin(phi) , ray*cos(phi) , (Lenght_Calo)/2);
[112]350 CALO->addDaughter(caloEndCap1b); DetIdCountCalo++;
351
352 //Partial cone - side
353 FROG_Element_Primitive_CustomSurface* caloEndCap2a = new FROG_Element_Primitive_CustomSurface(9200000+DetIdCountCalo*10,
[183]354 rayConeD*sin(phi) , rayConeD*cos(phi) , -(Lenght_Tracker)/2,
355 rayConeD*sin(phi+dphi) , rayConeD*cos(phi+dphi) , -(Lenght_Tracker)/2,
356 rayConeF*sin(phi+dphi) , rayConeF*cos(phi+dphi) , -(Lenght_Calo)/2,
357 rayConeF*sin(phi) , rayConeF*cos(phi) , -(Lenght_Calo)/2);
[112]358 CALO->addDaughter(caloEndCap2a); DetIdCountCalo++;
359
360 //Close Endcap - side
361 FROG_Element_Primitive_CustomSurface* caloEndCap2b = new FROG_Element_Primitive_CustomSurface(9200000+DetIdCountTracker*10,
[183]362 rayConeF*sin(phi) , rayConeF*cos(phi) , -(Lenght_Calo)/2,
363 rayConeF*sin(phi+dphi) , rayConeF*cos(phi+dphi) , -(Lenght_Calo)/2,
364 ray*sin(phi+dphi) , ray*cos(phi+dphi) , -(Lenght_Calo)/2,
365 ray*sin(phi) , ray*cos(phi) , -(Lenght_Calo)/2);
[112]366 CALO->addDaughter(caloEndCap2b); DetIdCountCalo++;
367
[97]368 }
[112]369
370 Rayon_Muon = ray*1.2;
371
372 //***********************************************Muon chambers********************************************
373 //********************************************************************************************************
374
375 FROG_Element_Base_With_DetId_And_Name* MUON = new FROG_Element_Base_With_DetId_And_Name(930000000,"Muon");
376 detector->addDaughter(MUON);
377 unsigned int DetIdCountMuon = 1;
378
379 rayConeD = tan(EtaToTheta(DET->CEN_max_mu))*Lenght_Calo/2;
380 rayConeF = tan(EtaToTheta(DET->CEN_max_mu))*Lenght_Muon/2;
381 ray=Rayon_Muon;
382 if(ray<rayConeF)ray=rayConeF;
383
[183]384 for(double phi=0; phi<=twopi/frac;phi+=dphi){
[112]385
386 //BARREL
387 FROG_Element_Primitive_CustomSurface* muonB = new FROG_Element_Primitive_CustomSurface(9300000+DetIdCountMuon*10,
[183]388 ray*sin(phi) ,ray*cos(phi) ,-Lenght_Muon*0.5,
389 ray*sin(phi+dphi) ,ray*cos(phi+dphi) ,-Lenght_Muon*0.5,
390 ray*sin(phi+dphi) ,ray*cos(phi+dphi) , Lenght_Muon*0.5,
391 ray*sin(phi) ,ray*cos(phi) , Lenght_Muon*0.5);
[112]392 MUON->addDaughter(muonB); DetIdCountMuon++;
393
394 //Partial cone + side
395 FROG_Element_Primitive_CustomSurface* muonEndCap1a = new FROG_Element_Primitive_CustomSurface(9300000+DetIdCountMuon*10,
[183]396 rayConeD*sin(phi) , rayConeD*cos(phi) , (Lenght_Calo)/2,
397 rayConeD*sin(phi+dphi) , rayConeD*cos(phi+dphi) , (Lenght_Calo)/2,
398 rayConeF*sin(phi+dphi) , rayConeF*cos(phi+dphi) , (Lenght_Muon)/2,
399 rayConeF*sin(phi) , rayConeF*cos(phi) , (Lenght_Muon)/2);
[112]400 MUON->addDaughter(muonEndCap1a); DetIdCountMuon++;
401
402 //Close Endcap + side
403 FROG_Element_Primitive_CustomSurface* muonEndCap1b = new FROG_Element_Primitive_CustomSurface(9300000+DetIdCountMuon*10,
[183]404 rayConeF*sin(phi) , rayConeF*cos(phi) , (Lenght_Muon)/2,
405 rayConeF*sin(phi+dphi) , rayConeF*cos(phi+dphi) , (Lenght_Muon)/2,
406 ray*sin(phi+dphi) , ray*cos(phi+dphi) , (Lenght_Muon)/2,
407 ray*sin(phi) , ray*cos(phi) , (Lenght_Muon)/2);
[112]408 MUON->addDaughter(muonEndCap1b); DetIdCountMuon++;
409
410 //Partial cone - side
411 FROG_Element_Primitive_CustomSurface* muonEndCap2a = new FROG_Element_Primitive_CustomSurface(9300000+DetIdCountMuon*10,
[183]412 rayConeD*sin(phi) , rayConeD*cos(phi) , -(Lenght_Calo)/2,
413 rayConeD*sin(phi+dphi) , rayConeD*cos(phi+dphi) , -(Lenght_Calo)/2,
414 rayConeF*sin(phi+dphi) , rayConeF*cos(phi+dphi) , -(Lenght_Muon)/2,
415 rayConeF*sin(phi) , rayConeF*cos(phi) , -(Lenght_Muon)/2);
[112]416 MUON->addDaughter(muonEndCap2a); DetIdCountMuon++;
417
418 //Close Endcap - side
419 FROG_Element_Primitive_CustomSurface* muonEndCap2b = new FROG_Element_Primitive_CustomSurface(9300000+DetIdCountMuon*10,
[183]420 rayConeF*sin(phi) , rayConeF*cos(phi) , -(Lenght_Muon)/2,
421 rayConeF*sin(phi+dphi) , rayConeF*cos(phi+dphi) , -(Lenght_Muon)/2,
422 ray*sin(phi+dphi) , ray*cos(phi+dphi) , -(Lenght_Muon)/2,
423 ray*sin(phi) , ray*cos(phi) , -(Lenght_Muon)/2);
[112]424 MUON->addDaughter(muonEndCap2b); DetIdCountMuon++;
[97]425 }
426
[112]427
[97]428 //******************************************Forward calorimeters******************************************
429 //********************************************************************************************************
430
[112]431 FROG_Element_Base_With_DetId_And_Name* CALOFWD = new FROG_Element_Base_With_DetId_And_Name(940000000,"CaloFwd");
432 detector->addDaughter(CALOFWD);
433 unsigned int DetIdCountCaloFwd = 1;
[97]434
[112]435 rayConeD = tan(EtaToTheta(DET->CEN_max_calo_fwd))*Lenght_Muon;
436 rayConeF = tan(EtaToTheta(DET->CEN_max_calo_fwd))*Lenght_CaloFwd;
437 ray=tan(EtaToTheta(DET->CEN_max_calo_cen))*Lenght_CaloFwd;
438
[183]439 for(double phi=0; phi<=twopi/frac;phi+=dphi){
[97]440
[112]441 //Partial cone + side
442 FROG_Element_Primitive_CustomSurface* caloFWDEndCap1a = new FROG_Element_Primitive_CustomSurface(9400000+DetIdCountCaloFwd*10,
[183]443 rayConeD*sin(phi) , rayConeD*cos(phi) , (Lenght_Muon)/2,
444 rayConeD*sin(phi+dphi) , rayConeD*cos(phi+dphi) , (Lenght_Muon)/2,
445 rayConeF*sin(phi+dphi) , rayConeF*cos(phi+dphi) , (Lenght_CaloFwd)/2,
446 rayConeF*sin(phi) , rayConeF*cos(phi) , (Lenght_CaloFwd)/2);
[112]447 CALOFWD->addDaughter(caloFWDEndCap1a); DetIdCountCaloFwd++;
448
449 //Close Endcap + side
450 FROG_Element_Primitive_CustomSurface* caloFWDEndCap1b = new FROG_Element_Primitive_CustomSurface(9400000+DetIdCountCaloFwd*10,
[183]451 rayConeF*sin(phi) , rayConeF*cos(phi) , (Lenght_CaloFwd)/2,
452 rayConeF*sin(phi+dphi) , rayConeF*cos(phi+dphi) , (Lenght_CaloFwd)/2,
453 ray*sin(phi+dphi) , ray*cos(phi+dphi) , (Lenght_CaloFwd)/2,
454 ray*sin(phi) , ray*cos(phi) , (Lenght_CaloFwd)/2);
[112]455 CALOFWD->addDaughter(caloFWDEndCap1b); DetIdCountCaloFwd++;
456
457 //BARREL
458 FROG_Element_Primitive_CustomSurface* caloFWDB1 = new FROG_Element_Primitive_CustomSurface(9400000+DetIdCountMuon*10,
[183]459 ray*sin(phi) ,ray*cos(phi) , Lenght_Muon*0.5,
460 ray*sin(phi+dphi) ,ray*cos(phi+dphi) , Lenght_Muon*0.5,
461 ray*sin(phi+dphi) ,ray*cos(phi+dphi) , Lenght_CaloFwd*0.5,
462 ray*sin(phi) ,ray*cos(phi) , Lenght_CaloFwd*0.5);
[112]463 CALOFWD->addDaughter(caloFWDB1); DetIdCountCaloFwd++;
464
465 //Partial cone - side
466 FROG_Element_Primitive_CustomSurface* caloFWDEndCap2a = new FROG_Element_Primitive_CustomSurface(9400000+DetIdCountCaloFwd*10,
[183]467 rayConeD*sin(phi) , rayConeD*cos(phi) , -(Lenght_Muon)/2,
468 rayConeD*sin(phi+dphi) , rayConeD*cos(phi+dphi) , -(Lenght_Muon)/2,
469 rayConeF*sin(phi+dphi) , rayConeF*cos(phi+dphi) , -(Lenght_CaloFwd)/2,
470 rayConeF*sin(phi) , rayConeF*cos(phi) , -(Lenght_CaloFwd)/2);
[112]471 CALOFWD->addDaughter(caloFWDEndCap2a); DetIdCountCaloFwd++;
472
473 //Close Endcap - side
474 FROG_Element_Primitive_CustomSurface* caloFWDEndCap2b = new FROG_Element_Primitive_CustomSurface(9400000+DetIdCountCaloFwd*10,
[183]475 rayConeF*sin(phi) , rayConeF*cos(phi) , -(Lenght_CaloFwd)/2,
476 rayConeF*sin(phi+dphi) , rayConeF*cos(phi+dphi) , -(Lenght_CaloFwd)/2,
477 ray*sin(phi+dphi) , ray*cos(phi+dphi) , -(Lenght_CaloFwd)/2,
478 ray*sin(phi) , ray*cos(phi) , -(Lenght_CaloFwd)/2);
[112]479 CALOFWD->addDaughter(caloFWDEndCap2b); DetIdCountCaloFwd++;
480
481 //BARREL
482 FROG_Element_Primitive_CustomSurface* caloFWDB2 = new FROG_Element_Primitive_CustomSurface(9400000+DetIdCountMuon*10,
[183]483 ray*sin(phi) ,ray*cos(phi) , -Lenght_Muon*0.5,
484 ray*sin(phi+dphi) ,ray*cos(phi+dphi) , -Lenght_Muon*0.5,
485 ray*sin(phi+dphi) ,ray*cos(phi+dphi) , -Lenght_CaloFwd*0.5,
486 ray*sin(phi) ,ray*cos(phi) , -Lenght_CaloFwd*0.5);
[112]487 CALOFWD->addDaughter(caloFWDB2); DetIdCountCaloFwd++;
[100]488 }
[219]489
[97]490 FROG_Geometry* CustomGeom = new FROG_Geometry(prim);
491 CustomGeom->Save("DelphesToFrog.geom");
[219]492
493 delete prim;
494 delete CustomGeom;
495
496/* pointer hierarchy
497prim
498 -> mygeom
499 -> detector
500 -> Tracker
501 -> CALO
502 -> MUON
503 -> CALOFWD
504pointers that have been added into others will be properly deleted by the destructor of their "mother"
505*/
506
507
[171]508 //FROG_ELEMENT::PrintTree(prim);
[97]509 return;
510
511}
Note: See TracBrowser for help on using the repository browser.