def run_ECRad_from_script(working_dir, shot, Config, InvokeECRad, args):
Results = ECRadResults()
Results.parse_config(Config, Config.time)
next_time_index_to_analyze = 0
print("Analyzing {0:d} time points".format(len(Config.time)))
while True:
if(next_time_index_to_analyze >= len(Config.time)):
break
print("Working on t = {0:1.4f}".format(Config.time[next_time_index_to_analyze]))
if(os.path.isdir(os.path.join(Config.working_dir, "ECRad_data"))):
shutil.rmtree(os.path.join(Config.working_dir, "ECRad_data"))
for diag_key in Config.used_diags_dict:
if("CT" in diag_key or "IEC" == diag_key):
if(str(Config.used_diags_dict[diag_key].beamline) not in Config.gy_dict):
new_gy = get_ECRH_viewing_angles(Config.shot, \
Config.used_diags_dict[diag_key].beamline)
if(new_gy.error == 0):
Config.gy_dict[str(Config.used_diags_dict[diag_key].beamline)] = new_gy
else:
print("Error when reading viewing angles")
print("Launch aborted")
return
if(not prepare_input_files(Config.time, next_time_index_to_analyze, Config)):
print("Error!! Launch aborted")
return None
print("-------- Launching ECRad -----------\n")
print("-------- INVOKE COMMAND------------\n")
print(InvokeECRad + " " + Config.working_dir)
ECRad_process = subprocess.Popen([InvokeECRad, args])
print("-----------------------------------\n")
ECRad_process = subprocess.Popen([InvokeECRad, Config.working_dir])
ECRad_process.wait()
next_time_index_to_analyze += 1
try:
Results.append_new_results()
except (IOError, IndexError) as e:
print("Error: Results of ECRad cannot be found")
print("Most likely cause is an error that occurred within the ECRad")
print("Please run the ECRad with current input parameters in a separate shell.")
print("The command to launch the ECRad can be found above.")
print("Afterwards please send any error messages that appear at sdenk|at|ipp.mpg.de")
print("If no errors occur make sure that you don't have another instance of ECRad GUI working in the same working directory")
print(e)
return None
Results.tidy_up()
print("Finished!")
return Results
class ECRadDriver():
'''
Driver for ECRad. Passes information to ECRad, runs ECRad and extracts results
'''
def __init__(self, Result=None, Scenario_file=None, Config_file=None):
'''
Constructor
'''
if (Result is None):
if (Scenario_file is None or Config_file is None):
raise ValueError(
"Either result or a Scenario and Config file must be present"
)
else:
self.Result = ECRadResults()
self.Result.Scenario = ECRadScenario(noLoad=True)
self.Result.Scenario.load(Scenario_file)
self.Result.Config = ECRadConfig(noLoad=True)
self.Result.Config.load(Config_file)
else:
self.Result = Result
self.ECRad_F2PY_interface = ECRadF2PYInterface()
def run(self, id=None):
itime = 0
self.Result.set_dimensions()
while itime < self.Result.Scenario["dimensions"]["N_time"]:
try:
self.process_time_point(itime)
itime += 1
except Exception as e:
print("Error when processing t = {0:1.4f}".format(
self.Result.Scenario["time"][itime]))
print("Removing this time point and continuing")
raise (e)
self.Result.Scenario.drop_time_point(itime)
self.Result.set_dimensions()
self.Result.tidy_up(autosave=False)
self.Result.to_netcdf(scratch=True, ed=id)
def process_time_point(self, itime):
self.Result = self.ECRad_F2PY_interface.process_single_timepoint(
self.Result, itime)
class Main_Panel(scrolled.ScrolledPanel):
def __init__(self, parent, ECRad_runner_process, ECRad_input_queue, ECRad_output_queue):
scrolled.ScrolledPanel.__init__(self, parent, wx.ID_ANY)
self.parent = parent
self.ECRad_running = False
self.ECRad_runner_process = ECRad_runner_process
self.ECRad_input_queue = ECRad_input_queue
self.ECRad_output_queue = ECRad_output_queue
self.sizer = wx.BoxSizer(wx.VERTICAL)
self.Results = ECRadResults(lastused=True)
self.Bind(EVT_MAKE_ECRAD, self.OnProcessTimeStep)
self.Bind(EVT_ECRAD_FINISHED, self.OnProcessEnded)
self.Bind(EVT_NEXT_TIME_STEP, self.OnNextTimeStep)
self.Bind(EVT_UPDATE_DATA, self.OnUpdate)
self.Bind(EVT_LOCK_EXPORT, self.OnLockExport)
self.Bind(wx.EVT_IDLE, self.OnIdle)
self.Bind(EVT_GENE_DATA_LOADED, self.OnGeneLoaded)
self.Bind(EVT_ECRAD_RESULT_LOADED, self.OnResultsImported)
self.Bind(EVT_LOAD_OLD_RESULT, self.OnImport)
self.SetSizer(self.sizer)
self.SetSize((400, 400))
self.SetMinSize((400, 400))
self.data = None
self.ControlSizer = wx.BoxSizer(wx.HORIZONTAL)
self.ButtonSizer = wx.BoxSizer(wx.VERTICAL)
self.sizer.Add(self.ControlSizer, 0, wx.EXPAND | wx.ALL , 5)
self.StartECRadButton = wx.Button(self, wx.ID_ANY, \
'Start ECRad')
self.KillECRadButton = wx.Button(self, wx.ID_ANY, 'Terminate ECRad')
self.StartECRadButton.Bind(wx.EVT_BUTTON, self.OnStartECRad)
self.KillECRadButton.Bind(wx.EVT_BUTTON, self.OnKillECRad)
self.KillECRadButton.Disable()
self.ExportButton = wx.Button(self, wx.ID_ANY, 'Save results')
self.ExportButton.SetToolTip("If this is grayed out there is no (new) data to save!")
self.ExportButton.Bind(wx.EVT_BUTTON, self.OnExport)
self.ExportButton.Disable()
self.NameButton = wx.Button(self, wx.ID_ANY, 'Comment Results')
self.NameButton.Bind(wx.EVT_BUTTON, self.OnName)
username = "******"
if(getpass.getuser() in ["sdenk", "g2sdenk", "denk"]):
username = "******"
elif(getpass.getuser() == "bva"):
username = "******"
elif(getpass.getuser() == "mwillens"):
username = "******"
elif(getpass.getuser() == "sfreethy"):
username = "******"
self.ButtonSizer.Add(self.StartECRadButton, 0, wx.ALL | \
wx.LEFT, 5)
self.ButtonSizer.Add(self.KillECRadButton, 0, wx.ALL | \
wx.LEFT, 5)
self.ButtonSizer.Add(self.ExportButton, 0, wx.ALL | \
wx.LEFT, 5)
self.ButtonSizer.Add(self.NameButton, 0, wx.ALL | \
wx.LEFT, 5)
self.ControlSizer.Add(self.ButtonSizer, 0, wx.ALIGN_TOP)
self.Log_Box = wx.TextCtrl(self, wx.ID_ANY, size=(200, 100), \
style=wx.TE_MULTILINE | wx.TE_READONLY)
self.Log_Box.AppendText('Welcome to the ECRad GUI' + username + os.linesep)
self.diag_box_sizer = wx.BoxSizer(wx.VERTICAL)
self.diag_box_label = wx.StaticText(self, wx.ID_ANY, "Diagnostics")
self.DiagBox = wx.ListBox(self, wx.ID_ANY, size=(100, 100))
for diag_key in list(self.Results.Scenario["used_diags_dict"]):
self.DiagBox.Append(diag_key)
self.diag_box_sizer.Add(self.diag_box_label, 0, wx.ALL | wx.ALIGN_CENTER_HORIZONTAL , 5)
self.diag_box_sizer.Add(self.DiagBox, 1, wx.ALL | wx.EXPAND , 5)
self.time_box_sizer = wx.BoxSizer(wx.VERTICAL)
self.time_box_label = wx.StaticText(self, wx.ID_ANY, "Time points")
self.TimeBox = wx.ListBox(self, wx.ID_ANY, size=(100, 100))
for time in self.Results.Scenario["time"]:
self.TimeBox.Append("{0:1.5f}".format(time))
self.time_box_sizer.Add(self.time_box_label, 0, wx.ALL | wx.ALIGN_CENTER_HORIZONTAL , 5)
self.time_box_sizer.Add(self.TimeBox, 1, wx.ALL | wx.EXPAND , 5)
self.ControlSizer.Add(self.Log_Box, 1, wx.ALL | wx.EXPAND , 5)
self.ControlSizer.Add(self.diag_box_sizer, 0, wx.ALL | wx.EXPAND, 5)
self.ControlSizer.Add(self.time_box_sizer, 0, wx.ALL | wx.EXPAND, 5)
self.ProgressBar = wx.Gauge(self, wx.ID_ANY, style=wx.GA_HORIZONTAL)
self.Progress_label = wx.StaticText(self, wx.ID_ANY, "No ECRad run in progress")
self.Progress_sizer = wx.BoxSizer(wx.HORIZONTAL)
self.Progress_sizer.Add(self.ProgressBar, 1, wx.ALL | wx.EXPAND, 5)
self.Progress_sizer.Add(self.Progress_label, 0, wx.ALL, 5)
self.Redirector = Redirect_Text(self.Log_Box)
sys.stdout = self.Redirector
self.index = 0 # Index for the iteration over timepoints
self.UpperBook = wx.Notebook(self)
self.scenario_select_panel = ScenarioSelectPanel(self.UpperBook, self.Results.Scenario, self.Results.Config)
self.UpperBook.AddPage(self.scenario_select_panel, "Select scenario time points")
self.launch_panel = LaunchPanel(self.UpperBook, self.Results.Scenario, self.Results.Config["Execution"]["working_dir"])
self.UpperBook.AddPage(self.launch_panel, "Diagnostic configuration")
self.config_panel = ConfigPanel(self.UpperBook, self.Results.Config)
self.UpperBook.AddPage(self.config_panel, "ECRad configuration")
self.plot_panel = PlotPanel(self.UpperBook)
self.UpperBook.AddPage(self.plot_panel, "Misc. Plots")
self.sizer.Add(self.Progress_sizer, 0, wx.ALL | wx.EXPAND, 5)
self.SetScrollRate(20, 20)
if(globalsettings.AUG):
self.calib_panel = CalibPanel(self.UpperBook, self.Results.Scenario)
self.UpperBook.AddPage(self.calib_panel, "ECRad Calibration")
self.calib_evolution_Panel = CalibEvolutionPanel(self.UpperBook, self.Results.Config["Execution"]["working_dir"])
self.UpperBook.AddPage(self.calib_evolution_Panel, "Plotting for calibration")
self.sizer.Add(self.UpperBook, 1, wx.ALL | \
wx.LEFT, 5)
# def __del__(self):
# if self.ECRad_process is not None:
# self.ECRad_process.Detach()
# self.ECRad_process.CloseOutput()
# self.ECRad_process = None
# Overwrite this to stop the window from constantly jumping
def OnChildFocus(self, evt):
pass
def OnStartECRad(self, evt):
if(self.ECRad_running):
print('ECRad is still running - please wait!')
return
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('Preparing run - GUI might be unresponsive for minutes - please wait!')
self.GetEventHandler().ProcessEvent(evt)
try:
self.Results.Config = self.config_panel.UpdateConfig(self.Results.Config)
except ValueError as e:
print("Failed to parse Configuration")
print("Reason: ", e)
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('')
self.GetEventHandler().ProcessEvent(evt)
return
if(self.Results.Config["Physics"]["dstf"] not in ["Th", "Re", "Lu", "Ge", "GB"]):
print("Invalid choice of distribution")
print("Possible options:")
print("Th -> thermal plasma")
print("Re -> distribution function computed by RELAX")
print("Lu -> distribution function computed by LUKE (deprecated)")
print("Ge, GB -> distribution function computed by GENE (deprecated)")
print("Please select a valid distribution function identifier.")
return
scenario_updated = False
# Reset only the result fields but leave Scenario and Config untouched
self.Results.reset(light=True)
# Sets time points and stores plasma data in Scenario
if(self.scenario_select_panel.FullUpdateNeeded() or not self.Results.Scenario.plasma_set):
try:
self.Results.Scenario = self.scenario_select_panel.UpdateScenario(self.Results.Scenario, self.Results.Config, None)
if(not self.Results.Scenario.plasma_set):
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('')
self.GetEventHandler().ProcessEvent(evt)
return
except ValueError as e:
print("Failed to load Scenario")
print("Reason: ", e)
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('')
self.GetEventHandler().ProcessEvent(evt)
return
self.TimeBox.Clear()
for t in self.Results.Scenario["time"]:
self.TimeBox.Append("{0:1.4f}".format(t))
scenario_updated = True
else:
self.Results.Scenario = self.scenario_select_panel.SetScaling(self.Results.Scenario)
if(self.launch_panel.UpdateNeeded() or not self.Results.Scenario.diags_set):
try:
self.Results.Scenario = self.launch_panel.UpdateScenario(self.Results.Scenario)
if(not self.Results.Scenario.diags_set):
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('')
self.GetEventHandler().ProcessEvent(evt)
return
except ValueError as e:
print("Failed to parse diagnostic info")
print("Reason: ", e)
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('')
self.GetEventHandler().ProcessEvent(evt)
return
self.DiagBox.Clear()
for diag in list(self.Results.Scenario["used_diags_dict"]):
self.DiagBox.Append(diag)
scenario_updated = True
self.Results.Scenario.set_up_dimensions()
# Stores launch data in Scenario
self.stop_current_evaluation = False
self.index = 0
old_comment = self.Results.comment
self.Results.comment = old_comment # Keep the comment
if(len(list(self.Results.Scenario["used_diags_dict"].keys())) == 0):
print("No diagnostics selected")
print("Run aborted")
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('')
self.GetEventHandler().ProcessEvent(evt)
return
if(len(self.Results.Scenario["time"]) == 0):
print("No time points selected")
print("Run aborted")
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('')
self.GetEventHandler().ProcessEvent(evt)
return
if(self.Results.Config["Physics"]["dstf"] == "Re"):
if(self.Results.Scenario["dist_obj"] is None):
fileDialog=wx.FileDialog(self, "Selectr file with bounce averaged distribution data", \
defaultDir = self.Results.Config["Execution"]["working_dir"], \
wildcard="matlab files (*.mat)|*.mat",
style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST)
if(fileDialog.ShowModal() == wx.ID_CANCEL):
print("Launch aborted")
return
else:
pathname = fileDialog.GetPath()
self.Results.Scenario.load_dist_obj(pathname)
fileDialog.Destroy()
self.Results.Config.autosave()
if(self.Results.Config["Physics"]["dstf"] not in ["Ge", "GB"]):
self.FinalECRadSetup(scenario_updated)
else:
if(len(self.Results.Scenario["time"]) != 1):
print("For GENE distributions please select only one time point, i.e. the time point of the gene calcuation")
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('')
self.GetEventHandler().ProcessEvent(evt)
return
fileDialog=wx.FileDialog(self, "Selectr file with GENE distribution data", \
defaultDir = self.Results.Config["Execution"]["working_dir"], \
wildcard="hdf5 files (*.h5)|*.h5",
style=wx.FD_OPEN | wx.FD_FILE_MUST_EXIST)
if(fileDialog.ShowModal() == wx.ID_CANCEL):
print("Launch aborted")
return
else:
pathname = fileDialog.GetPath()
WorkerThread(self.LoadGeneData, [pathname])
fileDialog.Destroy()
def LoadGeneData(self, args):
pathname = args[0]
evt = GENEDataEvt(Unbound_EVT_GENE_DATA_LOADED, wx.ID_ANY)
if(self.Results.Scenario.load_GENE_obj(pathname, self.Results.Config["Physics"]["dstf"])):
evt.set_state(0)
else:
evt.set_state(-1)
wx.PostEvent(self, evt)
def OnGeneLoaded(self, evt):
if(evt.state == 0):
gene_dlg = Select_GENE_timepoints_dlg(self, self.Results.Scenario["GENE_obj"].time)
if(gene_dlg.ShowModal() == wx.ID_OK):
if(len(gene_dlg.used) == 0):
print("No time point selected")
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('')
wx.PostEvent(self, evt)
return
if(not self.Results.Scenario.integrate_GeneData(np.asarray(gene_dlg.used, dtype=np.float) * 1.e-3)):
print("GENE object not properly initialized - this is most likely due to a bug in the GUI")
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('')
wx.PostEvent(self, evt)
return
evt_out_2 = UpdateDataEvt(Unbound_EVT_UPDATE_DATA, self.GetId())
evt_out_2.SetResults(self.Results)
wx.PostEvent(self.scenario_select_panel, evt_out_2)
gene_dlg.Destroy()
self.TimeBox.Clear()
for time in self.Results.Scenario["time"]:
self.TimeBox.Append("{0:1.5f}".format(time))
self.FinalECRadSetup(True)
else:
print("Aborted")
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('')
wx.PostEvent(self, evt)
return
else:
print("Error when loading GENE data - see above")
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('')
wx.PostEvent(self, evt)
return
def FinalECRadSetup(self, scenario_updated):
# Thinga we need to do just before we get going
if(scenario_updated):
self.Results.Scenario.autosave()
self.Results.set_dimensions()
self.ProgressBar.SetRange(self.Results.Scenario["dimensions"]["N_time"])
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('ECRad is running - please wait.')
wx.PostEvent(self, evt)
self.Progress_label.SetLabel("ECRad running - ({0:d}/{1:d})".format(self.index + 1,len(self.Results.Scenario["time"])))
self.ProgressBar.SetValue(self.index)
self.ExportButton.Disable()
self.NameButton.Disable()
self.StartECRadButton.Disable()
evt = wx.PyCommandEvent(Unbound_EVT_MAKE_ECRAD, self.GetId())
wx.PostEvent(self, evt)
def OnProcessTimeStep(self, evt):
self.ECRad_running = True
self.ECRad_input_queue.put(["run ECRad", self.Results, self.index])
def ECRadRunner(cls, input_queue, output_queue):
ECRad_inferface = ECRadF2PYInterface()
while True:
args = input_queue.get()
try:
command = args[0]
if(command == "close"):
break
Results = args[1]
if(Results.Config["Execution"]["batch"]):
scratch_dir = Results.Config["Execution"]["scratch_dir"]
Results.Scenario.to_netcdf(filename=os.path.join(scratch_dir, "Scenario.nc"))
Results.Config.to_netcdf(filename=os.path.join(scratch_dir, "Config.nc"))
run_ECRad = SetupECRadBatch(Results.Config, Results.Scenario, Results.Scenario["time"][args[2]])
ECRad_batch = Popen(run_ECRad)
ECRad_batch.wait()
try:
filename, ed = Results.get_default_filename_and_edition(True)
NewResults = ECRadResults(False)
NewResults.from_netcdf(filename)
NewResults.to_netcdf()
output_queue.put([True, NewResults])
except:
print("Failed to run remotely. Please check .o and .e files at")
print(scratch_dir)
output_queue.put([False, Results])
else:
Results = ECRad_inferface.process_single_timepoint(Results, args[2])
output_queue.put([True, Results])
except Exception as e:
print(e)
output_queue.put([False, Results])
ECRadRunner = classmethod(ECRadRunner)
def OnProcessEnded(self, evt):
self.ECRad_running = False
if(evt.success):
if(self.Results.Config["Execution"]["batch"]):
self.index = self.Results.Scenario["dimensions"]["N_time"]
# Append times twice to track which time points really do have results in case of crashes
self.index += 1
else:
if(self.Results.Config["Execution"]["batch"]):
# If batch crashes we have to dump everything
self.index = 0
while self.index < self.Results.Scenario["dimensions"]["N_time"]:
self.Results.Scenario.drop_time_point(self.index)
else:
print("Sorry, ECRad crashed. Please send the log console output to Severin Denk")
print("Skipping current time point {0:1.4f} and continuing".format(self.Results.Scenario["time"][self.index]))
self.Results.Scenario.drop_time_point(self.index)
self.ProgressBar.SetRange(self.Results.Scenario["dimensions"]["N_time"])
if(self.index < len(self.Results.Scenario["time"]) and not self.stop_current_evaluation):
evt = NewStatusEvt(Unbound_EVT_NEXT_TIME_STEP, self.GetId())
wx.PostEvent(self, evt)
else:
self.FinishUpECRad()
def OnNextTimeStep(self, evt):
if(not self.stop_current_evaluation):
evt = wx.PyCommandEvent(Unbound_EVT_MAKE_ECRAD, self.GetId())
wx.PostEvent(self, evt)
else:
self.FinishUpECRad()
def FinishUpECRad(self):
if(self.Results.Scenario["dimensions"]["N_time"] == 0):
# Unsuccessful termination
print("None of the ECRad runs were completed succesfully - sorry")
self.Results = ECRadResults()
# We have deleted all entries from the Scenario we need to rebuilt this
self.Results.Scenario.plasma_set = False
self.Progress_label.SetLabel("No ECRad run in progress")
self.ProgressBar.SetRange(1)
self.ProgressBar.SetValue(1)
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('ECRad failed - sorry!')
self.GetEventHandler().ProcessEvent(evt)
self.stop_current_evaluation = True
self.StartECRadButton.Enable()
return
if(not self.Results.Config["Execution"]["batch"]):
# Remove time points in case of early termination
for i in range(self.index, self.Results.Scenario["dimensions"]["N_time"]):
self.Results.Scenario.drop_time_point(i)
self.Results.tidy_up(False)
self.TimeBox.Clear()
for t in self.Results.Scenario["time"]:
self.TimeBox.Append("{0:1.4f}".format(t))
self.KillECRadButton.Disable()
self.stop_current_evaluation = False
self.ExportButton.Enable()
self.NameButton.Enable()
self.Progress_label.SetLabel("No ECRad run in progress")
self.ProgressBar.SetRange(self.index)
self.ProgressBar.SetValue(self.index)
evt = wx.PyCommandEvent()
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('ECRad has Finished!')
wx.PostEvent(self, evt)
if(not self.Results.Config["Execution"]["batch"]):
print("Now saving results")
print("This takes a moment please wait")
WorkerThread(self.SavingThread, [])
evt_2 = UpdateDataEvt(Unbound_EVT_UPDATE_DATA, self.GetId())
evt_2.SetResults(self.Results)
if(globalsettings.AUG):
wx.PostEvent(self.calib_panel, evt)
self.calib_panel.GetEventHandler().ProcessEvent(evt_2)
wx.PostEvent(self.scenario_select_panel, evt_2)
wx.PostEvent(self.plot_panel, evt_2)
self.StartECRadButton.Enable()
def SavingThread(self, args):
try:
self.Results.autosave()
except Exception as e:
print("ERROR: Failed to save results")
print(e)
def OnKillECRad(self, evt):
self.stop_current_evaluation = True
print("Waiting for current calculation to finish")
evt = NewStatusEvt(Unbound_EVT_NEW_STATUS, self.GetId())
evt.SetStatus('Termination scheduled - please wait!')
self.GetEventHandler().ProcessEvent(evt)
self.KillECRadButton.Disable()
def OnUpdate(self, evt):
self.Results = evt.Results
self.DiagBox.Clear()
for diag_key in list(self.Results.Scenario["used_diags_dict"]):
self.DiagBox.Append(diag_key)
self.TimeBox.Clear()
for time in self.Results.Scenario["time"]:
self.TimeBox.Append("{0:1.5f}".format(time))
evt_out = UpdateConfigEvt(Unbound_EVT_UPDATE_CONFIG, self.GetId())
self.GetEventHandler().ProcessEvent(evt_out)
self.ExportButton.Enable()
self.NameButton.Enable()
print("Updated main results")
def OnImport(self, evt):
self.Results = ECRadResults()
print("Now loarding: " + evt.filename)
print("This takes a moment please wait")
WorkerThread(self.ImportThread, [evt.filename])
def ImportThread(self,args):
evt = ThreadFinishedEvt(Unbound_EVT_ECRAD_RESULT_LOADED, self.GetId())
try:
self.Results.load(args[0])
evt.SetSuccess(True)
except Exception as e:
print(e)
evt.SetSuccess(False)
wx.PostEvent(self, evt)
def OnResultsImported(self, evt):
if(evt.success):
print("Results loaded")
self.DiagBox.Clear()
for diag_key in self.Results.Scenario["used_diags_dict"]:
self.DiagBox.Append(diag_key)
self.TimeBox.Clear()
for time in self.Results.Scenario["time"]:
self.TimeBox.Append("{0:1.5f}".format(time))
evt_out = UpdateConfigEvt(Unbound_EVT_UPDATE_CONFIG, self.GetId())
evt_out.set_config(self.Results.Config)
evt_out.set_data_origin(self.Results.data_origin)
wx.PostEvent(self.config_panel, evt_out)
evt_out_2 = UpdateDataEvt(Unbound_EVT_UPDATE_DATA, self.GetId())
evt_out_2.SetResults(self.Results)
evt_out_2.SetPath(self.Results.data_origin)
if(globalsettings.AUG):
wx.PostEvent(self.calib_panel, evt_out_2)
wx.PostEvent(self.scenario_select_panel, evt_out_2)
wx.PostEvent(self.launch_panel, evt_out_2)
wx.PostEvent(self.plot_panel, evt_out_2)
else:
print("ERROR: Failed to load Results")
def OnExport(self, evt):
try:
try:
NewConfig = self.config_panel.UpdateConfig(self.Results.Config)
self.Results.Config["Execution"]["working_dir"] = NewConfig["Execution"]["working_dir"]
except ValueError as e:
print("Failed to parse Configuration")
print("Reason: ", e)
print("Did not update working directory")
if(self.Results is not None):
WorkerThread(self.Results.to_netcdf)
else:
print("No results to save")
except AttributeError as e:
print("No results to save")
print(e)
except IOError as e:
print("Failed to save results")
print(e)
def OnLockExport(self, evt):
self.ExportButton.Disable()
self.NameButton.Disable()
def OnIdle(self, evt):
if(self.ECRad_running):
try:
success, self.Results = self.ECRad_output_queue.get(block=False)
evt = ProccessFinishedEvt(Unbound_EVT_ECRAD_FINISHED, self.GetId())
evt.SetSuccess(success)
wx.PostEvent(self, evt)
return
except queue.Empty:
pass
if(not self.ECRad_runner_process.is_alive()):
self.ECRad_runner_process = Process(target=Main_Panel.ECRadRunner, \
args=(self.ECRad_input_queue, \
self.ECRad_output_queue))
self.ECRad_runner_process.start()
success = False
evt = ProccessFinishedEvt(Unbound_EVT_ECRAD_FINISHED, self.GetId())
evt.SetSuccess(success)
wx.PostEvent(self, evt)
# if(self.ECRad_process is not None):
# stream = self.ECRad_process.GetInputStream()
# if stream is not None:
# if stream.CanRead():
# text = stream.read()
# self.Log_Box.AppendText(text)
# evt.RequestMore()
# elif(self.ECRad_running):
# print("ECRad seems to have crashed without the corresponding event chain firing")
# print("This might cause some weird stuff from here on out")
# print("Trying to fix it...")
# self.ECRad_running = False
# self.OnProcessEnded(None)
def OnName(self, evt):
if(self.Results.comment == None):
comment = ""
else:
print(self.Results.comment)
comment = self.Results.comment
comment_dialogue = wx.TextEntryDialog(self, 'Please type comment for your calculation!', value=comment)
if(comment_dialogue.ShowModal() == wx.ID_OK):
self.Results.comment = comment_dialogue.GetValue()