#include "Scene_implicit_function_item.h"
#include <QColor>
#include <QApplication>
#include <map>
#include <CGAL/gl.h>
#include <CGAL/Simple_cartesian.h>
#include <QGLViewer/manipulatedFrame.h>

#include "Color_ramp.h"
#include <CGAL/Three/Viewer_interface.h>

#include <CGAL/double.h>

inline
bool is_nan(double d)
{
    return !CGAL::Is_valid<double>()( d );
}

struct Scene_implicit_function_item_priv
{
  Scene_implicit_function_item_priv(Implicit_function_interface* f, Scene_implicit_function_item* parent)
    : function_(f)
    , frame_(new ManipulatedFrame())
    , need_update_(true)
    , grid_size_(SCENE_IMPLICIT_GRID_SIZE)
    , max_value_(0.)
    , min_value_(0.)
    , blue_color_ramp_()
    , red_color_ramp_()
    , textureId(-1)
  {
    init=false;
    item = parent;
    texture = new Texture(grid_size_-1,grid_size_-1);
    blue_color_ramp_.build_blue();
    red_color_ramp_.build_red();
  }
  ~Scene_implicit_function_item_priv()
  {
    delete frame_;
  }
  typedef qglviewer::Vec                  Point;
  typedef std::pair <Point,double>        Point_value;
  typedef qglviewer::ManipulatedFrame ManipulatedFrame;
  void compute_min_max();
  void initialize_buffers(CGAL::Three::Viewer_interface *viewer) const;
  void compute_vertices_and_texmap(void);
  void compute_texture(int, int);
  enum VAOs {
      Plane = 0,
      BBox,
      Grid,
      NbOfVaos
  };
  enum VBOs {
      Quad_vertices = 0,
      TexMap,
      Cube_vertices,
      Grid_vertices,
      NbOfVbos
  };
  Implicit_function_interface* function_;
  ManipulatedFrame* frame_;
  mutable bool need_update_;
  int grid_size_;
  double max_value_;
  double min_value_;
  bool init;
  mutable Point_value implicit_grid_[SCENE_IMPLICIT_GRID_SIZE][SCENE_IMPLICIT_GRID_SIZE];
  Color_ramp blue_color_ramp_;
  Color_ramp red_color_ramp_;
  std::vector<float> positions_cube;
  std::vector<float> positions_grid;
  std::vector<float> positions_tex_quad;
  std::vector<float> texture_map;
  Texture *texture;
  GLuint vao;
  GLuint buffer[4];
  mutable QOpenGLShaderProgram *program;
  mutable GLuint textureId;
  Scene_implicit_function_item* item;
};

void Scene_implicit_function_item_priv::initialize_buffers(CGAL::Three::Viewer_interface *viewer = 0) const
{
    if(GLuint(-1) == textureId) {
        viewer->glGenTextures(1, &textureId);
    }

    //vao fot the cutting plane
    {
        program = item->getShaderProgram(Scene_implicit_function_item::PROGRAM_WITH_TEXTURE, viewer);
        program->bind();
        item->vaos[Plane]->bind();


        item->buffers[Quad_vertices].bind();
        item->buffers[Quad_vertices].allocate(positions_tex_quad.data(),
                            static_cast<int>(positions_tex_quad.size()*sizeof(float)));
        program->enableAttributeArray("vertex");
        program->setAttributeBuffer("vertex",GL_FLOAT,0,3);
        item->buffers[Quad_vertices].release();

        item->buffers[TexMap].bind();
        item->buffers[TexMap].allocate(texture_map.data(),
                            static_cast<int>(texture_map.size()*sizeof(float)));
        program->enableAttributeArray("v_texCoord");
        program->setAttributeBuffer("v_texCoord",GL_FLOAT,0,2);
        item->buffers[TexMap].release();
        program->setAttributeValue("normal", QVector3D(0.f,0.f,0.f));

        program->release();
        item->vaos[Plane]->release();
    }
    //vao fot the bbox
    {
        program = item->getShaderProgram(Scene_implicit_function_item::PROGRAM_WITHOUT_LIGHT, viewer);
        program->bind();
        item->vaos[BBox]->bind();


        item->buffers[Cube_vertices].bind();
        item->buffers[Cube_vertices].allocate(positions_cube.data(),
                            static_cast<int>(positions_cube.size()*sizeof(float)));
        program->enableAttributeArray("vertex");
        program->setAttributeBuffer("vertex",GL_FLOAT,0,3);
        item->buffers[Cube_vertices].release();

        program->release();
        item->vaos[BBox]->release();
    }
    //vao fot the grid
    {
        program = item->getShaderProgram(Scene_implicit_function_item::PROGRAM_WITHOUT_LIGHT, viewer);
        program->bind();
        item->vaos[Grid]->bind();


        item->buffers[Grid_vertices].bind();
        item->buffers[Grid_vertices].allocate(positions_grid.data(),
                            static_cast<int>(positions_grid.size()*sizeof(float)));
        program->enableAttributeArray("vertex");
        program->setAttributeBuffer("vertex",GL_FLOAT,0,3);
        item->buffers[Grid_vertices].release();
        program->release();
        item->vaos[Grid]->release();
    }



    viewer->glPixelStorei(GL_UNPACK_ALIGNMENT, 1);
    viewer->glBindTexture(GL_TEXTURE_2D, textureId);
    viewer->glTexImage2D(GL_TEXTURE_2D,
                 0,
                 GL_RGB,
                 texture->getWidth(),
                 texture->getHeight(),
                 0,
                 GL_RGB,
                 GL_UNSIGNED_BYTE,
                 texture->getData());
       viewer->glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
       viewer->glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
       viewer->glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S,GL_CLAMP_TO_EDGE );
       viewer->glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T,GL_CLAMP_TO_EDGE );

       item->are_buffers_filled = true;
}

