ECLSplitterN1Module.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 <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
#include <framework/database/DBObjPtr.h>
#include <framework/gearbox/Unit.h>
 
/* Root headers */
#include <TH2F.h>
 
/* C++ headers. */
#include <vector>
 
namespace Belle2 {
  class ECLCalDigit;
  class ECLConnectedRegion;
  class ECLShower;
  class ECLLocalMaximum;
  class EventLevelClusteringInfo;
  class ECLnOptimal;
 
  namespace ECL {
    class ECLNeighbours;
    class ECLGeometryPar;
  }
 
  class ECLSplitterN1Module : public Module {
 
  public:
    ECLSplitterN1Module();
 
    ~ECLSplitterN1Module();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
  private:
    // Module parameters:
 
    // Splitter
    double m_threshold; 
    double m_expConstant; 
    int m_maxIterations; 
    double m_shiftTolerance; 
    double m_minimumSharedEnergy; 
    int m_maxSplits; 
    const double c_molierRadius = 3.581 *
                                  Belle2::Unit::cm; 
    double m_cutDigitEnergyForEnergy; 
    double m_cutDigitTimeResidualForEnergy; 
    int m_useOptimalNumberOfDigitsForEnergy; 
    DBObjPtr<ECLnOptimal> m_eclNOptimal;
    TH2F m_nOptimal2D; 
    std::vector<int> m_groupNumber; 
    const int m_nLeakReg = 3; 
    int m_nEnergyBins = 0; 
    std::vector< std::vector<float> > m_eBoundaries; 
    // Position
    std::string m_positionMethod;  
    double m_liloParameterA; 
    double m_liloParameterB; 
    double m_liloParameterC; 
    std::vector<double> m_liloParameters; 
    // Background
    int m_fullBkgdCount; 
    std::vector< int > m_StoreArrPosition;
 
    std::vector< int > m_StoreArrPositionLM;
 
    std::vector< int > m_cellIdInCR;
 
    ECL::ECLNeighbours* m_NeighbourMap9{nullptr}; 
    ECL::ECLNeighbours* m_NeighbourMap21{nullptr}; 
    StoreArray<ECLCalDigit> m_eclCalDigits;
 
    StoreArray<ECLConnectedRegion> m_eclConnectedRegions;
 
    StoreArray<ECLShower> m_eclShowers;
 
    StoreArray<ECLLocalMaximum> m_eclLocalMaximums;
 
    StoreObjPtr<EventLevelClusteringInfo> m_eventLevelClusteringInfo;
 
    virtual const char* eclCalDigitArrayName() const
    { return "ECLCalDigits" ; }
 
    virtual const char* eclConnectedRegionArrayName() const
    { return "ECLConnectedRegions" ; }
 
    virtual const char* eclLocalMaximumArrayName() const
    { return "ECLLocalMaximums" ; }
 
    virtual const char* eclShowerArrayName() const
    { return "ECLShowers" ; }
 
    virtual const char* eventLevelClusteringInfoName() const
    { return "EventLevelClusteringInfo" ; }
 
    ECL::ECLGeometryPar* m_geom{nullptr};
 
    void splitConnectedRegion(ECLConnectedRegion& aCR);
 
    int getNeighbourMap(const double energy, const double background);
 
    std::vector<int> getOptimalNumberOfDigits(const int cellid, const double energy);
 
    double getEnergySum(std::vector < std::pair<double, double> >& weighteddigits, const unsigned int n);
 
    double estimateEnergy(const int centerid);
 
  }; // end of ECLSplitterN1Module
 
 
  class ECLSplitterN1PureCsIModule : public ECLSplitterN1Module {
  public:
    virtual const char* eclCalDigitArrayName() const override
    { return "ECLCalDigitsPureCsI" ; }
 
    virtual const char* eclConnectedRegionArrayName() const override
    { return "ECLConnectedRegionsPureCsI" ; }
 
    virtual const char* eclLocalMaximumArrayName() const override
    { return "ECLLocalMaximumsPureCsI" ; }
 
    virtual const char* eclShowerArrayName() const override
    { return "ECLShowersPureCsI" ; }
 
    virtual const char* eventLevelClusteringInfoName() const override
    { return "EventLevelClusteringInfoPureCsI" ; }
 
  }; // end of ECLSplitterN1PureCsIModule
 
} // end of Belle2 namespace