| 1 |
/* |
| 2 |
* TrkChi2VsDeflHistoAction.cpp |
| 3 |
* |
| 4 |
* Created on: 14/ago/2009 |
| 5 |
* Author: Nicola Mori |
| 6 |
*/ |
| 7 |
|
| 8 |
/*! @file TrkChi2VsDeflHistoAction.cpp The TrkChi2VsDeflHistoAction class implementation file. */ |
| 9 |
|
| 10 |
#include "TrkChi2VsDeflHistoAction.h" |
| 11 |
|
| 12 |
TrkChi2VsDeflHistoAction::TrkChi2VsDeflHistoAction(const char *actionName, TString outFileName, TString mode, |
| 13 |
bool logChi2, float minDefl, float maxDefl, unsigned int nBinsDefl, float minChi2, float maxChi2, |
| 14 |
unsigned int nBinsChi2) : |
| 15 |
CollectionAction(actionName), _outFileName(outFileName), _histo(), _histoX(0), _histoY(0), _mode(mode), _logChi2( |
| 16 |
logChi2), _cout_sbuf(cout.rdbuf()), _fout("/dev/null"){ |
| 17 |
|
| 18 |
double deflBins[nBinsDefl + 1]; |
| 19 |
double maxExp = log10(maxDefl / minDefl); |
| 20 |
for (unsigned int i = 0; i < nBinsDefl + 1; i++) { |
| 21 |
deflBins[i] = minDefl * pow(10., (double) i / ((double) nBinsDefl) * maxExp); |
| 22 |
} |
| 23 |
|
| 24 |
double chi2Bins[nBinsChi2 + 1]; |
| 25 |
if (_logChi2) { |
| 26 |
//LINEAR scale for log10(chi2) values |
| 27 |
double chi2Step = (log10(maxChi2 / minChi2)) / (double) nBinsChi2; |
| 28 |
minChi2 = log10(minChi2); |
| 29 |
for (unsigned int i = 0; i < nBinsChi2 + 1; i++) { |
| 30 |
chi2Bins[i] = minChi2 + i * chi2Step; |
| 31 |
} |
| 32 |
} |
| 33 |
else { |
| 34 |
//LOG scale for chi2 values |
| 35 |
maxExp = log10(maxChi2 / minChi2); |
| 36 |
for (unsigned int i = 0; i < nBinsChi2 + 1; i++) { |
| 37 |
chi2Bins[i] = minChi2 * pow(10., (double) i / ((double) nBinsChi2) * maxExp); |
| 38 |
} |
| 39 |
} |
| 40 |
_histo.SetBins(nBinsDefl, deflBins, nBinsChi2, chi2Bins); |
| 41 |
|
| 42 |
// Set labels |
| 43 |
_histo.SetName(GetName()); |
| 44 |
_histoX = (TH2F*) _histo.Clone(TString(GetName()) + TString("_X")); |
| 45 |
_histoY = (TH2F*) _histo.Clone(TString(GetName()) + TString("_Y")); |
| 46 |
if (_logChi2) { |
| 47 |
_histo.SetTitle("Trk log10(Chi2) Vs Deflection"); |
| 48 |
_histo.GetYaxis()->SetTitle("log10(Chi2)"); |
| 49 |
_histoX->SetTitle("Trk log10(Chi2 X) Vs Deflection"); |
| 50 |
_histoX->GetYaxis()->SetTitle("log10(Chi2 X)"); |
| 51 |
_histoY->SetTitle("Trk log10(Chi2 Y) Vs Deflection"); |
| 52 |
_histoY->GetYaxis()->SetTitle("log10(Chi2 Y)"); |
| 53 |
} |
| 54 |
else { |
| 55 |
_histo.SetTitle("Trk Chi2 Vs Deflection"); |
| 56 |
_histo.GetYaxis()->SetTitle("Chi2"); |
| 57 |
_histoX->SetTitle("Trk Chi2 X Vs Deflection"); |
| 58 |
_histoX->GetYaxis()->SetTitle("Chi2 X"); |
| 59 |
_histoY->SetTitle("Trk Chi2 Y Vs Deflection"); |
| 60 |
_histoY->GetYaxis()->SetTitle("Chi2 Y"); |
| 61 |
} |
| 62 |
|
| 63 |
//_histo.SetBins(nBinsDefl, minDefl, maxDefl, nBinsChi2, minChi2, maxChi2); |
| 64 |
|
| 65 |
|
| 66 |
} |
| 67 |
|
| 68 |
void TrkChi2VsDeflHistoAction::OnGood(PamLevel2 *event) { |
| 69 |
|
| 70 |
cout.rdbuf(_fout.rdbuf()); |
| 71 |
|
| 72 |
if (_logChi2) { |
| 73 |
if (event->GetTrack(0)->GetTrkTrack()->chi2 > 0.) { |
| 74 |
_histo.Fill(event->GetTrack(0)->GetTrkTrack()->GetDeflection(), log10(event->GetTrack(0)->GetTrkTrack()->chi2)); |
| 75 |
_histoX->Fill(event->GetTrack(0)->GetTrkTrack()->GetDeflection(), log10( |
| 76 |
event->GetTrack(0)->GetTrkTrack()->GetChi2X())); |
| 77 |
_histoY->Fill(event->GetTrack(0)->GetTrkTrack()->GetDeflection(), log10( |
| 78 |
event->GetTrack(0)->GetTrkTrack()->GetChi2Y())); |
| 79 |
} |
| 80 |
} |
| 81 |
else { |
| 82 |
_histo.Fill(event->GetTrack(0)->GetTrkTrack()->GetDeflection(), event->GetTrack(0)->GetTrkTrack()->chi2); |
| 83 |
_histoX->Fill(event->GetTrack(0)->GetTrkTrack()->GetDeflection(), event->GetTrack(0)->GetTrkTrack()->GetChi2X()); |
| 84 |
_histoY->Fill(event->GetTrack(0)->GetTrkTrack()->GetDeflection(), event->GetTrack(0)->GetTrkTrack()->GetChi2Y()); |
| 85 |
} |
| 86 |
|
| 87 |
cout.rdbuf(_cout_sbuf); |
| 88 |
} |
| 89 |
void TrkChi2VsDeflHistoAction::Finalize() { |
| 90 |
|
| 91 |
TFile outFile(_outFileName, _mode); |
| 92 |
outFile.cd(); |
| 93 |
_histo.Write(); |
| 94 |
_histoX->Write(); |
| 95 |
_histoY->Write(); |
| 96 |
outFile.Close(); |
| 97 |
} |