mg-pso-gui 0.1.209__py3-none-any.whl → 0.1.213__py3-none-any.whl

{mg_pso_gui-0.1.209.dist-info → mg_pso_gui-0.1.213.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mg-pso-gui
-Version: 0.1.209
+Version: 0.1.213
 Summary: GUI for MG-PSO
 Author: Robert Cordingly
 Author-email: <rcording@uw.ed>

{mg_pso_gui-0.1.209.dist-info → mg_pso_gui-0.1.213.dist-info}/RECORD

@@ -21,7 +21,7 @@ mgpsogui/gui/SetupTab/StepView.py,sha256=C6DDinU9AgaUM1EOAHetFI4bE89ANJ247F0doks
 mgpsogui/gui/SetupTab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mgpsogui/gui/VisualizeTab/MatrixEditor.py,sha256=JPQmfljdddIwHtTKpFqUdk7hRzVdX-IKl4neOk3DsjM,3132
 mgpsogui/gui/VisualizeTab/SideBar.py,sha256=MtZfDScNFN0fQ8DyPm6bZGJSrFRgT14fZeUq-qc61O8,21936
-mgpsogui/gui/VisualizeTab/VisualizeTab.py,sha256=gLDuI6rS21rYZ9v-77qffmtTp3NS9DdohrMQgo_JZvY,3953
+mgpsogui/gui/VisualizeTab/VisualizeTab.py,sha256=JCm4qOe_l2vVTabBjGUHAK6fl21UQ4RhRLll_PIUFWI,6832
 mgpsogui/gui/VisualizeTab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mgpsogui/gui/images/IGOW 4 Logo.png,sha256=JixNXz5gOEj898VF-_PHthAlGU-6W-y9ucb4EVNPtjs,433752
 mgpsogui/gui/images/collapse.png,sha256=yicb16LaY7Nl5q9V8HHlht-Fbg5xU-l4_LJ_Arfmu1c,2989
@@ -34,7 +34,7 @@ mgpsogui/gui/images/stop.png,sha256=JPuxXQerCGpLikcp7cAj3iLCOjULMYYZ2sZe0lArh68,
 mgpsogui/gui/images/test.png,sha256=MUnVpRK-isxhEHzx4Q6Yh0M6FRZD1qvgCHH2XmiSBbk,3642
 mgpsogui/gui/images/trash.png,sha256=j8cf0kWbJd-4Jp20lUVV1o1NSeQ4v1Ej4gfcIA3DVRQ,2958
 mgpsogui/gui/images/up.png,sha256=AQvFWCUqSQNaQ1E6LKZ9zNfSvW6t4mgy8uswdg9T2Hg,2457
-mgpsogui/util/GraphGenerator.py,sha256=t2SS3dXyhYVZkeu1RskZlOjq85NYBXK6cdD0NbaMYTo,26310
+mgpsogui/util/GraphGenerator.py,sha256=dkrGl5rkKJdloi_Mm-TKw4gEddpFlTJxiF1HZEH8Pzc,28627
 mgpsogui/util/PSORunner.py,sha256=ELpRgEpUiLV4hW-A-y4yzfkFN1BXgH7LZ0qlwWBRz3s,11012
 mgpsogui/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mgpsogui/util/debug.py,sha256=Swzny9_1S-CCONWGssj9iDoSpbwu5Vr4f95g_9L8bec,19759
@@ -60,8 +60,8 @@ mgpsogui/util/recosu/utils/trace_writer.py,sha256=V9BJlOjCbNYGoXGEk3CF5wjifBxvar
 mgpsogui/util/recosu/utils/utils.py,sha256=QB8vftq3142ekG0ORjz0ZBHU5YknXbR0oTsrxrPAsF0,3951
 mgpsogui/util/recosu/utils/plot/__init__.py,sha256=h1KjM7_tNDv351pcwt8A6Ibb1jhwWyx5Gbu-zj-sI3Q,71
 mgpsogui/util/recosu/utils/plot/cost_steps.py,sha256=1Ce11AJyweWkmvjXPxEygzS-h8yVLmQEDLS53yjPLqQ,3779
-mg_pso_gui-0.1.209.dist-info/METADATA,sha256=2kJji5mcHBn5t1pjIoQzJ9zfz0UiDUlYSL1hbJGBHTg,9456
-mg_pso_gui-0.1.209.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
-mg_pso_gui-0.1.209.dist-info/entry_points.txt,sha256=jg82VOFjR1XDGrchs1wJSCqKYE4Ozv12aBcCSp--koA,117
-mg_pso_gui-0.1.209.dist-info/top_level.txt,sha256=y7JuS9xJN5YdxUsQ3PSVjN8MzQAnR146bP3ZN3PYWdE,9
-mg_pso_gui-0.1.209.dist-info/RECORD,,
+mg_pso_gui-0.1.213.dist-info/METADATA,sha256=q1SRoqQHAafc-bm4NWTM4RlNyqrhWYY04R-IfnDWfEw,9456
+mg_pso_gui-0.1.213.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
+mg_pso_gui-0.1.213.dist-info/entry_points.txt,sha256=jg82VOFjR1XDGrchs1wJSCqKYE4Ozv12aBcCSp--koA,117
+mg_pso_gui-0.1.213.dist-info/top_level.txt,sha256=y7JuS9xJN5YdxUsQ3PSVjN8MzQAnR146bP3ZN3PYWdE,9
+mg_pso_gui-0.1.213.dist-info/RECORD,,

mgpsogui/gui/VisualizeTab/VisualizeTab.py

@@ -4,6 +4,7 @@ from PIL import Image, ImageTk
 import os
 import platform
 import subprocess
+import shutil
 
 from . import SideBar
 
@@ -27,7 +28,60 @@ def create_tab(self, tab):
             subprocess.Popen(["xdg-open", file_path])
 
     def export_graph():
-        pass
+        
+        info = self.option_manager.get_project_data()
+        folder = os.path.join(info['path'], info['name'])
+        
+        if not os.path.exists(folder):
+            os.makedirs(folder)
+
+        html_file = os.path.join(folder, self.selected_graph_name + ".html")
+        json_file = os.path.join(folder, self.selected_graph_name + ".json")
+        png_file = os.path.join(folder, self.selected_graph_name + ".png")
+        pdf_file = os.path.join(folder, self.selected_graph_name + ".pdf")
+
+        # Make a directory call package
+        if not os.path.exists(os.path.join(folder, self.selected_graph_name + "_package")):
+            os.makedirs(os.path.join(folder, self.selected_graph_name + "_package"))
+
+        # Check if html_file exists and copy it to the package directory
+        if os.path.exists(html_file):
+            with open(html_file, 'r') as file:
+                data = file.read()
+                with open(os.path.join(folder, self.selected_graph_name + "_package", self.selected_graph_name + ".html"), 'w') as new_file:
+                    new_file.write(data)
+
+        # Check if json_file exists and copy it to the package directory
+        if os.path.exists(json_file):
+            with open(json_file, 'r') as file:
+                data = file.read()
+                with open(os.path.join(folder, self.selected_graph_name + "_package", self.selected_graph_name + ".json"), 'w') as new_file:
+                    new_file.write(data)
+        
+        # Check if png_file exists and copy it to the package directory
+        if os.path.exists(png_file):
+            with open(png_file, 'rb') as file:
+                data = file.read()
+                with open(os.path.join(folder, self.selected_graph_name + "_package", self.selected_graph_name + ".png"), 'wb') as new_file:
+                    new_file.write(data)
+
+        # Check if pdf_file exists and copy it to the package directory
+        if os.path.exists(pdf_file):
+            with open(pdf_file, 'rb') as file:
+                data = file.read()
+                with open(os.path.join(folder, self.selected_graph_name + "_package", self.selected_graph_name + ".pdf"), 'wb') as new_file:
+                    new_file.write(data)
+        
+        # Zip the package directory
+        shutil.make_archive(os.path.join(folder, self.selected_graph_name + "_package"), 'zip', os.path.join(folder, self.selected_graph_name + "_package"))
+
+        # Open the directory containing the package
+        if platform.system() == "Windows":
+            os.startfile(os.path.join(folder, self.selected_graph_name + "_package"))
+        elif platform.system() == "Darwin":
+            subprocess.Popen(["open", os.path.join(folder, self.selected_graph_name + "_package")])
+        else:
+            subprocess.Popen(["xdg-open", os.path.join(folder, self.selected_graph_name + "_package")])
     
     def _resize_image(event):
         self.graph_label.update_idletasks()

mgpsogui/util/GraphGenerator.py

@@ -9,6 +9,9 @@ import customtkinter
 import traceback
 import math
 
+baseFigureWidth = 700
+baseFigureHeight = 350
+
 theme_background_color = {
     "Dark": 'rgba(42, 42, 42, 0)',
     "Light": 'rgba(255, 255, 255, 0)',
@@ -139,6 +142,12 @@ def best_cost_stacked(homepage, config, data, option_manager):
             gridwidth=0.1
         )
     )