void Scene_implicit_function_item_priv::compute_vertices_and_texmap(void)
{
    QApplication::setOverrideCursor(Qt::WaitCursor);
    positions_tex_quad.resize(0);
    positions_cube.resize(0);
    positions_grid.resize(0);
    texture_map.resize(0);

    const CGAL::Three::Scene_item::Bbox& b = item->bbox();
    float x,y,z;
    z = (b.zmax()+b.zmin())/2.0;
    x = (b.xmax()+b.xmin())/2.0;
    y = (b.ymax()+b.ymin())/2.0;
    // The Quad
    {


        //A
        positions_tex_quad.push_back(b.xmin()-x);
        positions_tex_quad.push_back(b.ymin()-z);
        positions_tex_quad.push_back(0);


        //B
        positions_tex_quad.push_back(b.xmin()-x);
        positions_tex_quad.push_back(b.ymax()-y);
        positions_tex_quad.push_back(0);


        //C
        positions_tex_quad.push_back(b.xmax()-x);
        positions_tex_quad.push_back(b.ymax()-y);
        positions_tex_quad.push_back(0);



        //A
        positions_tex_quad.push_back(b.xmin()-x);
        positions_tex_quad.push_back(b.ymin()-y);
        positions_tex_quad.push_back(0);


        //C
        positions_tex_quad.push_back(b.xmax()-x);
        positions_tex_quad.push_back(b.ymax()-y);
        positions_tex_quad.push_back(0);


        //D
        positions_tex_quad.push_back(b.xmax()-x);
        positions_tex_quad.push_back(b.ymin()-y);
        positions_tex_quad.push_back(0);



        texture_map.push_back(0.0);
        texture_map.push_back(0.0);

        texture_map.push_back(0.0);
        texture_map.push_back(1.0);

        texture_map.push_back(1.0);
        texture_map.push_back(1.0);

        texture_map.push_back(0.0);
        texture_map.push_back(0.0);

        texture_map.push_back(1.0);
        texture_map.push_back(1.0);

        texture_map.push_back(1.0);
        texture_map.push_back(0.0);



    }
    //The grid
    {
      double dx((b.xmax()-b.xmin())/10.0), dy((b.ymax()-b.ymin())/10.0);
        for(int u = 0; u < 11; u++)
        {

            positions_grid.push_back(b.xmin()-x + dx* u);
            positions_grid.push_back(b.ymin()-y);
            positions_grid.push_back(0);

            positions_grid.push_back(b.xmin()-x + dx* u);
            positions_grid.push_back(b.ymax()-y);
            positions_grid.push_back(0);
        }
        for(int v=0; v<11; v++)
        {

            positions_grid.push_back(b.xmin()-x);
            positions_grid.push_back(b.ymin()-y + v * dy);
            positions_grid.push_back(0);

            positions_grid.push_back(b.xmax()-x);
            positions_grid.push_back(b.ymin()-y + v * dy);
            positions_grid.push_back(0);
        }

    }
    //the Box
    {
      const qglviewer::Vec offset = static_cast<CGAL::Three::Viewer_interface*>(QGLViewer::QGLViewerPool().first())->offset();

        positions_cube.push_back(b.xmin()+offset.x);
        positions_cube.push_back(b.ymin()+offset.y);
        positions_cube.push_back(b.zmin()+offset.z);

        positions_cube.push_back(b.xmin()+offset.x);
        positions_cube.push_back(b.ymin()+offset.y);
        positions_cube.push_back(b.zmax()+offset.z);


        positions_cube.push_back(b.xmin()+offset.x);
        positions_cube.push_back(b.ymin()+offset.y);
        positions_cube.push_back(b.zmin()+offset.z);


        positions_cube.push_back(b.xmin()+offset.x);
        positions_cube.push_back(b.ymax()+offset.y);
        positions_cube.push_back(b.zmin()+offset.z);


        positions_cube.push_back(b.xmin()+offset.x);
        positions_cube.push_back(b.ymin()+offset.y);
        positions_cube.push_back(b.zmin()+offset.z);


        positions_cube.push_back(b.xmax()+offset.x);
        positions_cube.push_back(b.ymin()+offset.y);
        positions_cube.push_back(b.zmin()+offset.z);


        positions_cube.push_back(b.xmax()+offset.x);
        positions_cube.push_back(b.ymin()+offset.y);
        positions_cube.push_back(b.zmin()+offset.z);


        positions_cube.push_back(b.xmax()+offset.x);
        positions_cube.push_back(b.ymax()+offset.y);
        positions_cube.push_back(b.zmin()+offset.z);


        positions_cube.push_back(b.xmax()+offset.x);
        positions_cube.push_back(b.ymin()+offset.y);
        positions_cube.push_back(b.zmin()+offset.z);


        positions_cube.push_back(b.xmax()+offset.x);
        positions_cube.push_back(b.ymin()+offset.y);
        positions_cube.push_back(b.zmax()+offset.z);


        positions_cube.push_back(b.xmin()+offset.x);
        positions_cube.push_back(b.ymax()+offset.y);
        positions_cube.push_back(b.zmin()+offset.z);


        positions_cube.push_back(b.xmin()+offset.x);
        positions_cube.push_back(b.ymax()+offset.y);
        positions_cube.push_back(b.zmax()+offset.z);


        positions_cube.push_back(b.xmin()+offset.x);
        positions_cube.push_back(b.ymax()+offset.y);
        positions_cube.push_back(b.zmin()+offset.z);


        positions_cube.push_back(b.xmax()+offset.x);
        positions_cube.push_back(b.ymax()+offset.y);
        positions_cube.push_back(b.zmin()+offset.z);


        positions_cube.push_back(b.xmax()+offset.x);
        positions_cube.push_back(b.ymax()+offset.y);
        positions_cube.push_back(b.zmin()+offset.z);


        positions_cube.push_back(b.xmax()+offset.x);
        positions_cube.push_back(b.ymax()+offset.y);
        positions_cube.push_back(b.zmax()+offset.z);


        positions_cube.push_back(b.xmin()+offset.x);
        positions_cube.push_back(b.ymin()+offset.y);
        positions_cube.push_back(b.zmax()+offset.z);


        positions_cube.push_back(b.xmin()+offset.x);
        positions_cube.push_back(b.ymax()+offset.y);
        positions_cube.push_back(b.zmax()+offset.z);


        positions_cube.push_back(b.xmin()+offset.x);
        positions_cube.push_back(b.ymin()+offset.y);
        positions_cube.push_back(b.zmax()+offset.z);


        positions_cube.push_back(b.xmax()+offset.x);
        positions_cube.push_back(b.ymin()+offset.y);
        positions_cube.push_back(b.zmax()+offset.z);


        positions_cube.push_back(b.xmax()+offset.x);
        positions_cube.push_back(b.ymax()+offset.y);
        positions_cube.push_back(b.zmax()+offset.z);


        positions_cube.push_back(b.xmin()+offset.x);
        positions_cube.push_back(b.ymax()+offset.y);
        positions_cube.push_back(b.zmax()+offset.z);


        positions_cube.push_back(b.xmax()+offset.x);
        positions_cube.push_back(b.ymax()+offset.y);
        positions_cube.push_back(b.zmax()+offset.z);


        positions_cube.push_back(b.xmax()+offset.x);
        positions_cube.push_back(b.ymin()+offset.y);
        positions_cube.push_back(b.zmax()+offset.z);

    }

    //The texture
    for( int i=0 ; i < texture->getWidth() ; i++ )
    {
        for( int j=0 ; j < texture->getHeight() ; j++)
        {
            compute_texture(i,j);
        }
    }
    QApplication::restoreOverrideCursor();
}

