From 2da2eedc0e477f94827b026aab0545521a8b0b34 Mon Sep 17 00:00:00 2001
From: Geoff Hutchison <geoff.hutchison@gmail.com>
Date: Wed, 20 Sep 2023 18:32:15 -0400
Subject: [PATCH] Add a preview panel when PNG images are present for molecules

Signed-off-by: Geoff Hutchison <geoff.hutchison@gmail.com>
---
 .../insertfragment/insertfragmentdialog.cpp   | 36 +++++++++++++++++--
 .../insertfragment/insertfragmentdialog.h     |  8 +++--
 .../insertfragment/insertfragmentdialog.ui    | 19 +++++++++-
 3 files changed, 56 insertions(+), 7 deletions(-)

diff --git a/avogadro/qtplugins/insertfragment/insertfragmentdialog.cpp b/avogadro/qtplugins/insertfragment/insertfragmentdialog.cpp
index 78a96cce52..800b0f1ebd 100644
--- a/avogadro/qtplugins/insertfragment/insertfragmentdialog.cpp
+++ b/avogadro/qtplugins/insertfragment/insertfragmentdialog.cpp
@@ -48,8 +48,8 @@ class InsertFragmentDialog::Private
 
   ~Private()
   {
-    
-      delete model; // proxy is handled through the model
+
+    delete model; // proxy is handled through the model
   }
 };
 
@@ -60,6 +60,8 @@ InsertFragmentDialog::InsertFragmentDialog(QWidget* aParent, QString directory,
 {
   setWindowFlags(Qt::Dialog | Qt::Tool);
   m_ui->setupUi(this);
+  // to start, hide the preview
+  m_ui->preview->hide();
 
   m_implementation->currentFileName.clear();
   if (directory.contains(QLatin1String("crystals")))
@@ -112,6 +114,13 @@ InsertFragmentDialog::InsertFragmentDialog(QWidget* aParent, QString directory,
   m_implementation->model->setReadOnly(true);
   QModelIndex rootIndex = m_implementation->model->setRootPath(m_directory);
 
+  QStringList filters;
+  if (m_implementation->crystalFiles)
+    filters << "*.cif";
+  else
+    filters << "*.cjson";
+  m_implementation->model->setNameFilters(filters);
+
   m_implementation->proxy = new SortFilterTreeProxyModel(this);
   m_implementation->proxy->setSourceModel(m_implementation->model);
   m_implementation->proxy->setSortLocaleAware(true); // important for files
@@ -144,6 +153,9 @@ InsertFragmentDialog::InsertFragmentDialog(QWidget* aParent, QString directory,
 
   connect(m_ui->filterLineEdit, SIGNAL(textChanged(const QString&)), this,
           SLOT(filterTextChanged(const QString&)));
+
+  connect(m_ui->directoryTreeView, SIGNAL(clicked(const QModelIndex&)), this,
+          SLOT(clicked(const QModelIndex&)));
 }
 
 InsertFragmentDialog::~InsertFragmentDialog()
@@ -169,6 +181,24 @@ QString InsertFragmentDialog::fileName()
   return selected.first().data(QFileSystemModel::FilePathRole).toString();
 }
 
+void InsertFragmentDialog::clicked(const QModelIndex& selected)
+{
+  if (m_implementation == nullptr || m_implementation->model == nullptr)
+    return;
+
+  // Remember to map to the source model
+  QString fileName = selected.data(QFileSystemModel::FilePathRole).toString();
+  QFileInfo info(fileName);
+  if (!info.isDir()) {
+    // get the PNG name
+    QString pngName = info.absolutePath() + '/' + info.baseName() + ".png";
+
+    m_ui->preview->setIcon(QIcon(pngName));
+    m_ui->preview->show();
+  } else
+    m_ui->preview->hide();
+}
+
 void InsertFragmentDialog::refresh()
 {
   m_ui->directoryTreeView->update();
@@ -201,4 +231,4 @@ void InsertFragmentDialog::activated()
   emit performInsert(currentFileName, m_implementation->crystalFiles);
 }
 
-} // namespace Avogadro
+} // namespace Avogadro::QtPlugins
diff --git a/avogadro/qtplugins/insertfragment/insertfragmentdialog.h b/avogadro/qtplugins/insertfragment/insertfragmentdialog.h
index e870148e4f..edc810c8de 100644
--- a/avogadro/qtplugins/insertfragment/insertfragmentdialog.h
+++ b/avogadro/qtplugins/insertfragment/insertfragmentdialog.h
@@ -33,18 +33,20 @@ class InsertFragmentDialog : public QDialog
 public Q_SLOTS:
   void refresh();
 
-  void filterTextChanged(const QString &);
+  void filterTextChanged(const QString&);
 
   void activated();
 
+  void clicked(const QModelIndex& selected);
+
 Q_SIGNALS:
-  void performInsert(const QString &fileName, bool crystal);
+  void performInsert(const QString& fileName, bool crystal);
 
 private:
   Ui::InsertFragmentDialog* m_ui;
 
   class Private;
-  Private *m_implementation;
+  Private* m_implementation;
 };
 
 } // namespace QtPlugins
diff --git a/avogadro/qtplugins/insertfragment/insertfragmentdialog.ui b/avogadro/qtplugins/insertfragment/insertfragmentdialog.ui
index 7ce1b05ac7..cc3313318a 100644
--- a/avogadro/qtplugins/insertfragment/insertfragmentdialog.ui
+++ b/avogadro/qtplugins/insertfragment/insertfragmentdialog.ui
@@ -39,7 +39,24 @@
     </layout>
    </item>
    <item>
-    <widget class="QTreeView" name="directoryTreeView"/>
+    <layout class="QGridLayout" name="gridLayout">
+     <item row="0" column="0">
+      <widget class="QTreeView" name="directoryTreeView"/>
+     </item>
+     <item row="0" column="1">
+      <widget class="QToolButton" name="preview">
+       <property name="text">
+        <string/>
+       </property>
+       <property name="iconSize">
+        <size>
+         <width>128</width>
+         <height>128</height>
+        </size>
+       </property>
+      </widget>
+     </item>
+    </layout>
    </item>
    <item>
     <widget class="Line" name="line_2">