[286] | 1 | // |
---|
| 2 | // This script shows how to read, with ROOT, a ELYSE.root |
---|
| 3 | // file having the "TTree within TTree" of the ELYSE_batch |
---|
| 4 | // progrem (see Analysis.cxx). |
---|
| 5 | // |
---|
| 6 | // Usage : |
---|
| 7 | // OS> <setup ROOT> |
---|
| 8 | // ( OS> root batch_dummy.C ) |
---|
| 9 | // OS> root analyis.C |
---|
| 10 | // |
---|
| 11 | // G.Barrand |
---|
| 12 | // |
---|
| 13 | |
---|
| 14 | ////////////////////////////////////////////////////////////////////////////// |
---|
| 15 | bool plot( |
---|
| 16 | TH1D& aHisto1D |
---|
| 17 | ,TH2D& aHisto2D |
---|
| 18 | ) |
---|
| 19 | ////////////////////////////////////////////////////////////////////////////// |
---|
| 20 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
---|
| 21 | { |
---|
| 22 | TCanvas* plotter = new TCanvas("canvas","",10,10,800,600); |
---|
| 23 | plotter->Divide(1,2); |
---|
| 24 | |
---|
| 25 | plotter->cd(1); |
---|
| 26 | aHisto1D.Draw(); |
---|
| 27 | |
---|
| 28 | plotter->cd(2); |
---|
| 29 | aHisto2D.Draw(); |
---|
| 30 | |
---|
| 31 | plotter->Update(); |
---|
| 32 | |
---|
| 33 | return true; |
---|
| 34 | } |
---|
| 35 | ////////////////////////////////////////////////////////////////////////////// |
---|
| 36 | void analysis() |
---|
| 37 | ////////////////////////////////////////////////////////////////////////////// |
---|
| 38 | //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!// |
---|
| 39 | { |
---|
| 40 | |
---|
| 41 | //////////////////////////////////////////////////////// |
---|
| 42 | // Create histograms /////////////////////////////////// |
---|
| 43 | //////////////////////////////////////////////////////// |
---|
| 44 | TH1D* hits_times = new TH1D("hits_times","Hits times",100,0,3000); |
---|
| 45 | |
---|
| 46 | TH2D* digits_time_pe = |
---|
| 47 | new TH2D("digits_pe_time","Digits PE time",100,0,3000,100,0,10); |
---|
| 48 | |
---|
| 49 | int hit_count = 0; |
---|
| 50 | bool dump = true; |
---|
| 51 | //bool dump = false; |
---|
| 52 | |
---|
| 53 | //////////////////////////////////////////////////////// |
---|
| 54 | /// Read data an file histos /////////////////////////// |
---|
| 55 | //////////////////////////////////////////////////////// |
---|
| 56 | TFile* f = new TFile("ELYSE.root"); |
---|
| 57 | TTree* tEvent = (TTree*)f->Get("Event"); |
---|
| 58 | |
---|
| 59 | {TObjArray* brs = tEvent->GetListOfBranches(); |
---|
| 60 | for(int i=0;i<brs->GetEntries();i++) { |
---|
| 61 | TBranch* b = (TBranch*)brs->At(i); |
---|
| 62 | std::cout << " Event branch : " << b->GetName() |
---|
| 63 | << std::endl; |
---|
| 64 | }} |
---|
| 65 | |
---|
| 66 | //declare Non subTuple part |
---|
| 67 | Int_t eventId,inputEvtId, interMode, vtxVol; |
---|
| 68 | tEvent->SetBranchAddress("eventId",&eventId); |
---|
| 69 | tEvent->SetBranchAddress("inputEvtId",&inputEvtId); |
---|
| 70 | tEvent->SetBranchAddress("interMode",&interMode); |
---|
| 71 | tEvent->SetBranchAddress("vtxVol",&vtxVol); |
---|
| 72 | |
---|
| 73 | Int_t nPart,leptonIndex, protonIndex; |
---|
| 74 | tEvent->SetBranchAddress("nPart",&nPart); |
---|
| 75 | tEvent->SetBranchAddress("leptonIndex",&leptonIndex); |
---|
| 76 | tEvent->SetBranchAddress("protonIndex",&protonIndex); |
---|
| 77 | |
---|
| 78 | Int_t nHits; |
---|
| 79 | tEvent->SetBranchAddress("nHits",&nHits); |
---|
| 80 | |
---|
| 81 | Int_t nDigits; |
---|
| 82 | Double_t sumPE; |
---|
| 83 | tEvent->SetBranchAddress("nDigits",&nDigits); |
---|
| 84 | tEvent->SetBranchAddress("sumPE",&sumPE); |
---|
| 85 | |
---|
| 86 | Int_t nEvent = (Int_t)tEvent->GetEntries(); |
---|
| 87 | //nEvent = 1; |
---|
| 88 | //nEvent = 10000; |
---|
| 89 | std::cout << " nEvents = " << nEvent << std::endl; |
---|
| 90 | |
---|
| 91 | for (Int_t i=0; i<nEvent; ++i) { |
---|
| 92 | |
---|
| 93 | TTree* Event_vtxPos = new TTree(); |
---|
| 94 | tEvent->SetBranchAddress("vtxPos",&Event_vtxPos); |
---|
| 95 | |
---|
| 96 | TTree* Event_track = new TTree(); |
---|
| 97 | tEvent->SetBranchAddress("track",&Event_track); |
---|
| 98 | |
---|
| 99 | TTree* Event_hit = new TTree(); |
---|
| 100 | tEvent->SetBranchAddress("hit",&Event_hit); |
---|
| 101 | |
---|
| 102 | TTree* Event_digit = new TTree(); |
---|
| 103 | tEvent->SetBranchAddress("digit",&Event_digit); |
---|
| 104 | |
---|
| 105 | tEvent->GetEntry(i); |
---|
| 106 | |
---|
| 107 | //{TObjArray* brs = Event_track->GetListOfBranches(); |
---|
| 108 | //for(int i=0;i<brs->GetEntries();i++) { |
---|
| 109 | // TBranch* b = (TBranch*)brs->At(i); |
---|
| 110 | // std::cout << " Event_track branch : " << b->GetName() << std::endl; |
---|
| 111 | // }} |
---|
| 112 | |
---|
| 113 | // Bind sub tuple variables |
---|
| 114 | Double_t vx,vy,vz; |
---|
| 115 | Event_vtxPos->SetBranchAddress("x",&vx); |
---|
| 116 | Event_vtxPos->SetBranchAddress("y",&vy); |
---|
| 117 | Event_vtxPos->SetBranchAddress("z",&vz); |
---|
| 118 | |
---|
| 119 | Int_t pId,parent; |
---|
| 120 | Float_t timeStart; |
---|
| 121 | Double_t mass, pTot, ETot; |
---|
| 122 | Int_t startVol, stopVol; |
---|
| 123 | Event_track->SetBranchAddress("pId",&pId); |
---|
| 124 | Event_track->SetBranchAddress("parent",&parent); |
---|
| 125 | Event_track->SetBranchAddress("timeStart",&timeStart); |
---|
| 126 | Event_track->SetBranchAddress("mass",&mass); |
---|
| 127 | Event_track->SetBranchAddress("pTot",&pTot); |
---|
| 128 | Event_track->SetBranchAddress("ETot",&ETot); |
---|
| 129 | Event_track->SetBranchAddress("startVol",&startVol); |
---|
| 130 | Event_track->SetBranchAddress("stopVol",&stopVol); |
---|
| 131 | |
---|
| 132 | Int_t tubeId_hit; //JEC 16/1/06 |
---|
| 133 | Int_t totalPE; |
---|
| 134 | Event_hit->SetBranchAddress("tubeId",&tubeId_hit); //JEC 16/1/06 |
---|
| 135 | Event_hit->SetBranchAddress("totalPE",&totalPE); |
---|
| 136 | |
---|
| 137 | Int_t tubeId; |
---|
| 138 | Double_t digit_pe, digit_time; |
---|
| 139 | Event_digit->SetBranchAddress("tubeId",&tubeId); |
---|
| 140 | Event_digit->SetBranchAddress("pe",&digit_pe); |
---|
| 141 | Event_digit->SetBranchAddress("time",&digit_time); |
---|
| 142 | |
---|
| 143 | if(dump) |
---|
| 144 | std::cout << ">>>>>>>>>>>>> Event{" << i << "}: " |
---|
| 145 | << " evt Id " << eventId |
---|
| 146 | // << " evt Input Id " << inputEvtId |
---|
| 147 | // << "\n interaction mode " << interMode |
---|
| 148 | // << " start in volume " << vtxVol << "\n" |
---|
| 149 | <<" #tracks: " << nPart |
---|
| 150 | <<" #hits: " << nHits |
---|
| 151 | <<" #digits: " << nDigits |
---|
| 152 | << std::endl; |
---|
| 153 | |
---|
| 154 | Int_t nVtx = (Int_t)Event_vtxPos->GetEntries(); |
---|
| 155 | Int_t nTracks = (Int_t)Event_track->GetEntries(); |
---|
| 156 | Int_t nTubeHits = (Int_t)Event_hit->GetEntries(); |
---|
| 157 | Int_t nTubeDigits = (Int_t)Event_digit->GetEntries(); |
---|
| 158 | |
---|
| 159 | if(dump) |
---|
| 160 | std::cout << "Verif :" |
---|
| 161 | << " nVtx = " << nVtx |
---|
| 162 | << " nTracks = " << nTracks |
---|
| 163 | << " nTube Hits = " << nTubeHits |
---|
| 164 | << " nTube Digits = " << nTubeDigits |
---|
| 165 | << std::endl; |
---|
| 166 | |
---|
| 167 | for (Int_t j=0; j<nTracks; ++j) { |
---|
| 168 | |
---|
| 169 | //Sub tuples of Event_tracks : |
---|
| 170 | TTree* Track_direction = new TTree(); |
---|
| 171 | Event_track->SetBranchAddress("direction",&Track_direction); |
---|
| 172 | TTree* Track_momentum = new TTree(); |
---|
| 173 | Event_track->SetBranchAddress("momentum",&Track_momentum); |
---|
| 174 | TTree* Track_startPos = new TTree(); |
---|
| 175 | Event_track->SetBranchAddress("startPos",&Track_startPos); |
---|
| 176 | TTree* Track_stopPos = new TTree(); |
---|
| 177 | Event_track->SetBranchAddress("stopPos",&Track_stopPos); |
---|
| 178 | |
---|
| 179 | Event_track->GetEntry(j); |
---|
| 180 | |
---|
| 181 | Double_t dx,dy,dz; |
---|
| 182 | Track_direction->SetBranchAddress("dx",&dx); |
---|
| 183 | Track_direction->SetBranchAddress("dy",&dy); |
---|
| 184 | Track_direction->SetBranchAddress("dz",&dz); |
---|
| 185 | |
---|
| 186 | Double_t px,py,pz; |
---|
| 187 | Track_momentum->SetBranchAddress("px",&px); |
---|
| 188 | Track_momentum->SetBranchAddress("py",&py); |
---|
| 189 | Track_momentum->SetBranchAddress("pz",&pz); |
---|
| 190 | |
---|
| 191 | Double_t start_x,start_y,start_z; |
---|
| 192 | Track_startPos->SetBranchAddress("x",&start_x); |
---|
| 193 | Track_startPos->SetBranchAddress("y",&start_y); |
---|
| 194 | Track_startPos->SetBranchAddress("z",&start_z); |
---|
| 195 | |
---|
| 196 | Double_t stop_x,stop_y,stop_z; |
---|
| 197 | Track_stopPos->SetBranchAddress("x",&stop_x); |
---|
| 198 | Track_stopPos->SetBranchAddress("y",&stop_y); |
---|
| 199 | Track_stopPos->SetBranchAddress("z",&stop_z); |
---|
| 200 | |
---|
| 201 | // One entry only : |
---|
| 202 | //std::cout << "debug : " |
---|
| 203 | // << " " << Track_direction->GetEntries() |
---|
| 204 | // << " " << Track_momentum->GetEntries() |
---|
| 205 | // << " " << Track_startPos->GetEntries() |
---|
| 206 | // << " " << Track_stopPos->GetEntries() |
---|
| 207 | // << std::endl; |
---|
| 208 | if(Track_direction->GetEntries()==1) Track_direction->GetEntry(0); |
---|
| 209 | if(Track_momentum->GetEntries()==1) Track_momentum->GetEntry(0); |
---|
| 210 | if(Track_startPos->GetEntries()==1) Track_startPos->GetEntry(0); |
---|
| 211 | if(Track_stopPos->GetEntries()==1) Track_stopPos->GetEntry(0); |
---|
| 212 | |
---|
| 213 | if(dump) |
---|
| 214 | std::cout << "----> Tk{"<<j<<"}: " |
---|
| 215 | << " pId " << pId |
---|
| 216 | << " parent " << parent |
---|
| 217 | << " creation time " << timeStart |
---|
| 218 | << " Volumes " << startVol << " " << stopVol << "\n" |
---|
| 219 | << " Start Pos (" << start_x << "," << start_y << "," << start_z << ")\n" |
---|
| 220 | << " Stop Pos (" << stop_x << "," << stop_y << "," << stop_z << ")\n" |
---|
| 221 | << " dx,dy,dz " << dx << " " << dy << " " << dz << "\n" |
---|
| 222 | << " m " << mass |
---|
| 223 | << " ETot " << ETot |
---|
| 224 | << " pTot " << pTot |
---|
| 225 | << " px,py,pz " << px << " " << py << " " << pz << "\n" |
---|
| 226 | << std::endl; |
---|
| 227 | |
---|
| 228 | delete Track_direction; |
---|
| 229 | delete Track_momentum; |
---|
| 230 | delete Track_startPos; |
---|
| 231 | delete Track_stopPos; |
---|
| 232 | |
---|
| 233 | }//loop on Tracks |
---|
| 234 | |
---|
| 235 | //-------- |
---|
| 236 | // The Hits |
---|
| 237 | //-------- |
---|
| 238 | |
---|
| 239 | |
---|
| 240 | for (Int_t k=0; k<nTubeHits; ++k) { |
---|
| 241 | |
---|
| 242 | TTree* Hit_pe = new TTree(); |
---|
| 243 | Event_hit->SetBranchAddress("pe",&Hit_pe); |
---|
| 244 | |
---|
| 245 | Event_hit->GetEntry(k); |
---|
| 246 | |
---|
| 247 | Float_t hit_time; |
---|
| 248 | //Float_t trk_length; //NV 13/6/06 |
---|
| 249 | Hit_pe->SetBranchAddress("time",&hit_time); |
---|
| 250 | //Hit_pe->SetBranchAddress("length",&trk_length); |
---|
| 251 | |
---|
| 252 | //JEC 16/1/06 add the tubeId_hit info |
---|
| 253 | if(dump) |
---|
| 254 | std::cout << "----> Hit{"<<k<<"}: tube[" << tubeId_hit << "] total #PE " << totalPE << std::endl; |
---|
| 255 | |
---|
| 256 | for (Int_t ki=0; ki<Hit_pe->GetEntries(); ++ki) { |
---|
| 257 | Hit_pe->GetEntry(ki); |
---|
| 258 | |
---|
| 259 | if(dump) |
---|
| 260 | std::cout << "<" << hit_time << ">" |
---|
| 261 | /* << "<" << trk_length << ">" */ ; |
---|
| 262 | |
---|
| 263 | hits_times->Fill(hit_time); |
---|
| 264 | |
---|
| 265 | hit_count++; |
---|
| 266 | } |
---|
| 267 | if(dump) |
---|
| 268 | std::cout << std::endl; |
---|
| 269 | |
---|
| 270 | delete Hit_pe; |
---|
| 271 | }//Loop on Hits |
---|
| 272 | |
---|
| 273 | //-------- |
---|
| 274 | // The Digits |
---|
| 275 | //-------- |
---|
| 276 | for (Int_t l=0; l<nTubeDigits; ++l) { |
---|
| 277 | Event_digit->GetEntry(l); |
---|
| 278 | |
---|
| 279 | if(dump) |
---|
| 280 | std::cout << "----> Digit{"<<l<<"}: " |
---|
| 281 | << "tube[" << tubeId << "] = " |
---|
| 282 | << " pe: " << digit_pe |
---|
| 283 | << " time: " << digit_time |
---|
| 284 | << std::endl; |
---|
| 285 | |
---|
| 286 | digits_time_pe->Fill(digit_time,digit_pe); |
---|
| 287 | |
---|
| 288 | }//Loop on Digits |
---|
| 289 | |
---|
| 290 | delete Event_vtxPos; |
---|
| 291 | delete Event_track; |
---|
| 292 | delete Event_hit; |
---|
| 293 | delete Event_digit; |
---|
| 294 | |
---|
| 295 | }//loop on event |
---|
| 296 | |
---|
| 297 | if(dump) |
---|
| 298 | std::cout << " nEvents = " << nEvent << " hits = " << hit_count << std::endl; |
---|
| 299 | |
---|
| 300 | //////////////////////////////////////////////////////// |
---|
| 301 | // Plot histograms ///////////////////////////////////// |
---|
| 302 | //////////////////////////////////////////////////////// |
---|
| 303 | plot(*hits_times,*digits_time_pe); |
---|
| 304 | |
---|
| 305 | }//read_event |
---|