#include "ProjectionMappingMode.h"

namespace ofx {
namespace piMapper {

ProjectionMappingMode::ProjectionMappingMode(){
	_surfaceScaleBeforeTransform = 1.0f;
	_bDrawLayerPanel = true;
}

ProjectionMappingMode * ProjectionMappingMode::_instance = 0;

ProjectionMappingMode * ProjectionMappingMode::instance(){
	if(_instance == 0){
		_instance = new ofx::piMapper::ProjectionMappingMode();
	}
	return _instance;
}

void ProjectionMappingMode::setup(Application *app){
	Gui::instance()->getSurfaceHighlightWidget().setSurfaceManager(app->getSurfaceManager());
	Gui::instance()->getLayerPanelWidget().setSurfaceManager(app->getSurfaceManager());
	Gui::instance()->getProjectionEditorWidget().setSurfaceManager(app->getSurfaceManager());
}

void ProjectionMappingMode::update(Application * app){
	Gui::instance()->getProjectionEditorWidget().update();
	Gui::instance()->getScaleWidget().update();
}

void ProjectionMappingMode::draw(Application * app){
	ofPushStyle();
	ofSetColor(255, 255, 255, 255);
	app->getSurfaceManager()->draw();
	ofPopStyle();
	
	Gui::instance()->getProjectionEditorWidget().draw();
	
	// Draw scale widget. The size of the widget is being set on surface select.
	BaseSurface * selectedSurface = app->getSurfaceManager()->getSelectedSurface();
	if(selectedSurface != 0){
		Gui::instance()->getScaleWidget().draw();
	}
	
	if(_bDrawLayerPanel){
		Gui::instance()->getLayerPanelWidget().draw();
	}
	
	Gui::instance()->getSurfaceHighlightWidget().draw();
}

void ProjectionMappingMode::onKeyPressed(Application * app, ofKeyEventArgs & args){
	switch(args.key){
		 
	 case 't':
		 createSurface(app, SurfaceType::TRIANGLE_SURFACE);
		 break;

	 case 'q':
		 createSurface(app, SurfaceType::QUAD_SURFACE);
		 break;
		 
	 case 'g':
		 createSurface(app, SurfaceType::GRID_WARP_SURFACE);
		 break;
		
	 case 'h':
		 createSurface(app, SurfaceType::HEXAGON_SURFACE);
		 break;

	 case OF_KEY_BACKSPACE:
		 app->eraseSurface(app->getSurfaceManager()->getSelectedSurfaceIndex());
		 break;
	 
	 case 'p':
		 app->togglePerspective();
		 break;
	 
	 case '}':
		 app->addGridRow();
		 break;
	
	 case '{':
		 app->removeGridRow();
		 break;
	 
	 case ']':
		 app->addGridColumn();
		 break;
		 
	 case '[':
		 app->removeGridColumn();
		 break;
		 
	 case '.':
		 selectNextSurface(app);
		 break;
		 
	 case ',':
		 selectPrevSurface(app);
		 break;
		 
	 case '>':
		 selectNextVertex(app);
		 break;
	 
	 case '<':
		 selectPrevVertex(app);
		 break;
	 
	 case OF_KEY_UP:
		 if(app->isShiftKeyDown()){
			moveSelection(app, ofVec2f(0.0f, -10.0f));
		 }else{
			moveSelection(app, ofVec2f(0.0f, -1.0f));
		 }
		 break;
		 
	 case OF_KEY_DOWN:
		 if(app->isShiftKeyDown()){
			moveSelection(app, ofVec2f(0.0f, 10.0f));
		 }else{
			moveSelection(app, ofVec2f(0.0f, 1.0f));
		 }
		 break;
		 
	 case OF_KEY_LEFT:
		 if(app->isShiftKeyDown()){
			moveSelection(app, ofVec2f(-10.0f, 0.0f));
		 }else{
			moveSelection(app, ofVec2f(-1.0f, 0.0f));
		 }
		 break;
	 
	 case OF_KEY_RIGHT:
		 if(app->isShiftKeyDown()){
			moveSelection(app, ofVec2f(10.0f, 0.0f));
		 }else{
			moveSelection(app, ofVec2f(1.0f, 0.0f));
		 }
		 break;
		 
	 case ' ':
		 if(app->getSurfaceManager()->getSelectedSurface() == 0){
			return;
		 }
		 
		 if(app->getSurfaceManager()->getSelectedSurface()->getSource()->getType() ==
            SourceType::SOURCE_TYPE_VIDEO){
            app->getCmdManager()->exec(
                new ToggleAnimatedSourceCmd(
                    app->getSurfaceManager()->getSelectedSurface()));
         }
		 break;
		 
	 case OF_KEY_TAB:
		 app->setNextSource();
		 break;
	
	 case 'd':
		 app->duplicateSurface();
		 break;
		 
	 case '0': // Move selected surface up the layer stack
		 app->moveLayerUp();
		 break;
		
	 case '9': // Move selected surface down the layer stack
		 app->moveLayerDown();
		 break;
		 
	 case '+': // Scale surface up
		 app->scaleUp();
		 break;

	 case '-': // Scale surface down
		 app->scaleDown();
		 break;

	 case 'l':
		 _bDrawLayerPanel = !_bDrawLayerPanel;
		 break;

	/*
	 case 'n': // Set next preset
		 app->getSurfaceManager()->setNextPreset();
		 break;
	*/
		 
	 default:
		 break;
	}
}

void ProjectionMappingMode::onMousePressed(Application * app, ofMouseEventArgs & args){
	Gui::instance()->onMousePressed(args);
	
	CircleJoint * hitJoint = 0;
	int hitJointIndex = -1;
	BaseSurface * hitSurface = 0;

	hitJoint = Gui::instance()->getProjectionEditorWidget().hitTestJoints(ofVec2f(args.x, args.y));
		
	if(hitJoint){
		for(int i = Gui::instance()->getProjectionEditorWidget().getJoints()->size() - 1; i >= 0 ; --i){
			if((*Gui::instance()->getProjectionEditorWidget().getJoints())[i] == hitJoint){
				hitJointIndex = i;
				break;
			}
		}
	}else{
		for(int i = app->getSurfaceManager()->size() - 1; i >= 0; --i){
			if(app->getSurfaceManager()->getSurface(i)->hitTest(ofVec2f(args.x, args.y))){
				hitSurface = app->getSurfaceManager()->getSurface(i);
				break;
			}
		}
	}
		
	if(Gui::instance()->getScaleWidget().inside(args.x, args.y)){
		//
	}else if(hitJoint){
		hitJoint->select();
		hitJoint->startDrag();
		Gui::instance()->notifyJointPressed(args, hitJointIndex);
	}else if(hitSurface){
		_clickPosition = ofVec2f(args.x, args.y); // TODO: redesign this so we can use a kind of
												  //       display stack.
		_bSurfaceDrag = true; // TODO: Should be something like `hitSurface->startDrag()`
		Gui::instance()->notifySurfacePressed(args, hitSurface);
	}else{
		Gui::instance()->notifyBackgroundPressed(args);
	}
}

void ProjectionMappingMode::onMouseReleased(Application * app, ofMouseEventArgs & args){
	Gui::instance()->onMouseReleased(args);
	_bSurfaceDrag = false; // TODO: handle this locally
	Gui::instance()->getProjectionEditorWidget().stopDragJoints();
}

void ProjectionMappingMode::onMouseDragged(Application * app, ofMouseEventArgs & args){
	Gui::instance()->onMouseDragged(args);
	Gui::instance()->getProjectionEditorWidget().mouseDragged(args);
	
	// TODO: Handle app->getGui()->clickPosition and app->getGui()->bDrag locally.
	if(_bSurfaceDrag){
		ofVec2f mousePosition = ofVec2f(args.x, args.y);
		ofVec2f distance = mousePosition - _clickPosition;
		Gui::instance()->getProjectionEditorWidget().moveSelectedSurface(distance);
		_clickPosition = mousePosition;
	}
}

void ProjectionMappingMode::onJointPressed(Application * app, GuiJointEvent & e){
	app->getCmdManager()->exec(new SelVertexCmd(app->getSurfaceManager(), e.jointIndex));
	app->getCmdManager()->exec(new MvSurfaceVertCmd(
		e.jointIndex,
		app->getSurfaceManager()->getSelectedSurface()));
}

void ProjectionMappingMode::onSurfacePressed(Application * app, GuiSurfaceEvent & e){
	if(app->getSurfaceManager()->getSelectedSurface() != e.surface){
		app->getCmdManager()->exec(new SelSurfaceCmd(app->getSurfaceManager(), e.surface ));
	}
	
	app->getCmdManager()->exec(new StartDragSurfaceCmd(e.surface));
}

void ProjectionMappingMode::onBackgroundPressed(Application * app, GuiBackgroundEvent & e){
	if(app->getSurfaceManager()->getSelectedSurface() != 0){
		app->getCmdManager()->exec(new DeselectSurfaceCmd(app->getSurfaceManager()));
	}
}

void ProjectionMappingMode::onGuiEvent(Application * app, GuiEvent & e){
	
	// Scale widget now. More widgets later.
	if(e.widget == &Gui::instance()->getScaleWidget()){
		if(e.args.type == e.args.Pressed){
			if(app->getSurfaceManager()->getSelectedSurface() == 0){
				return;
			}
			
			_surfaceScaleBeforeTransform =
				app->getSurfaceManager()->getSelectedSurface()->getScale();
		}else if(e.args.type == e.args.Released){
			if(app->getSurfaceManager()->getSelectedSurface() == 0){
				return;
			}
			
			if(_surfaceScaleBeforeTransform !=
				app->getSurfaceManager()->getSelectedSurface()->getScale()){
				
				app->getCmdManager()->exec(new ScaleSurfaceFromToCmd(
					app->getSurfaceManager()->getSelectedSurface(),
					_surfaceScaleBeforeTransform,
					app->getSurfaceManager()->getSelectedSurface()->getScale()));
			}
		}else if(e.args.type == e.args.Dragged){
			app->getSurfaceManager()->getSelectedSurface()->scaleTo(e.widget->getScale());
		}
	}
}

void ProjectionMappingMode::selectSurface(Application * app, int i){
	if(app->getSurfaceManager()->size()){
		if(app->getSurfaceManager()->getSelectedSurfaceIndex() == i){
			return;
		}
		app->getCmdManager()->exec(
			new SelSurfaceCmd(
				app->getSurfaceManager(),
				app->getSurfaceManager()->getSurface(i) ));
	}
}

void ProjectionMappingMode::selectNextSurface(Application * app){
	if(app->getSurfaceManager()->size()){
		if(	app->getSurfaceManager()->size() == 1 &&
			app->getSurfaceManager()->getSelectedSurface() ==
			app->getSurfaceManager()->getSurface(0)){
			return;
		}
		app->getCmdManager()->exec(new SelNextSurfaceCmd(app->getSurfaceManager()));
	}
}

void ProjectionMappingMode::selectPrevSurface(Application * app){
	if(app->getSurfaceManager()->size()){
		if(	app->getSurfaceManager()->size() == 1 &&
			app->getSurfaceManager()->getSelectedSurface() ==
			app->getSurfaceManager()->getSurface(0)){
			return;
		}
		app->getCmdManager()->exec(new SelPrevSurfaceCmd(app->getSurfaceManager()));
	}
}

void ProjectionMappingMode::selectNextVertex(Application * app){
	if(app->getSurfaceManager()->getSelectedSurface() != 0){
		app->getCmdManager()->exec(new SelNextVertexCmd(app->getSurfaceManager()));
	}
}

void ProjectionMappingMode::selectPrevVertex(Application * app){
	if(app->getSurfaceManager()->getSelectedSurface() != 0){
		app->getCmdManager()->exec(new SelPrevVertexCmd(app->getSurfaceManager()));
	}
}

void ProjectionMappingMode::moveSelection(Application * app, ofVec2f by){
	app->getCmdManager()->exec(
		new MvSelectionCmd(
			app->getSurfaceManager(), by));
}

void ProjectionMappingMode::createSurface(Application *app, SurfaceType type){
	app->getCmdManager()->exec(
		new AddSurfaceCmd(app->getSurfaceManager(), type));
}

} // namespace piMapper
} // namespace ofx