PWG/EMCAL/AliEsdSkimTask.h

   1 #ifndef ALIESDSKINTASK_H
   2 #define ALIESDSKINTASK_H
   3
   4 // $Id$
   5
   6 #include "AliPhysicsSelectionTask.h"
   7 #include "AliESDtrack.h"
   8
   9 class TTree;
  10 class AliESDEvent;
  11 class AliESDtrackCuts;
  12
  13 class AliEsdSkimTask : public AliAnalysisTaskSE {
  14  public:
  15   AliEsdSkimTask(const char *opt=0);
  16
  17   void UserExec(Option_t *opt);
  18   void UserCreateOutputObjects();
  19   void SetCheckCond(Int_t c)       { fCheckCond     = c; }
  20   void SetCuts(AliESDtrackCuts *c) { fCuts          = c; }
  21   void SetDoCent(Bool_t b)         { fDoCent        = b; }
  22   void SetDoClus(Bool_t b)         { fDoClus        = b; }
  23   void SetDoEmC(Bool_t b)          { fDoEmCs        = b; }
  24   void SetDoEmT(Bool_t b)          { fDoEmT         = b; }
  25   void SetDoFmd(Bool_t b)          { fDoFmd         = b; }
  26   void SetDoMiniTracks(Bool_t b)   { fDoMiniTracks  = b; }
  27   void SetDoMult(Bool_t b)         { fDoMult        = b; }
  28   void SetDoMuonTracks(Bool_t b)   { fDoMuonTracks  = b; }
  29   void SetDoPhC(Bool_t b)          { fDoPCs         = b; }
  30   void SetDoPhT(Bool_t b)          { fDoPT          = b; }
  31   void SetDoPicoTracks(Bool_t b)   { fDoPicoTracks  = b; }
  32   void SetDoPileup(Bool_t b)       { fDoPileup      = b; }
  33   void SetDoPriV(Bool_t b)         { fDoPriv        = b; }
  34   void SetDoRP(Bool_t b)           { fDoRP          = b; }
  35   void SetDoSaveBytes(Bool_t b)    { fDoSaveBytes   = b; }
  36   void SetDoSpdV(Bool_t b)         { fDoSPDv        = b; }
  37   void SetDoT0(Bool_t b)           { fDoT0          = b; }
  38   void SetDoTof(Bool_t b)          { fDoTof         = b; }
  39   void SetDoTpcV(Bool_t b)         { fDoTPCv        = b; }
  40   void SetDoTracks(Bool_t b)       { fDoTracks      = b; }
  41   void SetDoV0(Bool_t b)           { fDoV0          = b; }
  42   void SetDoZdc(Bool_t b)          { fDoZDC         = b; }
  43   void SetEmcNames(const char *n)  { fEmcNames      = n; }
  44   void SetEmcalClusOnly(Bool_t b)  { fEmcalClusOnly = b; }
  45   void SetPhosClusOnly(Bool_t b)   { fPhosClusOnly  = b; }
  46   void SetRemoveCP(Bool_t b)       { fRemoveCP      = b; }
  47   void SetResetCov(Bool_t b)       { fResetCov      = b; }
  48   void SetTracks(const char *n)    { fTracks        = n; }
  49
  50  protected:
  51   AliESDEvent     *fEvent;        //!esd event
  52   TTree           *fTree;         //!tree
  53   AliESDtrackCuts *fCuts;         // track cuts
  54   Bool_t           fDoZDC;        // do zdc
  55   Bool_t           fDoV0;         // do vzero
  56   Bool_t           fDoT0;         // do tzero
  57   Bool_t           fDoTPCv;       // do tpc vertex
  58   Bool_t           fDoSPDv;       // do spd vertex
  59   Bool_t           fDoPriv;       // do primary vertex
  60   Bool_t           fDoEmCs;       // do emcal cells
  61   Bool_t           fDoPCs;        // do phos cells
  62   Bool_t           fDoEmT;        // do emcal trigger
  63   Bool_t           fDoPT;         // do phos trigger
  64   Bool_t           fDoTracks;     // do tracks
  65   Bool_t           fDoFmd;        // do fmd
  66   Bool_t           fDoMult;       // do mult
  67   Bool_t           fDoTof;        // do TOF
  68   Bool_t           fDoPileup;     // do pileup
  69   Bool_t           fDoClus;       // do clusters
  70   Bool_t           fDoMuonTracks; // do muon tracks
  71   TString          fEmcNames;     // name of clusters
  72   Bool_t           fDoMiniTracks; // strip down tracks
  73   TString          fTracks;       // name of tracks (e.g. tracks propagated to EMCAL surface)
  74   Bool_t           fPhosClusOnly; // if true then only store PHOS clusters
  75   Bool_t           fEmcalClusOnly;// if true then only store EMCAL clusters
  76   Bool_t           fDoSaveBytes;  // if true then trim down some of the stored objects (mult, fmd)
  77   Bool_t           fDoCent;       // do centrality
  78   Bool_t           fDoRP;         // do reaction plane
  79   Bool_t           fRemoveCP;     // if false then keep constrained parameters (only reset covariance)
  80   Bool_t           fResetCov;     // if true reset covariance matrix of track
  81   Bool_t           fDoPicoTracks; // if true then do pico tracks
  82   Int_t            fCheckCond;    // if !=0 check certain conditions before event is accepted
  83
  84  private:
  85   AliEsdSkimTask(const AliEsdSkimTask&);            // not implemented
  86   AliEsdSkimTask &operator=(const AliEsdSkimTask&); // not implemented
  87
  88  ClassDef(AliEsdSkimTask, 5); // Esd trimming and skimming task
  89 };
  90 #endif