# Video file reader, using QuickTime # # This module was quickly ripped out of another software package, so there is a good # chance that it does not work as-is and it needs some hacking. # # Jack Jansen, August 2000 # from warnings import warnpy3k warnpy3k("In 3.x, the videoreader module is removed.", stacklevel=2) import sys from Carbon import Qt from Carbon import QuickTime from Carbon import Qd from Carbon import Qdoffs from Carbon import QDOffscreen from Carbon import Res try: import MediaDescr except ImportError: def _audiodescr(data): return None else: def _audiodescr(data): return MediaDescr.SoundDescription.decode(data) try: from imgformat import macrgb except ImportError: macrgb = "Macintosh RGB format" import os # import audio.format class VideoFormat: def __init__(self, name, descr, width, height, format): self.__name = name self.__descr = descr self.__width = width self.__height = height self.__format = format def getname(self): return self.__name def getdescr(self): return self.__descr def getsize(self): return self.__width, self.__height def getformat(self): return self.__format class _Reader: def __init__(self, path): fd = Qt.OpenMovieFile(path, 0) self.movie, d1, d2 = Qt.NewMovieFromFile(fd, 0, 0) self.movietimescale = self.movie.GetMovieTimeScale() try: self.audiotrack = self.movie.GetMovieIndTrackType(1, QuickTime.AudioMediaCharacteristic, QuickTime.movieTrackCharacteristic) self.audiomedia = self.audiotrack.GetTrackMedia() except Qt.Error: self.audiotrack = self.audiomedia = None self.audiodescr = {} else: handle = Res.Handle('') n = self.audiomedia.GetMediaSampleDescriptionCount() self.audiomedia.GetMediaSampleDescription(1, handle) self.audiodescr = _audiodescr(handle.data) self.audiotimescale = self.audiomedia.GetMediaTimeScale() del handle try: self.videotrack = self.movie.GetMovieIndTrackType(1, QuickTime.VisualMediaCharacteristic, QuickTime.movieTrackCharacteristic) self.videomedia = self.videotrack.GetTrackMedia() except Qt.Error: self.videotrack = self.videomedia = self.videotimescale = None if self.videotrack: self.videotimescale = self.videomedia.GetMediaTimeScale() x0, y0, x1, y1 = self.movie.GetMovieBox() self.videodescr = {'width':(x1-x0), 'height':(y1-y0)} self._initgworld() self.videocurtime = None self.audiocurtime = None def __del__(self): self.audiomedia = None self.audiotrack = None self.videomedia = None self.videotrack = None self.movie = None def _initgworld(self): old_port, old_dev = Qdoffs.GetGWorld() try: movie_w = self.videodescr['width'] movie_h = self.videodescr['height'] movie_rect = (0, 0, movie_w, movie_h) self.gworld = Qdoffs.NewGWorld(32, movie_rect, None, None, QDOffscreen.keepLocal) self.pixmap = self.gworld.GetGWorldPixMap() Qdoffs.LockPixels(self.pixmap) Qdoffs.SetGWorld(self.gworld.as_GrafPtr(), None) Qd.EraseRect(movie_rect) self.movie.SetMovieGWorld(self.gworld.as_GrafPtr(), None) self.movie.SetMovieBox(movie_rect) self.movie.SetMovieActive(1) self.movie.MoviesTask(0) self.movie.SetMoviePlayHints(QuickTime.hintsHighQuality, QuickTime.hintsHighQuality) # XXXX framerate finally: Qdoffs.SetGWorld(old_port, old_dev) def _gettrackduration_ms(self, track): tracktime = track.GetTrackDuration() return self._movietime_to_ms(tracktime) def _movietime_to_ms(self, time): value, d1, d2 = Qt.ConvertTimeScale((time, self.movietimescale, None), 1000) return value def _videotime_to_ms(self, time): value, d1, d2 = Qt.ConvertTimeScale((time, self.videotimescale, None), 1000) return value def _audiotime_to_ms(self, time): value, d1, d2 = Qt.ConvertTimeScale((time, self.audiotimescale, None), 1000) return value def _videotime_to_movietime(self, time): value, d1, d2 = Qt.ConvertTimeScale((time, self.videotimescale, None), self.movietimescale) return value def HasAudio(self): return not self.audiotrack is None def HasVideo(self): return not self.videotrack is None def GetAudioDuration(self): if not self.audiotrack: return 0 return self._gettrackduration_ms(self.audiotrack) def GetVideoDuration(self): if not self.videotrack: return 0 return self._gettrackduration_ms(self.videotrack) def GetAudioFormat(self): if not self.audiodescr: return None, None, None, None, None bps = self.audiodescr['sampleSize'] nch = self.audiodescr['numChannels'] if nch == 1: channels = ['mono'] elif nch == 2: channels = ['left', 'right'] else: channels = map(lambda x: str(x+1), range(nch)) if bps % 8: # Funny bits-per sample. We pretend not to understand blocksize = 0 fpb = 0 else: # QuickTime is easy (for as far as we support it): samples are always a whole # number of bytes, so frames are nchannels*samplesize, and there's one frame per block. blocksize = (bps/8)*nch fpb = 1 if self.audiodescr['dataFormat'] == 'raw ': encoding = 'linear-excess' elif self.audiodescr['dataFormat'] == 'twos': encoding = 'linear-signed' else: encoding = 'quicktime-coding-%s'%self.audiodescr['dataFormat'] ## return audio.format.AudioFormatLinear('quicktime_audio', 'QuickTime Audio Format', ## channels, encoding, blocksize=blocksize, fpb=fpb, bps=bps) return channels, encoding, blocksize, fpb, bps def GetAudioFrameRate(self): if not self.audiodescr: return None return int(self.audiodescr['sampleRate']) def GetVideoFormat(self): width = self.videodescr['width'] height = self.videodescr['height'] return VideoFormat('dummy_format', 'Dummy Video Format', width, height, macrgb) def GetVideoFrameRate(self): tv = self.videocurtime if tv is None: tv = 0 flags = QuickTime.nextTimeStep|QuickTime.nextTimeEdgeOK tv, dur = self.videomedia.GetMediaNextInterestingTime(flags, tv, 1.0) dur = self._videotime_to_ms(dur) return int((1000.0/dur)+0.5) def ReadAudio(self, nframes, time=None): if not time is None: self.audiocurtime = time flags = QuickTime.nextTimeStep|QuickTime.nextTimeEdgeOK if self.audiocurtime is None: self.audiocurtime = 0 tv = self.audiomedia.GetMediaNextInterestingTimeOnly(flags, self.audiocurtime, 1.0) if tv < 0 or (self.audiocurtime and tv < self.audiocurtime): return self._audiotime_to_ms(self.audiocurtime), None h = Res.Handle('') desc_h = Res.Handle('') size, actualtime, sampleduration, desc_index, actualcount, flags = \ self.audiomedia.GetMediaSample(h, 0, tv, desc_h, nframes) self.audiocurtime = actualtime + actualcount*sampleduration return self._audiotime_to_ms(actualtime), h.data def ReadVideo(self, time=None): if not time is None: self.videocurtime = time flags = QuickTime.nextTimeStep if self.videocurtime is None: flags = flags | QuickTime.nextTimeEdgeOK self.videocurtime = 0 tv = self.videomedia.GetMediaNextInterestingTimeOnly(flags, self.videocurtime, 1.0) if tv < 0 or (self.videocurtime and tv <= self.videocurtime): return self._videotime_to_ms(self.videocurtime), None self.videocurtime = tv moviecurtime = self._videotime_to_movietime(self.videocurtime) self.movie.SetMovieTimeValue(moviecurtime) self.movie.MoviesTask(0) return self._videotime_to_ms(self.videocurtime), self._getpixmapcontent() def _getpixmapcontent(self): """Shuffle the offscreen PixMap data, because it may have funny stride values""" rowbytes = Qdoffs.GetPixRowBytes(self.pixmap) width = self.videodescr['width'] height = self.videodescr['height'] start = 0 rv = [] for i in range(height): nextline = Qdoffs.GetPixMapBytes(self.pixmap, start, width*4) start = start + rowbytes rv.append(nextline) return ''.join(rv) def reader(url): try: rdr = _Reader(url) except IOError: return None return rdr def _test(): import EasyDialogs try: from PIL import Image except ImportError: Image = None import MacOS Qt.EnterMovies() path = EasyDialogs.AskFileForOpen(message='Video to convert') if not path: sys.exit(0) rdr = reader(path) if not rdr: sys.exit(1) dstdir = EasyDialogs.AskFileForSave(message='Name for output folder') if not dstdir: sys.exit(0) num = 0 os.mkdir(dstdir) videofmt = rdr.GetVideoFormat() imgfmt = videofmt.getformat() imgw, imgh = videofmt.getsize() timestamp, data = rdr.ReadVideo() while data: fname = 'frame%04.4d.jpg'%num num = num+1 pname = os.path.join(dstdir, fname) if not Image: print 'Not', print 'Writing %s, size %dx%d, %d bytes'%(fname, imgw, imgh, len(data)) if Image: img = Image.fromstring("RGBA", (imgw, imgh), data) img.save(pname, 'JPEG') timestamp, data = rdr.ReadVideo() MacOS.SetCreatorAndType(pname, 'ogle', 'JPEG') if num > 20: print 'stopping at 20 frames so your disk does not fill up:-)' break print 'Total frames:', num if __name__ == '__main__': _test() sys.exit(1)