Scene_implicit_function_item::
Scene_implicit_function_item(Implicit_function_interface* f)
    :CGAL::Three::Scene_item(4,3)
{
  d = new Scene_implicit_function_item_priv(f, this);
  //Generates an integer which will be used as ID for each buffer
  d->compute_min_max();
  connect(d->frame_, SIGNAL(modified()), this, SLOT(plane_was_moved()));
  plane_was_moved();
  invalidateOpenGLBuffers();
}


Scene_implicit_function_item::~Scene_implicit_function_item()
{
  delete d;
}


void
Scene_implicit_function_item::compute_bbox() const
{
    _bbox = d->function_->bbox();
}

void
Scene_implicit_function_item::draw(CGAL::Three::Viewer_interface* viewer) const
{
  if(!d->init)
  {
    compute_function_grid();
    double offset_x = (bbox().xmin() + bbox().xmax()) / 2;
    double offset_y = (bbox().ymin() + bbox().ymax()) / 2;
    double offset_z = (bbox().zmin() + bbox().zmax()) / 2;
    const qglviewer::Vec offset = static_cast<CGAL::Three::Viewer_interface*>(QGLViewer::QGLViewerPool().first())->offset();
    d->frame_->setPosition(offset_x+offset.x, offset_y+offset.y, offset_z+offset.z);
    d->frame_->setOrientation(1., 0, 0, 0);
    d->init = true;
  }
    if(!are_buffers_filled)
        d->initialize_buffers(viewer);
    if(d->frame_->isManipulated()) {
        if(d->need_update_) {
            compute_function_grid();
            d->need_update_ = false;
        }
    }
    vaos[Scene_implicit_function_item_priv::Plane]->bind();
    viewer->glActiveTexture(GL_TEXTURE0);
    viewer->glBindTexture(GL_TEXTURE_2D, d->textureId);
    attribBuffers(viewer, PROGRAM_WITH_TEXTURE);
    QMatrix4x4 f_mat;
    GLdouble d_mat[16];
    d->frame_->getMatrix(d_mat);
    //Convert the GLdoubles matrices in GLfloats
    for (int i=0; i<16; ++i){
        f_mat.data()[i] = GLfloat(d_mat[i]);
    }
    d->program = getShaderProgram(PROGRAM_WITH_TEXTURE);
    d->program->bind();
    d->program->setUniformValue("f_matrix", f_mat);
    d->program->setUniformValue("light_amb", QVector4D(1.f,1.f,1.f,1.f));
    d->program->setUniformValue("light_diff", QVector4D(0.f,0.f,0.f,1.f));
    d->program->setAttributeValue("color_facets", QVector3D(1.f,1.f,1.f));
    viewer->glDrawArrays(GL_TRIANGLES, 0, static_cast<GLsizei>(d->positions_tex_quad.size()/3));
    vaos[Scene_implicit_function_item_priv::Plane]->release();
    d->program->release();
}

