ECLShowerCorrectorModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * This module performs the correction for EM shower (mainly longitudinal *
 * leakage): corr = (Reconstructed / Truth).                              *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Torben Ferber (ferber@physics.ubc.ca) (TF)               *
 *               Alon Hershenhorn (hersehn@physics.ubc.ca)                *
 *               Suman Koirala (suman@ntu.edu.tw)                         *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
// STL
#include <vector>
 
//ROOT
#include <TGraph2D.h>
 
// FRAMEWORK
#include <framework/core/Module.h>
#include <framework/database/DBObjPtr.h>
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
 
namespace Belle2 {
  class EventLevelClusteringInfo;
  class ECLShower;
  class ECLShowerCorrectorLeakageCorrection;
  class ECLShowerEnergyCorrectionTemporary;
 
  class ECLShowerCorrectorModule : public Module {
 
  public:
    ECLShowerCorrectorModule();
 
    ~ECLShowerCorrectorModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
    void prepareLeakageCorrections();
 
    double getLeakageCorrection(const double theta, const double phi, const double energy, const double background) const;
 
    double getLeakageCorrectionTemporary(const double theta, const double phi, const double energy, const double background);
 
  private:
    DBObjPtr<ECLShowerCorrectorLeakageCorrection> m_leakageCorrectionPtr_bgx0;  
    // Elisa's and Claudia's corrections
    DBObjPtr<ECLShowerEnergyCorrectionTemporary>
    m_leakageCorrectionPtr_thetaGeo_phase2bgx1;  
    DBObjPtr<ECLShowerEnergyCorrectionTemporary>
    m_leakageCorrectionPtr_thetaGeo_phase3bgx1;  
    DBObjPtr<ECLShowerEnergyCorrectionTemporary>
    m_leakageCorrectionPtr_phiGeo_phase2bgx1;  
    DBObjPtr<ECLShowerEnergyCorrectionTemporary>
    m_leakageCorrectionPtr_phiGeo_phase3bgx1;  
    DBObjPtr<ECLShowerEnergyCorrectionTemporary>
    m_leakageCorrectionPtr_thetaEn_phase2bgx1;  
    DBObjPtr<ECLShowerEnergyCorrectionTemporary>
    m_leakageCorrectionPtr_thetaEn_phase3bgx1;  
    DBObjPtr<ECLShowerEnergyCorrectionTemporary>
    m_leakageCorrectionPtr_phiEn_phase2bgx1;  
    DBObjPtr<ECLShowerEnergyCorrectionTemporary>
    m_leakageCorrectionPtr_phiEn_phase3bgx1;  
    // TO BE UPDATED
    const double m_fullBkgdCount = 183.0; 
    // For temporary BGx1
    TGraph2D m_leakage_bgx1[8]; //0: theta_geo ph2, 1: theta_geo ph3, 2: phi_geo ph2, 3: phi_geo ph3, 4: theta_en ph2, 5: theta_en ph3, 6: phi_en ph2, 7 : phi_en ph3,
    std::vector<double> m_leakage_bgx1_limits[8];
 
    // For BGx0 (complicated theta and phi dependency)
    // Vectors with one entry each:
    int m_numOfBfBins = 0; 
    int m_numOfEnergyBins = 0; 
    int m_numOfPhiBins = 0; 
    int m_numOfReg1ThetaBins = 0; 
    int m_numOfReg2ThetaBins = 0; 
    int m_numOfReg3ThetaBins = 0; 
    int m_phiPeriodicity = 0; 
    float m_lReg1Theta = 0; 
    float m_hReg1Theta = 0; 
    float m_lReg2Theta = 0; 
    float m_hReg2Theta = 0; 
    float m_lReg3Theta = 0; 
    float m_hReg3Theta = 0; 
    // Vector with (right now) 15 entries
    std::vector<float> m_avgRecEn; 
    // Vectors with all corrections
    std::vector<int> m_bgFractionBinNum; 
    std::vector<int> m_regNum; 
    std::vector<int> m_phiBinNum; 
    std::vector<int> m_thetaBinNum; 
    std::vector<int> m_energyBinNum; 
    std::vector<float> m_correctionFactor; 
    std::vector < std::vector < std::vector < std::vector < float > > > > m_reg1CorrFactorArrays; 
    std::vector < std::vector < std::vector < std::vector < float > > > > m_reg2CorrFactorArrays; 
    std::vector < std::vector < std::vector < std::vector < float > > > > m_reg3CorrFactorArrays; 
    StoreArray<ECLShower> m_eclShowers;
 
    StoreObjPtr<EventLevelClusteringInfo> m_eventLevelClusteringInfo;
 
  public:
    virtual const char* eclShowerArrayName() const
    { return "ECLShowers" ; }
 
    virtual const char* eventLevelClusteringInfoName() const
    { return "EventLevelClusteringInfo" ; }
 
  };
 
  class ECLShowerCorrectorPureCsIModule : public ECLShowerCorrectorModule {
  public:
 
    virtual const char* eclShowerArrayName() const override
    { return "ECLShowersPureCsI" ; }
 
    virtual const char* eventLevelClusteringInfoName() const override
    { return "EventLevelClusteringInfoPureCsI" ; }
 
  };
 
} // end of Belle2 namespace