/* Copyright 2006-2026 The QElectroTech Team This file is part of QElectroTech. QElectroTech 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 2 of the License, or (at your option) any later version. QElectroTech 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 QElectroTech. If not, see . */ #include "qetscriptapi.h" #include "../ElementsCollection/elementslocation.h" #include "../QPropertyUndoCommand/qpropertyundocommand.h" #include "../cli_export.h" #include "../diagram.h" #include "../diagramcontent.h" #include "../diagramview.h" #include "../factory/elementfactory.h" #include "../qet.h" #include "../qetgraphicsitem/element.h" #include "../qetmessagebox.h" #include "../qetproject.h" #include "../qetresult.h" #include "../qetgraphicsitem/conductor.h" #include "../qetgraphicsitem/terminal.h" #include "../qetinformation.h" #include "../titleblockproperties.h" #include "../undocommand/addgraphicsobjectcommand.h" #include "../undocommand/changeelementinformationcommand.h" #include "../undocommand/changetitleblockcommand.h" #include "../undocommand/deleteqgraphicsitemcommand.h" #include "../undocommand/linkelementcommand.h" #include "../utils/conductorcreator.h" #include #include QetScriptApi::QetScriptApi(QETProject *project, DiagramView *view, QObject *parent) : QObject(parent), m_project(project), m_view(view) { } QString QetScriptApi::projectTitle() const { return m_project ? m_project->title() : QString(); } QString QetScriptApi::filePath() const { return m_project ? m_project->filePath() : QString(); } int QetScriptApi::folioCount() const { return m_project ? m_project->diagrams().count() : 0; } QString QetScriptApi::folioTitle(int index) const { if (!m_project) return QString(); const QList diagrams = m_project->diagrams(); if (index < 0 || index >= diagrams.count()) return QString(); return diagrams.at(index)->title(); } int QetScriptApi::elementCount(int folioIndex) const { if (!m_project) return 0; const QList diagrams = m_project->diagrams(); if (folioIndex < 0 || folioIndex >= diagrams.count()) return 0; DiagramContent content(diagrams.at(folioIndex), false); return content.m_elements.count(); } int QetScriptApi::conductorCount(int folioIndex) const { if (!m_project) return 0; const QList diagrams = m_project->diagrams(); if (folioIndex < 0 || folioIndex >= diagrams.count()) return 0; DiagramContent content(diagrams.at(folioIndex), false); return content.conductors(DiagramContent::AnyConductor).count(); } /** @brief QetScriptApi::runFlag Build a CLIExport::run() argument list from this project's own file path plus @p args, and run it. Reopens the project from disk -- see the class comment for why that trade-off was made. @param flag one of the --export-* / --resave / --set-titleblock flags @param args the flag's own positional arguments (output path, etc.) @return true if CLIExport::run() returned 0 (success) */ bool QetScriptApi::runFlag(const QString &flag, const QStringList &args) { if (!m_project) { log(QStringLiteral("qet.%1: no project").arg(flag)); return false; } const QString path = m_project->filePath(); if (path.isEmpty()) { log(QStringLiteral("qet.%1: project has no file path -- save it first").arg(flag)); return false; } QStringList full_args; full_args << flag << path << args; if (m_view) { // Interactive "Run Script...": CLIExport::run() opens a second, // temporary QETProject on the same file the GUI already has open. // Backups are only disabled on the headless --run path (main.cpp); // here that second, short-lived project would otherwise manage its // own KAutoSaveFile for the same path as the user's real, already // open project, racing it and risking a stale-restore prompt next // launch. Disable backups for just this one export and restore // them right after -- the headless path must never see this // change, since it depends on backups staying off for the whole // run (see the crash note next to the other setBackupEnabled(false) // call in main.cpp). QETProject::setBackupEnabled(false); const int result = CLIExport::run(full_args); QETProject::setBackupEnabled(true); return result == 0; } return CLIExport::run(full_args) == 0; } bool QetScriptApi::exportPdf(const QString &output, bool showTerminals) { QStringList args{output}; if (showTerminals) args << QStringLiteral("--show-terminals"); return runFlag(QStringLiteral("--export-pdf"), args); } bool QetScriptApi::exportPng(const QString &outDir, bool showTerminals) { QStringList args{outDir}; if (showTerminals) args << QStringLiteral("--show-terminals"); return runFlag(QStringLiteral("--export-png"), args); } bool QetScriptApi::exportSvg(const QString &outDir, bool showTerminals) { QStringList args{outDir}; if (showTerminals) args << QStringLiteral("--show-terminals"); return runFlag(QStringLiteral("--export-svg"), args); } bool QetScriptApi::exportCables(const QString &output) { return runFlag(QStringLiteral("--export-cables"), {output}); } bool QetScriptApi::exportWires(const QString &output) { return runFlag(QStringLiteral("--export-wires"), {output}); } bool QetScriptApi::exportBom(const QString &output) { return runFlag(QStringLiteral("--export-bom"), {output}); } bool QetScriptApi::exportWiring(const QString &output) { return runFlag(QStringLiteral("--export-wiring"), {output}); } bool QetScriptApi::exportNets(const QString &output) { return runFlag(QStringLiteral("--export-nets"), {output}); } bool QetScriptApi::exportLinks(const QString &output) { return runFlag(QStringLiteral("--export-links"), {output}); } bool QetScriptApi::exportInfo(const QString &output) { return runFlag(QStringLiteral("--info"), output.isEmpty() ? QStringList{} : QStringList{output}); } bool QetScriptApi::setTitleBlock(const QString &output, const QStringList &assignments) { QStringList args{output}; args << assignments; return runFlag(QStringLiteral("--set-titleblock"), args); } /** @brief QetScriptApi::save Write this project's own current in-memory state -- unlike every export method above, this does NOT go through runFlag()/CLIExport::run(): that reopens the project fresh from its file on disk, which would never see any addElement()/setElementPosition()/moveElement()/deleteElement() this script made, only silently rewrite the file exactly as it already was. Caught by testing this against a script that added an element and called save(): the element counted correctly in memory but the saved file didn't have it, since the old implementation opened an unrelated, unmodified second copy of the project to write. With no output path, this goes through QETProject::write() -- the same path "Enregistrer" uses -- so it honours read-only mode the same way, updates saveddate/savedtime, and clears the modified flag. A plain QFile write used to skip all of that and could tear the file on an interruption, where write() goes through QET::writeXmlFile()'s QSaveFile. With an explicit output path this is a save to a different file, so it goes through QET::writeXmlFile() directly without touching the project's own filePath() or read-only state, the same way "Save As" targeting a writable location is allowed even for a project opened read-only. @param output path to write to; the project's own file path if empty @return false if there is no output path to use, or the write failed */ bool QetScriptApi::save(const QString &output) { if (!m_project) return false; if (output.isEmpty()) { if (m_project->filePath().isEmpty()) { log(QStringLiteral("qet.save: no output path given and the project has none of its own -- pass one")); return false; } const QETResult result = m_project->write(); if (!result.isOk()) { log(QStringLiteral("qet.save: %1").arg(result.errorMessage())); return false; } return true; } QDomDocument xml_doc(m_project->toXml()); QString error_message; if (!QET::writeXmlFile(xml_doc, output, &error_message)) { log(QStringLiteral("qet.save: %1").arg(error_message)); return false; } return true; } void QetScriptApi::log(const QString &message) { QTextStream(stderr) << message << "\n"; } /** @brief QetScriptApi::findElement @param folioIndex @param elementUuid as returned by addElement(), or read from the element's own uuid attribute @return the matching element on that folio, or nullptr */ Element *QetScriptApi::findElement(int folioIndex, const QString &elementUuid) const { if (!m_project) return nullptr; const QList diagrams = m_project->diagrams(); if (folioIndex < 0 || folioIndex >= diagrams.count()) return nullptr; DiagramContent content(diagrams.at(folioIndex), false); for (Element *elmt : std::as_const(content.m_elements)) { if (elmt->uuid().toString() == elementUuid) return elmt; } return nullptr; } Terminal *QetScriptApi::findTerminal(int folioIndex, const QString &elementUuid, int terminalIndex, const QString &caller) { Element *element = findElement(folioIndex, elementUuid); if (!element) { log(QStringLiteral("qet.%1: no element %2 on folio %3").arg(caller, elementUuid).arg(folioIndex)); return nullptr; } const QList terminals = element->terminals(); if (terminalIndex < 0 || terminalIndex >= terminals.count()) { log(QStringLiteral("qet.%1: %2 has %3 terminal(s), no index %4") .arg(caller, element->name()).arg(terminals.count()).arg(terminalIndex)); return nullptr; } return terminals.at(terminalIndex); } /** @brief QetScriptApi::findConductor The single conductor attached to a terminal, or nullptr. Conductors carry no persisted uuid, and the terminal1/terminal2 ids the file uses for their ends are folio-scoped integers QElectroTech renumbers on every save, so a conductor has no name that survives a save/load cycle. Naming one by a terminal it is attached to does, and it reads the way the question is usually asked ("the wire on A1 of KM1"). A terminal with several conductors on it does not name one, so refuse rather than silently take the first. */ Conductor *QetScriptApi::findConductor(int folioIndex, const QString &elementUuid, int terminalIndex, const QString &caller) { Terminal *terminal = findTerminal(folioIndex, elementUuid, terminalIndex, caller); if (!terminal) return nullptr; const QList conductors = terminal->conductors(); if (conductors.isEmpty()) { log(QStringLiteral("qet.%1: terminal %2 of %3 has no conductor on it") .arg(caller).arg(terminalIndex).arg(elementUuid)); return nullptr; } if (conductors.count() > 1) { log(QStringLiteral("qet.%1: terminal %2 of %3 carries %4 conductors, so it does " "not name one -- use a terminal with a single conductor") .arg(caller).arg(terminalIndex).arg(elementUuid).arg(conductors.count())); return nullptr; } return conductors.first(); } namespace { /** Read or write one named conductor property. The names are the ones the project file uses for the same fields (ConductorProperties::toXml), so that what a script sets is what a reader of the .qet sees, rather than a third spelling invented here. */ QString conductorPropertyValue(const ConductorProperties &p, const QString &name) { if (name == QLatin1String("num")) return p.text; if (name == QLatin1String("formula")) return p.m_formula; if (name == QLatin1String("function")) return p.m_function; if (name == QLatin1String("bus")) return p.m_bus; if (name == QLatin1String("cable")) return p.m_cable; if (name == QLatin1String("tension_protocol")) return p.m_tension_protocol; if (name == QLatin1String("conductor_color")) return p.m_wire_color; if (name == QLatin1String("conductor_section")) return p.m_wire_section; if (name == QLatin1String("color")) return p.color.name(); if (name == QLatin1String("text_color")) return p.text_color.name(); return QString(); } bool setConductorPropertyValue(ConductorProperties &p, const QString &name, const QString &value) { if (name == QLatin1String("num")) { p.text = value; return true; } if (name == QLatin1String("formula")) { p.m_formula = value; return true; } if (name == QLatin1String("function")) { p.m_function = value; return true; } if (name == QLatin1String("bus")) { p.m_bus = value; return true; } if (name == QLatin1String("cable")) { p.m_cable = value; return true; } if (name == QLatin1String("tension_protocol")) { p.m_tension_protocol = value; return true; } if (name == QLatin1String("conductor_color")) { p.m_wire_color = value; return true; } if (name == QLatin1String("conductor_section")) { p.m_wire_section = value; return true; } // The two real colours are QColor, not free text: an unparseable name // would otherwise be stored as an invalid colour and drawn as black. if (name == QLatin1String("color") || name == QLatin1String("text_color")) { const QColor c(value); if (!c.isValid()) return false; if (name == QLatin1String("color")) p.color = c; else p.text_color = c; return true; } return false; } const QStringList &conductorPropertyNames() { static const QStringList names { QStringLiteral("num"), QStringLiteral("formula"), QStringLiteral("function"), QStringLiteral("bus"), QStringLiteral("cable"), QStringLiteral("tension_protocol"), QStringLiteral("conductor_color"), QStringLiteral("conductor_section"), QStringLiteral("color"), QStringLiteral("text_color")}; return names; } } // namespace bool QetScriptApi::setInfoKey(int folioIndex, const QString &elementUuid, const QString &key, const QString &value, const QString &caller) { if (!m_project) return false; if (m_project->isReadOnly()) { log(QStringLiteral("qet.%1: project is read-only").arg(caller)); return false; } if (key.isEmpty()) { log(QStringLiteral("qet.%1: empty information key").arg(caller)); return false; } Element *element = findElement(folioIndex, elementUuid); if (!element) return false; const DiagramContext old_info = element->elementInformations(); if (old_info.value(key).toString() == value) return true; // nothing to push DiagramContext new_info = old_info; new_info.addValue(key, value); auto *cmd = new ChangeElementInformationCommand(element, old_info, new_info); m_project->undoStack()->push(cmd); return true; } /** @brief QetScriptApi::addElement Place a new element on a folio, through the same AddGraphicsObjectCommand the interactive drag-from-collection-panel path uses (see DiagramEventAddElement::addElement()) -- so Ctrl+Z undoes it exactly as it would a manually dropped element. Like that path, the element is first imported into the project's own embedded collection (QETProject::importElement()): building straight from a common://custom:// location without embedding it left the saved .qet referencing a definition outside the project, missing on any machine that doesn't have that same collection installed. importElement() can, on a name collision with a different, already embedded element, pop a modal dialog asking the user to choose -- with nobody there to answer it in a script, headless or interactive, that is exactly the hang class this whole API is built to avoid (see the class comment). Detected and refused before it can happen, rather than risked. @param folioIndex @param locationPath an element collection path, e.g. "embed://some/path.elmt" or "common://10_electric/...elmt" @param x @param y target position, in the diagram's own coordinates @return the new element's uuid (empty string on failure -- read-only project, bad folio index, an import collision, or the location could not be resolved/built) */ QString QetScriptApi::addElement(int folioIndex, const QString &locationPath, double x, double y) { if (!m_project) return QString(); if (m_project->isReadOnly()) { log(QStringLiteral("qet.addElement: project is read-only")); return QString(); } const QList diagrams = m_project->diagrams(); if (folioIndex < 0 || folioIndex >= diagrams.count()) return QString(); Diagram *diagram = diagrams.at(folioIndex); // ElementsLocation::setPath() forces ANY path to embed:// as soon as a // non-null project is given, common://custom:// included -- passing // m_project unconditionally here (as the pre-fix code did) silently // turned every non-embed:: locationPath into a lookup for an embedded // element that was never there. Caught by actually running this // against a common:// path: "does not resolve to an element" even // though the file plainly exists. ElementsLocation location = locationPath.startsWith(QStringLiteral("embed://")) ? ElementsLocation(locationPath, m_project) : ElementsLocation(locationPath); if (!location.isElement() || !location.exist()) { log(QStringLiteral("qet.addElement: '%1' does not resolve to an element").arg(locationPath)); return QString(); } ElementsLocation import_location = location; if (!(location.isProject() && location.project() == m_project)) { const QString import_path = location.isFileSystem() ? QStringLiteral("import/") + location.collectionPath(false) : location.collectionPath(false); const ElementsLocation existing(import_path, m_project); if (existing.exist() && existing.uuid() != location.uuid()) { log(QStringLiteral("qet.addElement: '%1' would collide with a different element " "already embedded under the same name -- refusing rather than " "risk the interactive import-conflict dialog").arg(locationPath)); return QString(); } import_location = m_project->importElement(location); if (!import_location.exist()) { log(QStringLiteral("qet.addElement: could not import '%1' into the project").arg(locationPath)); return QString(); } } int state = 0; Element *element = ElementFactory::Instance()->createElement(import_location, nullptr, &state); if (state) { delete element; log(QStringLiteral("qet.addElement: could not build element from '%1'").arg(locationPath)); return QString(); } const QPointF pos(x, y); element->setPos(pos); diagram->addItem(element); auto *undo_group = new QUndoCommand(QObject::tr("Ajouter %1").arg(element->name())); new AddGraphicsObjectCommand(element, diagram, pos, undo_group); diagram->undoStack().push(undo_group); return element->uuid().toString(); } bool QetScriptApi::setElementPosition(int folioIndex, const QString &elementUuid, double x, double y) { if (m_project && m_project->isReadOnly()) { log(QStringLiteral("qet.setElementPosition: project is read-only")); return false; } Element *element = findElement(folioIndex, elementUuid); if (!element) return false; const QVariant old_value = element->pos(); const QVariant new_value = QPointF(x, y); if (old_value == new_value) return true; // already there; nothing to push auto *cmd = new QPropertyUndoCommand(element, "pos", old_value, new_value); cmd->setText(QObject::tr("Déplacer %1").arg(element->name())); m_project->undoStack()->push(cmd); return true; } bool QetScriptApi::moveElement(int folioIndex, const QString &elementUuid, double dx, double dy) { Element *element = findElement(folioIndex, elementUuid); if (!element) return false; const QPointF p = element->pos(); return setElementPosition(folioIndex, elementUuid, p.x() + dx, p.y() + dy); } /** @brief QetScriptApi::deleteElement Through DeleteQGraphicsItemCommand -- the same command the Delete key uses -- so any conductors attached to the element's terminals are cleaned up the same way, not left dangling. */ bool QetScriptApi::deleteElement(int folioIndex, const QString &elementUuid) { if (!m_project) return false; if (m_project->isReadOnly()) { log(QStringLiteral("qet.deleteElement: project is read-only")); return false; } Element *element = findElement(folioIndex, elementUuid); if (!element) return false; Diagram *diagram = m_project->diagrams().at(folioIndex); DiagramContent content; content.m_elements << element; if (DeleteQGraphicsItemCommand::hasNonDeletableTerminal(content)) { log(QStringLiteral("qet.deleteElement: %1 has a non-deletable terminal (linked master/slave?), refusing").arg(elementUuid)); return false; } auto *cmd = new DeleteQGraphicsItemCommand(diagram, content); diagram->undoStack().push(cmd); return true; } bool QetScriptApi::rotateElement(int folioIndex, const QString &elementUuid, double angle) { if (m_project && m_project->isReadOnly()) { log(QStringLiteral("qet.rotateElement: project is read-only")); return false; } Element *element = findElement(folioIndex, elementUuid); if (!element) return false; // The same property command RotateSelectionCommand pushes for an // Element -- deliberately not RotateSelectionCommand itself, which // works on diagram->selectedItems() and would mean quietly rewriting // the user's selection to rotate one element by uuid. For a single // element the two are mechanically identical: that class special-cases // Element::Type to exactly this one command, and only adds a second, // positional one when rotating a multi-item selection as a group. auto *cmd = new QPropertyUndoCommand(element, "rotation", QVariant(element->rotation()), QVariant(element->rotation() + angle)); cmd->setText(QObject::tr("Pivoter %1").arg(element->name())); m_project->undoStack()->push(cmd); return true; } QStringList QetScriptApi::elementUuids(int folioIndex) const { QStringList uuids; if (!m_project) return uuids; const QList diagrams = m_project->diagrams(); if (folioIndex < 0 || folioIndex >= diagrams.count()) return uuids; DiagramContent content(diagrams.at(folioIndex), false); for (Element *elmt : std::as_const(content.m_elements)) { uuids << elmt->uuid().toString(); } return uuids; } QString QetScriptApi::elementName(int folioIndex, const QString &elementUuid) const { Element *element = findElement(folioIndex, elementUuid); return element ? element->name() : QString(); } /** @brief QetScriptApi::elementTerminals The element's terminals, in the order addConductor() indexes them: one entry per terminal, ": ( conductor(s))". Descriptive rather than structured because its only job is to let a script -- or a human reading a script's output -- see which index is which before wiring anything to it. Indexes, not uuids, because a terminal uuid does not address a terminal on a folio. Terminal::uuid() comes from the catalog .elmt definition (see Terminal::stableUuid()), so it is empty for most of the installed base, and where it is not, every instance of that same element carries the same one -- two coils of one type placed side by side have byte-identical terminal uuids, which is plainly visible in the saved file of any project written through this API. The order of Element::terminals() also comes from the definition, but it is at least unambiguous within the element the caller has already named by uuid. */ QStringList QetScriptApi::elementTerminals(int folioIndex, const QString &elementUuid) const { QStringList list; Element *element = findElement(folioIndex, elementUuid); if (!element) return list; const QList terminals = element->terminals(); for (int i = 0 ; i < terminals.count() ; ++i) { Terminal *t = terminals.at(i); list << QStringLiteral("%1: %2 (%3 conductor(s))") .arg(i) .arg(t->name().isEmpty() ? QStringLiteral("-") : t->name()) .arg(t->conductorsCount()); } return list; } QString QetScriptApi::elementInfo(int folioIndex, const QString &elementUuid, const QString &key) const { Element *element = findElement(folioIndex, elementUuid); if (!element) return QString(); return element->elementInformations().value(key).toString(); } bool QetScriptApi::setElementInfo(int folioIndex, const QString &elementUuid, const QString &key, const QString &value) { return setInfoKey(folioIndex, elementUuid, key, value, QStringLiteral("setElementInfo")); } QString QetScriptApi::elementLabel(int folioIndex, const QString &elementUuid) const { return elementInfo(folioIndex, elementUuid, QETInformation::ELMT_LABEL); } bool QetScriptApi::setElementLabel(int folioIndex, const QString &elementUuid, const QString &label) { return setInfoKey(folioIndex, elementUuid, QETInformation::ELMT_LABEL, label, QStringLiteral("setElementLabel")); } /** @brief QetScriptApi::addConductor Wire terminal terminalIndexA of one element to terminalIndexB of another, on the same folio, through ConductorCreator -- the same class the "draw a selection rectangle over terminals" GUI path uses. Going through it rather than constructing a Conductor directly is what makes the new conductor inherit an existing potential's properties and take part in conductor auto-numbering; a hand-built one would be silently outside both. Refuses, rather than creating anything, when the two terminals sit on two different existing potentials: ConductorCreator then has to ask which one's properties the new conductor should inherit, and it asks with a plain modal QDialog that QET::QetMessageBox's non-interactive mode does not cover -- so under headless --run there would be nobody to answer it and the script would hang forever. Same reasoning, and the same choice, as addElement() makes about the import-conflict dialog. @return true if a conductor was created */ bool QetScriptApi::addConductor(int folioIndex, const QString &elementUuidA, int terminalIndexA, const QString &elementUuidB, int terminalIndexB) { if (!m_project) return false; if (m_project->isReadOnly()) { log(QStringLiteral("qet.addConductor: project is read-only")); return false; } const QString caller = QStringLiteral("addConductor"); Terminal *t1 = findTerminal(folioIndex, elementUuidA, terminalIndexA, caller); Terminal *t2 = findTerminal(folioIndex, elementUuidB, terminalIndexB, caller); if (!t1 || !t2) return false; if (t1 == t2) { log(QStringLiteral("qet.addConductor: both ends are the same terminal")); return false; } if (t1->isLinkedTo(t2)) { log(QStringLiteral("qet.addConductor: those two terminals are already wired together")); return false; } if (!t1->canBeLinkedTo(t2)) { log(QStringLiteral("qet.addConductor: those two terminals cannot be linked")); return false; } const QList terminals {t1, t2}; if (ConductorCreator::needsPotentialChoice(terminals)) { log(QStringLiteral("qet.addConductor: those terminals are on two different existing " "potentials, so creating a conductor would ask which one to inherit " "-- refusing rather than open a dialog no script can answer")); return false; } Diagram *diagram = m_project->diagrams().at(folioIndex); ConductorCreator creator(diagram, terminals); Q_UNUSED(creator) // ConductorCreator has no return value and several ways to decline // quietly, so report what actually happened rather than that it ran. return t1->isLinkedTo(t2); } /** @brief QetScriptApi::conductors One line per conductor on the folio: which terminals it joins and its number, in the form setConductorProperty() addresses them. Descriptive rather than structured for the same reason elementTerminals() is -- it exists so a script, or a person reading its output, can see what is there before changing it. */ QStringList QetScriptApi::conductors(int folioIndex) const { QStringList list; if (!m_project) return list; const QList diagrams = m_project->diagrams(); if (folioIndex < 0 || folioIndex >= diagrams.count()) return list; auto describe = [](Terminal *t) -> QString { if (!t || !t->parentElement()) return QStringLiteral("?"); return QStringLiteral("%1 terminal %2") .arg(t->parentElement()->uuid().toString()) .arg(t->parentElement()->terminals().indexOf(t)); }; DiagramContent content(diagrams.at(folioIndex), false); const QList all = content.conductors(DiagramContent::AnyConductor); for (Conductor *c : all) { list << QStringLiteral("%1 -- %2 : num='%3'") .arg(describe(c->terminal1), describe(c->terminal2), c->properties().text); } return list; } QString QetScriptApi::conductorProperty(int folioIndex, const QString &elementUuid, int terminalIndex, const QString &property) const { // const_cast: findConductor logs, and log() writes to stderr, which is // not a const operation on this object. The lookup itself changes // nothing. auto *self = const_cast(this); Conductor *conductor = self->findConductor(folioIndex, elementUuid, terminalIndex, QStringLiteral("conductorProperty")); if (!conductor) return QString(); return conductorPropertyValue(conductor->properties(), property); } /** @brief QetScriptApi::setConductorProperty Set one property on the conductor attached to a terminal -- and on every other conductor of the same electrical potential. That is not a convenience, it is the rule the application already follows: SearchAndReplaceWorker does exactly this, pushing one QPropertyUndoCommand per conductor of relatedPotentialConductors() inside a single macro, because a wire number, colour or section describes a potential and not one drawn segment. Setting it on one conductor and leaving the rest of the potential disagreeing would produce a file no GUI action could have produced. @return true if anything was changed, or if it already held that value */ bool QetScriptApi::setConductorProperty(int folioIndex, const QString &elementUuid, int terminalIndex, const QString &property, const QString &value) { if (!m_project) return false; const QString caller = QStringLiteral("setConductorProperty"); if (m_project->isReadOnly()) { log(QStringLiteral("qet.%1: project is read-only").arg(caller)); return false; } if (!conductorPropertyNames().contains(property)) { log(QStringLiteral("qet.%1: unknown property '%2'; expected one of %3") .arg(caller, property, conductorPropertyNames().join(QStringLiteral(", ")))); return false; } Conductor *conductor = findConductor(folioIndex, elementUuid, terminalIndex, caller); if (!conductor) return false; ConductorProperties properties = conductor->properties(); if (!setConductorPropertyValue(properties, property, value)) { log(QStringLiteral("qet.%1: '%2' is not a valid value for %3") .arg(caller, value, property)); return false; } if (properties == conductor->properties()) return true; // already so QSet potential = conductor->relatedPotentialConductors(true); potential << conductor; m_project->undoStack()->beginMacro(QObject::tr("Modifier les propriétés du conducteur")); for (Conductor *c : std::as_const(potential)) { QVariant old_value, new_value; old_value.setValue(c->properties()); new_value.setValue(properties); m_project->undoStack()->push(new QPropertyUndoCommand(c, "properties", old_value, new_value)); } m_project->undoStack()->endMacro(); return true; } QString QetScriptApi::elementLinkType(int folioIndex, const QString &elementUuid) const { Element *element = findElement(folioIndex, elementUuid); if (!element) return QString(); switch (element->linkType()) { case Element::Simple: return QStringLiteral("simple"); case Element::NextReport: return QStringLiteral("next_report"); case Element::PreviousReport: return QStringLiteral("previous_report"); case Element::Master: return QStringLiteral("master"); case Element::Slave: return QStringLiteral("slave"); case Element::Terminale: return QStringLiteral("terminal"); default: return QStringLiteral("unknown"); } } QStringList QetScriptApi::linkedElements(int folioIndex, const QString &elementUuid) const { QStringList list; Element *element = findElement(folioIndex, elementUuid); if (!element) return list; const QList linked = element->linkedElements(); for (Element *e : linked) { list << e->uuid().toString(); } return list; } /** @brief QetScriptApi::linkElements Link two elements -- a master to a slave, or one report to its counterpart. Two folio indices because a master and its slave normally sit on different folios; that is the usual case, not the exception. Whether a given pair may be linked is not decided here. LinkElementCommand::isLinkable() already holds those rules -- that a master takes a slave and not another master, that a PLC master pairs only with a PLC slave, that a next-report pairs only with a previous-report, and that the target is free -- and asking it rather than re-deriving them is what keeps a script from producing a link the GUI would refuse to make. */ bool QetScriptApi::linkElements(int folioIndexA, const QString &elementUuidA, int folioIndexB, const QString &elementUuidB) { if (!m_project) return false; if (m_project->isReadOnly()) { log(QStringLiteral("qet.linkElements: project is read-only")); return false; } Element *a = findElement(folioIndexA, elementUuidA); Element *b = findElement(folioIndexB, elementUuidB); if (!a || !b) { log(QStringLiteral("qet.linkElements: %1 does not resolve to an element") .arg(a ? elementUuidB : elementUuidA)); return false; } if (a == b) { log(QStringLiteral("qet.linkElements: an element cannot be linked to itself")); return false; } if (!LinkElementCommand::isLinkable(a, b)) { log(QStringLiteral("qet.linkElements: %1 (%2) cannot be linked to %3 (%4) -- " "check the two link types, and that the target is still free") .arg(elementUuidA, elementLinkType(folioIndexA, elementUuidA), elementUuidB, elementLinkType(folioIndexB, elementUuidB))); return false; } auto *cmd = new LinkElementCommand(a); cmd->setLink(b); m_project->undoStack()->push(cmd); return a->linkedElements().contains(b); } bool QetScriptApi::unlinkElement(int folioIndex, const QString &elementUuid) { if (!m_project) return false; if (m_project->isReadOnly()) { log(QStringLiteral("qet.unlinkElement: project is read-only")); return false; } Element *element = findElement(folioIndex, elementUuid); if (!element) return false; if (element->linkedElements().isEmpty()) return true; // nothing to undo auto *cmd = new LinkElementCommand(element); cmd->unlinkAll(); m_project->undoStack()->push(cmd); return element->linkedElements().isEmpty(); } int QetScriptApi::addFolio() { if (!m_project) return -1; if (m_project->isReadOnly()) { log(QStringLiteral("qet.addFolio: project is read-only")); return -1; } Diagram *diagram = m_project->addNewDiagram(); if (!diagram) return -1; return m_project->diagrams().indexOf(diagram); } bool QetScriptApi::setFolioTitle(int folioIndex, const QString &title) { if (!m_project) return false; if (m_project->isReadOnly()) { log(QStringLiteral("qet.setFolioTitle: project is read-only")); return false; } const QList diagrams = m_project->diagrams(); if (folioIndex < 0 || folioIndex >= diagrams.count()) return false; Diagram *diagram = diagrams.at(folioIndex); // The folio title is one field of the title block properties, so it // changes the way the title block dialog changes it: read the whole // struct, set one member, push the command with both versions. const TitleBlockProperties old_properties = diagram->border_and_titleblock.exportTitleBlock(); if (old_properties.title == title) return true; TitleBlockProperties new_properties = old_properties; new_properties.title = title; auto *cmd = new ChangeTitleBlockCommand(diagram, old_properties, new_properties); m_project->undoStack()->push(cmd); return true; } bool QetScriptApi::undo() { if (!m_project || !m_project->undoStack()->canUndo()) return false; m_project->undoStack()->undo(); return true; } bool QetScriptApi::redo() { if (!m_project || !m_project->undoStack()->canRedo()) return false; m_project->undoStack()->redo(); return true; } bool QetScriptApi::canUndo() const { return m_project && m_project->undoStack()->canUndo(); } bool QetScriptApi::canRedo() const { return m_project && m_project->undoStack()->canRedo(); } bool QetScriptApi::selectElement(const QString &elementUuid) { if (!m_project) return false; for (Diagram *diagram : m_project->diagrams()) { DiagramContent content(diagram, false); for (Element *elmt : std::as_const(content.m_elements)) { if (elmt->uuid().toString() == elementUuid) { elmt->setSelected(true); return true; } } } return false; } void QetScriptApi::deselectAll(int folioIndex) { if (!m_project) return; const QList diagrams = m_project->diagrams(); if (folioIndex < 0 || folioIndex >= diagrams.count()) return; diagrams.at(folioIndex)->clearSelection(); } bool QetScriptApi::zoomFit() { if (!m_view) return false; m_view->zoomFit(); return true; } bool QetScriptApi::zoomToContent() { if (!m_view) return false; m_view->zoomContent(); return true; } bool QetScriptApi::zoomReset() { if (!m_view) return false; m_view->zoomReset(); return true; } /** @brief QetScriptApi::showMessage A modal QET::QetMessageBox::information() -- safe to call headless because non-interactive mode is already set for the whole process before any script runs (main.cpp), which is exactly the mechanism that keeps --export-* and --resave from hanging on a warning dialog. Interactively, a real person is there to see and dismiss it. */ void QetScriptApi::showMessage(const QString &text) { QET::QetMessageBox::information(nullptr, QObject::tr("Script"), text); }