PRDtoTrackMap.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef _Trk_PRDtoTrackMap_H_
#define _Trk_PRDtoTrackMap_H_
#include "AthAllocators/ArenaPoolSTLAllocator.h"
#include <unordered_map>
#include <functional>
 
namespace Trk {
 
class PrepRawData;
class Track;
 
class PRDtoTrackMap
{
public:
 using TrackPrepRawDataMap =
     std::unordered_map<const Track*,                     // Key
                        std::vector<const PrepRawData*>,  // T
                        std::hash<const Track*>,          // Hash
                        std::equal_to<const Track*>,      // KeyEqual
                        SG::ArenaPoolSTLAllocator<std::pair<
                            const Track* const,
                            std::vector<const PrepRawData*>>>  // Allocator
                        >;
 
 using PrepRawDataTrackMap = std::unordered_multimap<
     const PrepRawData*,                 // Key
     const Track*,                       // T
     std::hash<const PrepRawData*>,      // Hash
     std::equal_to<const PrepRawData*>,  // KeyEqual
     SG::ArenaPoolSTLAllocator<
         std::pair<const PrepRawData* const, const Track*>>  // Allocator
 
     >;
 
 using PrepRawDataTrackMapRange =
     std::pair<PrepRawDataTrackMap::iterator, PrepRawDataTrackMap::iterator>;
 
 using ConstPrepRawDataTrackMapRange =
     std::pair<PrepRawDataTrackMap::const_iterator,
               PrepRawDataTrackMap::const_iterator>;
 
 PRDtoTrackMap() = default;
 virtual ~PRDtoTrackMap() = default;
 PRDtoTrackMap(const PRDtoTrackMap& a) = default;
 PRDtoTrackMap(PRDtoTrackMap&& a) noexcept = default;
 PRDtoTrackMap& operator=(const PRDtoTrackMap& a) = default;
 PRDtoTrackMap& operator=(PRDtoTrackMap&& a) noexcept = default;
 
 virtual const std::type_info& getType() const;

 bool isUsed(const PrepRawData& prd) const;

 bool isShared(const PrepRawData& prd) const;

 PrepRawDataTrackMapRange onTracks(const PrepRawData& prd);

 ConstPrepRawDataTrackMapRange onTracks(const PrepRawData& prd) const;

 std::vector<const Trk::PrepRawData*> getPrdsOnTrack(const Track& track) const;
 
 void clear();
 
 void reserve(size_t nTracks, size_t nRawData);
 
protected:
  PrepRawDataTrackMap m_prepRawDataTrackMap;

  TrackPrepRawDataMap m_trackPrepRawDataMap;
};
 
}
 
#include "PRDtoTrackMap.icc"
#ifndef CLIDSVC_CLASSDEF_H
#include "AthenaKernel/CLASS_DEF.h"
CLASS_DEF(Trk::PRDtoTrackMap, 51736277, 1)
#endif
#endif