/*
 * Copyright (c) 2015 Tricoire Sebastien 3dsman@free.fr
 *
 * This software is provided 'as-is', without any express or implied
 * warranty.  In no event will the authors be held liable for any damages
 * arising from the use of this software.
 *
 * Permission is granted to anyone to use this software for any purpose,
 * including commercial applications, and to alter it and redistribute it
 * freely, subject to the following restrictions:
 *
 * 1. The origin of this software must not be misrepresented; you must not
 * claim that you wrote the original software. If you use this software
 * in a product, an acknowledgment in the product documentation would be
 * appreciated but is not required.
 * 2. Altered source versions must be plainly marked as such, and must not be
 * misrepresented as being the original software.
 * 3. This notice may not be removed or altered from any source distribution.
 *
 */

#include "OE_interfaceDisplay.h"
#include "OE_utils.h"

#include "OE_controller.h"

#include <iostream>

#include <GL/gl.h>
#include <cstdlib>
#include <math.h>
#include <cstdio>
#include <cstring>
#include <algorithm> 

#include "actions/OE_actionsCurves.h"
#include "actions/OE_actionsStitchs.h"
#include "actions/OE_actionsLineStitchs.h"
#include "actions/OE_actionsBirailStitchs.h"
#include "actions/OE_actionsSelection.h"
#include "actions/OE_actionsThreads.h"

OE_interfaceDisplay::OE_interfaceDisplay(OE_document* document)
{
    editDisplay = new OE_editDisplay();
	commandDisplay = new OE_commandDisplay();
	curDisplay = editDisplay;
	setDocument(document);
}

OE_interfaceDisplay::~OE_interfaceDisplay()
{
    delete editDisplay;
    delete commandDisplay;
}

bool OE_interfaceDisplay::setDocument(OE_document * document)
{

    this->document = document;
    if(document)
    {
        editDisplay->setDocument(document);
        commandDisplay->setDocument(document);
    }
    return true;
}


bool OE_interfaceDisplay::setController(OE_controller* controller)
{
    this->controller = controller;
    if(controller)
    {
        if (editDisplay) editDisplay->setController(controller);
        if (commandDisplay) commandDisplay->setController(controller);
    }
    return true;
}

void OE_interfaceDisplay::showAll()
{
    if (curDisplay) curDisplay->showAll();
}

void OE_interfaceDisplay::mouse_Pos(double x, double y)
{
    OE_display::mouse_Pos( x, y);

    vector_2dt oldClosestPoint = editClosestPoint;
    if (curDisplay) curDisplay->mouse_Pos(x, y);


	if(toolAction == 1)
	{
		 controller->editActionMoveSelection(vector_2d((mouse.x - clicOldMouse.x)*zoom*2,(mouse.y - clicOldMouse.y)*zoom*2));
		 return;
	}

    if (editSubcurve)
    {
        editClosestPoint = controller->getClosestPoint(editSubcurve->getCurve(),absMouse);
    }

    if ((editionState == NewSubcurve))
    {

        closestCurve = controller->getClosestCurve(absMouse, editClosestPoint);
    }

    if ((selControlCurve)&&(selControlType == 0))
        controller->editActionMovePointCurve(vector_2d((mouse.x - clicOldMouse.x)*zoom*2,(mouse.y - clicOldMouse.y)*zoom*2));

    if (selControlStitch)
    {
        OE_linestitch* tmpLineStitch = dynamic_cast<OE_linestitch*>(selControlStitch);
        OE_birailstitch* tmpBirailStitch = dynamic_cast<OE_birailstitch*>(selControlStitch);
        if (((tmpLineStitch)&&(selControlType == 0))||((tmpBirailStitch)&&((selControlType == 0)||(selControlType == 1))))
        {
            if (editionState == NewSubcurve)
            {
                editionState = TraceSubcurve;
                if(oldClosestPoint.t>editClosestPoint.t) controller->editActionAddSubcurvePosSwitchDir();
            }

            if (editionState == TraceSubcurve)
            {
                controller->editActionAddSubcurvePosEnd(editClosestPoint.t);
            }
            else
                controller->editActionSetSubcurvePos(editClosestPoint.t);
        }
    }

}

