AliJetEmbeddingFromAODTask.h

Go to the documentation of this file.

#ifndef ALIJETEMBEDDINGFROMAODTASK_H
#define ALIJETEMBEDDINGFROMAODTASK_H

// $Id$

class TFile;
class TObjArray;
class TClonesArray;
class TString;
class AliVCaloCells;
class AliVHeader;
class TH2;
class TH1;
class TLorentzVector;
class AliNamedString;

#include "AliJetModelBaseTask.h"

class AliJetEmbeddingFromAODTask : public AliJetModelBaseTask {
 public:
  AliJetEmbeddingFromAODTask();
  AliJetEmbeddingFromAODTask(const char *name, Bool_t drawqa=kFALSE); 
  virtual ~AliJetEmbeddingFromAODTask();

  void           UserCreateOutputObjects();
  Bool_t         UserNotify();

  void           SetFileList(TObjArray *list)                      { fFileList           = list  ; }
  void           SetAODTreeName(const char *t)                     { fAODTreeName        = t     ; }
  void           SetAODHeaderName(const char *t)                   { fAODHeaderName      = t     ; }
  void           SetAODTracksName(const char *n)                   { fAODTrackName       = n     ; }
  void           SetAODClusName(const char *n)                     { fAODClusName        = n     ; }
  void           SetAODCellsName(const char *n)                    { fAODCellsName       = n     ; }
  void           SetAODMCParticlesName(const char *n)              { fAODMCParticlesName = n     ; }
  void           SetCentralityRange(Double_t min, Double_t max)    { fMinCentrality      = min   ; fMaxCentrality    = max ; }
  void           SetTriggerMask(UInt_t mask)                       { fTriggerMask        = mask  ; }
  void           SetAODfilterBits(Int_t b0 = 0, Int_t b1 = 0)      { fAODfilterBits[0]   = b0    ; fAODfilterBits[1] = b1  ; }
  void           SetIncludeNoITS(Bool_t f)                         { fIncludeNoITS       = f     ; }
  void           SetCutMaxFractionSharedTPCClusters(Double_t c = 0.4) { fCutMaxFractionSharedTPCClusters  = c ; }
  void           SetUseNegativeLabels(Bool_t f)                    { fUseNegativeLabels  = f     ; }
  void           SetTrackEfficiency(Double_t eff = 0.95)           { fTrackEfficiency    = new TF1("eff", "[0]", 0, 500); fTrackEfficiency->FixParameter(0,eff); }
  void           SetTrackEfficiency(TF1* eff)                      { fTrackEfficiency    = eff   ; }
  void           SetTotalFiles(Int_t n)                            { fTotalFiles         = n     ; }
  void           SetAttempts(Int_t n)                              { fAttempts           = n     ; }
  void           SetRandomAccess(Bool_t r=kTRUE)                   { fRandomAccess       = r     ; }
  void           SetAODMC(Bool_t a)                                { fIsAODMC            = a     ; }
  void           SetJetMinPt(Double_t pt)                          { fJetMinPt           = pt    ; }
  void           SetJetEtaRange(Double_t emi, Double_t ema)        { fJetMinEta = emi; fJetMaxEta = ema; }
  void           SetJetPhiRange(Double_t pmi, Double_t pma)        { fJetMinPhi = pmi; fJetMaxPhi = pma; }
  void           SetJetConstituentMinPt(Double_t pt)               { fJetConstituentMinPt= pt    ; }
  void           SetJetType(Byte_t t)                              { fJetType            = t     ; }
  void           SetJetAlgo(Byte_t t)                              { fJetAlgo            = t     ; }
  void           SetZVertexCut(Double_t z)                         { fZVertexCut         = z     ; }
  void           SetMaxVertexDist(Double_t d)                      { fMaxVertexDist      = d     ; }
  void           SetParticlePtRange(Double_t min, Double_t max, Byte_t t=1) { fParticleMinPt = min; fParticleMaxPt = max; fParticleSelection = t; }
  void           SetEmbedCentrality(Bool_t d)                      { fEmbedCentrality    = d     ; }

 protected:
  Bool_t          ExecOnce()            ;// intialize task
  void            Run()                 ;// do jet model action
  virtual TFile  *GetNextFile()         ;// get next file from fFileList
  virtual Bool_t  OpenNextFile()        ;// open next file
  virtual Bool_t  GetNextEntry()        ;// get next entry in current tree
  virtual Bool_t  IsAODEventSelected()  ;// AOD event trigger/centrality selection
  TLorentzVector  GetLeadingJet(TClonesArray *tracks, TClonesArray *clusters=0);  // get the leading jet
  Bool_t          FindParticleInRange(TClonesArray *array);// Find particle in array within range (fParticleMinPt, fParticleMaxPt)

