SVDNNShapeReconstructorModule.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 SVDNNShapeReconstructorModule_H
#define SVDNNShapeReconstructorModule_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 <svd/dataobjects/SVDEventInfo.h>
#include <map>
#include <vector>
 
namespace Belle2 {
  namespace SVD {
 
    class SVDNNShapeReconstructorModule : public Module {
 
    public:
 
      typedef std::vector<const RelationElement*> RelationLookup;
 
      SVDNNShapeReconstructorModule();
 
      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
      StoreObjPtr<SVDEventInfo> m_storeSVDEvtInfo;
 
      //1. Collections
      std::string m_storeMCParticlesName;
      std::string m_storeTrueHitsName;
      std::string m_storeShaperDigitsName;
      std::string m_storeRecoDigitsName;
      std::string m_relShaperDigitMCParticleName;
      std::string m_relShaperDigitTrueHitName;
      std::string m_relRecoDigitMCParticleName;
      std::string m_relRecoDigitTrueHitName;
      std::string m_relRecoDigitShaperDigitName;
      std::string m_svdEventInfoName;
 
      //2. Strip and time fitter data
      std::string m_timeFitterName;
      bool m_calibratePeak = false;
 
      SVDNoiseCalibrations m_noiseCal;
      SVDPulseShapeCalibrations m_pulseShapeCal;
 
      NNWaveFitter m_fitter;
 
      // Relation lookups
      RelationLookup m_mcRelation;
      RelationLookup m_trueRelation;
 
      bool m_writeRecoDigits = false;
 
      float m_cutAdjacent = 3.0;
 
    };//end class declaration
 
  } //end SVD namespace;
} // end namespace Belle2
 
#endif // SVDNNShapeReconstructorModule_H