Class AliasSampling

src/sampler/sampling/alias_sampling.cc:22–47

namespace embedx {

class AliasSampling : public Sampling {
 private:
  vec_float_t alias_probs_;
  vec_int_t alias_tables_;

 public:
  static std::unique_ptr<Sampling> Create(const vec_float_t& probs);

 public:
  int_t Next() const noexcept override;
  int_t Next(int begin, int end) const noexcept override;

 private:
  // Always return true.
  bool Init(const vec_float_t& probs);

  void Clear() noexcept {
    alias_probs_.clear();
    alias_tables_.clear();
  }

  void Resize(size_t size) {
    alias_probs_.resize(size, 0);
    alias_tables_.resize(size, 0);
  }
};

std::unique_ptr<Sampling> AliasSampling::Create(const vec_float_t& probs) {
  std::unique_ptr<Sampling> sampling(new AliasSampling);