AliEmcalCorrectionClusterTrackMatcher.h

Go to the documentation of this file.

#ifndef ALIEMCALCORRECTIONCLUSTERTRACKMATCHER_H
#define ALIEMCALCORRECTIONCLUSTERTRACKMATCHER_H

#include "AliEmcalCorrectionComponent.h"

#if !(defined(__CINT__) || defined(__MAKECINT__))
#include "AliEmcalContainerIndexMap.h"
#endif

class TH1;
class TClonesArray;

class AliVParticle;

class AliEmcalCorrectionClusterTrackMatcher : public AliEmcalCorrectionComponent {
 public:
  AliEmcalCorrectionClusterTrackMatcher();
  virtual ~AliEmcalCorrectionClusterTrackMatcher();

  // Sets up and runs the task
  Bool_t Initialize();
  void UserCreateOutputObjects();
  void ExecOnce();
  Bool_t Run();
  
 protected:
  Int_t         GetMomBin(Double_t p) const;
  void          GenerateEmcalParticles();
  void          DoMatching();
  void          UpdateTracks();
  void          UpdateClusters();
  Bool_t        IsTrackInEmcalAcceptance(AliVParticle* part, Double_t edges=0.9) const;
  
  void          SetNumberOfMCGeneratorsToAccept(Int_t nGen){ fNMCGenerToAccept = nGen ;
                    if      ( nGen > 5 ) fNMCGenerToAccept = 5 ;
                    else if ( nGen < 0 ) fNMCGenerToAccept = 0 ; }
  void          SetNameOfMCGeneratorsToAccept(Int_t ig, TString name) { if ( ig < 5 || ig >= 0 ) fMCGenerToAccept[ig] = name ; }
  
  Double_t      fPropDist;              
  Bool_t        fDoPropagation;         
  Bool_t        fAttemptProp;           
  Bool_t        fAttemptPropMatch;      
  Double_t      fMaxDistance;           
  Bool_t        fUsePIDmass;            
  Bool_t        fUseDCA;                
  Bool_t        fUseOuterParamInESDs;   
  Bool_t        fUpdateTracks;          
  Bool_t        fUpdateClusters;        
  
#if !(defined(__CINT__) || defined(__MAKECINT__))
  // Handle mapping between index and containers
  AliEmcalContainerIndexMap <AliClusterContainer, AliVCluster> fClusterContainerIndexMap;    
  AliEmcalContainerIndexMap <AliParticleContainer, AliVParticle> fParticleContainerIndexMap; 
#endif

  TClonesArray *fEmcalTracks;           
  TClonesArray *fEmcalClusters;         
  Int_t         fNEmcalTracks;          
  Int_t         fNEmcalClusters;        
  TH1          *fHistMatchEtaAll;       
  TH1          *fHistMatchPhiAll;       
  TH1          *fHistMatchEta[10][9][2]; 
  TH1          *fHistMatchPhi[10][9][2]; 
  
  Int_t      fNMCGenerToAccept;          
  TString    fMCGenerToAccept[5];        
  Bool_t     fMCGenerToAcceptForTrack;   
  
private:
  AliEmcalCorrectionClusterTrackMatcher(const AliEmcalCorrectionClusterTrackMatcher &);               // Not implemented
  AliEmcalCorrectionClusterTrackMatcher &operator=(const AliEmcalCorrectionClusterTrackMatcher &);    // Not implemented

  // Allows the registration of the class so that it is available to be used by the correction task.
  static RegisterCorrectionComponent<AliEmcalCorrectionClusterTrackMatcher> reg;

  ClassDef(AliEmcalCorrectionClusterTrackMatcher, 5); // EMCal cluster track matcher correction component
};

#endif /* ALIEMCALCORRECTIONCLUSTERTRACKMATCHER_H */