#include "tools/tool.h"
#include "tools/toolutils.h"
#include "tstroke.h"
#include "tstrokeutil.h"
#include "tstrokedeformations.h"
#include "tmathutil.h"
#include "tools/cursors.h"
#include "drawutil.h"
#include "toonz/tobjecthandle.h"
#include "toonz/txshlevelhandle.h"
#include "toonz/tstageobject.h"
using namespace ToolUtils;
//=============================================================================
namespace
{
const UINT MAX_SAMPLE = 2;
const int MY_ERROR = -1;
const double c_LenghtOfBenderRegion = 10.0;
const int IS_BEGIN = 0;
const int IS_END = 1;
const int IS_ALL = 2;
//-----------------------------------------------------------------------------
double _extractFirst(DoublePair val)
{
return val.first;
}
//-----------------------------------------------------------------------------
double _extractSecond(DoublePair val)
{
return val.second;
}
//-----------------------------------------------------------------------------
bool strokeIsConnected(const TStroke &s, double toll = TConsts::epsilon)
{
bool out = true;
int count = s.getChunkCount();
long double toll2 = sq(TConsts::epsilon);
if (count > 0) {
const TThickQuadratic *refTQ = s.getChunk(0);
for (int i = 1; i < count; ++i) {
if (out &&
tdistance2(refTQ->getP2(), s.getChunk(i)->getP0()) > toll2)
out = false;
}
}
return out;
}
//-----------------------------------------------------------------------------
inline bool strokeIsConnected(const TStroke *s, double toll = TConsts::epsilon)
{
assert(s);
return strokeIsConnected(*s, toll);
}
//-----------------------------------------------------------------------------
/*
Extract an value of a pair.
*/
void extract(const std::vector<DoublePair> &s, ArrayOfDouble &d, bool isFirstOrSecond = true)
{
if (isFirstOrSecond)
std::transform(s.begin(), s.end(), back_inserter(d), _extractFirst);
else
std::transform(s.begin(), s.end(), back_inserter(d), _extractSecond);
}
//-----------------------------------------------------------------------------
double retrieveInitLength(double strokeLength, int beginEndOrAll)
{
double initLength = MY_ERROR;
switch (beginEndOrAll) {
case IS_BEGIN:
initLength = 0.0;
break;
case IS_END:
initLength = strokeLength;
break;
case IS_ALL:
initLength = strokeLength * 0.5;
break;
}
return initLength;
}
//-----------------------------------------------------------------------------
inline bool isOdd(UINT val)
{
return val & 1 ? true : false;
}
inline bool isEven(UINT val)
{
return !isOdd(val);
}
void clearPointerMap(std::map<TStroke *, std::vector<int> *> &corners)
{
std::map<TStroke *, std::vector<int> *>::iterator it = corners.begin();
for (; it != corners.end(); ++it) {
delete it->second;
}
}
//=============================================================================
// Bender Tool
//-----------------------------------------------------------------------------
class BenderTool : public TTool
{
private:
TUndo *m_undo;
bool m_atLeastOneIsChanged;
std::vector<bool> m_directionIsChanged;
std::vector<TPointD> m_accumulator;
void increaseCP(TStroke *, int);
bool m_active;
int m_cursor;
enum PNT_SELECTED {
BNDR_NULL = 0,
BNDR_P0 = 1,
BNDR_P1 = 2
};
struct benderStrokeInfo {
TStroke *m_stroke;
DoublePair m_extremes;
int m_isBeginEndOrAll;
benderStrokeInfo(TStroke *stroke,
DoublePair &info,
int isBeginEndOrAll)
{
m_stroke = stroke;
m_extremes = info;
m_isBeginEndOrAll = isBeginEndOrAll;
}
};
std::vector<benderStrokeInfo> m_info;
// contains information about stroke which have
// intersection with benderSegment
std::map<TStroke *, ArrayOfStroke> m_metaStroke;
std::map<TStroke *, std::vector<int> *> m_hitStrokeCorners;
bool m_showTangents;
// value selected
int m_buttonDownCounter;
TSegment m_benderSegment;
TPointD m_prevPoint;
double m_rotationVersus;
//!
ArrayOfStroke m_strokesToRotate;
//!
ArrayOfStroke m_strokesToBend;
std::vector<int> m_changedStrokes;
bool m_enableDragSelection;
void findCurves(TVectorImageP &);
void findVersus(const TPointD &p);
double computeRotationVersus(const TPointD &,
const TPointD &);
public:
void initBenderAction(TVectorImageP &, const TPointD &);
// BenderTool(TVectorImageP vimage, GLTestWidget* ref );
BenderTool();
virtual ~BenderTool();
ToolType getToolType() const { return TTool::LevelWriteTool; }
void draw();
void leftButtonDown(const TPointD &, const TMouseEvent &);
void leftButtonDrag(const TPointD &, const TMouseEvent &);
void leftButtonUp(const TPointD &, const TMouseEvent &);
void onEnter();
void onActivate()
{
m_buttonDownCounter = 1;
m_prevPoint = TConsts::napd;
m_benderSegment.setP0(TConsts::napd);
m_benderSegment.setP1(TConsts::napd);
m_metaStroke.clear();
clearPointerMap(m_hitStrokeCorners);
m_hitStrokeCorners.clear();
}
int getCursorId() const { return m_cursor; }
} BenderTool;
//-----------------------------------------------------------------------------
BenderTool::BenderTool()
: TTool("T_Bender"), m_showTangents(false), m_buttonDownCounter(1), m_rotationVersus(0.0), m_enableDragSelection(false), m_undo(0), m_cursor(ToolCursor::BenderCursor)
{
bind(TTool::Vectors);
m_prevPoint = TConsts::napd;
m_benderSegment.setP0(TConsts::napd);
m_benderSegment.setP1(TConsts::napd);
}
//-----------------------------------------------------------------------------
BenderTool::~BenderTool()
{
}
//-----------------------------------------------------------------------------
void BenderTool::onEnter()
{
if ((TVectorImageP)getImage(false))
m_cursor = ToolCursor::BenderCursor;
else
m_cursor = ToolCursor::CURSOR_NO;
}
//-----------------------------------------------------------------------------
void BenderTool::leftButtonUp(const TPointD &pos, const TMouseEvent &)
{
m_active = false;
TVectorImageP vi = TImageP(getImage(true));
if (!vi)
return;
QMutexLocker lock(vi->getMutex());
m_active = true;
std::vector<TStroke *> oldStrokesArray(m_changedStrokes.size());
int i;
for (i = 0; i < (int)m_changedStrokes.size(); i++)
oldStrokesArray[i] = new TStroke(*(vi->getStroke(m_changedStrokes[i])));
if (3 == m_buttonDownCounter) {
m_rotationVersus = 0.0;
// hide bender tool
m_prevPoint = TConsts::napd;
m_benderSegment.setP0(TConsts::napd);
m_benderSegment.setP1(TConsts::napd);
typedef std::map<TStroke *, ArrayOfStroke>::iterator itAATS;
//UINT count=0;
for (itAATS it1 = m_metaStroke.begin(); it1 != m_metaStroke.end(); ++it1) {
UINT i;
// copy vector and...
ArrayOfStroke &refA = it1->second;
TStroke *s = it1->first;
TStroke *out = merge(refA);
out->reduceControlPoints(getPixelSize(), *(m_hitStrokeCorners[s]));
if (it1->first->isSelfLoop()) {
int cpCount = out->getControlPointCount();
TThickPoint p1 = out->getControlPoint(0);
TThickPoint p2 = out->getControlPoint(cpCount - 1);
TThickPoint midP = (p1 + p2) * 0.5;
out->setControlPoint(0, midP);
out->setControlPoint(cpCount - 1, midP);
out->setSelfLoop(true);
}
it1->first->swap(*out);
it1->first->setStyle(out->getStyle());
it1->first->outlineOptions() = out->outlineOptions();
it1->first->invalidate();
for (i = 0; i < refA.size(); ++i)
delete refA[i];
delete out;
}
m_metaStroke.clear();
clearPointerMap(m_hitStrokeCorners);
m_hitStrokeCorners.clear();
m_buttonDownCounter = 1;
UINT lastCpIndex, i, size = m_changedStrokes.size();
TStroke *stroke;
TThickPoint p1, p2, middleP;
double loopError = 0.5 * getPixelSize();
for (i = 0; i < size; i++) {
stroke = vi->getStroke(m_changedStrokes[i]);
if (m_directionIsChanged[i])
stroke->changeDirection();
lastCpIndex = stroke->getControlPointCount() - 1;
p1 = stroke->getControlPoint(0);
p2 = stroke->getControlPoint(lastCpIndex);
if (isAlmostZero(p1.x - p2.x, loopError) && isAlmostZero(p1.y - p2.y, loopError)) {
middleP = (p1 + p2) * 0.5;
stroke->setControlPoint(0, middleP);
stroke->setControlPoint(lastCpIndex, middleP);
} else {
stroke->setSelfLoop(false);
}
}
vi->notifyChangedStrokes(m_changedStrokes, oldStrokesArray);
notifyImageChanged();
if (m_undo)
TUndoManager::manager()->add(m_undo);
m_undo = 0;
}
clearPointerContainer(oldStrokesArray);
m_active = false;
invalidate();
}
//-----------------------------------------------------------------------------
void BenderTool::leftButtonDrag(const TPointD &pos, const TMouseEvent &)
{
if (!m_active)
return;
TVectorImageP vi = TImageP(getImage(true));
if (!vi)
return;
QMutexLocker lock(vi->getMutex());
double pixelSize = getPixelSize();
if (tdistance2(pos, m_prevPoint) < 9.0 * pixelSize * pixelSize)
return;
if (m_buttonDownCounter < 3)
return;
if (m_enableDragSelection) {
m_accumulator.push_back(pos);
if (m_accumulator.size() < 3)
return;
TPointD middlePnt;
accumulate(m_accumulator.begin(), m_accumulator.end(), middlePnt);
static const double inv_of_3 = 1.0 / 2.0;
middlePnt = inv_of_3 * middlePnt;
m_accumulator.clear();
initBenderAction(vi, pos + middlePnt);
m_enableDragSelection = false;
m_buttonDownCounter = 3;
}
TPointD p = pos;
TPointD
vc(p - m_benderSegment.getP0()), // current vector
vp(m_prevPoint - m_benderSegment.getP0()); // previous vector
TPointD s2v = m_benderSegment.getSpeed();
// some check to jump invalid case
double norm2BenderSeg = norm2(s2v);
double norm2CurrentVect = norm2(vc);
double norm2PreviousVect = norm2(vp);
// invalid segments
if (0.0 == norm2BenderSeg ||
0.0 == norm2PreviousVect ||
0.0 == norm2CurrentVect)
return;
// invalid rotation versus
if (tsign(cross(s2v, vc)) != m_rotationVersus)
return;
// compute delta of rotation angle
double diff = asin(cross(normalize(vp), normalize(vc)));
// make a rototranslation matrix
TRotation rot(m_benderSegment.getP0(), rad2degree(diff));
// rotate references
for (itAS its = m_strokesToRotate.begin(); its != m_strokesToRotate.end(); ++its)
(*its)->transform(rot);
// deform strokes
for (UINT i = 0; i < m_info.size(); ++i) {
TStroke &ref = *m_info[i].m_stroke;
// DoublePair &extr = m_info[i].m_extremes;
double strokeLength = ref.getLength();
double initLength = retrieveInitLength(strokeLength, m_info[i].m_isBeginEndOrAll); // ? 0.0 : strokeLength;
if (MY_ERROR == initLength)
return;
int innerOrOuter = m_info[i].m_isBeginEndOrAll == IS_ALL ? TStrokeBenderDeformation::OUTER : TStrokeBenderDeformation::INNER;
TStrokeBenderDeformation
def(&ref,
m_benderSegment.getP0(),
diff,
initLength,
innerOrOuter,
strokeLength);
modifyControlPoints(ref, def);
}
// fix previous point
m_prevPoint = p;
invalidate();
}
//-----------------------------------------------------------------------------
void BenderTool::leftButtonDown(const TPointD &p, const TMouseEvent &)
{
m_active = false;
TVectorImageP vi = TImageP(getImage(true));
if (!vi)
return;
QMutexLocker lock(vi->getMutex());
m_active = true;
switch (m_buttonDownCounter) {
case 1:
findCurves(vi);
m_strokesToRotate.clear();
// le curve puntate sono state eliminate dalla clear precedente
m_info.clear();
m_benderSegment.setP0(p);
m_benderSegment.setP1(p);
break;
case 2: // second buttonDown
m_prevPoint = p;
m_benderSegment.setP1(p);
m_enableDragSelection = true;
break;
/*
case 3: // third buttonDown
// initBenderAction( vi, p );
break;
*/
}
++m_buttonDownCounter;
invalidate();
//vi->validateRegionEdges(vi->getStroke( m_strokeIndex ), true);
}
//-----------------------------------------------------------------------------
void BenderTool::findVersus(const TPointD &p)
{
TPointD
v1 = m_benderSegment.getSpeed(),
v2 = p - m_benderSegment.getP0();
// fix rotation versus
m_rotationVersus = tsign(cross(v1, v2));
if (isAlmostZero(m_rotationVersus))
m_rotationVersus = 1.0;
}
//-----------------------------------------------------------------------------
inline void one_minus_x(double &x)
{
x = 1.0 - x;
}
//-----------------------------------------------------------------------------
void BenderTool::findCurves(TVectorImageP &vi)
{
ArrayOfStroke strokeToModify;
m_changedStrokes.clear();
m_directionIsChanged.clear();
UINT j;
for (UINT i = 0; i < vi->getStrokeCount(); ++i) // for all stroke in image
{
if (!vi->inCurrentGroup(i))
continue;
TStroke *s = vi->getStroke(i); // a useful reference
std::vector<DoublePair> pair_intersection; // informations about extremes
// if there is interesection between stroke and bender tool
if (intersect(*s, m_benderSegment, pair_intersection)) {
if (s->isSelfLoop()) {
//make the semgnet longer
// such as the points are just a littleBit
// outside the stroke bbox
const double littleBit = 0.1;
TRectD bbox = s->getBBox();
TPointD bboxP0 = bbox.getP00();
TPointD bboxP1 = bbox.getP11();
double bboxX0 = std::min(bboxP0.x, bboxP1.x);
double bboxX1 = std::max(bboxP0.x, bboxP1.x);
double bboxY0 = std::min(bboxP0.y, bboxP1.y);
double bboxY1 = std::max(bboxP0.y, bboxP1.y);
TSegment segment;
TPointD pp0 = m_benderSegment.getP0();
TPointD pp1 = m_benderSegment.getP1();
TPointD newP0;
TPointD newP1;
if (bbox.contains(pp1)) {
double x = (pp0.x < pp1.x) ? bboxX1 : bboxX0;
double y = (pp0.y < pp1.y) ? bboxY1 : bboxY0;
double t;
if (pp1.x == pp0.x && pp1.y == pp0.y) {
assert(!"segmento del bender puntiforme");
return;
}
if (pp1.x == pp0.x)
t = (y - pp0.y) / (pp1.y - pp0.y);
else if (pp1.y == pp0.y)
t = (x - pp0.x) / (pp1.x - pp0.x);
else
t = std::min((x - pp0.x) / (pp1.x - pp0.x), (y - pp0.y) / (pp1.y - pp0.y));
newP1 = (t > 1) ? m_benderSegment.getPoint(t + littleBit) : pp1;
} else
newP1 = pp1;
if (bbox.contains(pp0)) {
double x = (pp0.x > pp1.x) ? bboxX1 : bboxX0;
double y = (pp0.y > pp1.y) ? bboxY1 : bboxY0;
double t;
if (pp1.x == pp0.x && pp1.y == pp0.y) {
assert(!"segmento del bender puntiforme");
return;
}
if (pp1.x == pp0.x)
t = (y - pp0.y) / (pp1.y - pp0.y);
else if (pp1.y == pp0.y)
t = (x - pp0.x) / (pp1.x - pp0.x);
else
t = std::max((x - pp0.x) / (pp1.x - pp0.x), (y - pp0.y) / (pp1.y - pp0.y));
newP0 = (t < 0) ? m_benderSegment.getPoint(t - littleBit) : pp0;
} else
newP0 = pp0;
segment = TSegment(newP0, newP1);
pair_intersection.clear();
intersect(*s, segment, pair_intersection);
assert(isEven(pair_intersection.size()));
assert(!pair_intersection.empty());
if (pair_intersection.empty()) //non dovrebbe accadere
continue;
}
//-----------------------------------------------------------------------------
strokeToModify.push_back(s);
m_changedStrokes.push_back(i);
//-----------------------------------------------------------------------------
m_metaStroke[s] = ArrayOfStroke();
ArrayOfStroke &tempAS = m_metaStroke[s];
// extract information about intersection parameter
ArrayOfDouble intersection;
extract(pair_intersection, intersection);
// now add stroke to rotate in m_info struct
TPointD v = s->getSpeed(intersection[0]);
TPointD normalToBenderSeg;
normalToBenderSeg = m_rotationVersus * rotate90(m_benderSegment.getSpeed());
m_atLeastOneIsChanged = false;
//m_directionIsChanged = false;
if (tsign(v * normalToBenderSeg) < 0) {
m_atLeastOneIsChanged = true;
m_directionIsChanged.push_back(true);
std::for_each(intersection.begin(), intersection.end(), one_minus_x);
std::reverse(intersection.begin(), intersection.end());
s->changeDirection();
} else
m_directionIsChanged.push_back(false);
splitStroke(*s, intersection, tempAS);
// number of curves is number of intersection plus one
UINT numberOfCurves = intersection.size() + 1;
// and begin increase of control point
if (isEven(intersection.size()) && m_atLeastOneIsChanged) // if solution are even
{
for (j = 0; j < numberOfCurves; ++j) {
if (isOdd(j))
increaseCP(tempAS[j], IS_ALL);
else
m_strokesToRotate.push_back(tempAS[j]);
}
} else {
increaseCP(tempAS[0], IS_END);
for (j = 1; j < numberOfCurves - 1; ++j) {
if (isEven(j))
increaseCP(tempAS[j], IS_ALL);
else
m_strokesToRotate.push_back(tempAS[j]);
}
if (isOdd(numberOfCurves))
increaseCP(tempAS.back(), IS_BEGIN);
else
m_strokesToRotate.push_back(tempAS.back());
}
TStroke *tempForCorners = merge(tempAS);
std::vector<int> *corners = new std::vector<int>;
corners->push_back(0);
detectCorners(tempForCorners, 20, *corners);
corners->push_back(tempForCorners->getChunkCount());
m_hitStrokeCorners[s] = corners;
delete tempForCorners;
}
}
if (!strokeToModify.empty()) {
UINT i, size = strokeToModify.size();
for (i = 0; i < size; i++)
if (m_directionIsChanged[i])
strokeToModify[i]->changeDirection();
if (TTool::getApplication()->getCurrentObject()->isSpline())
m_undo = new UndoPath(getXsheet()->getStageObject(getObjectId())->getSpline());
else {
TXshSimpleLevel *sl = TTool::getApplication()->getCurrentLevel()->getSimpleLevel();
assert(sl);
TFrameId id = getCurrentFid();
m_undo = new UndoModifyListStroke(sl, id, strokeToModify);
}
for (i = 0; i < size; i++)
if (m_directionIsChanged[i])
strokeToModify[i]->changeDirection();
}
}
//-----------------------------------------------------------------------------
void BenderTool::increaseCP(TStroke *tmpStroke, int beginEndOrAll)
{
DoublePair extremes;
double strokeLength = tmpStroke->getLength();
double initLength = retrieveInitLength(strokeLength, beginEndOrAll);
if (MY_ERROR == initLength)
return;
TStrokeBenderDeformation deformer(tmpStroke, initLength, strokeLength * 0.5);
increaseControlPoints(*tmpStroke, deformer, getPixelSize());
tmpStroke->disableComputeOfCaches();
m_info.push_back(benderStrokeInfo(tmpStroke, extremes, beginEndOrAll));
}
//-----------------------------------------------------------------------------
void BenderTool::draw()
{
//TAffine viewMatrix = getViewer()->getViewMatrix();
//glPushMatrix();
//tglMultMatrix(viewMatrix);
double pixelSize = getPixelSize();
typedef std::map<TStroke *, ArrayOfStroke>::const_iterator mapTACit;
for (mapTACit cit1 = m_metaStroke.begin(); cit1 != m_metaStroke.end(); ++cit1) {
const ArrayOfStroke &tmp = (*cit1).second;
tglColor(TPixel32::Red);
for (citAS cit2 = tmp.begin(); cit2 != tmp.end(); ++cit2)
drawStrokeCenterline(**cit2, pixelSize);
//drawStroke(**cit2, TRect(0,0,1000,1000), TAffine() );
}
double length = m_benderSegment.getLength();
// pnt to draw rotation vector
TPointD pnt;
if (length != 0.0) {
TPointD v = m_prevPoint - m_benderSegment.getP0();
double tmp = norm2(v);
if (tmp != 0.0) {
pnt = v * (length / sqrt(tmp));
pnt += m_benderSegment.getP0();
}
} else
pnt = m_prevPoint;
// rotation vector
if (m_buttonDownCounter == 3) {
tglColor(TPixel::Black);
tglDrawSegment(m_benderSegment.getP0(), pnt);
drawPoint(pnt, pixelSize);
}
// bender vector
tglColor(TPixel::Red);
tglDrawSegment(m_benderSegment.getP0(), m_benderSegment.getP1());
drawPoint(m_benderSegment.getP0(), pixelSize);
drawPoint(m_benderSegment.getP1(), pixelSize);
// point where is mouse pointer
drawPoint(m_prevPoint, pixelSize);
// arrow in up direction
TPointD vDir = m_benderSegment.getSpeed();
double length2 = norm2(vDir);
if (length2 == 0.0) {
//glPopMatrix();
return;
}
TPointD vUp = 15.0 * normalize(rotate90(vDir));
tglColor(TPixel::Magenta);
TPointD middlePnt = 0.5 * (m_benderSegment.getP0() + m_benderSegment.getP1());
drawArrow(TSegment(middlePnt, m_rotationVersus * vUp + middlePnt), pixelSize);
//glPopMatrix();
}
//-----------------------------------------------------------------------------
void BenderTool::initBenderAction(TVectorImageP &vi, const TPointD &p)
{
// reset counter
// m_buttonDownCounter = 0;
// Versus of bender segment depends from
// point selected to do rotation
// P0 is always center of rotation
if (tdistance2(p, m_benderSegment.getP0()) <
tdistance2(p, m_benderSegment.getP1())) {
TPointD tmpPnt = m_benderSegment.getP1();
m_benderSegment.setP1(m_benderSegment.getP0());
m_benderSegment.setP0(tmpPnt);
}
// fix information about versus
findVersus(p);
// find curves to bender and init data structures
findCurves(vi);
// now it's possible set first position of rotation
// in the same position of extremes
m_benderSegment.setP1(p);
m_prevPoint = p;
}
//-----------------------------------------------------------------------------
} //namespace
//-----------------------------------------------------------------------------
//TTool *getBenderTool() {return &BenderTool;}