PWGLF/STRANGENESS/LambdaK0/AliAnalysisTaskExtractV0.h

   1 #ifndef ALIANALYSISTASKEXTRACTV0_H
   2 #define ALIANALYSISTASKEXTRACTV0_H
   3
   4 /**************************************************************************
   5  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   6  *                                                                        *
   7  * Author: The ALICE Off-line Project.                                    *
   8  * Contributors are mentioned in the code where appropriate.              *
   9  *                                                                        *
  10  * Permission to use, copy, modify and distribute this software and its   *
  11  * documentation strictly for non-commercial purposes is hereby granted   *
  12  * without fee, provided that the above copyright notice appears in all   *
  13  * copies and that both the copyright notice and this permission notice   *
  14  * appear in the supporting documentation. The authors make no claims     *
  15  * about the suitability of this software for any purpose. It is          *
  16  * provided "as is" without express or implied warranty.                  *
  17  **************************************************************************/
  18
  19 //-----------------------------------------------------------------
  20 //      AliAnalysisTaskExtractV0 class
  21 //      ------------------------------
  22 //
  23 //    Please see cxx file for more details.
  24 //
  25 //-----------------------------------------------------------------
  26
  27 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
  28 //
  29 // --- This version: 23rd March 2012
  30 //
  31 // +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
  32
  33 class TList;
  34 class TH1F;
  35 class TH2F;
  36 class TH3F;
  37 class TVector3;
  38 class THnSparse;
  39
  40 class AliESDpid;
  41 class AliESDtrackCuts;
  42 class AliAnalysisUtils;
  43 class AliESDEvent;
  44 class AliPhysicsSelection;
  45 class AliCFContainer;
  46
  47 //#include "TString.h"
  48 //#include "AliESDtrackCuts.h"
  49 #include "AliAnalysisTaskSE.h"
  50
  51 class AliAnalysisTaskExtractV0 : public AliAnalysisTaskSE {
  52  public:
  53         AliAnalysisTaskExtractV0();
  54         AliAnalysisTaskExtractV0(const char *name);
  55         virtual ~AliAnalysisTaskExtractV0();
  56
  57         virtual void     UserCreateOutputObjects();
  58         virtual void     UserExec(Option_t *option);
  59         virtual void     Terminate(Option_t *);
  60   void CheckChargeV0(AliESDv0 *thisv0);
  61
  62 //---------------------------------------------------------------------------------------
  63
  64   void SetIsNuclear           (Bool_t lIsNuclear   = kTRUE ) { fkIsNuclear   = lIsNuclear;   }
  65   void SetINT7Trigger         (Bool_t lSwitchINT7  = kTRUE ) { fkSwitchINT7  = lSwitchINT7; }
  66   void SetUseOnTheFly         (Bool_t lUseOnTheFly = kTRUE ) { fkUseOnTheFly = lUseOnTheFly; }
  67   void SetTakeAllTracks       (Bool_t lTakeAllTracks = kTRUE ) { fkTakeAllTracks = lTakeAllTracks; }
  68   void SetCentralityEstimator (TString lCentralityEstimator = "V0M" ) { fCentralityEstimator = lCentralityEstimator; }
  69   void SetLightWeightAnalysis (Bool_t lLightWeight = kTRUE) {fkLightWeight = lLightWeight;  }
  70   void SetFastOnly (TString lFastOnly = "kFastOnly") {fkFastOnly = lFastOnly;  }
  71   void SetpAVertexSelection   (Bool_t lpAVertexSelection = kTRUE) {fkpAVertexSelection = lpAVertexSelection;  }
  72   void SetRunV0Vertexer ( Bool_t lRunV0Vertexer = kTRUE) { fkRunV0Vertexer = lRunV0Vertexer; }
  73   void SetRejectPileup ( Bool_t lRejectPileup = kTRUE) { fkRejectPileup = lRejectPileup; }
  74   void SetSpecialExecution ( Bool_t lSpecialExecution = kTRUE) { fkSpecialExecution = lSpecialExecution; }
  75   void SetSkipTrigger ( Bool_t lSkipTrigger = kTRUE ){ fkSkipTrigger = lSkipTrigger; }
  76
  77 //---------------------------------------------------------------------------------------
  78 //Setters for the V0 Vertexer Parameters
  79   void SetV0VertexerMaxChisquare   ( Double_t lParameter ){ fV0Sels[0] = lParameter; }
  80   void SetV0VertexerDCAFirstToPV   ( Double_t lParameter ){ fV0Sels[1] = lParameter; }
  81   void SetV0VertexerDCASecondtoPV  ( Double_t lParameter ){ fV0Sels[2] = lParameter; }
  82   void SetV0VertexerDCAV0Daughters ( Double_t lParameter ){ fV0Sels[3] = lParameter; }
  83   void SetV0VertexerCosinePA       ( Double_t lParameter ){ fV0Sels[4] = lParameter; }
  84   void SetV0VertexerMinRadius      ( Double_t lParameter ){ fV0Sels[5] = lParameter; }
  85   void SetV0VertexerMaxRadius      ( Double_t lParameter ){ fV0Sels[6] = lParameter; }
  86 //---------------------------------------------------------------------------------------
  87
  88   void SetTPCdEdxSelection ( Bool_t lTPCdEdxSelection = kTRUE ) { fTPCdEdxSelection = lTPCdEdxSelection; }
  89   void SetEtaRefMult ( Double_t lEtaRefMult = 0.5 ) { fEtaRefMult = lEtaRefMult; }
  90
  91  private:
  92                                 // Note : In ROOT, "//!" means "do not stream the data from Master node to Worker node" ...
  93                                 // your data member object is created on the worker nodes and streaming is not needed.
  94                                 // http://root.cern.ch/download/doc/11InputOutput.pdf, page 14
  95         TList   *fListHistV0;   //! List of output objects
  96         TTree   *fTree;                                                 //! Output Tree
  97
  98   AliPIDResponse *fPIDResponse;     // PID response object
  99   AliESDtrackCuts *fESDtrackCuts;   // ESD track cuts used for primary track definition
 100   AliAnalysisUtils *fUtils;         // analysis utils (for pA vertex selection)
 101
 102   //Objects Controlling Task Behaviour
 103
 104   Bool_t fkIsNuclear;   // if true, replace multiplicity est. by centrality (default FALSE)
 105   Bool_t fkSwitchINT7; // if true, skip FASTOnly (default FALSE)
 106   Bool_t fkUseOnTheFly; // if true, will use On-the-fly V0s instead of Offline V0s (default FALSE)
 107   Bool_t fkTakeAllTracks; // if true, no TPC crossed rows and ratio cut
 108   TString fCentralityEstimator; //Centrality Estimator String value (default V0M, DEPRECATED)
 109   Bool_t fkLightWeight; // if true, analysis output will exclude some non-fundamental
 110                         // debugging information. This creates smaller output.
 111   TString fkFastOnly; //"" if no extra selection, "kFastOnly" -> without SDD, "NotkFastOnly" -> With SDD
 112   Bool_t fkpAVertexSelection; //if true, select vertex with pPb Methods
 113   Bool_t fkRunV0Vertexer; //if true, re-run vertexer with loose cuts. CARE MUST BE TAKEN in PbPb!
 114   Bool_t fkRejectPileup; //Reject pileup or not
 115   Bool_t fkSpecialExecution; //Special Debug / Exploratory mode
 116   Bool_t fkSkipTrigger; //To be used with ::SetCollisionCandidates
 117
 118   Double_t        fV0Sels[7];                     // Array to store the 7 values for the different selections V0 related
 119 //  Double_t fEtaRefMult; //Reference multiplicity eta
 120
 121   //Extra selections
 122   Bool_t fTPCdEdxSelection; //Configuration to apply extra TPC dE/dx selection for better filling of tree
 123   Double_t fEtaRefMult; //Reference multiplicity eta
 124
 125   //Variables for Tree
 126         Float_t fTreeVariableChi2V0;         //!
 127         Float_t fTreeVariableDcaV0Daughters; //!
 128         Float_t fTreeVariableDcaV0ToPrimVertex; //!
 129         Float_t fTreeVariableDcaPosToPrimVertex; //!
 130         Float_t fTreeVariableDcaNegToPrimVertex; //!
 131         Float_t fTreeVariableV0CosineOfPointingAngle; //!
 132         Float_t fTreeVariableV0Radius; //!
 133         Float_t fTreeVariablePt; //!
 134         Float_t fTreeVariableRapK0Short; //!
 135         Float_t fTreeVariableRapLambda; //!
 136         Float_t fTreeVariableInvMassK0s; //!
 137         Float_t fTreeVariableInvMassLambda; //!
 138         Float_t fTreeVariableInvMassAntiLambda; //!
 139         Float_t fTreeVariableAlphaV0; //!
 140         Float_t fTreeVariablePtArmV0;//!
 141         Float_t fTreeVariableNegTotMomentum; //!
 142         Float_t fTreeVariablePosTotMomentum; //!
 143         Float_t fTreeVariableNegdEdxSig; //!
 144         Float_t fTreeVariablePosdEdxSig; //!
 145         Float_t fTreeVariableNegEta; //!
 146         Float_t fTreeVariablePosEta; //!
 147
 148         Float_t fTreeVariableNSigmasPosProton; //!
 149         Float_t fTreeVariableNSigmasPosPion; //!
 150         Float_t fTreeVariableNSigmasNegProton; //!
 151         Float_t fTreeVariableNSigmasNegPion; //!
 152
 153         Float_t fTreeVariableDistOverTotMom;//!
 154         Int_t   fTreeVariableLeastNbrCrossedRows;//!
 155         Float_t fTreeVariableLeastRatioCrossedRowsOverFindable;//!
 156         Int_t   fTreeVariableMultiplicity ;//!
 157   Int_t   fTreeVariableMultiplicityV0A ;//!
 158   Int_t   fTreeVariableMultiplicityZNA ;//!
 159   Int_t   fTreeVariableMultiplicityTRK ;//!
 160   Int_t   fTreeVariableMultiplicitySPD ;//!
 161
 162   Int_t   fTreeVariableRunNumber; //!
 163   ULong64_t fTreeVariableEventNumber; //!
 164
 165   Float_t fTreeVariableV0x; //!
 166   Float_t fTreeVariableV0y; //!
 167   Float_t fTreeVariableV0z; //!
 168
 169   Float_t fTreeVariableV0Px; //!
 170   Float_t fTreeVariableV0Py; //!
 171   Float_t fTreeVariableV0Pz; //!
 172
 173   Float_t fTreeVariablePVx; //!
 174   Float_t fTreeVariablePVy; //!
 175   Float_t fTreeVariablePVz; //!
 176
 177   //Decay Length issue debugging: ULong_t with track status
 178   ULong64_t fTreeVariableNegTrackStatus; //!
 179   ULong64_t fTreeVariablePosTrackStatus; //!
 180
 181   //Special dEdx debug/plotting
 182   Float_t fTreeVariableNegTPCSignal; //!
 183   Float_t fTreeVariablePosTPCSignal; //!
 184   Float_t fTreeVariableNegInnerP; //!
 185   Float_t fTreeVariablePosInnerP; //!
 186
 187   Float_t fTreeVariableNegPx; //!
 188   Float_t fTreeVariableNegPy; //!
 189   Float_t fTreeVariableNegPz; //!
 190   Float_t fTreeVariablePosPx; //!
 191   Float_t fTreeVariablePosPy; //!
 192   Float_t fTreeVariablePosPz; //!
 193
 194
 195
 196 //Note: TDistOverTotMom needs a mass hypothesis to be converted to proper decaylength.
 197
 198 //---------------------------------------------------------------------------------------
 199
 200         TH1F    *fHistV0MultiplicityBeforeTrigSel;             //! V0 multiplicity distribution
 201         TH1F    *fHistV0MultiplicityForTrigEvt;                //! V0 multiplicity distribution
 202         TH1F    *fHistV0MultiplicityForSelEvt;                 //! V0 multiplicity distribution
 203         TH1F    *fHistV0MultiplicityForSelEvtNoTPCOnly;        //! V0 multiplicity distribution
 204         TH1F    *fHistV0MultiplicityForSelEvtNoTPCOnlyNoPileup;//! V0 multiplicity distribution
 205
 206         TH1F    *fHistMultiplicityBeforeTrigSel;                //! multiplicity distribution
 207         TH1F    *fHistMultiplicityForTrigEvt;                   //! multiplicity distribution
 208         TH1F    *fHistMultiplicity;                                                     //! multiplicity distribution
 209         TH1F    *fHistMultiplicityNoTPCOnly;                            //! multiplicity distribution
 210         TH1F    *fHistMultiplicityNoTPCOnlyNoPileup;                    //! multiplicity distribution
 211
 212   //V0A Centrality
 213   TH1F    *fHistMultiplicityV0ABeforeTrigSel;           //! multiplicity distribution
 214         TH1F    *fHistMultiplicityV0AForTrigEvt;                        //! multiplicity distribution
 215         TH1F    *fHistMultiplicityV0A;                                                  //! multiplicity distribution
 216         TH1F    *fHistMultiplicityV0ANoTPCOnly;                         //! multiplicity distribution
 217         TH1F    *fHistMultiplicityV0ANoTPCOnlyNoPileup;                 //! multiplicity distribution
 218
 219   //ZNA Centrality
 220   TH1F    *fHistMultiplicityZNABeforeTrigSel;           //! multiplicity distribution
 221         TH1F    *fHistMultiplicityZNAForTrigEvt;                        //! multiplicity distribution
 222         TH1F    *fHistMultiplicityZNA;                                                  //! multiplicity distribution
 223         TH1F    *fHistMultiplicityZNANoTPCOnly;                         //! multiplicity distribution
 224         TH1F    *fHistMultiplicityZNANoTPCOnlyNoPileup;                 //! multiplicity distribution
 225
 226   //TRK Centrality
 227   TH1F    *fHistMultiplicityTRKBeforeTrigSel;           //! multiplicity distribution
 228         TH1F    *fHistMultiplicityTRKForTrigEvt;                        //! multiplicity distribution
 229         TH1F    *fHistMultiplicityTRK;                                                  //! multiplicity distribution
 230         TH1F    *fHistMultiplicityTRKNoTPCOnly;                         //! multiplicity distribution
 231         TH1F    *fHistMultiplicityTRKNoTPCOnlyNoPileup;                 //! multiplicity distribution
 232
 233   //SPD Centrality
 234   TH1F    *fHistMultiplicitySPDBeforeTrigSel;           //! multiplicity distribution
 235         TH1F    *fHistMultiplicitySPDForTrigEvt;                        //! multiplicity distribution
 236         TH1F    *fHistMultiplicitySPD;                                                  //! multiplicity distribution
 237         TH1F    *fHistMultiplicitySPDNoTPCOnly;                         //! multiplicity distribution
 238         TH1F    *fHistMultiplicitySPDNoTPCOnlyNoPileup;                 //! multiplicity distribution
 239
 240 //---------------------------------------------------------------------------------------
 241
 242   //Raw Data for Vertex Z position estimator change
 243         TH2F    *f2dHistMultiplicityVsVertexZBeforeTrigSel;             //! multiplicity distribution
 244         TH2F    *f2dHistMultiplicityVsVertexZForTrigEvt;                        //! multiplicity distribution
 245         TH2F    *f2dHistMultiplicityVsVertexZ;                                                  //! multiplicity distribution
 246         TH2F    *f2dHistMultiplicityVsVertexZNoTPCOnly;                         //! multiplicity distribution
 247         TH2F    *f2dHistMultiplicityVsVertexZNoTPCOnlyNoPileup;                 //! multiplicity distribution
 248
 249         TH1F    *fHistPVx;                                              //! multiplicity distribution
 250         TH1F    *fHistPVy;                                              //! multiplicity distribution
 251         TH1F    *fHistPVz;                                              //! multiplicity distribution
 252         TH1F    *fHistPVxAnalysis;                                              //! multiplicity distribution
 253         TH1F    *fHistPVyAnalysis;                                              //! multiplicity distribution
 254         TH1F    *fHistPVzAnalysis;                                              //! multiplicity distribution
 255         TH1F    *fHistSwappedV0Counter;                                                 //! Swapped V0 Counter
 256
 257    AliAnalysisTaskExtractV0(const AliAnalysisTaskExtractV0&);            // not implemented
 258    AliAnalysisTaskExtractV0& operator=(const AliAnalysisTaskExtractV0&); // not implemented
 259
 260    ClassDef(AliAnalysisTaskExtractV0, 11);
 261 };
 262
 263 #endif