+    if (theme == "Publication"):
+        fig.update_layout(
+            font=dict(
+                size=16
+            )
+        )
     
     info = option_manager.get_project_data()
     folder = os.path.join(info['path'], info['name'])
@@ -149,6 +158,7 @@ def best_cost_stacked(homepage, config, data, option_manager):
     fig.write_image(os.path.join(folder, "best_cost_stacked.png"), width=1280, height=720)
     fig.write_html(os.path.join(folder, "best_cost_stacked.html"), include_plotlyjs='cdn', auto_open=False)
     fig.write_json(os.path.join(folder, "best_cost_stacked.json"))
+    fig.write_image(os.path.join(folder, "best_cost_stacked.pdf"), engine="kaleido", width=baseFigureWidth, height=baseFigureHeight)
     with open(os.path.join(folder, "best_cost_stacked.html"), "r") as f:
         html = f.read()
         html = html.replace("<body>", "<body bgcolor='#2a2a2a'>")
@@ -192,6 +202,12 @@ def table(homepage, config, dataframe, option_manager):
             gridwidth=0.1
         )
     )
+    if (theme == "Publication"):
+        fig.update_layout(
+            font=dict(
+                size=16
+            )
+        )
 
     info = option_manager.get_project_data()
     folder = os.path.join(info['path'], info['name'])
