Merge branch 'feature-movesplit'

[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / ChargedHadrons / multPbPb / AliAnalysisTaskMultPbTracks.h

#ifndef ALIANALYSISTASKMULTPBTRACKS_H
#define ALIANALYSISTASKMULTPBTRACKS_H

#include "AliAnalysisTaskSE.h"
#include "AliESDtrackCuts.h" // if I don't include this, nothing compiles

//-------------------------------------------------------------------------
//                      AliAnalysisTaskMultPbTracks
// 
// 
//
//
// Author: Michele Floris, CERN
//-------------------------------------------------------------------------


class AliESDEvent;
class AliESDtrackCuts;
class AliAnalysisMultPbCentralitySelector;
class AliAnalysisMultPbTrackHistoManager;
class AliTriggerAnalysis;
class AliPIDResponse;

class AliAnalysisTaskMultPbTracks : public AliAnalysisTaskSE {

public:

  AliAnalysisTaskMultPbTracks();
  AliAnalysisTaskMultPbTracks(const char * name);
  AliAnalysisTaskMultPbTracks(const AliAnalysisTaskMultPbTracks& obj) ;
  ~AliAnalysisTaskMultPbTracks();
  void SetTrackCuts(AliESDtrackCuts * cuts) { fTrackCuts = cuts;}
  void SetCentralitySelector(AliAnalysisMultPbCentralitySelector * centr) { fCentrSelector=centr;}
  void SetTPCOnly(Bool_t flag=kTRUE) {fIsTPCOnly=flag;}
  // void SetCentralityBin(Int_t bin = 0) { fCentrBin = bin; }
  // void SetCentralityEstimator(const char * centr) { fCentralityEstimator = centr; }

  void SetIsMC(Bool_t flag=kTRUE) { fIsMC = flag;}
  AliAnalysisMultPbTrackHistoManager * GetHistoManager() { return fHistoManager;}
  Bool_t IsPhysicalPrimaryAndTransportBit(Int_t ipart) ;
  Bool_t IsElectron(AliESDtrack *  esdTrack);
  void RejectElectrons(Bool_t rejElectron) { fRejectElectrons = rejElectron;}
  void SetOfflineTrigger(UInt_t mask) { fOfflineTrigger = mask; }

  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *option);
  virtual void   Terminate(Option_t *);
  
  

private:

  //
  AliESDEvent *  fESD;    //! ESD object  AliVEvent*     fEvent;
  //  TList * fListHisto;     // list of output object
  AliAnalysisMultPbTrackHistoManager  * fHistoManager; // wrapper for the list, takes care of merging + histo booking and getters
  AliAnalysisMultPbCentralitySelector * fCentrSelector; // centrality selector
  // Int_t fCentrBin; // centrality bin selected (5% XS percentiles)
  // TString fCentralityEstimator; // Name of the centrality estimator, for AliESDCentrality
  AliESDtrackCuts * fTrackCuts; // track cuts
  AliESDtrackCuts * fTrackCutsNoDCA; // copy of the previous one, but with no DCA cuts
  UInt_t fOfflineTrigger ; // offline trigger mask
  Bool_t fIsMC; // true if processing montecarlo
  Bool_t fIsTPCOnly; // if you only want to use TPC tracks
  AliTriggerAnalysis * fTriggerAnalysis; // for offline triggers
  AliPIDResponse *fPIDResponse;     //! PID response object
  Bool_t fRejectElectrons; // reject electrons

  AliAnalysisTaskMultPbTracks& operator=(const AliAnalysisTaskMultPbTracks& task);
  
  ClassDef(AliAnalysisTaskMultPbTracks, 3)


};

#endif