# Live video output (display video on the screen, presumably from the net)

import gl
from VFile import Displayer


# Video output (displayer) class.

class LiveVideoOut:

	# Call this to initialize things.  Arguments:
	# wid:    the window id where the video is to be displayed (centered)
	# vw, vh: size of the video image to be displayed

	def init(self, wid, vw, vh, type):
		##print 'Init', wid, xywh
		##print 'video', vw, vw
		self.vw = vw
		self.vh = vh
		self.disp = Displayer().init()
		if not type in ('rgb8', 'grey', 'mono', 'grey2', 'grey4'):
			raise 'Incorrent live video output type', type
		info = (type, vw, vh, 1, 8, 0, 0, 0, 0)
		self.disp.setinfo(info)
		self.wid = wid
		oldwid = gl.winget()
		gl.winset(wid)
		self.disp.initcolormap()
		self.reshapewindow()
		gl.winset(oldwid)
		return self

	# Call this in response to every REDRAW event for the window
	# or if the window size has changed for other reasons.

	def reshapewindow(self):
		oldwid = gl.winget()
		gl.winset(self.wid)
		gl.reshapeviewport()
		w, h = gl.getsize()
		self.disp.xorigin = (w-self.vw)/2
		self.disp.yorigin = (h-self.vh)/2
		self.disp.clear()
		gl.winset(oldwid)

	# Call this to change the size of the video images being displayed.
	# Implies reshapewindow().

	def resizevideo(self, vw, vh):
		self.vw, self.vh = vw, vh
		self.disp.setsize(vw, vh)
		self.reshapewindow()

	# Call this to display the next video packet.  Arguments:
	# pos:  line number where the packet begins
	# data: image data of the packet
	# (these correspond directly to the return values from
	# LiveVideoIn.getnextpacket()).

	def putnextpacket(self, pos, data):
		if self.disp.format == 'mono':
			# Unfortunately size-check is difficult for
			# packed video
			nline = (len(data)*8)/self.vw
		elif self.disp.format == 'grey2':
			nline = (len(data)*4)/self.vw
		elif self.disp.format == 'grey4':
			nline = (len(data)*2)/self.vw
		else:
			nline = len(data)/self.vw
			if nline*self.vw <> len(data):
				print 'Incorrect-sized video fragment ignored'
				return
		oldwid = gl.winget()
		gl.winset(self.wid)
		self.disp.showpartframe(data, None, (0, pos, self.vw, nline))
		gl.winset(oldwid)

	# Call this to close the window.

	def close(self):
		pass

	# Call this to set optional mirroring of video
	def setmirror(self, mirrored):
		f, w, h, pf, c0, c1, c2, o, cp = self.disp.getinfo()
		if type(pf) == type(()):
			xpf, ypf = pf
		else:
			xpf = ypf = pf
		xpf = abs(xpf)
		if mirrored:
			xpf = -xpf
		info = (f, w, h, (xpf, ypf), c0, c1, c2, o, cp)
		self.disp.setinfo(inf0)