print('\n Analysing the following folders:\n') print(expDirs) # generate fly color map numFlies = len(expDirs) # Go through directories of experiments with different flies ........................................................... for currExpDir in expDirs: expDir = dataDir + currExpDir + sep print('\n Analysing the following folder:\n') print(expDir) FODataFiles = [filepath.split(sep)[-1] for filepath in glob(expDir + '*.txt')] FODataFiles = sorted(FODataFiles) print('\n Analysing the following log files:\n') print(FODataFiles) # Run single trial analysis on each file in folder ..................................................... for fileToAnalyse in FODataFiles: flyID = fileToAnalyse.split('_')[0] trial = FODataFiles.index(fileToAnalyse) + 1 returnVal = singleVRStripeTrialAnalysis(expDir + fileToAnalyse) print('Ran analysis, now loading saved *.npy file.') FODatLoad = np.load(expDir + fileToAnalyse[:-3] + 'npy')[()] print('Analysis ran successfully')
root.withdraw() # choose experiment folder expDir = askdirectory(initialdir=baseDir, title='Select experiment directory of a single fly') + sep dataDir = sep.join(expDir.split(sep)[0:-2]) + sep flyID = expDir.split(sep)[-2] # create analysis dir analysisDir = sep.join(dataDir.split(sep)[:-2]) + sep + 'analysis' + sep try: mkdir(analysisDir) except OSError: print('Analysis directory already exists.') FODataFiles = [filepath.split(sep)[-1] for filepath in glob(expDir + '*.txt')] FODataFiles = sorted(FODataFiles) print('\n Analysing the following log files:\n') print(FODataFiles) for FODataFile in FODataFiles: # ------------------------------------------------------------------------------------------------------------------ # Run for each file in folder # ------------------------------------------------------------------------------------------------------------------ returnVal = singleVRStripeTrialAnalysis(expDir+FODataFile) # test that npy file got saved FODatLoad = np.load(expDir + FODataFile[:-3] + 'npy')[()] print('Analysis ran successfully!')