#include "CurveEdit.h"
#include "../../SDK/PlatformIndependent/Core/Timer.h"

#include <Wt/WStandardItemModel>
#include <Wt/WPointF>
#include <Wt/WStandardItem>

using namespace Wt;
using namespace Wt::Chart;
using namespace TVNG;

//-------------------------------------------------------------------------
CurveEdit::CurveEdit(const std::string& label)
: 
WCartesianChart(),
label_(label),
selectedRowIndex_(-1),
lastEmitTime_(0)
{
	model_ = new WStandardItemModel(0, 3);
	labelFont_.setFamily(WFont::SansSerif);
	labelFont_.setSize(WFont::FixedSize, WLength(7, WLength::Point));

	// needed to capture key events (HTML5)
	setAttributeValue("tabindex", "0");

	setModel(model_); 
	setXSeriesColumn(0);
	setLegendEnabled(false); 
	setType(ScatterPlot);

	WPen pen;
	pen.setColor(WColor(0, 0, 0, 100));
	pen.setWidth(1);

	WAxis& xAxis = axis(XAxis);
	WAxis& yAxis = axis(YAxis);
	xAxis.setLocation(ZeroValue);
	yAxis.setLocation(ZeroValue);

	WFont labelFont;
	labelFont.setFamily(WFont::SansSerif);
	labelFont.setSize(WFont::FixedSize, WLength(0, WLength::Point));
	
	xAxis.setLabelFont(labelFont);
	yAxis.setLabelFont(labelFont);
	xAxis.setLabelFormat("%.2f");
	yAxis.setLabelFormat("%.2f");
	xAxis.setLabelInterval(0.05);
	yAxis.setLabelInterval(0.05);
	xAxis.setRange(-0.1, 1.1);
	yAxis.setRange(-0.1, 1.1);
	xAxis.setGridLinesEnabled(true);
	yAxis.setGridLinesEnabled(true);
	xAxis.setGridLinesPen(pen);
	yAxis.setGridLinesPen(pen);

	setPlotAreaPadding(0, Left);
	setPlotAreaPadding(0, Right);
	setPlotAreaPadding(0, Bottom);
	setPlotAreaPadding(0, Top);

	setMargin(5, Top | Bottom);
	setMargin(WLength::Auto, Left | Right);
	setBackground(WBrush(WColor(150, 150, 150)));

	// Add the curves
	WDataSeries s(1, LineSeries);
	s.setShadow(WShadow(3, 3, WColor(0, 0, 0, 127), 2));
	s.setMarker(Chart::SquareMarker);
	s.setMarkerSize(5);
	s.setMarkerPen(WPen(WColor(0, 0, 0)));
	s.setMarkerBrush(WBrush(WColor(255, 255, 0)));
	s.setPen(WPen(WColor(0, 255, 0, 255)));
	
	s.setLabelsEnabled(XAxis);
	s.setLabelsEnabled(YAxis);
	addSeries(s);

	WDataSeries selection(2, PointSeries);
	selection.setMarker(Chart::SquareMarker);
	selection.setMarkerPen(WPen(WColor(0, 0, 0)));
	selection.setMarkerBrush(WBrush(WColor(255, 0, 0)));
	selection.setMarkerSize(5);
	addSeries(selection);

	mouseWentDown().connect(this, &CurveEdit::MouseWentDown);
	doubleClicked().connect(this, &CurveEdit::DoubleClicked);
	mouseDragged().connect(this, &CurveEdit::MouseDragged);
	keyWentUp().connect(this, &CurveEdit::KeyWentUp);

	resize(300, 135);
}

//-------------------------------------------------------------------------
CurveEdit::~CurveEdit()
{
	delete model_;
}

//-------------------------------------------------------------------------
void CurveEdit::DoubleClicked(const WMouseEvent& e)
{
	if (e.button() != WMouseEvent::LeftButton) return;

	WPointF pointF = mapFromDevice(WPointF(e.widget().x, e.widget().y));
	Vector2f p( static_cast<float>(pointF.x()), static_cast<float>(pointF.y()) );
	if (p.X() < 0.) p.SetX(0.);
	if (p.X() > 1.) p.SetX(1.);
	if (p.Y() < 0.) p.SetY(0.);
	if (p.Y() > 1.) p.SetY(1.);

	for (int i(0); i<model_->rowCount(); i++)
	{
		Vector2f elem( boost::any_cast<float>(model_->data(i, 0)), boost::any_cast<float>(model_->data(i, 1)) );
		if (SamePoints(p, elem)) return;
		if (p.X() == elem.X()) return;
	}
	
	int row = model_->rowCount();
	model_->insertRow(row);
	model_->setData(row, 0, boost::any(p.X()));
	model_->setData(row, 1, boost::any(p.Y()));
	model_->setData(row, 2, boost::any(p.Y()));

	if (selectedRowIndex_ != -1)
		model_->setData(selectedRowIndex_, 2, boost::any());
	model_->sort(0);
	for (int i(0); i<model_->rowCount(); i++)
	{
		float x( boost::any_cast<float>(model_->data(i, 0)) );
		if (x == p.X()) 
		{
			selectedRowIndex_ = i;
			break;
		}
	}
	UpdateCurve();
	changed_.emit();
}