void OE_interfaceDisplay::mouse_Button(int button, int action, int mods)
{
    modState = mods;
    if (curDisplay) curDisplay->mouse_Button(button, action, mods);
    if((curDisplay == editDisplay)&&(button == GLFW_MOUSE_BUTTON_LEFT)&&(action==GLFW_PRESS))
    {
        if(editionState==NewSubcurve)
        {
            if(!closestCurve)
                editionState=None;
            else
            {
                controller->addAction(new OE_actionJoincurveAddSubCurve(curJoincurve,-1,closestCurve,editClosestPoint.t,editClosestPoint.t,false));
                selControlStitch = controller->getStitch(-1);
                selControlType = 0;
                editSubcurve = curJoincurve->getCurve(-1);
            }
        }
        else
        {
			clicOldMouse = mouse;
			selection = true;
			select();
        }
    }

    if((curDisplay == editDisplay)&&(button == GLFW_MOUSE_BUTTON_LEFT)&&(action==GLFW_RELEASE))
    {
        if(selection)
        {
            if ((mouse-clicOldMouse).len()>5)
                selectBox();

            selectApply(selectedCurves, selectedStitches);
            selection = false;
        }
        if(selControlCurve)selControlCurve = 0;
        if(selControlStitch)selControlStitch = 0;

        if((editionState == MoveSubcurve)||(editionState == MovePointcurve)) editionState = None;

        if (editionState == TraceSubcurve)
        {
            editionState = NewSubcurve;
            editSubcurve = 0;
        }
		if(toolAction) toolAction = 0;
    }
}

void OE_interfaceDisplay::scroll(double xoffset, double yoffset)
{
    if (curDisplay) curDisplay->scroll(xoffset, yoffset);
}

void OE_interfaceDisplay::resize(int width, int height)
{
    if (curDisplay) curDisplay->resize(width, height);
}

