PostProcessingRunner.h

Go to the documentation of this file.

// Copyright CERN and copyright holders of ALICE O2. This software is
// distributed under the terms of the GNU General Public License v3 (GPL
// Version 3), copied verbatim in the file "COPYING".
//
// See http://alice-o2.web.cern.ch/license for full licensing information.
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.


#ifndef QUALITYCONTROL_POSTPROCESSINGRUNNER_H
#define QUALITYCONTROL_POSTPROCESSINGRUNNER_H

#include <memory>
#include <functional>
#include <Framework/ServiceRegistry.h>
#include <boost/property_tree/ptree_fwd.hpp>
#include "QualityControl/PostProcessingInterface.h"
#include "QualityControl/PostProcessingConfig.h"
#include "QualityControl/Triggers.h"
#include "QualityControl/DatabaseInterface.h"
#include "QualityControl/ObjectsManager.h"
#include "QualityControl/MonitorObjectCollection.h"

namespace o2::framework
{
class DataAllocator;
} // namespace o2::framework

namespace o2::quality_control::postprocessing
{

using MOCPublicationCallback = std::function<void(const o2::quality_control::core::MonitorObjectCollection*, long from, long to)>;

class PostProcessingRunner
{
 public:
  PostProcessingRunner(std::string name);
  ~PostProcessingRunner() = default;

  void init(const boost::property_tree::ptree& config);
  bool run();
  void start();
  void stop();
  void reset();
  void runOverTimestamps(const std::vector<uint64_t>& t);

  void setPublicationCallback(MOCPublicationCallback callback);

  const std::string& getName();

 private:
  void doInitialize(Trigger trigger);
  void doUpdate(Trigger trigger);
  void doFinalize(Trigger trigger);

  enum class TaskState {
    INVALID,
    Created,
    Running,
    Finished
  };
  TaskState mTaskState = TaskState::INVALID;
  std::vector<TriggerFcn> mInitTriggers;
  std::vector<TriggerFcn> mUpdateTriggers;
  std::vector<TriggerFcn> mStopTriggers;

  std::unique_ptr<PostProcessingInterface> mTask;
  framework::ServiceRegistry mServices;
  std::shared_ptr<o2::quality_control::core::ObjectsManager> mObjectManager;
  // TODO in a longer run, we should store from/to in the MonitorObject itself and use them.
  std::function<void(const o2::quality_control::core::MonitorObjectCollection*, long /*from*/, long /*to*/)> mPublicationCallback = nullptr;

  std::string mName = "";
  std::string mConfigPath = "";
  PostProcessingConfig mConfig;
  std::shared_ptr<o2::quality_control::repository::DatabaseInterface> mDatabase;
};

MOCPublicationCallback publishToDPL(o2::framework::DataAllocator&, std::string outputBinding);
MOCPublicationCallback publishToRepository(o2::quality_control::repository::DatabaseInterface&);

} // namespace o2::quality_control::postprocessing

#endif //QUALITYCONTROL_POSTPROCESSINGRUNNER_H