mg-pso-gui 0.1.20__tar.gz → 0.1.22__tar.gz

{mg-pso-gui-0.1.20 → mg-pso-gui-0.1.22}/PKG-INFO

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

{mg-pso-gui-0.1.20 → mg-pso-gui-0.1.22}/mg_pso_gui.egg-info/PKG-INFO

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

{mg-pso-gui-0.1.20 → mg-pso-gui-0.1.22}/mgpsogui/gui/HomePage.py

@@ -175,6 +175,8 @@ class App(customtkinter.CTk):
 		ctt(self.download_button, delay=0.1, alpha=0.95, message="Download Results")
 		self.download_button.grid(row=0, column=10, padx=(5, 20), pady=header_padding_y)
 
+		self.after(1000, self.running_loop)
+
 	'''def _resize_image(self, event):
 		self.graph_label.update_idletasks()
 		new_width = self.graph_label.winfo_width()
@@ -267,13 +269,15 @@ class App(customtkinter.CTk):
 			if "service_parameters" in metrics:
 				self.option_manager.set_service_parameters(metrics["service_parameters"])
 				self.tabview.configure(state="enabled")
-    
+	
 			if "calibration_progress" in metrics:
 				self.progress_data = pd.read_json(metrics["calibration_progress"])
 				self.calibration_data = metrics["calibration_data"]
-    
+				print(self.progress_data)
+	
 			if "calibration_data" in metrics:
 				self.calibration_data = metrics["calibration_data"]
+				print(self.calibration_data)
 			
 			print(metrics)
 			
@@ -347,6 +351,9 @@ class App(customtkinter.CTk):
 		metrics = self.option_manager.get_metrics()
 		self.running_config = metrics
 		
+		self.progress_data = None
+		self.calibration_data = None
+  
 		#self.progress_bar.configure(mode="indeterminnate")
 		#self.progress_bar.start()
 		self.footer_progress_bar.configure(mode="indeterminnate")
@@ -364,6 +371,15 @@ class App(customtkinter.CTk):
 		try:
 			info = self.option_manager.get_project_data()
 			folder = os.path.join(info['path'], info['name'])
+
+			if not os.path.exists(folder):
+				os.makedirs(folder)
+
+			if (os.path.exists(os.path.join(folder, 'output.txt'))):
+				os.remove(os.path.join(folder, 'output.txt'))
+				
+			if (os.path.exists(os.path.join(folder, 'error.txt'))):
+				os.remove(os.path.join(folder, 'error.txt'))
 			
 			self.train_process = Process(target=PSORunner.run_process, args=(stdout_queue, stderr_queue, results_queue, cosu_queue, metrics, folder))
 			self.train_process.daemon = True
@@ -418,9 +434,18 @@ class App(customtkinter.CTk):
 		self.footer_progress_label.configure(text="Stopped")
 		self.progress_message_right.configure(text="")
 		
+	def running_loop(self):
+		try:
+			active_tab = self.tabview.get()
+			if (active_tab == "Results"):
+				GraphGenerator.generate_graphs(self)
+			else:
+				print(active_tab)
+		finally:
+			self.after(1000, self.running_loop)
+
 	def watch_loop(self):
 		# Check if file exists:
-		
 		info = self.option_manager.get_project_data()
 		folder = os.path.join(info['path'], info['name'])
 			
@@ -492,11 +517,11 @@ class App(customtkinter.CTk):
 				progress_values = [tuple(map(float, i)) for i in progress_values]
 				
 				#print("Current params: " + str(calibrated_params), flush=True)
-    
+	
 				self.calibration_data = calibrated_params
 				
 				#GraphGenerator.calibrated_params_by_round(self.running_config['steps'], calibrated_params, self.option_manager)
-				self.update_graph("")
+				#self.update_graph("")
 			except Exception as e:
 				# Print stack trace
 				traceback.print_exc()
@@ -523,12 +548,6 @@ class App(customtkinter.CTk):
 					df = df.drop_duplicates(subset=['completed_rounds', 'round_step'])
 					self.progress_data = df
 
-					#GraphGenerator.update_progress_graphs(self.running_config['steps'], self.current_data, self.option_manager)
-					#GraphGenerator.best_cost_stacked(self.running_config['steps'], df, self.option_manager)
-					#GraphGenerator.best_cost_by_round(self.running_config['steps'], df, self.option_manager)
-					#GraphGenerator.table(self.running_config['steps'], df, self.option_manager)
-					#self.update_graph("")
-
 					match = matches[-1]
 					percent = int(match[0])
 					completed_rounds = int(match[1])
@@ -561,18 +580,6 @@ class App(customtkinter.CTk):
 		if self.train_process.is_alive():
 			self.after(1000, self.watch_loop)
 		else:
-			# Check if queue is empty
-			"""
-			if (self.process_queue.empty()):
-				print("PROCESS FINISHED BUT WE GOT NO OBJECTS :(")
-				self.footer_progress_label.configure(text="Failed")
-			else:
-				self.final_results = self.process_queue.get()
-				print("PROCESS FINISHED WE GOT THINGS")
-				print(self.final_results)
-				self.footer_progress_label.configure(text="Finished")
-			"""
-			
 			#self.progress_bar.stop()
 			self.footer_progress_bar.stop()
 			#self.progress_bar.configure(mode="indeterminate")
@@ -584,6 +591,9 @@ class App(customtkinter.CTk):
 			self.progress_message_right.configure(text="")
 			self.textbox.insert("0.0", "\nCalibration finished!\n")
 
+		
+	
+
 def start():
 	app = App()
 	app.mainloop()

{mg-pso-gui-0.1.20 → mg-pso-gui-0.1.22}/mgpsogui/gui/RunTab/RunTab.py

@@ -19,21 +19,7 @@ def create_tab(self, tab):
     tab.grid_columnconfigure(0, weight=1)
     tab.grid_rowconfigure(0, weight=1)
     tab.grid_rowconfigure(1, weight=200)
-    
-    """
-    self.url = customtkinter.CTkEntry(tab, textvariable=self.option_manager.get_arguments()['url'])
-    self.url.grid(row=0, column=0, columnspan=1, padx=(20, 20), pady=(20, 20), sticky="nsew")
 
-    self.run_button = customtkinter.CTkButton(tab, text="Run", command=self.run)
-    self.run_button.grid(row=0, column=1, padx=(20, 20), pady=(20, 20), sticky="nsew")
-    ctt(self.run_button, delay=0.5, message="Start calibration...")
-    
-        self.stop_button = customtkinter.CTkButton(tab, text="Stop", command=self.stop)
-    self.stop_button.grid(row=0, column=2, padx=(20, 20), pady=(20, 20), sticky="nsew")
-    ctt(self.stop_button, delay=0.5, message="Stop calibration...")
-    
-    """
-    
     self.progress_container = customtkinter.CTkFrame(tab)
     self.progress_container.grid_columnconfigure(0, weight=1)
     self.progress_container.grid_columnconfigure(1, weight=1)
@@ -50,12 +36,7 @@ def create_tab(self, tab):
     self.progress_message_right = customtkinter.CTkLabel(self.progress_container, text="")
     self.progress_message_right.grid(row=0, column=2, padx=(10, 10), pady=(10, 10), sticky="e")
     
-    #self.progress_bar = customtkinter.CTkProgressBar(self.progress_container)
-    #self.progress_bar.grid(row=1, column=0, columnspan=3, padx=(10, 10), pady=(10, 10), sticky="ew")
-    #self.progress_bar.set(0)
-    #ctt(self.progress_bar, delay=0.5, message="Current calibration progress")
-    
     self.textbox = customtkinter.CTkTextbox(tab)
     self.textbox.grid(row=1, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
-    self.textbox.insert("0.0", "Welcome to the CSIP PSO Calibration Tool!\n\nUse the Calibration tab to define steps and calibration parameters. Use this tab to run and observe calibration progress. Once finished, use the Visualization tab to generate figures and graphs.")
+    self.textbox.insert("0.0", "Welcome to the CSIP PSO Calibration Tool!\n\nUse the Setup tab to define steps and calibration parameters. Use this tab to view logs and observe calibration progress. Once finished, use the Results tab to generate figures and graphs.")
         

{mg-pso-gui-0.1.20 → mg-pso-gui-0.1.22}/mgpsogui/util/GraphGenerator.py

@@ -3,9 +3,34 @@ import plotly.graph_objs as go
 import pandas as pd
 import numpy as np
 import os
+from PIL import Image, ImageTk
 
-def generate_graphs(config, dataframe, option_manager):
-    pass
+def generate_graphs(HomePage):
+    selected_graph = HomePage.graph_selector_value.get()
+    info = HomePage.option_manager.get_project_data()
+    folder = os.path.join(info['path'], info['name'])
+    if not os.path.exists(folder):
+        os.makedirs(folder)
+        
+    if (selected_graph == "Best Cost Stacked"):
+        HomePage.selected_graph_name = "best_cost_stacked"
+        best_cost_stacked(HomePage.running_config['steps'], HomePage.progress_data, HomePage.option_manager)
+    elif (selected_graph == "Best Cost by Round"):
+        HomePage.selected_graph_name = "best_cost_by_round"
+        best_cost_by_round(HomePage.running_config['steps'], HomePage.progress_data, HomePage.option_manager)
+    elif (selected_graph == "Iteration Table"):
+        HomePage.selected_graph_name = "table"
+        table(HomePage.running_config['steps'], HomePage.progress_data, HomePage.option_manager)
+    elif (selected_graph == "Calibrated Parameters"):
+        HomePage.selected_graph_name = "calibrated_params_by_round"
+        calibrated_params_by_round(HomePage.running_config['steps'], HomePage.calibration_data, HomePage.option_manager)
+        
+    image_path = os.path.join(folder, HomePage.selected_graph_name + ".png")
+    
+    if not os.path.exists(image_path):
+        image_path = os.path.join("./images", "up.png")
+  
+    HomePage.graph_image_obj = Image.open(image_path)
 
 def best_cost_stacked(config, dataframe, option_manager):
     fig = go.Figure()

{mg-pso-gui-0.1.20 → mg-pso-gui-0.1.22}/setup.py

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