parameters for export figure can now be controlled using export fig window popup, all functionality working for X-Y plot export for two traces

Author: SarthakJariwala

PythonGUI_apps/Lifetime_analysis/Lifetime_plot_fit.py

@@ -67,7 +67,7 @@ def __init__(self):
         self.ui.plot_pushButton.clicked.connect(self.plot)
         self.ui.fit_pushButton.clicked.connect(self.call_fit_and_plot)
         self.ui.clear_pushButton.clicked.connect(self.clear_plot)
-        self.ui.export_plot_pushButton.clicked.connect(self.pub_ready_plot_export)
+        self.ui.export_plot_pushButton.clicked.connect(self.export_window)#pub_ready_plot_export
         self.ui.calculate_srv_pushButton.clicked.connect(self.calculate_srv)
 
         self.ui.log_checkBox.stateChanged.connect(self.make_semilog)
@@ -518,6 +518,10 @@ def export_data(self):
 
     def clear_export_data(self):
         self.data_list = []
+    
+    def export_window(self):
+        self.exportplotwindow = ExportPlotWindow()
+        self.exportplotwindow.export_fig_signal.connect(self.pub_ready_plot_export)
 
     def pub_ready_plot_export(self):
         try:
@@ -526,19 +530,27 @@ def pub_ready_plot_export(self):
             plt.figure(figsize=(8,6))
             plt.tick_params(direction='out', length=8, width=3.5)
             if self.ui.save_w_fit_checkBox.isChecked():
-                plt.plot(self.out[:,0],self.out[:,1]/np.max(self.out[:,1]))
-                plt.plot(self.out[:,0],self.out[:,2]/np.max(self.out[:,1]),'k')
+                plt.plot(self.out[:,0],self.out[:,1]/np.max(self.out[:,1]),self.exportplotwindow.ui.traceColor_comboBox.currentText())
+                plt.plot(self.out[:,0],self.out[:,2]/np.max(self.out[:,1]),self.exportplotwindow.ui.fitColor_comboBox.currentText())
+                if self.exportplotwindow.ui.legend_checkBox.isChecked():
+                    plt.legend([self.exportplotwindow.ui.legend1_lineEdit.text(),self.exportplotwindow.ui.legend2_lineEdit.text()])
             else:
-                plt.plot(self.acquire_settings()[0],self.acquire_settings()[1]/np.max(self.acquire_settings()[1]))
+                plt.plot(self.acquire_settings()[0],self.acquire_settings()[1]/np.max(self.acquire_settings()[1]),
+                         self.exportplotwindow.ui.traceColor_comboBox.currentText())
+                if self.exportplotwindow.ui.legend_checkBox.isChecked():
+                    plt.legend([self.exportplotwindow.ui.legend1_lineEdit.text()])
             plt.yscale('log')
             plt.xlabel("Time (ns)", fontsize=20, fontweight='bold')
             plt.ylabel("Intensity (norm.)", fontsize=20, fontweight='bold')
             plt.tight_layout()
+            plt.xlim([self.exportplotwindow.ui.lowerX_spinBox.value(),self.exportplotwindow.ui.upperX_spinBox.value()])
+            plt.ylim([10**(self.exportplotwindow.ui.lowerY_spinBox.value()),self.exportplotwindow.ui.upperY_doubleSpinBox.value()])
 
             plt.savefig(filename[0],bbox_inches='tight', dpi=300)
             plt.close()
 
-        except:
+        except Exception as e:
+            self.ui.Result_textBrowser.append(format(e))
             pass
 
     def close_application(self):
@@ -572,6 +584,38 @@ def done(self):
         self.skip_rows_signal.emit()
         self.close()
 
