-
Notifications
You must be signed in to change notification settings - Fork 1
/
extractRungLocation.py
44 lines (37 loc) · 1.94 KB
/
extractRungLocation.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
37
38
39
40
41
42
43
44
from oauth2client import tools
tools.argparser.add_argument("-m","--mouse", help="specify name of the mouse", required=False)
tools.argparser.add_argument("-d","--date", help="specify name of the mouse", required=False)
args = tools.argparser.parse_args()
import tools.extractSaveData as extractSaveData
import tools.dataAnalysis as dataAnalysis
import tools.openCVImageProcessingTools as openCVImageProcessingTools
import pdb
mouseD = '210214_m15'
expDateD = 'all910' # specific date e.g. '180214', 'some' for manual selection or 'all'
recordings='all910' # 'all or 'some'
# in case mouse, and date were specified as input arguments
if args.mouse == None:
mouse = mouseD
else:
mouse = args.mouse
if args.date == None:
try:
expDate = expDateD
except :
expDate = 'all'
else:
expDate = args.date
eSD = extractSaveData.extractSaveData(mouse)
(foldersRecordings,dataFolder) = eSD.getRecordingsList(expDate=expDate,recordings=recordings) # get recordings for specific mouse and date
cv2Tools = openCVImageProcessingTools.openCVImageProcessingTools(eSD.analysisLocation,eSD.figureLocation,eSD.f,showI=True)
# loop over all folders, mostly days but sometimes there were two recording sessions per day
for f in range(len(foldersRecordings)) :
# loop over all recordings in that folder
for r in range(len(foldersRecordings[f][2])): # for r in recordings[f][1]:
#print foldersRecordings[f][2][r]
(existence,fileHandle) = eSD.checkIfDeviceWasRecorded(foldersRecordings[f][0],foldersRecordings[f][1],foldersRecordings[f][2][r],'CameraGigEBehavior')
if existence:
rungPositions = cv2Tools.trackRungs(mouse,foldersRecordings[f][0],foldersRecordings[f][2][r],defineROI=False)
eSD.saveRungMotionData(mouse,foldersRecordings[f][0],foldersRecordings[f][2][r],rungPositions)
#cv2Tools.trackPawsAndRungs(mouse,foldersRecordings[f][0],foldersRecordings[f][2][r])
#pdb.set_trace()