eclLeakageCollectorModule.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
 
/* Basf2 headers. */
#include <calibration/CalibrationCollectorModule.h>
#include <framework/database/DBObjPtr.h>
#include <framework/datastore/StoreArray.h>
#include <ecl/dbobjects/ECLnOptimal.h>
#include <TH2F.h>
 
namespace Belle2 {
  class ECLCrystalCalib;
  class ECLShower;
  class MCParticle;
  class EventMetaData;
  class EventLevelClusteringInfo;
  class ECLCluster;
 
  namespace ECL {
    class ECLLeakagePosition;
  }
 
 

  class eclLeakageCollectorModule : public CalibrationCollectorModule {
 
  public:

    eclLeakageCollectorModule();

    virtual void prepare() override;

    virtual void collect() override;
 
  private:

    int m_position_bins; 
    int m_number_energies; 
    std::vector<double> m_energies_forward; 
    std::vector<double> m_energies_barrel; 
    std::vector<double> m_energies_backward; 

    std::string m_showerArrayName;  
    StoreArray<ECLShower> m_eclShowerArray; 
    StoreArray<MCParticle> m_mcParticleArray; 
    StoreObjPtr<EventMetaData> m_evtMetaData; 
    StoreObjPtr<EventLevelClusteringInfo> m_eventLevelClusteringInfo; 
    StoreArray<ECLCluster> m_eclClusterArray; 
 

    const int nLeakReg = 3; 
    bool storeCalib = true; 
    std::vector< std::vector<int> > i_energies;
 
    ECL::ECLLeakagePosition* leakagePosition{nullptr}; 
    int m_nDump = 0; 

    DBObjPtr<ECLnOptimal> m_eclNOptimal;
    TH2F m_nOptimal2D; 
    TH2F m_peakFracEnergy; 
    TH2F m_bias; 
    TH2F m_logPeakEnergy; 
 
    std::vector<int> m_groupNumber; 
    const int m_nLeakReg = 3; 
    int m_nEnergyBins = 0; 
    std::vector< std::vector<float> > m_eBoundaries; 

    int t_cellID = 0; 
    int t_thetaID = 0; 
    int t_region = 0; 
    int t_thetaBin = -1; 
    int t_phiBin =
      -1; 
    int t_phiMech = -1; 
    int t_energyBin = -1; 
    int t_nCrys = -1; 
    float t_energyFrac = 0.; 
    float t_origEnergyFrac = 0.; 
    float t_locationError = 999.; 
  };

}