| 1 |
//////////////////////////////////////////////////////////////// |
| 2 |
// |
| 3 |
// Histogram set for tracker-efficiency study. |
| 4 |
// |
| 5 |
// It contains the functions: |
| 6 |
// |
| 7 |
// - CreateHistos() |
| 8 |
// - RetrieveHistos() |
| 9 |
// - FillHistos(PamLevel2* event) |
| 10 |
// - Save Histos() |
| 11 |
// |
| 12 |
// Usage (example): |
| 13 |
// [] .L My-histo-set-macro.C+ |
| 14 |
// [] .L My-selection.C+ |
| 15 |
// [] .L Loop.C+ |
| 16 |
// [] Loop("data-dir/","file-list.txt",100000,"+ALL fillTree fillHistos","output-file.root") |
| 17 |
// |
| 18 |
//////////////////////////////////////////////////////////////// |
| 19 |
|
| 20 |
#if !defined(__CINT__) || defined(__MAKECINT__) |
| 21 |
|
| 22 |
#include <TString.h> |
| 23 |
#include <TH1F.h> |
| 24 |
#include <TH2F.h> |
| 25 |
#include <TMath.h> |
| 26 |
#include <TLine.h> |
| 27 |
#include <TPolyMarker.h> |
| 28 |
#include <TSelector.h> |
| 29 |
#include <TFile.h> |
| 30 |
|
| 31 |
#include <stdlib.h> |
| 32 |
#include <iostream> |
| 33 |
using namespace std; |
| 34 |
|
| 35 |
#include <PamLevel2.h> |
| 36 |
#include <CaloAxis.h> |
| 37 |
|
| 38 |
#endif |
| 39 |
//====================== |
| 40 |
// HITOGRAMS declaration |
| 41 |
//====================== |
| 42 |
|
| 43 |
TH2F *hchi2def; |
| 44 |
TH2F *hchi2rig; |
| 45 |
TH2F *hdedxtofvsdedxtrk; |
| 46 |
TH2F *hdedxtrkvsrig ; |
| 47 |
TH2F *hdedxtofvsrig ; |
| 48 |
TH2F *hbetavsrig ; |
| 49 |
TH2F *hdedxtrkvsdef ; |
| 50 |
TH2F *hbetavsdef ; |
| 51 |
TH2F *hbetavsdef2 ; |
| 52 |
|
| 53 |
TH1F *herrdef; |
| 54 |
TH1F *herrx[6]; |
| 55 |
TH1F *herry[6]; |
| 56 |
|
| 57 |
TH1F *hdef ; |
| 58 |
TH1F *hrig1 ; |
| 59 |
TH1F *hrig2 ; |
| 60 |
TH1F *hen1 ; |
| 61 |
TH1F *hen2 ; |
| 62 |
|
| 63 |
|
| 64 |
Int_t tot=0; |
| 65 |
Int_t sel[]={0,0,0,0,0,0,0,0}; |
| 66 |
|
| 67 |
//=============================================================== |
| 68 |
// Get ToF dE/dx |
| 69 |
//=============================================================== |
| 70 |
// Float_t GetdEdx( PamTrack *trk ){ |
| 71 |
// Float_t dedx = 0.; |
| 72 |
// Int_t nval=0; |
| 73 |
// Bool_t ISNULL = false; |
| 74 |
// for (Int_t i=0; i<trk->GetToFTrack()->npmtadc; i++){ |
| 75 |
// if(trk->GetToFTrack()->dedx[i] == 0)ISNULL=true; |
| 76 |
// dedx += (trk->GetToFTrack()->dedx).At(i)*(Int_t)(!ISNULL); |
| 77 |
// nval += (Int_t)(!ISNULL); |
| 78 |
// }; |
| 79 |
// // |
| 80 |
// if(nval)dedx=dedx/nval; |
| 81 |
// // cout <<" -- dedx "<<dedx<<endl; |
| 82 |
// return(dedx); |
| 83 |
// } |
| 84 |
|
| 85 |
// float GetZ(float beta, float dedx){ |
| 86 |
|
| 87 |
// int nspec=2; |
| 88 |
// float p0[]={0.904,3.373}; |
| 89 |
// float p1[]={0.295,2.756}; |
| 90 |
// float zz[]={1.,4.}; |
| 91 |
|
| 92 |
// if(beta>1)beta=1; |
| 93 |
// float f0 = p0[0]/beta+p1[0]; |
| 94 |
// float f1 = p0[1]/beta+p1[1]; |
| 95 |
// float z0 = zz[0]; |
| 96 |
// float z1 = zz[1]; |
| 97 |
// float aa = (z0-z1)/(f0-f1); |
| 98 |
// float bb = -aa*f0+z0; |
| 99 |
// if( (bb+aa*dedx)==0 )return 0; |
| 100 |
// return sqrt(bb+aa*dedx); |
| 101 |
|
| 102 |
// } |
| 103 |
|
| 104 |
|
| 105 |
//=============================================================== |
| 106 |
// Create histograms |
| 107 |
//=============================================================== |
| 108 |
void CreateHistos( PamLevel2* event, TFile* outf ){ |
| 109 |
|
| 110 |
gROOT->cd();//create histos in memory |
| 111 |
|
| 112 |
TString hid; |
| 113 |
TString title; |
| 114 |
//=========================================== |
| 115 |
// rigidity binning |
| 116 |
Int_t nbins = 200; |
| 117 |
Float_t rmin = .1; |
| 118 |
Float_t rmax = 1000.; |
| 119 |
Float_t dr = (log10(rmax)-log10(rmin))/nbins; |
| 120 |
TArrayD* arr = new TArrayD(nbins+1); |
| 121 |
arr->AddAt((Double_t)rmin,0); |
| 122 |
for(Int_t i=1; i<=nbins; i++){ |
| 123 |
Double_t previous = arr->At(i-1); |
| 124 |
Double_t value=0; |
| 125 |
value = (Double_t) pow(10,log10((Float_t)previous)+dr); |
| 126 |
arr->AddAt(value,i); |
| 127 |
}; |
| 128 |
Double_t *rbinpt = arr->GetArray(); |
| 129 |
//=========================================== |
| 130 |
|
| 131 |
hchi2def = new TH2F("hchi2def","(chi**2)**0.25 vs deflection",100,0.,2.5,200,0.,10.); |
| 132 |
hchi2rig = new TH2F("hchi2rig","#chi^{2} vs R",nbins,rbinpt,1000,0.,500.); |
| 133 |
hdedxtofvsdedxtrk = new TH2F("hdedxtofvsdedxtrk","dE/dx (ToF-average) vs dE/dx (trk-average)",600,0.,40.,600,0.,40.); |
| 134 |
hdedxtrkvsrig = new TH2F("hdedxtrkvsrig","dE/dx (Trk-average) vs R",nbins,rbinpt,600,0.,40.); |
| 135 |
hdedxtofvsrig = new TH2F("hdedxtofvsrig","dE/dx (ToF-average) vs R",nbins,rbinpt,600,0.,40.); |
| 136 |
hbetavsrig = new TH2F("hbetavsrig","beta vs R",nbins,rbinpt,200,0.4,1.5); |
| 137 |
hrig1 = new TH1F("hrig1","Rigidity distribution Z=1",nbins,rbinpt); |
| 138 |
hrig2 = new TH1F("hrig2","Rigidity distribution Z=2",nbins,rbinpt); |
| 139 |
hen1 = new TH1F("hen1","Kin.Energy per nucleon Z=1",nbins,rbinpt); |
| 140 |
hen2 = new TH1F("hen2","Kin.Energy per nucleon Z=2",nbins,rbinpt); |
| 141 |
|
| 142 |
hdedxtrkvsdef = new TH2F("hdedxtrkvsdef","dE/dx (Trk-average) vs deflection",500,-10.,10.,600,0.,40.); |
| 143 |
hbetavsdef = new TH2F("hbetavsdef","beta vs deflection",500,-10.,10.,200,0.4,1.5); |
| 144 |
hbetavsdef2 = new TH2F("hbetavsdef2","beta vs deflection (Z=1)",500,-10.,10.,200,0.4,1.5); |
| 145 |
hdef = new TH1F("hdef","Deflection distribution",5000,-2.5,2.5); |
| 146 |
|
| 147 |
herrdef = new TH1F("herrdef","Deflection error",1000,0.,0.01); |
| 148 |
|
| 149 |
for(Int_t ip=0; ip<6; ip++){ |
| 150 |
hid.Form("herrx%i",ip); |
| 151 |
title = ""; title.Form("Spatial resolution - plane %i (x)",ip); |
| 152 |
herrx[ip] = new TH1F(hid.Data(),title.Data(),500,0.,0.005); |
| 153 |
hid.Form("herry%i",ip); |
| 154 |
title = ""; title.Form("Spatial resolution - plane %i (y)",ip); |
| 155 |
herry[ip] = new TH1F(hid.Data(),title.Data(),500,0.,0.005); |
| 156 |
} |
| 157 |
|
| 158 |
} |
| 159 |
//=============================================================== |
| 160 |
// Retrieve histograms |
| 161 |
//=============================================================== |
| 162 |
void RetrieveHistos(){ |
| 163 |
|
| 164 |
TString hid; |
| 165 |
|
| 166 |
hchi2def = dynamic_cast<TH2F*>(gDirectory->FindObject("hchi2def")); |
| 167 |
hchi2rig = dynamic_cast<TH2F*>(gDirectory->FindObject("hchi2rig")); |
| 168 |
hdedxtofvsdedxtrk = dynamic_cast<TH2F*>(gDirectory->FindObject("hdedxtofvsdedxtrk")); |
| 169 |
hdedxtrkvsrig = dynamic_cast<TH2F*>(gDirectory->FindObject("hdedxtrkvsrig")); |
| 170 |
hdedxtofvsrig = dynamic_cast<TH2F*>(gDirectory->FindObject("hdedxtofvsrig")); |
| 171 |
hbetavsrig = dynamic_cast<TH2F*>(gDirectory->FindObject("hbetavsrig")); |
| 172 |
hdedxtrkvsdef = dynamic_cast<TH2F*>(gDirectory->FindObject("hdedxtrkvsdef")); |
| 173 |
hbetavsdef = dynamic_cast<TH2F*>(gDirectory->FindObject("hbetavsdef")); |
| 174 |
hbetavsdef2 = dynamic_cast<TH2F*>(gDirectory->FindObject("hbetavsdef2")); |
| 175 |
hdef = dynamic_cast<TH1F*>(gDirectory->FindObject("hdef")); |
| 176 |
hrig1 = dynamic_cast<TH1F*>(gDirectory->FindObject("hrig1")); |
| 177 |
hrig2 = dynamic_cast<TH1F*>(gDirectory->FindObject("hrig2")); |
| 178 |
hen1 = dynamic_cast<TH1F*>(gDirectory->FindObject("hen1")); |
| 179 |
hen2 = dynamic_cast<TH1F*>(gDirectory->FindObject("hen2")); |
| 180 |
herrdef = dynamic_cast<TH1F*>(gDirectory->FindObject("herrdef")); |
| 181 |
for(Int_t ip=0; ip<6; ip++){ |
| 182 |
hid.Form("herrx%i",ip); |
| 183 |
herrx[ip] = dynamic_cast<TH1F*>(gDirectory->FindObject(hid.Data())); |
| 184 |
hid.Form("herry%i",ip); |
| 185 |
herry[ip] = dynamic_cast<TH1F*>(gDirectory->FindObject(hid.Data())); |
| 186 |
} |
| 187 |
} |
| 188 |
//=============================================================== |
| 189 |
// Fill histograms (called event-by-event) |
| 190 |
//=============================================================== |
| 191 |
|
| 192 |
void FillHistos( PamLevel2* event ){ |
| 193 |
// ---------------------------------- |
| 194 |
// retrieve histos |
| 195 |
// ---------------------------------- |
| 196 |
RetrieveHistos(); |
| 197 |
|
| 198 |
// ---------------------------------- |
| 199 |
// single-track events |
| 200 |
// ---------------------------------- |
| 201 |
float beta = 0; |
| 202 |
float dedxtof = 0; |
| 203 |
float dedxtrk = 0; |
| 204 |
float rigidity = 0; |
| 205 |
float deflection = 0; |
| 206 |
float chi2 = 0; |
| 207 |
float errdef = 0.; |
| 208 |
float errx[] = {10.,10.,10.,10.,10.,10.}; |
| 209 |
float erry[] = {10.,10.,10.,10.,10.,10.}; |
| 210 |
|
| 211 |
tot++; |
| 212 |
|
| 213 |
if( event->GetTrkLevel2()->GetNTracks()==0 ) return; |
| 214 |
PamTrack *track = event->GetTrack(0); |
| 215 |
//------------------------------ |
| 216 |
// track selection |
| 217 |
//------------------------------ |
| 218 |
if( |
| 219 |
true |
| 220 |
){ |
| 221 |
|
| 222 |
sel[0]++; |
| 223 |
//------------------- |
| 224 |
// Calo variables |
| 225 |
//------------------- |
| 226 |
|
| 227 |
|
| 228 |
//------------------- |
| 229 |
// ToF variables |
| 230 |
//------------------- |
| 231 |
beta = track->GetToFTrack()->beta[12]; |
| 232 |
// dedxtof = GetdEdx(track); |
| 233 |
|
| 234 |
//------------------- |
| 235 |
// tracker variables |
| 236 |
//------------------- |
| 237 |
dedxtrk = track->GetTrkTrack()->GetDEDX()/1.3; //<<<ATTENZIONE TEMPORANEO!!!! |
| 238 |
rigidity = track->GetTrkTrack()->GetRigidity(); |
| 239 |
deflection = track->GetTrkTrack()->GetDeflection(); |
| 240 |
chi2 = track->GetTrkTrack()->chi2; |
| 241 |
errdef = sqrt(track->GetTrkTrack()->coval[4][4]); |
| 242 |
for(Int_t ip=0; ip<6; ip++){ |
| 243 |
if(track->GetTrkTrack()->XGood(ip))errx[ip]=track->GetTrkTrack()->resx[ip]; |
| 244 |
if(track->GetTrkTrack()->YGood(ip))erry[ip]=track->GetTrkTrack()->resy[ip]; |
| 245 |
} |
| 246 |
|
| 247 |
//------------------- |
| 248 |
// chi2 selection |
| 249 |
//------------------- |
| 250 |
hchi2def->Fill( fabs(deflection), pow((double)chi2,0.25) ); |
| 251 |
hchi2rig->Fill( rigidity, chi2 ); |
| 252 |
if(pow((double)chi2,0.25) > (2.5+1.85*fabs(deflection)) )return; |
| 253 |
sel[1]++; |
| 254 |
|
| 255 |
// hdedxtofvsdedxtrk->Fill(dedxtrk,dedxtof); |
| 256 |
hdedxtrkvsrig->Fill(rigidity,dedxtrk); |
| 257 |
// hdedxtofvsrig->Fill(rigidity,dedxtof); |
| 258 |
hbetavsrig->Fill(rigidity,beta); |
| 259 |
|
| 260 |
hdedxtrkvsdef->Fill(deflection,dedxtrk); |
| 261 |
hbetavsdef->Fill(deflection,beta); |
| 262 |
|
| 263 |
//------------------------ |
| 264 |
// rough charge selection |
| 265 |
//------------------------ |
| 266 |
if(dedxtrk > (6.+16./(rigidity*rigidity)))return; |
| 267 |
//-------- |
| 268 |
// helium |
| 269 |
//-------- |
| 270 |
if( dedxtrk > (2.8+3.3/(rigidity*rigidity)) && deflection>0 ){ |
| 271 |
sel[5]++; |
| 272 |
hrig2->Fill(rigidity); |
| 273 |
float m=3.725; |
| 274 |
float a=4.; |
| 275 |
float z=2.; |
| 276 |
float en=(sqrt( (z*rigidity)*(z*rigidity)+m*m )-m)/a; |
| 277 |
hen2->Fill(en); |
| 278 |
} |
| 279 |
|
| 280 |
//-------- |
| 281 |
// Z=1 |
| 282 |
//-------- |
| 283 |
if(dedxtrk > (2.8+3.3/(rigidity*rigidity)))return; |
| 284 |
if(dedxtrk < (0.4+0.35/(rigidity*rigidity)))return; |
| 285 |
sel[2]++; |
| 286 |
|
| 287 |
if(deflection>0){ |
| 288 |
hrig1->Fill(rigidity); |
| 289 |
float m=0.938; |
| 290 |
float a=1.; |
| 291 |
float z=1.; |
| 292 |
float en=(sqrt( (z*rigidity)*(z*rigidity)+m*m )-m)/a; |
| 293 |
hen1->Fill(en); |
| 294 |
} |
| 295 |
//------------------------ |
| 296 |
// 6-planes lever-arm |
| 297 |
//------------------------ |
| 298 |
if( !(track->GetTrkTrack()->XGood(0) && track->GetTrkTrack()->XGood(5)) )return; |
| 299 |
sel[3]++; |
| 300 |
|
| 301 |
for(Int_t ip=0; ip<6; ip++)herrx[ip]->Fill(errx[ip]); |
| 302 |
for(Int_t ip=0; ip<6; ip++)herry[ip]->Fill(erry[ip]); |
| 303 |
|
| 304 |
//--------------------------- |
| 305 |
// cut on spatial resolution |
| 306 |
//--------------------------- |
| 307 |
if( errx[0] > 0.0004 )return; |
| 308 |
if( errx[5] > 0.0004 )return; |
| 309 |
sel[4]++; |
| 310 |
|
| 311 |
herrdef->Fill(errdef); |
| 312 |
hbetavsdef2->Fill(deflection,beta); |
| 313 |
hdef->Fill(deflection); |
| 314 |
} |
| 315 |
|
| 316 |
// delete track; |
| 317 |
|
| 318 |
|
| 319 |
} |
| 320 |
//=============================================================== |
| 321 |
// Save histograms |
| 322 |
//=============================================================== |
| 323 |
void SaveHistos( TFile * outf ){ |
| 324 |
|
| 325 |
TString hid; |
| 326 |
|
| 327 |
gROOT->cd(); |
| 328 |
// ---------------------------------- |
| 329 |
// retrieve histos |
| 330 |
// ---------------------------------- |
| 331 |
RetrieveHistos(); |
| 332 |
|
| 333 |
// ---------------------------------- |
| 334 |
// save on file |
| 335 |
// ---------------------------------- |
| 336 |
outf->cd(); |
| 337 |
|
| 338 |
hchi2def->Write(); |
| 339 |
hchi2rig->Write(); |
| 340 |
hdedxtofvsdedxtrk->Write(); |
| 341 |
hdedxtrkvsrig->Write(); |
| 342 |
hdedxtofvsrig->Write(); |
| 343 |
hbetavsrig->Write(); |
| 344 |
hdedxtrkvsdef->Write(); |
| 345 |
hbetavsdef->Write(); |
| 346 |
hbetavsdef2->Write(); |
| 347 |
hdef->Write(); |
| 348 |
hrig1->Write(); |
| 349 |
hrig2->Write(); |
| 350 |
hen1->Write(); |
| 351 |
hen2->Write(); |
| 352 |
herrdef->Write(); |
| 353 |
for(Int_t ip=0; ip<6; ip++){ |
| 354 |
herrx[ip]->Write(); |
| 355 |
herry[ip]->Write(); |
| 356 |
} |
| 357 |
|
| 358 |
cout << "------------------------------------------------------------------" <<endl; |
| 359 |
cout << "Pre-selected events :"<<tot<<endl; |
| 360 |
if(!tot)return; |
| 361 |
cout << "chi2 selection :"<<sel[1]<< " (" << ((float)sel[1]/(float)tot) <<")"<<endl; |
| 362 |
cout << "Z=2 selection :"<<sel[5]<< " (" << ((float)sel[5]/(float)tot) <<")"<<endl; |
| 363 |
cout << "Z=1 selection :"<<sel[2]<< " (" << ((float)sel[2]/(float)tot) <<")"<<endl; |
| 364 |
cout << "(6-plane lever-arm) :"<<sel[3]<< " (" << ((float)sel[3]/(float)tot) <<")"<<endl; |
| 365 |
cout << "ext.spatial res. on planes 1-6 < 4 micron :"<<sel[4]<< " (" << ((float)sel[4]/(float)tot) <<")"<<endl; |
| 366 |
cout << "------------------------------------------------------------------" <<endl; |
| 367 |
|
| 368 |
} |
| 369 |
|