ECLCRFinderModule.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
 
// FRAMEWORK
#include <framework/core/Module.h>
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/StoreObjPtr.h>
 
namespace Belle2 {
  class ECLConnectedRegion;
  class ECLCalDigit;
  class EventLevelClusteringInfo;
 
  namespace ECL {
    class ECLNeighbours;
  }
 
  class ECLCRFinderModule : public Module {
 
  public:
    ECLCRFinderModule();
 
    virtual ~ECLCRFinderModule();
 
    virtual void initialize() override;
 
    virtual void beginRun() override;
 
    virtual void event() override;
 
    virtual void endRun() override;
 
    virtual void terminate() override;
 
    StoreArray<ECLCalDigit> m_eclCalDigits;
 
    StoreArray<ECLConnectedRegion> m_eclConnectedRegions;
 
    StoreObjPtr<EventLevelClusteringInfo> m_eventLevelClusteringInfo;
 
    virtual const char* eclCalDigitArrayName() const
    { return "ECLCalDigits" ; }
 
    virtual const char* eclConnectedRegionArrayName() const
    { return "ECLConnectedRegions" ; }
 
    virtual const char* eventLevelClusteringInfoName() const
    { return "EventLevelClusteringInfo" ; }
 
  private:
 
    // Module parameters
    double m_energyCut[3]; 
    double m_energyCutBkgd[3]; 
    double m_timeCut[3]; 
    std::string m_mapType[2]; 
    double m_mapPar[2]; 
    int m_useBackgroundLevel; 
    int m_skipFailedTimeFitDigits; 
    int m_fullBkgdCount; 
    bool m_isOnlineProcessing{false}; 
    double m_energyCutMod[3] {}; 
    int m_tempCRId = -1; 
    std::vector <int>  m_cellIdToCheckVec; 
    std::vector <int>  m_cellIdToSeedVec; 
    std::vector <int>  m_cellIdToGrowthVec; 
    std::vector <int>  m_cellIdToDigitVec; 
    std::vector< int > m_calDigitStoreArrPosition;
 
    // USE POSITION IN STORE ARRAY!!!
 
    std::vector < int > m_cellIdToTempCRIdVec; 
    std::map < int, int > m_cellIdToTempCRIdMap; 
    std::vector<ECL::ECLNeighbours*> m_neighbourMaps;
 
    void checkNeighbours(const int cellid, const int tempcrid, const int type);
 
    void updateCRs(int cellid, int tempcr);
 
  };
 
  class ECLCRFinderPureCsIModule : public ECLCRFinderModule {
  public:
    virtual const char* eclCalDigitArrayName() const override
    { return "ECLCalDigitsPureCsI" ; }
 
    virtual const char* eclConnectedRegionArrayName() const override
    { return "ECLConnectedRegionsPureCsI" ; }
 
    virtual const char* eventLevelClusteringInfoName() const override
    { return "EventLevelClusteringInfoPureCsI" ; }
 
  }; // end of ECLCovarianceMatrixPureCsIModule
 
} // end of Belle2 namespace