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>
#include <framework/database/DBObjPtr.h>
 
// ECL
#include <ecl/dbobjects/ECLClusteringParameters.h>
 
// C++
#include <set>
 
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_timeCut[3]; 
    double m_timeCut_maxEnergy[3]; 
    std::string m_mapType[2]; 
    double m_mapPar[2]; 
    int m_skipFailedTimeFitDigits; 
    bool m_useParametersFromDatabase; 

    DBObjPtr<ECLClusteringParameters> m_eclClusteringParameters;

    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;

    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);
    bool areNeighbours(const int cellid1, const int cellid2, const int maptype);

    std::vector<int> oneHotVector(std::vector<int>& A, const int n);

    std::vector<int> flattenVector(std::vector<std::vector<int>>& A);

    std::vector<std::set<int>> mergeVectorsUsingSets(std::vector<std::vector<int>>& A);

    std::vector<std::vector<int>> getConnectedRegions(const std::vector<int>& A, const std::vector<int>& B, const int maptype);
 
  };

  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