-
Notifications
You must be signed in to change notification settings - Fork 2
/
dataProcess.py~
38 lines (31 loc) · 1.01 KB
/
dataProcess.py~
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import numpy as np
import matplotlib.pylab as plt
import matplotlib.cm as colormap
import os.path
import tkFileDialog
import sys
import videoAnalysisLib as va
import dataAnalysisLib as da
#activeDir = '/lmb/home/tbranco/analysis/114491a_foraging'
activedir = sys.argv[1]
datafname = activedir+'/trackingData.txt'
avifname = activedir+'/trackingMovie.avi'
aviProps = va.getAVIinfo(avifname)
try:
bg = np.load(activedir+'/bg.npy')
print("Background file loaded")
except IOError:
print("Cannot find background file ")
try:
pmts = np.load(activedir+'/pmts.npy')
print("Parameters file loaded")
except IOError:
print("Cannot find parameters file ")
# Basic tracking data
print("Processing data...")
data, resultsDict, fig = va.analyseData(datafname, aviProps, bg, pmts)
va.writeDict(resultsDict, activedir+'/trackingAnalysis.txt')
fig.savefig(activedir+'/trackingFig.pdf')
# Further analysis
pMatrix, xhist, yhist = da.getArenaMatrix(data, aviProps, plot=False)
np.save(activedir+'/arenaMatrix', [pMatrix, xhist, yhist])