void OE_interfaceDisplay::key(int key, int scancode, int action, int mods)
{
    modState = mods;
    if((action == GLFW_PRESS)&&(curDisplay == editDisplay))
    {
        if ((key == GLFW_KEY_Z)&&(mods == GLFW_MOD_CONTROL))
        {
			if (controller) controller->undoAction();
		}else if ((key == GLFW_KEY_S)&&(mods == GLFW_MOD_CONTROL))
		{
			if (document) document->saveToFile("../roger.oe");
		}else if ((key == GLFW_KEY_O)&&(mods == GLFW_MOD_CONTROL))
		{
			if (document) document->loadFromFile("../roger.oe");
		}else if ((key == GLFW_KEY_Z)&&(mods == GLFW_MOD_CONTROL+GLFW_MOD_SHIFT))
        {
			if (controller) controller->redoAction();
		}else if (mods == 0)
        {
            if (key == GLFW_KEY_C)
            {
				if (controller) controller->toggleCloseSelectedCurve();
            }
		}else if (key == GLFW_KEY_R)
            {
            if (editionState == TraceSubcurve)
            {
				controller->editActionAddSubcurvePosSwitchDir();
            }else if (editionState == MoveSubcurve)
			{
				controller->editActionSetSubcurvePosSwitchDir();
			}
        }
    }

    if(action == GLFW_RELEASE)
    {
        if(key == GLFW_KEY_F5)
        {
            if(curDisplay == editDisplay)
            {
                if (controller) controller->generateInstructions();
                curDisplay = commandDisplay;
            }
            else
                curDisplay = editDisplay;
		}else if(key == GLFW_KEY_L)
        {
            if (controller)
            {
                controller->newLineStitch(0.4,1.0,2,0);
                document->selectedStitchs.push_back(controller->getLineStitch(-1));
                curJoincurve = controller->getLineStitch(-1)->getJoincurve();
                closestCurve = controller->getClosestCurve(absMouse, editClosestPoint);

                editionState = NewSubcurve;
                joincurve_id = 0;
            }
		}else if(key == GLFW_KEY_B)
        {
            if (controller)
            {
                controller->newBirailStitch(false, false, 0,0,0.7);
                document->selectedStitchs.push_back(controller->getBirailStitch(-1));
                curJoincurve = controller->getBirailStitch(-1)->getJoincurve1();
                closestCurve = controller->getClosestCurve(absMouse, editClosestPoint);

                editionState = NewSubcurve;
                joincurve_id = 1;
            }
		}else if(key == GLFW_KEY_ENTER)
        {
            if(editionState == NewSubcurve)
            {
                if (joincurve_id == 0)
                {
                    editionState = None;
                    curJoincurve = 0;
                }else
                {
                    curJoincurve = controller->getBirailStitch(-1)->getJoincurve2();
                    closestCurve = controller->getClosestCurve(absMouse, editClosestPoint);
                    joincurve_id -= 1;
                }
            }
		}else if (key == GLFW_KEY_R)
            {
            if (editionState == TraceSubcurve)
            {
                controller->editActionAddSubcurvePosSwitchDir();
            }else if (editionState == MoveSubcurve)
                controller->editActionSetSubcurvePosSwitchDir();
		}else if (key == GLFW_KEY_KP_ADD)
		{
			if(mods == GLFW_MOD_CONTROL)
				controller->addAction(new OE_actionScaleSelectedLineStitchLen(1.1));
			else
				controller->addAction(new OE_actionScaleSelectedLineStitchWidth(1.1));
		}else if (key == GLFW_KEY_KP_SUBTRACT)
		{
			if(mods == GLFW_MOD_CONTROL)
				controller->addAction(new OE_actionScaleSelectedLineStitchLen(1/1.1));
			else
				controller->addAction(new OE_actionScaleSelectedLineStitchWidth(1/1.1));
		}else if (key == GLFW_KEY_KP_1)
		{
			controller->addAction(new OE_actionSetSelectedLineStitchMotif(0));
		}else if (key == GLFW_KEY_KP_2)
		{
			controller->addAction(new OE_actionSetSelectedLineStitchMotif(1));
		}else if (key == GLFW_KEY_KP_3)
		{
			controller->addAction(new OE_actionSetSelectedLineStitchMotif(2));
		}else if (key == GLFW_KEY_KP_4)
		{
			controller->addAction(new OE_actionSetSelectedLineStitchMotif(3));
		}else if (key == GLFW_KEY_KP_5)
		{
			controller->addAction(new OE_actionSetSelectedLineStitchMotif(4));
		}else if (key == GLFW_KEY_KP_6)
		{
			controller->addAction(new OE_actionSetSelectedLineStitchMotif(5));
		}



    }

    curDisplay->key( key,  scancode,  action,  mods);
}

bool OE_interfaceDisplay::selectBox()
{
    selectedCurves.clear();
    selectedStitches.clear();
    vector_2d clicAbsMouse = viewPos+(clicOldMouse-vector_2d(width,height)/2)*zoom*2;
    float tmpXMin, tmpYMin, tmpXMax, tmpYMax;

    std::list<OE_pointcurve*>::iterator curve = document->curves.begin();
    while (curve != document->curves.end())
    {
         BoundingBox box = (*curve)->getBound();
         if ((box.getMin().x>minf(clicAbsMouse.x,absMouse.x))
             &&(box.getMax().x<maxf(clicAbsMouse.x,absMouse.x))
             &&(box.getMin().y>minf(clicAbsMouse.y,absMouse.y))
             &&(box.getMax().y<maxf(clicAbsMouse.y,absMouse.y)))
             selectedCurves.push_back(*curve);
         curve++;
    }

    std::list<OE_stitchs*>::iterator stitch = document->stitchs.begin();
    while (stitch != document->stitchs.end())
    {

         BoundingBox box = (*stitch)->getBound();
         if ((box.getMin().x>minf(clicAbsMouse.x,absMouse.x))
             &&(box.getMax().x<maxf(clicAbsMouse.x,absMouse.x))
             &&(box.getMin().y>minf(clicAbsMouse.y,absMouse.y))
             &&(box.getMax().x<maxf(clicAbsMouse.y,absMouse.y)))
             selectedStitches.push_back(*stitch);
         stitch++;
    }

}


