//

// Created by Sumedhe Dissanayake on 10/28/18.

//

#include "arg_parser.h"

#include <iostream>

#include <string>

#include <stdlib.h>

#include "../helpers/learning_rate.h"

#include "../helpers/neighborhood.h"

#include "../core/som.h"

using namespace std;

void parse_config_from_args(SOM som, int argc, char** argv){

try{

// For each argument

for (int i = 1; i < argc; ++i){

string opt = argv[i];

if (opt == "-x" || opt == "-xdim") {

// X dimension size

som.x_dim = atoi(argv[++i]);

} else if (opt == "-y" || opt == "-ydim") {

// Y dimension size

som.y_dim = atoi(argv[++i]);

} else if (opt == "-d" || opt == "-dimension") {

// Dimension

som.dimension = atoi(argv[++i]);

} else if (opt == "-g" || opt == "-grid-type") {

// Grid Type

som.grid_type = argv[++i];

} else if (opt == "-t" || opt == "-iteration-limit"){

// Iteration limit

som.iteration_limit = atoi(argv[++i]);

} else if (opt == "-lr" || opt == "-starting-learning-rate"){

// Starting learning rate

som.starting_learning_rate = (float)atof(argv[++i]);

} else if (opt == "-m" || opt == "-node-init-method"){

// Node initialization method

som.node_initialization_method = argv[++i];

} else if (opt == "-lf" || opt == "-learning-rate-type"){

// Learning rate type

som.learning_rate_type = (LearningRate::type_map[argv[++i]]);

} else if (opt == "-nf" || opt == "-neighborhood-type"){

// Neighborhood function

som.neighborhood_type = Neighborhood::type_map[argv[++i]];

} else if (opt == "-min" || opt == "-min-node-weight"){

// Minimum node weight (For random initialization)

som.min_node_weight = (float)atof(argv[++i]);

} else if (opt == "-max" || opt == "-max-node-weight"){

// Maximum node weight (For random initialization)

som.max_node_weight = (float)atof(argv[++i]);

}

}

} catch (int e){

cout << "Error in parsing args!" << endl;

}

}