PXDClusterizerModule.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
 
#include <framework/core/Module.h>
#include <framework/database/DBObjPtr.h>
#include <vxd/dataobjects/VxdID.h>
#include <pxd/dbobjects/PXDClusterPositionErrorPar.h>
#include <pxd/reconstruction/ClusterCache.h>
#include <pxd/reconstruction/ClusterProjection.h>
#include <pxd/reconstruction/NoiseMap.h>
#include <string>
#include <memory>
 
namespace Belle2 {
  class RelationArray;
  class RelationElement;
  class EventLevelTrackingInfo;
 
 
  namespace PXD {

    class PXDClusterizerModule : public Module {
 
    public:
      typedef std::vector<const RelationElement*> RelationLookup;

      PXDClusterizerModule();

      virtual void initialize() override;
      virtual void beginRun() override;
      virtual void event() override;
 
    private:
      void createRelationLookup(const RelationArray& relation, RelationLookup& lookup, size_t digits);

      void fillRelationMap(const RelationLookup& lookup, std::map<unsigned int, float>& relation, unsigned int index);

      void writeClusters(VxdID sensorID);

      void calculatePositionError(const ClusterCandidate& cls, ClusterProjection& primary, const ClusterProjection& secondary,
                                  double minPitch, double centerPitch, double maxPitch);

      void assignPositionErrorFromDB(ClusterProjection& primary, PXDClusterPositionErrorPar errorPar,
                                     VxdID sensorID, unsigned int uCell, unsigned int vCell, double centerPitch,
                                     bool isAtUEdge = false, bool isAtVEdge = false, bool isAdjacentDead = false);

      double m_elNoise;
      double m_cutSeed;
      double m_cutAdjacent;
      double m_cutCluster;
      double m_cutAdjacentSignal;
      std::string m_storeDigitsName;
      std::string m_storeClustersName;
      std::string m_storeTrueHitsName;
      std::string m_storeMCParticlesName;
      std::string m_relDigitMCParticleName;
      std::string m_relClusterMCParticleName;
      std::string m_relClusterDigitName;
      std::string m_relDigitTrueHitName;
      std::string m_relClusterTrueHitName;

      int m_sizeHeadTail;

      int m_clusterCacheSize;
      std::unique_ptr<ClusterCache> m_cache;
      NoiseMap m_noiseMap;

      RelationLookup m_mcRelation;
      RelationLookup m_trueRelation;

      bool m_errorFromDB;
      std::string m_positionErrorUName;
      std::string m_positionErrorVName;
      std::unique_ptr<DBObjPtr<PXDClusterPositionErrorPar>> m_clusterPositionErrorUPar;
      std::unique_ptr<DBObjPtr<PXDClusterPositionErrorPar>> m_clusterPositionErrorVPar;
      StoreObjPtr<EventLevelTrackingInfo> m_eventLevelTrackingInfo;
      bool m_createPXDClustersForAbortedTrackingEvents = false;
 
 
    };//end class declaration
 
 
  } //end PXD namespace;
} // end namespace Belle2