PayloadDetectorElement.hpp

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

// 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/.

#pragma once

#include <map>

#include "ACTFW/GenericDetector/GenericDetectorElement.hpp"
#include "Acts/Geometry/GeometryContext.hpp"
#include "Acts/Geometry/GeometryID.hpp"
#include "Acts/Plugins/Identification/IdentifiedDetectorElement.hpp"
#include "Acts/Plugins/Identification/Identifier.hpp"
#include "Acts/Surfaces/Surface.hpp"
#include "Acts/Utilities/Definitions.hpp"

namespace FW {

namespace Contextual {

class PayloadDetectorElement : public Generic::GenericDetectorElement {
 public:
  struct ContextType {
    // The alignment store of this event
    // not the fastest, but good enough for a demonstrator
    std::vector<Acts::Transform3D> alignmentStore;
  };

  template <typename... Args>
  PayloadDetectorElement(Args&&... args)
      : Generic::GenericDetectorElement(std::forward<Args>(args)...) {}

  const Acts::Transform3D& transform(
      const Acts::GeometryContext& gctx) const final override;
};

inline const Acts::Transform3D& PayloadDetectorElement::transform(
    const Acts::GeometryContext& gctx) const {
  // cast into the right context object
  auto alignContext = std::any_cast<ContextType>(gctx);
  identifier_type idValue = identifier_type(identifier());

  // check if we have the right alignment parameter in hand
  if (idValue < alignContext.alignmentStore.size()) {
    return alignContext.alignmentStore[idValue];
  }
  // Return the standard transform if not found
  return GenericDetectorElement::transform(gctx);
}

}  // end of namespace Contextual
}  // end of namespace FW