AliEmcalCorrectionClusterizer.h

Go to the documentation of this file.

#ifndef ALIEMCALCORRECTIONCLUSTERIZER_H
#define ALIEMCALCORRECTIONCLUSTERIZER_H

#include "AliEmcalCorrectionComponent.h"

#include "AliEMCALRecParam.h"

class AliEmcalCorrectionClusterizer : public AliEmcalCorrectionComponent {
 public:
  enum InputCellType {
    kFEEData = 0,
    kFEEDataMCOnly,
    kFEEDataExcludeMC,
    kPattern,
    kL0FastORs,
    kL0FastORsTC,
    kL1FastORs
  };

#if !(defined(__CINT__) || defined(__MAKECINT__))
  std::map <std::string, AliEMCALRecParam::AliEMCALClusterizerFlag> clusterizerTypeMap = {
    {"kClusterizerv1", AliEMCALRecParam::kClusterizerv1 },
    {"kClusterizerNxN", AliEMCALRecParam::kClusterizerNxN },
    {"kClusterizerv2", AliEMCALRecParam::kClusterizerv2 },
    {"kClusterizerFW", AliEMCALRecParam::kClusterizerFW }
  };
#endif

  AliEmcalCorrectionClusterizer();
  virtual ~AliEmcalCorrectionClusterizer();

  // Sets up and runs the task
  Bool_t Initialize();
  void UserCreateOutputObjects();
  Bool_t Run();
  
protected:
  void           Clusterize();
  void           FillDigitsArray();
  void           Init();
  void           RecPoints2Clusters(TClonesArray *clus);
  void           UpdateClusters();
  void           CalibrateClusters();
  
  void           RemapMCLabelForAODs(Int_t &label);
  void           SetClustersMCLabelFromOriginalClusters();
  
  TClonesArray          *fDigitsArr;                      
  TObjArray             *fClusterArr;                     
  AliEMCALRecParam      *fRecParam;                       
  AliEMCALClusterizer   *fClusterizer;                    
  AliEMCALAfterBurnerUF *fUnfolder;                       
  Bool_t                 fJustUnfold;                     
  TString                fGeomName;                       
  Bool_t                 fGeomMatrixSet;                  
  Bool_t                 fLoadGeomMatrices;               
  TGeoHMatrix           *fGeomMatrix[AliEMCALGeoParams::fgkEMCALModules]; 
  TString                fOCDBpath;                       
  AliEMCALCalibData     *fCalibData;                      
  AliCaloCalibPedestal  *fPedestalData;                   
  Bool_t                 fLoadCalib;                      
  Bool_t                 fLoadPed;                        
  Bool_t                 fSubBackground;                  
  Int_t                  fNPhi;                           
  Int_t                  fNEta;                           
  Int_t                  fShiftPhi;                       
  Int_t                  fShiftEta;                       
  Bool_t                 fTRUShift;                       
  InputCellType          fInputCellType;                  
  
  // MC labels
  static const Int_t     fgkTotalCellNumber = 17664 ;     
  
  Int_t                  fOrgClusterCellId[fgkTotalCellNumber]; 
  Int_t                  fCellLabels      [fgkTotalCellNumber]; 
  
  Int_t                  fSetCellMCLabelFromCluster;      
  // 0 - get the MC label stored in cells (not available for productions done with aliroot < v5-02-Rev09)
  // 1 - assign to the cell the MC label of the cluster
  
  Bool_t                 fSetCellMCLabelFromEdepFrac;     
  // stored in ESDs/AODs to set the cell MC labels
  
  Bool_t                 fRemapMCLabelForAODs;            
  
  Bool_t                 fRecalDistToBadChannels;         
  Bool_t                 fRecalShowerShape;               
  
  TClonesArray          *fCaloClusters;                   
  AliESDEvent           *fEsd;                            
  AliAODEvent           *fAod;                            

 private:
  AliEmcalCorrectionClusterizer(const AliEmcalCorrectionClusterizer &);               // Not implemented
  AliEmcalCorrectionClusterizer &operator=(const AliEmcalCorrectionClusterizer &);    // Not implemented

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

  ClassDef(AliEmcalCorrectionClusterizer, 1); // EMCal correction clusterizer component
};

#endif /* ALIEMCALCORRECTIONCLUSTERIZER_H */