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() override;
    ECLShowerCorrectorModule(const ECLShowerCorrectorModule&) = delete;

    ECLShowerCorrectorModule& operator=(const ECLShowerCorrectorModule&) = delete;

    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