#pragma once

#include "vw_slim_predict.h"

namespace vw_slim
{
class example_predict_builder
{
  example_predict* _ex;
  namespace_index _namespace_idx;
  uint64_t _namespace_hash;
  uint64_t _feature_index_bit_mask;

  void add_namespace(namespace_index feature_group);

 public:
  example_predict_builder(example_predict* ex, char* namespace_name, uint32_t feature_index_num_bits = 18);
  example_predict_builder(example_predict* ex, namespace_index namespace_idx, uint32_t feature_index_num_bits = 18);

  void push_feature_string(char* feature_idx, feature_value value);
  void push_feature(feature_index feature_idx, feature_value value);
};
}  // namespace vw_slim