mg-pso-gui 0.1.177__tar.gz → 0.1.178__tar.gz

{mg-pso-gui-0.1.177 → mg-pso-gui-0.1.178}/PKG-INFO

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

{mg-pso-gui-0.1.177 → mg-pso-gui-0.1.178}/mg_pso_gui.egg-info/PKG-INFO

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

{mg-pso-gui-0.1.177 → mg-pso-gui-0.1.178}/mgpsogui/util/GraphGenerator.py

@@ -1,5 +1,6 @@
 import plotly.express as px
 import plotly.graph_objs as go
+from plotly.subplots import make_subplots
 import pandas as pd
 import numpy as np
 import os
@@ -38,6 +39,7 @@ def generate_graphs(HomePage):
             sampling_csv(HomePage, HomePage.option_manager)
         elif (selected_graph == "Matrix Editor"):
             HomePage.selected_graph_name = "matrix_editor"
+            matrix_editor(HomePage, HomePage.option_manager)
             pass
             
         image_path = os.path.join(folder, HomePage.selected_graph_name + ".png")
@@ -525,4 +527,143 @@ def sampling_csv(homepage, option_manager):
     with open(os.path.join(folder, "sampling_csv.html"), "w") as f:
         f.write(html)
 
+    return fig
+
+def matrix_editor(homepage, option_manager):
+    style = homepage.figure_style.get()
+    data = homepage.csv_data
+    x = homepage.csv_x_selector.get()
+
+    all_figures = []
+    figure_parameters = option_manager.get_arguments()['figure_parameters']
+
+    color_list = ['#636EFA', '#EF553B', '#00CC96', '#AB63FA', '#FFA15A', '#19D3F3', '#FF6692', '#B6E880', '#FF97FF', '#FECB52']
+    color_index = 0
+    color_dict = {}
+
+    total_figures = 0
+    for parameter in figure_parameters:
+        name = parameter['name'].get()
+        if ("Fig" not in name):
+            continue
+        total_figures += 1
+
+    figures_per_row = 1
+    if (total_figures == 1):
+        figures_per_row = 1
+    elif (total_figures == 2):
+        figures_per_row = 2
+    elif (total_figures == 3 or total_figures == 4):
+        figures_per_row = 2
+    elif (total_figures == 5 or total_figures == 6):
+        figures_per_row = 3
+    elif (total_figures == 7 or total_figures == 8):
+        figures_per_row = 4
+    elif (total_figures == 9 or total_figures == 10):
+        figures_per_row = 5
+
+    fig_combined = make_subplots(rows=(total_figures // figures_per_row) + 1, cols=figures_per_row, shared_xaxes=False, shared_yaxes=False)
+
+    for parameter in figure_parameters:
+        name = parameter['name'].get()
+
+        if ("Fig" not in name):
+            continue
+
+        fig = go.Figure()
+        val = parameter['value'].get()
+        
+        xx = None
+        if x == "time":
+            xx = pd.to_datetime(data["time"], format='%Y-%m-%d', errors='coerce')
+        elif x == "date":
+            xx = pd.to_datetime(data["date"], format='%d-%m-%Y', errors='coerce')
+        else:
+            xx = pd.to_numeric(data[x], errors="coerce")
+            
+        yy = pd.to_numeric(data[val], errors="coerce")
+        
+        yy_unit = ""
+
+        if (style == "Scatter"):
+            fig.add_trace(go.Scatter(x=xx, y=yy, name=val, mode='markers'))
+        elif (style == "Bars"):
+            fig.add_trace(go.Bar(x=xx, y=yy, name=val))
+        elif (style == "Lines"):
+            fig.add_trace(go.Scatter(x=xx, y=yy, name=val))
+        elif (style == "Area"):
+            fig.add_trace(go.Scatter(x=xx, y=yy, name=val, fill='tozeroy'))
+        elif (style == "Box"):
+            fig.add_trace(go.Box(x=xx, y=yy, name=val))
+
+        fig.update_layout(
+            title="",
+            xaxis_title=x,
+            yaxis_title=val,
+            font=dict(color='white'),
+            paper_bgcolor='rgba(42, 42, 42, 0)',
+            plot_bgcolor='rgb(62, 62, 62)',
+            xaxis=dict(
+                gridcolor='rgb(72, 72, 72)',
+                gridwidth=1
+            ),
+            yaxis=dict(
+                title=val,
+                autorange=True,
+                gridcolor='rgb(72, 72, 72)',
+                gridwidth=0.1
+            )
+        )
+        all_figures.append(fig)
+
+    row = 1
+    col = 1
+    for fig in all_figures:
+        for trace in fig.data:
+
+            if trace.name not in color_dict:
+                color_dict[trace.name] = color_list[color_index % len(color_list)]
+                color_index += 1
+
+            trace.marker.color = color_dict[trace.name]
+
+            if fig_combined.data:
+                trace.showlegend = not any(t.name == trace.name for t in fig_combined.data)
+            fig_combined.add_trace(trace, row=row, col=col)
+        
+        # Get the layout titles from the individual figure
+        xaxis_title = fig.layout.xaxis.title.text
+        yaxis_title = fig.layout.yaxis.title.text
+
+        # Update the combined figure's subplot with the titles
+        fig_combined.update_xaxes(title_text=xaxis_title, row=row, col=col)
+        fig_combined.update_yaxes(title_text=yaxis_title, row=row, col=col)
+
+        col += 1
+        if col > figures_per_row:
+            row += 1
+            col = 1
+
+    info = option_manager.get_project_data()
+    folder = os.path.join(info['path'], info['name'])
+    
+    fig = fig_combined
+    fig.update_layout(
+        title="",
+        font=dict(color='white'),
+        paper_bgcolor='rgba(42, 42, 42, 0)',
+        plot_bgcolor='rgb(62, 62, 62)'
+    )
+
+    if not os.path.exists(folder):
+        os.makedirs(folder)
+
+    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)
+    with open(os.path.join(folder, "matrix_editor.html"), "r") as f:
+        html = f.read()
+        html = html.replace("<body>", "<body bgcolor='#2a2a2a'>")
+    with open(os.path.join(folder, "matrix_editor.html"), "w") as f:
+        f.write(html)
+
     return fig

{mg-pso-gui-0.1.177 → mg-pso-gui-0.1.178}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 import codecs
 import os
 
-VERSION = '0.1.177'
+VERSION = '0.1.178'
 DESCRIPTION = 'GUI for MG-PSO'
 LONG_DESCRIPTION = open('../README.md').read()