PWGUD/multVScentPbPb/AliTrackletTaskMultipp.h

   1 #ifndef ALITRACKLETTASKMULTIPP_H
   2 #define ALITRACKLETTASKMULTIPP_H
   3
   4 ///////////////////////////////////////////////////////////////////////////
   5 // Class AliTrackletTaskMultipp                                            //
   6 // Analysis task to produce data and MC histos needed for tracklets      //
   7 // dNdEta extraction in multiple bins in one go                          //
   8 // Author:  ruben.shahoyan@cern.ch                                       //
   9 ///////////////////////////////////////////////////////////////////////////
  10
  11 class TH1F;
  12 class TH2F;
  13 class TH3F;
  14 class AliESDEvent;
  15 class TList;
  16 class TNtuple;
  17
  18 class AliMCParticle;
  19 class AliITSMultRecBg;
  20 class AliESDTrackCuts;
  21 class AliITSsegmentationSPD;
  22 #include "AliAnalysisTaskSE.h"
  23 #include "AliTriggerAnalysis.h"
  24 #include <TMath.h>
  25
  26 class AliTrackletTaskMultipp : public AliAnalysisTaskSE {
  27  public:
  28   enum {kData,kBgInj,kBgRot,kBgMix,kMC};
  29   enum {kCentV0M,kCentFMD,kCentTRK,kCentTKL,kCentCL0,kCentCL1,kCentV0MvsFMD,kCentTKLvsV0,kCentZEMvsZDC,kNCentTypes}; // what is used to define centrality
  30   //
  31   enum {  // define here id's of the standard histos in corresponding TObjArray* fHistosTr...
  32     kHEtaZvCut,       // histo zv vs eta for tracklets passing final selection (dist<1 or |dPhi|<narrowWindow ...)
  33     kHDPhiDTheta,     // measured dTheta vs dPhi
  34     kHDPhiSDThetaX,   // dTheta (1/sin^2 scaled if needed) vs dPhi (bending subtracted)
  35     kHWDist,          // Weighted distance
  36     kHEtaZvSPD1,      // histo zv vs eta for SPD1 single clusters
  37     kNStandardH       // number of standard histos per centrality bin
  38   };
  39   enum { // define here id's of any custom histos to be added to fHistosCustom
  40     kHStat,            // job info (meaning of bins defined in the enum below)
  41     //
  42     kHStatCent,        // events per centrality bin with real values on the axis
  43     kHStatCentBin,     // events per centrality bin
  44     //
  45     kHNPrimMeanMC,     // <n> primaries per mult bin
  46     kHNPrim2PartMC,    // <n> prim per part.pair per mult bin
  47     kHNPrim2BCollMC,   // <n> prim per bin.coll per mult bin
  48     kHNPrim2PartNpMC,  // <n> prim per n part vs npart
  49     kHNPrim2BCollNpMC, // <n> prim per n part vs npart
  50     kHNPartMC,         // n.part.pairs according to MC
  51     kHNPartMeanMC,     // <n> part pairs per mult bin
  52     kHNBCollMC,        // n.bin.colls according to MC
  53     kHNBCollMeanMC,
  54     //
  55     kHZVtxNoSel,       // Z vertex distribution before event selection
  56     kHV0NoSel,         // V0 before selection
  57     kHNClSPD2NoSel,    // NSPD2 before selection
  58     kHZDCZEMNoSel,     // ZDC ZEM before selection
  59     //
  60     kHZVtx,            // Z vertex distribution
  61     kHV0,              // V0 before selection
  62     kHNClSPD2,         // NSPD2 before selection
  63     kHZDCZEM,          // ZDC ZEM before selection
  64
  65
  66     kHZVtxMixDiff,     // difference in Z vtx of mixed events
  67     kHNTrMixDiff,      // difference in N tracklets of mixed events
  68     //
  69     kHPrimPDG,         // PDG code of prim tracklet
  70     kHSecPDG,          // PDG code of sec tracklet
  71     kHPrimParPDG,      // PDG code of prim tracklet parent
  72     kHSecParPDG,       // PDG code of sec tracklet parent
  73     //
  74     kHClUsedInfoL0,    // used clusters of lr0
  75     kHClUsedInfoL1,    // used clusters of lr1
  76     kHClAllInfoL0,     // all clusters of lr0
  77     kHClAllInfoL1,     // all clusters of lr1
  78     //
  79     //-------------------------------------- tmp
  80     kHclDstZAll,       // Distance between SPD1 clusters vs Z, all cl
  81     kHclDstPhiAll,     // Distance between SPD1 clusters vs phi, all cl
  82     kHclDstZUsed,      // Distance between SPD1 clusters vs Z, used-unused cl
  83     kHclDstPhiUsed,    // Distance between SPD1 clusters vs phi, used-unused cl
  84     //--------------------------------------
  85     // This MUST be last one: this is just beginning of many histos (one per bin)
  86     kHZVEtaPrimMC      // Zv vs eta for all primary tracks (true MC multiplicity)
  87   }; // custom histos
  88
  89   // bins for saved parameters
  90   enum {kDummyBin,
  91         kEvTot0,      // events read
  92         kEvTot,       // events read after vertex quality selection
  93         kOneUnit,     // just 1 to track primate merges
  94         kNWorkers,    // n workers
  95         //
  96         kCentVar,     // cetrality var. used
  97         kDPhi,        // dphi window
  98         kDTht,        // dtheta window
  99         kNStd,        // N.standard deviations to keep
 100         kPhiShift,    // bending shift
 101         kThtS2,       // is dtheta scaled by 1/sin^2
 102         kThtCW,       // on top of w.dist cut cut also on 1 sigma dThetaX
 103         kPhiOvl,      // overlap params
 104         kZEtaOvl,     // overlap params
 105         kNoOvl,       // flag that overlap are suppressed
 106         //
 107         kPhiRot,      // rotation phi
 108         kInjScl,      // injection scaling
 109         kEtaMin,      // eta cut
 110         kEtaMax,      // eta cut
 111         kZVMin,       // min ZVertex to process
 112         kZVMax,       // max ZVertex to process
 113         //
 114         kDPiSCut,     // cut on dphi used to extract signal (when WDist is used in analysis, put it equal to kDPhi
 115         kNStdCut,     // cut on weighted distance (~1) used to extract signal
 116         //
 117         kMCV0Scale,   // scaling value for V0 in MC
 118         //
 119         // here we put entries for each mult.bin
 120         kBinEntries = 50,
 121         kEvProcData,  // events with data mult.object (ESD or reco)
 122         kEvProcInj,   // events Injected, total
 123         kEvProcRot,   // events Rotated
 124         kEvProcMix,   // events Mixed
 125         kEntriesPerBin
 126   };
 127
 128   //
 129   AliTrackletTaskMultipp(const char *name = "AliTrackletTaskMultipp");
 130   virtual ~AliTrackletTaskMultipp();
 131
 132   virtual void  UserCreateOutputObjects();
 133   virtual void  UserExec(Option_t *option);
 134   virtual void  Terminate(Option_t *);
 135
 136   void       SetUseCentralityVar(Int_t v=kCentV0M)     {fUseCentralityVar = v;}
 137   void       SetUseMC(Bool_t mc = kFALSE)              {fUseMC = mc;}
 138   void       SetCheckReconstructables(Bool_t c=kFALSE) {fCheckReconstructables = c;}
 139   TObjArray* BookHistosSet(const char* pref, UInt_t selHistos=0xffffffff);
 140   TObjArray* BookCustomHistos();
 141   void       AddHisto(TObjArray* histos, TObject* h, Int_t at=-1);
 142   void       FillHistosSet(TObjArray* histos, double eta, /*double phi,double theta,*/double dphi,double dtheta,double dthetaX,double dist);
 143   // RS
 144   void       SetNStdDev(Float_t f=1.)           {fNStdDev = f<1e-5 ? 1e-5:f;}
 145   void       SetScaleDThetaBySin2T(Bool_t v=kFALSE) {fScaleDTBySin2T = v;}
 146   void       SetCutOnDThetaX(Bool_t v=kFALSE)   {fCutOnDThetaX = v;}
 147   void       SetPhiWindow(float w=0.08)         {fDPhiWindow   = w<1e-5 ? 1e-5:w;}
 148   void       SetThetaWindow(float w=0.025)      {if (w<0) fCutOnDThetaX=kTRUE; fDThetaWindow = TMath::Abs(w)<1e-5 ? 1e-5:TMath::Abs(w);}
 149   void       SetPhiShift(float w=0.0045)        {fDPhiShift = w;}
 150   void       SetPhiOverlapCut(float w=0.005)    {fPhiOverlapCut = w;}
 151   void       SetZetaOverlapCut(float w=0.05)    {fZetaOverlap = w;}
 152   void       SetPhiRot(float w=0)               {fPhiRot = w;}
 153   void       SetInjScale(Float_t s=1.)          {fInjScale = s>0? s:1.;}
 154   void       SetRemoveOverlaps(Bool_t w=kFALSE) {fRemoveOverlaps = w;}
 155   //
 156   void       SetDPhiSCut(Float_t c=0.06)        {fDPhiSCut = c;}
 157   void       SetNStdCut(Float_t c=1.0)          {fNStdCut = c;}
 158   void       SetScaleMCV0(Float_t s=1.0)        {fMCV0Scale = s;}
 159   //
 160   void       SetEtaCut(Float_t etaCut)          {fEtaMax = TMath::Abs(etaCut); fEtaMin= -fEtaMax;}
 161   void       SetEtaMin(Float_t etaMin)          {fEtaMin = etaMin;}
 162   void       SetEtaMax(Float_t etaMax)          {fEtaMax = etaMax;}
 163   void       SetZVertexMin(Float_t z)           {fZVertexMin = z;}
 164   void       SetZVertexMax(Float_t z)           {fZVertexMax = z;}
 165   //
 166   Bool_t     GetDoNormalReco()             const {return fDoNormalReco;}
 167   Bool_t     GetDoInjection()              const {return fDoInjection;}
 168   Bool_t     GetDoRotation()               const {return fDoRotation;}
 169   Bool_t     GetDoMixing()                 const {return fDoMixing;}
 170   //
 171   void       SetDoNormalReco(Bool_t v=kTRUE)    {fDoNormalReco = v;}
 172   void       SetDoInjection(Bool_t v=kTRUE)     {fDoInjection = v;}
 173   void       SetDoRotation(Bool_t v=kTRUE)      {fDoRotation = v;}
 174   void       SetDoMixing(Bool_t v=kTRUE)        {fDoMixing = v;}
 175   //
 176   //
 177  protected:
 178   void       InitMultReco();
 179   Bool_t     HaveCommonParent(const float* clLabs0,const float* clLabs1);
 180   void       FillHistos(Int_t type, const AliMultiplicity* mlt);
 181   void       FillMCPrimaries();
 182   void       FillSpecies(Int_t primsec, Int_t id);
 183   void       FillClusterInfo();
 184   void       FillClusterInfoFromMult(const AliMultiplicity* mlt, double zVertex);
 185   void       FillClusterAutoCorrelationFromMult(const AliMultiplicity* mlt, double zVertex);
 186   Int_t      GetPdgBin(Int_t pdgCode);
 187   void       CheckReconstructables();
 188   Int_t      GetCentralityBin(Float_t percentile) const;
 189   //
 190  protected:
 191   TList*       fOutput;                   // output list send on output slot 1
 192   //
 193   Bool_t       fDoNormalReco;              // do normal reco
 194   Bool_t       fDoInjection;               // do injection
 195   Bool_t       fDoRotation;                // do rotation
 196   Bool_t       fDoMixing;                  // do mixing
 197   //
 198   Bool_t       fUseMC;                     // flag of MC processing
 199   Bool_t       fCheckReconstructables;     // request check
 200   //
 201   TObjArray*   fHistosTrData;              //! all tracklets in data
 202   TObjArray*   fHistosTrInj;               //! injected
 203   TObjArray*   fHistosTrRot;               //! rotated
 204   TObjArray*   fHistosTrMix;               //! mixed
 205   //
 206   TObjArray*   fHistosTrPrim;              //! primary
 207   TObjArray*   fHistosTrSec;               //! secondary
 208   TObjArray*   fHistosTrComb;              //! combinatorials
 209   TObjArray*   fHistosTrCombU;             //! combinatorials uncorrelated
 210   //
 211   TObjArray*   fHistosTrRcblPrim;          //! Primary Reconstructable
 212   TObjArray*   fHistosTrRcblSec;           //! Secondary Reconstructable
 213   TObjArray*   fHistosCustom;              //! custom histos
 214   //
 215   // Settings for the reconstruction
 216   // tracklet reco settings
 217   Float_t      fEtaMin;                    // histos filled only for this eta range
 218   Float_t      fEtaMax;                    // histos filled only for this eta range
 219   Float_t      fZVertexMin;                // min Z vtx to process
 220   Float_t      fZVertexMax;                // max Z vtx to process
 221   //
 222   Bool_t       fScaleDTBySin2T;            // request dTheta scaling by 1/sin^2(theta)
 223   Bool_t       fCutOnDThetaX;              // if true, apart from NStdDev cut apply also the cut on dThetaX
 224   Float_t      fNStdDev;                   // cut on weighted distance
 225   Float_t      fDPhiWindow;                // max dPhi
 226   Float_t      fDThetaWindow;              // max dTheta
 227   Float_t      fDPhiShift;                 // mean bend
 228   Float_t      fPhiOverlapCut;             // overlaps cut in phi
 229   Float_t      fZetaOverlap;               // overlaps cut in Z
 230   Float_t      fPhiRot;                    // rotate L1 wrt L2
 231   Float_t      fInjScale;                  // scaling factor for injection
 232   Bool_t       fRemoveOverlaps;            // request overlaps removal
 233   //
 234   Float_t      fDPhiSCut;                  // cut on signal dphiS
 235   Float_t      fNStdCut;                   // cut on signal weighted distance
 236   Float_t      fMCV0Scale;                 // scaling factor for V0 in MC
 237   //
 238   AliITSMultRecBg *fMultReco;              //! mult.reco object
 239   TTree*       fRPTree;                    //! tree of recpoints
 240   TTree*       fRPTreeMix;                 //! tree of recpoints for mixing
 241   AliStack*    fStack;                     //! MC stack
 242   AliMCEvent*  fMCevent;                   //! MC Event
 243   Float_t      fESDVtx[3];                 //  ESD vertex
 244   //
 245   Float_t fNPart;                          // number of participant pairs from MC
 246   Float_t fNBColl;                         // number of bin. collision from MC
 247   Int_t  fCurrCentBin;                     // current centrality bin
 248   Int_t  fNCentBins;                       // N of mult bins
 249   Int_t  fUseCentralityVar;                // what is used to determine the centrality
 250   //
 251   static const Float_t fgkCentPerc[];               //! centrality in percentiles
 252   //
 253   static const char*  fgkCentSelName[];             //!centrality types
 254   static const char*  fgkPDGNames[];                //!pdg names
 255   static const Int_t  fgkPDGCodes[];                //!pdg codes
 256   //
 257  private:
 258   AliTrackletTaskMultipp(const AliTrackletTaskMultipp&); // not implemented
 259   AliTrackletTaskMultipp& operator=(const AliTrackletTaskMultipp&); // not implemented
 260
 261   ClassDef(AliTrackletTaskMultipp, 1);
 262 };
 263
 264
 265 #endif