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