Python Writer.stop方法代码示例

# 需要导入模块: from writer import Writer [as 别名]
# 或者: from writer.Writer import stop [as 别名]

#.........这里部分代码省略.........
		settings.update(kwargs)
		trialDir = "{0}_{1}_{2}".format(time.strftime("%y%m%d%H%M%S"),
				settings['experimentName'], settings['stimulator']['activeProtocolName'])
		settings['outputDataDir'] = os.path.join(settings['outputRootDir'], trialDir)
		self.keyBindings = settings['keyBindings']
		self.keycodeBindings = {k: getattr(opencvgui.keycodes, v)
				for k,v in self.keyBindings.items() }
		os.makedirs(os.path.join(settings['outputDataDir'], 'audio'))
		self.writeExperimentData(settings)
		for x in ['audio', 'videoOut', 'writer']:
			kwargs[x] = {} if not kwargs.has_key(x) else kwargs[x]
			kwargs[x]['outputDataDir'] = settings['outputDataDir']

		# gui and writer both need a reference to the app instance because they
		# draw on the state of so many other controllers
		self.audio = Audio(**settings['audio'])
		self.gui = Gui(self, **settings['gui'])
		self.stimulator = Stimulator(**settings['stimulator'])
		self.tracker = Tracker(**settings['tracker'])
		self.videoIn = VideoIn(**settings['videoIn'])
		self.videoOut = VideoOut(**settings['videoOut'])
		self.writer = Writer(self, **settings['writer'])

	def run(self):
		"""Respond to user input and run the master application loop.

		This method has three basic sections. Before the loop starts, the start()
		method is called on all controllers. The central loop calls update() on
		each controller until it is terminated by a user-issued quit keystroke.
		Keystrokes are listened for at the start of each loop iteration.
		"""
		self.printKeybindings()
		self.startTime = self.lastTime = time.clock()
		self.audio.start()
		self.gui.start()
		self.stimulator.start()
		self.tracker.start()
		self.videoIn.start()
		self.videoOut.start()
		self.writer.start()

		while True:

			# respond to user input
			lastKeyStroke = cv2.waitKey(20)  # 20 is the number of ms to wait for key
			if lastKeyStroke != -1: # -1 means there was no keystroke
				if lastKeyStroke == self.keycodeBindings['quit']:
					break
				if lastKeyStroke == self.keycodeBindings['triggerStimulator']:
					self.stimulator.trigger()
				if lastKeyStroke == self.keycodeBindings['toggleStimulator']:
					self.stimulator.toggle()

			# update state
			self.currTime = time.clock()
			self.totalTimeElapsed = self.currTime - self.startTime
			self.audio.update()
			self.videoIn.update()
			self.tracker.update(self.videoIn, self.currTime)
			self.stimulator.update(self.currTime, self.tracker)
			self.videoOut.update(self.videoIn, self.tracker, self.stimulator)
			self.gui.update()
			self.lastTime = self.currTime
			self.writer.update()

		# closing
		self.audio.stop()
		self.videoIn.stop()
		self.videoOut.stop()
		self.gui.stop()
		self.stimulator.stop()
		self.tracker.stop()
		self.writer.stop()

	def writeExperimentData(self, settings, console=True):
		"""Write experiment metadata to a file in the output directory.
		
		Args:
			settings: The full settings dictionary passed to the App's __init__
			console: Optional. Boolean. Also print experiment data to console.
		"""
		with open(os.path.join(settings['outputDataDir'], 'experiment.txt'), 'w') as f:
			lines = [
					"name: " + settings['experimentName'],
					"date: " + time.strftime("%m/%y/%d"),
					"time: " + time.strftime("%H:%M"),
					"protocol: "  + settings['stimulator']['activeProtocolName'],
					str(settings['stimulator']['protocols'][settings['stimulator']['activeProtocolName']])
					]
			f.write("\n".join(lines))
			f.close()
		if console:
			print "\n".join(lines)
	
	def printKeybindings(self):
		"""Print the available keyboard commands to the terminal."""
		print "Key Bindings:"
		for k,v in self.keyBindings.items():
			print "{0}: {1}".format(k,v)
		print ""