weightencoder.h

Go to the documentation of this file.

#ifndef DYSCO_WEIGHT_ENCODER_H
#define DYSCO_WEIGHT_ENCODER_H
 
#include <cmath>
#include <vector>
 
namespace dyscostman {
 
template <typename T>
class WeightEncoder {
 public:
  typedef T value_t;
  typedef unsigned symbol_t;
 
  explicit WeightEncoder(unsigned quantCount) : _quantCount(quantCount) {}
 
  void Encode(value_t &scaleVal, std::vector<symbol_t> &dest,
              const std::vector<value_t> &input) const {
    // Find max value (implicit assumption input is not empty)
    typename std::vector<value_t>::const_iterator i = input.begin();
    scaleVal = *i;
    ++i;
    while (i != input.end()) {
      if (*i > scaleVal) scaleVal = *i;
      ++i;
    }
 
    i = input.begin();
    typename std::vector<symbol_t>::iterator d = dest.begin();
    const value_t scaleFact = value_t(_quantCount - 1) / scaleVal;
    while (i != input.end()) {
      *d = roundf(scaleFact * (*i));
 
      ++i;
      ++d;
    }
  }
 
  void Decode(std::vector<value_t> &dest, value_t scaleVal,
              const std::vector<symbol_t> &input) const {
    typename std::vector<symbol_t>::const_iterator i = input.begin();
    typename std::vector<value_t>::iterator d = dest.begin();
    const value_t scaleFact = scaleVal / value_t(_quantCount - 1);
    while (i != input.end()) {
      *d = (*i) * scaleFact;
 
      ++i;
      ++d;
    }
  }
 
 private:
  unsigned _quantCount;
};
 
}  // namespace dyscostman
 
#endif