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"

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