dyscoweightcolumn.h

Go to the documentation of this file.

#ifndef DYSCO_WEIGHT_COLUMN_H
#define DYSCO_WEIGHT_COLUMN_H
 
#include "stochasticencoder.h"
#include "threadeddyscocolumn.h"
#include "weightblockencoder.h"
 
namespace dyscostman {
 
class DyscoStMan;
 
class DyscoWeightColumn final : public ThreadedDyscoColumn<float> {
 public:
  DyscoWeightColumn(DyscoStMan *parent, int dtype)
      : ThreadedDyscoColumn(parent, dtype) {}
 
  DyscoWeightColumn(const DyscoWeightColumn &source) = delete;
 
  void operator=(const DyscoWeightColumn &source) = delete;
 
  virtual ~DyscoWeightColumn() { shutdown(); }
 
  virtual void Prepare(DyscoDistribution distribution,
                       Normalization normalization, double studentsTNu,
                       double distributionTruncation) override;
 
 protected:
  virtual void initializeDecode(TimeBlockBuffer<data_t> *buffer,
                                const float *metaBuffer, size_t nRow,
                                size_t nAntennae) override;
 
  virtual void decode(TimeBlockBuffer<data_t> *buffer, const symbol_t *data,
                      size_t blockRow, size_t a1, size_t a2) override;
 
  virtual std::unique_ptr<ThreadDataBase> initializeEncodeThread() override {
    return nullptr;
  }
 
  virtual void encode(ThreadDataBase *threadData,
                      TimeBlockBuffer<data_t> *buffer, float *metaBuffer,
                      symbol_t *symbolBuffer, size_t nAntennae) override;
 
  virtual size_t metaDataFloatCount(size_t /*nRows*/, size_t /*nPolarizations*/,
                                    size_t /*nChannels*/,
                                    size_t /*nAntennae*/) const override {
    return _encoder->MetaDataFloatCount();
  }
 
  virtual size_t symbolCount(size_t nRowsInBlock, size_t /*nPolarizations*/,
                             size_t /*nChannels*/) const override {
    return _encoder->SymbolCount(nRowsInBlock);
  }
 
 private:
  std::unique_ptr<WeightBlockEncoder> _encoder;
};
 
}  // namespace dyscostman
 
#endif