#include "train_config.h"


namespace data_structures {

    TrainConfig::TrainConfig() {
        num_bars = 4;
        min_tracks = 1;
        max_tracks = 12;
        max_mask_percentage = 0.75;
        use_microtiming = false;
        microtiming = 0.9;
        no_max_length = false;
        resolution = 12;
        delta_resolution = 1920;
        decode_resolution = delta_resolution;
    }

    std::map<std::string, std::string> TrainConfig::ToJson() {
        std::map<std::string, std::string> json_config;
        json_config["num_bars"] = std::to_string(num_bars);
        json_config["min_tracks"] = std::to_string(min_tracks);
        json_config["max_tracks"] = std::to_string(max_tracks);
        json_config["max_mask_percentage"] = std::to_string(max_mask_percentage);
        json_config["use_microtiming"] = std::to_string((int)use_microtiming);
        json_config["microtiming"] = std::to_string(microtiming);
        json_config["no_max_length"] = std::to_string((int)no_max_length);
        json_config["resolution"] = std::to_string(resolution);
        json_config["decode_resolution"] = std::to_string(decode_resolution);
        json_config["delta_resolution"] = std::to_string(delta_resolution);
        return json_config;
    }

    void TrainConfig::FromJson(std::map<std::string, std::string>& json_config) {
        num_bars = stoi(json_config["num_bars"]);
        min_tracks = stoi(json_config["min_tracks"]);
        max_tracks = stoi(json_config["max_tracks"]);
        max_mask_percentage = stof(json_config["max_mask_percentage"]);
        microtiming = stoi(json_config["microtiming"]);
        use_microtiming = (bool)stoi(json_config["use_microtiming"]);
        no_max_length = (bool)stoi(json_config["no_max_length"]);
        resolution = stoi(json_config["resolution"]);
        decode_resolution = stoi(json_config["decode_resolution"]);
        delta_resolution = stoi(json_config["delta_resolution"]);
    }
}