#include "ext/Selector.h"
#include "tgl.h"
//#include "drawutil.h"
#include "tmathutil.h"
 
using namespace ToonzExt;
 
//-----------------------------------------------------------------------------
 
namespace
{
const GLfloat s_normalColor[] = {250.0 / 255.0, 127 / 255.0, 240 / 255.0};
const GLfloat s_highlightedColor[] = {150.0 / 255.0, 255.0 / 255.0, 140.0 / 255.0};
const double s_sqrt_2 = sqrt(2.0);
const double s_radius = 5.0;
const double s_over_size = 10;
const double s_length = 15;
const double s_square_size = 5;
const double s_arrow_ratio = 2.5;
 
void drawArrow(const TPointD &from,
			   const TPointD &direction,
			   double length,
			   const TPointD &arrowDirection = TPointD(0.0, -1.0),
			   double arrowLength = 0.0)
{
	if (length <= 0 ||
		arrowLength < 0)
		return;
 
	if (arrowLength == 0.0)
		arrowLength = 0.2 * length;
 
	TPointD
		pnt(from),
		myDir = normalize(direction);
 
	TPointD
		myArrowDirection(arrowDirection);
 
	if (arrowDirection.y <= 0.0)
		myArrowDirection = normalize(TPointD(-0.5, 0.8));
 
	if (direction * TPointD(1, 0) <= 0.0)
		myArrowDirection.x = -myArrowDirection.x;
 
	myDir = myDir * length;
	myArrowDirection = myArrowDirection * arrowLength;
 
	glBegin(GL_LINES);
	tglVertex(pnt);
	pnt += myDir;
	tglVertex(pnt);
	glEnd();
	glBegin(GL_TRIANGLES);
	TPointD
		up = myDir + myArrowDirection;
	tglVertex(from + up);
	//tglVertex(pnt);
	tglVertex(pnt);
	myArrowDirection.y = -myArrowDirection.y;
	TPointD
		down = myDir + myArrowDirection;
	tglVertex(from + down);
	glEnd();
}
}
 
//-----------------------------------------------------------------------------
 
Selector::Selector(double stroke_length,
				   double min_val,
				   double max_val)
{
	original_stroke_length_ = stroke_length;
	range_ = TPointD(min_val, max_val);
	isVisible_ = false;
	pixel_size_ = 1.0;
	w_ = 0.5;
	this->init();
}
 
//-----------------------------------------------------------------------------
 
void Selector::init()
{
	strokeRef_ = 0;
	signum_ = 1.0;
	isSelected_ = NONE;
}
 
//-----------------------------------------------------------------------------
 
Selector::~Selector()
{
}
 
//-----------------------------------------------------------------------------
 
TPointD
Selector::getUp() const
{
	if (!strokeRef_)
		return TPointD();
 
	TThickPoint
		pnt = strokeRef_->getPoint(w_);
 
	TPointD
		p = convert(pnt),
		p1;
 
	TPointD
		v0,
		v1,
		v = normalize(rotate90(strokeRef_->getSpeed(w_)));
 
	const double delta = TConsts::epsilon;
 
	if (w_ - delta >= 0.0)
		v0 = rotate90(strokeRef_->getSpeed(w_ - delta));
	else
		v0 = rotate90(strokeRef_->getSpeed(0));
 
	if (w_ + delta <= 1.0)
		v1 = rotate90(strokeRef_->getSpeed(w_ + delta));
	else
		v1 = rotate90(strokeRef_->getSpeed(1.0));
 
	if (!isAlmostZero(v * v0) ||
		!isAlmostZero(v * v1))
		v = normalize(v1 + v0);
 
	return v;
}
 
//-----------------------------------------------------------------------------
 
