#include "ext/SquarePotential.h"
#include <tmathutil.h>
#include <algorithm>

using namespace std;

//-----------------------------------------------------------------------------

void ToonzExt::SquarePotential::setParameters_(const TStroke *ref,
											   double par,
											   double al)
{
	ref_ = ref;
	par_ = par;
	actionLength_ = al;
	assert(ref_);

	strokeLength_ = ref->getLength();
	lenghtAtParam_ = ref->getLength(par);

	// lunghezza dal pto di click all'inizio della curva
	leftFactor_ = min(lenghtAtParam_,
					  actionLength_ * 0.5); //lenghtAtParam_ / strokeLength_;

	// lunghezza dal pto di click alla fine
	rightFactor_ = min(strokeLength_ - lenghtAtParam_,
					   actionLength_ * 0.5);

	// considero come intervallo di mapping [-range,range].
	//  4 ha come valore c.a. 10exp-6
	//  cioé sposterei un pixel su un movimento di un milione di pixel
	range_ = 2.8;
}

//-----------------------------------------------------------------------------

ToonzExt::SquarePotential::~SquarePotential()
{
}

//-----------------------------------------------------------------------------

double
ToonzExt::SquarePotential::value_(double value2test) const
{
	return this->compute_value(value2test);
}

//-----------------------------------------------------------------------------

// normalization of parameter in range interval
double ToonzExt::SquarePotential::compute_shape(double value2test) const
{
	double x = ref_->getLength(value2test);
	double shape = this->actionLength_ * 0.5;
	if (isAlmostZero(shape))
		shape = 1.0;
	x = ((x - lenghtAtParam_) * range_) / shape;
	return x;
}

//-----------------------------------------------------------------------------

double
ToonzExt::SquarePotential::compute_value(double value2test) const
{
	// use
	//     2
	//  1-x
	//

	double x = 0.0;
	double res = 0.0;

	// lenght  at parameter
	x = ref_->getLength(value2test);

	double tmp_al = actionLength_ * 0.5;

	// compute correct parameter considering offset
	// try to have a square curve like shape
	//
	//       2
	//  m = x
	//
	if (leftFactor_ == 0.0)
		x = 1.0 - x / tmp_al;
	else if (rightFactor_ == 0.0)
		x = (x - (strokeLength_ - tmp_al)) / tmp_al;
	else {
		if (x <= lenghtAtParam_ &&
			((lenghtAtParam_ - x) <= leftFactor_))
			x = (x - (lenghtAtParam_ - leftFactor_)) / leftFactor_;
		else if (x > lenghtAtParam_ &&
				 ((x - lenghtAtParam_) < rightFactor_))
			x = (rightFactor_ - (x - lenghtAtParam_)) / rightFactor_;
		else
			x = -1;
	}

	if (x < 0.0)
		return 0.0;
	//assert( 0.0 <= x &&
	//  x <= 1.0 + TConsts::epsilon );
	res = sq(x);

	return res;
}

//-----------------------------------------------------------------------------

ToonzExt::Potential *
ToonzExt::SquarePotential::clone()
{
	return new SquarePotential;
}

//-----------------------------------------------------------------------------
//  End Of File
//-----------------------------------------------------------------------------