//
//  addBizCode.cpp
//  MNNConverter
//
//  Created by MNN on 2019/01/31.
//  Copyright © 2018, Alibaba Group Holding Limited
//

#include "addBizCode.hpp"
#include <fstream>
#include <iostream>
#include "logkit.h"

int addBizCode(const std::string modelFile, const std::string bizCode,
               std::unique_ptr<MNN::NetT>& netT) {
    std::ifstream inputFile(modelFile, std::ios::binary);
    inputFile.seekg(0, std::ios::end);
    auto size = inputFile.tellg();
    inputFile.seekg(0, std::ios::beg);

    char* buffer = new char[size];
    inputFile.read((char*)buffer, size);
    inputFile.close();
    netT = MNN::UnPackNet(buffer);
    CHECK(netT->oplists.size() > 0) << "MNN Molde ERROR: " << modelFile;

    // if no set bizCode and has original bizCode, then keep using original bizCode
    if (bizCode != "MNNTest" || netT->bizCode.empty()) {
        netT->bizCode = bizCode;
    }

    delete[] buffer;

    return 0;
}