void Selector::draw(Designer *designer)
{
	if (!strokeRef_ ||
		!isVisible_)
		return;
 
	pixel_size_ = designer ? sqrt(designer->getPixelSize2()) : 1.0;
 
	TPointD
		v = this->getUp(),
		n = normalize(rotate90(v));
 
	TThickPoint
		pnt = strokeRef_->getThickPoint(w_);
 
	height_ = (pnt.thick + s_over_size) * pixel_size_;
 
	TPointD
		p(pnt.x, pnt.y),
		up = p + v * height_,
		down = p - v * height_;
 
	glColor3fv(s_normalColor);
	glBegin(GL_LINES);
	tglVertex(down);
	tglVertex(up);
	glEnd();
 
	if (isSelected_ == POSITION)
		glColor3fv(s_highlightedColor);
	else
		glColor3fv(s_normalColor);
 
	double
		radius = s_radius * pixel_size_;
 
	if (isSelected_ == POSITION)
		tglDrawDisk(up + v * radius,
					radius);
 
	tglDrawCircle(up + v * radius,
				  radius);
 
	if (isSelected_ == LENGTH)
		glColor3fv(s_highlightedColor);
	else
		glColor3fv(s_normalColor);
 
	{
		// the circle center is in
		TPointD
			down = -this->getUp(),
			center = pnt + down * (height_);
		const double
			length = s_square_size * 0.5 * pixel_size_;
 
		TPointD
			//  offset(length, 0.0),
			offset(length,
				   length),
			bottom_left = center - offset,
			top_rigth = center + offset;
 
		TRectD
			rectangle(bottom_left,
					  top_rigth);
 
		if (isSelected_ == LENGTH)
			tglFillRect(rectangle);
 
		tglDrawRect(rectangle);
	}
	/*
  TPointD
    arrowDirection(-0.5,0.8);
  
  if( signum_ < 0.0 )
    arrowDirection.x = -arrowDirection.x;
  const double
    length = s_length  * pixel_size_,
    arrow_length = length/s_arrow_ratio;
 
  drawArrow(down,
            TPointD(1,0),
            length,
            arrowDirection,
            arrow_length);
  drawArrow(down,
            TPointD(-1,0),
            length,
            arrowDirection,
            arrow_length);
  */
	// draw stroke related information
	if (designer &&
		this->isSelected())
		designer->draw(this);
 
#if 0 //def  _DEBUG
  {
  TThickPoint
    pnt = strokeRef_->getThickPoint(w_);
 
  TPointD
      up = this->getUp(),
      down = -up;
 
  double
    radius = s_radius * pixel_size_;
 
  // the circle center is in 
  TPointD 
    center = pnt + up * ( height_ + radius);
 
  glColor3d(1.0,0.0,1.0);
  tglDrawCircle(center,radius+pixel_size_);
 
  // one pixel of tolerance
  center = pnt + down * ( height_ );
 
  const double
    length = s_length * pixel_size_;
 
  TPointD
    offset(length, 0.0),
    bottom_left = center - offset,
    top_rigth   = center + offset;
 
  TRectD
    rectangle(bottom_left,
              top_rigth);
 
  rectangle = rectangle.enlarge( 2.0* pixel_size_);
 
  tglDrawRect(rectangle);
  }
#endif
}
 
//-----------------------------------------------------------------------------
 
void Selector::mouseDown(const TPointD &pos)
{
	click_ =
		curr_ =
			pos;
	if (!strokeRef_)
		return;
	stroke_length_ =
		original_stroke_length_;
	prev_ = curr_;
}
 
//-----------------------------------------------------------------------------
 
void Selector::mouseUp(const TPointD &pos)
{
	curr_ = pos;
	if (!strokeRef_)
		return;
	original_stroke_length_ =
		stroke_length_;
	prev_ = curr_;
}
 
//-----------------------------------------------------------------------------
 
void Selector::mouseMove(const TPointD &pos)
{
	curr_ = pos;
	if (!strokeRef_)
		return;
 
	isSelected_ = this->getSelection(pos);
	prev_ = curr_;
}
 
//-----------------------------------------------------------------------------
 
void Selector::mouseDrag(const TPointD &pos)
{
	curr_ = pos;
	if (!strokeRef_)
		return;
 
	double
		val = original_stroke_length_;
 
	const double
		stroke_length = strokeRef_->getLength();
 
	double
		max_val = std::min(stroke_length, range_.y);
 
	TPointD
		v = curr_ - prev_;
 
	signum_ = 1.0;
 
	switch (isSelected_) {
	case POSITION:
		w_ = strokeRef_->getW(pos);
		break;
	case LENGTH:
		signum_ = v * TPointD(1.0, 0.0) < 0.0 ? -1.0 : 1.0;
		val = original_stroke_length_ + signum_ * pixel_size_ * norm(v);
		stroke_length_ = std::max(std::min(max_val, val), range_.x);
		break;
	default:
		assert(false);
	}
	prev_ = curr_;
}
 
//-----------------------------------------------------------------------------
 
void Selector::setStroke(const TStroke *ref)
{
	//if(strokeRef_  &&
	//   strokeRef_ != ref )
	//  w_=strokeRef_->getParameterAtLength( strokeRef_->getLength()*0.5);
	//else
	//  w_=0.5;
	this->init();
	strokeRef_ = ref;
}
 
//-----------------------------------------------------------------------------
 
Selector::Selection
Selector::getSelection(const TPointD &pos)
{
	if (!strokeRef_ ||
		!isVisible_)
		return NONE;
 
	TThickPoint
		pnt = strokeRef_->getThickPoint(w_);
 
	TPointD
		up = this->getUp(),
		down = -up;
 
	double
		radius = s_radius * pixel_size_;
 
	// the circle center is in
	TPointD
		center = pnt + up * (height_ + radius);
 
	// one pixel of tolerance
	if (tdistance2(center, pos) <= sq(radius + pixel_size_))
		return POSITION;
 
	center = pnt + down * (height_);
 
	// const double  length = s_length * pixel_size_;
	const double length = s_square_size * 0.5 * pixel_size_;
 
	TPointD
		//  offset(length, 0.0),
		offset(length,
			   length),
		bottom_left = center - offset,
		top_rigth = center + offset;
 
	TRectD
		rectangle(bottom_left,
				  top_rigth);
 
	rectangle = rectangle.enlarge(2.0 * pixel_size_);
 
	if (rectangle.contains(pos))
		return LENGTH;
 
	return NONE;
}
 
//-----------------------------------------------------------------------------
 
double
Selector::getLength() const
{
	return stroke_length_;
}
 
//-----------------------------------------------------------------------------
 
void Selector::setLength(double length)
{
	stroke_length_ =
		original_stroke_length_ = length;
}
 
//-----------------------------------------------------------------------------
//  End Of File
//-----------------------------------------------------------------------------