PXDClusterizerModule.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Ritter                                            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <framework/core/Module.h>
#include <vxd/dataobjects/VxdID.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;
 
 
  namespace PXD {
 
    class PXDClusterizerModule : public Module {
 
    public:
      typedef std::vector<const RelationElement*> RelationLookup;
 
      PXDClusterizerModule();
 
      virtual void initialize() 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);
 
      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;
 
    };//end class declaration
 
 
  } //end PXD namespace;
} // end namespace Belle2