  TObjArray     *fFileList            ;//  List of AOD files 
  Bool_t         fRandomAccess        ;//  Random access to file number and event
  TString        fAODTreeName         ;//  Name of the tree in the AOD file
  TString        fAODHeaderName       ;//  Name of the header in the AOD tree
  TString        fAODVertexName       ;//  Name of the vertex branch in the AOD tree
  TString        fAODTrackName        ;//  Name of the track collection branch in the AOD tree
  TString        fAODClusName         ;//  Name of the cluster collection branch in the AOD tree
  TString        fAODCellsName        ;//  Name of the cell collection branch in the AOD tree
  TString        fAODMCParticlesName  ;//  Name of the cell collection branch in the AOD tree
  Double_t       fMinCentrality       ;//  Minimum centrality
  Double_t       fMaxCentrality       ;//  Maximum centrality
  UInt_t         fTriggerMask         ;//  Trigger selection mask
  Double_t       fZVertexCut          ;//  Z vertex cut
  Double_t       fMaxVertexDist       ;//  Maximum distance allowed between the vertices of the current and the embedded events
  Double_t       fJetMinPt            ;//  Select events with a minimum jet pt
  Double_t       fJetMinEta           ;//  Min eta for jets
  Double_t       fJetMaxEta           ;//  Max eta for jets
  Double_t       fJetMinPhi           ;//  Min phi for jets
  Double_t       fJetMaxPhi           ;//  Max phi for jets
  Double_t       fJetConstituentMinPt ;//  Jet constituent min pt
  Double_t       fJetRadius           ;//  Jet radius
  Byte_t         fJetType             ;//  Jet type (0=full, 1=charged, 2=neutral)
  Byte_t         fJetAlgo             ;//  Jet algorithm (0=kT, 1=anti-kT)
  Bool_t         fJetParticleLevel    ;//  Trigger, look at particle level jets
  Double_t       fParticleMinPt       ;//  Select events with a particle pt between fParticleMinPt and fParticleMaxPt (see fParticleSelection)
  Double_t       fParticleMaxPt       ;//  Select events with a particle pt between fParticleMinPt and fParticleMaxPt (see fParticleSelection)
  Byte_t         fParticleSelection   ;//  Particles used to select events (def=0=none, 1=tracks, 2=clusters, 3=MC particles)
  Int_t          fAODfilterBits[2]    ;//  AOD track filter bit map
  Bool_t         fIncludeNoITS        ;//  True = includes tracks with failed ITS refit
  Double_t       fCutMaxFractionSharedTPCClusters;  // max fraction of shared TPC clusters
  Bool_t         fUseNegativeLabels   ;//  Whether or not should use negative MC labels
  TF1           *fTrackEfficiency     ;//  Track efficiency
  Bool_t         fIsAODMC             ;//  Whether the embedding AOD is MC or not
  Int_t          fTotalFiles          ;//  Total number of files per pt hard bin
  Int_t          fAttempts            ;//  Attempts to be tried before giving up in opening the next file
  Bool_t         fEmbedCentrality     ;//  If true, embed centrality (only works when running on AOD) - carefull: it overwrites the event centrality (if any) 
  Bool_t         fEsdTreeMode         ;
  Int_t          fCurrentFileID       ;
  Int_t          fCurrentAODFileID    ;
  TFile         *fCurrentAODFile      ;
  Int_t          fPicoTrackVersion    ;
  TTree         *fCurrentAODTree      ;
  AliVHeader    *fAODHeader           ;
  TClonesArray  *fAODVertex           ;
  TClonesArray  *fAODTracks           ;
  TClonesArray  *fAODClusters         ;
  AliVCaloCells *fAODCaloCells        ;
  TClonesArray  *fAODMCParticles      ;
  Int_t          fCurrentAODEntry     ;
  Int_t          fFirstAODEntry       ;
  Int_t          fLastAODEntry        ;
  AliNamedString *fAODFilePath        ;
  TH2           *fHistFileMatching    ;
  TH1           *fHistAODFileError    ;
  TH1           *fHistNotEmbedded     ;
  TH1           *fHistEmbeddingQA     ;
  TH1           *fHistRejectedEvents  ;
  Int_t          fEmbeddingCount      ;

 private:
  AliJetEmbeddingFromAODTask(const AliJetEmbeddingFromAODTask&);            // not implemented
  AliJetEmbeddingFromAODTask &operator=(const AliJetEmbeddingFromAODTask&); // not implemented

  ClassDef(AliJetEmbeddingFromAODTask, 13) // Jet embedding from AOD task
};
#endif