#include "plastictool.h"
// TnzTools includes
#include "tw/keycodes.h" // Obsolete by now... still currently used though
#include "tooloptionscontrols.h"
#include "tools/toolcommandids.h"
// TnzQt includes
#include "toonzqt/selection.h"
#include "toonzqt/tselectionhandle.h"
#include "toonzqt/dvmimedata.h"
#include "toonzqt/dvdialog.h"
#include "toonzqt/selectioncommandids.h"
// TnzLib includes
#include "toonz/tframehandle.h"
#include "toonz/tcolumnhandle.h"
#include "toonz/txsheethandle.h"
#include "toonz/tobjecthandle.h"
#include "toonz/tonionskinmaskhandle.h"
#include "toonz/tstageobject.h"
#include "toonz/doubleparamcmd.h"
#include "toonz/palettecontroller.h"
#include "toonz/txshsimplelevel.h"
// TnzExt includes
#include "ext/plasticskeleton.h"
#include "ext/plasticdeformerstorage.h"
// TnzCore includes
#include "tgl.h"
#include "tundo.h"
#include "tfunctorinvoker.h"
// Qt includes
#include <QApplication>
#include <QString>
#include <QToolBar>
#include <QPushButton>
#include <QLabel>
#include <QClipboard>
// tcg includes
#include "tcg/tcg_macros.h"
#include "tcg/tcg_point_ops.h"
#include "tcg/tcg_list.h"
#include "tcg/tcg_function_types.h"
#include "tcg/tcg_iterator_ops.h"
//****************************************************************************************
// Local namespace
//****************************************************************************************
namespace {
static const double l_dmax = (std::numeric_limits<double>::max)();
} // namespace
//****************************************************************************************
// PlasticToolLocals namespace
//****************************************************************************************
using namespace PlasticToolLocals;
namespace PlasticToolLocals {
PlasticTool l_plasticTool;
bool l_suspendParamsObservation = false;
//------------------------------------------------------------------------
TPointD projection(const PlasticSkeleton &skeleton, int e, const TPointD &pos) {
const PlasticSkeleton::edge_type &ed = skeleton.edge(e);
const TPointD &p0 = skeleton.vertex(ed.vertex(0)).P();
const TPointD &p1 = skeleton.vertex(ed.vertex(1)).P();
return tcg::point_ops::projection(pos, p0, tcg::point_ops::direction(p0, p1));
}
//------------------------------------------------------------------------
double frame() {
return TTool::getApplication()->getCurrentFrame()->getFrame();
}
//------------------------------------------------------------------------
int row() { return int(frame()) + 1; }
//------------------------------------------------------------------------
int column() {
return TTool::getApplication()->getCurrentColumn()->getColumnIndex();
}
//------------------------------------------------------------------------
void setCell(int row, int col) {
TTool::Application *app = TTool::getApplication();
app->getCurrentFrame()->setCurrentFrame(row);
app->getCurrentColumn()->setColumnIndex(col);
}
//------------------------------------------------------------------------
TXshColumn *xshColumn() {
TXsheet *xsh = TTool::getApplication()->getCurrentXsheet()->getXsheet();
return xsh->getColumn(column());
}
//------------------------------------------------------------------------
TStageObject *stageObject() {
TXsheet *xsh = TTool::getApplication()->getCurrentXsheet()->getXsheet();
return xsh->getStageObject(TStageObjectId::ColumnId(column()));
}
//------------------------------------------------------------------------
const TXshCell &xshCell() {
TXsheet *xsh = TTool::getApplication()->getCurrentXsheet()->getXsheet();
return xsh->getCell(frame(), column());
}
//------------------------------------------------------------------------
int skeletonId() {
TStageObject *obj = stageObject();
const PlasticSkeletonDeformationP &def = obj->getPlasticSkeletonDeformation();
return def ? def->skeletonId(obj->paramsTime(frame()))
: 1; // 1 (not -1) is intended.
} // Means '' (empty string)
//------------------------------------------------------------------------
double sdFrame() { return stageObject()->paramsTime(frame()); }
//------------------------------------------------------------------------
void setKeyframe(TDoubleParamP ¶m, double frame) {
if (!param->isKeyframe(frame)) {
KeyframeSetter setter(param.getPointer(), -1,
false); // Not placing undos through this setter
setter.createKeyframe(frame);
}
}
//------------------------------------------------------------------------
void setKeyframe(SkVD *vd, double frame) {
// vd->setKeyframe(frame); // Nope. In fact...
// Keyframe set is performed with a special tool that is NOT AVAILABLE in
// TnzExt
// (thus, not available to m_sd). It deals with specifying the correct
// interpolation
// type (by user preference, which is TnzLib stuff), etc...
// Traverse vd's parameters. In case they don't have a keyframe at current
// frame, add one.
for (int p = 0; p < SkVD::PARAMS_COUNT; ++p)
setKeyframe(vd->m_params[p], frame);
}
//------------------------------------------------------------------------
void setKeyframe(const PlasticSkeletonDeformationP &sd, double frame) {
// NOTE: The skeleton ids parameter is NOT affected
SkD::vd_iterator vdt, vdEnd;
sd->vertexDeformations(vdt, vdEnd);
for (; vdt != vdEnd; ++vdt) setKeyframe((*vdt).second, frame);
}
//------------------------------------------------------------------------
void invalidateXsheet() {
TTool::getApplication()->getCurrentXsheet()->notifyXsheetChanged();
stageObject()->updateKeyframes();
l_plasticTool.storeDeformation();
l_plasticTool.invalidate();
}
} // namespace PlasticToolLocals
//****************************************************************************************
// Mime definitions
//****************************************************************************************
struct PlasticSkeletonPMime : public DvMimeData {
PlasticSkeletonP m_skeleton;
public:
PlasticSkeletonPMime(const PlasticSkeletonP &skeleton)
: m_skeleton(skeleton) {}
virtual DvMimeData *clone() const {
return new PlasticSkeletonPMime(m_skeleton);
}
virtual void releaseData() { m_skeleton = PlasticSkeletonP(); }
};
struct SkDPMime : public DvMimeData {
SkDP m_sd;
public:
SkDPMime(const SkDP &sd) : m_sd(sd) {}
virtual DvMimeData *clone() const { return new SkDPMime(m_sd); }
virtual void releaseData() { m_sd = SkDP(); }
};
//****************************************************************************************
// Undo definitions
//****************************************************************************************
// NOTE: Some of the following UNDOs have complex and dynamic contents, and I
// don't want to
// trace their size thoroughly. So, I'll follow this guideline: given the
// standard
// 100 MB undos pool, how many undos of one specific type I'd want the
// pool to be able
// to store?
namespace {
class SetVertexNameUndo : public TUndo {
int m_row, m_col; //!< Xsheet coordinates
int m_v; //!< Changed vertex
public:
mutable QString m_oldName, m_newName; //!< Vertex names
mutable SkVD
m_oldVd; //!< Old Vertex deformation (SHARE-OWNED, rather than CLONED)
public:
SetVertexNameUndo(int v, const QString &newName)
: m_row(::row()), m_col(::column()), m_v(v), m_newName(newName) {
const PlasticSkeletonP &skeleton = l_plasticTool.skeleton();
const PlasticSkeletonVertex &vx = skeleton->vertex(v);
m_oldName = vx.name();
}
int getSize() const { return sizeof(*this); } // sizeof this is roughly ok
void redo() const {
PlasticTool::TemporaryActivation tempActivate(m_row, m_col);
// Store the vertex deformation before it's released (possibly destroyed)
{
const SkDP &sd = l_plasticTool.deformation();
TCG_ASSERT(sd, return );
const SkVD *vd = sd->vertexDeformation(m_oldName);
TCG_ASSERT(vd, return );
m_oldVd = *vd;
}
if (m_v >= 0) l_plasticTool.setSkeletonSelection(m_v);
l_plasticTool.setVertexName(m_newName);
::invalidateXsheet();
}
void undo() const {
PlasticTool::TemporaryActivation tempActivate(m_row, m_col);
const SkDP &sd = l_plasticTool.deformation();
TCG_ASSERT(sd, return );
if (m_v >= 0) l_plasticTool.setSkeletonSelection(m_v);
l_plasticTool.setVertexName(m_oldName);
// Restore the vertex deformation.
SkVD *vd = sd->vertexDeformation(m_oldName);
assert(vd);
*vd = m_oldVd;
::invalidateXsheet();
}
};
//========================================================================
class PasteDeformationUndo : public TUndo {
int m_col; //!< Affected column
SkDP m_oldSd, m_newSd; //!< The skeleton deformations
public:
PasteDeformationUndo(const SkDP &newSd)
: m_col(column())
, m_oldSd(stageObject()->getPlasticSkeletonDeformation())
, m_newSd(newSd) {}
int getSize() const { return 1 << 20; }
void redo() const {
TTool::getApplication()->getCurrentColumn()->setColumnIndex(m_col);
stageObject()->setPlasticSkeletonDeformation(m_newSd);
::invalidateXsheet();
}
void undo() const {
TTool::getApplication()->getCurrentColumn()->setColumnIndex(m_col);
stageObject()->setPlasticSkeletonDeformation(m_oldSd);
::invalidateXsheet();
}
};
} // namespace
//****************************************************************************************
// PlasticTool::TemporaryActivation implementation
//****************************************************************************************
PlasticTool::TemporaryActivation::TemporaryActivation(int row, int col)
: m_activate(!l_plasticTool.isActive()) {
if (m_activate) l_plasticTool.onActivate();
::setCell(row, col);
}
//------------------------------------------------------------------------
PlasticTool::TemporaryActivation::~TemporaryActivation() {
if (m_activate) l_plasticTool.onDeactivate();
}
//****************************************************************************************
// PlasticToolOptionsBox::SkelIdComboBox definition
//****************************************************************************************
class PlasticToolOptionsBox::SkelIdsComboBox : public QComboBox {
public:
SkelIdsComboBox(QWidget *parent = 0) : QComboBox(parent) {
updateSkeletonsList();
}
void updateSkeletonsList();
void updateCurrentSkeleton();
};
//------------------------------------------------------------------------
void PlasticToolOptionsBox::SkelIdsComboBox::updateSkeletonsList() {
clear();
const SkDP &sd = l_plasticTool.deformation();
if (!sd) return;
QStringList skeletonsList;
SkD::skelId_iterator st, sEnd;
sd->skeletonIds(st, sEnd);
for (; st != sEnd; ++st) skeletonsList.push_back(QString::number(*st));
QComboBox::insertItems(0, skeletonsList);
updateCurrentSkeleton();
}
//------------------------------------------------------------------------
void PlasticToolOptionsBox::SkelIdsComboBox::updateCurrentSkeleton() {
setCurrentIndex(findText(QString::number(::skeletonId())));
}
//****************************************************************************************
// PlasticToolOptionsBox implementation
//****************************************************************************************
PlasticToolOptionsBox::PlasticToolOptionsBox(QWidget *parent, TTool *tool,
TPaletteHandle *pltHandle)
: GenericToolOptionsBox(parent, tool, pltHandle, PlasticTool::MODES_COUNT)
, m_tool(tool)
, m_subToolbars(new GenericToolOptionsBox *[PlasticTool::MODES_COUNT])
//, m_subToolbarActions(new QAction*[PlasticTool::MODES_COUNT])
{
struct locals {
static inline QWidget *newSpace(QWidget *parent = 0) {
QWidget *space = new QWidget(parent);
space->setFixedWidth(TOOL_OPTIONS_LEFT_MARGIN);
return space;
}
};
// Create Mesh button
QPushButton *meshifyButton = new QPushButton(tr("Create Mesh"));
// Add skeleton id-related widgets
QLabel *skelIdLabel = new QLabel(tr("Skeleton:"));
m_skelIdComboBox = new SkelIdsComboBox;
m_addSkelButton = new QPushButton("+"); // Connected in the show event
m_removeSkelButton = new QPushButton("-"); // Connected in the show event
// Add sub-options for each mode group
for (int m = 0; m != PlasticTool::MODES_COUNT; ++m)
m_subToolbars[m] = new GenericToolOptionsBox(0, tool, pltHandle, m);
meshifyButton->setFixedHeight(20);
QAction *meshifyAction =
CommandManager::instance()->getAction("A_ToolOption_Meshify");
meshifyButton->addAction(meshifyAction);
skelIdLabel->setFixedHeight(20);
m_skelIdComboBox->setFixedWidth(50);
m_addSkelButton->setFixedSize(20, 20);
m_removeSkelButton->setFixedSize(20, 20);
for (int m = 0; m != PlasticTool::MODES_COUNT; ++m)
m_subToolbars[m]->setContentsMargins(0, 0, 0, 0);
/*- Layout -*/
// Add created widgets to the toolbar (in reverse order since we're inserting
// at 0)
m_layout->insertWidget(0, m_removeSkelButton);
m_layout->insertWidget(0, m_addSkelButton);
m_layout->insertWidget(0, m_skelIdComboBox);
m_layout->insertWidget(0, skelIdLabel);
m_layout->insertWidget(0, locals::newSpace(this));
m_layout->insertWidget(0, meshifyButton);
m_layout->insertWidget(0, locals::newSpace(this));
for (int m = 0; m != PlasticTool::MODES_COUNT; ++m)
m_layout->insertWidget(m_layout->count() - 1, m_subToolbars[m], 1);
bool ret = true;
ret = ret && connect(meshifyButton, SIGNAL(clicked()), meshifyAction,
SLOT(trigger()));
assert(ret);
// Add Animation mode fields corresponding to vertex properties
GenericToolOptionsBox *animateOptionsBox =
m_subToolbars[PlasticTool::ANIMATE_IDX];
// Adjust some specific controls first
{
ToolOptionTextField *minAngleField = static_cast<ToolOptionTextField *>(
animateOptionsBox->control("minAngle"));
assert(minAngleField);
minAngleField->setFixedWidth(40);
ToolOptionTextField *maxAngleField = static_cast<ToolOptionTextField *>(
animateOptionsBox->control("maxAngle"));
assert(maxAngleField);
maxAngleField->setFixedWidth(40);
}
// Distance
ToolOptionParamRelayField *distanceField = new ToolOptionParamRelayField(
&l_plasticTool, &l_plasticTool.m_distanceRelay);
distanceField->setGlobalKey(&l_plasticTool.m_globalKey,
&l_plasticTool.m_relayGroup);
QLabel *distanceLabel = new QLabel(tr("Distance"));
distanceLabel->setFixedHeight(20);
// Angle
ToolOptionParamRelayField *angleField = new ToolOptionParamRelayField(
&l_plasticTool, &l_plasticTool.m_angleRelay);
angleField->setGlobalKey(&l_plasticTool.m_globalKey,
&l_plasticTool.m_relayGroup);
QLabel *angleLabel = new QLabel(tr("Angle"));
angleLabel->setFixedHeight(20);
// SO
ToolOptionParamRelayField *soField =
new ToolOptionParamRelayField(&l_plasticTool, &l_plasticTool.m_soRelay);
soField->setGlobalKey(&l_plasticTool.m_globalKey,
&l_plasticTool.m_relayGroup);
QLabel *soLabel = new QLabel(tr("SO"));
soLabel->setFixedHeight(20);
QHBoxLayout *animateLayout = animateOptionsBox->hLayout();
animateLayout->insertWidget(0, soField);
animateLayout->insertWidget(0, soLabel);
animateLayout->insertWidget(0, angleField);
animateLayout->insertWidget(0, angleLabel);
animateLayout->insertWidget(0, distanceField);
animateLayout->insertWidget(0, distanceLabel);
onPropertyChanged();
}
//------------------------------------------------------------------------
void PlasticToolOptionsBox::showEvent(QShowEvent *se) {
bool ret = true;
ret = ret && connect(&l_plasticTool, SIGNAL(skelIdsListChanged()),
SLOT(onSkelIdsListChanged()));
ret = ret && connect(&l_plasticTool, SIGNAL(skelIdChanged()),
SLOT(onSkelIdChanged()));
ret = ret && connect(m_skelIdComboBox, SIGNAL(activated(int)),
SLOT(onSkelIdEdited()));
ret = ret &&
connect(m_addSkelButton, SIGNAL(released()), SLOT(onAddSkeleton()));
ret = ret && connect(m_removeSkelButton, SIGNAL(released()),
SLOT(onRemoveSkeleton()));
assert(ret);
m_skelIdComboBox->updateSkeletonsList();
}
//------------------------------------------------------------------------
void PlasticToolOptionsBox::hideEvent(QHideEvent *he) {
disconnect(&l_plasticTool, 0, this, 0);
disconnect(m_skelIdComboBox, 0, this, 0);
disconnect(m_addSkelButton, 0, this, 0);
disconnect(m_removeSkelButton, 0, this, 0);
}
//------------------------------------------------------------------------
void PlasticToolOptionsBox::onPropertyChanged() {
// Fetch current mode index
TPropertyGroup *pGroup = m_tool->getProperties(PlasticTool::MODES_COUNT);
assert(pGroup);
TEnumProperty *prop =
dynamic_cast<TEnumProperty *>(pGroup->getProperty("mode"));
assert(prop);
int mode = prop->getIndex();
// Show the specified mode options, hide all the others
for (int m = 0; m != PlasticTool::MODES_COUNT; ++m)
m_subToolbars[m]->setVisible(m == mode);
}
//------------------------------------------------------------------------
void PlasticToolOptionsBox::onSkelIdsListChanged() {
m_skelIdComboBox->updateSkeletonsList();
}
//------------------------------------------------------------------------
void PlasticToolOptionsBox::onSkelIdChanged() {
m_skelIdComboBox->updateCurrentSkeleton();
}
//------------------------------------------------------------------------
void PlasticToolOptionsBox::onSkelIdEdited() {
int skelId = m_skelIdComboBox->currentText().toInt();
if (skelId == ::skeletonId()) return;
if (!l_plasticTool.deformation()) return;
l_plasticTool.editSkelId_undo(skelId);
}
//------------------------------------------------------------------------
void PlasticToolOptionsBox::onAddSkeleton() {
if (l_plasticTool.isEnabled())
l_plasticTool.addSkeleton_undo(new PlasticSkeleton);
}
//------------------------------------------------------------------------
void PlasticToolOptionsBox::onRemoveSkeleton() {
if (l_plasticTool.isEnabled() && l_plasticTool.deformation())
l_plasticTool.removeSkeleton_withKeyframes_undo(::skeletonId());
}
//****************************************************************************************
// PlasticTool implementation
//****************************************************************************************
PlasticTool::PlasticTool()
: TTool(T_Plastic)
, m_skelId(-(std::numeric_limits<int>::max)())
, m_propGroup(new TPropertyGroup[MODES_COUNT + 1])
, m_mode("mode")
, m_vertexName("vertexName", L"")
, m_interpolate("interpolate", false)
, m_snapToMesh("snapToMesh", false)
, m_thickness("Thickness", 1, 100, 5)
, m_rigidValue("rigidValue")
, m_globalKey("globalKeyframe", true)
, m_keepDistance("keepDistance", true)
, m_minAngle("minAngle", L"")
, m_maxAngle("maxAngle", L"")
, m_distanceRelay("distanceRelay")
, m_angleRelay("angleRelay")
, m_soRelay("soRelay")
, m_skelIdRelay("skelIdRelay")
, m_pressedPos(TConsts::napd)
, m_dragged(false)
, m_svHigh(-1)
, m_seHigh(-1)
, m_mvHigh(-1)
, m_meHigh(-1)
, m_rigidityPainter(createRigidityPainter())
, m_showSkeletonOS(true)
, m_recompileOnMouseRelease(false) {
// And now, a little trick about tool binding
bind(TTool::AllImages); // Attach the tool to all types :)
bind(TTool::MeshLevels); // But disable it for all but meshes :0
// This little trick is needed to associate the tool to common levels (the
// toolbar must appear), in
// order to make them meshable.
// Bind properties to the appropriate property group (needed by the automatic
// toolbar builder)
m_propGroup[MODES_COUNT].bind(m_mode);
m_propGroup[MODES_COUNT].bind(m_vertexName);
m_propGroup[RIGIDITY_IDX].bind(m_thickness);
m_propGroup[RIGIDITY_IDX].bind(m_rigidValue);
m_propGroup[BUILD_IDX].bind(m_interpolate);
m_propGroup[BUILD_IDX].bind(m_snapToMesh);
m_propGroup[ANIMATE_IDX].bind(m_globalKey);
m_propGroup[ANIMATE_IDX].bind(m_keepDistance);
m_propGroup[ANIMATE_IDX].bind(m_minAngle);
m_propGroup[ANIMATE_IDX].bind(m_maxAngle);
m_relayGroup.bind(m_distanceRelay);
m_relayGroup.bind(m_angleRelay);
m_relayGroup.bind(m_soRelay);
m_mode.setId("SkeletonMode");
m_vertexName.setId("VertexName");
m_interpolate.setId("Interpolate");
m_snapToMesh.setId("SnapToMesh");
m_thickness.setId("Thickness");
m_rigidValue.setId("RigidValue");
m_globalKey.setId("GlobalKey");
m_keepDistance.setId("KeepDistance");
m_minAngle.setId("MinAngle");
m_maxAngle.setId("MaxAngle");
m_distanceRelay.setId("DistanceRelay");
m_angleRelay.setId("AngleRelay");
m_soRelay.setId("SoRelay");
m_skelIdRelay.setId("SkelIdRelay");
// Attach to selections
m_svSel.setView(this);
m_mvSel.setView(this), m_meSel.setView(this);
}
//------------------------------------------------------------------------
PlasticTool::~PlasticTool() {
if (m_sd) m_sd->removeObserver(this);
}
//------------------------------------------------------------------------
TTool::ToolType PlasticTool::getToolType() const {
switch (m_mode.getIndex()) {
case MESH_IDX:
case RIGIDITY_IDX:
return TTool::LevelWriteTool;
case BUILD_IDX:
case ANIMATE_IDX:
return TTool::ColumnTool;
}
assert(false);
return TTool::GenericTool;
}
//------------------------------------------------------------------------
void PlasticTool::updateTranslation() {
m_mode.setQStringName(tr("Mode:"));
m_mode.deleteAllValues();
m_mode.addValue(tr("Edit Mesh").toStdWString());
m_mode.addValue(tr("Paint Rigid").toStdWString());
m_mode.addValue(tr("Build Skeleton").toStdWString());
m_mode.addValue(tr("Animate").toStdWString());
m_mode.setIndex(BUILD_IDX);
m_vertexName.setQStringName(tr("Vertex Name:"));
m_interpolate.setQStringName(tr("Allow Stretching"));
m_snapToMesh.setQStringName(tr("Snap To Mesh"));
m_thickness.setQStringName(tr("Thickness"));
m_rigidValue.setQStringName("");
m_rigidValue.deleteAllValues();
m_rigidValue.addValue(tr("Rigid").toStdWString());
m_rigidValue.addValue(tr("Flex").toStdWString());
m_globalKey.setQStringName(tr("Global Key"));
m_keepDistance.setQStringName(tr("Keep Distance"));
m_minAngle.setQStringName(tr("Angle Bounds"));
m_maxAngle.setQStringName("");
}
//-------------------------------------------------------------------------------------------------------
ToolOptionsBox *PlasticTool::createOptionsBox() {
// Create the options box
TPaletteHandle *currPalette =
TTool::getApplication()->getPaletteController()->getCurrentLevelPalette();
PlasticToolOptionsBox *optionsBox =
new PlasticToolOptionsBox(0, this, currPalette);
// Connect it to receive m_mode notifications
m_mode.addListener(optionsBox);
return optionsBox;
}
//------------------------------------------------------------------------
PlasticSkeletonP PlasticTool::skeleton() const {
return m_sd ? m_sd->skeleton(::sdFrame()) : PlasticSkeletonP();
}
//------------------------------------------------------------------------
PlasticSkeleton &PlasticTool::deformedSkeleton() {
typedef tcg::function<void (PlasticTool::*)(PlasticSkeleton &),
&PlasticTool::updateDeformedSkeleton>
Func;
return m_deformedSkeleton(tcg::bind1st(Func(), *this));
}
//------------------------------------------------------------------------
void PlasticTool::touchSkeleton() {
touchDeformation();
int skelId = ::skeletonId();
if (!m_sd->skeleton(skelId)) {
m_sd->attach(skelId, new PlasticSkeleton);
emit skelIdsListChanged();
}
}
//------------------------------------------------------------------------
void PlasticTool::touchDeformation() {
if (m_sd) return;
// Store a new deformation in the column's stage object
stageObject()->setPlasticSkeletonDeformation(new PlasticSkeletonDeformation);
storeDeformation(); // Builds the deformed skeleton too
}
//------------------------------------------------------------------------
void PlasticTool::storeDeformation() {
const SkDP &sd = stageObject()->getPlasticSkeletonDeformation();
if (m_sd != sd) {
clearSkeletonSelections();
if (m_sd) {
m_sd->removeObserver(this);
m_skelIdRelay.setParam(TDoubleParamP());
}
// Store the deformation, retrieving it from current column stage object
m_sd = sd;
if (m_sd) {
m_sd->addObserver(this);
m_skelIdRelay.setParam(m_sd->skeletonIdsParam());
}
m_skelIdRelay.notifyListeners();
}
storeSkeletonId();
if (m_mode.getIndex() == ANIMATE_IDX)
m_deformedSkeleton.invalidate(); // Store the deformed skeleton too
emit skelIdsListChanged();
}
//------------------------------------------------------------------------
void PlasticTool::storeSkeletonId() {
int skelId = m_sd ? m_sd->skeletonIdsParam()->getValue(::sdFrame())
: -(std::numeric_limits<int>::max)();
if (m_skelId != skelId) {
m_skelId = skelId;
clearSkeletonSelections();
emit skelIdChanged();
}
}
//------------------------------------------------------------------------
void PlasticTool::updateDeformedSkeleton(PlasticSkeleton &deformedSkeleton) {
if (m_sd)
m_sd->storeDeformedSkeleton(::skeletonId(), ::sdFrame(), deformedSkeleton);
else
deformedSkeleton.clear();
}
//------------------------------------------------------------------------
void PlasticTool::onFrameSwitched() {
storeSkeletonId();
storeMeshImage();
switch (m_mode.getIndex()) {
case ANIMATE_IDX:
m_deformedSkeleton.invalidate();
}
// Update the relays' current frame
double frame = ::sdFrame();
m_distanceRelay.frame() = frame;
m_angleRelay.frame() = frame;
m_soRelay.frame() = frame;
m_skelIdRelay.frame() = frame;
m_distanceRelay.notifyListeners();
m_angleRelay.notifyListeners();
m_soRelay.notifyListeners();
m_skelIdRelay.notifyListeners();
}
//------------------------------------------------------------------------
void PlasticTool::onColumnSwitched() {
switch (m_mode.getIndex()) {
case MESH_IDX:
case BUILD_IDX:
case RIGIDITY_IDX:
m_pvs.m_showOriginalColumn = xshColumn();
}
storeDeformation();
storeMeshImage();
}
//------------------------------------------------------------------------
void PlasticTool::onXsheetChanged() {
onColumnSwitched();
TTool::updateEnabled(); // Current cell may no longer be a mesh one (or
// viceversa),
} // so tool enabled status must be updated.
//------------------------------------------------------------------------
void PlasticTool::onChange() {
// Since parameters are typically coupled, we could pass multiple, consecutive
// times in
// this notification function. We have to employ counter-measures to prevent
// multiple
// calls from affecting performance.
static bool refresh = false; // Accessible from locals since static
struct locals {
struct RefreshFunctor : public TFunctorInvoker::BaseFunctor {
void operator()() {
refresh = false;
l_plasticTool.storeSkeletonId(); // Calls ::sdFrame()
// This is needed to repaint the xsheet (not automatic otherwise)
TTool::getApplication()->getCurrentObject()->notifyObjectIdChanged(
false);
}
};
}; // locals
// Using invalidate/update and delayed invocation to prevent multiple calls to
// ::sdFrame()
m_deformedSkeleton.invalidate();
if (!refresh) {
refresh = true;
QMetaObject::invokeMethod(TFunctorInvoker::instance(), "invoke",
Qt::QueuedConnection,
Q_ARG(void *, new locals::RefreshFunctor));
}
// Passing through Qt's event system to compress repaints in a single one
TTool::Viewer *viewer = getViewer();
if (viewer) // This goes through paintEvent(),
viewer->invalidateAll(); // \a unlike TTool::invalidate()
}
//------------------------------------------------------------------------
void PlasticTool::onChange(const TParamChange &pc) {
if (l_suspendParamsObservation) return;
onChange();
}
//------------------------------------------------------------------------
void PlasticTool::onSetViewer() {
Viewer *viewer = getViewer();
if (viewer) {
PlasticVisualSettings &pvs =
viewer->visualSettings().m_plasticVisualSettings;
pvs = m_pvs;
// Force options if needed
if (m_mode.getIndex() == RIGIDITY_IDX) pvs.m_drawRigidity = true;
}
}
//------------------------------------------------------------------------
void PlasticTool::onActivate() {
bool ret;
ret = connect(TTool::m_application->getCurrentFrame(),
SIGNAL(frameSwitched()), this, SLOT(onFrameSwitched())),
assert(ret);
ret = connect(TTool::m_application->getCurrentColumn(),
SIGNAL(columnIndexSwitched()), this, SLOT(onColumnSwitched())),
assert(ret);
ret = connect(TTool::m_application->getCurrentXsheet(),
SIGNAL(xsheetChanged()), this, SLOT(onXsheetChanged())),
assert(ret);
ret = connect(TTool::m_application->getCurrentXsheet(),
SIGNAL(xsheetSwitched()), this, SLOT(onXsheetChanged())),
assert(ret);
onSetViewer();
onColumnSwitched();
onFrameSwitched();
setActive(true);
}
//------------------------------------------------------------------------
void PlasticTool::onDeactivate() {
setActive(false);
bool ret;
ret = disconnect(TTool::m_application->getCurrentFrame(),
SIGNAL(frameSwitched()), this, SLOT(onFrameSwitched())),
assert(ret);
ret =
disconnect(TTool::m_application->getCurrentColumn(),
SIGNAL(columnIndexSwitched()), this, SLOT(onColumnSwitched())),
assert(ret);
ret = disconnect(TTool::m_application->getCurrentXsheet(),
SIGNAL(xsheetChanged()), this, SLOT(onXsheetChanged())),
assert(ret);
ret = disconnect(TTool::m_application->getCurrentXsheet(),
SIGNAL(xsheetSwitched()), this, SLOT(onXsheetChanged())),
assert(ret);
Viewer *viewer = getViewer();
if (viewer)
viewer->visualSettings().m_plasticVisualSettings = PlasticVisualSettings();
m_sd = PlasticSkeletonDeformationP();
}
//------------------------------------------------------------------------
void PlasticTool::onEnter() {}
//------------------------------------------------------------------------
void PlasticTool::onLeave() {
// Clear visualization vars
m_pos = TConsts::napd;
m_svHigh = m_seHigh = -1;
m_mvHigh = m_meHigh = MeshIndex();
}
//------------------------------------------------------------------------
void PlasticTool::onSelectionChanged() {
SkVD *vd = 0;
if (m_sd && m_svSel.hasSingleObject()) {
int skelId = ::skeletonId();
const PlasticSkeleton::vertex_type &vx =
m_sd->skeleton(skelId)->vertex(m_svSel);
m_vertexName.setValue(vx.name().toStdWString());
m_interpolate.setValue(vx.m_interpolate);
m_minAngle.setValue((vx.m_minAngle == -l_dmax)
? L""
: QString::number(vx.m_minAngle).toStdWString());
m_maxAngle.setValue((vx.m_maxAngle == l_dmax)
? L""
: QString::number(vx.m_maxAngle).toStdWString());
vd = m_sd->vertexDeformation(skelId, m_svSel);
} else {
m_vertexName.setValue(L"");
m_interpolate.setValue(false);
m_minAngle.setValue(L"");
m_maxAngle.setValue(L"");
}
// Attach or detach relays depending on selected vertex's parameters
m_soRelay.setParam(vd ? vd->m_params[SkVD::SO] : TDoubleParamP());
if (vd && m_svSel.hasSingleObject() && m_svSel.objects().front() > 0) {
m_distanceRelay.setParam(vd->m_params[SkVD::DISTANCE]);
m_angleRelay.setParam(vd->m_params[SkVD::ANGLE]);
} else {
m_distanceRelay.setParam(TDoubleParamP());
m_angleRelay.setParam(TDoubleParamP());
}
m_vertexName.notifyListeners();
m_interpolate.notifyListeners();
m_minAngle.notifyListeners();
m_maxAngle.notifyListeners();
m_distanceRelay.notifyListeners();
m_angleRelay.notifyListeners();
m_soRelay.notifyListeners();
}
//------------------------------------------------------------------------
void PlasticTool::enableCommands() {
if (TSelection::getCurrent() == &m_svSel)
m_svSel.enableCommand(this, MI_Clear,
&PlasticTool::deleteSelectedVertex_undo);
else if (TSelection::getCurrent() == &m_meSel) {
m_meSel.enableCommand(this, MI_Clear, &PlasticTool::collapseEdge_mesh_undo);
m_meSel.enableCommand(this, MI_Insert, &PlasticTool::splitEdge_mesh_undo);
}
}
//------------------------------------------------------------------------
void PlasticTool::setSkeletonSelection(const PlasticVertexSelection &vSel) {
if (vSel.isEmpty()) {
m_svSel.selectNone();
m_svSel.makeNotCurrent();
return;
}
assert(m_sd);
m_svSel.skeletonId() = m_skelId;
m_svSel.setObjects(vSel.objects());
m_svSel.notifyView();
m_svSel.makeCurrent();
// Okay, the following is cheap - we have to update the Function Editor
// (specifically)
// since current vertex is shown in a special color. We know that the same
// happens for
// the current stage object, so... we'll attach there.
TTool::getApplication()->getCurrentObject()->notifyObjectIdChanged(
false); // Carry on, you've seen nothing ;)
}
//------------------------------------------------------------------------
void PlasticTool::toggleSkeletonSelection(
const PlasticVertexSelection &addition) {
const std::vector<int> &storedIdxs = m_svSel.objects();
const std::vector<int> &addedIdxs = addition.objects();
// Build new selection
std::vector<int> selectedIdxs;
if (m_svSel.contains(addition)) {
std::set_difference(storedIdxs.begin(), storedIdxs.end(), addedIdxs.begin(),
addedIdxs.end(), std::back_inserter(selectedIdxs));
} else {
std::set_union(storedIdxs.begin(), storedIdxs.end(), addedIdxs.begin(),
addedIdxs.end(), std::back_inserter(selectedIdxs));
}
setSkeletonSelection(selectedIdxs);
}
//------------------------------------------------------------------------
void PlasticTool::clearSkeletonSelections() {
m_svHigh = m_seHigh = -1;
m_svSel.selectNone();
m_svSel.makeNotCurrent();
}
//------------------------------------------------------------------------
PlasticVertexSelection PlasticTool::branchSelection(int vIdx) const {
struct locals {
static void addBranch(const PlasticSkeleton &skeleton, int v,
std::vector<int> &branch) {
branch.push_back(v);
const PlasticSkeletonVertex &vx = skeleton.vertex(v);
PlasticSkeletonVertex::edges_const_iterator et, eEnd = vx.edgesEnd();
for (et = vx.edgesBegin(); et != eEnd; ++et) {
int child = skeleton.edge(*et).vertex(1);
if (v != child) // The edge to parent is in the list
addBranch(skeleton, child,
branch); // I wonder if it's ensured to be always at begin?
}
}
};
assert(skeleton());
std::vector<int> selectedIdxs;
locals::addBranch(*skeleton(), vIdx, selectedIdxs);
return selectedIdxs;
}
//------------------------------------------------------------------------
void PlasticTool::copySkeleton() {
if (!m_sd) return;
const PlasticSkeletonP &skel = m_sd->skeleton(::skeletonId());
if (!skel) return;
// Copy a CLONE of currently addressed skeleton in the app clipboard
QMimeData *data = new PlasticSkeletonPMime(new PlasticSkeleton(*skel));
QApplication::clipboard()->setMimeData(data, QClipboard::Clipboard);
}
//------------------------------------------------------------------------
void PlasticTool::pasteSkeleton_undo() {
const PlasticSkeletonPMime *data = dynamic_cast<const PlasticSkeletonPMime *>(
QApplication::clipboard()->mimeData());
if (!data) return;
PlasticSkeletonP newSkeleton(new PlasticSkeleton(*data->m_skeleton));
touchDeformation();
assert(m_sd);
int skelId = ::skeletonId();
const PlasticSkeletonP &oldSkel = m_sd->skeleton(skelId);
if (oldSkel && !oldSkel->empty()) {
// In case there exists a not-empty skeleton, add a NEW skeleton
addSkeleton_undo(newSkeleton);
} else {
TUndoManager *manager = TUndoManager::manager();
manager->beginBlock();
{
removeSkeleton_undo(skelId); // No problem - it's eventually empty
addSkeleton_undo(skelId, newSkeleton.getPointer());
}
manager->endBlock();
}
}
//------------------------------------------------------------------------
void PlasticTool::copyDeformation() {
if (!m_sd) return;
// Copy a reference to currently addressed skeleton in the app clipboard
QMimeData *data = new SkDPMime(m_sd);
QApplication::clipboard()->setMimeData(data, QClipboard::Clipboard);
}
//------------------------------------------------------------------------
void PlasticTool::pasteDeformation_undo() {
const SkDPMime *data =
dynamic_cast<const SkDPMime *>(QApplication::clipboard()->mimeData());
if (!data) return;
// Given a skeleton, attempt to assign it to the current stage object
TStageObject *obj = ::stageObject();
assert(obj);
const PlasticSkeletonDeformationP &oldSd =
obj->getPlasticSkeletonDeformation();
if (oldSd) {
// A skeleton already exists. Ask the user if it has to be replaced.
bool replace = DVGui::MsgBox(tr("A group of skeletons already exists for "
"current column. Replacing it will also "
"substitute any existing vertex "
"animation.\n\nDo you want to continue?"),
tr("Ok"), tr("Cancel")) == 1;
if (!replace) return;
}
// Clone the whole skeleton deformation (skeleton itself included)
SkDP newSd(new PlasticSkeletonDeformation(*data->m_sd));
// Insert the undo and perform the op
TUndoManager::manager()->add(new PasteDeformationUndo(newSd));
obj->setPlasticSkeletonDeformation(newSd);
::invalidateXsheet();
}
//------------------------------------------------------------------------
void PlasticTool::setKey() {
assert(m_svSel.hasSingleObject());
SkVD *vd = m_sd->vertexDeformation(::skeletonId(), m_svSel);
double frame = ::frame();
if (vd->isFullKeyframe(frame))
vd->deleteKeyframe(frame);
else
::setKeyframe(vd, frame);
}
//------------------------------------------------------------------------
void PlasticTool::setGlobalKey() {
struct locals {
inline static bool isFullKeyframe(const SkDP &sd, double frame) {
SkD::vd_iterator vdt, vdEnd;
sd->vertexDeformations(vdt, vdEnd);
for (; vdt != vdEnd; ++vdt)
if (!(*vdt).second->isFullKeyframe(frame)) return false;
return true;
}
};
double frame = ::frame();
if (locals::isFullKeyframe(m_sd, frame))
m_sd->deleteKeyframe(frame);
else
::setKeyframe(m_sd, frame);
}
//------------------------------------------------------------------------
void PlasticTool::setRestKey() {
assert(m_svSel.hasSingleObject());
SkVD *vd = m_sd->vertexDeformation(::skeletonId(), m_svSel);
double frame = ::frame();
for (int p = 0; p != SkVD::PARAMS_COUNT; ++p)
vd->m_params[p]->setValue(frame, vd->m_params[p]->getDefaultValue());
}
//------------------------------------------------------------------------
void PlasticTool::setGlobalRestKey() {
double frame = ::frame();
SkD::vd_iterator vdt, vdEnd;
m_sd->vertexDeformations(vdt, vdEnd);
for (; vdt != vdEnd; ++vdt) {
SkVD *vd = (*vdt).second;
for (int p = 0; p != SkVD::PARAMS_COUNT; ++p)
vd->m_params[p]->setValue(frame, vd->m_params[p]->getDefaultValue());
}
}
//------------------------------------------------------------------------
void PlasticTool::setKey_undo() { keyFunc_undo(&PlasticTool::setKey); }
//------------------------------------------------------------------------
void PlasticTool::setGlobalKey_undo() {
keyFunc_undo(&PlasticTool::setGlobalKey);
}
//------------------------------------------------------------------------
void PlasticTool::setRestKey_undo() { keyFunc_undo(&PlasticTool::setRestKey); }
//------------------------------------------------------------------------
void PlasticTool::setGlobalRestKey_undo() {
keyFunc_undo(&PlasticTool::setGlobalRestKey);
}
//------------------------------------------------------------------------
void PlasticTool::setVertexName(QString &name) {
const PlasticSkeletonP &skeleton = this->skeleton();
assert(skeleton && m_svSel.hasSingleObject() && m_svSel > 0);
// Update the selected vertex's name
while (!m_sd->skeleton(::skeletonId())->setVertexName(m_svSel, name))
name += "_";
m_vertexName.setValue(name.toStdWString());
m_vertexName.notifyListeners(); // NOTE: This should NOT invoke this function
// recursively
// Re-store the deformed skeleton. This is necessary since any follow-up
// vertex
// manipulation must refer the correct vd name.
m_deformedSkeleton.invalidate();
PlasticDeformerStorage::instance()->invalidateSkeleton(
m_sd.getPointer(), ::skeletonId(), PlasticDeformerStorage::NONE);
}
//------------------------------------------------------------------------
void PlasticTool::mouseMove(const TPointD &pos, const TMouseEvent &me) {
// Discriminate mode
switch (m_mode.getIndex()) {
case MESH_IDX:
mouseMove_mesh(pos, me);
break;
case BUILD_IDX:
mouseMove_build(pos, me);
break;
case RIGIDITY_IDX:
mouseMove_rigidity(pos, me);
break;
case ANIMATE_IDX:
mouseMove_animate(pos, me);
break;
}
}
//------------------------------------------------------------------------
void PlasticTool::leftButtonDown(const TPointD &pos, const TMouseEvent &me) {
switch (m_mode.getIndex()) {
case MESH_IDX:
leftButtonDown_mesh(pos, me);
break;
case BUILD_IDX:
leftButtonDown_build(pos, me);
break;
case RIGIDITY_IDX:
leftButtonDown_rigidity(pos, me);
break;
case ANIMATE_IDX:
leftButtonDown_animate(pos, me);
break;
}
}
//------------------------------------------------------------------------
void PlasticTool::leftButtonDrag(const TPointD &pos, const TMouseEvent &me) {
// Track dragging status
m_dragged = true;
switch (m_mode.getIndex()) {
case MESH_IDX:
leftButtonDrag_mesh(pos, me);
break;
case BUILD_IDX:
leftButtonDrag_build(pos, me);
break;
case RIGIDITY_IDX:
leftButtonDrag_rigidity(pos, me);
break;
case ANIMATE_IDX:
leftButtonDrag_animate(pos, me);
break;
}
}
//------------------------------------------------------------------------
void PlasticTool::leftButtonUp(const TPointD &pos, const TMouseEvent &me) {
switch (m_mode.getIndex()) {
case MESH_IDX:
leftButtonUp_mesh(pos, me);
break;
case BUILD_IDX:
leftButtonUp_build(pos, me);
break;
case RIGIDITY_IDX:
leftButtonUp_rigidity(pos, me);
break;
case ANIMATE_IDX:
leftButtonUp_animate(pos, me);
break;
}
m_pressedPos = TConsts::napd;
m_pressedVxsPos.clear();
m_dragged = false;
}
//------------------------------------------------------------------------
void PlasticTool::addContextMenuItems(QMenu *menu) {
bool ret = true;
// Add global actions
if (m_sd && m_sd->skeleton(::skeletonId())) {
QAction *copySkeleton = menu->addAction(tr("Copy Skeleton"));
ret = ret && connect(copySkeleton, SIGNAL(triggered()), &l_plasticTool,
SLOT(copySkeleton()));
}
if (dynamic_cast<const PlasticSkeletonPMime *>(
QApplication::clipboard()->mimeData())) {
QAction *pasteSkeleton = menu->addAction(tr("Paste Skeleton"));
ret = ret && connect(pasteSkeleton, SIGNAL(triggered()), &l_plasticTool,
SLOT(pasteSkeleton_undo()));
}
menu->addSeparator(); // Separate actions type
// Add editing actions
switch (m_mode.getIndex()) {
case MESH_IDX:
addContextMenuActions_mesh(menu);
break;
case BUILD_IDX:
addContextMenuActions_build(menu);
break;
case RIGIDITY_IDX:
addContextMenuActions_rigidity(menu);
break;
case ANIMATE_IDX:
addContextMenuActions_animate(menu);
break;
}
// Add view actions
QAction *showMesh = menu->addAction(tr("Show Mesh"));
showMesh->setCheckable(true);
showMesh->setChecked(m_pvs.m_drawMeshesWireframe);
ret = ret && connect(showMesh, SIGNAL(triggered(bool)), &l_plasticTool,
SLOT(onShowMeshToggled(bool)));
QAction *showRigidity = menu->addAction(tr("Show Rigidity"));
showRigidity->setCheckable(true);
showRigidity->setChecked(m_pvs.m_drawRigidity);
ret = ret && connect(showRigidity, SIGNAL(triggered(bool)), &l_plasticTool,
SLOT(onShowRigidityToggled(bool)));
QAction *showSO = menu->addAction(tr("Show SO"));
showSO->setCheckable(true);
showSO->setChecked(m_pvs.m_drawSO);
ret = ret && connect(showSO, SIGNAL(triggered(bool)), &l_plasticTool,
SLOT(onShowSOToggled(bool)));
QAction *showSkeletonOS = menu->addAction(tr("Show Skeleton Onion Skin"));
showSkeletonOS->setCheckable(true);
showSkeletonOS->setChecked(m_showSkeletonOS);
ret = ret && connect(showSkeletonOS, SIGNAL(triggered(bool)), &l_plasticTool,
SLOT(onShowSkelOSToggled(bool)));
assert(ret);
menu->addSeparator(); // Separate from common view options
}
//------------------------------------------------------------------------
void PlasticTool::reset() {
// NOTE: This is an inherited virtual. Please leave it even if it's empty.
}
//------------------------------------------------------------------------
bool PlasticTool::onPropertyChanged(std::string propertyName) {
struct locals {
static bool alreadyContainsVertexName(const PlasticSkeleton &skel,
const QString &name) {
tcg::list<PlasticSkeletonVertex>::const_iterator vt,
vEnd(skel.vertices().end());
for (vt = skel.vertices().begin(); vt != vEnd; ++vt)
if (vt->name() == name) return true;
return false;
}
static int vdCount(const SkDP &sd, const QString &name) {
SkD::vx_iterator vxBegin, vxEnd;
sd->vdSkeletonVertices(name, vxBegin, vxEnd);
return std::distance(vxBegin, vxEnd);
}
}; // locals
if (propertyName == "mode") {
switch (m_mode.getIndex()) {
case MESH_IDX:
case BUILD_IDX:
case RIGIDITY_IDX:
m_pvs.m_showOriginalColumn = xshColumn();
break;
case ANIMATE_IDX:
m_pvs.m_showOriginalColumn = 0;
if (m_svSel.objects().size() > 1) setSkeletonSelection(-1);
storeDeformation(); // Rebuild deformed skeleton
break;
}
m_mode.notifyListeners(); // You thought that was automatic, eh? BTW, this
// means
// we're requesting toolbars to update options visibility
onSetViewer(); // Store m_pvs in the viewer's visual settings
invalidate();
} else if (propertyName == "vertexName") {
if (m_sd && m_svSel >= 0) {
// Update the selected vertex's name
QString newName(QString::fromStdWString(m_vertexName.getValue()));
const PlasticSkeletonP skeleton = this->skeleton();
assert(skeleton);
const QString &oldName = skeleton->vertex(m_svSel).name();
bool doRename = true;
if (oldName != newName &&
!locals::alreadyContainsVertexName(*skeleton, newName) &&
m_sd->vertexDeformation(newName) &&
locals::vdCount(m_sd, oldName) == 1)
doRename =
(DVGui::MsgBox(
tr("The previous vertex name will be discarded, and all "
"associated keys will be lost.\n\nDo you want to proceed?"),
QObject::tr("Ok"), QObject::tr("Cancel")) == 1);
if (doRename) {
TUndo *undo = new SetVertexNameUndo(m_svSel, newName);
TUndoManager::manager()->add(undo);
undo->redo();
} else {
m_vertexName.setValue(oldName.toStdWString());
m_vertexName.notifyListeners();
}
}
} else if (propertyName == "interpolate") {
if (m_sd && m_svSel >= 0) {
// Set interpolation property to the associated skeleton vertex
int skelId = ::skeletonId();
m_sd->skeleton(skelId)->vertex(m_svSel).m_interpolate =
m_interpolate.getValue();
m_interpolate.notifyListeners(); // NOTE: This should NOT invoke this
// function recursively
PlasticDeformerStorage::instance()->invalidateSkeleton(
m_sd.getPointer(), skelId, PlasticDeformerStorage::ALL);
}
} else if (propertyName == "minAngle") {
if (m_sd && m_svSel >= 0) {
// Set maxAngle property to the associated skeleton vertex
int skelId = ::skeletonId();
bool ok;
double value =
QString::fromStdWString(m_minAngle.getValue()).toDouble(&ok);
if (!ok) value = -l_dmax, m_minAngle.setValue(L"");
m_sd->skeleton(skelId)->vertex(m_svSel).m_minAngle = value;
if (m_mode.getIndex() == ANIMATE_IDX)
deformedSkeleton().vertex(m_svSel).m_minAngle = value;
m_minAngle.notifyListeners(); // NOTE: This should NOT invoke this
// function recursively
}
} else if (propertyName == "maxAngle") {
if (m_sd && m_svSel >= 0) {
// Set maxAngle property to the associated skeleton vertex
int skelId = ::skeletonId();
bool ok;
double value =
QString::fromStdWString(m_maxAngle.getValue()).toDouble(&ok);
if (!ok) value = l_dmax, m_maxAngle.setValue(L"");
m_sd->skeleton(skelId)->vertex(m_svSel).m_maxAngle = value;
if (m_mode.getIndex() == ANIMATE_IDX)
deformedSkeleton().vertex(m_svSel).m_maxAngle = value;
m_maxAngle.notifyListeners(); // NOTE: This should NOT invoke this
// function recursively
}
}
return true;
}
//------------------------------------------------------------------------
void PlasticTool::onShowMeshToggled(bool on) {
m_pvs.m_drawMeshesWireframe = on;
invalidate();
}
//------------------------------------------------------------------------
void PlasticTool::onShowSOToggled(bool on) {
m_pvs.m_drawSO = on;
invalidate();
}
//------------------------------------------------------------------------
void PlasticTool::onShowRigidityToggled(bool on) {
m_pvs.m_drawRigidity = on;
invalidate();
}
//------------------------------------------------------------------------
void PlasticTool::onShowSkelOSToggled(bool on) {
m_showSkeletonOS = on;
invalidate();
}
//****************************************************************************************
// Drawing functions
//****************************************************************************************
namespace PlasticToolLocals {
void drawSquare(const TPointD &pos, double radius) {
glBegin(GL_LINE_LOOP);
glVertex2d(pos.x - radius, pos.y - radius);
glVertex2d(pos.x + radius, pos.y - radius);
glVertex2d(pos.x + radius, pos.y + radius);
glVertex2d(pos.x - radius, pos.y + radius);
glEnd();
}
//------------------------------------------------------------------------
void drawFullSquare(const TPointD &pos, double radius) {
glBegin(GL_QUADS);
glVertex2d(pos.x - radius, pos.y - radius);
glVertex2d(pos.x + radius, pos.y - radius);
glVertex2d(pos.x + radius, pos.y + radius);
glVertex2d(pos.x - radius, pos.y + radius);
glEnd();
}
//------------------------------------------------------------------------
void drawFilledSquare(const TPointD &pos, double radius) {
glBegin(GL_QUADS);
glVertex2d(pos.x - radius, pos.y - radius);
glVertex2d(pos.x + radius, pos.y - radius);
glVertex2d(pos.x + radius, pos.y + radius);
glVertex2d(pos.x - radius, pos.y + radius);
glEnd();
}
//------------------------------------------------------------------------
void drawHandle(const TPointD &pos, double radius, const TPixel32 &color) {
glColor4ub(0, 0, 0, color.m); // Black border
glLineWidth(4.0f);
drawSquare(pos, radius);
glColor4ub(color.r, color.g, color.b, color.m);
glLineWidth(2.0f);
drawSquare(pos, radius);
}
//------------------------------------------------------------------------
void drawFilledHandle(const TPointD &pos, double radius, double pixelSize,
const TPixel32 &color) {
glColor4ub(0, 0, 0, color.m);
drawFilledSquare(pos, radius + pixelSize);
glColor4ub(color.r, color.g, color.b, color.m);
drawFilledSquare(pos, radius);
}
//------------------------------------------------------------------------
void drawText(const TPointD &pos, const QString &text, double fontScale) {
// Get the world-to-window affine
double matrix[16];
glGetDoublev(GL_MODELVIEW_MATRIX, matrix);
TAffine worldToWindowAff(matrix[0], matrix[4], matrix[12], matrix[1],
matrix[5], matrix[13]);
// Push the window reference
glPushMatrix();
glLoadIdentity();
glScaled(fontScale, fontScale, 1.0);
tglDrawText(TScale(1.0 / fontScale) * worldToWindowAff * pos,
text.toStdWString());
// Bottom-left fixed text version
// double origin = 10.0 / fontScale
// tglDrawText(TPointD(origin, origin), text.toStdWString());
glPopMatrix();
}
} // namespace PlasticToolLocals
//========================================================================
void PlasticTool::drawHighlights(const SkDP &sd,
const PlasticSkeleton *skeleton,
double pixelSize) {
glColor3f(1.0f, 0.0f, 0.0f); // Red
glLineWidth(1.0f);
// Vertex highlights
if (m_svHigh >= 0) {
double handleRadius = HIGHLIGHTED_HANDLE_SIZE * pixelSize;
const PlasticSkeleton::vertex_type &vx = skeleton->vertex(m_svHigh);
int hookNumber = sd->hookNumber(vx.name());
assert(hookNumber >= 0);
{
glPushAttrib(GL_LINE_BIT);
glEnable(GL_LINE_STIPPLE);
glLineStipple(1, 0xCCCC);
drawSquare(vx.P(), handleRadius);
glPopAttrib();
}
drawText(vx.P() + TPointD(2.0 * handleRadius, 2.0 * handleRadius),
QString("(%1) ").arg(hookNumber) + vx.name(), 1.7);
} else if (m_seHigh >= 0) {
// Draw a handle at the projection of current mouse position towards the
// highlighted edge
double handleRadius = HANDLE_SIZE * pixelSize;
drawSquare(projection(*skeleton, m_seHigh, m_pos), handleRadius);
}
}
//------------------------------------------------------------------------
void PlasticTool::drawSelections(const SkDP &sd,
const PlasticSkeleton &skeleton,
double pixelSize) {
glColor3f(1.0f, 0.0f, 0.0f); // Red
glLineWidth(1.0f);
double handleRadius = SELECTED_HANDLE_SIZE * pixelSize;
if (!m_svSel.isEmpty()) {
typedef PlasticVertexSelection::objects_container objects_container;
const objects_container &vIdxs = m_svSel.objects();
// Draw a handle square for each selected vertex
objects_container::const_iterator vst, vsEnd = vIdxs.end();
for (vst = vIdxs.begin(); vst != vsEnd; ++vst)
drawSquare(skeleton.vertex(*vst).P(), handleRadius);
// Draw vertex descriptions (only in the single selection case - to avoid
// text pollution)
if (vIdxs.size() == 1) {
const PlasticSkeleton::vertex_type &vx = skeleton.vertex(m_svSel);
int hookNumber = sd->hookNumber(vx.name());
assert(hookNumber >= 0);
drawText(vx.P() + TPointD(2.0 * handleRadius, 2.0 * handleRadius),
QString("(%1) ").arg(hookNumber) + vx.name(), 1.7);
}
}
}
//------------------------------------------------------------------------
void PlasticTool::drawSkeleton(const PlasticSkeleton &skel, double pixelSize,
UCHAR alpha) {
struct locals {
inline static void drawLine(const TPointD &p0, const TPointD &p1) {
glVertex2d(p0.x, p0.y);
glVertex2d(p1.x, p1.y);
}
}; // locals
const tcg::list<PlasticSkeleton::vertex_type> &vertices = skel.vertices();
if (vertices.size() > 0) {
// Draw edges
{
const tcg::list<PlasticSkeleton::edge_type> &edges = skel.edges();
tcg::list<PlasticSkeleton::edge_type>::const_iterator et,
eEnd(edges.end());
glColor4ub(0, 0, 0, alpha);
glLineWidth(4.0f); // Black border
glBegin(GL_LINES);
{
for (et = edges.begin(); et != eEnd; ++et)
locals::drawLine(skel.vertex(et->vertex(0)).P(),
skel.vertex(et->vertex(1)).P());
}
glEnd();
glColor4ub(250, 184, 70, alpha);
glLineWidth(2.0f); // Yellow/Orange-ish line center
glBegin(GL_LINES);
{
for (et = edges.begin(); et != eEnd; ++et)
locals::drawLine(skel.vertex(et->vertex(0)).P(),
skel.vertex(et->vertex(1)).P());
}
glEnd();
}
// Draw vertices
{
const TPixel32 magenta(255, 0, 255, alpha);
const TPixel32 yellow(255, 255, 0, alpha);
double handleRadius = HANDLE_SIZE * pixelSize;
float intHandleThick = 2.0f, extHandleThick = 4.0f;
// Draw root
drawFilledHandle(vertices.begin()->P(), handleRadius, pixelSize, magenta);
// Draw remaining vertices
tcg::list<PlasticSkeleton::vertex_type>::const_iterator vt(
vertices.begin()),
vEnd(vertices.end());
if (vt != vEnd) {
for (vt = ++vertices.begin(); vt != vEnd; ++vt)
drawHandle(vt->P(), handleRadius,
vt->m_interpolate ? magenta : yellow);
}
}
}
}
//------------------------------------------------------------------------
void PlasticTool::drawOnionSkinSkeletons_build(double pixelSize) {
if (!(m_showSkeletonOS && m_sd)) return;
const OnionSkinMask &os =
TTool::getApplication()->getCurrentOnionSkin()->getOnionSkinMask();
std::vector<int> osRows;
int currentRow = ::row();
os.getAll(currentRow, osRows);
TStageObject *obj = ::stageObject();
// Sieve osRows' associated skeleton ids first
std::map<int, UCHAR> skelAlphas;
int r, rCount = int(osRows.size());
for (r = 0; r != rCount; ++r) {
assert(osRows[r] != currentRow);
double sdFrame = obj->paramsTime(double(osRows[r] - 1));
int skelId = m_sd->skeletonId(sdFrame);
UCHAR &skelAlpha = skelAlphas[skelId];
UCHAR alpha =
255 -
UCHAR(255.0 * OnionSkinMask::getOnionSkinFade(osRows[r] - currentRow));
skelAlpha = std::max(skelAlpha, alpha);
}
std::map<int, UCHAR>::iterator st, sEnd(skelAlphas.end());
for (st = skelAlphas.begin(); st != sEnd; ++st) {
const PlasticSkeletonP &skel = m_sd->skeleton(st->first);
drawSkeleton(*skel, pixelSize, st->second);
}
}
//------------------------------------------------------------------------
void PlasticTool::drawOnionSkinSkeletons_animate(double pixelSize) {
if (!(m_showSkeletonOS && m_sd)) return;
const OnionSkinMask &os =
TTool::getApplication()->getCurrentOnionSkin()->getOnionSkinMask();
std::vector<int> osRows;
int currentRow = ::row();
os.getAll(currentRow, osRows);
TStageObject *obj = ::stageObject();
int r, rCount = int(osRows.size());
for (r = 0; r != rCount; ++r) {
assert(osRows[r] != currentRow);
double sdFrame = obj->paramsTime(double(osRows[r] - 1));
PlasticSkeleton skel;
m_sd->storeDeformedSkeleton(m_sd->skeletonId(sdFrame), sdFrame, skel);
UCHAR alpha =
255 -
255.0 * OnionSkinMask::getOnionSkinFade(abs(osRows[r] - currentRow));
drawSkeleton(skel, pixelSize, alpha);
}
}
//------------------------------------------------------------------------
void PlasticTool::drawAngleLimits(const SkDP &sd, int skelId, int v,
double pixelSize) {
struct {
PlasticTool *m_this;
void drawAnnulusArc(const TPointD ¢er, double angleStart,
double angleEnd, double radiusA, double radiusB,
double pixelSize) {
double angleDelta = acos(1.0 - pixelSize / std::max(radiusA, radiusB)) *
((angleStart <= angleEnd) ? 1.0 : -1.0);
int a, aCount = tcg::numeric_ops::grow(
fabs((angleEnd - angleStart) / angleDelta));
glBegin(GL_QUAD_STRIP);
{
for (a = 0; a != aCount; ++a) {
double angle = angleStart + a * angleDelta;
TPointD direction(cos(angle), sin(angle));
tglVertex(center + radiusA * direction);
tglVertex(center + radiusB * direction);
}
TPointD direction(cos(angleEnd), sin(angleEnd));
tglVertex(center + radiusA * direction);
tglVertex(center + radiusB * direction);
}
glEnd();
}
void drawLimit(const SkDP &sd, int skelId, int v, double angleLimit,
double pixelSize) {
const PlasticSkeleton &skel = *sd->skeleton(skelId);
const PlasticSkeleton &defSkel = m_this->deformedSkeleton();
const PlasticSkeletonVertex &vx = skel.vertex(v);
const PlasticSkeletonVertex &defVx = defSkel.vertex(v);
int vParent = vx.parent();
const PlasticSkeletonVertex &vxParent = skel.vertex(vParent);
const PlasticSkeletonVertex &defVxParent = defSkel.vertex(vParent);
// Build directions
int vGrandParent = vxParent.parent();
TPointD dirFromParent(vx.P() - vxParent.P()), dirFromGrandParent(1, 0),
dirFromDeformedGrandParent(1, 0);
if (vGrandParent >= 0) {
const PlasticSkeletonVertex &vxGrandParent = skel.vertex(vGrandParent),
&defVxGrandParent =
defSkel.vertex(vGrandParent);
dirFromGrandParent = vxParent.P() - vxGrandParent.P();
dirFromDeformedGrandParent = defVxParent.P() - defVxGrandParent.P();
}
// Retrieve angular data
double angleShift =
sd->vertexDeformation(skelId, v)->m_params[SkVD::ANGLE]->getValue(
::frame());
double defaultAngleValue =
tcg::point_ops::angle(dirFromGrandParent, dirFromParent) * M_180_PI;
// Convert to radians
double currentBranchAngle_rad =
tcg::point_ops::rad(dirFromDeformedGrandParent);
double currentAngle_rad =
currentBranchAngle_rad + (angleShift + defaultAngleValue) * M_PI_180;
double limitDirection_rad =
currentBranchAngle_rad + (angleLimit + defaultAngleValue) * M_PI_180;
glColor4ub(0, 0, 255, 128);
// Draw limit lines
if (angleShift - 180.0 <= angleLimit &&
angleLimit <= angleShift + 180.0) {
TPointD limitDirection(cos(limitDirection_rad),
sin(limitDirection_rad));
glBegin(GL_LINES);
{
tglVertex(defVxParent.P());
tglVertex(defVxParent.P() + 1e4 * limitDirection);
}
glEnd();
}
// Draw limit annulus arc
angleLimit = tcrop(angleLimit, angleShift - 180.0, angleShift + 180.0);
limitDirection_rad =
currentBranchAngle_rad + (angleLimit + defaultAngleValue) * M_PI_180;
double radius = tcg::point_ops::dist(defVx.P(), defVxParent.P()) * 0.25;
drawAnnulusArc(defVxParent.P(), limitDirection_rad, currentAngle_rad,
radius - 5.0 * pixelSize, radius + 5.0 * pixelSize,
pixelSize);
}
} locals = {this};
// Dismiss no-ops
const PlasticSkeletonP &skel = sd->skeleton(skelId);
if (!skel || v < 0) return;
// Ensure we're editing a vertex with an existing parent
if (m_dragged) {
const PlasticSkeletonVertex &vx = skel->vertex(v);
int vParent = vx.parent();
if (vParent >= 0) {
// Draw angular limits
if (vx.m_minAngle != -l_dmax)
locals.drawLimit(sd, skelId, v, vx.m_minAngle, pixelSize);
if (vx.m_maxAngle != l_dmax)
locals.drawLimit(sd, skelId, v, vx.m_maxAngle, pixelSize);
}
}
}
//------------------------------------------------------------------------
void PlasticTool::draw() {
glPushAttrib(GL_LINE_BIT | GL_COLOR_BUFFER_BIT | GL_ENABLE_BIT);
glEnable(GL_BLEND);
glEnable(GL_LINE_SMOOTH);
switch (m_mode.getIndex()) {
case MESH_IDX:
draw_mesh();
break;
case BUILD_IDX:
draw_build();
break;
case RIGIDITY_IDX:
draw_rigidity();
break;
case ANIMATE_IDX:
draw_animate();
break;
};
glPopAttrib();
}