TGeoITkModuleSplitter.hpp

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

// This file is part of the Acts project.
//
// Copyright (C) 2021 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/Geometry/GeometryContext.hpp"
#include "Acts/Plugins/TGeo/ITGeoDetectorElementSplitter.hpp"
#include "Acts/Utilities/Logger.hpp"

#include <map>
#include <memory>
#include <regex>
#include <string>
#include <tuple>
#include <utility>
#include <vector>

class TGeoNode;

namespace Acts {
class TGeoDetectorElement;
}

namespace ActsExamples {

class TGeoITkModuleSplitter : public Acts::ITGeoDetectorElementSplitter {
 public:
  using SplitRange = std::pair<double, double>;

  struct Config {
    // Map the nodes name to the splitting parameters
    std::map<std::string, unsigned int> barrelMap = {};
    std::map<std::string, std::vector<SplitRange>> discMap = {};
    std::map<std::string, std::string> splitPatterns;
  };

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

  ~TGeoITkModuleSplitter() override = default;

  std::vector<std::shared_ptr<const Acts::TGeoDetectorElement>> split(
      const Acts::GeometryContext& gctx,
      std::shared_ptr<const Acts::TGeoDetectorElement> detElement)
      const override;

 private:
  void initSplitCategories();

  std::vector<std::shared_ptr<const Acts::TGeoDetectorElement>>
  splitBarrelModule(
      const Acts::GeometryContext& gctx,
      const std::shared_ptr<const Acts::TGeoDetectorElement>& detElement,
      unsigned int nSegments) const;

  std::vector<std::shared_ptr<const Acts::TGeoDetectorElement>> splitDiscModule(
      const Acts::GeometryContext& gctx,
      const std::shared_ptr<const Acts::TGeoDetectorElement>& detElement,
      const std::vector<SplitRange>& splitRanges) const;

  Config m_cfg;

  std::vector<std::tuple<std::regex, std::string, bool>> m_splitCategories;

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

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

}  // namespace ActsExamples