ECLShowerCorrectorModule.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
#include <framework/database/DBObjPtr.h>
#include <framework/datastore/StoreArray.h>
#include <ecl/dbobjects/ECLnOptimal.h>
#include <mdst/dataobjects/EventLevelClusteringInfo.h>
#include <TH2F.h>
 
namespace Belle2 {
  class ECLLeakageCorrections;
  class ECLShower;
 
  namespace ECL {
    class ECLLeakagePosition;
  }
 
  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;
 
  private:
 
    StoreArray<ECLShower> m_eclShowers;
 
    StoreObjPtr<EventLevelClusteringInfo> m_eventLevelClusteringInfo;
 
    ECL::ECLLeakagePosition* m_leakagePosition{nullptr};
 
    DBObjPtr<ECLLeakageCorrections> m_eclLeakageCorrections;
    TH2F thetaCorrection; 
    TH2F phiCorrection; 
    //..Parameters derived from payload (except for nThetaID & nLeakReg, which are fixed)
    const unsigned int nThetaID = 69; 
    static constexpr unsigned int nLeakReg = 3; 
    int nPositionBins = 0; 
    int nXBins = 0; 
    int nEnergies = 0; 
    std::vector< std::vector<float> > leakLogE; 
    DBObjPtr<ECLnOptimal> m_eclNOptimal;
    TH2F m_bias; 
    TH2F m_logPeakEnergy; 
    TH2F m_peakFracEnergy; 
  public:
    virtual const char* eclShowerArrayName() const
    { return "ECLShowers" ; }
 
  };
 
  class ECLShowerCorrectorPureCsIModule : public ECLShowerCorrectorModule {
  public:
 
    virtual const char* eclShowerArrayName() const override
    { return "ECLShowersPureCsI" ; }
 
  };
 
} // end of Belle2 namespace