GeometryIdGenerator.hpp

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

// This file is part of the Acts project.
//
// Copyright (C) 2023 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/Detector/interface/IGeometryIdGenerator.hpp"
#include "Acts/Geometry/GeometryIdentifier.hpp"
#include "Acts/Utilities/Logger.hpp"

#include <any>

namespace Acts {

class Surface;

namespace Experimental {

class Portal;
class DetectorVolume;

class GeometryIdGenerator final : public IGeometryIdGenerator {
 public:
  struct Config {
    bool containerMode = false;
    unsigned int containerId = 0u;
    bool resetSubCounters = true;
    bool overrideExistingIds = false;
  };

  struct Cache {
    unsigned int volumeCount = 0u;
    unsigned int layerCount = 0u;
    unsigned int portalCount = 0u;
    unsigned int passiveCount = 0u;
    unsigned int sensitiveCount = 0u;
  };

  GeometryIdGenerator(const Config& cfg,
                      std::unique_ptr<const Logger> mlogger = getDefaultLogger(
                          "GeometryIdGenerator", Logging::INFO))
      : m_cfg(cfg), m_logger(std::move(mlogger)) {}

  ~GeometryIdGenerator() override = default;

  IGeometryIdGenerator::GeoIdCache generateCache() const final;

  void assignGeometryId(IGeometryIdGenerator::GeoIdCache& cache,
                        DetectorVolume& dVolume) const final;

  void assignGeometryId(IGeometryIdGenerator::GeoIdCache& cache,
                        Portal& portal) const final;

  void assignGeometryId(IGeometryIdGenerator::GeoIdCache& cache,
                        Surface& surface) const final;

 private:
  GeometryIdentifier volumeId(Cache& cache, bool incrementLayer = true) const;

  Config m_cfg;

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

  std::unique_ptr<const Logger> m_logger;
};

}  // namespace Experimental
}  // namespace Acts