import matplotlib.pylab as plt import numpy as np import os from load_hdf5 import load_hdf5 import spectrum_wwind as spec import indexfinderfuncs as iff import compute_wavelet as cw # Sample rate and desired cutoff frequencies (in Hz). fs = 125e6 lowcut = 50e3 highcut = 2e6 directory = 'C:\\Users\\dschaffner\\Dropbox\\Data\\BMPL\\BMX\\2022\\01122022\\' datafilename = 'Dataset_01122022.h5' data = load_hdf5(directory + datafilename, verbose=True) time_s = data['time']['time_s'] timeB_s = time_s[1:] time_us = data['time']['time_us'] timeB_us = time_us[1:] analysis_start_time = 60 analysis_end_time = 160 start_time_index = iff.tindex_min(analysis_start_time, timeB_us) end_time_index = iff.tindex_min(analysis_end_time, timeB_us) timerange_limit = 3e-6 #s port_sep = 0.0254 #m numshots = 94 direction_list = ['r', 't', 'z'] probelist = ['probe5', 'probe7', 'probe19', 'probe21', 'probe33', 'probe35']
# Directory style depends on Mac vs PC. For PC, use a double backslash. # for a Mac, use a single forward slash. ### PC Style ### data_directory_location = 'C:\\Users\\dschaffner\\Dropbox\\Data\\BMPL\\BMX\\2019\\Correlation Campaign\\Encoding Converted for PC\\10142019\\processed\\' data_directory_location = 'C:\\Users\\dschaffner\\Dropbox\\Data\\BMPL\\BMX\\2019\\Correlation Campaign\\Encoding Converted for PC\\07022019\\processed\\' ### Mac Style ### #data_director_location = '/user/dschaffner/data/' ####################################################################### #place the following file in the directory indicated above datafilename = 'sample_2kV_oddpos1to13_2ms_stuffdelay_17shots_04232019.h5' datafilename = '2kV_0p5ms_1msGas_20shots_bdot_10142019.h5' #datafilename='2kV_0p5ms_10msGas_20shots_bdot_10142019.h5' datafilename = '2kV_oddpos9to21zr_2ms_stuffdelay_40_07022019.h5' #load hdf5 file data = load_hdf5(data_directory_location + datafilename, verbose=True) timeB_us = data['time']['timeB_us'][:] timeB_s = timeB_us / 1e6 #select time range for FFT (in us) start_time = 40 end_time = 90 time_range = [start_time, end_time] #compute indices from time range start_time_index = iff.tindex_min(start_time, timeB_us) end_time_index = iff.tindex_min(end_time, timeB_us) #select shots to analyze first_shot = 1 last_shot = 16 numshots = 40 shot_range = [first_shot, last_shot]
