ModelConfig.h

Go to the documentation of this file.

#pragma once

#include <string>
#include <vector>

#include <boost/property_tree/ptree.hpp>
namespace pt = boost::property_tree;


const std::string CONFIG_NAME = "config.json";
const std::string MODEL_NAME  = "model.pb";


struct InputConfigKeys
{
    std::string nodeName;
    std::string varNames;
};

struct InputConfig
{
    std::string              nodeName;
    std::vector<std::string> varNames;
};

class ModelConfig
{
private:
    std::string savedir;
    bool loaded;

    void parseConfig(const std::string &fname);

    static InputConfig parseInput(
        pt::ptree &tree, const InputConfigKeys &keys
    );

    std::vector<InputConfigKeys> scalarInputKeys;
    std::vector<InputConfigKeys> vectorInputKeys;
    std::vector<std::string>     outputKeys;

public:
    ModelConfig(
        const std::string &savedir,
        const std::vector<InputConfigKeys> &scalarInputKeys,
        const std::vector<InputConfigKeys> &vectorInputKeys,
        const std::vector<std::string>     &outputKeys
    );

    void load();
    bool isLoaded() const;

    std::string getConfigPath() const;
    std::string getModelPath()  const;
    std::string getSavedir()    const;

    std::vector<InputConfig> scalarInputs;
    std::vector<InputConfig> vectorInputs;
    std::vector<std::string> outputNodes;
};