// Copyright (c) OpenMMLab. All rights reserved.

#include "test_utils.h"
using namespace std;

namespace mmdeploy::test {
unique_ptr<Transform> CreateTransform(const Value& cfg, Device device, Stream stream) {
  auto op_type = cfg.value<string>("type", "");
  auto op_version = cfg.value<int>("version", 1);

  try {
    auto creator = Registry<Transform>::Get().GetCreator(op_type, op_version);
    if (creator == nullptr) {
      return nullptr;
    }
    auto _cfg = cfg;
    _cfg["context"]["device"] = device;
    _cfg["context"]["stream"] = stream;
    return creator->Create(_cfg);
  } catch (std::exception& e) {
    cout << "exception: " << e.what() << endl;
    return nullptr;
  } catch (...) {
    cout << "unexpected exception" << endl;
    return nullptr;
  }
}

vector<int64_t> Shape(const Value& value, const string& shape_key) {
  vector<int64_t> shape;
  for (auto& v : value[shape_key]) {
    shape.push_back(v.get<int>());
  }
  return shape;
}

vector<float> ImageNormCfg(const Value& value, const std::string& key) {
  vector<float> res;
  for (auto& v : value["img_norm_cfg"][key]) {
    res.push_back(v.get<float>());
  }
  return res;
}

}  // namespace mmdeploy::test