/**
* @author Fabrizio Morciano <fabrizio.morciano@gmail.com>
*/
#ifdef _DEBUG
#define _STLP_DEBUG 1
#endif
#include "ext/StrokeDeformationImpl.h"
#include "ext/StrokeDeformation.h"
#include "ext/SquarePotential.h"
#include "ext/StrokeParametricDeformer.h"
//#include "ext/NotSymmetricBezierPotential.h"
#include "ext/ContextStatus.h"
#include "ext/Designer.h"
//#include "ext/TriParam.h"
#include <tcurves.h>
#include <tstrokeutil.h>
//#include <tvectorimage.h>
#include <tmathutil.h>
#include <algorithm>
#include <iterator>
#include <vector>
#include "DeformationSelector.h"
using namespace ToonzExt;
namespace {
/*
* Avoid to reduce curves with great zoom out.
* assert that pixelSize and factor > 0.0
*/
double computeReductionFactor(double pixelSize, double factor) {
assert(pixelSize > 0.0 && factor > 0.0);
if (factor <= 0.0) factor = 1.0;
if (pixelSize <= 0.0 || pixelSize > 1.0) pixelSize = 1.0;
return pixelSize * factor;
}
//---------------------------------------------------------------------------
double retrieveParamAtLengthWithOffset(const TStroke *stroke2change,
double length, // = 0.5 * strokeLength,
double offset) {
if (!isValid(stroke2change) || !isValid(offset) || length < 0) return -1;
double strokeLength = stroke2change->getLength();
assert(strokeLength >= 0.0 && "Not valid length");
if (strokeLength < 0) {
return -1;
}
double lengthAtW = stroke2change->getLength(offset);
assert(strokeLength >= lengthAtW &&
"Position of parameter is greater than stroke length!!!");
if (strokeLength < lengthAtW) return -1;
double newLength = -1;
if (stroke2change->isSelfLoop()) {
if (length >= 0)
newLength = length > lengthAtW ? length + lengthAtW : lengthAtW - length;
} else
newLength = std::min(length + lengthAtW, strokeLength);
return stroke2change->getParameterAtLength(newLength);
}
//---------------------------------------------------------------------------
bool rotateStroke(const TStroke *stroke2change, TStroke *&rotated, double &from,
double &to, TPointD &old_w0_pos) {
if (!stroke2change || !isValid(from) || !isValid(to)) return false;
rotated = 0;
// save position of w=0 (useful to retrieve this
// position after changes)
old_w0_pos = convert(stroke2change->getControlPoint(0));
double rotateAtLength = stroke2change->getLength(to);
assert(rotateAtLength >= 0.0);
if (rotateAtLength < 0) return false;
// TStroke*
rotated = ToonzExt::rotateControlPoint(stroke2change, ToonzExt::EvenInt(0),
rotateAtLength);
if (!rotated) return false;
from = rotated->getW(stroke2change->getPoint(from));
to = rotated->getW(stroke2change->getPoint(to));
// save some other information (style, etc..)
ToonzExt::cloneStrokeStatus(stroke2change, rotated);
return true; // rotated;
}
//---------------------------------------------------------------------------
TStroke *rotateControlPointAtPoint(const TStroke *stroke, const TPointD &pnt) {
if (!stroke || tdistance2(stroke->getPoint(0.0), pnt) < sq(TConsts::epsilon))
return 0;
double w = stroke->getW(pnt);
TPointD theSamePnt = stroke->getPoint(w);
double length = stroke->getLength(w);
return ToonzExt::rotateControlPoint(stroke, ToonzExt::EvenInt(0), length);
}
//---------------------------------------------------------------------------
bool findExtremesFromActionLength(double toolActionLength,
const TStroke *stroke, double w,
ToonzExt::Interval &out) {
out = ToonzExt::Interval(-1.0, -1.0);
if (!stroke || 0.0 > w || w > 1.0) return false;
double emiToolSize = 0.5 * toolActionLength;
double strokelength = stroke->getLength();
double lengthAtParam = stroke->getLength(w);
assert(emiToolSize >= 0.0 && strokelength >= 0.0 && lengthAtParam >= 0.0);
if (emiToolSize > strokelength * 0.5) {
if (stroke->isSelfLoop()) {
emiToolSize = strokelength * 0.5;
// first and second are the same
lengthAtParam += emiToolSize;
if (lengthAtParam > strokelength) lengthAtParam -= strokelength;
out.first = out.second = stroke->getParameterAtLength(lengthAtParam);
} else {
out.first = 0.0;
out.second = 1.0;
}
return true;
}
if (emiToolSize < 0.0 || strokelength < 0.0 || lengthAtParam < 0.0)
return false;
// [out] now is length, range 0,length
out.first = lengthAtParam - emiToolSize;
out.second = lengthAtParam + emiToolSize;
if (stroke->isSelfLoop()) {
if (out.first < 0.0) {
out.first = strokelength + out.first;
assert(out.first < strokelength);
}
if (out.second > strokelength) {
out.second = out.second - strokelength;
assert(0.0 < out.second && out.second < strokelength);
}
} else {
out.first = std::max(out.first, 0.0);
out.second = std::min(out.second, strokelength);
}
// [out] now is parameter, range 0,1
out.first = stroke->getParameterAtLength(out.first);
out.second = stroke->getParameterAtLength(out.second);
#ifdef _DEBUG
if (!stroke->isSelfLoop()) assert(out.first <= out.second);
#endif
return true;
}
//---------------------------------------------------------------------------
bool areDifferent(const TStroke *ref1, const TStroke *ref2) {
if (!isValid(ref1) || !isValid(ref2)) return true;
int cpCount1 = ref1->getControlPointCount();
int cpCount2 = ref2->getControlPointCount();
if (cpCount1 != cpCount2) return true;
while (--cpCount1 >= 0) {
if (ref1->getControlPoint(cpCount1) != ref2->getControlPoint(cpCount1))
return true;
}
assert(cpCount1 == -1);
return false;
}
//---------------------------------------------------------------------------
}
//-----------------------------------------------------------------------------
TStroke *StrokeDeformationImpl::copyOfLastSelectedStroke_ =
0; // deep copy stroke selected previously
//-----------------------------------------------------------------------------
const ContextStatus *&StrokeDeformationImpl::getImplStatus() {
// if multi threading this datas need to be serialized
// a ref to status
static const ContextStatus *contextStatus_instance;
return contextStatus_instance;
}
//-----------------------------------------------------------------------------
TStroke *&StrokeDeformationImpl::getLastSelectedStroke() {
static TStroke *lastSelectedStroke_instance;
return lastSelectedStroke_instance;
}
//-----------------------------------------------------------------------------
void StrokeDeformationImpl::setLastSelectedStroke(TStroke *stroke) {
TStroke *&lastSelStroke = getLastSelectedStroke();
lastSelStroke = stroke;
if (lastSelStroke) {
delete copyOfLastSelectedStroke_;
copyOfLastSelectedStroke_ = new TStroke(*lastSelStroke);
}
}
//-----------------------------------------------------------------------------
int &StrokeDeformationImpl::getLastSelectedDegree() {
static int lastSelectedDegree_instance;
return lastSelectedDegree_instance;
}
//-----------------------------------------------------------------------------
void StrokeDeformationImpl::setLastSelectedDegree(int degree) {
getLastSelectedDegree() = degree;
}
//-----------------------------------------------------------------------------
ToonzExt::Intervals &StrokeDeformationImpl::getSpiresList() {
static ToonzExt::Intervals listOfSpire_instance;
return listOfSpire_instance;
}
//-----------------------------------------------------------------------------
ToonzExt::Intervals &StrokeDeformationImpl::getStraightsList() {
static ToonzExt::Intervals listOfStraight_instance;
return listOfStraight_instance;
}
//-----------------------------------------------------------------------------
StrokeDeformationImpl::StrokeDeformationImpl() {
shortcutKey_ = ContextStatus::NONE;
cursorId_ = -1;
this->reset();
}
//-----------------------------------------------------------------------------
StrokeDeformationImpl::~StrokeDeformationImpl() {
clearPointerContainer(strokes_);
delete potential_;
potential_ = 0;
delete deformer_;
deformer_ = 0;
delete copyOfLastSelectedStroke_;
copyOfLastSelectedStroke_ = 0;
}
//-----------------------------------------------------------------------------
bool StrokeDeformationImpl::activate_impl(const ContextStatus *status) {
assert(status && "Not status available");
if (!status || !this->init(status)) return false;
double w = status->w_;
ToonzExt::Interval extremes = this->getExtremes();
TStroke *stroke2transform;
if (!this->computeStroke2Transform(status, stroke2transform, w, extremes))
return false;
// to avoid strange behaviour in limit's value
if (areAlmostEqual(extremes.first, w)) w = extremes.first;
if (areAlmostEqual(extremes.second, w)) w = extremes.second;
assert(extremes.first <= w && w <= extremes.second);
if (extremes.first > w || w > extremes.second) {
return false;
}
std::vector<double> splitParameter;
splitParameter.push_back(extremes.first);
splitParameter.push_back(extremes.second);
assert(strokes_.empty());
if (!strokes_.empty()) clearPointerContainer(strokes_);
splitStroke(*stroke2transform, splitParameter, strokes_);
assert(strokes_.size() == 3);
// I don't know how to manage properly this case
if (strokes_.size() != 3) {
clearPointerContainer(strokes_);
return false;
}
// stroke to change
stroke2manipulate_ = strokes_[1];
assert(stroke2manipulate_ && " Not valid reference to stroke to move!!!");
if (!stroke2manipulate_) return false;
// remove empty stroke
{
TStroke *tmp_stroke = strokes_[2];
if (isAlmostZero(tmp_stroke->getLength())) {
std::vector<TStroke *>::iterator it = strokes_.begin();
std::advance(it, 2);
strokes_.erase(it);
delete tmp_stroke;
}
tmp_stroke = strokes_[0];
if (isAlmostZero(tmp_stroke->getLength())) {
strokes_.erase(strokes_.begin());
delete tmp_stroke;
}
// little movement of control points to have not empty
// length
if (isAlmostZero(stroke2manipulate_->getLength())) {
int count = stroke2manipulate_->getControlPointCount() - 1;
assert(count > 0);
TThickPoint pnt0 = stroke2manipulate_->getControlPoint(0),
pntn = stroke2manipulate_->getControlPoint(count),
delta = TThickPoint(2.0 * TConsts::epsilon,
-2.0 * TConsts::epsilon, 0.0);
stroke2manipulate_->setControlPoint(0, pnt0 - delta);
stroke2manipulate_->setControlPoint(count, pntn + delta);
}
}
// simple check for empty chunk
int cp_count = stroke2manipulate_->getChunkCount();
const TThickQuadratic *tq;
bool haveEmpty = false;
while (--cp_count >= 0) {
tq = stroke2manipulate_->getChunk(cp_count);
// assert( tq->getLength() != 0.0 );
if (tq->getLength() == 0.0) haveEmpty = true;
}
if (haveEmpty) {
const double reductionFactor =
computeReductionFactor(getImplStatus()->pixelSize_, 1.0);
stroke2manipulate_->reduceControlPoints(reductionFactor);
}
TPointD pntOnStroke = stroke2transform->getPoint(w);
// parameter need to be recomputed to match
// with new stroke
w = stroke2manipulate_->getW(pntOnStroke);
// prevStyle_ = stroke2transform->getStyle();
// set a style for error
// stroke2manipulate_->setStyle( 4 );
// in this deformer all stroke need to be moved
double lengthOfAction = this->findActionLength();
bool exception = false;
try {
// create a new stroke deformer
delete deformer_;
deformer_ = new StrokeParametricDeformer(
lengthOfAction, w, stroke2manipulate_, potential_->clone());
} catch (std::invalid_argument &) {
exception = true;
assert(!"Invalid argument");
} catch (...) {
exception = true;
}
assert(deformer_ && "Deformer is not available");
if (exception) {
deformer_ = 0;
return false;
}
assert(getImplStatus() != 0 && "ContextStatus is null???");
if (!getImplStatus()) {
delete deformer_;
this->reset();
return false;
}
// change the threshold value of increser,
// if value is greater than diff add control points
deformer_->setDiff(getImplStatus()->deformerSensitivity_);
// just to be sure to have a control point where
// stroke is selected
stroke2manipulate_->insertControlPoints(w);
// two different case:
// (1) point at w=0 is not in stroke to transform
// (2) point at w=0 is in stroke to transform
// in case (2) it is mandatory to retrieve the
// parameter where stroke2transform is equal to
// w=0 and store it.
// old_w0_pos_ will be used in deactivate method
if (old_w0_pos_ != TConsts::napd) {
double w = stroke2manipulate_->getW(old_w0_pos_);
TPointD pnt = stroke2manipulate_->getPoint(w);
if (tdistance2(pnt, old_w0_pos_) < sq(TConsts::epsilon)) {
// the this position can be modifyed during drag
old_w0_ = w;
} else
old_w0_ = -1;
}
bool test = increaseControlPoints(*stroke2manipulate_, *deformer_,
getImplStatus()->pixelSize_);
/*
// New Increaser behaviour
deformer_->setMouseMove(0,100);
extremes_ = Toonz::increase_cp (stroke2manipulate_,
deformer_);
//*/
if (!test) return false;
stroke2manipulate_->disableComputeOfCaches();
return true;
}
//-----------------------------------------------------------------------------
void StrokeDeformationImpl::update_impl(const TPointD &delta) {
assert(stroke2manipulate_ && deformer_ &&
"Stroke and Deformer are available");
if (!stroke2manipulate_ || !deformer_) return;
// update delta in deformer
deformer_->setMouseMove(delta.x, delta.y);
modifyControlPoints(*stroke2manipulate_, *deformer_);
}
//-----------------------------------------------------------------------------
TStroke *StrokeDeformationImpl::deactivate_impl() {
if (!stroke2manipulate_ || !getImplStatus()) {
this->reset();
return 0;
}
// retrieve the point
if (old_w0_ != -1) old_w0_pos_ = stroke2manipulate_->getPoint(old_w0_);
const double reductionFactor =
computeReductionFactor(getImplStatus()->pixelSize_, 3.0);
// stroke2manipulate_->enableComputeOfCaches ();
// stroke2manipulate_->reduceControlPoints (reductionFactor);
const int size = stroke2manipulate_->getControlPointCount();
std::vector<TThickPoint> pnt(size);
for (int i = 0; i < size; ++i)
pnt[i] = stroke2manipulate_->getControlPoint(i);
std::vector<TStroke *>::iterator it =
std::find(strokes_.begin(), strokes_.end(), stroke2manipulate_);
assert(it != strokes_.end());
int pos = std::distance(strokes_.begin(), it);
stroke2manipulate_ = 0;
delete strokes_[pos];
TStroke *tmpStroke = new TStroke(pnt);
assert((tmpStroke != 0) && "Not valid stroke!");
strokes_[pos] = tmpStroke;
strokes_[pos]->reduceControlPoints(reductionFactor);
// strokes_[pos] = TStroke::interpolate(pnt,
// reductionFactor,
// false);
TStroke *ref = Toonz::merge(strokes_);
ToonzExt::cloneStrokeStatus(getImplStatus()->stroke2change_, ref);
if (old_w0_pos_ != TConsts::napd) {
// restore (if necessary) the initial position of
// a stroke
TStroke *rotated = rotateControlPointAtPoint(ref, old_w0_pos_);
if (rotated) {
delete ref;
ref = rotated;
if (getImplStatus())
ToonzExt::cloneStrokeStatus(getImplStatus()->stroke2change_, ref);
else {
delete ref;
ref = 0;
}
}
}
this->reset();
return ref;
}
//-----------------------------------------------------------------------------
void StrokeDeformationImpl::reset() {
old_w0_ = -1;
old_w0_pos_ = TConsts::napd;
deformer_ = 0;
getImplStatus() = 0;
this->setLastSelectedDegree(-1);
this->setLastSelectedStroke(0);
this->getSpiresList().clear();
this->getStraightsList().clear();
stroke2manipulate_ = 0;
clearPointerContainer(strokes_); // stroke2move is deleted here (it
// is in vector)
}
//-----------------------------------------------------------------------------
TStroke *StrokeDeformationImpl::getTransformedStroke() {
return this->stroke2manipulate_;
}
//-----------------------------------------------------------------------------
ToonzExt::Potential *StrokeDeformationImpl::getPotential() {
return this->potential_;
}
//-----------------------------------------------------------------------------
void StrokeDeformationImpl::draw(Designer *designer) {}
//-----------------------------------------------------------------------------
void StrokeDeformationImpl::setPotential(Potential *potential) {
assert(potential);
potential_ = potential;
}
//-----------------------------------------------------------------------------
bool StrokeDeformationImpl::check(const ContextStatus *status) {
if (!status || !this->init(status)) return false;
return this->check_(status);
}
//-----------------------------------------------------------------------------
TPointD &StrokeDeformationImpl::oldW0() { return old_w0_pos_; }
//-----------------------------------------------------------------------------
ToonzExt::Interval StrokeDeformationImpl::getExtremes() {
ToonzExt::Interval extremes = ToonzExt::Interval(-1.0, -1.0);
if (!getImplStatus()) return extremes;
// if it is manual
if (getImplStatus()->isManual_ == true) {
findExtremesFromActionLength(getImplStatus()->lengthOfAction_,
getImplStatus()->stroke2change_,
getImplStatus()->w_, extremes);
} else {
// if there are some special key down all will be managed in deformation
this->findExtremes_(getImplStatus(), extremes);
}
return extremes;
}
//-----------------------------------------------------------------------------
bool StrokeDeformationImpl::init(const ContextStatus *status) {
if (!status || !isValid(status->stroke2change_) || !isValid(status->w_)) {
this->reset();
return false;
}
getImplStatus() = status;
if (!this->getLastSelectedStroke() ||
this->getLastSelectedStroke() != status->stroke2change_ ||
areDifferent(copyOfLastSelectedStroke_, status->stroke2change_) ||
(this->getLastSelectedDegree() == -1) ||
(this->getLastSelectedDegree() != status->cornerSize_)) {
this->getSpiresList().clear();
this->getStraightsList().clear();
// recompute cache
ToonzExt::findCorners(status->stroke2change_, this->getSpiresList(),
this->getStraightsList(), status->cornerSize_,
TConsts::epsilon);
this->setLastSelectedStroke(status->stroke2change_);
this->setLastSelectedDegree(status->cornerSize_);
}
return true;
}
//-----------------------------------------------------------------------------
bool StrokeDeformationImpl::computeStroke2Transform(
const ContextStatus *status, TStroke *&stroke2transform, double &w,
ToonzExt::Interval &extremes) {
if (!status || !isValid(w)) return false;
stroke2transform = 0;
if (status->stroke2change_->isSelfLoop()) {
if (extremes.first > extremes.second) {
double new_w = (extremes.first + extremes.second) * 0.5;
if (!rotateStroke(status->stroke2change_, stroke2transform, w, new_w,
old_w0_pos_))
return false;
ContextStatus tmpStatus = *status;
tmpStatus.stroke2change_ = stroke2transform;
tmpStatus.w_ = w;
this->check(&tmpStatus);
extremes = this->getExtremes();
this->init(status);
return true;
} else if (extremes.first == extremes.second) {
double positionOfFixPoint = -1;
// there are two different cases
// a) only one corner
// b) only two corners
switch (this->getSpiresList().size()) {
case 0:
assert(extremes.first == -1);
positionOfFixPoint = retrieveParamAtLengthWithOffset(
status->stroke2change_, status->stroke2change_->getLength() * 0.5,
w);
break;
case 1:
if (extremes.first != -1)
positionOfFixPoint = retrieveParamAtLengthWithOffset(
status->stroke2change_, status->stroke2change_->getLength() * 0.5,
w);
else
positionOfFixPoint = this->getSpiresList()[0].first;
break;
case 2:
// also forced mode
default:
assert(extremes.first != -1);
// if( extremes.first == -1 )
// positionOfFixPoint=retrieveParamAtLengthWithOffset(status->stroke2change_,
// status->stroke2change_->getLength()*0.5,
// w);
// else
positionOfFixPoint = extremes.first;
break;
}
if (!rotateStroke(status->stroke2change_, stroke2transform, w,
positionOfFixPoint, old_w0_pos_))
return false;
extremes = ToonzExt::Interval(0.0, 1.0);
return true;
}
}
if (!isValid(extremes.first) || !isValid(extremes.second)) return false;
try {
if (!stroke2transform)
stroke2transform = new TStroke(*status->stroke2change_);
} catch (...) {
return false;
}
return true;
}
//-----------------------------------------------------------------------------
// End Of File
//-----------------------------------------------------------------------------