//-------------------------------------------------------------------------
void CurveEdit::MouseWentDown(const WMouseEvent& e)
{
	if (e.button() != WMouseEvent::LeftButton) return;
	WPointF pointF = mapFromDevice(WPointF(e.widget().x, e.widget().y));
	Vector2f p( static_cast<float>(pointF.x()), static_cast<float>(pointF.y()) );
	if (p.X() < 0.) p.SetX(0.);
	if (p.X() > 1.) p.SetX(1.);
	if (p.Y() < 0.) p.SetY(0.);
	if (p.Y() > 1.) p.SetY(1.);

	int row(-1);
	for (int i(0); i<model_->rowCount(); i++)
	{
		Vector2f elem( boost::any_cast<float>(model_->data(i, 0)), boost::any_cast<float>(model_->data(i, 1)) );
		if (SamePoints(elem, p))
		{
			row = i;
			p = elem;
			break;
		}
	}

	if (selectedRowIndex_ != -1 && row != selectedRowIndex_)
		model_->setData(selectedRowIndex_, 2, boost::any());

	if (row != -1 && selectedRowIndex_ != row)
		model_->setData(row, 2, boost::any(p.Y()));
	selectedRowIndex_ = row;
}

//-------------------------------------------------------------------------
void CurveEdit::MouseDragged(const WMouseEvent& e)
{
	if (e.button() != WMouseEvent::LeftButton) return;
	if (selectedRowIndex_ == -1) return;

	WPointF pointF = mapFromDevice(WPointF(e.widget().x, e.widget().y));
	Vector2f p( static_cast<float>(pointF.x()), static_cast<float>(pointF.y()) );
	if (p.X() < 0.) p.SetX(0.);
	if (p.X() > 1.) p.SetX(1.);
	if (p.Y() < 0.) p.SetY(0.);
	if (p.Y() > 1.) p.SetY(1.);

	for (int i(0); i<model_->rowCount(); i++)
	{
		if (i == selectedRowIndex_) continue;
		float x = boost::any_cast<float>(model_->data(i, 0));
		if (x == p.X()) return;
	}

	model_->setData(selectedRowIndex_, 0, boost::any(p.X()));
	model_->setData(selectedRowIndex_, 1, boost::any(p.Y()));
	model_->setData(selectedRowIndex_, 2, boost::any(p.Y()));
	model_->sort(0);

	for (int i(0); i<model_->rowCount(); i++)
	{
		float x( boost::any_cast<float>(model_->data(i, 0)) );
		if (x == p.X()) 
		{
			selectedRowIndex_ = i;
			break;
		}
	}
	UpdateCurve();
	double emitTime( Timer::DGetTime() );
	if (emitTime - lastEmitTime_ > 0.04)
	{
		lastEmitTime_ = emitTime;
		changed_.emit();
	}
}

//-------------------------------------------------------------------------
void CurveEdit::KeyWentUp(const WKeyEvent& e)
{
	if (selectedRowIndex_ == -1) return;

	if (e.key() == Wt::Key_Delete)
	{
		model_->removeRow(selectedRowIndex_);
		selectedRowIndex_ = -1;
		UpdateCurve();
		changed_.emit();
	}
}

//-------------------------------------------------------------------------
void CurveEdit::SetCurve( const FloatAnimationCurvePtr& curve )
{
	model_->removeRows(0, model_->rowCount());
	if (curve != NULL)
	{
		for (unsigned int i(0); i<curve->GetNumKeyFrames(); i++)
		{
			int row = model_->rowCount();
			model_->insertRow(row);
			const std::pair< float, float >& keyframe = curve->GetKeyFrame(i);
			model_->setData(row, 0, boost::any(keyframe.first));
			model_->setData(row, 1, boost::any(keyframe.second));
			model_->setData(row, 2, boost::any());
		}
	}
	UpdateCurve();
	selectedRowIndex_ = -1;
}

//-------------------------------------------------------------------------
void CurveEdit::UpdateCurve()
{
	curve_.RemoveAllKeyFrames();
	for (int i(0); i<model_->rowCount(); i++)
	{
		Vector2f elem( boost::any_cast<float>(model_->data(i, 0)), boost::any_cast<float>(model_->data(i, 1)) );
		curve_.AddKeyFrame(elem.X(), elem.Y());
	}
}

//-------------------------------------------------------------------------
bool CurveEdit::SamePoints(const Vector2f& p1, const Vector2f& p2)
{
	float d = sqrt((p1.X() - p2.X()) * (p1.X() - p2.X()) + (p1.Y() - p2.Y()) * (p1.Y() - p2.Y()));
	if (d < 0.05) return true;
	return false;
}

//-------------------------------------------------------------------------
void CurveEdit::paint(Wt::WPainter& painter, const Wt::WRectF& rectangle) const
{
	painter.setFont(labelFont_);
	WCartesianChart::paint(painter, rectangle);
}