tahoma2d/toonz/sources/tnztools/plastictool.cpp
Toshihiro Shimizu 890dddabbd first commit
2016-03-19 02:57:51 +09:00

2072 lines
60 KiB
C++

#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 &param, 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);
CASE BUILD_IDX : mouseMove_build(pos, me);
CASE RIGIDITY_IDX : mouseMove_rigidity(pos, me);
CASE ANIMATE_IDX : mouseMove_animate(pos, me);
}
}
//------------------------------------------------------------------------
void PlasticTool::leftButtonDown(const TPointD &pos, const TMouseEvent &me)
{
switch (m_mode.getIndex()) {
case MESH_IDX:
leftButtonDown_mesh(pos, me);
CASE BUILD_IDX : leftButtonDown_build(pos, me);
CASE RIGIDITY_IDX : leftButtonDown_rigidity(pos, me);
CASE ANIMATE_IDX : leftButtonDown_animate(pos, me);
}
}
//------------------------------------------------------------------------
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);
CASE BUILD_IDX : leftButtonDrag_build(pos, me);
CASE RIGIDITY_IDX : leftButtonDrag_rigidity(pos, me);
CASE ANIMATE_IDX : leftButtonDrag_animate(pos, me);
}
}
//------------------------------------------------------------------------
void PlasticTool::leftButtonUp(const TPointD &pos, const TMouseEvent &me)
{
switch (m_mode.getIndex()) {
case MESH_IDX:
leftButtonUp_mesh(pos, me);
CASE BUILD_IDX : leftButtonUp_build(pos, me);
CASE RIGIDITY_IDX : leftButtonUp_rigidity(pos, me);
CASE ANIMATE_IDX : leftButtonUp_animate(pos, me);
}
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);
CASE BUILD_IDX : addContextMenuActions_build(menu);
CASE RIGIDITY_IDX : addContextMenuActions_rigidity(menu);
CASE ANIMATE_IDX : addContextMenuActions_animate(menu);
}
// 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();
CASE ANIMATE_IDX:
{
m_pvs.m_showOriginalColumn = 0;
if (m_svSel.objects().size() > 1)
setSkeletonSelection(-1);
storeDeformation(); // Rebuild deformed skeleton
}
};
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 = tmax(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 &center, double angleStart, double angleEnd,
double radiusA, double radiusB, double pixelSize)
{
double angleDelta = acos(1.0 - pixelSize / tmax(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::consts::rad_to_deg * tcg::point_ops::angle(
dirFromGrandParent, dirFromParent);
// Convert to radians
double currentBranchAngle_rad = tcg::point_ops::rad(dirFromDeformedGrandParent);
double currentAngle_rad = currentBranchAngle_rad + tcg::consts::deg_to_rad * (angleShift + defaultAngleValue);
double limitDirection_rad = currentBranchAngle_rad + tcg::consts::deg_to_rad * (angleLimit + defaultAngleValue);
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 + tcg::consts::deg_to_rad * (angleLimit + defaultAngleValue);
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();
CASE BUILD_IDX : draw_build();
CASE RIGIDITY_IDX : draw_rigidity();
CASE ANIMATE_IDX : draw_animate();
};
glPopAttrib();
}