SurfaceIndexMap.h

Go to the documentation of this file.

// Copyright 2019-2026 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#ifndef ALICEO2_ACTS_SURFACEINDEXMAP_H
#define ALICEO2_ACTS_SURFACEINDEXMAP_H
 
#include <cstddef>
#include <functional>
#include <string>
#include <unordered_map>
#include <vector>
 
#include "Acts/Geometry/GeometryIdentifier.hpp"
 
namespace Acts
{
class Surface;
class TrackingGeometry;
class GeometryContext;
} // namespace Acts
 
namespace o2::detectors
{
class DetMatrixCache;
}
 
namespace o2::acts
{
 
class SurfaceIndexMap
{
 public:
  static constexpr int InvalidSensorID = -1;
 
  void add(int sensorID, const Acts::Surface& surface);
 
  const Acts::Surface* getSurface(int sensorID) const;
  Acts::GeometryIdentifier getGeometryId(int sensorID) const;
 
  const std::vector<int>& getSensorIDs(Acts::GeometryIdentifier geoId) const;
 
  bool empty() const { return mBySensor.empty(); }
  std::size_t size() const { return mBySensor.size(); }
  std::size_t getNSurfaces() const { return mByGeoId.size(); }
  void clear();
 
 private:
  struct Entry {
    const Acts::Surface* surface = nullptr;
    Acts::GeometryIdentifier geoId{};
  };
 
  std::unordered_map<int, Entry> mBySensor;
  std::unordered_map<Acts::GeometryIdentifier, std::vector<int>> mByGeoId;
};
 
using SensorPathProvider = std::function<std::string(int sensorID)>;
 
SurfaceIndexMap makeSurfaceIndex(const Acts::TrackingGeometry& geometry,
                                 const Acts::GeometryContext& gctx,
                                 const o2::detectors::DetMatrixCache& cache,
                                 double toleranceCm = 1e-3,
                                 SensorPathProvider pathProvider = {});
 
} // namespace o2::acts
 
#endif // ALICEO2_ACTS_SURFACEINDEXMAP_H