class TestHyperlapse(unittest.TestCase):
def setUp(self):
video = Video()
video.setVideoFile('/home/victorhugomoura/Documents/example.mp4')
self.hyperlapse = SemanticHyperlapse()
self.hyperlapse.setVideo(video)
self.hyperlapse.setPaths()
def testVideo(self):
self.assertIsInstance(self.hyperlapse.getVideo(), Video)
def testPath(self):
self.assertEqual(self.hyperlapse.getPath(), os.getcwd())
self.hyperlapse.setPath('/')
self.assertEqual(self.hyperlapse.getPath(), '/')
def testVelocity(self):
self.assertRaises(InputError, self.hyperlapse.setVelocity, '')
self.assertRaises(InputError, self.hyperlapse.setVelocity, 'A')
self.assertRaises(InputError, self.hyperlapse.setVelocity, '1')
self.hyperlapse.setVelocity('10')
self.assertEqual(self.hyperlapse.getVelocity(), 10)
def testMaxVel(self):
self.hyperlapse.setMaxVel(100.0)
self.assertEqual(self.hyperlapse.getMaxVel(), 100)
def testCheckWeights(self):
self.assertRaises(ValueError, self.hyperlapse.convertWeights, ['', ''])
self.assertRaises(ValueError, self.hyperlapse.convertWeights,
['a', '10'])
self.assertRaises(InputError, self.hyperlapse.checkAndSetWeights,
['', ''])
self.assertListEqual(self.hyperlapse.checkAndSetWeights(['10', '10']),
[10, 10])
def testWeights(self):
self.hyperlapse.setAlpha(['50', '50'])
self.hyperlapse.setBeta(['20', '20'])
self.hyperlapse.setGama(['30', '30'])
self.hyperlapse.setEta(['40', '50'])
self.assertListEqual(self.hyperlapse.getAlpha(), [50, 50])
self.assertListEqual(self.hyperlapse.getBeta(), [20, 20])
self.assertListEqual(self.hyperlapse.getGama(), [30, 30])
self.assertListEqual(self.hyperlapse.getEta(), [40, 50])
def testOpticalFlowCommand(self):
command = self.hyperlapse.opticalFlowCommand()
expectedCommand = './optflow -v /home/victorhugomoura/Documents/example.mp4 ' + \
'-c default-config.xml -o /home/victorhugomoura/Documents/example.csv'
self.assertEqual(command, expectedCommand)
def testCheckVideoInput(self):
self.hyperlapse.getVideo().setVideoFile('')
self.assertRaises(InputError, self.hyperlapse.checkVideoInput)
self.hyperlapse.getVideo().setVideoFile(
'/home/victorhugomoura/Documents/example.csv')
self.assertRaises(InputError, self.hyperlapse.checkVideoInput)
def testSetup(self):
alpha = ['1', '2']
beta = ['4', '3']
gama = ['5', '6']
eta = ['8', '7']
speed = '10'
# check if it don't raise an exception
self.hyperlapse.setup(speed, alpha, beta, gama, eta)
speed = '1'
self.assertRaises(InputError, self.hyperlapse.setup, speed, alpha,
beta, gama, eta)
def testInputError(self):
self.hyperlapse.getVideo().setVideoFile('')
try:
self.hyperlapse.checkVideoInput()
except InputError as IE:
self.assertEqual(IE.__str__(), 'Please insert input video first')
class MainWindow(object):
def __init__(self, root):
self.root = root
self.inputFile = []
self.speedUp = None
self.weights = []
self.errorLabel = None
self.buttons = []
self.logText = None
self.hyperlapse = None
def start(self):
self.setInputFileLabel()
self.setSpeedUpEntry()
self.setWeightsEntries()
self.setErrorLabel()
self.setButtons()
self.setHyperlapse()
self.setMenu()
self.root.title('Semantic Hyperlapse')
self.root.mainloop()
def setInputFileLabel(self):
Label(self.root, text = 'Input File: ').grid(row=0, sticky=W)
self.inputFile.append(StringVar())
self.inputFile.append(Label(self.root, textvariable = self.inputFile[0]))
self.inputFile[1].grid(row=0, column=1, columnspan=2)
def setSpeedUpEntry(self):
Label(self.root, text = 'Final Speed: ').grid(row=1, sticky=W)
self.speedUp = tk.Entry(self.root, width=4)
self.speedUp.grid(row=1, column=1, columnspan=2, sticky=W+E+N+S)
def setWeightsEntries(self):
Label(self.root, text = 'α Weights: ').grid(row=2, sticky=W)
self.weights.append([tk.Entry(self.root, width=4) for _ in range(2)])
[alpha.insert(0, '50') for alpha in self.weights[0]]
Label(self.root, text = 'β Weights: ').grid(row=3, sticky=W)
self.weights.append([tk.Entry(self.root, width=4) for _ in range(2)])
[beta.insert(0, '50') for beta in self.weights[1]]
Label(self.root, text = 'γ Weights: ').grid(row=4, sticky=W)
self.weights.append([tk.Entry(self.root, width=4) for _ in range(2)])
[gama.insert(0, '50') for gama in self.weights[2]]
Label(self.root, text = 'η Weights: ').grid(row=5, sticky=W)
self.weights.append([tk.Entry(self.root, width=4) for _ in range(2)])
[eta.insert(0, '50') for eta in self.weights[3]]
for i in range(len(self.weights)):
for j in range(len(self.weights[i])):
self.weights[i][j].grid(row=i+2, column=j+1)
def setErrorLabel(self):
self.errorLabel = Label(self.root, text = '')
def setButtons(self):
self.buttons.append(Button(root, text='Speed Up Video', command=self.preProcessAndRun))
self.buttons[0].grid(row=6, column=1)
def setHyperlapse(self):
self.hyperlapse = SemanticHyperlapse()
self.hyperlapse.setVideo(Video())
def setMenu(self):
menubar = Menu(self.root)
filemenu = Menu(menubar, tearoff=0)
filemenu.add_command(label='OpenFile', command=self.openFile)
filemenu.add_separator()
filemenu.add_command(label='Exit', command=self.root.quit)
menubar.add_cascade(label='File', menu=filemenu)
helpmenu = Menu(menubar, tearoff=0)
helpmenu.add_command(label='Help Index', command=self.help)
helpmenu.add_command(label='About...', command=self.about)
menubar.add_cascade(label='Help', menu=helpmenu)
self.root.config(menu=menubar)
def about(self):
aboutscreen = tk.Toplevel(self.root)
text = tk.Text(aboutscreen, wrap=WORD)
scroll = tk.Scrollbar(self.root, command=text.yview)
text.configure(yscrollcommand=scroll.set)
text.tag_configure('italics', font=('Arial', 10, 'italic'))
text.tag_configure('title', font=('Arial', 15, 'bold'))
text.tag_configure('normal', font=('Arial', 10))
text.insert(tk.END, 'About\n\n', 'title')
text.insert(tk.END, 'See more about the project on:\n', 'normal')
text.insert(tk.END, '\thttps://github.com/verlab/SemanticFastForward_ICIP_2016\n', 'italics')
text.configure(state='disabled')
text.pack()
aboutscreen.title('About')
def help(self):
helpscreen = tk.Toplevel(self.root)
text = tk.Text(helpscreen, wrap=WORD)
scroll = tk.Scrollbar(self.root, command=text.yview)
text.configure(yscrollcommand=scroll.set)
text.tag_configure('bold_italics', font=('Arial', 10, 'bold', 'italic'))
text.tag_configure('title', font=('Arial', 15, 'bold'))
text.tag_configure('normal', font=('Arial', 10))
text.insert(tk.END, 'Instructions:\n\n', 'title')
text.insert(tk.END, '1: ', 'bold_italics')
text.insert(tk.END, 'Go to File->OpenFile and select the video file that will be accelerated.\n', 'normal')
text.insert(tk.END, '2: ', 'bold_italics')
text.insert(tk.END, 'Choose the speedup (speedup > 1).\n', 'normal')
text.insert(tk.END, '3: ', 'bold_italics')
text.insert(tk.END, 'Choose graph weights α, β, γ, η. If you don\'t change anything, the default parameters will be used.\n', 'normal')
text.insert(tk.END, '4: ', 'bold_italics')
text.insert(tk.END, 'Click on \'Speed Up Video\' and see the progress while your video is being accelerated.\n', 'normal')
text.configure(state='disabled')
text.pack()
helpscreen.title('Help')
def openFile(self):
self.hyperlapse.video.setVideoFile(tkFileDialog.askopenfilename(filetypes=([('All files', '*.*'), ('MP4 files', '*.mp4'), ('AVI files', '*.avi')])))
self.hyperlapse.setPaths()
self.inputFile[0].set(self.hyperlapse.video.getVideoName())
def preProcess(self):
speed = self.speedUp.get()
alpha = [a.get() for a in self.weights[0]]
beta = [b.get() for b in self.weights[1]]
gama = [g.get() for g in self.weights[2]]
eta = [e.get() for e in self.weights[3]]
self.hyperlapse.setup(speed, alpha, beta, gama, eta)
def run(self):
self.createLogWindow()
self.hyperlapse.run(self.addLog)
def preProcessAndRun(self):
try:
self.preProcess()
self.run()
except InputError as IE:
self.errorLabel['text'] = IE.msg
self.errorLabel.grid(row=7, columnspan=3)
def addLog(self, text, textType):
self.logText.configure(state='normal')
self.logText.insert('end', text, textType)
self.logText.configure(state='disabled')
self.logText.update_idletasks()
def createLogWindow(self):
log = tk.Toplevel(self.root)
self.logText = tk.Text(log, wrap=WORD)
scroll = tk.Scrollbar(log, command=self.logText.yview)
self.logText.configure(yscrollcommand=scroll.set)
self.logText.tag_configure('italics', font=('Arial', 10, 'italic'))
self.logText.tag_configure('title', font=('Arial', 12, 'bold'))
self.logText.tag_configure('normal', font=('Arial', 10))
self.logText.pack()
self.logText.update_idletasks()
log.title('Semantic Hyperlapse Status')