bool OE_interfaceDisplay::checkPicking()
{
	if(toolAction == 1)
	{
		selection = false;
		return controller->addAction(new OE_actionMoveSelection(vector_2d(0,0)));
	}
    if(selControlStitch)
	{
		selection = false;

        OE_linestitch* tmpLineStitch = dynamic_cast<OE_linestitch*>(selControlStitch);

        if ((tmpLineStitch)&&(selControlType == 0))
        {
            // get the corresponding subcurve
            editSubcurve = tmpLineStitch->getJoincurve()->getCurve(selControlIndex/2);
            if (editSubcurve)
            {
                // get the closest point t value
                vector_2dt pos = editSubcurve->getCurve()->closestPoint(absMouse);
                // and add an action to move the point
                controller->addAction(new OE_actionSetSubcurvePos( editSubcurve, selControlIndex%2, pos.t));
                editionState = MoveSubcurve;
                return true;
            }
            return false;
        }

        OE_birailstitch* tmpBirailStitch = dynamic_cast<OE_birailstitch*>(selControlStitch);

        if ((tmpBirailStitch)&&((selControlType == 0)||(selControlType == 1)))
        {
            // get the corresponding subcurve
            if (selControlType == 0)
                editSubcurve = tmpBirailStitch->getJoincurve1()->getCurve(selControlIndex/2);
            else if(selControlType == 1)
                editSubcurve = tmpBirailStitch->getJoincurve2()->getCurve(selControlIndex/2);
            if (editSubcurve)
            {
                // get the closest point t value
                vector_2dt pos = editSubcurve->getCurve()->closestPoint(absMouse);
                // and add an action to move the point
                controller->addAction(new OE_actionSetSubcurvePos( editSubcurve, selControlIndex%2, pos.t));
                editionState = MoveSubcurve;
                return true;
            }
            return false;
        }
        selControlStitch = 0;
    }

    if(selControlCurve)
	{

		selection = false;
        OE_pointcurve* tmpPointCurve = dynamic_cast<OE_pointcurve*>(selControlCurve);

        if ((tmpPointCurve)&&(selControlType == 0))
        {
            return controller->addAction(new OE_actionMovePointCurve( tmpPointCurve, selControlIndex + 1, vector_2d(0,0)));
            editionState = MovePointcurve;
        }
        selControlCurve = 0;

		return true;
	}
}

/** \brief set selection
 *
 * \return true if all is ok
 *
 */

