Python code coverage for Lib/plat-mac/videoreader.py

#	count	content
1	n/a	# Video file reader, using QuickTime
2	n/a	#
3	n/a	# This module was quickly ripped out of another software package, so there is a good
4	n/a	# chance that it does not work as-is and it needs some hacking.
5	n/a	#
6	n/a	# Jack Jansen, August 2000
7	n/a	#
8	n/a
9	n/a	from warnings import warnpy3k
10	n/a	warnpy3k("In 3.x, the videoreader module is removed.", stacklevel=2)
11	n/a
12	n/a
13	n/a	import sys
14	n/a	from Carbon import Qt
15	n/a	from Carbon import QuickTime
16	n/a	from Carbon import Qd
17	n/a	from Carbon import Qdoffs
18	n/a	from Carbon import QDOffscreen
19	n/a	from Carbon import Res
20	n/a	try:
21	n/a	from Carbon import MediaDescr
22	n/a	except ImportError:
23	n/a	def _audiodescr(data):
24	n/a	return None
25	n/a	else:
26	n/a	def _audiodescr(data):
27	n/a	return MediaDescr.SoundDescription.decode(data)
28	n/a	try:
29	n/a	from imgformat import macrgb
30	n/a	except ImportError:
31	n/a	macrgb = "Macintosh RGB format"
32	n/a	import os
33	n/a	# import audio.format
34	n/a
35	n/a	class VideoFormat:
36	n/a	def __init__(self, name, descr, width, height, format):
37	n/a	self.__name = name
38	n/a	self.__descr = descr
39	n/a	self.__width = width
40	n/a	self.__height = height
41	n/a	self.__format = format
42	n/a
43	n/a	def getname(self):
44	n/a	return self.__name
45	n/a
46	n/a	def getdescr(self):
47	n/a	return self.__descr
48	n/a
49	n/a	def getsize(self):
50	n/a	return self.__width, self.__height
51	n/a
52	n/a	def getformat(self):
53	n/a	return self.__format
54	n/a
55	n/a	class _Reader:
56	n/a	def __init__(self, path):
57	n/a	fd = Qt.OpenMovieFile(path, 0)
58	n/a	self.movie, d1, d2 = Qt.NewMovieFromFile(fd, 0, 0)
59	n/a	self.movietimescale = self.movie.GetMovieTimeScale()
60	n/a	try:
61	n/a	self.audiotrack = self.movie.GetMovieIndTrackType(1,
62	n/a	QuickTime.AudioMediaCharacteristic, QuickTime.movieTrackCharacteristic)
63	n/a	self.audiomedia = self.audiotrack.GetTrackMedia()
64	n/a	except Qt.Error:
65	n/a	self.audiotrack = self.audiomedia = None
66	n/a	self.audiodescr = {}
67	n/a	else:
68	n/a	handle = Res.Handle('')
69	n/a	n = self.audiomedia.GetMediaSampleDescriptionCount()
70	n/a	self.audiomedia.GetMediaSampleDescription(1, handle)
71	n/a	self.audiodescr = _audiodescr(handle.data)
72	n/a	self.audiotimescale = self.audiomedia.GetMediaTimeScale()
73	n/a	del handle
74	n/a
75	n/a	try:
76	n/a	self.videotrack = self.movie.GetMovieIndTrackType(1,
77	n/a	QuickTime.VisualMediaCharacteristic, QuickTime.movieTrackCharacteristic)
78	n/a	self.videomedia = self.videotrack.GetTrackMedia()
79	n/a	except Qt.Error:
80	n/a	self.videotrack = self.videomedia = self.videotimescale = None
81	n/a	if self.videotrack:
82	n/a	self.videotimescale = self.videomedia.GetMediaTimeScale()
83	n/a	x0, y0, x1, y1 = self.movie.GetMovieBox()
84	n/a	self.videodescr = {'width':(x1-x0), 'height':(y1-y0)}
85	n/a	self._initgworld()
86	n/a	self.videocurtime = None
87	n/a	self.audiocurtime = None
88	n/a
89	n/a
90	n/a	def __del__(self):
91	n/a	self.audiomedia = None
92	n/a	self.audiotrack = None
93	n/a	self.videomedia = None
94	n/a	self.videotrack = None
95	n/a	self.movie = None
96	n/a
97	n/a	def _initgworld(self):
98	n/a	old_port, old_dev = Qdoffs.GetGWorld()
99	n/a	try:
100	n/a	movie_w = self.videodescr['width']
101	n/a	movie_h = self.videodescr['height']
102	n/a	movie_rect = (0, 0, movie_w, movie_h)
103	n/a	self.gworld = Qdoffs.NewGWorld(32, movie_rect, None, None, QDOffscreen.keepLocal)
104	n/a	self.pixmap = self.gworld.GetGWorldPixMap()
105	n/a	Qdoffs.LockPixels(self.pixmap)
106	n/a	Qdoffs.SetGWorld(self.gworld.as_GrafPtr(), None)
107	n/a	Qd.EraseRect(movie_rect)
108	n/a	self.movie.SetMovieGWorld(self.gworld.as_GrafPtr(), None)
109	n/a	self.movie.SetMovieBox(movie_rect)
110	n/a	self.movie.SetMovieActive(1)
111	n/a	self.movie.MoviesTask(0)
112	n/a	self.movie.SetMoviePlayHints(QuickTime.hintsHighQuality, QuickTime.hintsHighQuality)
113	n/a	# XXXX framerate
114	n/a	finally:
115	n/a	Qdoffs.SetGWorld(old_port, old_dev)
116	n/a
117	n/a	def _gettrackduration_ms(self, track):
118	n/a	tracktime = track.GetTrackDuration()
119	n/a	return self._movietime_to_ms(tracktime)
120	n/a
121	n/a	def _movietime_to_ms(self, time):
122	n/a	value, d1, d2 = Qt.ConvertTimeScale((time, self.movietimescale, None), 1000)
123	n/a	return value
124	n/a
125	n/a	def _videotime_to_ms(self, time):
126	n/a	value, d1, d2 = Qt.ConvertTimeScale((time, self.videotimescale, None), 1000)
127	n/a	return value
128	n/a
129	n/a	def _audiotime_to_ms(self, time):
130	n/a	value, d1, d2 = Qt.ConvertTimeScale((time, self.audiotimescale, None), 1000)
131	n/a	return value
132	n/a
133	n/a	def _videotime_to_movietime(self, time):
134	n/a	value, d1, d2 = Qt.ConvertTimeScale((time, self.videotimescale, None),
135	n/a	self.movietimescale)
136	n/a	return value
137	n/a
138	n/a	def HasAudio(self):
139	n/a	return not self.audiotrack is None
140	n/a
141	n/a	def HasVideo(self):
142	n/a	return not self.videotrack is None
143	n/a
144	n/a	def GetAudioDuration(self):
145	n/a	if not self.audiotrack:
146	n/a	return 0
147	n/a	return self._gettrackduration_ms(self.audiotrack)
148	n/a
149	n/a	def GetVideoDuration(self):
150	n/a	if not self.videotrack:
151	n/a	return 0
152	n/a	return self._gettrackduration_ms(self.videotrack)
153	n/a
154	n/a	def GetAudioFormat(self):
155	n/a	if not self.audiodescr:
156	n/a	return None, None, None, None, None
157	n/a	bps = self.audiodescr['sampleSize']
158	n/a	nch = self.audiodescr['numChannels']
159	n/a	if nch == 1:
160	n/a	channels = ['mono']
161	n/a	elif nch == 2:
162	n/a	channels = ['left', 'right']
163	n/a	else:
164	n/a	channels = map(lambda x: str(x+1), range(nch))
165	n/a	if bps % 8:
166	n/a	# Funny bits-per sample. We pretend not to understand
167	n/a	blocksize = 0
168	n/a	fpb = 0
169	n/a	else:
170	n/a	# QuickTime is easy (for as far as we support it): samples are always a whole
171	n/a	# number of bytes, so frames are nchannels*samplesize, and there's one frame per block.
172	n/a	blocksize = (bps/8)*nch
173	n/a	fpb = 1
174	n/a	if self.audiodescr['dataFormat'] == 'raw ':
175	n/a	encoding = 'linear-excess'
176	n/a	elif self.audiodescr['dataFormat'] == 'twos':
177	n/a	encoding = 'linear-signed'
178	n/a	else:
179	n/a	encoding = 'quicktime-coding-%s'%self.audiodescr['dataFormat']
180	n/a	## return audio.format.AudioFormatLinear('quicktime_audio', 'QuickTime Audio Format',
181	n/a	## channels, encoding, blocksize=blocksize, fpb=fpb, bps=bps)
182	n/a	return channels, encoding, blocksize, fpb, bps
183	n/a
184	n/a	def GetAudioFrameRate(self):
185	n/a	if not self.audiodescr:
186	n/a	return None
187	n/a	return int(self.audiodescr['sampleRate'])
188	n/a
189	n/a	def GetVideoFormat(self):
190	n/a	width = self.videodescr['width']
191	n/a	height = self.videodescr['height']
192	n/a	return VideoFormat('dummy_format', 'Dummy Video Format', width, height, macrgb)
193	n/a
194	n/a	def GetVideoFrameRate(self):
195	n/a	tv = self.videocurtime
196	n/a	if tv is None:
197	n/a	tv = 0
198	n/a	flags = QuickTime.nextTimeStep\|QuickTime.nextTimeEdgeOK
199	n/a	tv, dur = self.videomedia.GetMediaNextInterestingTime(flags, tv, 1.0)
200	n/a	dur = self._videotime_to_ms(dur)
201	n/a	return int((1000.0/dur)+0.5)
202	n/a
203	n/a	def ReadAudio(self, nframes, time=None):
204	n/a	if not time is None:
205	n/a	self.audiocurtime = time
206	n/a	flags = QuickTime.nextTimeStep\|QuickTime.nextTimeEdgeOK
207	n/a	if self.audiocurtime is None:
208	n/a	self.audiocurtime = 0
209	n/a	tv = self.audiomedia.GetMediaNextInterestingTimeOnly(flags, self.audiocurtime, 1.0)
210	n/a	if tv < 0 or (self.audiocurtime and tv < self.audiocurtime):
211	n/a	return self._audiotime_to_ms(self.audiocurtime), None
212	n/a	h = Res.Handle('')
213	n/a	desc_h = Res.Handle('')
214	n/a	size, actualtime, sampleduration, desc_index, actualcount, flags = \
215	n/a	self.audiomedia.GetMediaSample(h, 0, tv, desc_h, nframes)
216	n/a	self.audiocurtime = actualtime + actualcount*sampleduration
217	n/a	return self._audiotime_to_ms(actualtime), h.data
218	n/a
219	n/a	def ReadVideo(self, time=None):
220	n/a	if not time is None:
221	n/a	self.videocurtime = time
222	n/a	flags = QuickTime.nextTimeStep
223	n/a	if self.videocurtime is None:
224	n/a	flags = flags \| QuickTime.nextTimeEdgeOK
225	n/a	self.videocurtime = 0
226	n/a	tv = self.videomedia.GetMediaNextInterestingTimeOnly(flags, self.videocurtime, 1.0)
227	n/a	if tv < 0 or (self.videocurtime and tv <= self.videocurtime):
228	n/a	return self._videotime_to_ms(self.videocurtime), None
229	n/a	self.videocurtime = tv
230	n/a	moviecurtime = self._videotime_to_movietime(self.videocurtime)
231	n/a	self.movie.SetMovieTimeValue(moviecurtime)
232	n/a	self.movie.MoviesTask(0)
233	n/a	return self._videotime_to_ms(self.videocurtime), self._getpixmapcontent()
234	n/a
235	n/a	def _getpixmapcontent(self):
236	n/a	"""Shuffle the offscreen PixMap data, because it may have funny stride values"""
237	n/a	rowbytes = Qdoffs.GetPixRowBytes(self.pixmap)
238	n/a	width = self.videodescr['width']
239	n/a	height = self.videodescr['height']
240	n/a	start = 0
241	n/a	rv = []
242	n/a	for i in range(height):
243	n/a	nextline = Qdoffs.GetPixMapBytes(self.pixmap, start, width*4)
244	n/a	start = start + rowbytes
245	n/a	rv.append(nextline)
246	n/a	return ''.join(rv)
247	n/a
248	n/a	def reader(url):
249	n/a	try:
250	n/a	rdr = _Reader(url)
251	n/a	except IOError:
252	n/a	return None
253	n/a	return rdr
254	n/a
255	n/a	def _test():
256	n/a	import EasyDialogs
257	n/a	try:
258	n/a	from PIL import Image
259	n/a	except ImportError:
260	n/a	Image = None
261	n/a	import MacOS
262	n/a	Qt.EnterMovies()
263	n/a	path = EasyDialogs.AskFileForOpen(message='Video to convert')
264	n/a	if not path: sys.exit(0)
265	n/a	rdr = reader(path)
266	n/a	if not rdr:
267	n/a	sys.exit(1)
268	n/a	dstdir = EasyDialogs.AskFileForSave(message='Name for output folder')
269	n/a	if not dstdir: sys.exit(0)
270	n/a	num = 0
271	n/a	os.mkdir(dstdir)
272	n/a	videofmt = rdr.GetVideoFormat()
273	n/a	imgfmt = videofmt.getformat()
274	n/a	imgw, imgh = videofmt.getsize()
275	n/a	timestamp, data = rdr.ReadVideo()
276	n/a	while data:
277	n/a	fname = 'frame%04.4d.jpg'%num
278	n/a	num = num+1
279	n/a	pname = os.path.join(dstdir, fname)
280	n/a	if not Image: print 'Not',
281	n/a	print 'Writing %s, size %dx%d, %d bytes'%(fname, imgw, imgh, len(data))
282	n/a	if Image:
283	n/a	img = Image.fromstring("RGBA", (imgw, imgh), data)
284	n/a	img.save(pname, 'JPEG')
285	n/a	timestamp, data = rdr.ReadVideo()
286	n/a	MacOS.SetCreatorAndType(pname, 'ogle', 'JPEG')
287	n/a	if num > 20:
288	n/a	print 'stopping at 20 frames so your disk does not fill up:-)'
289	n/a	break
290	n/a	print 'Total frames:', num
291	n/a
292	n/a	if __name__ == '__main__':
293	n/a	_test()
294	n/a	sys.exit(1)