Sequencer.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file Sequencer.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/IAlgorithm.hpp"
#include "ActsExamples/Framework/IContextDecorator.hpp"
#include "ActsExamples/Framework/IReader.hpp"
#include "ActsExamples/Framework/IService.hpp"
#include "ActsExamples/Framework/IWriter.hpp"
#include <Acts/Utilities/Logger.hpp>

#include <cstddef>
#include <memory>
#include <string>
#include <utility>
#include <vector>

namespace ActsExamples {

class Sequencer {
 public:
  struct Config {
    size_t skip = 0;
    size_t events = SIZE_MAX;
    Acts::Logging::Level logLevel = Acts::Logging::INFO;
    int numThreads = -1;
    std::string outputDir;
  };

  Sequencer(const Config& cfg);

  void addService(std::shared_ptr<IService> service);
  void addContextDecorator(std::shared_ptr<IContextDecorator> decorator);
  void addReader(std::shared_ptr<IReader> reader);
  void addAlgorithm(std::shared_ptr<IAlgorithm> algorithm);
  void addWriter(std::shared_ptr<IWriter> writer);

  int run();

 private:
  std::vector<std::string> listAlgorithmNames() const;
  std::pair<size_t, size_t> determineEventsRange() const;

  Config m_cfg;
  std::vector<std::shared_ptr<IService>> m_services;
  std::vector<std::shared_ptr<IContextDecorator>> m_decorators;
  std::vector<std::shared_ptr<IReader>> m_readers;
  std::vector<std::shared_ptr<IAlgorithm>> m_algorithms;
  std::vector<std::shared_ptr<IWriter>> m_writers;
  std::unique_ptr<const Acts::Logger> m_logger;

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

}  // namespace ActsExamples