#include "Attractor.hpp"
using namespace std;

Attractor::Attractor() {
    myAttractor = new Lorenz3D();
    // TODO: add default parameters
}

Attractor::Attractor(const char* const fileName) {
    cout << "Reading file " << fileName << "..." << endl;

    stfu::node system;
    system.read(fileName);
    stfu::node attractor = system.getChild("attractor");

    string attractorType = attractor.getValue("type");
    const string attractorDimension = attractor.getValue("dimensions");

    for ( unsigned int i = 0; attractorType[i] != '\0'; i++ ) {
        attractorType[i] = tolower(attractorType[i]);
    }
    const unsigned int dimension = atoi(attractorDimension.c_str());

    cout << " Formula: " << attractorType << endl;
    cout << " Dimensions: " << dimension << endl;

    if ( attractorType == "lorenz" ){
        if ( dimension == 3 ) {
            myAttractor = new Lorenz3D();
        } else {
            cerr << "something wrong";
            exit(37);
        }
    } else if ( attractorType == "polynomial" ) {
        const string attractorOrde = attractor.getValue("orde");
        const unsigned int orde = atoi(attractorOrde.c_str());
        cout << " Orde: " << orde << endl;
        myAttractor = new Polynomial(dimension, orde);

    } else if ( attractorType == "polynomial a" ) {
        if ( dimension == 3 ) {
            myAttractor = new PolynomialA3D();
        } else {
            cerr << "something wrong";
            exit(37);
        }
    } else if ( attractorType == "logistic" ) {
        myAttractor = new Logistic(dimension);

    } else if ( attractorType == "unravel" ) {
        if ( dimension == 3 ) {
            myAttractor = new Unravel3D();
        } else {
            cerr << "somtheing wrong";
            exit(37);
        }
    } else {
        cout << "'" << attractorType << "' not recognized" << endl;
        exit(3);
    }

    const unsigned int numberOfParameters = myAttractor->getNumberOfParameters();
    double * & parameters = myAttractor->parameters();

    for ( unsigned int i = 0; i < numberOfParameters; i++ ) {
        stfu::node attractorParameters = attractor.getChild("parameters");
        parameters[i] = atof(attractorParameters.getValue(i).c_str());
        cout << " Parameter " << i << " set to " << parameters[i] << ",  ";
    }

    cout << endl << " Reading file complete" << endl;
}

void Attractor::init_range() {
    // stabilize attractor
    for ( unsigned int i = 0; i < 100000; i++ ) {
        iterate();
    }

    // initialize projectors with dimension and first point
    const unsigned int* dim = (unsigned int*)myAttractor->getProperty("dimension");
    const unsigned int dimension = *dim;
    delete dim;
    const double * point = myAttractor->vector();
    for ( vector<Projector*>::iterator it = projectors.begin(); it != projectors.end(); it++ ) {
        (*it)->extern_dim = dimension;
        (*it)->intern_dim = 2;
        (*it)->init(point);
    }

    // update ranges
    for ( unsigned int i = 0; i < 100000; i++ ) {
        iterate();

        for ( vector<Projector*>::iterator it = projectors.begin(); it != projectors.end(); it++ ) {
            (*it)->update_range(point);
        }
    }
    for ( vector<Projector*>::iterator it = projectors.begin(); it != projectors.end(); it++ ) {
        (*it)->finish_range();
    }

}

bool Attractor::is_chaos() {
    /*
        check existence of attractor:
            Escaping
            Single point attractor
            Lyapunov exponent
    */
/*
    double sum = 0;
    for ( unsigned int i = 0; i < dim; i++ ) {
        const double dist = 0; //new_point[i] - point[i];
        sum += dist*dist;
    }
    if ( sum >= 1.0e7 ) {
        // big change => Escaping
        return false;
    }
    if ( sum <= 1.0e-7 ) {
        // small change => singularity
        return false;
    }
    return true;
*/
    return true;
}

void Attractor::iterate() {
    myAttractor->iterate();
}

void Attractor::plot() {
    for ( vector<Projector *>::iterator it = projectors.begin(); it != projectors.end(); it++ ) {
        const double * point = myAttractor->vector();
        (*it)->plot(point);
    }
}


/*
    IO & control
*/
void Attractor::output() {
    const unsigned int* dim = (unsigned int*)myAttractor->getProperty("dimension");
    const unsigned int dimension = *dim;
    delete dim;
    const double * point = myAttractor->vector();

    for ( unsigned int i = 0; i < dimension; i++ ) {
        cout << point[i] << " ";
    }
    cout << endl;
}