void
Scene_implicit_function_item::drawEdges(CGAL::Three::Viewer_interface* viewer) const
{
    if(!are_buffers_filled)
        d->initialize_buffers(viewer);
    vaos[Scene_implicit_function_item_priv::BBox]->bind();
    attribBuffers(viewer, PROGRAM_WITHOUT_LIGHT);
    d->program = getShaderProgram(PROGRAM_WITHOUT_LIGHT);
    d->program->bind();
    d->program->setAttributeValue("colors", QVector3D(0.f,0.f,0.f));
    viewer->glDrawArrays(GL_LINES, 0, static_cast<GLsizei>(d->positions_cube.size()/3));
    vaos[Scene_implicit_function_item_priv::BBox]->release();
    vaos[Scene_implicit_function_item_priv::Grid]->bind();
    QMatrix4x4 f_mat;
    GLdouble d_mat[16];
    d->frame_->getMatrix(d_mat);
    //Convert the GLdoubles matrices in GLfloats
    for (int i=0; i<16; ++i){
        f_mat.data()[i] = double(d_mat[i]);
    }
    d->program->setUniformValue("f_matrix", f_mat);
    d->program->setAttributeValue("colors", QVector3D(0.6f, 0.6f, 0.6f));
    viewer->glDrawArrays(GL_LINES, 0, static_cast<GLsizei>(d->positions_grid.size()/3));
    vaos[Scene_implicit_function_item_priv::Grid]->release();
    d->program->release();
}

QString
Scene_implicit_function_item::toolTip() const
{
    return tr("<p>Function <b>%1</b>")
            .arg(this->name());
}

bool
Scene_implicit_function_item::supportsRenderingMode(RenderingMode m) const
{
    switch ( m )
    {
    case Splatting:
    case Gouraud:
        return false;

    case Points:
    case Wireframe:
    case Flat:
    case FlatPlusEdges:
        return true;

    default:
        return false;
    }

    return false;
}

