ModelConfig Class Reference

#include <ModelConfig.h>

Public Member Functions
	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
const std::vector< InputConfig > &	getScalarInputs () const
const std::vector< InputConfig > &	getVectorInputs () const
const std::vector< std::string > &	getOutputNodes () const

Static Public Member Functions
static std::string	expandPath (const std::string &path)

Private Member Functions
void	parseConfig (const std::string &fname)

Static Private Member Functions
static InputConfig	parseInput (pt::ptree &tree, const InputConfigKeys &keys)

Private Attributes
std::string	savedir
bool	loaded
std::vector< InputConfigKeys >	scalarInputKeys
std::vector< InputConfigKeys >	vectorInputKeys
std::vector< std::string >	outputKeys
std::vector< InputConfig >	scalarInputs
std::vector< InputConfig >	vectorInputs
std::vector< std::string >	outputNodes

Detailed Description

Definition at line 88 of file ModelConfig.h.

Constructor & Destructor Documentation

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

Definition at line 23 of file ModelConfig.cxx.

  : savedir(ModelConfig::expandPath(savedir)),
    loaded(false),
    scalarInputKeys(scalarInputKeys),
    vectorInputKeys(vectorInputKeys),
    outputKeys(outputKeys)
{
    scalarInputs.reserve(scalarInputKeys.size());
    vectorInputs.reserve(vectorInputKeys.size());
    outputNodes.reserve(outputKeys.size());
}

Member Function Documentation

std::string ModelConfig::expandPath ( const std::string &  path )

static

Definition at line 6 of file ModelConfig.cxx.

{
    wordexp_t p;

    int s = wordexp(path.c_str(), &p, WRDE_NOCMD | WRDE_SHOWERR);

    if ((s != 0) || (p.we_wordc != 1)) {
        wordfree(&p);
        return path;
    }

    std::string result(p.we_wordv[0]);
    wordfree(&p);

    return result;
}

std::string ModelConfig::getConfigPath

(

)

const

Definition at line 100 of file ModelConfig.cxx.

{
    return savedir + "/" + CONFIG_NAME;
}

std::string ModelConfig::getModelPath

(

)

const

Definition at line 105 of file ModelConfig.cxx.

{
    return savedir + "/" + MODEL_NAME;
}

const std::vector< std::string > & ModelConfig::getOutputNodes

(

)

const

Definition at line 120 of file ModelConfig.cxx.

{
    return outputNodes;
}

std::string ModelConfig::getSavedir

(

)

const

Definition at line 95 of file ModelConfig.cxx.

{
    return savedir;
}

const std::vector< InputConfig > & ModelConfig::getScalarInputs

(

)

const

Definition at line 110 of file ModelConfig.cxx.

{
    return scalarInputs;
}

const std::vector< InputConfig > & ModelConfig::getVectorInputs

(

)

const

Definition at line 115 of file ModelConfig.cxx.

{
    return vectorInputs;
}

bool ModelConfig::isLoaded

(

)

const

Definition at line 39 of file ModelConfig.cxx.

{
    return loaded;
}

void ModelConfig::load

(

)

Definition at line 44 of file ModelConfig.cxx.

{
    if (loaded) {
        return;
    }

    const std::string configPath = getConfigPath();
    try {
        parseConfig(configPath);
    }
    catch (const std::exception& e) {
        throw std::runtime_error(
            "Failed to parse config file: " + configPath + ". : " + e.what()
        );
    }

    loaded = true;
}

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

private

Definition at line 77 of file ModelConfig.cxx.

{
    pt::ptree tree;
    pt::read_json(fname, tree);

    for (const auto &inputKeys : scalarInputKeys) {
        scalarInputs.emplace_back(parseInput(tree, inputKeys));
    }

    for (const auto &inputKeys : vectorInputKeys) {
        vectorInputs.emplace_back(parseInput(tree, inputKeys));
    }

    for (const auto &outputKey : outputKeys) {
        outputNodes.emplace_back(tree.get<std::string>(outputKey));
    }
}

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

staticprivate

Definition at line 63 of file ModelConfig.cxx.

{
    std::string nodeName = tree.get<std::string>(keys.nodeName);
    std::vector<std::string> varNames;

    for (auto &stree : tree.get_child(keys.varNames)) {
        varNames.push_back(stree.second.get_value<std::string>());
    }

    return InputConfig{ std::move(nodeName), std::move(varNames) };
}

Member Data Documentation

bool ModelConfig::loaded

private

Definition at line 119 of file ModelConfig.h.

std::vector<std::string> ModelConfig::outputKeys

private

Definition at line 129 of file ModelConfig.h.

std::vector<std::string> ModelConfig::outputNodes

private

Definition at line 133 of file ModelConfig.h.

std::string ModelConfig::savedir

private

Definition at line 118 of file ModelConfig.h.

std::vector<InputConfigKeys> ModelConfig::scalarInputKeys

private

Definition at line 127 of file ModelConfig.h.

std::vector<InputConfig> ModelConfig::scalarInputs

private

Definition at line 131 of file ModelConfig.h.

std::vector<InputConfigKeys> ModelConfig::vectorInputKeys

private

Definition at line 128 of file ModelConfig.h.

std::vector<InputConfig> ModelConfig::vectorInputs

private

Definition at line 132 of file ModelConfig.h.

---

The documentation for this class was generated from the following files:

• dunereco/dunereco/VLNets/models/tf_model/ModelConfig.h
• dunereco/dunereco/VLNets/models/tf_model/ModelConfig.cxx