KDTreeTrackingGeometryBuilder.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file KDTreeTrackingGeometryBuilder.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2022 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

#include "Acts/Definitions/Algebra.hpp"
#include "Acts/Detector/ProtoDetector.hpp"
#include "Acts/Geometry/GeometryContext.hpp"
#include "Acts/Geometry/GeometryIdentifier.hpp"
#include "Acts/Geometry/ITrackingGeometryBuilder.hpp"
#include "Acts/Geometry/TrackingGeometryBuilder.hpp"
#include "Acts/Utilities/KDTree.hpp"
#include "Acts/Utilities/Logger.hpp"

#include <array>
#include <cstddef>
#include <memory>
#include <string>
#include <vector>

namespace Acts {

class TrackingGeometry;
class Layer;
class LayerCreator;
class Surface;
class ITrackingVolumeHelper;
class TrackingVolume;

class KDTreeTrackingGeometryBuilder : public ITrackingGeometryBuilder {
 public:
  struct Config {
    std::shared_ptr<const ITrackingVolumeHelper> trackingVolumeHelper = nullptr;
    std::shared_ptr<const LayerCreator> layerCreator = nullptr;
    std::vector<std::shared_ptr<Surface>> surfaces = {};
    ProtoDetector protoDetector;
    std::shared_ptr<const GeometryIdentifierHook> geometryIdentifierHook =
        std::make_shared<GeometryIdentifierHook>();
    std::string hierarchyIndent = "  ";
  };

  using SurfaceKDT =
      KDTree<2u, std::shared_ptr<Surface>, ActsScalar, std::array, 100>;

  KDTreeTrackingGeometryBuilder(
      const Config& cfg,
      std::unique_ptr<const Logger> logger =
          getDefaultLogger("KDTreeTrackingGeometryBuilder", Logging::INFO));

  std::unique_ptr<const TrackingGeometry> trackingGeometry(
      const GeometryContext& gctx) const final;

 private:
  Config m_cfg;

  const Logger& logger() const { return *m_logger; }

  std::unique_ptr<const Logger> m_logger;

  struct Cache {
    size_t surfaceCounter = 0;
  };

  std::shared_ptr<TrackingVolume> translateVolume(
      Cache& cCache, const GeometryContext& gctx, const SurfaceKDT& kdt,
      const ProtoVolume& ptVolume, const std::string& indent = "") const;

  std::shared_ptr<const Layer> translateLayer(
      Cache& cCache, const GeometryContext& gctx, const SurfaceKDT& kdt,
      const ProtoVolume& plVolume, const std::string& indent = "") const;
};

}  // namespace Acts