CsvBFieldWriter.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CsvBFieldWriter.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/MagneticField/InterpolatedBFieldMap.hpp"
#include "Acts/MagneticField/MagneticFieldProvider.hpp"
#include "Acts/Utilities/Logger.hpp"

#include <array>
#include <cstddef>
#include <memory>
#include <optional>
#include <string>
#include <type_traits>

namespace Acts {
class InterpolatedMagneticField;
class MagneticFieldProvider;
}  // namespace Acts

namespace ActsExamples {
class CsvBFieldWriter {
 public:
  enum class CoordinateType { XYZ, RZ };

  template <CoordinateType Coord, bool Grid>
  struct Config {
    template <typename T>
    using Functor = std::conditional_t<Grid, std::optional<T>, T>;

    static constexpr std::size_t NDims = Coord == CoordinateType::RZ ? 2 : 3;

    std::string fileName = "bfield.csv";

    std::array<std::array<Functor<double>, 2>, NDims> range{};

    std::array<Functor<std::size_t>, NDims> bins{};

    std::shared_ptr<std::add_const_t<std::conditional_t<
        Grid, Acts::InterpolatedMagneticField, Acts::MagneticFieldProvider>>>
        bField;
  };

  template <CoordinateType Coord, bool Grid>
  static void run(const Config<Coord, Grid>& config,
                  Acts::Logging::Level level);
};
}  // namespace ActsExamples