bool OE_interfaceDisplay::select()
{
    GLuint selectBuf[64];
    GLint hits;

    glSelectBuffer(64, selectBuf);

    draw(true);
    hits = glRenderMode(GL_RENDER);

    selectedCurves.clear();
    selectedStitches.clear();

    GLuint i, nb_names, name, *ptr;

    //std::cout<<"hits : "<<"\r\n";
    ptr = (GLuint *) selectBuf;
    for (GLuint i = 0; i < hits; i++)  //  for each hit
    {
        bool recognized =false;
        GLuint names = *ptr;

        ptr+=3; //jump near and far info and goto first index
        if (*ptr == 0) //tools index
        {
            std::cout<<"  tool"<< std::endl;
            ptr++;
            if(*ptr == 0)
                std::cout<<"    zero"<< std::endl;
            if(*ptr == 1)
			{
                std::cout<<"    select tools"<< std::endl;
				toolAction = 1;
				checkPicking();
				return true;
			}
            ptr++;
        }else if (*ptr == 1) //stitches index
        {
            std::cout<<"  stitch"<< std::endl;
            ptr++;
            if (names==2)
            {
                std::cout<<"    select"<< std::endl;
                  //ptr++;
                  //std::cout<<"    stitch "<<*ptr;
                  if (document->stitchs.size() > *ptr)
                  {
                      std::list<OE_stitchs*>::iterator it = document->stitchs.begin();
                      std::advance(it, *ptr);
                      selectedStitches.push_back(*it);
                      return true;
                  }
                  ptr++;
            } else if (names==4) //if name count correspond to a sub element (like control point)
            {
                std::cout<<"    sub element"<< std::endl;
                if(*ptr < document->selectedStitchs.size())
                {
                    std::list<OE_stitchs*>::iterator it = document->selectedStitchs.begin();
                    std::advance(it, *ptr++);
                    selControlStitch = *it;
                    selControlType = *ptr++;
                    selControlIndex = *ptr++;

                    checkPicking();

                    return true;
                }
                ptr+=4;
            }
        }else if (*ptr == 2) //curves index
        {
            std::cout<<"  curve"<< std::endl;
            ptr++;
            if (names==2)
            {
                std::cout<<"    select"<< std::endl;
                //ptr++;
                //std::cout<<"    curve "<<*ptr;
                if (document->curves.size() > *ptr)
                {
                    std::list<OE_pointcurve*>::iterator it = document->curves.begin();
                    std::advance(it, *ptr);
                    selectedCurves.push_back(*it);
                    return true;
                }
                ptr++;
            } else if (names==4) //if name count correspond to a sub element (like control point)
            {
                std::cout<<"    sub element"<< std::endl;

                if(*ptr < document->selectedCurves.size())
                {
                    std::cout<<"      curve id "<<*ptr << std::endl;
                    std::list<OE_pointcurve*>::iterator it = document->selectedCurves.begin();
                    std::advance(it, *ptr++);
                    selControlCurve = *it;
                    std::cout<<"      control type "<<*ptr <<std::endl;
                    selControlType = *ptr++;
                    std::cout<<"      control id "<<*ptr << std::endl;
                    selControlIndex = *ptr++;

                    checkPicking();

                    return true;
                }
                ptr+=4;
            }
        }

        //if (!recognized)
        //{
            ptr++;
            //std::cout<<" strange hit! probably a bug : "<< names<<std::endl; ptr+=3;
            //std::cout<<"   names are ";
            for (GLuint j = 0; j < names; j++) { /*  for each name */
               //std::cout<< *ptr<<" ";
               ptr++;
            }
            //std::cout<<std::endl;
        //}
    }

    return true;
}

bool OE_interfaceDisplay::selectApply(std::list<OE_pointcurve*> selectedCurves, std::list<OE_stitchs*> selectedStitches)
{
    if(selectedCurves.size())
        if(modState&GLFW_MOD_CONTROL)
            controller->unselectCurves(selectedCurves);
        else
            controller->selectCurves(selectedCurves, !(modState&GLFW_MOD_SHIFT));

    if(selectedStitches.size())
        if(modState&GLFW_MOD_CONTROL)
            controller->unselectStitches(selectedStitches);
        else
            controller->selectStitches(selectedStitches, !(modState&GLFW_MOD_SHIFT));

    if(!(selectedStitches.size())&&!(selectedCurves.size()))
    {
        controller->clearSelection();
    }

    return true;
}

/** \brief draw the document on screen
 *
 * \return true if all is ok
 *
 */

bool OE_interfaceDisplay::draw()
{
    draw(false);
}

/** \brief draw the document on screen
 *
 * \return true if all is ok
 *
 */

