174 lines
6.0 KiB
C++
174 lines
6.0 KiB
C++
#include "bone_property_widget.h"
|
|
#include "float_number_widget.h"
|
|
#include "image_preview_widget.h"
|
|
#include "theme.h"
|
|
#include <QComboBox>
|
|
#include <QGroupBox>
|
|
#include <QHBoxLayout>
|
|
#include <QLineEdit>
|
|
#include <QPushButton>
|
|
#include <QVBoxLayout>
|
|
#include <unordered_set>
|
|
|
|
BonePropertyWidget::BonePropertyWidget(Document* document,
|
|
const std::vector<dust3d::Uuid>& boneIds,
|
|
QWidget* parent)
|
|
: QWidget(parent)
|
|
, m_document(document)
|
|
, m_boneIds(boneIds)
|
|
{
|
|
prepareBoneIds();
|
|
|
|
QVBoxLayout* mainLayout = new QVBoxLayout;
|
|
|
|
if (nullptr != m_bone) {
|
|
m_parentBoneComboBox = new QComboBox;
|
|
m_parentBoneComboBox->addItem(tr("None"));
|
|
|
|
m_parentJointComboBox = new QComboBox;
|
|
m_parentJointComboBox->addItem(tr("1"));
|
|
m_parentJointComboBox->setVisible(false);
|
|
|
|
QHBoxLayout* parentBoneAndJointLayout = new QHBoxLayout;
|
|
parentBoneAndJointLayout->addWidget(m_parentBoneComboBox);
|
|
parentBoneAndJointLayout->addWidget(m_parentJointComboBox);
|
|
|
|
for (size_t i = 0; i < m_document->boneIdList.size(); ++i) {
|
|
const auto& boneId = m_document->boneIdList[i];
|
|
if (boneId == m_boneId)
|
|
continue;
|
|
const Document::Bone* bone = m_document->findBone(boneId);
|
|
if (nullptr == bone)
|
|
continue;
|
|
m_parentBoneComboBox->addItem(QIcon(bone->previewPixmap), bone->name, QVariant(QString::fromStdString(boneId.toString())));
|
|
if (m_bone->attachBoneId == boneId)
|
|
m_parentBoneComboBox->setCurrentIndex(m_parentBoneComboBox->count() - 1);
|
|
}
|
|
updateBoneJointComboBox();
|
|
|
|
connect(m_parentBoneComboBox, QOverload<int>::of(&QComboBox::currentIndexChanged), this, &BonePropertyWidget::updateBoneJointComboBox);
|
|
connect(m_parentJointComboBox, QOverload<int>::of(&QComboBox::currentIndexChanged), this, &BonePropertyWidget::synchronizeBoneAttachmentFromControls);
|
|
|
|
QHBoxLayout* parentLayout = new QHBoxLayout;
|
|
parentLayout->addWidget(new QLabel(tr("Parent")));
|
|
parentLayout->addLayout(parentBoneAndJointLayout);
|
|
parentLayout->addStretch();
|
|
|
|
m_nameEdit = new QLineEdit;
|
|
Theme::initLineEdit(m_nameEdit);
|
|
m_nameEdit->setFixedWidth(Theme::partPreviewImageSize * 2.7);
|
|
m_nameEdit->setText(m_bone->name);
|
|
|
|
connect(m_nameEdit, &QLineEdit::textChanged, this, &BonePropertyWidget::nameEditChanged);
|
|
|
|
QHBoxLayout* renameLayout = new QHBoxLayout;
|
|
renameLayout->addWidget(new QLabel(tr("Name")));
|
|
renameLayout->addWidget(m_nameEdit);
|
|
renameLayout->addStretch();
|
|
|
|
m_jointsWidget = new IntNumberWidget;
|
|
m_jointsWidget->setRange(2, 10);
|
|
|
|
QPushButton* nodePicker = new QPushButton(QChar(fa::eyedropper));
|
|
nodePicker->setToolTip(tr("Click node one by one on canvas as joints in order"));
|
|
Theme::initIconButton(nodePicker);
|
|
|
|
connect(nodePicker, &QPushButton::clicked, this, [this]() {
|
|
emit this->pickBoneJoints(this->m_boneId, (size_t)this->m_jointsWidget->value());
|
|
});
|
|
|
|
QHBoxLayout* jointsLayout = new QHBoxLayout;
|
|
jointsLayout->addWidget(new QLabel(tr("Joints")));
|
|
jointsLayout->addWidget(m_jointsWidget);
|
|
jointsLayout->addStretch();
|
|
jointsLayout->addWidget(nodePicker);
|
|
|
|
mainLayout->addLayout(parentLayout);
|
|
mainLayout->addLayout(renameLayout);
|
|
mainLayout->addLayout(jointsLayout);
|
|
}
|
|
|
|
mainLayout->setSizeConstraint(QLayout::SetFixedSize);
|
|
|
|
connect(this, &BonePropertyWidget::renameBone, m_document, &Document::renameBone);
|
|
connect(this, &BonePropertyWidget::setBoneAttachment, m_document, &Document::setBoneAttachment);
|
|
connect(this, &BonePropertyWidget::groupOperationAdded, m_document, &Document::saveSnapshot);
|
|
|
|
setLayout(mainLayout);
|
|
|
|
setFixedSize(minimumSizeHint());
|
|
}
|
|
|
|
dust3d::Uuid BonePropertyWidget::editingParentBoneId()
|
|
{
|
|
return dust3d::Uuid(m_parentBoneComboBox->currentData().toString().toStdString());
|
|
}
|
|
|
|
int BonePropertyWidget::editingParentJointIndex()
|
|
{
|
|
return dust3d::String::toInt(m_parentJointComboBox->currentText().toStdString()) - 1;
|
|
}
|
|
|
|
void BonePropertyWidget::synchronizeBoneAttachmentFromControls()
|
|
{
|
|
if (nullptr == m_bone)
|
|
return;
|
|
|
|
auto parentBoneId = editingParentBoneId();
|
|
int parentJointIndex = editingParentJointIndex();
|
|
|
|
if (-1 == parentJointIndex) {
|
|
// The joint control is initializing, ignore
|
|
return;
|
|
}
|
|
|
|
if (m_bone->attachBoneId == parentBoneId && parentJointIndex == m_bone->attachBoneJointIndex)
|
|
return;
|
|
|
|
emit setBoneAttachment(m_boneId, parentBoneId, parentJointIndex);
|
|
}
|
|
|
|
void BonePropertyWidget::prepareBoneIds()
|
|
{
|
|
if (1 == m_boneIds.size()) {
|
|
m_bone = m_document->findBone(m_boneIds.front());
|
|
if (nullptr != m_bone)
|
|
m_boneId = m_boneIds.front();
|
|
}
|
|
}
|
|
|
|
void BonePropertyWidget::nameEditChanged()
|
|
{
|
|
const Document::Bone* bone = m_document->findBone(m_boneId);
|
|
if (nullptr == bone)
|
|
return;
|
|
emit renameBone(m_boneId, m_nameEdit->text());
|
|
emit groupOperationAdded();
|
|
}
|
|
|
|
void BonePropertyWidget::updateBoneJointComboBox()
|
|
{
|
|
if (nullptr == m_bone)
|
|
return;
|
|
|
|
auto parentBoneId = editingParentBoneId();
|
|
const Document::Bone* parentBone = m_document->findBone(parentBoneId);
|
|
if (nullptr == parentBone || parentBone->joints.size() < 2) {
|
|
if (1 == m_parentJointComboBox->count())
|
|
return;
|
|
m_parentJointComboBox->clear();
|
|
m_parentJointComboBox->addItem(tr("1"));
|
|
m_parentJointComboBox->setVisible(false);
|
|
synchronizeBoneAttachmentFromControls();
|
|
return;
|
|
}
|
|
m_parentJointComboBox->clear();
|
|
for (size_t i = 0; i < parentBone->joints.size(); ++i) {
|
|
m_parentJointComboBox->addItem(QString::number(i + 1));
|
|
if (i == (size_t)m_bone->attachBoneJointIndex)
|
|
m_parentJointComboBox->setCurrentIndex(m_parentJointComboBox->count() - 1);
|
|
}
|
|
m_parentJointComboBox->setVisible(true);
|
|
synchronizeBoneAttachmentFromControls();
|
|
}
|