// TnzTools includes
#include <tools/replicator.h>
#include <tools/modifiers/modifierclone.h>
// TnzCore includes
#include <tgl.h>
//*****************************************************************************************
// TReplicatorAffine implementation
//*****************************************************************************************
class TReplicatorAffine final : public TReplicator {
Q_DECLARE_TR_FUNCTIONS(TReplicatorAffine)
public:
const TStringId m_idFixScale;
const TStringId m_idFixAspect;
const TStringId m_idFixAngle;
const TStringId m_idFixSkew;
const TStringId m_idCount;
const TStringId m_idCountInv;
const TStringId m_idPressure;
protected:
TAssistantPoint &m_center0;
TAssistantPoint &m_a0;
TAssistantPoint &m_b0;
TAssistantPoint &m_center1;
TAssistantPoint &m_a1;
TAssistantPoint &m_b1;
public:
TReplicatorAffine(TMetaObject &object):
TReplicator(object),
m_idFixScale("fixScale"),
m_idFixAspect("fixAspect"),
m_idFixAngle("fixAngle"),
m_idFixSkew("fixSkew"),
m_idCount("count"),
m_idCountInv("countInv"),
m_idPressure("pressure"),
m_center0( addPoint("center0", TAssistantPoint::CircleCross) ),
m_a0 ( addPoint("a0", TAssistantPoint::CircleFill, TPointD(40, 0)) ),
m_b0 ( addPoint("b0", TAssistantPoint::Circle, TPointD( 0, 40)) ),
m_center1( addPoint("center1", TAssistantPoint::Circle, TPointD(50, 0)) ),
m_a1 ( addPoint("a1", TAssistantPoint::CircleFill, TPointD(90, 0)) ),
m_b1 ( addPoint("b1", TAssistantPoint::Circle, TPointD(50, 40)) )
{
addProperty( new TBoolProperty(m_idFixScale.str(), getFixScale()) );
addProperty( new TBoolProperty(m_idFixAspect.str(), getFixAspect()) );
addProperty( new TBoolProperty(m_idFixAngle.str(), getFixAngle()) );
addProperty( new TBoolProperty(m_idFixSkew.str(), getFixSkew()) );
addProperty( createCountProperty(m_idCount, getCount(), 0) );
addProperty( createCountProperty(m_idCountInv, getCountInv(), 0) );
addProperty( new TDoubleProperty(m_idPressure.str(), 0.0, 2.0, getPressure()) );
}
static QString getLocalName()
{ return tr("Replicator Affine"); }
void updateTranslation() const override {
TReplicator::updateTranslation();
setTranslation(m_idFixScale, tr("Fix Scale"));
setTranslation(m_idFixAspect, tr("Fix Aspect"));
setTranslation(m_idFixAngle, tr("Fix Angle"));
setTranslation(m_idFixSkew, tr("Fix Skew"));
setTranslation(m_idCount, tr("Count"));
setTranslation(m_idCountInv, tr("Inv. Count"));
setTranslation(m_idPressure, tr("Pressure"));
}
inline bool getFixScale() const
{ return data()[m_idFixScale].getBool(); }
inline bool getFixAspect() const
{ return data()[m_idFixAspect].getBool(); }
inline bool getFixAngle() const
{ return data()[m_idFixAngle].getBool(); }
inline bool getFixSkew() const
{ return data()[m_idFixSkew].getBool(); }
inline int getCount() const
{ return (int)data()[m_idCount].getDouble(); }
inline int getCountInv() const
{ return (int)data()[m_idCountInv].getDouble(); }
inline double getPressure() const
{ return data()[m_idPressure].getDouble(); }
protected:
inline void setCount(int x)
{ if (getCount() != (double)x) data()[m_idCount].setDouble((double)x); }
inline void setCountInv(int x)
{ if (getCountInv() != (double)x) data()[m_idCountInv].setDouble((double)x); }
inline void setPressure(double x)
{ if (getPressure() != x) data()[m_idPressure].setDouble(x); }
void onSetDefaults() override {
setCount(1);
setPressure(1);
TReplicator::onSetDefaults();
}
void onFixData() override {
TReplicator::onFixData();
setCount( std::max(0, std::min(multiplierSoftLimit - 1, getCount())) );
setCountInv( std::max(0, std::min(multiplierSoftLimit - 1, getCountInv())) );
setPressure( std::max(0.0, std::min(2.0, getPressure())) );
}
TAffine getAffine(const TAffine &toTool = TAffine()) const {
TPointD c, x, y;
c = toTool*m_center0.position;
x = toTool*m_a0.position - c;
y = toTool*m_b0.position - c;
TAffine t0( x.x, y.x, c.x,
x.y, y.y, c.y );
c = toTool*m_center1.position;
x = toTool*m_a1.position - c;
y = toTool*m_b1.position - c;
TAffine t1( x.x, y.x, c.x,
x.y, y.y, c.y );
return t1*t0.inv();
}
void onFixPoints() override {
TPointD &c0 = m_center0.position;
TPointD &a0 = m_a0.position;
TPointD &b0 = m_b0.position;
TPointD &c1 = m_center1.position;
TPointD &a1 = m_a1.position;
TPointD &b1 = m_b1.position;
if (getFixScale()) {
double la0 = tdistance(a0, c0);
double lb0 = tdistance(b0, c0);
double la1 = tdistance(a1, c1);
double lb1 = tdistance(b1, c1);
a1 = la1 > TConsts::epsilon
? (a1 - c1)*(la0/la1) + c1
: a0 - c0 + c1;
b1 = lb1 > TConsts::epsilon
? (b1 - c1)*(lb0/lb1) + c1
: b0 - c0 + c1;
} else
if (getFixAspect()) {
double la0 = tdistance(a0, c0);
double lb0 = tdistance(b0, c0);
double la1 = tdistance(a1, c1);
double lb1 = tdistance(b1, c1);
if (la0 > TConsts::epsilon)
b1 = lb1 > TConsts::epsilon
? (b1 - c1)*(lb0/lb1*la1/la0) + c1
: (b0 - c0)*(la1/la0) + c1;
}
if (getFixAngle()) {
double la0 = tdistance(a0, c0);
double lb0 = tdistance(b0, c0);
double la1 = tdistance(a1, c1);
double lb1 = tdistance(b1, c1);
if (la0 > TConsts::epsilon)
a1 = (a0 - c0)*(la1/la0) + c1;
if (lb0 > TConsts::epsilon)
b1 = (b0 - c0)*(lb1/lb0) + c1;
} else
if (getFixSkew()) {
TPointD da0 = a0 - c0;
TPointD pa0(-da0.y, da0.x);
double x = (b0 - c0)*da0;
double y = (b0 - c0)*pa0;
TPointD da1 = a1 - c1;
TPointD pa1(-da1.y, da1.x);
TPointD p = da1*x + pa1*y;
double l = norm2(p);
if (l > TConsts::epsilon*TConsts::epsilon)
b1 = p*sqrt(tdistance2(b1, c1)/l) + c1;
}
}
void onMovePoint(TAssistantPoint &point, const TPointD &position) override {
TPointD pc0 = m_center0.position;
TPointD pc1 = m_center1.position;
point.position = position;
if (&point == &m_center0) {
TPointD d = m_center0.position - pc0;
m_a0.position += d;
m_b0.position += d;
m_a1.position += d;
m_b1.position += d;
m_center1.position += d;
} else
if (&point == &m_center1) {
TPointD d = m_center1.position - pc1;
m_a1.position += d;
m_b1.position += d;
}
onFixPoints();
}
void onDataFieldChanged(const TStringId &name, const TVariant &value) override {
TReplicator::onDataFieldChanged(name, value);
if ( name == m_idFixScale
|| name == m_idFixAspect
|| name == m_idFixAngle
|| name == m_idFixSkew )
fixPoints();
}
public:
int getMultipler() const override
{ return getCount() + 1; }
void getModifiers(
const TAffine &toTool,
TInputModifier::List &outModifiers ) const override
{
TAffine aff = getAffine(toTool);
double pressure = getPressure();
double pressureInv = fabs(pressure) > TConsts::epsilon ? 1/pressure : 0;
struct {
int count;
TAffine aff;
double pressure;
} t[] = {
{ getCount(), aff, pressure },
{ getCountInv(), aff.inv(), pressureInv },
};
TModifierClone *modifier = new TModifierClone();
for(int i = 0; i < 2; ++i) {
TTrackTransform tt;
for(int j = 0; j < t[i].count; ++j) {
tt.transform = t[i].aff * tt.transform;
tt.tiltTransform = TTrackTransform::makeTiltTransform(tt.transform);
tt.pressureScale *= t[i].pressure;
modifier->transforms.push_back(tt);
}
}
outModifiers.push_back(modifier);
}
void draw(TToolViewer*, bool enabled) const override {
double alpha = getDrawingAlpha(enabled);
double pixelSize = sqrt(tglGetPixelSize2());
TPointD c = m_center0.position;
TPointD a = m_a0.position;
TPointD b = m_b0.position;
TAffine aff = getAffine();
// draw base
drawSegment(c, a, pixelSize, alpha);
drawSegment(c, b, pixelSize, alpha);
// draw clones
TAffine t;
for(int i = getCount(); i > 0; --i) {
t = aff * t;
drawSegment(t*c, t*a, pixelSize, alpha);
drawSegment(t*c, t*b, pixelSize, alpha);
}
// draw inverted clones
t = TAffine();
aff = aff.inv();
for(int i = getCountInv(); i > 0; --i) {
t = aff * t;
drawSegment(t*c, t*a, pixelSize, alpha);
drawSegment(t*c, t*b, pixelSize, alpha);
}
}
};
//*****************************************************************************************
// Registration
//*****************************************************************************************
static TAssistantTypeT<TReplicatorAffine> replicatorAffine("replicatorAffine");