encoder.h

#ifndef SINGA_IO_ENCODER_H_
#define SINGA_IO_ENCODER_H_

#include <vector>
#include <string>
#include "singa/core/tensor.h"
#include "singa/proto/io.pb.h"

namespace singa {

class Encoder {
 public:
  Encoder() {}
  virtual ~Encoder() {}

  virtual void Setup(const EncoderConf& conf) {}

  virtual std::string Encode(vector<Tensor>& data) = 0;
};

#ifdef USE_OPENCV
class JPGEncoder : public Encoder {
 public:
  void Setup(const EncoderConf& conf) override {
    image_dim_order_ = conf.image_dim_order();
  }
  std::string Encode(vector<Tensor>& data) override;

  const std::string image_dim_order() const { return image_dim_order_; }

 private:
  std::string image_dim_order_ = "CHW";
};
#endif  // USE_OPENCV

class CSVEncoder : public Encoder {
 public:
  void Setup(const EncoderConf& conf) override {}
  std::string Encode(vector<Tensor>& data) override;
};
} // namespace singa
#endif  // SINGA_IO_ENCODER_H_