WriterT.hpp

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

// This file is part of the Acts project.
//
// Copyright (C) 2017 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 "ActsExamples/Framework/IWriter.hpp"
#include "ActsExamples/Framework/WhiteBoard.hpp"
#include <Acts/Utilities/Logger.hpp>

#include <memory>
#include <string>

namespace ActsExamples {

template <typename write_data_t>
class WriterT : public IWriter {
 public:
  WriterT(std::string objectName, std::string writerName,
          Acts::Logging::Level level);

  std::string name() const final override;

  ProcessCode write(const AlgorithmContext& context) final override;

  ProcessCode endRun() override;

 protected:
  virtual ProcessCode writeT(const AlgorithmContext& context,
                             const write_data_t& t) = 0;

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

 private:
  std::string m_objectName;
  std::string m_writerName;
  std::unique_ptr<const Acts::Logger> m_logger;
};

}  // namespace ActsExamples

template <typename write_data_t>
ActsExamples::WriterT<write_data_t>::WriterT(std::string objectName,
                                             std::string writerName,
                                             Acts::Logging::Level level)
    : m_objectName(std::move(objectName)),
      m_writerName(std::move(writerName)),
      m_logger(Acts::getDefaultLogger(m_writerName, level)) {
  if (m_objectName.empty()) {
    throw std::invalid_argument("Missing input collection");
  } else if (m_writerName.empty()) {
    throw std::invalid_argument("Missing writer name");
  }
}

template <typename write_data_t>
inline std::string ActsExamples::WriterT<write_data_t>::name() const {
  return m_writerName;
}

template <typename write_data_t>
inline ActsExamples::ProcessCode ActsExamples::WriterT<write_data_t>::endRun() {
  return ProcessCode::SUCCESS;
}

template <typename write_data_t>
inline ActsExamples::ProcessCode ActsExamples::WriterT<write_data_t>::write(
    const AlgorithmContext& context) {
  return writeT(context, context.eventStore.get<write_data_t>(m_objectName));
}