Partition.h

Go to the documentation of this file.

 
#ifndef _SEFRAMEWORK_PIPELINE_PARTITION_H
#define _SEFRAMEWORK_PIPELINE_PARTITION_H
 
#include "SEFramework/Pipeline/PipelineStage.h"
#include "SEFramework/Source/SourceInterface.h"
#include <vector>
 
namespace SourceXtractor {

class PartitionStep {
public:
  virtual ~PartitionStep() = default;
 
  virtual std::vector<std::unique_ptr<SourceInterface>> partition(std::unique_ptr<SourceInterface> source) const = 0;
};

class Partition : public PipelineReceiver<SourceInterface>, public PipelineEmitter<SourceInterface> {
 
public:
  virtual ~Partition() = default;

  explicit Partition(std::vector<std::shared_ptr<PartitionStep>> steps);
 
  void receiveSource(std::unique_ptr<SourceInterface> source) override;
  void receiveProcessSignal(const ProcessSourcesEvent& event) override;
 
private:
  std::vector<std::shared_ptr<PartitionStep>> m_steps;
 
}; /* End of Partition class */
 
} /* namespace SourceXtractor */
 
#endif