ECLCRFinderModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Belle II Connected Region Finder (CRF). Starting with 'seed' cells     *
 * above an energy energyCut0. Add neighbouring crystals above energyCut2.*
 * If neighbouring crystal is above energyCut1, repeat this. Timing cuts  *
 * can be set for each digit type and the energy cuts can be made         *
 * background dependent using the event-by-event background measurements. *
 * Digits with failed time fits automatically pass timing cuts by default.*
 * The CRF must run once before the splitters and splitters must only use *
 * digits contained in a CR. Digits from different CRs must not be mixed. *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Torben Ferber (ferber@physics.ubc.ca)                    *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#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; 
    double m_energyCutMod[3] {}; 
    int m_tempCRId = -1; 
    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