PandoraPFACalibrator.h

#ifndef PANDORA_PFA_CALIBRATOR_H
#define PANDORA_PFA_CALIBRATOR_H

#include "marlin/Processor.h"

#include <vector>

class TFile;
class TH1F;
class TH2F;

//------------------------------------------------------------------------------------------------------------------------------------------

class PandoraPFACalibrator : public marlin::Processor
{
public:
    virtual Processor *newProcessor();
    PandoraPFACalibrator() ;

    virtual void init() ;

    virtual void processRunHeader( LCRunHeader* run ) ;

    virtual void processEvent( LCEvent * evt ) ; 

    virtual void check( LCEvent * evt ) ; 

    virtual void end() ;

private:
    void ReadMCParticles(LCEvent *pLCEvent, const LCStrVec &collectionNames, float &cosTheta) const;

    void ReadHitEnergies(LCEvent *pLCEvent, const LCStrVec &collectionNames, float &hitEnergySum, const float mipConstant = -1.f,
        TH1F *const pMipPlot = NULL, TH1F *const pMipPlotCorrected = NULL, const char *const pLayerEncoding = NULL, TH1F *const pEnergyByLayerPlot = NULL) const;

    void ReadPfoCollections(LCEvent *pLCEvent, const LCStrVec &collectionNames, float &pfoEnergySum, float &cosTheta) const;

    LCStrVec        m_inputMCParticleCollections;                   
    std::string     m_particleCollectionName;                       
    LCStrVec        m_mcPfoCollections;                             
    LCStrVec        m_recoPfoCollections;                           

    LCStrVec        m_ecalBarrelCollections;                        
    LCStrVec        m_ecalEndCapCollections;                        
    LCStrVec        m_hcalCollections;                              
    LCStrVec        m_muonCollections;                              
    LCStrVec        m_lcalCollections;                              
    LCStrVec        m_bcalCollections;                              
    LCStrVec        m_lhcalCollections;                             

    int             m_nRun;                                         
    int             m_nEvt;                                         

    float           m_ecalToMIP;                                    
    float           m_hcalToMIP;                                    
    float           m_muonToMIP;                                    
    float           m_ecalToEMGeVCalibration;                       
    float           m_hcalToHadGeVCalibration;                      
    float           m_ecalToHadGeVCalibrationBarrel;                
    float           m_ecalToHadGeVCalibrationEndCap;                
    float           m_hcalToEMGeVCalibration;                       
    float           m_maxHCalHitHadronicEnergy;                     

    float           m_zOfEndCap;                                    

    TFile          *m_pTFile;                                       
    std::string     m_rootFile;                                     

    TH1F           *m_hPfoEnergy;                                   
    TH1F           *m_hPfoEnergyBarrel;                             
    TH1F           *m_hPfoEnergy95ECal;                             
    TH1F           *m_hPfoEnergy95HCal;                             
    TH1F           *m_hPfoEnergy95Muon;                             
    TH2F           *m_hPfoEnergyVsCosTheta;                         
    TH2F           *m_hPfoEnergyVsCosThetaReco;                     

    TH1F           *m_hCaloEnergy;                                  
    TH1F           *m_hCaloEnergyECal;                              
    TH1F           *m_hCaloEnergyHCal;                              
    TH1F           *m_hCaloEnergyMuon;                              
    TH1F           *m_hCaloEnergy95ECal;                            
    TH1F           *m_hCaloEnergy95HCal;                            
    TH1F           *m_hCaloEnergy95Muon;                            
    TH1F           *m_hEcalBarrelEnergyByLayer;                     
    TH1F           *m_hEcalEndCapEnergyByLayer;                     
    TH2F           *m_hECalHCalEnergyEM;                            
    TH2F           *m_hECalHcalEnergyHAD;                           
    TH2F           *m_hECalBarrelHCalEnergyEM;                      
    TH2F           *m_hECalEndCapHCalEnergyEM;                      
    TH2F           *m_hECalBarrelHCalEnergyHAD;                     
    TH2F           *m_hECalEndCapHCalEnergyHAD;                     
    TH2F           *m_hCaloEnergyVsCosTheta;                        
    TH2F           *m_hCaloEnergyVsCosThetaReco;                    

    TH1F           *m_hECalBarrelMIP;                               
    TH1F           *m_hECalEndCapMIP;                               
    TH1F           *m_hHCalMIP;                                     
    TH1F           *m_hMuonMIP;                                     
    TH1F           *m_hECalBarrelMIPCorr;                           
    TH1F           *m_hECalEndCapMIPCorr;                           
    TH1F           *m_hHCalMIPCorr;                                 
    TH1F           *m_hMuonMIPCorr;                                 
};

//------------------------------------------------------------------------------------------------------------------------------------------

inline marlin::Processor *PandoraPFACalibrator::newProcessor()
{
    return new PandoraPFACalibrator;
}

#endif // #ifndef PANDORA_PFA_CALIBRATOR_H