--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/src/lddocument.cpp Tue Feb 14 14:52:01 2017 +0200
@@ -0,0 +1,528 @@
+/*
+ * LDForge: LDraw parts authoring CAD
+ * Copyright (C) 2013 - 2017 Teemu 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 <http://www.gnu.org/licenses/>.
+ */
+
+#include <QMessageBox>
+#include <QFileDialog>
+#include "lddocument.h"
+#include "miscallenous.h"
+#include "mainwindow.h"
+#include "canvas.h"
+#include "documentloader.h"
+#include "dialogs/openprogressdialog.h"
+#include "documentmanager.h"
+#include "linetypes/comment.h"
+
+LDDocument::LDDocument (DocumentManager* parent) :
+ Model {parent},
+ HierarchyElement (parent),
+ m_history (new EditHistory (this)),
+ m_savePosition(-1),
+ m_tabIndex(-1),
+ m_manager (parent) {}
+
+LDDocument::~LDDocument()
+{
+ m_isBeingDestroyed = true;
+ delete m_history;
+}
+
+QString LDDocument::name() const
+{
+ return m_name;
+}
+
+void LDDocument::setName (QString value)
+{
+ m_name = value;
+}
+
+EditHistory* LDDocument::history() const
+{
+ return m_history;
+}
+
+QString LDDocument::fullPath()
+{
+ return m_fullPath;
+}
+
+void LDDocument::setFullPath (QString value)
+{
+ m_fullPath = value;
+}
+
+int LDDocument::tabIndex() const
+{
+ return m_tabIndex;
+}
+
+void LDDocument::setTabIndex (int value)
+{
+ m_tabIndex = value;
+}
+
+const QList<LDPolygon>& LDDocument::polygonData() const
+{
+ return m_polygonData;
+}
+
+long LDDocument::savePosition() const
+{
+ return m_savePosition;
+}
+
+void LDDocument::setSavePosition (long value)
+{
+ m_savePosition = value;
+}
+
+QString LDDocument::defaultName() const
+{
+ return m_defaultName;
+}
+
+void LDDocument::setDefaultName (QString value)
+{
+ m_defaultName = value;
+}
+
+void LDDocument::setFrozen(bool value)
+{
+ m_isFrozen = value;
+}
+
+bool LDDocument::isFrozen() const
+{
+ return m_isFrozen;
+}
+
+void LDDocument::addHistoryStep()
+{
+ history()->addStep();
+}
+
+void LDDocument::undo()
+{
+ history()->undo();
+}
+
+void LDDocument::redo()
+{
+ history()->redo();
+}
+
+void LDDocument::clearHistory()
+{
+ history()->clear();
+}
+
+void LDDocument::addToHistory (AbstractHistoryEntry* entry)
+{
+ history()->add (entry);
+}
+
+void LDDocument::close()
+{
+ if (not isFrozen())
+ {
+ setFrozen(true);
+ m_manager->documentClosed(this);
+ }
+}
+
+// =============================================================================
+//
+// Performs safety checks. Do this before closing any files!
+//
+bool LDDocument::isSafeToClose()
+{
+ using msgbox = QMessageBox;
+ setlocale (LC_ALL, "C");
+
+ // If we have unsaved changes, warn and give the option of saving.
+ if (hasUnsavedChanges())
+ {
+ QString message = format (tr ("There are unsaved changes to %1. Should it be saved?"), getDisplayName());
+
+ int button = msgbox::question (m_window, QObject::tr ("Unsaved Changes"), message,
+ (msgbox::Yes | msgbox::No | msgbox::Cancel), msgbox::Cancel);
+
+ switch (button)
+ {
+ case msgbox::Yes:
+ {
+ // If we don't have a file path yet, we have to ask the user for one.
+ if (name().isEmpty())
+ {
+ QString newpath = QFileDialog::getSaveFileName (m_window, QObject::tr ("Save As"),
+ name(), QObject::tr ("LDraw files (*.dat *.ldr)"));
+
+ if (newpath.isEmpty())
+ return false;
+
+ setName (newpath);
+ }
+
+ if (not save())
+ {
+ message = format (QObject::tr ("Failed to save %1 (%2)\nDo you still want to close?"),
+ name(), strerror (errno));
+
+ if (msgbox::critical (m_window, QObject::tr ("Save Failure"), message,
+ (msgbox::Yes | msgbox::No), msgbox::No) == msgbox::No)
+ {
+ return false;
+ }
+ }
+ break;
+ }
+
+ case msgbox::Cancel:
+ return false;
+
+ default:
+ break;
+ }
+ }
+
+ return true;
+}
+
+// =============================================================================
+//
+bool LDDocument::save (QString path, qint64* sizeptr)
+{
+ if (isFrozen())
+ return false;
+
+ if (path.isEmpty())
+ path = fullPath();
+
+ // If the second object in the list holds the file name, update that now.
+ LDObject* nameObject = getObject (1);
+
+ if (nameObject and nameObject->type() == LDObjectType::Comment)
+ {
+ LDComment* nameComment = static_cast<LDComment*> (nameObject);
+
+ if (nameComment->text().left (6) == "Name: ")
+ {
+ QString newname = shortenName (path);
+ nameComment->setText (format ("Name: %1", newname));
+ m_window->buildObjectList();
+ }
+ }
+
+ QByteArray data;
+
+ if (sizeptr)
+ *sizeptr = 0;
+
+ // File is open, now save the model to it. Note that LDraw requires files to have DOS line endings.
+ for (LDObject* obj : objects())
+ {
+ QByteArray subdata ((obj->asText() + "\r\n").toUtf8());
+ data.append (subdata);
+
+ if (sizeptr)
+ *sizeptr += countof(subdata);
+ }
+
+ QFile f (path);
+
+ if (not f.open (QIODevice::WriteOnly))
+ return false;
+
+ f.write (data);
+ f.close();
+
+ // We have successfully saved, update the save position now.
+ setSavePosition (history()->position());
+ setFullPath (path);
+ setName (shortenName (path));
+ m_window->updateDocumentListItem (this);
+ m_window->updateTitle();
+ return true;
+}
+
+// =============================================================================
+//
+void LDDocument::reloadAllSubfiles()
+{
+ print ("Reloading subfiles of %1", getDisplayName());
+
+ // Go through all objects in the current file and reload the subfiles
+ for (LDObject* obj : objects())
+ {
+ if (obj->type() == LDObjectType::SubfileReference)
+ {
+ LDSubfileReference* reference = static_cast<LDSubfileReference*> (obj);
+ LDDocument* fileInfo = m_documents->getDocumentByName (reference->fileInfo()->name());
+
+ if (fileInfo)
+ reference->setFileInfo (fileInfo);
+ else
+ emplaceReplacement<LDError>(reference, reference->asText(), format("Could not open %1", reference->fileInfo()->name()));
+ }
+
+ // Reparse gibberish files. It could be that they are invalid because
+ // of loading errors. Circumstances may be different now.
+ if (obj->type() == LDObjectType::Error)
+ replaceWithFromString(obj, static_cast<LDError*> (obj)->contents());
+ }
+
+ m_needsRecache = true;
+
+ if (this == m_window->currentDocument())
+ m_window->buildObjectList();
+}
+
+// =============================================================================
+//
+void LDDocument::insertObject (int pos, LDObject* obj)
+{
+ Model::insertObject(pos, obj);
+ history()->add(new AddHistoryEntry {pos, obj});
+ connect(obj, SIGNAL(codeChanged(QString,QString)), this, SLOT(objectChanged(QString,QString)));
+
+#ifdef DEBUG
+ if (not isFrozen())
+ dprint ("Inserted object #%1 (%2) at %3\n", obj->id(), obj->typeName(), pos);
+#endif
+}
+
+void LDDocument::objectChanged(QString before, QString after)
+{
+ LDObject* object = static_cast<LDObject*>(sender());
+ addToHistory(new EditHistoryEntry {object->lineNumber(), before, after});
+ redoVertices();
+ emit objectModified(object);
+}
+
+LDObject* LDDocument::withdrawAt(int position)
+{
+ LDObject* object = getObject(position);
+
+ if (not isFrozen() and not m_isBeingDestroyed)
+ {
+ history()->add(new DelHistoryEntry {position, object});
+ m_objectVertices.remove(object);
+ }
+
+ m_selection.remove(object);
+ return Model::withdrawAt(position);
+}
+
+// =============================================================================
+//
+bool LDDocument::hasUnsavedChanges() const
+{
+ return not isFrozen() and history()->position() != savePosition();
+}
+
+// =============================================================================
+//
+QString LDDocument::getDisplayName()
+{
+ if (not name().isEmpty())
+ return name();
+
+ if (not defaultName().isEmpty())
+ return "[" + defaultName() + "]";
+
+ return QObject::tr ("untitled");
+}
+
+// =============================================================================
+//
+void LDDocument::initializeCachedData()
+{
+ if (m_needsRecache)
+ {
+ m_vertices.clear();
+ Model model {m_documents};
+ inlineContents(model, true, true);
+
+ for (LDObject* obj : model.objects())
+ {
+ if (obj->type() == LDObjectType::SubfileReference)
+ {
+ print ("Warning: unable to inline %1 into %2",
+ static_cast<LDSubfileReference*> (obj)->fileInfo()->getDisplayName(),
+ getDisplayName());
+ continue;
+ }
+
+ LDPolygon* data = obj->getPolygon();
+
+ if (data)
+ {
+ m_polygonData << *data;
+ delete data;
+ }
+ }
+
+ m_needsRecache = false;
+ }
+
+ if (m_verticesOutdated)
+ {
+ m_objectVertices.clear();
+ Model model {m_documents};
+ inlineContents(model, true, false);
+
+ for (LDObject* object : model)
+ {
+ auto iterator = m_objectVertices.find (object);
+
+ if (iterator == m_objectVertices.end())
+ iterator = m_objectVertices.insert (object, QSet<Vertex>());
+ else
+ iterator->clear();
+
+ object->getVertices (*iterator);
+ }
+
+ m_vertices.clear();
+
+ for (const QSet<Vertex>& vertices : m_objectVertices)
+ m_vertices.unite(vertices);
+
+ m_verticesOutdated = false;
+ }
+}
+
+// =============================================================================
+//
+QList<LDPolygon> LDDocument::inlinePolygons()
+{
+ initializeCachedData();
+ return polygonData();
+}
+
+// =============================================================================
+// -----------------------------------------------------------------------------
+void LDDocument::inlineContents(Model& model, bool deep, bool renderinline)
+{
+ if (m_manager->preInline(this, model, deep, renderinline))
+ return; // Manager dealt with this inline
+
+ for (LDObject* object : objects())
+ {
+ // Skip those without scemantic meaning
+ if (not object->isScemantic())
+ continue;
+
+ // Got another sub-file reference, inline it if we're deep-inlining. If not,
+ // just add it into the objects normally. Yay, recursion!
+ if (deep and object->type() == LDObjectType::SubfileReference)
+ static_cast<LDSubfileReference*>(object)->inlineContents(model, deep, renderinline);
+ else
+ model.addFromString(object->asText());
+ }
+}
+
+// =============================================================================
+//
+void LDDocument::addToSelection (LDObject* obj) // [protected]
+{
+ if (not m_selection.contains(obj) and obj->model() == this)
+ {
+ m_selection.insert(obj);
+ emit objectModified(obj);
+
+ // If this object is inverted with INVERTNEXT, select the INVERTNEXT as well.
+ LDBfc* invertnext;
+
+ if (obj->previousIsInvertnext(invertnext))
+ addToSelection(invertnext);
+ }
+}
+
+// =============================================================================
+//
+void LDDocument::removeFromSelection (LDObject* obj) // [protected]
+{
+ if (m_selection.contains(obj))
+ {
+ m_selection.remove(obj);
+ emit objectModified(obj);
+
+ // If this object is inverted with INVERTNEXT, deselect the INVERTNEXT as well.
+ LDBfc* invertnext;
+
+ if (obj->previousIsInvertnext(invertnext))
+ removeFromSelection(invertnext);
+ }
+}
+
+// =============================================================================
+//
+void LDDocument::clearSelection()
+{
+ for (LDObject* object : m_selection.toList())
+ removeFromSelection(object);
+}
+
+// =============================================================================
+//
+const QSet<LDObject*>& LDDocument::getSelection() const
+{
+ return m_selection;
+}
+
+// =============================================================================
+//
+bool LDDocument::swapObjects (LDObject* one, LDObject* other)
+{
+ if (Model::swapObjects(one, other))
+ {
+ addToHistory(new SwapHistoryEntry {one->id(), other->id()});
+ return true;
+ }
+ else
+ {
+ return false;
+ }
+}
+
+// =============================================================================
+//
+QString LDDocument::shortenName (QString a) // [static]
+{
+ QString shortname = Basename (a);
+ QString topdirname = Basename (Dirname (a));
+
+ if (DocumentManager::specialSubdirectories.contains (topdirname))
+ shortname.prepend (topdirname + "\\");
+
+ return shortname;
+}
+
+// =============================================================================
+//
+const QSet<Vertex>& LDDocument::inlineVertices()
+{
+ initializeCachedData();
+ return m_vertices;
+}
+
+void LDDocument::redoVertices()
+{
+ m_verticesOutdated = true;
+}