ReferenceGenerators.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file ReferenceGenerators.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/Definitions/Algebra.hpp"
#include "Acts/Geometry/GeometryContext.hpp"
#include "Acts/Geometry/Polyhedron.hpp"
#include "Acts/Surfaces/Surface.hpp"
#include "Acts/Utilities/BinningData.hpp"

#include <vector>

namespace Acts {
namespace Experimental {
namespace detail {

struct CenterReferenceGenerator {
  const std::vector<Vector3> references(const GeometryContext& gctx,
                                        const Surface& surface) const {
    return {surface.center(gctx)};
  }
};

template <BinningValue bVAL = BinningValue::binValues>
struct BinningValueReferenceGenerator {
  const std::vector<Vector3> references(const GeometryContext& gctx,
                                        const Surface& surface) const {
    return {surface.binningPosition(gctx, bVAL)};
  }
};

template <size_t nSEGS = 1u, bool aBARY = true>
struct PolyhedronReferenceGenerator {
  const std::vector<Vector3> references(const GeometryContext& gctx,
                                        const Surface& surface) const {
    // Create the return  vector
    std::vector<Vector3> rPositions;
    auto pHedron = surface.polyhedronRepresentation(gctx, nSEGS);
    rPositions.insert(rPositions.end(), pHedron.vertices.begin(),
                      pHedron.vertices.end());
    // Add the barycenter if configured
    if constexpr (aBARY) {
      Vector3 bc(0., 0., 0.);
      std::for_each(rPositions.begin(), rPositions.end(),
                    [&](const auto& p) { bc += p; });
      bc *= 1. / rPositions.size();
      rPositions.push_back(bc);
    }
    return rPositions;
  }
};

}  // namespace detail
}  // namespace Experimental
}  // namespace Acts