diff -r 2f383e88acf4 -r 185eb297dc1e src/model.cpp --- a/src/model.cpp Mon Sep 27 21:04:45 2021 +0300 +++ b/src/model.cpp Tue Sep 28 00:10:29 2021 +0300 @@ -18,6 +18,7 @@ #include #include +#include #include #include "model.h" #include "modeleditcontext.h" @@ -36,7 +37,7 @@ */ Model::Model(const QString& path, QObject *parent) : QAbstractListModel{parent}, - path{path} + storedPath{path} { connect(this, &Model::dataChanged, [&](){ this->needRecache = true; }); } @@ -187,6 +188,39 @@ } /** + * @brief Gets the path to the model + * @return path + */ +const QString& Model::path() const +{ + return this->storedPath; +} + +/** + * @brief Sets the path to the model + * @param path New path to use + */ +void Model::setPath(const QString &path) +{ + this->storedPath = path; + this->header.name = QFileInfo{path}.fileName(); + // Update the "Name: 1234.dat" comment + if (this->body.size() >= 2) + { + const ldraw::id_t id = this->body[1]->id; + if (this->isA(id)) + { + const QString& textBody = this->body[1]->getProperty(); + if (textBody.startsWith("Name: ")) + { + auto editor = this->edit(); + editor.setObjectProperty(id, "Name: " + this->header.name); + } + } + } +} + +/** * @brief Gets the GL polygons of the object at the specified position in the model * @param index Index of object in the model * @param polygons_out Vector to add polygons into @@ -270,16 +304,60 @@ /** * @brief Write out the model as text */ -void Model::save() const +bool Model::save(QTextStream &errors) const { - QFile file{this->path}; + QFile file{this->storedPath}; if (file.open(QIODevice::WriteOnly)) { QTextStream out{&file}; for (const ModelObjectPointer& object : this->body) { - out << object.get()->textRepresentation() << "\r\n"; + out << object.get()->toLDrawCode() << "\r\n"; } file.close(); + if (out.status() != QTextStream::Ok) + { + errors << tr("Write error while writing to %1").arg(this->storedPath); + return false; + } + else + { + return true; + } + } + else + { + errors << tr("Could not open %1 for writing: %2") + .arg(this->storedPath) + .arg(file.errorString()); + return false; + } +} + +/** + * @brief Modifies the !LDRAW_ORG line so that it becomes unofficial + */ +void Model::makeUnofficial() +{ + if (this->body.size() >= 4) + { + const ldraw::id_t id = this->body[3]->id; + if (this->isA(id)) + { + const QString& body = this->body[3]->getProperty(); + if (body.startsWith("!LDRAW_ORG ") and not body.startsWith("!LDRAW_ORG Unofficial_")) + { + QStringList tokens = body.split(" "); + tokens[1] = "Unofficial_" + tokens[1]; + // Remove the UPDATE tag + if (tokens.size() >= 4 && tokens[2] == "UPDATE") + { + tokens.removeAt(3); + tokens.removeAt(2); + } + EditContext editor = this->edit(); + editor.setObjectProperty(id, tokens.join(" ")); + } + } } } \ No newline at end of file