CommonGeometry.cpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CommonGeometry.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 <Acts/Material/IMaterialDecorator.hpp>
#include <Acts/Plugins/Json/JsonGeometryConverter.hpp>
#include <Acts/Plugins/Json/JsonMaterialDecorator.hpp>
#include <Acts/Utilities/Logger.hpp>
#include <boost/program_options.hpp>
#include <string>

#include "ACTFW/Detector/IBaseDetector.hpp"
#include "ACTFW/Geometry/MaterialWiper.hpp"
#include "ACTFW/Io/Root/RootMaterialDecorator.hpp"

namespace FW {
namespace Geometry {

std::pair<std::shared_ptr<const Acts::TrackingGeometry>,
          std::vector<std::shared_ptr<FW::IContextDecorator>>>
build(const boost::program_options::variables_map& vm,
      IBaseDetector& detector) {
  // Material decoration
  std::shared_ptr<const Acts::IMaterialDecorator> matDeco = nullptr;
  auto matType = vm["mat-input-type"].template as<std::string>();
  if (matType == "none") {
    matDeco = std::make_shared<const Acts::MaterialWiper>();
  } else if (matType == "file") {
    // Retrieve the filename
    auto fileName = vm["mat-input-file"].template as<std::string>();
    // json or root based decorator
    if (fileName.find(".json") != std::string::npos) {
      // Set up the converter first
      Acts::JsonGeometryConverter::Config jsonGeoConvConfig;
      // Set up the json-based decorator
      matDeco = std::make_shared<const Acts::JsonMaterialDecorator>(
          jsonGeoConvConfig, fileName);
    } else if (fileName.find(".root") != std::string::npos) {
      // Set up the root-based decorator
      FW::RootMaterialDecorator::Config rootMatDecConfig;
      rootMatDecConfig.fileName = fileName;
      matDeco =
          std::make_shared<const FW::RootMaterialDecorator>(rootMatDecConfig);
    }
  }

  return detector.finalize(vm, matDeco);
}

}  // namespace Geometry
}  // namespace FW