-
Notifications
You must be signed in to change notification settings - Fork 41
/
02_4_final-cut.py
70 lines (53 loc) · 1.5 KB
/
02_4_final-cut.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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
# -*- coding: utf-8 -*-
import sys
import os
import glob
import cv
startframe = 0
def main():
global startframe
os.chdir(sys.argv[1])
os.chdir("shot_slitscans")
'''cv.NamedWindow("win", cv.CV_WINDOW_AUTOSIZE)
cv.MoveWindow("win", 500, 200)
cv.SetMouseCallback("win", mouse_callback)'''
bg_img = cv.CreateImage((576, 576), cv.IPL_DEPTH_8U, 1)
#cv.Set(bg_img, (180))
files = sorted( glob.glob("*.png") )
i = 0
while i < len(files):
file = files[i]
startframe = int( file.split("_")[3].split(".")[0] )
print startframe
cap = cv.CreateFileCapture(file)
img = cv.QueryFrame(cap)
win_name = "%d" % (int(float(i+1)*100.0/len(files))) + "% - " + file
cv.NamedWindow(win_name, cv.CV_WINDOW_AUTOSIZE)
cv.MoveWindow(win_name, 500, 200)
cv.SetMouseCallback(win_name, mouse_callback)
cv.ShowImage(win_name, bg_img)
cv.ShowImage(win_name, img)
key = cv.WaitKey(0)
if key == 2555904: # right arrow
i += 1
elif key == 2424832: # left arrow
i -= 1
if i < 0:
i = 0
elif key == 27: # ESC
break
cv.DestroyWindow(win_name)
os.chdir("../../..")
os.system("python 02_2_save-shots.py \"" + sys.argv[1] + "\"")
def mouse_callback(event, x, y, flags, param):
if event == 1: # left mouse down
# draw line?
pass
elif event == 4: # left mouse up
f = open("../shot_snapshots/%06d.png" % (startframe + x), "w")
print "cut added at %06d" % (startframe + x)
f.close()
# #########################
if __name__ == "__main__":
main()
# #########################