+/**************************************************************************
+ * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ * *
+ * Author: The ALICE Off-line Project. *
+ * Contributors are mentioned in the code where appropriate. *
+ * *
+ * Permission to use, copy, modify and distribute this software and its *
+ * documentation strictly for non-commercial purposes is hereby granted *
+ * without fee, provided that the above copyright notice appears in all *
+ * copies and that both the copyright notice and this permission notice *
+ * appear in the supporting documentation. The authors make no claims *
+ * about the suitability of this software for any purpose. It is *
+ * provided "as is" without express or implied warranty. *
+ **************************************************************************/
+
/* $Id$ */
#include "AlidNdEtaCorrection.h"
+#include <AliLog.h>
#include <TCanvas.h>
#include <TH3F.h>
#include <TH1D.h>
+#include <TDirectory.h>
+#include <AliCorrection.h>
+#include <AliCorrectionMatrix2D.h>
+#include <AliCorrectionMatrix3D.h>
//____________________________________________________________________
ClassImp(AlidNdEtaCorrection)
}
//____________________________________________________________________
-AlidNdEtaCorrection::AlidNdEtaCorrection(const Char_t* name, const Char_t* title)
+AlidNdEtaCorrection::AlidNdEtaCorrection(const Char_t* name, const Char_t* title, AliPWG0Helper::AnalysisMode analysis)
: TNamed(name, title),
fTrack2ParticleCorrection(0),
fVertexRecoCorrection(0),
fTriggerBiasCorrectionMBToNSD(0),
fTriggerBiasCorrectionMBToND(0)
{
+ //
// constructor
//
- Float_t binLimitsPt[] = {0.0, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0, 1.5, 2.0, 10.0, 100.0};
-
- TString matrixName;
- matrixName.Form("%s_nTrackToNPart", name);
-
- fTrack2ParticleCorrection = new AliCorrectionMatrix3D(matrixName, matrixName, 40, -20, 20, 60, -6, 6, 14, binLimitsPt);
-
- Float_t binLimitsN[] = {-0.5, 0.5, 1.5, 2.5, 3.5, 4.5, 5.5, 6.5, 7.5, 8.5, 9.5,
- 10.5, 12.5, 14.5, 16.5, 18.5, 20.5, 25.5, 30.5, 40.5, 50.5, 100.5, 300.5};
- Float_t binLimitsVtx[] = {-20,-15,-10,-6,-3,0,3,6,10,15,20};
-
- matrixName.Form("%s_vtxReco", name);
- fVertexRecoCorrection = new AliCorrectionMatrix2D(matrixName, matrixName, 10,binLimitsVtx ,22,binLimitsN);
-
- matrixName.Form("%s_triggerBias_MBToINEL", name);
- fTriggerBiasCorrectionMBToINEL = new AliCorrectionMatrix2D(matrixName, matrixName, 10,binLimitsVtx ,22,binLimitsN);
- matrixName.Form("%s_triggerBias_MBToNSD", name);
- fTriggerBiasCorrectionMBToNSD = new AliCorrectionMatrix2D(matrixName, matrixName, 10,binLimitsVtx ,22,binLimitsN);
- matrixName.Form("%s_triggerBias_MBToND", name);
- fTriggerBiasCorrectionMBToND = new AliCorrectionMatrix2D(matrixName, matrixName, 10,binLimitsVtx ,22,binLimitsN);
-
- fTrack2ParticleCorrection ->SetAxisTitles("vtx z [cm]", "#eta", "p_{T} [GeV/c]");
- fVertexRecoCorrection ->SetAxisTitles("vtx z [cm]", "Ntracks");
- fTriggerBiasCorrectionMBToINEL ->SetAxisTitles("vtx z [cm]", "Ntracks");
- fTriggerBiasCorrectionMBToNSD ->SetAxisTitles("vtx z [cm]", "Ntracks");
- fTriggerBiasCorrectionMBToND ->SetAxisTitles("vtx z [cm]", "Ntracks");
+ fTrack2ParticleCorrection = new AliCorrection("Track2Particle", "Track2Particle", analysis);
+ fVertexRecoCorrection = new AliCorrection("VertexReconstruction", "VertexReconstruction", analysis);
+ fTriggerBiasCorrectionMBToINEL = new AliCorrection("TriggerBias_MBToINEL", "TriggerBias_MBToINEL", analysis);
+ fTriggerBiasCorrectionMBToNSD = new AliCorrection("TriggerBias_MBToNSD", "TriggerBias_MBToNSD", analysis);
+ fTriggerBiasCorrectionMBToND = new AliCorrection("TriggerBias_MBToND", "TriggerBias_MBToND", analysis);
}
//____________________________________________________________________
// divide the histograms in the AliCorrectionMatrix2D objects to get the corrections
fTrack2ParticleCorrection->Divide();
-
- TH3F* hist = fTrack2ParticleCorrection->GetCorrectionHistogram();
- Int_t emptyBins = 0;
- for (Int_t x=hist->GetXaxis()->FindBin(-10); x<=hist->GetXaxis()->FindBin(10); ++x)
- for (Int_t y=hist->GetYaxis()->FindBin(-0.8); y<=hist->GetYaxis()->FindBin(0.8); ++y)
- for (Int_t z=hist->GetZaxis()->FindBin(0.3); z<=hist->GetZaxis()->FindBin(9.9); ++z)
- if (hist->GetBinContent(x, y, z) == 0)
- {
- printf("Empty bin in fTrack2ParticleCorrection at vtx = %f, eta = %f, pt = %f\n", hist->GetXaxis()->GetBinCenter(x), hist->GetYaxis()->GetBinCenter(y), hist->GetZaxis()->GetBinCenter(z));
- ++emptyBins;
- }
-
- printf("INFO: In the central region fTrack2ParticleCorrection has %d empty bins\n", emptyBins);
-
fVertexRecoCorrection->Divide();
fTriggerBiasCorrectionMBToINEL->Divide();
fTriggerBiasCorrectionMBToNSD->Divide();
}
//____________________________________________________________________
-Long64_t
-AlidNdEtaCorrection::Merge(TCollection* list) {
+Long64_t AlidNdEtaCorrection::Merge(TCollection* list)
+{
// Merge a list of dNdEtaCorrection objects with this (needed for
// PROOF).
// Returns the number of merged objects (including this).
if (entry == 0)
continue;
- collectionNtrackToNparticle ->Add(entry->GetTrack2ParticleCorrection());
- collectionVertexReco ->Add(entry->GetVertexRecoCorrection());
- collectionTriggerBiasMBToINEL->Add(entry->GetTriggerBiasCorrection("INEL"));
- collectionTriggerBiasMBToNSD ->Add(entry->GetTriggerBiasCorrection("NSD"));
- collectionTriggerBiasMBToND ->Add(entry->GetTriggerBiasCorrection("ND"));
+ collectionNtrackToNparticle ->Add(entry->fTrack2ParticleCorrection);
+ collectionVertexReco ->Add(entry->fVertexRecoCorrection);
+ collectionTriggerBiasMBToINEL->Add(entry->fTriggerBiasCorrectionMBToINEL);
+ collectionTriggerBiasMBToNSD ->Add(entry->fTriggerBiasCorrectionMBToNSD);
+ collectionTriggerBiasMBToND ->Add(entry->fTriggerBiasCorrectionMBToND);
count++;
-
- //fNEvents += entry->fNEvents;
- //fNTriggeredEvents += entry->fNTriggeredEvents;
}
fTrack2ParticleCorrection ->Merge(collectionNtrackToNparticle);
fVertexRecoCorrection ->Merge(collectionVertexReco);
return count+1;
}
+//____________________________________________________________________
+void AlidNdEtaCorrection::Add(AlidNdEtaCorrection* aCorrectionsToAdd, Float_t c) {
+ //
+ // adds the measured and generated of aCorrectionsToAdd to measured and generated
+ // of all corrections in this
+
+ fTrack2ParticleCorrection ->Add(aCorrectionsToAdd->GetTrack2ParticleCorrection() ,c);
+ fVertexRecoCorrection ->Add(aCorrectionsToAdd->GetVertexRecoCorrection() ,c);
+ fTriggerBiasCorrectionMBToINEL ->Add(aCorrectionsToAdd->GetTriggerBiasCorrectionINEL(),c);
+ fTriggerBiasCorrectionMBToNSD ->Add(aCorrectionsToAdd->GetTriggerBiasCorrectionNSD() ,c);
+ fTriggerBiasCorrectionMBToND ->Add(aCorrectionsToAdd->GetTriggerBiasCorrectionND() ,c);
+
+}
+
+//____________________________________________________________________
+void AlidNdEtaCorrection::Reset(void) {
+ //
+ // reset all corrections
+ //
+
+ fTrack2ParticleCorrection ->Reset();
+ fVertexRecoCorrection ->Reset();
+ fTriggerBiasCorrectionMBToINEL ->Reset();
+ fTriggerBiasCorrectionMBToNSD ->Reset();
+ fTriggerBiasCorrectionMBToND ->Reset();
+
+}
+
//____________________________________________________________________
-Bool_t
-AlidNdEtaCorrection::LoadHistograms(const Char_t* fileName, const Char_t* dir) {
+Bool_t AlidNdEtaCorrection::LoadHistograms(const Char_t* dir)
+{
//
// loads the histograms
+ // if dir is empty a directory with the name of this object is taken (like in SaveHistogram)
//
- fTrack2ParticleCorrection ->LoadHistograms(fileName, dir);
- fVertexRecoCorrection ->LoadHistograms(fileName, dir);
- fTriggerBiasCorrectionMBToINEL ->LoadHistograms(fileName, dir);
- fTriggerBiasCorrectionMBToNSD ->LoadHistograms(fileName, dir);
- fTriggerBiasCorrectionMBToND ->LoadHistograms(fileName, dir);
+ if (!dir)
+ dir = GetName();
-
+ if (!gDirectory->cd(dir))
+ return kFALSE;
+
+ fTrack2ParticleCorrection ->LoadHistograms();
+ fVertexRecoCorrection ->LoadHistograms();
+ fTriggerBiasCorrectionMBToINEL ->LoadHistograms();
+ fTriggerBiasCorrectionMBToNSD ->LoadHistograms();
+ fTriggerBiasCorrectionMBToND ->LoadHistograms();
+
+ gDirectory->cd("..");
return kTRUE;
}
//____________________________________________________________________
-void
-AlidNdEtaCorrection::SaveHistograms() {
+void AlidNdEtaCorrection::SaveHistograms()
+{
//
// save the histograms
//
fTriggerBiasCorrectionMBToNSD ->SaveHistograms();
fTriggerBiasCorrectionMBToND ->SaveHistograms();
- gDirectory->cd("../");
+ gDirectory->cd("..");
}
//____________________________________________________________________
void AlidNdEtaCorrection::DrawHistograms()
{
//
- // call the draw histogram method of the two AliCorrectionMatrix2D objects
+ // call the draw histogram method of the corrections
+ //
fTrack2ParticleCorrection ->DrawHistograms();
fVertexRecoCorrection ->DrawHistograms();
fTriggerBiasCorrectionMBToINEL->DrawHistograms();
fTriggerBiasCorrectionMBToNSD ->DrawHistograms();
fTriggerBiasCorrectionMBToND ->DrawHistograms();
-
}
//____________________________________________________________________
-void AlidNdEtaCorrection::FillEventWithTrigger(Float_t vtx, Float_t n)
+void AlidNdEtaCorrection::DrawOverview(const char* canvasName)
{
- // fill events with trigger.
- // used to calculate vertex reco and trigger bias corrections
+ //
+ // call the DrawOverview histogram method of the corrections
+ //
- fVertexRecoCorrection->FillGene(vtx, n);
- fTriggerBiasCorrectionMBToINEL ->FillMeas(vtx, n);
- fTriggerBiasCorrectionMBToNSD ->FillMeas(vtx, n);
- fTriggerBiasCorrectionMBToND ->FillMeas(vtx, n);
+ fTrack2ParticleCorrection ->DrawOverview(canvasName);
+ fVertexRecoCorrection ->DrawOverview(canvasName);
+ fTriggerBiasCorrectionMBToINEL->DrawOverview(canvasName);
+ fTriggerBiasCorrectionMBToNSD ->DrawOverview(canvasName);
+ fTriggerBiasCorrectionMBToND ->DrawOverview(canvasName);
}
//____________________________________________________________________
-void AlidNdEtaCorrection::FillEventAll(Float_t vtx, Float_t n, Char_t* opt)
+void AlidNdEtaCorrection::FillMCParticle(Float_t vtx, Float_t eta, Float_t pt, Bool_t trigger, Bool_t vertex, Int_t processType)
{
- // fill all events
- // used to calculate trigger bias corrections
-
- if (strcmp(opt,"INEL")==0)
- fTriggerBiasCorrectionMBToINEL->FillGene(vtx, n);
- else if (strcmp(opt,"NSD")==0)
- fTriggerBiasCorrectionMBToNSD->FillGene(vtx, n);
- else if (strcmp(opt,"ND")==0)
- fTriggerBiasCorrectionMBToND->FillGene(vtx, n);
- else
- AliDebug(AliLog::kWarning, Form(" event type %s unknown (use INEL, NSD or ND)",opt));
+ // fills a particle in the corrections
+ // it is filled in generated or measured depending of the flags
+
+ fTriggerBiasCorrectionMBToINEL->GetTrackCorrection()->FillGene(vtx, eta, pt);
+
+ if (processType != 92 && processType != 93)
+ fTriggerBiasCorrectionMBToNSD->GetTrackCorrection()->FillGene(vtx, eta, pt);
+
+ if (processType!=92 && processType!=93 && processType!=94)
+ fTriggerBiasCorrectionMBToND->GetTrackCorrection()->FillGene(vtx, eta, pt);
+
+ if (!trigger)
+ return;
+
+ fTriggerBiasCorrectionMBToINEL->GetTrackCorrection()->FillMeas(vtx, eta, pt);
+ fTriggerBiasCorrectionMBToNSD->GetTrackCorrection()->FillMeas(vtx, eta, pt);
+ fTriggerBiasCorrectionMBToND->GetTrackCorrection()->FillMeas(vtx, eta, pt);
+ fVertexRecoCorrection->GetTrackCorrection()->FillGene(vtx, eta, pt);
+
+ if (!vertex)
+ return;
+
+ fVertexRecoCorrection->GetTrackCorrection()->FillMeas(vtx, eta, pt);
+ fTrack2ParticleCorrection->GetTrackCorrection()->FillGene(vtx, eta, pt);
}
//____________________________________________________________________
-AliCorrectionMatrix2D* AlidNdEtaCorrection::GetTriggerBiasCorrection(Char_t* opt)
+void AlidNdEtaCorrection::FillTrackedParticle(Float_t vtx, Float_t eta, Float_t pt)
{
- // returning the trigger bias correction matrix
- // option can be used to specify to which collision process (INEL, NSD or ND)
- if (strcmp(opt,"INEL")==0)
- return fTriggerBiasCorrectionMBToINEL;
- else if (strcmp(opt,"NSD")==0)
- return fTriggerBiasCorrectionMBToNSD;
- else if (strcmp(opt,"ND")==0)
- return fTriggerBiasCorrectionMBToND;
- else
- {
- AliDebug(AliLog::kWarning, Form(" %s is unknown (use INEL, NSD or ND). returning INEL ",opt));
- return fTriggerBiasCorrectionMBToINEL;
- }
+ // fills a tracked particle in the corrections
+
+ fTrack2ParticleCorrection->GetTrackCorrection()->FillMeas(vtx, eta, pt);
}
//____________________________________________________________________
-Float_t AlidNdEtaCorrection::GetTriggerBiasCorrection(Float_t vtx, Float_t n, Char_t* opt)
+void AlidNdEtaCorrection::FillEvent(Float_t vtx, Float_t n, Bool_t trigger, Bool_t vertex, Int_t processType)
{
- // returning the trigger bias correction matrix
- // 3rd option can be used to specify to which collision process (INEL, NSD or ND)
-
- if (strcmp(opt,"INEL")==0)
- return fTriggerBiasCorrectionMBToINEL->GetCorrection(vtx, n);
- else if (strcmp(opt,"NSD")==0)
- return fTriggerBiasCorrectionMBToNSD->GetCorrection(vtx, n);
- else if (strcmp(opt,"ND")==0)
- return fTriggerBiasCorrectionMBToND->GetCorrection(vtx, n);
- else {
- AliDebug(AliLog::kWarning, Form(" %s unknown (use INEL, NSD or ND). returning corr for INEL",opt));
- return fTriggerBiasCorrectionMBToINEL->GetCorrection(vtx, n);
- }
-}
+ // fills an event int he correction
+ // it is filled in generated or measured depending of the flags
+
+ fTriggerBiasCorrectionMBToINEL->GetEventCorrection()->FillGene(vtx, n);
+
+ if (processType != 92 && processType != 93)
+ fTriggerBiasCorrectionMBToNSD->GetEventCorrection()->FillGene(vtx, n);
+
+ if (processType!=92 && processType!=93 && processType!=94)
+ fTriggerBiasCorrectionMBToND->GetEventCorrection()->FillGene(vtx, n);
+
+ if (!trigger)
+ return;
+ fTriggerBiasCorrectionMBToINEL->GetEventCorrection()->FillMeas(vtx, n);
+ fTriggerBiasCorrectionMBToNSD->GetEventCorrection()->FillMeas(vtx, n);
+ fTriggerBiasCorrectionMBToND->GetEventCorrection()->FillMeas(vtx, n);
+ fVertexRecoCorrection->GetEventCorrection()->FillGene(vtx, n);
+
+ if (!vertex)
+ return;
+
+ fVertexRecoCorrection->GetEventCorrection()->FillMeas(vtx, n);
+}
//____________________________________________________________________
-Float_t AlidNdEtaCorrection::GetMeasuredFraction(Float_t ptCutOff, Float_t eta, Bool_t debug)
+Float_t AlidNdEtaCorrection::GetMeasuredFraction(CorrectionType correctionType, Float_t ptCutOff, Float_t eta, Bool_t debug)
{
// calculates the fraction of particles measured (some are missed due to the pt cut off)
- // uses the generated particle histogram from fTrack2ParticleCorrection
+ //
+ // uses the generated particle histogram from the correction passed, e.g. pass GetTrack2ParticleCorrection()
- const TH3F* generated = fTrack2ParticleCorrection->GetGeneratedHistogram();
+ if (!GetCorrection(correctionType))
+ return -1;
+
+ const TH3F* generated = GetCorrection(correctionType)->GetTrackCorrection()->GetGeneratedHistogram();
// find eta borders, if eta is negative assume -0.8 ... 0.8
Int_t etaBegin = 0;
etaEnd = etaBegin;
}
- Int_t vertexBegin = generated->GetXaxis()->FindBin(-10);
- Int_t vertexEnd = generated->GetXaxis()->FindBin(10);
+ //Printf("AlidNdEtaCorrection::GetMeasuredFraction: Using vtx range of +- 10 cm");
+ Int_t vertexBegin = generated->GetXaxis()->FindBin(-9.99);
+ Int_t vertexEnd = generated->GetXaxis()->FindBin(9.99);
TH1D* ptProj = dynamic_cast<TH1D*> (generated->ProjectionZ(Form("%s_pt", generated->GetName()), vertexBegin, vertexEnd, etaBegin, etaEnd));
+ //printf("GetMeasuredFraction: bin range %d %d %d %d\n", vertexBegin, vertexEnd, etaBegin, etaEnd);
ptProj->GetXaxis()->SetTitle(generated->GetZaxis()->GetTitle());
Int_t ptBin = ptProj->FindBin(ptCutOff);
+ //printf("GetMeasuredFraction: bin range %d %d\n", ptBin, ptProj->GetNbinsX());
Float_t abovePtCut = ptProj->Integral(ptBin, ptProj->GetNbinsX());
Float_t all = ptProj->Integral();
Float_t fraction = abovePtCut / all;
+ //printf("GetMeasuredFraction: all %f above %f fraction %f\n", all, abovePtCut, fraction);
+
if (debug)
{
new TCanvas;
return fraction;
}
+//____________________________________________________________________
+TH1* AlidNdEtaCorrection::GetMeasuredEventFraction(CorrectionType correctionType, Int_t multCut)
+{
+ // calculates the fraction of events above multCut (but including it)
+ //
+ // uses the generated event histogram from the correction passed, e.g. pass GetTrack2ParticleCorrection()
+
+ if (!GetCorrection(correctionType))
+ return 0;
+
+ const TH2F* generated = GetCorrection(correctionType)->GetEventCorrection()->GetGeneratedHistogram();
+
+ TH1* allEvents = generated->ProjectionX(Form("%s_all", generated->GetName()), 1, generated->GetNbinsY());
+ TH1* aboveEvents = generated->ProjectionX(Form("%s_above", generated->GetName()), generated->GetYaxis()->FindBin(multCut), generated->GetNbinsY());
+
+ aboveEvents->Divide(aboveEvents, allEvents, 1, 1, "B");
+
+ return aboveEvents;
+}
+
+//____________________________________________________________________
void AlidNdEtaCorrection::ReduceInformation()
{
// this function deletes the measured and generated histograms from the corrections to reduce the amount of data
fTriggerBiasCorrectionMBToND ->ReduceInformation();
}
+//____________________________________________________________________
+AliCorrection* AlidNdEtaCorrection::GetCorrection(CorrectionType correctionType)
+{
+ // returns the given correction
+
+ switch (correctionType)
+ {
+ case kNone : return 0;
+ case kTrack2Particle : return fTrack2ParticleCorrection;
+ case kVertexReco : return fVertexRecoCorrection;
+ case kINEL : return fTriggerBiasCorrectionMBToINEL;
+ case kNSD : return fTriggerBiasCorrectionMBToNSD;
+ case kND : return fTriggerBiasCorrectionMBToND;
+ }
+
+ return 0;
+}