PWG0/multPbPb/AliAnalysisMultPbCentralitySelector.h

   1 #ifndef ALIANALYSISMULTPBCENTRALITYSELECTOR_H
   2 #define ALIANALYSISMULTPBCENTRALITYSELECTOR_H
   3
   4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 //-------------------------------------------------------------------------
   8 //                    AliAnalysisMultPbCentralitySelector
   9 //
  10 // This class selects collision candidates from data runs, applying selection cuts on triggers
  11 // and background rejection based on the content of the ESD
  12 //
  13 // Author: Michele Floris, CERN
  14 //-------------------------------------------------------------------------
  15
  16 #include <AliAnalysisCuts.h>
  17 #include <AliLog.h>
  18
  19 #define VERBOSE_STAT
  20
  21 class AliESDEvent;
  22 class TH2F;
  23 class TH1F;
  24 class TCollection;
  25 class AliTriggerAnalysis;
  26 class AliAnalysisTaskSE;
  27 class AliESDtrackCuts;
  28
  29 class AliAnalysisMultPbCentralitySelector : public AliAnalysisCuts
  30 {
  31 public:
  32
  33   AliAnalysisMultPbCentralitySelector() : fIsMC (0), fCentrEstimator(""), fCentrBin(-1), fMultMin(0), fMultMax(1000000), fFile1(""), fFile2(""), fUseMultRange(kFALSE), fUseV0CutRange(kFALSE) {;}
  34   virtual ~AliAnalysisMultPbCentralitySelector(){}
  35
  36   // AliAnalysisCuts interface
  37   virtual UInt_t GetSelectionMask(const TObject* obj) { return (UInt_t) IsCentralityBinSelected((AliESDEvent*) obj, NULL); }
  38   virtual Bool_t IsSelected(TList*) { AliFatal("Not implemented"); return kFALSE; }
  39   virtual Bool_t IsSelected(TObject* obj)  {return (UInt_t) IsCentralityBinSelected ( (AliESDEvent*) obj, NULL);}
  40
  41   Bool_t IsCentralityBinSelected(AliESDEvent* aEsd, AliESDtrackCuts * trackCuts);
  42
  43   void SetIsMC(Bool_t flag = kTRUE, Int_t multMin = 0, Int_t multMax=10000) { fIsMC = flag; fMultMin = multMin; fMultMax = multMax; }
  44   void SetMultRange(Int_t multMin = 0, Int_t multMax=10000) { fMultMin = multMin; fMultMax = multMax; }
  45   void SetUseMultRange(Bool_t flag = kTRUE) {fUseMultRange = flag;}
  46   void SetUseV0Range(Bool_t flag = kTRUE) {fUseV0CutRange = flag;}
  47   void SetCentralityEstimator(const char * estimator) { fCentrEstimator = estimator; }
  48   void SetCentralityBin(Int_t bin) { fCentrBin = bin; }
  49   void SetCentrTaskFiles(const char * file1, const char * file2) { fFile1 = file1; fFile2 = file2; }
  50   virtual void Print(Option_t* option = "") const ;
  51   virtual Long64_t Merge(TCollection* list){list->GetEntries();return 0;}
  52
  53 protected:
  54   Bool_t fIsMC;             // flag if MC is analyzed
  55   TString fCentrEstimator;  // Centrality estimator for AliESDCentrality
  56   Int_t   fCentrBin; // centrality bin to be selected
  57   Float_t fMultMin ; // Minimum multiplicity, because on MC we cut on tracks rather than on the estimator . Also used for other estimators
  58   Float_t fMultMax ; // Maximum multiplicity, because on MC we cut on tracks rather than on the estimator . Also used for other estimators
  59   TString fFile1; // file used by centrality task. Set here for bookkeeping
  60   TString fFile2; // file used by centrality task. Set here for bookkeeping
  61   Bool_t fUseMultRange; // if true, use track bins rather than multiplicity estimator
  62   Bool_t fUseV0CutRange; // if true, use v0 range rather than multiplicity estimator
  63
  64   ClassDef(AliAnalysisMultPbCentralitySelector, 2)
  65
  66   private:
  67   AliAnalysisMultPbCentralitySelector(const AliAnalysisMultPbCentralitySelector&); // not implemented
  68   AliAnalysisMultPbCentralitySelector& operator=(const AliAnalysisMultPbCentralitySelector&); // not implemented
  69 };
  70
  71 #endif