mg-pso-gui 0.1.210__py3-none-any.whl → 0.1.214__py3-none-any.whl

{mg_pso_gui-0.1.210.dist-info → mg_pso_gui-0.1.214.dist-info}/METADATA

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

{mg_pso_gui-0.1.210.dist-info → mg_pso_gui-0.1.214.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=6g_UfMMiy2c8-dP9lAvcMRa-wACvmC1u65DcwvL8z30,28600
+mgpsogui/util/GraphGenerator.py,sha256=7JRq4um5UJd5tAYmffHWi558pcCMAwd6k3yvO0kEbPk,31779
 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.210.dist-info/METADATA,sha256=L9cFP4nDIU7yyQgWvpSKIBhwQaFgwrxp4PRNKHCnhlY,9456
-mg_pso_gui-0.1.210.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
-mg_pso_gui-0.1.210.dist-info/entry_points.txt,sha256=jg82VOFjR1XDGrchs1wJSCqKYE4Ozv12aBcCSp--koA,117
-mg_pso_gui-0.1.210.dist-info/top_level.txt,sha256=y7JuS9xJN5YdxUsQ3PSVjN8MzQAnR146bP3ZN3PYWdE,9
-mg_pso_gui-0.1.210.dist-info/RECORD,,
+mg_pso_gui-0.1.214.dist-info/METADATA,sha256=U-WtoiGnr87Cbil-UFkE5f8IvE-72q2BW3ZDxkKLAq8,9456
+mg_pso_gui-0.1.214.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
+mg_pso_gui-0.1.214.dist-info/entry_points.txt,sha256=jg82VOFjR1XDGrchs1wJSCqKYE4Ozv12aBcCSp--koA,117
+mg_pso_gui-0.1.214.dist-info/top_level.txt,sha256=y7JuS9xJN5YdxUsQ3PSVjN8MzQAnR146bP3ZN3PYWdE,9
+mg_pso_gui-0.1.214.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

@@ -146,6 +146,20 @@ def best_cost_stacked(homepage, config, data, option_manager):
         fig.update_layout(
             font=dict(
                 size=16
+            ),
+            legend=dict(
+                orientation="h",
+                yanchor="bottom",
+                y=1.02,
+                xanchor="center",
+                x=0.47
+            ),
+            margin=dict(
+                t=0,
+                b=1,
+                l=1,
+                r=1,
+                autoexpand=True
             )
         )
     
@@ -206,6 +220,20 @@ def table(homepage, config, dataframe, option_manager):
         fig.update_layout(
             font=dict(
                 size=16
+            ),
+            legend=dict(
+                orientation="h",
+                yanchor="bottom",
+                y=1.02,
+                xanchor="center",
+                x=0.47
+            ),
+            margin=dict(
+                t=0,
+                b=1,
+                l=1,
+                r=1,
+                autoexpand=True
             )
         )
 
@@ -298,6 +326,20 @@ def best_cost_by_round(homepage, config, data, option_manager):
         fig.update_layout(
             font=dict(
                 size=16
+            ),
+            legend=dict(
+                orientation="h",
+                yanchor="bottom",
+                y=1.02,
+                xanchor="center",
+                x=0.47
+            ),
+            margin=dict(
+                t=0,
+                b=1,
+                l=1,
+                r=1,
+                autoexpand=True
             )
         )
     
@@ -373,6 +415,20 @@ def calibrated_params_by_round(homepage, config, list_of_objs, option_manager):
         fig.update_layout(
             font=dict(
                 size=16
+            ),
+            legend=dict(
+                orientation="h",
+                yanchor="bottom",
+                y=1.02,
+                xanchor="center",
+                x=0.47
+            ),
+            margin=dict(
+                t=0,
+                b=1,
+                l=1,
+                r=1,
+                autoexpand=True
             )
         )
 
@@ -455,6 +511,20 @@ def custom_csv(homepage, option_manager):
         fig.update_layout(
             font=dict(
                 size=16
+            ),
+            legend=dict(
+                orientation="h",
+                yanchor="bottom",
+                y=1.02,
+                xanchor="center",
+                x=0.47
+            ),
+            margin=dict(
+                t=0,
+                b=1,
+                l=1,
+                r=1,
+                autoexpand=True
             )
         )
 
@@ -546,6 +616,20 @@ def compare_csv(homepage, option_manager):
         fig.update_layout(
             font=dict(
                 size=16
+            ),
+            legend=dict(
+                orientation="h",
+                yanchor="bottom",
+                y=1.02,
+                xanchor="center",
+                x=0.47
+            ),
+            margin=dict(
+                t=0,
+                b=1,
+                l=1,
+                r=1,
+                autoexpand=True
             )
         )
 
@@ -639,6 +723,20 @@ def sampling_csv(homepage, option_manager):
         fig.update_layout(
             font=dict(
                 size=16
+            ),
+            legend=dict(
+                orientation="h",
+                yanchor="bottom",
+                y=1.02,
+                xanchor="center",
+                x=0.47
+            ),
+            margin=dict(
+                t=0,
+                b=1,
+                l=1,
+                r=1,
+                autoexpand=True
             )
         )
 
@@ -670,7 +768,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 = {}
 
@@ -755,6 +853,20 @@ def matrix_editor(homepage, option_manager):
             fig.update_layout(
                 font=dict(
                     size=16
+                ),
+                legend=dict(
+                    orientation="h",
+                    yanchor="bottom",
+                    y=1.02,
+                    xanchor="center",
+                    x=0.47
+                ),
+                margin=dict(
+                    t=0,
+                    b=1,
+                    l=1,
+                    r=1,
+                    autoexpand=True
                 )
             )
 
@@ -766,7 +878,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]
@@ -802,6 +914,20 @@ def matrix_editor(homepage, option_manager):
         fig.update_layout(
             font=dict(
                 size=16
+            ),
+            legend=dict(
+                orientation="h",
+                yanchor="bottom",
+                y=1.02,
+                xanchor="center",
+                x=0.47
+            ),
+            margin=dict(
+                t=0,
+                b=1,
+                l=1,
+                r=1,
+                autoexpand=True
             )
         )