CDCMCMap.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 <tracking/trackingUtilities/utilities/Range.h>
#include <tracking/trackingUtilities/utilities/MayBePtr.h>
 
#include <map>
#include <set>
 
namespace Belle2 {
  class MCParticle;
  class CDCSimHit;
  class CDCHit;
 
  namespace TrackFindingCDC {

    class CDCMCMap {
 
    public:

      CDCMCMap() = default;
      CDCMCMap& operator=(const CDCMCMap&) = delete;
      CDCMCMap(CDCMCMap&) = delete;
 
    public:

      void clear();

      void fill();
 
    private:
      static bool indicatesReassignedSecondary(double weight)
      {
        return weight <= 0;
      }

      void fillSimHitByHitMap();

      void fillMCParticleByHitMap();

      void fillMCParticleBySimHitMap();

      void validateRelations() const;

      void validateReassignedSecondaries() const;
 
    public:
      TrackingUtilities::MayBePtr<const CDCSimHit> getSimHit(const CDCHit* hit) const;

      TrackingUtilities::MayBePtr<const CDCHit> getHit(const CDCSimHit* simHit) const;

      static bool isBackground(const CDCSimHit* simHit);

      bool isBackground(const CDCHit* hit) const;

      TrackingUtilities::MayBePtr<const MCParticle> getMCParticle(const CDCHit* hit) const;

      TrackingUtilities::MayBePtr<const MCParticle> getMCParticle(const CDCSimHit* simHit) const;

      auto getSimHits(const MCParticle* mcParticle) const
      {
        return TrackingUtilities::asRange(m_simHitsByMCParticle.equal_range(mcParticle));
      }

      auto getHits(const MCParticle* mcParticle) const
      {
        return TrackingUtilities::asRange(m_hitsByMCParticle.equal_range(mcParticle));
      }

      bool isReassignedSecondary(const CDCSimHit* ptrSimHit) const
      {
        return m_reassignedSecondarySimHits.count(ptrSimHit) > 0;
      }

      bool isReassignedSecondary(const CDCHit* ptrHit) const
      {
        return m_reassignedSecondaryHits.count(ptrHit) > 0;
      }

      const std::set<const CDCHit*>& getReassignedSecondaryHits() const
      {
        return m_reassignedSecondaryHits;
      }

      const std::set<const CDCSimHit*>& getReassignedSecondarySimHits() const
      {
        return m_reassignedSecondarySimHits;
      }

      const std::multimap<const CDCHit*, const CDCSimHit*>& getSimHitsByHit() const
      {
        return m_simHitsByHit;
      }

      const std::multimap<const MCParticle*, const CDCHit*>& getHitsByMCParticle() const
      {
        return m_hitsByMCParticle;
      }

      const std::multimap<const MCParticle*, const CDCSimHit*>& getSimHitsByMCParticle() const
      {
        return m_simHitsByMCParticle;
      }
 
    private:
      std::multimap<const CDCHit*, const CDCSimHit*> m_simHitsByHit;

      std::multimap<const MCParticle*, const CDCHit*> m_hitsByMCParticle;

      std::multimap<const MCParticle*, const CDCSimHit*> m_simHitsByMCParticle;

      std::set<const CDCHit*> m_reassignedSecondaryHits;

      std::set<const CDCSimHit*> m_reassignedSecondarySimHits;
    };
  }
}