SVDNNClusterizerModule.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.                  *
 **************************************************************************/
 
#ifndef SVDNNClusterizerModule_H
#define SVDNNClusterizerModule_H
 
#include <framework/core/Module.h>
#include <framework/datastore/RelationArray.h>
#include <svd/calibration/SVDNoiseCalibrations.h>
#include <svd/calibration/SVDPulseShapeCalibrations.h>
#include <svd/reconstruction/NNWaveFitter.h>
#include <map>
#include <vector>
 
namespace Belle2 {
  namespace SVD {
 
    class SVDNNClusterizerModule : public Module {
 
    public:
 
      typedef std::vector<const RelationElement*> RelationLookup;
 
      SVDNNClusterizerModule();
 
      virtual void initialize() override;
 
      virtual void event() override;
 
    protected:
 
      void createRelationLookup(const RelationArray& relation, RelationLookup& lookup,
                                size_t digits);
 
      void fillRelationMap(const RelationLookup& lookup, std::map<unsigned int, float>&
                           relation, unsigned int index);
 
      // Data members
      //1. Collections
      std::string m_storeMCParticlesName;
      std::string m_storeTrueHitsName;
      std::string m_storeRecoDigitsName;
      std::string m_storeClustersName;
      std::string m_relRecoDigitMCParticleName;
      std::string m_relRecoDigitTrueHitName;
      std::string m_relClusterMCParticleName;
      std::string m_relClusterTrueHitName;
      std::string m_relClusterRecoDigitName;
 
      //2. Strip and time fitter data
      std::string m_timeFitterName;
      bool m_calibratePeak = false;
 
      //3. Clustering
      double m_cutAdjacent = 3.0;
      double m_cutSeed = 5.0;
      double m_cutCluster = 5.0;
      int m_sizeHeadTail = 3;
 
      // 4. Strip map and time fitter handles
      SVDNoiseCalibrations m_noiseCal;
      SVDPulseShapeCalibrations m_pulseShapeCal;
      NNWaveFitter m_fitter;
 
      // 5. Relation lookups
      RelationLookup m_mcRelation;
      RelationLookup m_trueRelation;
 
    };//end class declaration
 
  } //end SVD namespace;
} // end namespace Belle2
 
#endif // SVDNNClusterizerModule_H