diff -r 9374fea8f77f -r f1b8cb53d2a2 src/addObjectDialog.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/addObjectDialog.cpp Wed May 08 15:19:06 2013 +0300 @@ -0,0 +1,511 @@ +/* + * LDForge: LDraw parts authoring CAD + * Copyright (C) 2013 Santeri Piippo + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ + +#include +#include +#include +#include "gui.h" +#include "addObjectDialog.h" +#include "file.h" +#include "colors.h" +#include "colorSelectDialog.h" +#include "history.h" +#include "setContentsDialog.h" + +#define APPLY_COORDS(OBJ, N) \ + for (short i = 0; i < N; ++i) \ + for (const Axis ax : g_Axes) \ + OBJ->vaCoords[i][ax] = dlg.dsb_coords[(i * 3) + ax]->value (); + +// ============================================================================= +class SubfileListItem : public QTreeWidgetItem { +public: + SubfileListItem (QTreeWidgetItem* parent, int subfileID) : + QTreeWidgetItem (parent), subfileID (subfileID) {} + SubfileListItem (QTreeWidget* parent, int subfileID) : + QTreeWidgetItem (parent), subfileID (subfileID) {} + + int subfileID; +}; + +// ============================================================================= +// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * +// ============================================================================= +AddObjectDialog::AddObjectDialog (const LDObject::Type type, LDObject* obj, QWidget* parent) : + QDialog (parent) +{ + setlocale (LC_ALL, "C"); + + short coordCount = 0; + + switch (type) { + case LDObject::Comment: + le_comment = new QLineEdit; + if (obj) + le_comment->setText (static_cast (obj)->text); + break; + + case LDObject::Line: + coordCount = 6; + break; + + case LDObject::Triangle: + coordCount = 9; + break; + + case LDObject::Quad: + case LDObject::CondLine: + coordCount = 12; + break; + + case LDObject::Vertex: + coordCount = 3; + break; + + case LDObject::BFC: + rb_bfcType = new RadioBox ("Statement", {}, 0, Qt::Vertical); + + for (int i = 0; i < LDBFC::NumStatements; ++i) + rb_bfcType->addButton (new QRadioButton (LDBFC::statements[i])); + + if (obj) + rb_bfcType->setValue ((int) static_cast (obj)->type); + break; + + case LDObject::Subfile: + coordCount = 3; + + enum { + Parts, + Subparts, + Primitives, + HiRes, + }; + + tw_subfileList = new QTreeWidget (); + for (int i : vector ({Parts, Subparts, Primitives, HiRes})) { + SubfileListItem* parentItem = new SubfileListItem (tw_subfileList, -1); + parentItem->setText (0, (i == Parts) ? "Parts" : + (i == Subparts) ? "Subparts" : + (i == Primitives) ? "Primitives" : + "Hi-Res"); + + ulong j = 0; + for (partListEntry& part : g_PartList) { + QList subfileItems; + + str fileName = part.sName; + const bool isSubpart = fileName.substr (0, 2) == "s\\"; + const bool isPrimitive = str (part.sTitle).substr (0, 9) == "Primitive"; + const bool isHiRes = fileName.substr (0, 3) == "48\\"; + + if ((i == Subparts && isSubpart) || + (i == Primitives && isPrimitive) || + (i == HiRes && isHiRes) || + (i == Parts && !isSubpart && !isPrimitive && !isHiRes)) + { + SubfileListItem* item = new SubfileListItem (parentItem, j); + item->setText (0, fmt ("%s - %s", part.sName, part.sTitle)); + subfileItems.append (item); + } + + j++; + } + + tw_subfileList->addTopLevelItem (parentItem); + } + + connect (tw_subfileList, SIGNAL (itemSelectionChanged ()), this, SLOT (slot_subfileTypeChanged ())); + lb_subfileName = new QLabel ("File:"); + le_subfileName = new QLineEdit; + le_subfileName->setFocus (); + + if (obj) { + LDSubfile* ref = static_cast (obj); + le_subfileName->setText (ref->zFileName); + } + break; + + case LDObject::Radial: + coordCount = 3; + + lb_radType = new QLabel ("Type:"); + lb_radResolution = new QLabel ("Resolution:"); + lb_radSegments = new QLabel ("Segments:"); + lb_radRingNum = new QLabel ("Ring number:"); + + rb_radType = new RadioBox ("Type", {}, 0, Qt::Vertical); + + for (int i = 0; i < LDRadial::NumTypes; ++i) { + if (i % (LDRadial::NumTypes / 2) == 0) + rb_radType->rowBreak (); + + rb_radType->addButton (new QRadioButton (LDRadial::radialTypeName ((LDRadial::Type) i))); + } + + connect (rb_radType, SIGNAL (sig_buttonPressed (int)), this, SLOT (slot_radialTypeChanged (int))); + + cb_radHiRes = new QCheckBox ("Hi-Res"); + + sb_radSegments = new QSpinBox; + sb_radSegments->setMinimum (1); + + sb_radRingNum = new QSpinBox; + sb_radRingNum->setEnabled (false); + + if (obj) { + LDRadial* rad = static_cast (obj); + + rb_radType->setValue (rad->eRadialType); + sb_radSegments->setValue (rad->dSegments); + cb_radHiRes->setChecked ((rad->dDivisions == 48) ? Qt::Checked : Qt::Unchecked); + sb_radRingNum->setValue (rad->dRingNum); + } + break; + + default: + assert (false); + return; + } + + QPixmap icon = getIcon (fmt ("add-%s", g_saObjTypeIcons[type])); + LDObject* defaults = LDObject::getDefault (type); + + lb_typeIcon = new QLabel; + lb_typeIcon->setPixmap (icon); + + // Show a color edit dialog for the types that actually use the color + if (defaults->isColored ()) { + if (obj != null) + dColor = obj->dColor; + else + dColor = (type == LDObject::CondLine || type == LDObject::Line) ? edgecolor : maincolor; + + pb_color = new QPushButton; + setButtonBackground (pb_color, dColor); + connect (pb_color, SIGNAL (clicked ()), this, SLOT (slot_colorButtonClicked ())); + } + + for (short i = 0; i < coordCount; ++i) { + dsb_coords[i] = new QDoubleSpinBox; + dsb_coords[i]->setDecimals (5); + dsb_coords[i]->setMinimum (-10000.0); + dsb_coords[i]->setMaximum (10000.0); + } + + IMPLEMENT_DIALOG_BUTTONS + + QGridLayout* const layout = new QGridLayout; + layout->addWidget (lb_typeIcon, 0, 0); + + switch (type) { + case LDObject::Line: + case LDObject::CondLine: + case LDObject::Triangle: + case LDObject::Quad: + // Apply coordinates + if (obj) { + for (short i = 0; i < coordCount / 3; ++i) + for (short j = 0; j < 3; ++j) + dsb_coords[(i * 3) + j]->setValue (obj->vaCoords[i].coord (j)); + } + break; + + case LDObject::Comment: + layout->addWidget (le_comment, 0, 1); + break; + + case LDObject::BFC: + layout->addWidget (rb_bfcType, 0, 1); + break; + + case LDObject::Radial: + layout->addWidget (rb_radType, 1, 1, 3, 2); + layout->addWidget (cb_radHiRes, 1, 3); + layout->addWidget (lb_radSegments, 2, 3); + layout->addWidget (sb_radSegments, 2, 4); + layout->addWidget (lb_radRingNum, 3, 3); + layout->addWidget (sb_radRingNum, 3, 4); + + if (obj) + for (short i = 0; i < 3; ++i) + dsb_coords[i]->setValue (static_cast (obj)->vPosition.coord (i)); + break; + + case LDObject::Subfile: + layout->addWidget (tw_subfileList, 1, 1, 1, 2); + layout->addWidget (lb_subfileName, 2, 1); + layout->addWidget (le_subfileName, 2, 2); + + if (obj) + for (short i = 0; i < 3; ++i) + dsb_coords[i]->setValue (static_cast (obj)->vPosition.coord (i)); + break; + + default: + break; + } + + if (type == LDObject::Subfile || type == LDObject::Radial) { + QLabel* lb_matrix = new QLabel ("Matrix:"); + le_matrix = new QLineEdit; + // le_matrix->setValidator (new QDoubleValidator); + matrix<3> defval = g_identity; + + if (obj) { + if (obj->getType () == LDObject::Subfile) + defval = static_cast (obj)->mMatrix; + else + defval = static_cast (obj)->mMatrix; + } + + le_matrix->setText (defval.stringRep ()); + layout->addWidget (lb_matrix, 4, 1); + layout->addWidget (le_matrix, 4, 2, 1, 3); + } + + if (defaults->isColored ()) + layout->addWidget (pb_color, 1, 0); + + if (coordCount > 0) { + QGridLayout* const qCoordLayout = new QGridLayout; + + for (short i = 0; i < coordCount; ++i) + qCoordLayout->addWidget (dsb_coords[i], (i / 3), (i % 3)); + + layout->addLayout (qCoordLayout, 0, 1, (coordCount / 3), 3); + } + + layout->addWidget (bbx_buttons, 5, 0, 1, 4); + setLayout (layout); + setWindowTitle (fmt (APPNAME ": New %s", + g_saObjTypeNames[type]).chars()); + + setWindowIcon (icon); + delete defaults; +} + +// ============================================================================= +// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * +// ============================================================================= +void AddObjectDialog::setButtonBackground (QPushButton* button, short color) { + button->setIcon (getIcon ("palette")); + button->setAutoFillBackground (true); + button->setStyleSheet ( + fmt ("background-color: %s", getColor (color)->zColorString.chars()).chars() + ); +} + +// ============================================================================= +// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * +// ============================================================================= +char* AddObjectDialog::currentSubfileName() { + SubfileListItem* item = static_cast (tw_subfileList->currentItem ()); + + if (item->subfileID == -1) + return null; // selected a heading + + return g_PartList[item->subfileID].sName; +} + +// ============================================================================= +// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * +// ============================================================================= +void AddObjectDialog::slot_colorButtonClicked () { + ColorSelectDialog::staticDialog (dColor, dColor, this); + setButtonBackground (pb_color, dColor); +} + +// ============================================================================= +// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * +// ============================================================================= +void AddObjectDialog::slot_radialTypeChanged (int dType) { + LDRadial::Type eType = (LDRadial::Type) dType; + sb_radRingNum->setEnabled (eType == LDRadial::Ring || eType == LDRadial::Cone); +} + +// ============================================================================= +// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * +// ============================================================================= +void AddObjectDialog::slot_subfileTypeChanged () { + char* name = currentSubfileName (); + + if (name) + le_subfileName->setText (name); +} + +// ============================================================================= +template T* initObj (LDObject*& obj) { + if (obj == null) + obj = new T; + + return static_cast (obj); +} + +// ============================================================================= +// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * +// ============================================================================= +void AddObjectDialog::staticDialog (const LDObject::Type type, LDObject* obj) { + setlocale (LC_ALL, "C"); + + // Redirect editing of gibberish to the set contents dialog + if (obj && obj->getType () == LDObject::Gibberish) { + SetContentsDialog::staticDialog (obj); + return; + } + + if (type == LDObject::Empty) + return; // Nothing to edit with empties + + const bool newObject = (obj == null); + AddObjectDialog dlg (type, obj); + + if (obj) + assert (obj->getType () == type); + + if (dlg.exec () == false) + return; + + LDObject* backup = null; + if (!newObject) + backup = obj->clone (); + + matrix<3> transform = g_identity; + if (type == LDObject::Subfile || type == LDObject::Radial) { + vector matrixstrvals = str (dlg.le_matrix->text ()).split (" ", true); + + if (matrixstrvals.size () == 9) { + double matrixvals[9]; + int i = 0; + + for (str val : matrixstrvals) + matrixvals[i++] = atof (val); + + transform = matrix<3> (matrixvals); + } + } + + switch (type) { + case LDObject::Comment: + { + LDComment* comm = initObj (obj); + comm->text = dlg.le_comment->text (); + } + break; + + case LDObject::Line: + { + LDLine* line = initObj (obj); + line->dColor = dlg.dColor; + APPLY_COORDS (line, 2) + } + break; + + case LDObject::Triangle: + { + LDTriangle* tri = initObj (obj); + tri->dColor = dlg.dColor; + APPLY_COORDS (tri, 3) + } + break; + + case LDObject::Quad: + { + LDQuad* quad = initObj (obj); + quad->dColor = dlg.dColor; + APPLY_COORDS (quad, 4) + } + break; + + case LDObject::CondLine: + { + LDCondLine* line = initObj (obj); + line->dColor = dlg.dColor; + APPLY_COORDS (line, 4) + } + break; + + case LDObject::BFC: + { + LDBFC* bfc = initObj (obj); + bfc->type = (LDBFC::Type) dlg.rb_bfcType->value (); + } + break; + + case LDObject::Vertex: + { + LDVertex* vert = initObj (obj); + vert->dColor = dlg.dColor; + + for (const Axis ax : g_Axes) + vert->vPosition[ax] = dlg.dsb_coords[ax]->value (); + } + break; + + case LDObject::Radial: + { + LDRadial* pRad = initObj (obj); + pRad->dColor = dlg.dColor; + + for (const Axis ax : g_Axes) + pRad->vPosition[ax] = dlg.dsb_coords[ax]->value (); + + pRad->dDivisions = (dlg.cb_radHiRes->checkState () != Qt::Checked) ? 16 : 48; + pRad->dSegments = min (dlg.sb_radSegments->value (), pRad->dDivisions); + pRad->eRadialType = (LDRadial::Type) dlg.rb_radType->value (); + pRad->dRingNum = dlg.sb_radRingNum->value (); + pRad->mMatrix = transform; + } + break; + + case LDObject::Subfile: + { + str name = dlg.le_subfileName->text (); + if (~name == 0) + return; // no subfile filename + + OpenFile* file = loadSubfile (name); + if (!file) + return; + + LDSubfile* ref = initObj (obj); + ref->dColor = dlg.dColor; + + for (const Axis ax : g_Axes) + ref->vPosition[ax] = dlg.dsb_coords[ax]->value (); + + ref->zFileName = name; + ref->mMatrix = transform; + ref->pFile = file; + } + break; + + default: + break; + } + + if (newObject) { + ulong idx = g_win->getInsertionPoint (); + g_curfile->insertObj (idx, obj); + History::addEntry (new AddHistory ({(ulong) idx}, {obj->clone ()})); + } else { + History::addEntry (new EditHistory ({(ulong) obj->getIndex (g_curfile)}, {backup}, {obj->clone ()})); + } + + g_win->refresh (); +} \ No newline at end of file