Deblending.h

Go to the documentation of this file.

#ifndef _SEFRAMEWORK_PIPELINE_DEBLENDING_H
#define _SEFRAMEWORK_PIPELINE_DEBLENDING_H

#include "SEFramework/Pipeline/PipelineStage.h"
#include "SEFramework/Source/SourceGroupInterface.h"
#include "SEFramework/Task/TaskProvider.h"

namespace SourceXtractor {

class DeblendStep {
public:

  virtual ~DeblendStep() = default;

  virtual void deblend(SourceGroupInterface& group) const = 0;

  virtual std::set<PropertyId> requiredProperties() const { return {};}
};

class Deblending : public PipelineReceiver<SourceGroupInterface>, public PipelineEmitter<SourceGroupInterface> {

public:

  virtual ~Deblending() = default;

  explicit Deblending(std::vector<std::shared_ptr<DeblendStep>> deblend_steps);

  void receiveSource(std::unique_ptr<SourceGroupInterface> group) override;

  void receiveProcessSignal(const ProcessSourcesEvent& event) override;

  std::set<PropertyId> requiredProperties() const;

private:
  std::vector<std::shared_ptr<DeblendStep>> m_deblend_steps;
}; /* End of Deblending class */

} /* namespace SourceXtractor */


#endif