/**
 * DESCRIPTION OF THE FILE
 *
 * @author Michal Kravčenko
 * @date 4.9.18 -
 */

#include <boost/serialization/export.hpp>

#include "exprtk.hpp"
#include "ExprtkWrapper.h"
#include "ExprtkWrapperSerialization.h"
#include "../Exception/Exceptions.h"

BOOST_CLASS_EXPORT_IMPLEMENT(ExprtkWrapper);

ExprtkWrapper::ExprtkWrapper() {
    // Because of serialization
    // TODO implement?
    throw lib4neuro::NotImplementedException();
}

ExprtkWrapper::ExprtkWrapper( std::string expression_string ) {

    this->p_impl = new ExprtkWrapperImpl();

    this->p_impl->expression_str = expression_string;

    this->p_impl->symbol_table = new symbol_table_t( );

    this->p_impl->symbol_table->add_variable("x", this->p_impl->x);
    this->p_impl->symbol_table->add_variable("y", this->p_impl->y);
    this->p_impl->symbol_table->add_variable("z", this->p_impl->z);
    this->p_impl->symbol_table->add_variable("t", this->p_impl->t);
    this->p_impl->symbol_table->add_variable("f", this->p_impl->z);

    this->p_impl->expression = new expression_t( );
    this->p_impl->expression->register_symbol_table( *this->p_impl->symbol_table );

    this->p_impl->parser = new parser_t( );
    this->p_impl->parser->compile(this->p_impl->expression_str, *this->p_impl->expression );
}

ExprtkWrapper::~ExprtkWrapper() {

    if( this->p_impl->expression ){
        delete this->p_impl->expression;
        this->p_impl->expression = nullptr;
    }

    if( this->p_impl->symbol_table ){
        delete this->p_impl->symbol_table;
        this->p_impl->symbol_table = nullptr;
    }

    if( this->p_impl->parser ){
        delete this->p_impl->parser;
        this->p_impl->parser = nullptr;
    }

    delete this->p_impl;
    this->p_impl = nullptr;

}

double ExprtkWrapper::eval(double x1, double x2, double x3, double x4) {

    this->p_impl->x = x1;
    this->p_impl->y = x2;
    this->p_impl->z = x3;
    this->p_impl->t = x4;

    return this->p_impl->expression->value( );

}

double ExprtkWrapper::eval(std::vector<double> &p) {


    if(p.size() > 0){
        this->p_impl->x = p[0];
    }
    if(p.size() > 1){
        this->p_impl->y = p[1];
    }
    if(p.size() > 2){
        this->p_impl->z = p[2];
    }
    if(p.size() > 3){
        this->p_impl->t = p[3];
    }

    double result = this->p_impl->expression->value();

    return result;
}