bool OE_interfaceDisplay::draw(bool select)
{
    if (select)
    {
        glRenderMode(GL_SELECT);
        //Init of name stack
        glInitNames();
        glPushName(0);
    }
	float wzoom, hzoom;
	
	glViewport(0, 0, width, height);
    glClearColor(220.0f/255.0f, 220.0f/255.0f, 220.0f/255.0f, 1.0f);

	glClear(GL_COLOR_BUFFER_BIT|GL_DEPTH_BUFFER_BIT);
	glEnable(GL_BLEND);
	glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
	glDisable(GL_TEXTURE_2D);
	glMatrixMode(GL_PROJECTION);
	glLoadIdentity();

    glEnd();

	wzoom = (float)width*zoom;
	hzoom = (float)height*zoom;

    if(select)
        glOrtho(absMouse.x-7*zoom, absMouse.x+7*zoom, absMouse.y+7*zoom, absMouse.y-7*zoom, -1, 1);
    else
        glOrtho(viewPos.x-wzoom, viewPos.x+wzoom, viewPos.y+hzoom, viewPos.y-hzoom, -1, 1);
	

	glMatrixMode(GL_MODELVIEW);
	glLoadIdentity();
	glDisable(GL_DEPTH_TEST);
	glColor4ub(255,255,255,255);
	glEnable(GL_BLEND);
	glBlendFunc(GL_SRC_ALPHA,GL_ONE_MINUS_SRC_ALPHA);
	
    if (document && curDisplay)
    {
        if(select && (curDisplay == editDisplay))
            editDisplay->select(document->selectedCurves, document->selectedStitchs);
        else
            curDisplay->draw();
    }

    if(!select)
    {
        if (editionState==NewSubcurve)
        {
            // Points
            glPointSize(5.0f);

            glBegin(GL_POINTS);
            if (joincurve_id == 0)
                glColor4d(0.8f, 0.0f, 0.0f, 1.0f);
            else
                glColor4d(0.0f, 0.8f, 0.0f, 1.0f);
            glVertex2f(editClosestPoint.v.x,editClosestPoint.v.y);
            glEnd();
        }

        // cursor cross
        glLineWidth(1.5);
        glBegin(GL_LINES);
            if (editionState)
            {
                if (joincurve_id == 0)
                    glColor4d(0.8f, 0.0f, 0.0f, 0.5f);
                else
                    glColor4d(0.0f, 0.8f, 0.0f, 0.5f);
            }else
                glColor4d(0.0f, 0.0f, 0.0f, 0.3f);
            glVertex2f(absMouse.x,-10000);
            glVertex2f(absMouse.x,10000);

            glVertex2f(-10000,absMouse.y);
            glVertex2f(10000,absMouse.y);
        glEnd();

        //cursor select zone
        glLineWidth(2);
        glBegin(GL_LINE_STRIP);
            glColor4d(0.0f, 0.0f, 0.0f, 0.3f);
            glVertex2f(absMouse.x-5*zoom,absMouse.y+5*zoom);
            glVertex2f(absMouse.x-5*zoom,absMouse.y-5*zoom);
            glVertex2f(absMouse.x+5*zoom,absMouse.y-5*zoom);
            glVertex2f(absMouse.x+5*zoom,absMouse.y+5*zoom);
            glVertex2f(absMouse.x-5*zoom,absMouse.y+5*zoom);
        glEnd();

        if(selection)
        {
            vector_2d clicAbsMouse = viewPos+(clicOldMouse-vector_2d(width,height)/2)*zoom*2;
            //float clicAbsMouseX = viewPos.x+(clicOldMouse.x-width/2)*zoom*2;
            //float clicAbsMouseY = viewPos.y+(clicOldMouse.y-height/2)*zoom*2;
            glLineWidth(2);
            glBegin(GL_LINE_STRIP);
                glColor4d(0.0f, 0.0f, 0.0f, 0.3f);
                glVertex2f(absMouse.x,clicAbsMouse.y);
                glVertex2f(absMouse.x,absMouse.y);
                glVertex2f(clicAbsMouse.x,absMouse.y);
                glVertex2f(clicAbsMouse.x,clicAbsMouse.y);
                glVertex2f(absMouse.x,clicAbsMouse.y);
            glEnd();
        }
    }
	return true;
}