void Scene_implicit_function_item_priv::compute_texture(int i, int j)
{
    double v = (implicit_grid_[i][j]).second;

    if(is_nan(v)) {
        texture->setData(i,j,51,51,51);
    } else
        // determines grey level
        if ( v > 0 )
        {
            v = v/max_value_;
            GLdouble r = red_color_ramp_.r(v), g = red_color_ramp_.g(v), b = red_color_ramp_.b(v);
            texture->setData(i,j,255*r,255*g,255*b);
        }
        else
        {
            v = v/min_value_;
            GLdouble r = blue_color_ramp_.r(v), g = blue_color_ramp_.g(v), b = blue_color_ramp_.b(v);
            texture->setData(i,j,255*r,255*g,255*b);
        }
}


void
Scene_implicit_function_item::
compute_function_grid() const
{
    const qglviewer::Vec offset = static_cast<CGAL::Three::Viewer_interface*>(QGLViewer::QGLViewerPool().first())->offset();
    typedef CGAL::Simple_cartesian<double>  K;
    typedef K::Aff_transformation_3         Aff_transformation;
    typedef K::Point_3                      Point_3;

    // Get transformation
    const GLdouble* m = d->frame_->matrix();

    // OpenGL matrices are row-major matrices
    Aff_transformation t (m[0], m[4], m[8], m[12]-offset.x,
            m[1], m[5], m[9], m[13]-offset.y,
            m[2], m[6], m[10], m[14]-offset.z);

    double diag = (CGAL::sqrt((bbox().xmax()-bbox().xmin())*(bbox().xmax()-bbox().xmin()) + (bbox().ymax()-bbox().ymin())*(bbox().ymax()-bbox().ymin())  + (bbox().zmax()-bbox().zmin())*(bbox().zmax()-bbox().zmin()) )) * .6;

    const double dx = diag;
    const double dy = diag;
    const double z (0);

    int nb_quad = d->grid_size_ - 1;

    for(int i=0 ; i<d->grid_size_ ; ++i)
    {
        double x = -diag/2. + double(i)/double(nb_quad) * dx;

        for(int j=0 ; j<d->grid_size_ ; ++j)
        {
            double y = -diag/2. + double(j)/double(nb_quad) * dy;

            Point_3 query = t( Point_3(x, y, z) );
            double v = d->function_->operator()(query.x(), query.y(), query.z());

            d->implicit_grid_[i][j] = Scene_implicit_function_item_priv::Point_value(Scene_implicit_function_item_priv::Point(query.x(),query.y(),query.z()),v);
        }
    }

    // Update
    const_cast<Scene_implicit_function_item*>(this)->invalidateOpenGLBuffers();

}

void
Scene_implicit_function_item_priv::
compute_min_max()
{
    if(function_->get_min_max(min_value_, max_value_))
        return;

    double probes_nb = double(grid_size_) / 2;

    // Probe bounding box
    const CGAL::Three::Scene_item::Bbox& b = item->bbox();

    for ( int i = 0 ; i <= probes_nb ; ++i )
    {
        double x = b.xmin() + double(i) * (b.xmax() - b.xmin()) / probes_nb;

        for ( int j = 0 ; j <= probes_nb ; ++j )
        {
            double y = b.ymin() + double(j) * (b.ymax() - b.ymin()) / probes_nb;

            for ( int k = 0 ; k <= probes_nb ; ++k )
            {
                double z = b.zmin() + double(k) * (b.zmax() - b.zmin()) / probes_nb;

                double v = (*function_)(x,y,z);
                if(is_nan(v)) continue;
                max_value_ = (std::max)(v,max_value_);
                min_value_ = (std::min)(v,min_value_);
            }
        }
    }
}

void
Scene_implicit_function_item::invalidateOpenGLBuffers()
{
    Scene_item::invalidateOpenGLBuffers();
    compute_bbox();
    d->compute_vertices_and_texmap();
    are_buffers_filled = false;
}


void Scene_implicit_function_item::updateCutPlane()
{ // just handle deformation - paint like selection is handled in eventFilter()
  if(d->need_update_) {
    compute_function_grid();
    d->compute_vertices_and_texmap();
    d->need_update_= false;
  }
}

Implicit_function_interface* Scene_implicit_function_item::function() const { return d->function_; }
Scene_implicit_function_item::ManipulatedFrame* Scene_implicit_function_item::manipulatedFrame() { return d->frame_; }
void Scene_implicit_function_item::plane_was_moved() { d->need_update_ = true; QTimer::singleShot(0, this, SLOT(updateCutPlane()));}