PayloadDecorator.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PayloadDecorator.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 "Acts/Geometry/GeometryContext.hpp"
#include "Acts/Utilities/Definitions.hpp"
#include "Acts/Utilities/Logger.hpp"
#include "ActsExamples/Framework/AlgorithmContext.hpp"
#include "ActsExamples/Framework/IContextDecorator.hpp"

#include <vector>

namespace Acts {
class TrackingGeometry;
}

namespace ActsExamples {

namespace Contextual {

class PayloadDecorator : public IContextDecorator {
 public:
  struct Config {
    std::shared_ptr<const Acts::TrackingGeometry> trackingGeometry = nullptr;

    double rotationStep = 0.2 * M_PI;

    unsigned int iovSize = 100;
  };

  PayloadDecorator(const Config& cfg,
                   std::unique_ptr<const Acts::Logger> logger =
                       Acts::getDefaultLogger("PayloadDecorator",
                                              Acts::Logging::INFO));

  virtual ~PayloadDecorator() = default;

  ProcessCode decorate(AlgorithmContext& context) final override;

  const std::string& name() const final override { return m_name; }

 private:
  Config m_cfg;                                  
  std::unique_ptr<const Acts::Logger> m_logger;  
  std::string m_name = "PayloadDecorator";

  std::vector<Acts::Transform3D> m_nominalStore;

  const Acts::Logger& logger() const { return *m_logger; }

  void parseGeometry(const Acts::TrackingGeometry& tGeometry);
};
}  // namespace Contextual

}  // namespace ActsExamples