PWGCF/Correlations/Base/AliUEHistograms.h

   1 #ifndef AliUEHistograms_H
   2 #define AliUEHistograms_H
   3
   4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 /* $Id: AliUEHistograms.h 20164 2007-08-14 15:31:50Z morsch $ */
   8
   9 // encapsulates several AliUEHist objects for a full UE analysis plus additional control histograms
  10
  11 #include "TNamed.h"
  12 #include "AliUEHist.h"
  13 #include "TMath.h"
  14
  15 class AliVParticle;
  16
  17 class TList;
  18 class TSeqCollection;
  19 class TObjArray;
  20 class TH1F;
  21 class TH2F;
  22 class TH3F;
  23
  24 class AliUEHistograms : public TNamed
  25 {
  26  public:
  27   AliUEHistograms(const char* name = "AliUEHistograms", const char* histograms = "");
  28   virtual ~AliUEHistograms();
  29
  30   void Fill(Int_t eventType, Float_t zVtx, AliUEHist::CFStep step, AliVParticle* leading, TList* toward, TList* away, TList* min, TList* max);
  31   void FillCorrelations(Double_t centrality, Float_t zVtx, AliUEHist::CFStep step, TObjArray* particles, TObjArray* mixed = 0, Float_t weight = 1, Bool_t firstTime = kTRUE, Bool_t twoTrackEfficiencyCut = kFALSE, Float_t bSign = 0);
  32   void Fill(AliVParticle* leadingMC, AliVParticle* leadingReco);
  33   void FillEvent(Int_t eventType, Int_t step);
  34   void FillEvent(Double_t centrality, Int_t step);
  35   void FillTrackingEfficiency(TObjArray* mc, TObjArray* recoPrim, TObjArray* recoAll, Int_t particleType, Double_t centrality = 0);
  36
  37   void CopyReconstructedData(AliUEHistograms* from);
  38   void DeepCopy(AliUEHistograms* from);
  39
  40   AliUEHist* GetUEHist(Int_t id);
  41
  42   AliUEHist* GetNumberDensitypT() { return fNumberDensitypT; }
  43   AliUEHist* GetSumpT() { return fSumpT; }
  44   AliUEHist* GetNumberDensityPhi() { return fNumberDensityPhi; }
  45
  46   void SetNumberDensitypT(AliUEHist* obj) { fNumberDensitypT = obj; }
  47   void SetSumpT(AliUEHist* obj) { fSumpT = obj; }
  48   void SetNumberDensityPhi(AliUEHist* obj) { fNumberDensityPhi = obj; }
  49
  50   void SetRunNumber(Long64_t runNumber) { fRunNumber = runNumber; }
  51
  52   TH2F* GetCorrelationpT()  { return fCorrelationpT; }
  53   TH2F* GetCorrelationEta() { return fCorrelationEta; }
  54   TH2F* GetCorrelationPhi() { return fCorrelationPhi; }
  55   TH2F* GetCorrelationR()   { return fCorrelationR; }
  56   TH2F* GetCorrelationLeading2Phi() { return fCorrelationLeading2Phi; }
  57   TH2F* GetCorrelationMultiplicity() { return fCorrelationMultiplicity; }
  58
  59   TH2F* GetEventCount()     { return fEventCount; }
  60   TH3F* GetEventCountDifferential() { return fEventCountDifferential; }
  61   TH1F* GetVertexContributors() { return fVertexContributors; }
  62   TH1F* GetCentralityDistribution() { return fCentralityDistribution; }
  63   Long64_t GetRunNumber() { return fRunNumber; }
  64   TH3F* GetTwoTrackDistance(Int_t i) { return fTwoTrackDistancePt[i]; }
  65
  66   void Correct(AliUEHistograms* corrections);
  67
  68   void SetEtaRange(Float_t etaMin, Float_t etaMax);
  69   void SetPtRange(Float_t ptMin, Float_t ptMax);
  70   void SetZVtxRange(Float_t min, Float_t max);
  71   void SetContaminationEnhancement(TH1F* hist);
  72   void SetCombineMinMax(Bool_t flag);
  73   void SetSelectCharge(Int_t selectCharge) { fSelectCharge = selectCharge; }
  74   void SetTriggerRestrictEta(Float_t eta) { fTriggerRestrictEta = eta; }
  75   void SetEtaOrdering(Bool_t flag) { fEtaOrdering = flag; }
  76   void SetPairCuts(Bool_t conversions, Bool_t resonances) { fCutConversions = conversions; fCutResonances = resonances; }
  77   void SetOnlyOneEtaSide(Int_t flag)    { fOnlyOneEtaSide = flag; }
  78
  79   void ExtendTrackingEfficiency(Bool_t verbose = kFALSE);
  80   void Reset();
  81
  82   AliUEHistograms(const AliUEHistograms &c);
  83   AliUEHistograms& operator=(const AliUEHistograms& c);
  84   virtual void Copy(TObject& c) const;
  85
  86   virtual Long64_t Merge(TCollection* list);
  87   void Scale(Double_t factor);
  88
  89 protected:
  90   void FillRegion(AliUEHist::Region region, Float_t zVtx, AliUEHist::CFStep step, AliVParticle* leading, TList* list, Int_t multiplicity);
  91   Int_t CountParticles(TList* list, Float_t ptMin);
  92   void DeleteContainers();
  93   Float_t GetInvMassSquared(Float_t pt1, Float_t eta1, Float_t phi1, Float_t pt2, Float_t eta2, Float_t phi2, Float_t m0);
  94   inline Float_t GetDPhiStar(Float_t phi1, Float_t pt1, Float_t charge1, Float_t phi2, Float_t pt2, Float_t charge2, Float_t radius, Float_t bSign);
  95
  96   static const Int_t fgkUEHists; // number of histograms
  97
  98   AliUEHist* fNumberDensitypT;   // d^2N/dphideta vs pT,lead
  99   AliUEHist* fSumpT;             // d^2 sum(pT)/dphideta vs pT,lead
 100   AliUEHist* fNumberDensityPhi;  // d^2N/dphideta vs delta phi,lead (in pT,lead bins)
 101
 102   TH2F* fCorrelationpT;         // pT,lead: true vs reco
 103   TH2F* fCorrelationEta;        // #eta,lead; true vs reco
 104   TH2F* fCorrelationPhi;        // #phi,lead; true vs reco
 105   TH2F* fCorrelationR;          // R = sqrt(delta eta^2 + delta phi^2) (true vs reco) vs pT,lead,MC
 106   TH2F* fCorrelationLeading2Phi;// delta phi (true vs reco) vs pT,lead,MC
 107   TH2F* fCorrelationMultiplicity; // number of mc particls vs reco particles (for pT > 0.5 GeV/c)
 108
 109   TH2F* fEventCount;            // event count as function of step, (for pp: event type (plus additional step -1 for all events without vertex range even in MC)) (for PbPb: centrality)
 110   TH3F* fEventCountDifferential;// event count as function of leading pT, step, event type
 111
 112   TH1F* fVertexContributors;    // number of contributors to the vertex
 113   TH1F* fCentralityDistribution; // distribution of the variable used for centrality selection
 114   TH2F* fCentralityCorrelation;  // centrality vs multiplicity
 115
 116   TH3F* fITSClusterMap;          // its cluster map vs centrality vs pT
 117
 118   TH3F* fTwoTrackDistancePt[2];    // control histograms for two-track efficiency study: dphi*_min vs deta (0 = before cut, 1 = after cut)
 119
 120   Int_t fSelectCharge;           // (un)like sign selection when building correlations: 0: no selection; 1: unlike sign; 2: like sign
 121   Float_t fTriggerRestrictEta;   // restrict eta range for trigger particle (default: -1 [off])
 122   Bool_t fEtaOrdering;           // activate eta ordering to prevent shape distortions. see FillCorrelation for the details
 123   Bool_t fCutConversions;        // cut on conversions (inv mass)
 124   Bool_t fCutResonances;         // cut on resonances (inv mass)
 125   Int_t fOnlyOneEtaSide;       // decides that only trigger particle from one eta side are considered (0 = all; -1 = negative, 1 = positive)
 126
 127   Long64_t fRunNumber;           // run number that has been processed
 128
 129   ClassDef(AliUEHistograms, 14)  // underlying event histogram container
 130 };
 131
 132 Float_t AliUEHistograms::GetDPhiStar(Float_t phi1, Float_t pt1, Float_t charge1, Float_t phi2, Float_t pt2, Float_t charge2, Float_t radius, Float_t bSign)
 133 {
 134   //
 135   // calculates dphistar
 136   //
 137
 138   Float_t dphistar = phi1 - phi2 - charge1 * bSign * TMath::ASin(0.075 * radius / pt1) + charge2 * bSign * TMath::ASin(0.075 * radius / pt2);
 139
 140   static const Double_t kPi = TMath::Pi();
 141
 142   // circularity
 143 //   if (dphistar > 2 * kPi)
 144 //     dphistar -= 2 * kPi;
 145 //   if (dphistar < -2 * kPi)
 146 //     dphistar += 2 * kPi;
 147
 148   if (dphistar > kPi)
 149     dphistar = kPi * 2 - dphistar;
 150   if (dphistar < -kPi)
 151     dphistar = -kPi * 2 - dphistar;
 152   if (dphistar > kPi) // might look funny but is needed
 153     dphistar = kPi * 2 - dphistar;
 154
 155   return dphistar;
 156 }
 157
 158 #endif