| 1 |
// |
| 2 |
// Plot the energy [MIP], nstrip and qtot distributions - Emiliano Mocchiutti |
| 3 |
// |
| 4 |
// FCaloMIP.cxx version 1.01 (2006-06-03) |
| 5 |
// |
| 6 |
// The only input needed is the path to the directory created by YODA for the data file you want to analyze. |
| 7 |
// |
| 8 |
// Changelog: |
| 9 |
// |
| 10 |
// 1.00 - 1.01 (2006-06-03): Flight version, read unique YODA file. |
| 11 |
// |
| 12 |
// 0.00 - 1.00 (2006-06-03): Clone of CaloMIP.c v2r12 |
| 13 |
// |
| 14 |
#include <iostream> |
| 15 |
#include <sstream> |
| 16 |
#include <fstream> |
| 17 |
#include <TTree.h> |
| 18 |
#include <TClassEdit.h> |
| 19 |
#include <TObject.h> |
| 20 |
#include <TList.h> |
| 21 |
#include <TSystem.h> |
| 22 |
#include <TSystemDirectory.h> |
| 23 |
#include <TString.h> |
| 24 |
#include <TFile.h> |
| 25 |
#include <TClass.h> |
| 26 |
#include <TCanvas.h> |
| 27 |
#include <TH1.h> |
| 28 |
#include <TH1F.h> |
| 29 |
#include <TH2D.h> |
| 30 |
#include <TLatex.h> |
| 31 |
#include <TPad.h> |
| 32 |
#include <TPaveLabel.h> |
| 33 |
#include <TChain.h> |
| 34 |
#include <TStyle.h> |
| 35 |
#include <TF1.h> |
| 36 |
// |
| 37 |
extern bool existfile(TString); |
| 38 |
extern void stringcopy(TString&, const TString&, Int_t, Int_t); |
| 39 |
TF1 *langaufit(TH1F *, Double_t *, Double_t *, Double_t *, Double_t *, Double_t *, Double_t *, Double_t *, Int_t *, TString); |
| 40 |
Double_t langaupro(Double_t *); |
| 41 |
// |
| 42 |
#include <caloclassesfun.h> |
| 43 |
// |
| 44 |
using namespace std; |
| 45 |
|
| 46 |
void FCaloMIP(const TString filename, Int_t view = 0, Int_t plane = 0, Int_t strip = 0, Int_t fromevent = 0, Int_t toevent = 0, TString outDir = "", TString saveas = "gif"){ |
| 47 |
gStyle->SetOptStat(1111); |
| 48 |
const char* startingdir = gSystem->WorkingDirectory(); |
| 49 |
if ( !strcmp(outDir.Data(),"") ) outDir = startingdir; |
| 50 |
if ( !existfile(filename) ){ |
| 51 |
printf(" %s :no such file or directory \n",filename.Data()); |
| 52 |
printf("\n ERROR: No calorimeter LEVEL1 file! \n\n Run FCaloLEVEL1 to generate level1 data. \n\n"); |
| 53 |
return; |
| 54 |
}; |
| 55 |
// |
| 56 |
TFile *File = new TFile(filename.Data()); |
| 57 |
TTree *otr = (TTree*)File->Get("CaloLevel1"); |
| 58 |
if ( !otr ){ |
| 59 |
printf(" %s are you sure? I need a LEVEL1 file! \n",filename.Data()); |
| 60 |
return; |
| 61 |
}; |
| 62 |
// |
| 63 |
CalorimeterLevel1 *calo = new CalorimeterLevel1(); |
| 64 |
otr->SetBranchAddress("Event", &calo); |
| 65 |
// |
| 66 |
Long64_t nevents = otr->GetEntries(); |
| 67 |
// |
| 68 |
// input limits |
| 69 |
// |
| 70 |
Int_t minview; |
| 71 |
Int_t maxview; |
| 72 |
if ( view > 2 || view < 0 ) { |
| 73 |
printf("You can choose view = 1 (X-views), view = 2 (Y-views) or view = 0 (both)\n"); |
| 74 |
return; |
| 75 |
}; |
| 76 |
if ( view == 0 ) { |
| 77 |
minview = 0; |
| 78 |
maxview = 2; |
| 79 |
} else { |
| 80 |
minview = view - 1; |
| 81 |
maxview = view; |
| 82 |
}; |
| 83 |
|
| 84 |
if ( plane > 22 || plane < 0 ) { |
| 85 |
printf("You can choose plane between 1 and 22 or plane = 0 (all)\n"); |
| 86 |
return; |
| 87 |
}; |
| 88 |
Int_t minplane; |
| 89 |
Int_t maxplane; |
| 90 |
if ( plane == 0 ) { |
| 91 |
minplane = 0; |
| 92 |
maxplane = 22; |
| 93 |
} else { |
| 94 |
minplane = plane - 1; |
| 95 |
maxplane = plane; |
| 96 |
}; |
| 97 |
Int_t minstrip; |
| 98 |
Int_t maxstrip; |
| 99 |
if ( strip > 96 || strip < 0 ) { |
| 100 |
printf("You can choose strip between 0 and 96 or strip = 0 (all)\n"); |
| 101 |
return; |
| 102 |
}; |
| 103 |
if ( strip == 0 ) { |
| 104 |
minstrip = 0; |
| 105 |
maxstrip = 96; |
| 106 |
} else { |
| 107 |
minstrip = strip - 1; |
| 108 |
maxstrip = strip; |
| 109 |
}; |
| 110 |
|
| 111 |
if ( fromevent > toevent && toevent ){ |
| 112 |
printf("It must be fromevent < toevent \n"); |
| 113 |
return; |
| 114 |
}; |
| 115 |
|
| 116 |
|
| 117 |
if ( fromevent > nevents+1 || fromevent < 0 ) { |
| 118 |
printf("You can choose fromevent between 0 (all) and %i \n",(int)nevents+1); |
| 119 |
return; |
| 120 |
}; |
| 121 |
|
| 122 |
if ( toevent > nevents+1 || toevent < 0 ) { |
| 123 |
printf("You can choose toevent between 0 (all) and %i \n",(int)nevents+1); |
| 124 |
return; |
| 125 |
}; |
| 126 |
Int_t minevent; |
| 127 |
Int_t maxevent; |
| 128 |
if ( fromevent == 0 ) { |
| 129 |
minevent = 0; |
| 130 |
maxevent = nevents; |
| 131 |
} else { |
| 132 |
minevent = fromevent - 1; |
| 133 |
if ( toevent > 0 ){ |
| 134 |
maxevent = toevent; |
| 135 |
} else { |
| 136 |
maxevent = fromevent - 1; |
| 137 |
}; |
| 138 |
}; |
| 139 |
|
| 140 |
// |
| 141 |
// Book the histograms: |
| 142 |
// |
| 143 |
// |
| 144 |
TH1F *gmip; |
| 145 |
gmip = new TH1F("the MIP","",55,-0.19,1.9); |
| 146 |
//gmip->SetBit(TH1F::kCanRebin); |
| 147 |
// |
| 148 |
TH1F *diffbase; |
| 149 |
diffbase = new TH1F("diffbase","",55,-0.19,0.19); |
| 150 |
diffbase->SetBit(TH1F::kCanRebin); |
| 151 |
// |
| 152 |
TH1F *Qtot; |
| 153 |
Qtot = new TH1F("qtot","",70,0.,60.); |
| 154 |
Qtot->SetBit(TH1F::kCanRebin); |
| 155 |
// |
| 156 |
TH1F *Nstrip; |
| 157 |
Nstrip = new TH1F("nstrip","",70,0.,60.); |
| 158 |
Nstrip->SetBit(TH1F::kCanRebin); |
| 159 |
// |
| 160 |
// run over all the events |
| 161 |
// |
| 162 |
// |
| 163 |
Int_t tot2 = 0; |
| 164 |
Int_t baseDIFFfull = 0; |
| 165 |
printf("\n Processed events: \n\n"); |
| 166 |
// |
| 167 |
for (Int_t i = minevent; i < maxevent; i++){ |
| 168 |
if ( i%1000 == 0 && i > 0 ) printf(" %iK \n",i/1000); |
| 169 |
|
| 170 |
// |
| 171 |
// read from the header of the event the absolute time at which it was recorded |
| 172 |
// |
| 173 |
otr->GetEntry(i); |
| 174 |
|
| 175 |
// |
| 176 |
// run over views and planes |
| 177 |
// |
| 178 |
for (Int_t l = minview; l < maxview; l++){ |
| 179 |
for (Int_t m = minplane; m < maxplane; m++){ |
| 180 |
Int_t pre = -1; |
| 181 |
for (Int_t n = minstrip; n < maxstrip; n++){ |
| 182 |
// |
| 183 |
if ( n%16 == 0 ) pre++; |
| 184 |
if ( n == minstrip && pre == -1 ){ |
| 185 |
for (Int_t nm = minstrip; pre == -1 ; nm--){ |
| 186 |
if ( nm%16 == 0 ) pre++; |
| 187 |
}; |
| 188 |
}; |
| 189 |
if ( pre == -1 ) { |
| 190 |
printf("Unable to determine the preamplifier number. Quit.\n"); |
| 191 |
return; |
| 192 |
}; |
| 193 |
// |
| 194 |
if ( calo->nobase ) tot2++; |
| 195 |
if ( calo->diffbas[l][m][pre] != 0. ) { |
| 196 |
// printf("differenza %f %i %i %i %i %i \n",calo->diffbas[l][m][pre],i,l,m,pre,n); |
| 197 |
baseDIFFfull = 1; |
| 198 |
diffbase->Fill(calo->diffbas[l][m][pre]); |
| 199 |
// return; |
| 200 |
}; |
| 201 |
if ( calo->estrip[l][m][n] != 0. ) gmip->Fill(calo->estrip[l][m][n]); |
| 202 |
}; |
| 203 |
}; |
| 204 |
}; |
| 205 |
if ( calo->nstrip > 0. ) Nstrip->Fill(calo->nstrip); |
| 206 |
if ( calo->qtot > 0. ) Qtot->Fill(calo->qtot); |
| 207 |
}; |
| 208 |
if ( tot2 ){ |
| 209 |
printf("WARNING: events with no baselines (event lost): %i\n",tot2); |
| 210 |
}; |
| 211 |
// |
| 212 |
// figures: |
| 213 |
// |
| 214 |
// THE MIP |
| 215 |
TCanvas *figura = new TCanvas("The_calorimeter_MIP", "The_calorimeter_MIP", 1100, 900); |
| 216 |
figura->SetFillColor(10); |
| 217 |
figura->Range(0,0,100,100); |
| 218 |
figura->cd(); |
| 219 |
gPad->SetLogy(); |
| 220 |
gmip->SetXTitle("MIP"); |
| 221 |
gmip->SetYTitle("Number of events"); |
| 222 |
gPad->SetTicks(); |
| 223 |
printf("\n Fitting...\n\n"); |
| 224 |
// Setting fit range and start values |
| 225 |
Double_t fr[2]; |
| 226 |
Double_t sv[4], pllo[4], plhi[4], fp[4], fpe[4]; |
| 227 |
fr[0] = 0.7; |
| 228 |
fr[1] = 2.0; |
| 229 |
pllo[0]=0.01; pllo[1]=0.5; pllo[2]=1.0; pllo[3]=0.05; |
| 230 |
plhi[0]=5.0; plhi[1]=5.0; plhi[2]=1000000.0; plhi[3]=1.0; |
| 231 |
sv[0]=1.; sv[1]=1.0; sv[2]=500.0; sv[3]=0.2; |
| 232 |
Double_t chisqr; |
| 233 |
Int_t ndf; |
| 234 |
TF1 *fitsnr = langaufit(gmip,fr,sv,pllo,plhi,fp,fpe,&chisqr,&ndf,"QR0"); |
| 235 |
// |
| 236 |
Double_t SNRPeak = langaupro(fp); |
| 237 |
printf("\n Conversion factor: %f \n",SNRPeak); |
| 238 |
// |
| 239 |
// gStyle->SetOptStat(1111); |
| 240 |
gStyle->SetOptFit(111); |
| 241 |
printf("\n ...done!\n"); |
| 242 |
gmip->Draw(); |
| 243 |
fitsnr->Draw("lsame"); |
| 244 |
// Differences between DSP and calculated baselines in full mode |
| 245 |
TCanvas *base = 0; |
| 246 |
if ( baseDIFFfull ){ |
| 247 |
base = new TCanvas("WARNING", "WARNING_full mode,_differences_between_calculated_and_DSP_baselines!!", 1100, 900); |
| 248 |
base->SetFillColor(10); |
| 249 |
base->Range(0,0,100,100); |
| 250 |
base->cd(); |
| 251 |
gStyle->SetOptStat(0); |
| 252 |
gPad->SetLogy(); |
| 253 |
diffbase->Draw(); |
| 254 |
gStyle->SetOptStat(1111); |
| 255 |
}; |
| 256 |
// nstrip and qtot WARNING: only GOOD strips considered! |
| 257 |
TCanvas *figura2 = new TCanvas("nstrip_and_qtot", "nstrip_and_qtot", 1100, 900); |
| 258 |
figura2->SetFillColor(10); |
| 259 |
figura2->Range(0,0,100,100); |
| 260 |
TPad *pd1; |
| 261 |
TPad *pd2; |
| 262 |
pd1 = new TPad("pd1","This is pad1",0.02,0.02,0.49,0.98,10); |
| 263 |
pd2 = new TPad("pd2","This is pad2",0.51,0.02,0.98,0.98,10); |
| 264 |
figura2->cd(); |
| 265 |
gStyle->SetOptStat(1111); |
| 266 |
pd1->Range(0,0,100,100); |
| 267 |
pd2->Range(0,0,100,100); |
| 268 |
pd1->SetGrid(); |
| 269 |
pd2->SetGrid(); |
| 270 |
pd1->SetTicks(); |
| 271 |
pd2->SetTicks(); |
| 272 |
pd1->Draw(); |
| 273 |
pd2->Draw(); |
| 274 |
pd1->cd(); |
| 275 |
gPad->SetLogy(); |
| 276 |
Nstrip->SetXTitle("Number of hit (E>0.7 MIP)"); |
| 277 |
Nstrip->SetYTitle("Number of events"); |
| 278 |
Nstrip->Draw(); |
| 279 |
pd2->cd(); |
| 280 |
gPad->SetLogy(); |
| 281 |
Qtot->SetXTitle("MIP"); |
| 282 |
Qtot->SetYTitle("Number of events"); |
| 283 |
Qtot->Draw(); |
| 284 |
printf("\n"); |
| 285 |
// |
| 286 |
const string fil = (const char*)filename; |
| 287 |
Int_t posiz = fil.find("dw_"); |
| 288 |
if ( posiz == -1 ) posiz = fil.find("DW_"); |
| 289 |
Int_t posiz2 = posiz+13; |
| 290 |
TString file2; |
| 291 |
stringcopy(file2,filename,posiz,posiz2); |
| 292 |
// |
| 293 |
const char *figrec = file2; |
| 294 |
const char *outdir = outDir; |
| 295 |
stringstream figsave; |
| 296 |
const char *format = saveas; |
| 297 |
figsave.str(""); |
| 298 |
figsave << outdir << "/" ; |
| 299 |
figsave << figrec << "_mip1."; |
| 300 |
figsave << format; |
| 301 |
// |
| 302 |
figura->SaveAs(figsave.str().c_str()); |
| 303 |
figsave.str(""); |
| 304 |
figsave << outdir << "/" ; |
| 305 |
figsave << figrec << "_mip2."; |
| 306 |
figsave << format; |
| 307 |
// |
| 308 |
figura2->SaveAs(figsave.str().c_str()); |
| 309 |
if ( baseDIFFfull ){ |
| 310 |
figsave.str(""); |
| 311 |
figsave << outdir << "/" ; |
| 312 |
figsave << figrec << "_mip3."; |
| 313 |
figsave << format; |
| 314 |
// |
| 315 |
base->SaveAs(figsave.str().c_str()); |
| 316 |
}; |
| 317 |
// |
| 318 |
gSystem->ChangeDirectory(startingdir); |
| 319 |
} |