def _intial_gui_setup(sys_argv):
global inifname # name of .ini file
inifname = os.curdir + '/.' + os.path.basename(__file__).replace(
'.py', '.ini')
for i, argv in enumerate(sys_argv):
print('arg {}: '.format(i) + argv)
if len(sys_argv) == 1:
default_ini = wpu.load_ini_file(inifname)
foldername = default_ini['Files']['Folder Name']
if easyqt.get_yes_or_no('Select new folder?\n' +
'[ESC load folder of previous run]'):
foldername = easyqt.get_directory_name(title='Select Folder that' +
'\ncontains *csv files')
elif len(sys_argv) == 2:
foldername = sys_argv[1]
else:
print('ERROR: wrong number of inputs: {} \n'.format(len(argv) - 1) +
'Usage: \n'
'\n' + os.path.basename(__file__) +
' : (no inputs) load dialogs \n'
'\n' + os.path.basename(__file__) + ' [args] \n'
'\n'
'arg1: folder that contains the *csv files\n'
'\n')
exit(-1)
wpu.set_at_ini_file(inifname, 'Files', 'Folder Name', foldername)
list_of_files = sorted(glob.glob(foldername + '/*csv'))
nfiles = len(list_of_files)
data, header_list = wpu.load_csv_file(list_of_files[0])
wpu.print_blue('MESSAGE: Header of data files:')
wpu.print_blue(header_list)
ncurves = data.shape[1] - 1
label_list = [
fname.rsplit('_', 1)[1].split('.')[0] for fname in list_of_files
]
#[fname.rsplit('/', 1)[1].split('_Y_integrated')[0]
# for fname in list_of_files]
return list_of_files, nfiles, ncurves, label_list, header_list
listOfFiles = glob.glob(dirName + '/*.csv')
listOfFiles.sort()
n_files = len(listOfFiles)
# %% Load data
listOfArrays = []
listOfShapes = []
for fname in listOfFiles:
wpu.print_blue('MESSAGE: Open File ' + fname)
if use_cap:
fileContent = load_cap_file(fname)
else:
fileContent = wpu.load_csv_file(fname)
listOfArrays.append(fileContent[0])
listOfShapes.append(np.shape(fileContent[0]))
headers = fileContent[1]
# %%
for data, fname in zip(listOfArrays, listOfFiles):
label = fname.rsplit('/', 1)[1].split('.')[0]
print(
label +
', rms value: {:.4f} nm'.format(np.std(data[:, profilenumber]) * 1e9))
# here to process the cap sensor data to get the relative response
if use_cap:
# %%
dirName = easyqt.get_directory_name()
print(dirName)
listOfFiles = glob.glob(dirName + '/*.csv')
listOfFiles.sort()
n_files = len(listOfFiles)
# %% Load data
listOfArrays = []
listOfShapes = []
for fname in listOfFiles:
fileContent = wpu.load_csv_file(fname)
listOfArrays.append(fileContent[0])
listOfShapes.append(np.shape(fileContent[0]))
headers = fileContent[1]
# %%
plt.figure(figsize=(12, 8))
for data, fname in zip(listOfArrays, listOfFiles):
label = fname.rsplit('/', 1)[1].split('.')[0]
plt.plot(data[:, 0] * 1e6, data[:, 1], 'o', label=label)
header_list] = _intial_gui_setup(sys.argv)
saveFlag = True
step_files = 1
addThisToTitle = '\nVertical Profile, Projected WF, grazing_angle = 2.6 mrad'
# %%
xmax = 0
xmin = 1e6
list_all_data = []
for fname in list_of_files:
data, _ = wpu.load_csv_file(fname)
# data[:,1:] *= np.pi
list_all_data.append(data)
xmin = np.min([xmin, np.min(data[:, 0])])
xmax = np.max([xmax, np.max(data[:, 0])])
# cant convert list_all_data to array because each array in the list can
# have different sizes. It is necessary to do the spline to have same
# vector sizes
# %%
lstyle_cycle, lc_cycle = wpu.line_style_cycle(ls=['-', '--'],