@@ -202,6 +218,7 @@ def table(homepage, config, dataframe, option_manager):
     fig.write_image(os.path.join(folder, "table.png"), width=1280, height=720)
     fig.write_html(os.path.join(folder, "table.html"), include_plotlyjs='cdn', auto_open=False)
     fig.write_json(os.path.join(folder, "table.json"))
+    fig.write_image(os.path.join(folder, "table.pdf"), engine="kaleido", width=baseFigureWidth, height=baseFigureHeight)
     with open(os.path.join(folder, "table.html"), "r") as f:
         html = f.read()
         html = html.replace("<body>", "<body bgcolor='#2a2a2a'>")
@@ -277,6 +294,12 @@ def best_cost_by_round(homepage, config, data, option_manager):
             gridwidth=0.1
         )
     )
+    if (theme == "Publication"):
+        fig.update_layout(
+            font=dict(
+                size=16
+            )
+        )
     
     info = option_manager.get_project_data()
     folder = os.path.join(info['path'], info['name'])
@@ -287,6 +310,7 @@ def best_cost_by_round(homepage, config, data, option_manager):
     fig.write_image(os.path.join(folder, "best_cost_by_round.png"), width=1280, height=720)
     fig.write_html(os.path.join(folder, "best_cost_by_round.html"), include_plotlyjs='cdn', auto_open=False)
     fig.write_json(os.path.join(folder, "best_cost_by_round.json"))
+    fig.write_image(os.path.join(folder, "best_cost_by_round.pdf"), engine="kaleido", width=baseFigureWidth, height=baseFigureHeight)
     with open(os.path.join(folder, "best_cost_by_round.html"), "r") as f:
         html = f.read()
         html = html.replace("<body>", "<body bgcolor='#2a2a2a'>")
@@ -345,6 +369,12 @@ def calibrated_params_by_round(homepage, config, list_of_objs, option_manager):
             gridwidth=0.1
         )
     )
+    if (theme == "Publication"):
+        fig.update_layout(
+            font=dict(
+                size=16
+            )
+        )
 
     info = option_manager.get_project_data()
     folder = os.path.join(info['path'], info['name'])
@@ -355,6 +385,7 @@ def calibrated_params_by_round(homepage, config, list_of_objs, option_manager):
     fig.write_image(os.path.join(folder, "calibrated_params_by_round.png"), width=1280, height=720)
     fig.write_html(os.path.join(folder, "calibrated_params_by_round.html"), include_plotlyjs='cdn', auto_open=False)
     fig.write_json(os.path.join(folder, "calibrated_params_by_round.json"))
+    fig.write_image(os.path.join(folder, "calibrated_params_by_round.pdf"), engine="kaleido", width=baseFigureWidth, height=baseFigureHeight)
     with open(os.path.join(folder, "calibrated_params_by_round.html"), "r") as f:
         html = f.read()
         html = html.replace("<body>", "<body bgcolor='#2a2a2a'>")
@@ -420,6 +451,12 @@ def custom_csv(homepage, option_manager):
             side='right'
         )
     )
+    if (theme == "Publication"):
+        fig.update_layout(
+            font=dict(
+                size=16
+            )
+        )
 
     info = option_manager.get_project_data()
     folder = os.path.join(info['path'], info['name'])
@@ -430,6 +467,7 @@ def custom_csv(homepage, option_manager):
     fig.write_image(os.path.join(folder, "custom_csv.png"), width=1280, height=720)
     fig.write_html(os.path.join(folder, "custom_csv.html"), include_plotlyjs='cdn', auto_open=False)
     fig.write_json(os.path.join(folder, "custom_csv.json"))
