def generate_error_graph( err_dirs, temperature_array, velocity_array, md_threshold_array, temperature_dict, flow_velocity_dict, md_threshold_dict, legendPosition='best', legendParam='None', graphPties='None', width=3, figPath='None', show=True, logScale=True, plot_ylim='None'): ''' @description generate the graph showing the maximum of the error as a function of time ''' # extract the error data data = [] for md_threshold in md_threshold_array: for flow_velocity in flow_velocity_array: for temperature in temperature_array: # get the path to the error data T_i = temperature_dict[str(temperature)] v_i = flow_velocity_dict[str(flow_velocity)] md_i = md_threshold_dict[str(md_threshold)] errorPath = err_dirs[T_i][v_i][md_i] # extract the error data [time_rescaled,error] = extract_max_error_in_time(errorPath) # stored the error data for the graph data.append([time_rescaled,error]) # create the graph create_error_graph( data, legendPosition=legendPosition, legendParam=legendParam, graphPties=graphPties, width=width, figPath=figPath, show=show, logScale=logScale, plot_ylim=plot_ylim)
def generate_error_position_graph( temperature, flow_velocity, md_threshold=0, width=3, figPath='None', show=True, logScale=False, plot_ylim='None'): ''' @description generate the figure showing the maximum of the mass error and the y-position as functions of time ''' T_i = temperature_dict[str(temperature)] v_i = flow_velocity_dict[str(flow_velocity)] md_i = md_threshold_dict[str(0)] errorPath = err_dirs[T_i][v_i][md_i] data = extract_max_error_in_time( errorPath, var_names=['time','max_error_mass','y_max_error_mass']) dataError = [data[0],data[1]] dataPosition = [data[0],data[2]] create_error_graph_with_location( dataError, dataPosition, legendParam='None', graphPties=[['black','-'],[grayscale_to_RGB(0.50),'-']], width=width, figPath=figPath_error, show=show, logScale=logScale, plot_ylim=plot_ylim)
data_error.append([]) data_error.append([]) for dct_distance in dctDistance_array: # get the path for the error file dir_name = get_simulation_dir(temperature, flow_velocity, dct_distance=dct_distance) errorPath = os.path.join(mainDir,dir_name,'error','error_max.nc') if(os.path.isfile(errorPath)): # extract the error data [time_rescaled,error] = extract_max_error_in_time(errorPath) # stored the error data for the graph data.append([time_rescaled,error]) #set the legend properties legendParam.append(dct_distance) # set the graph properties T_i = temperature_dict[str(temperature)] dct_i = dctDistance_dict[str(dct_distance)] graphPties.append([color_temperature[T_i],style_detector[dct_i]]) else: print errorPath+' does not exist'