BDAResultStep.h

Go to the documentation of this file.

// Copyright (C) 2021 ASTRON (Netherlands Institute for Radio Astronomy)
// SPDX-License-Identifier: GPL-3.0-or-later
 
#ifndef DP3_BDARESULTSTEP_H
#define DP3_BDARESULTSTEP_H
 
#include "Step.h"
#include "base/BdaBuffer.h"
 
#include <cassert>
#include <vector>
 
namespace dp3 {
namespace steps {
 
class BDAResultStep : public Step {
 public:
  BDAResultStep() : buffers_() {}
 
  ~BDAResultStep() override {}
 
  common::Fields getRequiredFields() const override { return {}; }
 
  common::Fields getProvidedFields() const override { return {}; }
 
  bool process(std::unique_ptr<base::BdaBuffer> buffer) override {
    assert(!getNextStep());  // The result step should be the last step.
    buffers_.push_back(std::move(buffer));
    return true;
  }
 
  void finish() override {}
 
  void show(std::ostream&) const override {}
 
  std::vector<std::unique_ptr<base::BdaBuffer>> Extract() {
    std::vector<std::unique_ptr<base::BdaBuffer>> result;
    result.swap(buffers_);
    return result;
  }
 
 private:
  std::vector<std::unique_ptr<base::BdaBuffer>> buffers_;
};
 
}  // namespace steps
}  // namespace dp3
 
#endif