#include <iostream></iostream>

#include "simulator.h"

void SimulatorXYZ::simulate(const Track &track, const Tool &tool) {

    points.clear();

    simu_index = 0;

    raster.x0 = raster.x1 = raster.y0 = raster.y1 = raster.min_value = raster.max_value = 0;

    if (!raster.raster.get_data() || track.points.empty()) return;

    std::cout << "SimulatorXYZ::simulate" << std::endl;

    std::cout << "SimulatorXYZ::simulate: track points: " << track.points.size() << std::endl;

    const TrackPoint &first = track.points.front();

    raster.x0 = raster.x1 = first.position.x;

    raster.y0 = raster.y1 = first.position.y;

    raster.min_value = raster.max_value = first.position.z;

    for(TrackPointList::const_iterator i = track.points.begin(); i != track.points.end(); ++i) {

        if (raster.x0 > i->position.x) raster.x0 = i->position.x;

        if (raster.x1 < i->position.x) raster.x1 = i->position.x;

        if (raster.y0 > i->position.y) raster.y0 = i->position.y;

        if (raster.y1 < i->position.y) raster.y1 = i->position.y;

        if (raster.min_value > i->position.z) raster.min_value = i->position.z;

        if (raster.max_value < i->position.z) raster.max_value = i->position.z;

    }

    const Real precision = 1e-5;

    if ( raster.x1 + precision < raster.x0

      || raster.y1 + precision < raster.y0

      || raster.max_value + precision < raster.min_value )

    {

        std::cout << "SimulatorXYZ::simulate: zero bounds" << std::endl;

        return;

    }

    raster.raster.fill(raster.max_value);

    Real x0 = raster.x0;

    Real y0 = raster.y0;

    Real dx = raster.x1 - x0;

    Real dy = raster.y1 - y0;

    Real w = raster.raster.get_width();

    Real h = raster.raster.get_height();

    Vector2 scale_to_raster(w/dx , h/dy);

    Vector2 scale_to_tool  (dx/w , dy/h);

    FlatToolPainter painter(&tool, scale_to_tool);

    unsigned long long max_count = 2ull*1024*1024*1024/sizeof(points.front());

    int index = 0;

    for(TrackPointList::const_iterator i = track.points.begin(); i != track.points.end(); ++i, ++index) {

        painter.distance = i->position.z;

        Vector2 position(

            scale_to_raster.x * (i->position.x - x0),

            scale_to_raster.y * (i->position.y - y0) );

        raster.raster.touch(painter, position, index, points);

        if (points.size() >= max_count) {

            std::cerr << "SimulatorXYZ::simulate: reached memory limit at time " <<  i->time << std::endl;

            break;

        }

        if (index % 1000 == 0)

            std::cout << "." << std::flush;

        if (index % 100000 == 0)

            std::cout << std::endl << std::flush;

    }

    simu_index = (int)points.size();

    std::cout << std::endl;

    std::cout << "SimulatorXYZ::simulate: done" << std::endl;

}

void SimulatorXYZ::scroll_to(int index) {

    while(simu_index < (int)points.size() && points[simu_index].index <= index)

        points[simu_index++].apply_next(raster.raster);

    while(simu_index > 0 && points[simu_index-1].index > index)

        points[--simu_index].apply_prev(raster.raster);

}

void SimulatorXYZA::simulate(const Track &track, const Tool &tool) {

    points.clear();

    simu_index = 0;

    raster.x0 = raster.x1 = raster.min_value = raster.max_value = 0;

    if (!raster.raster.get_data() || track.points.empty()) return;

    std::cout << "SimulatorXYZA::simulate" << std::endl;

    std::cout << "SimulatorXYZA::simulate: track points: " << track.points.size() << std::endl;

    raster.x0 = raster.x1 = track.points.front().position.x;

    for(TrackPointList::const_iterator i = track.points.begin(); i != track.points.end(); ++i) {

        Real distance = sqrt(i->position.y*i->position.y + i->position.z*i->position.z);

        if (raster.max_value < distance) raster.max_value = distance;

        if (raster.x0 > i->position.x) raster.x0 = i->position.x;

        if (raster.x1 < i->position.x) raster.x1 = i->position.x;

    }

    raster.raster.fill(raster.max_value);

    if (raster.x1 + 1e-5 < raster.x0) {

        std::cout << "SimulatorXYZA::simulate: zero bound" << std::endl;

        return;

    }

    Real x0 = raster.x0;

    Real dx = raster.x1 - x0;

    Real h = raster.raster.get_height();

    Real w = raster.raster.get_width();

    Vector2 scale_to_raster( w/dx , 1.0/360.0 );

    Vector2 scale_to_tool  ( dx/w , 2*M_PI/h  );

    PolarToolPainter painter(&tool, scale_to_tool, 1);

    unsigned long long max_count = 2ull*1024*1024*1024/sizeof(points.front());

    int index = 0;

    for(TrackPointList::const_iterator i = track.points.begin(); i != track.points.end(); ++i, ++index) {

        painter.distance = sqrt(i->position.y*i->position.y + i->position.z*i->position.z);

        if (painter.distance < 1e-5) painter.distance = 1e-5;

        Vector2 position(

            scale_to_raster.x * (i->position.x - x0),

            scale_to_raster.y * i->angle );

        position.y = (position.y - floor(position.y))*h;

        Vector2 rxy = painter.get_radius_xy();

        raster.raster.touch(painter, position, index, points);

        if (position.y - rxy.y < 0)

            raster.raster.touch(painter, Vector2(position.x, position.y + h), index, points);

        if (position.y + rxy.y > h)

            raster.raster.touch(painter, Vector2(position.x, position.y - h), index, points);

        if (points.size() >= max_count) {

            std::cerr << "SimulatorXYZA::simulate: reached memory limit at time " <<  i->time << std::endl;

            break;

        }

        if (index % 1000 == 0)

            std::cout << "." << std::flush;

        if (index % 100000 == 0)

            std::cout << std::endl << std::flush;

    }

    simu_index = (int)points.size();

    std::cout << std::endl;

    std::cout << "SimulatorXYZA::simulate: done" << std::endl;

}

void SimulatorXYZA::scroll_to(int index) {

    while(simu_index < (int)points.size() && points[simu_index].index <= index)

        points[simu_index++].apply_next(raster.raster);

    while(simu_index > 0 && points[simu_index-1].index > index)

        points[--simu_index].apply_prev(raster.raster);

}