Connect to molecule changes (e.g., in basis sets)

Signed-off-by: Geoff Hutchison <[email protected]>
ghutchis · Nov 27, 2023 · 55628fb · 55628fb
1 parent 323086e
commit 55628fb
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/avogadro/qtplugins/surfaces/surfaces.cpp b/avogadro/qtplugins/surfaces/surfaces.cpp
@@ -224,6 +224,10 @@ bool Surfaces::handleCommand(const QString& command, const QVariantMap& options)
 
 void Surfaces::setMolecule(QtGui::Molecule* mol)
 {
+  if (m_molecule != nullptr) {
+    m_molecule->disconnect(this);
+  }
+
   if (mol->basisSet()) {
     m_basis = mol->basisSet();
   } else if (mol->cubes().size() != 0) {
@@ -234,6 +238,18 @@ void Surfaces::setMolecule(QtGui::Molecule* mol)
   m_mesh1 = nullptr;
   m_mesh2 = nullptr;
   m_molecule = mol;
+
+  if (m_molecule != nullptr) {
+    connect(m_molecule, SIGNAL(changed(uint)), SLOT(moleculeChanged(uint)));
+  }
+}
+
+void Surfaces::moleculeChanged(unsigned int changes)
+{
+  if (changes & Molecule::Added || changes & Molecule::Removed) {
+    m_cubes = m_molecule->cubes();
+    m_basis = m_molecule->basisSet();
+  }
 }
 
 QList<QAction*> Surfaces::actions() const

diff --git a/avogadro/qtplugins/surfaces/surfaces.h b/avogadro/qtplugins/surfaces/surfaces.h
@@ -85,6 +85,7 @@ class Surfaces : public QtGui::ExtensionPlugin
 public slots:
   bool handleCommand(const QString& command,
                      const QVariantMap& options) override;
+  void moleculeChanged(unsigned int changes);
 
 private slots:
   void surfacesActivated();