Method parse_string

lite/load_and_run/src/helpers/data_parser.cpp:183–265

}

void DataParser::parse_string(const std::string& name, const std::string& str) {
    //! parse shape
    if ('{' == str[0]) {
        parse_shape(name, str);
        return;
    }

    // data type
    megdnn::DType data_type = mgb::dtype::Int32();
    if (str.find(".") != std::string::npos or str.find(".") != std::string::npos) {
        data_type = mgb::dtype::Float32();
    }
    // shape
    size_t number_cnt = 0;

    std::shared_ptr<Brace> brace_root = std::make_shared<Brace>();
    std::shared_ptr<Brace> cur = brace_root;
    for (size_t i = 0; i < str.size(); ++i) {
        char c = str[i];
        if (c == '[') {
            std::shared_ptr<Brace> child = std::make_shared<Brace>();
            child->parent = cur;
            cur->chidren.emplace_back(child);
            cur = child;
        } else if (c == ']') {
            cur = cur->parent.lock();
        } else if (c == ',') {
            number_cnt++;
        }
        continue;
    }
    ++number_cnt;

    mgb_assert(cur == brace_root, "braces not closed for --input");
    megdnn::SmallVector<size_t> shape;
    cur = brace_root;
    while (not cur->chidren.empty()) {
        shape.append({cur->chidren.size()});
        number_cnt /= cur->chidren.size();
        cur = cur->chidren[0];
    }
    mgb_assert(number_cnt > 0);
    shape.append({number_cnt});

    // data
    std::string json_arr;
    for (size_t i = 0; i < str.size(); ++i) {
        char c = str[i];
        if (c != '[' and c != ']') {
            json_arr += c;
        }
    }
    json_arr = "[" + json_arr + "]";

    // reuse json parser to resolve raw data
    mgb::JsonLoader json;
    std::shared_ptr<mgb::JsonLoader::Value> json_root =
            json.load(json_arr.data(), json_arr.size());