ScaleData.h

Go to the documentation of this file.

// ScaleData.h: DPPP step class for freq-dependent scaling of the data
// Copyright (C) 2020 ASTRON (Netherlands Institute for Radio Astronomy)
// SPDX-License-Identifier: GPL-3.0-or-later
 
 
#ifndef DPPP_SCALEDATA_H
#define DPPP_SCALEDATA_H
 
#include "InputStep.h"
 
#include <xtensor/containers/xtensor.hpp>
 
#include "base/DPBuffer.h"
#include "base/BdaBuffer.h"
 
#include <casacore/casa/Arrays/Cube.h>
 
namespace dp3 {
namespace common {
class ParameterSet;
}
 
namespace steps {
 
 
class ScaleData : public Step {
 public:
  ScaleData(const common::ParameterSet&, const std::string& prefix,
            MsType input_type);
 
  ~ScaleData() override;
 
  common::Fields getRequiredFields() const override { return kDataField; }
 
  common::Fields getProvidedFields() const override { return kDataField; }
 
  bool process(std::unique_ptr<base::DPBuffer>) override;
 
  bool process(std::unique_ptr<base::BdaBuffer>) override;
 
  void finish() override;
 
  void updateInfo(const base::DPInfo&) override;
 
  void show(std::ostream&) const override;
 
  void showTimings(std::ostream&, double duration) const override;
 
  bool accepts(MsType dt) const override { return dt == itsInputType; }
 
  MsType outputs() const override { return itsInputType; }
 
 private:
  void fillSizeScaleFactors(unsigned int nNominal, std::vector<float>& fact);
 
  std::string itsName;
  const MsType itsInputType;
  bool itsScaleSizeGiven;
  bool itsScaleSize;
  std::vector<std::string> itsStationExp;  
  std::vector<std::string> itsCoeffStr;    
  std::vector<std::vector<float>>
      itsStationFactors;             
  xt::xtensor<float, 3> itsFactors;  
  common::NSTimer itsTimer;
};
 
}  // namespace steps
}  // namespace dp3
 
#endif