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) {;}
  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 SetCentralityEstimator(const char * estimator) { fCentrEstimator = estimator; }
  47   void SetCentralityBin(Int_t bin) { fCentrBin = bin; }
  48   void SetCentrTaskFiles(const char * file1, const char * file2) { fFile1 = file1; fFile2 = file2; }
  49   virtual void Print(Option_t* option = "") const ;
  50   virtual Long64_t Merge(TCollection* list){list->GetEntries();return 0;}
  51
  52 protected:
  53   Bool_t fIsMC;             // flag if MC is analyzed
  54   TString fCentrEstimator;  // Centrality estimator for AliESDCentrality
  55   Int_t   fCentrBin; // centrality bin to be selected
  56   Int_t fMultMin ; // Minimum multiplicity, because on MC we cut on tracks rather than on the estimator
  57   Int_t fMultMax ; // Maximum multiplicity, because on MC we cut on tracks rather than on the estimator
  58   TString fFile1; // file used by centrality task. Set here for bookkeeping
  59   TString fFile2; // file used by centrality task. Set here for bookkeeping
  60   Bool_t fUseMultRange; // if true, use track bins rather than multiplicity estimator
  61
  62   ClassDef(AliAnalysisMultPbCentralitySelector, 2)
  63
  64   private:
  65   AliAnalysisMultPbCentralitySelector(const AliAnalysisMultPbCentralitySelector&); // not implemented
  66   AliAnalysisMultPbCentralitySelector& operator=(const AliAnalysisMultPbCentralitySelector&); // not implemented
  67 };
  68
  69 #endif