+"""Export plot GUI"""
+ui_file_path = (base_path / "export_plot.ui").resolve()
+export_WindowTemplate, export_TemplateBaseClass = pg.Qt.loadUiType(ui_file_path)
+
+class ExportPlotWindow(export_TemplateBaseClass):
+    
+    export_fig_signal = QtCore.pyqtSignal()
+    
+    def __init__(self):
+        export_TemplateBaseClass.__init__(self)
+        
+        self.ui = export_WindowTemplate()
+        self.ui.setupUi(self)
+        self.ui.traceColor_comboBox.addItems(["C0","C1","C2","C3","C4","C5","C6","C7", "r", "g", "b", "y", "k"])
+        self.ui.fitColor_comboBox.addItems(["k", "r", "b", "y", "g","C0","C1","C2","C3","C4","C5","C6","C7"])
+        self.ui.export_pushButton.clicked.connect(self.export)
+        self.ui.legend_checkBox.stateChanged.connect(self.legend_title)
+        self.show()
+    
+    def legend_title(self):
+        if self.ui.legend_checkBox.isChecked():
+            self.ui.legend1_lineEdit.setEnabled(True)
+            self.ui.legend2_lineEdit.setEnabled(True)
+        else:
+            self.ui.legend1_lineEdit.setEnabled(False)
+            self.ui.legend2_lineEdit.setEnabled(False)
+    
+    def export(self):
+        self.export_fig_signal.emit()
+        self.close()
+
+
 def run():
     win = MainWindow()
     QtGui.QApplication.instance().exec_()

PythonGUI_apps/Lifetime_analysis/export_plot.ui

@@ -6,8 +6,8 @@
    <rect>
     <x>0</x>
     <y>0</y>
-    <width>573</width>
-    <height>322</height>
+    <width>714</width>
+    <height>435</height>
    </rect>
   </property>
   <property name="windowTitle">
@@ -167,7 +167,19 @@
      </property>
     </widget>
    </item>
-   <item row="3" column="3">
+   <item row="5" column="0" colspan="5">
+    <widget class="QPushButton" name="export_pushButton">
+     <property name="font">
+      <font>
+       <pointsize>17</pointsize>
+      </font>
+     </property>
+     <property name="text">
+      <string>Export Graph</string>
+     </property>
+    </widget>
+   </item>
+   <item row="4" column="3">
     <widget class="QLabel" name="label_11">
      <property name="font">
       <font>
@@ -179,8 +191,11 @@
      </property>
     </widget>
    </item>
-   <item row="3" column="2">
+   <item row="4" column="2">
     <widget class="QLineEdit" name="legend1_lineEdit">
+     <property name="enabled">
+      <bool>false</bool>
+     </property>
      <property name="font">
       <font>
        <pointsize>12</pointsize>
@@ -218,16 +233,19 @@
      </property>
     </widget>
    </item>
-   <item row="3" column="4">
+   <item row="4" column="4">
     <widget class="QLineEdit" name="legend2_lineEdit">
+     <property name="enabled">
+      <bool>false</bool>
+     </property>
      <property name="font">
       <font>
        <pointsize>12</pointsize>
       </font>
      </property>
     </widget>
    </item>
-   <item row="3" column="0">
+   <item row="4" column="0">
     <widget class="QLabel" name="label_12">
      <property name="font">
       <font>
@@ -239,7 +257,7 @@
      </property>
     </widget>
    </item>
-   <item row="3" column="1">
+   <item row="4" column="1">
     <widget class="QLabel" name="label_10">
      <property name="font">
       <font>
@@ -251,15 +269,15 @@
      </property>
     </widget>
    </item>
-   <item row="4" column="0" colspan="5">
-    <widget class="QPushButton" name="pushButton">
+   <item row="3" column="0" colspan="2">
+    <widget class="QCheckBox" name="legend_checkBox">
      <property name="font">
       <font>
-       <pointsize>17</pointsize>
+       <pointsize>15</pointsize>
       </font>
      </property>
      <property name="text">
-      <string>Export Graph</string>
+      <string>Add Legends to the Plot</string>
      </property>
     </widget>
    </item>