+    fig.write_image(os.path.join(folder, "custom_csv.pdf"), engine="kaleido", width=baseFigureWidth, height=baseFigureHeight)
     with open(os.path.join(folder, "custom_csv.html"), "r") as f:
         html = f.read()
         html = html.replace("<body>", "<body bgcolor='#2a2a2a'>")
@@ -504,6 +542,12 @@ def compare_csv(homepage, option_manager):
             side='right'
         )
     )
+    if (theme == "Publication"):
+        fig.update_layout(
+            font=dict(
+                size=16
+            )
+        )
 
     info = option_manager.get_project_data()
     folder = os.path.join(info['path'], info['name'])
@@ -514,6 +558,7 @@ def compare_csv(homepage, option_manager):
     fig.write_image(os.path.join(folder, "compare_csv.png"), width=1280, height=720)
     fig.write_html(os.path.join(folder, "compare_csv.html"), include_plotlyjs='cdn', auto_open=False)
     fig.write_json(os.path.join(folder, "compare_csv.json"))
+    fig.write_image(os.path.join(folder, "compare_csv.pdf"), engine="kaleido", width=baseFigureWidth, height=baseFigureHeight)
     with open(os.path.join(folder, "compare_csv.html"), "r") as f:
         html = f.read()
         html = html.replace("<body>", "<body bgcolor='#2a2a2a'>")
@@ -590,6 +635,12 @@ def sampling_csv(homepage, option_manager):
             side='right'
         )
     )
+    if (theme == "Publication"):
+        fig.update_layout(
+            font=dict(
+                size=16
+            )
+        )
 
     info = option_manager.get_project_data()
     folder = os.path.join(info['path'], info['name'])
@@ -600,6 +651,7 @@ def sampling_csv(homepage, option_manager):
     fig.write_image(os.path.join(folder, "sampling_csv.png"), width=1280, height=720)
     fig.write_html(os.path.join(folder, "sampling_csv.html"), include_plotlyjs='cdn', auto_open=False)
     fig.write_json(os.path.join(folder, "sampling_csv.json"))
+    fig.write_image(os.path.join(folder, "sampling_csv.pdf"), engine="kaleido", width=baseFigureWidth, height=baseFigureHeight)
     with open(os.path.join(folder, "sampling_csv.html"), "r") as f:
         html = f.read()
         html = html.replace("<body>", "<body bgcolor='#2a2a2a'>")
@@ -618,7 +670,7 @@ def matrix_editor(homepage, option_manager):
     all_figures = []
     figure_parameters = option_manager.get_arguments()['figure_parameters']
 
-    color_list = ['#636EFA', '#EF553B', '#00CC96', '#AB63FA', '#FFA15A', '#19D3F3', '#FF6692', '#B6E880', '#FF97FF', '#FECB52']
+    #color_list = ['#636EFA', '#EF553B', '#00CC96', '#AB63FA', '#FFA15A', '#19D3F3', '#FF6692', '#B6E880', '#FF97FF', '#FECB52']
     color_index = 0
     color_dict = {}
 
@@ -699,6 +751,13 @@ def matrix_editor(homepage, option_manager):
                 gridwidth=0.1
             )
         )
+        if (theme == "Publication"):
+            fig.update_layout(
+                font=dict(
+                    size=16
+                )
+            )
+
         all_figures.append(fig)
 
     row = 1
@@ -707,7 +766,7 @@ def matrix_editor(homepage, option_manager):
         for trace in fig.data:
 
             if trace.name not in color_dict:
-                color_dict[trace.name] = color_list[color_index % len(color_list)]
+                color_dict[trace.name] = theme_plot_color_pallet[color_index % len(theme_plot_color_pallet)]
                 color_index += 1
 
             trace.marker.color = color_dict[trace.name]
@@ -739,6 +798,12 @@ def matrix_editor(homepage, option_manager):
         paper_bgcolor=theme_background_color[theme],
         plot_bgcolor=theme_plot_color[theme]
     )
+    if (theme == "Publication"):
+        fig.update_layout(
+            font=dict(
+                size=16
+            )
+        )
 
     if not os.path.exists(folder):
         os.makedirs(folder)
@@ -746,6 +811,7 @@ def matrix_editor(homepage, option_manager):
     fig.write_image(os.path.join(folder, "matrix_editor.png"), width=1280, height=720)
     fig.write_html(os.path.join(folder, "matrix_editor.html"), include_plotlyjs='cdn', auto_open=False)
     fig.write_json(os.path.join(folder, "matrix_editor.json"))
+    fig.write_image(os.path.join(folder, "matrix_editor.pdf"), engine="kaleido", width=baseFigureWidth, height=baseFigureHeight)
     with open(os.path.join(folder, "matrix_editor.html"), "r") as f:
         html = f.read()
         html = html.replace("<body>", "<body bgcolor='#2a2a2a'>")