return lags, msd_vals
if __name__ == '__main__':
# Load dataframe
# filename = filedialog.askopenfilename(initialdir='/media/ppzmis/data/BouncingBall_Data/newMovies/ProcessedData/finalProcessed/',title='Select Data File', filetypes = (('DataFrames', '*finaldata.hdf5'),))
basepath = '/media/ppzmis/data/BouncingBall_Data/newMovies/ProcessedData/finalProcessed_new10mm/*'
names = ['P120_045']
#names = ['800_050','400_050','240_050','120_050','80_050','800_077','800_062','800_040','400_077','400_062','400_040','240_077','240_062','240_040','120_077','120_062','120_040','80_077','80_062','80_040']
#names = ['P80_077']
pathnames = [basepath + name + '_data_finaldata.hdf5' for name in names]
# each path get the 3 files for a particular experiment.
print(pathnames)
for path in pathnames:
filenames = get_files_directory(path, full_filenames=True)
print(filenames)
# Read dataframe from file
msd_av = np.array([])
filename_op = min(filenames)[:-5]
for i, filename in enumerate(filenames):
data = pd.read_hdf(filename)
lags, msd_val = msd(data, show=True)
if i == 0:
msd_av = msd_val
elif i == 1:
msd_av = (msd_av + msd_val)
elif i == 2:
pass #msd_av = (msd_av + msd_val)/3
data = calcGradient(data, 'ballXMM', 'xVelMM')
return data
if __name__ == '__main__':
#Set to True if the ball doesn't bounce. Analysis will assume a mean bounce time of 10 frames.
no_bouncing = False
show_plots = False
#smoothing window length
window_length = 12
#Load dataframe
#filename = filedialog.askopenfilename(initialdir='/media/ppzmis/data/BouncingBall_Data/newMovies/ProcessedData/InitProcessed/',title='Select Data File', filetypes = (('DataFrames', '*040_data_initialprocessed.hdf5'),))
path = '/media/ppzmis/data/BouncingBall_Data/newMovies/ProcessedData/InitProcessed/*initialprocessed.hdf5' #*.initialprocessed.hdf5'
#path = path + 'bP240_077*initialprocessed.hdf5'
file_list = get_files_directory(path)
for filename in file_list:
#'/media/ppzmis/data/BouncingBall_Data/newMovies/ProcessedData/InitProcessed/P80_062_data_initialprocessed.hdf5'
new_folder = filename[:-22] + '_analyse'
try:
os.mkdir(new_folder) #Make directory to store fit graphs
except:
pass
#if '_040_' in filename:
# no_bouncing = True
#else:
# no_bouncing = False