i = T_min if i == T_min: if i < 10: T_id = '0000' + str(i) elif i < 100 and i > 9: T_id = '000' + str(i) elif i < 1000 and i > 99: T_id = '00' + str(i) elif i < 10000 and i > 999: T_id = '0' + str(i) else: print('T > 9999?') sys.exit(0) #read file #1 outputs = srd.rd_file(T_id, var_tit1, path_id, denor) barh = np.max(outputs[8]) barl = np.min(outputs[8]) if i == T_min: bar1h = barh bar1l = barl else: bar1h = cptwo(bar1h, barh, 1) bar1l = cptwo(bar1l, barl, 2) #file #2 outputs = srd.rd_file(T_id, var_tit2, path_id, denor) barh = np.max(outputs[8]) barl = np.min(outputs[8]) if i == T_min:
else: print('Dimension should be <= 2D!') sys.exit(0) return p ###################################################################'read data' denor = 0 'denormalize in def-Input' plog = -1 'plog = 1 logspace levels, =0, linspace levels, =2 pcolormesh' print('--- [', var_tit, '] reading time:', T_id, ' ---') '''start here''' outputs = srd.rd_file(T_id, var_tit, path_id, denor) title = outputs[0] x_title = outputs[1] y_title = outputs[2] t_out = outputs[3] num_x = outputs[4] num_y = outputs[5] x_axis = outputs[6] y_axis = outputs[7] data = outputs[8] print() print('--- Dimension @time= %3.2f: ---' % (t_out)) print('X(%i): [%6.2f,%6.2f], mid x: (%4.1e --- %4.1e)'%(\
pos_x = 0 pos_y = 0 v_drift = 0 for i in range(0, T_max): if i < 10: T_id = '0000' + str(i) elif i < 100 and i > 9: T_id = '000' + str(i) elif i < 1000 and i > 99: T_id = '00' + str(i) else: print('T > 9999?') sys.exit(0) #read files outputs1 = srd.rd_file(T_id, var_tit1, path_id) outputs2 = srd.rd_file(T_id, var_tit2, path_id) data1[i] = outputs1[8][pos_x, pos_y] data2[i] = outputs2[8][pos_x, pos_y] tim[i] = outputs1[3] if i == 1: dt = tim[1] - tim[0] posx[1] = posx[0] + dt * data1[0] posy[1] = posy[0] + dt * data2[0] if i > 0: posx[i + 1] = posx[i] + dt * data1[i] posy[i + 1] = posy[i] + dt * (data2[i] + v_drift) ''' ''' vmin = np.min([np.min(data1), np.min(data2)]) vmax = np.max([np.max(data1), np.max(data2)])
import matplotlib.pyplot as plt import matplotlib as mpl from matplotlib import cm, colors from mpl_toolkits.mplot3d import Axes3D #for my mac... import warnings warnings.simplefilter(action="ignore", category=FutureWarning) warnings.simplefilter(action="ignore", category=UserWarning) #debug import sys import imp ''' for o_fsp ''' '''start here''' outputs = srd.rd_file(T_id, var_tit, path_id, 0) title = outputs[0] x_title = outputs[1] y_title = outputs[2] t_out = outputs[3] num_x = outputs[4] num_y = outputs[5] x_axis = outputs[6] y_axis = outputs[7] data0 = outputs[8] outputs = srd.rd_file(T_id, var_tit, path_id, 1) data1 = outputs[8] outputs = srd.rd_file(T_id, var_tit, path_id, 2) data2 = outputs[8]