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=['-', '--'],