/**
 * MOMULPER - Multi-Objective MULtilayer PERceptron
 *
 * This is the main "driver" program for experiments. Algorithmic
 * details are to be encapsulated in the classes.
 */
#include<iostream>
#include<fstream>
#include<string>
#include<memory>
#include<random>
#include<chrono>

#include <boost/property_tree/ptree.hpp>
#include <boost/property_tree/json_parser.hpp>
#define BOOST_FILESYSTEM_NO_DEPRECATED
#include <boost/filesystem.hpp>

#include "MlpLearnAlgorithm.hpp"
#include "Algorithm.hpp"

using boost::property_tree::ptree;
using boost::property_tree::json_parser::read_json;

using namespace std;

void usage(ostream & os, const char* arg0){
    os << "Usage:" << endl;
    os << "  " << arg0 << " [OPTIONS] CASEDEF" << endl << endl;
    os << "where CASEDEF is a case definition file. By default, " <<endl;
    os << "data file is sought in relative path data/ and results" <<endl;
    os << "are stored in directory res/ which must exist." << endl;
}

int main(int argc, char **argv){
    chrono::time_point<chrono::system_clock> time_start, time_end;
    time_start = chrono::system_clock::now();

    if (argc < 2){
        usage(cerr, argv[0]);
        return 1;
    }

    ptree pt;
    ifstream casedef(argv[1]);
    read_json(casedef, pt);
    casedef.close();

    string tag, brief, task, alg, comm;
    tag   = pt.get<string>("tag");
    brief = pt.get<string>("brief");
    task  = pt.get<string>("task");
    alg   = pt.get<string>("algorithm");
    comm  = pt.get<string>("comment");

    cout << "Run case:        " << endl;
    cout << tag << " (" << brief << ")" << endl;
    cout << "Using algorithm: " << alg << endl;

    //string rundirname = "runs/" + pt.get<string>("tag") + "/";
    string rundirname = "runs/" + pt.get<string>("tag");
    boost::filesystem::path dir(rundirname);

    // FIXME: Something wrong here.. doesn't notice that we have the dir already:
    if( !(boost::filesystem::exists(dir))){
        if(!(boost::filesystem::create_directories(dir))) {
            std::cerr << "Could not create output directory "
                      << rundirname << "\n";
            exit(1);
        }
    }

    if (alg == "MLP_learn") {
        MlpLearnAlgorithm alg;
        alg.setup(pt, rundirname + "/", "datasets/");
        alg.run();
    } else {
        cerr << "Unknown algorithm: " << alg << endl;
        return 1;
    }

    time_end = chrono::system_clock::now();
    chrono::duration<double> duration = time_end - time_start;
    cout << "Elapsed wall-clock seconds: " << duration.count() << endl;
    return 0;
}