PayloadDetector.cpp

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

// This file is part of the Acts project.
//
// Copyright (C) 2019 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/.

#include "ActsExamples/ContextualDetector/PayloadDetector.hpp"

#include "Acts/Geometry/TrackingGeometry.hpp"
#include "Acts/Utilities/Logger.hpp"
#include "ActsExamples/ContextualDetector/PayloadDecorator.hpp"
#include "ActsExamples/ContextualDetector/PayloadDetectorElement.hpp"
#include "ActsExamples/Framework/IContextDecorator.hpp"
#include "ActsExamples/GenericDetector/BuildGenericDetector.hpp"
#include "ActsExamples/GenericDetector/GenericDetectorOptions.hpp"
#include "ActsExamples/Plugins/BField/BFieldOptions.hpp"
#include "ActsExamples/Plugins/BField/BFieldScalor.hpp"
#include "ActsExamples/Plugins/BField/ScalableBField.hpp"

#include <boost/program_options.hpp>

void PayloadDetector::addOptions(
    boost::program_options::options_description& opt) const {
  ActsExamples::Options::addGenericGeometryOptions(opt);
  // Add the bfield options for the magnetic field scaling
  ActsExamples::Options::addBFieldOptions(opt);
  // specify the rotation setp
  opt.add_options()(
      "align-rotation-step",
      boost::program_options::value<double>()->default_value(0.25 * M_PI),
      "Rotation step of the RotationDecorator")(
      "align-loglevel",
      boost::program_options::value<size_t>()->default_value(3),
      "Output log level of the alignment decorator.");
}

auto PayloadDetector::finalize(
    const boost::program_options::variables_map& vm,
    std::shared_ptr<const Acts::IMaterialDecorator> mdecorator)
    -> std::pair<TrackingGeometryPtr, ContextDecorators> {
  // --------------------------------------------------------------------------------
  DetectorElement::ContextType nominalContext;
  // set geometry building logging level
  Acts::Logging::Level surfaceLogLevel =
      Acts::Logging::Level(vm["geo-surface-loglevel"].template as<size_t>());
  Acts::Logging::Level layerLogLevel =
      Acts::Logging::Level(vm["geo-layer-loglevel"].template as<size_t>());
  Acts::Logging::Level volumeLogLevel =
      Acts::Logging::Level(vm["geo-volume-loglevel"].template as<size_t>());

  bool buildProto =
      (vm["mat-input-type"].template as<std::string>() == "proto");

  TrackingGeometryPtr pTrackingGeometry =
      ActsExamples::Generic::buildDetector<DetectorElement>(
          nominalContext, detectorStore, 0, std::move(mdecorator), buildProto,
          surfaceLogLevel, layerLogLevel, volumeLogLevel);

  ContextDecorators pContextDecorators = {};

  // Alignment service
  Decorator::Config agcsConfig;
  agcsConfig.trackingGeometry = pTrackingGeometry;
  agcsConfig.rotationStep = vm["align-rotation-step"].template as<double>();

  Acts::Logging::Level decoratorLogLevel =
      Acts::Logging::Level(vm["align-loglevel"].template as<size_t>());

  // Create the service
  auto agcDecorator = std::make_shared<Decorator>(
      agcsConfig,
      Acts::getDefaultLogger("PayloadDecorator", decoratorLogLevel));
  pContextDecorators.push_back(agcDecorator);

  if (vm["bf-context-scalable"].template as<bool>()) {
    ActsExamples::BField::BFieldScalor::Config bfsConfig;
    bfsConfig.scalor = vm["bf-bscalor"].template as<double>();

    auto bfDecorator =
        std::make_shared<ActsExamples::BField::BFieldScalor>(bfsConfig);

    pContextDecorators.push_back(bfDecorator);
  }

  // return the pair of geometry and the alignment decorator(s)
  return std::make_pair<TrackingGeometryPtr, ContextDecorators>(
      std::move(pTrackingGeometry), std::move(pContextDecorators));
}