Dispatcher.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// 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 ALICEO2_DISPATCHER_H
#define ALICEO2_DISPATCHER_H
 
#include <string>
#include <vector>
#include <memory>
 
#include "Framework/DataProcessorSpec.h"
#include "Framework/DeviceSpec.h"
#include "Framework/Task.h"
 
#include <fairmq/FwdDecls.h>
#include "DataSampling/DataSamplingHeader.h"
 
namespace o2::monitoring
{
class Monitoring;
}
 
namespace o2::utilities
{
 
class DataSamplingHeader;
class DataSamplingPolicy;
 
class Dispatcher : public framework::Task
{
 public:
  Dispatcher(const std::string name, const std::string reconfigurationSource);
  ~Dispatcher() override;
 
  void init(framework::InitContext& ctx) override;
  void run(framework::ProcessingContext& ctx) override;
 
  void registerPolicy(std::unique_ptr<DataSamplingPolicy>&&);
  size_t numberOfPolicies();
 
  const std::string& getName();
  framework::Inputs getInputSpecs();
  framework::Outputs getOutputSpecs();
  framework::Options getOptions();
 
 private:
  DataSamplingHeader prepareDataSamplingHeader(const DataSamplingPolicy& policy);
  header::Stack extractAdditionalHeaders(const char* inputHeaderStack) const;
  void reportStats(monitoring::Monitoring& monitoring) const;
  void send(framework::DataAllocator& dataAllocator, const framework::DataRef& inputData, const framework::Output& output) const;
 
  std::string mName;
  DataSamplingHeader::DeviceIDType mDeviceID = "invalid";
  std::string mReconfigurationSource;
  // policies should be shared between all pipeline threads
  std::vector<std::shared_ptr<DataSamplingPolicy>> mPolicies;
};
 
} // namespace o2::utilities
 
#endif //ALICEO2_DISPATCHER_H