move string data members to int; add more calo specific information: time, ncells...

[u/mrichter/AliRoot.git] / TOF / AliTOFAnalysisTaskCalibTree.h

#ifndef ALITOFANALYSISTASKCALIBTREE_H
#define ALITOFANALYSISTASKCALIBTREE_H

#include "AliAnalysisTaskSE.h"
#include "TString.h"
#include "TMath.h"

class AliESDEvent;
class AliESDtrackCuts;
class AliESDtrackCuts;
class AliESDtrack;
class AliESDVertex;
class AliESDpid;

class AliPhysicsSelection;

class AliTOFcalibHisto;
class AliTOFcalib;
class AliTOFT0maker;
class AliTOFT0v1;

class AliGRPManager;
class AliGRPObject;

class TH2F;

#define MAXHITS 100000

class AliTOFAnalysisTaskCalibTree :
public AliAnalysisTaskSE
{

 public:

  AliTOFAnalysisTaskCalibTree(const Char_t* name = "TOFcalibTree"); // default constructor
  virtual ~AliTOFAnalysisTaskCalibTree(); // default destructor

  virtual void UserCreateOutputObjects(); // user create output objects
  virtual void UserExec(Option_t *option); // user exec

  // getters
  AliPhysicsSelection *GetEventCuts() const {return fEventCuts;}; // getter
  AliESDtrackCuts *GetTrackCuts() const {return fTrackCuts;}; // getter
  AliTOFcalib *GetTOFcalib() const {return fTOFcalib;}; // getter
  AliTOFT0maker *GetTOFT0maker() const {return fTOFT0maker;}; // getter
  AliTOFT0v1 *GetTOFT0v1() const {return fTOFT0v1;}; // getter
  Float_t GetTimeResolution() const {return fTimeResolution;}; // time resolution
  AliESDpid *GetESDpid() const {return fESDpid;}; // get ESD PID

  // setters
  void SetEventSelectionFlag(Bool_t value = kTRUE) {fEventSelectionFlag = value;}; // setter
  void SetVertexSelectionFlag(Bool_t value = kTRUE) {fVertexSelectionFlag = value;}; // setter
  void SetVertexCut(Double_t value) {fVertexCut = value;}; // setter
  void SetDiscardPileupEventFlag(Bool_t value = kTRUE) {fDiscardPileupEventFlag = value;}; // setter
  void SetPrimaryDCASelectionFlag(Bool_t value = kTRUE) {fPrimaryDCASelectionFlag = value;}; // setter
  void SetTimeResolution(Float_t value) {fTimeResolution = value;}; // set time resolution
  void SetCalibrateTOFsignal(Bool_t value = kTRUE) {fCalibrateTOFsignal = value;}; // setter
  void SetComputeT0TOF(Bool_t value = kTRUE) {fComputeT0TOF = value;}; // setter
  void SetUseT0TOF(Bool_t value = kTRUE) {fUseT0TOF = value;}; // setter
  void SetUseLHCClockPhase(Bool_t value = kTRUE) {fUseLHCClockPhase = value;}; // setter
  void SetSpecificStorageParOffline(Char_t *value) {fSpecificStorageParOffline = value;}; // set specific storage ParOffline
  void SetSpecificStorageRunParams(Char_t *value) {fSpecificStorageRunParams = value;}; // set specific storage RunParams

 protected:

  AliTOFAnalysisTaskCalibTree(const AliTOFAnalysisTaskCalibTree &); // copy constructor
  AliTOFAnalysisTaskCalibTree &operator=(const AliTOFAnalysisTaskCalibTree &); // operator=

  // methods:

  Bool_t InitRun(); // init run
  Bool_t InitEvent(); // init event
  Bool_t HasTOFMeasurement(AliESDtrack *track); // has TOF
  Bool_t HasPrimaryDCA(AliESDtrack *track); // has primary DCA

  // members:

  // flags and cuts
  Bool_t fInitFlag;                   // init flag
  Bool_t fEventSelectionFlag;         // event selection flag
  Bool_t fVertexSelectionFlag;        // vertex selection flag
  Double_t fVertexCut;                // vertex cut
  Bool_t fDiscardPileupEventFlag;     // discard pile-up event flag
  Bool_t fCalibrateTOFsignal;         // calibrate TOF signal
  Bool_t fComputeT0TOF;               // compute T0-TOF
  Bool_t fUseT0TOF;                   // use T0-TOF
  Bool_t fUseLHCClockPhase;           // use LHC-clock phase
  Bool_t fPrimaryDCASelectionFlag;    // selection of primaries with DCA cut

  // ESD related stuff
  Int_t fRunNumber;                   // run number
  AliESDEvent *fESDEvent;             // ESD event
  AliPhysicsSelection *fEventCuts;    // event cuts
  AliESDtrackCuts *fTrackCuts;        // track cuts
  AliESDpid *fESDpid;                 // ESD PID
  UInt_t fStartTime;                  // start time
  UInt_t fEndTime;                    // end time
  UInt_t fElapsedTime;                // event time since start
  Bool_t fIsCollisionCandidate;       // is collision candidate
  Bool_t fHasVertex;                  // has vertex
  const AliESDVertex *fVertex;        // vertex

  // GRP related stuff
  AliGRPManager *fGRPManager;         // GRP manager
  const AliGRPObject *fGRPObject;     // GRP object

  // TOF related stuff
  TString fSpecificStorageParOffline; // specific storage ParOffline
  TString fSpecificStorageRunParams;  // specific storage RunParams
  Float_t fTimeResolution;            // time resolution
  AliTOFcalib *fTOFcalib;             // TOF calib
  AliTOFT0maker *fTOFT0maker;         // TOF-T0 maker
  AliTOFT0v1 *fTOFT0v1;               // TOF-T0 v1

  // task related stuff
  UInt_t ftimestamp;
  Float_t fVertexZ;
  Float_t ftimezero;
  Int_t fnhits;
  Float_t fmomentum[MAXHITS];
  Float_t flength[MAXHITS];
  Int_t findex[MAXHITS];
  Float_t ftime[MAXHITS];
  Float_t ftot[MAXHITS];
  Float_t ftexp[MAXHITS];

  TTree* fOutputTree;                 // output tree

  ClassDef(AliTOFAnalysisTaskCalibTree, 2);
};

#endif /* ALIANALYSISTASKTOFCOMPACTCALIB_H */