tools/vizTools/make_movie.py

#!/usr/bin/env python

"""
Make an mpeg movie from the pnm files in the current directory
"""

import os, sys, re
import getopt

(opts, args) = getopt.getopt(sys.argv[1:], 
    "s:o:f:h", 
    ["framestem=", "output=", "format=", "help"],
    )

def usage():
    print "Usage:"
    print "  make_movie -s <framestem> -o <output filename> -f <format>\n"
    print "  Arguments:"
    print "    -s/--framestem: The frame filename stem (the stuff before .pnm) (required)"
    print "    -o/--output: Output mpeg filename (optional)"
    print "    -f/--format: Input frame image format (optional)"

mpegName = None
fnameStem = None
format = "pnm"  # if format not specified assume pnm

for option, arg in opts:
    if option in ('-s', '--stem'):
    fnameStem = arg
    elif option in ('-o', '--output'):
    mpegName = arg
    elif option in ('-f', '--format'):
    format = arg
    elif option in ('-h', '--help'):
    usage()
    sys.exit(0)

if fnameStem is None:
    print "You must supply the stem of the frame filenames\n"
    usage()
    sys.exit(0)
if mpegName is None:
    mpegName = fnameStem + ".mpg"

# determine the number of files to convert
dirList = os.listdir('./')
r = re.compile( "%s\\d+\\.%s"%(fnameStem,format) )
count = 0  # counter for the number of files found
fnames = []
for fname in dirList:
    if r.match(fname):
    fnames.append(fname)
    count += 1

# do a conversion if necessary
if format != "pnm":
    print "Converting frames to pnm"
    r = re.compile("%s\\d+\\." % (fnameStem))
    for fname in fnames:
    # strip off the bit after the last dot
    front = r.findall(fname)
    front = front[0]
    # make the conversion string to pass to 'convert'
    convString = "convert %s%s %spnm" % (front, format, front)
    retVal = os.system(convString)
    if retVal != 0:
        raise SystemError, "Conversion of %s%s failed" % (front, format)

# use ppmtompeg to convert the pnms to a movie
print "Making movie"

# now automatically generate the ppmtompeg params file
print "Generating params file..."

# old pattern
# PATTERN IBBBPBBBPBBBPBBBPBBB

paramsFileString = """REFERENCE_FRAME DECODED
FRAME_RATE 24
OUTPUT %s
PATTERN IBBPBBPBB
FORCE_ENCODE_LAST_FRAME
GOP_SIZE 20
BSEARCH_ALG CROSS2
PSEARCH_ALG TWOLEVEL
IQSCALE 10
PQSCALE 11
BQSCALE 16
RANGE 8
SLICES_PER_FRAME 1
BASE_FILE_FORMAT PNM
INPUT_DIR .
INPUT_CONVERT *
INPUT
""" % (mpegName)

# need to determine the first number and last number of the list of files
# sort the files first just in case
fnames.sort()
firstFile = fnames[0]
lastFile = fnames[-1]

r = re.compile("([a-zA-Z])(\\d+)(\\.\\w+)")
firstNum = r.findall(firstFile)
firstNum = firstNum[0][1]
lastNum = r.findall(lastFile)
lastNum = lastNum[0][1]

# finish off the params file string
paramsFileString += "%s*.pnm [%s-%s]\n" % (fnameStem, firstNum, lastNum)

paramsFileString += """END_INPUT
PIXEL HALF
ASPECT_RATIO 1
"""

# write the string to file
fp = open( "%s.params" % (fnameStem,), "w" )
fp.write(paramsFileString + '\n')
fp.close()
print "Done params file generation"

# now do the conversion to mpeg
print "Performing conversion to mpeg"
convertString = "ppmtompeg %s.params" % (fnameStem,)
result = os.system(convertString)
if result != 0:
    print "An error occurred in mpeg conversion"

# now clean up a bit
#for i in range(count):
#    rmFname = "%s_%04d.ppm" % (fnameStem,i)
#    os.unlink(rmFname)

print "\nDone!"
 
1	jgs	123	#!/usr/bin/env python
2
3			"""
4			Make an mpeg movie from the pnm files in the current directory
5			"""
6
7			import os, sys, re
8			import getopt
9
10			(opts, args) = getopt.getopt(sys.argv[1:],
11			"s:o:f:h",
12			["framestem=", "output=", "format=", "help"],
13			)
14
15			def usage():
16			print "Usage:"
17			print " make_movie -s <framestem> -o <output filename> -f <format>\n"
18			print " Arguments:"
19			print " -s/--framestem: The frame filename stem (the stuff before .pnm) (required)"
20			print " -o/--output: Output mpeg filename (optional)"
21			print " -f/--format: Input frame image format (optional)"
22
23			mpegName = None
24			fnameStem = None
25			format = "pnm" # if format not specified assume pnm
26
27			for option, arg in opts:
28			if option in ('-s', '--stem'):
29			fnameStem = arg
30			elif option in ('-o', '--output'):
31			mpegName = arg
32			elif option in ('-f', '--format'):
33			format = arg
34			elif option in ('-h', '--help'):
35			usage()
36			sys.exit(0)
37
38			if fnameStem is None:
39			print "You must supply the stem of the frame filenames\n"
40			usage()
41			sys.exit(0)
42			if mpegName is None:
43			mpegName = fnameStem + ".mpg"
44
45			# determine the number of files to convert
46			dirList = os.listdir('./')
47			r = re.compile( "%s\\d+\\.%s"%(fnameStem,format) )
48			count = 0 # counter for the number of files found
49			fnames = []
50			for fname in dirList:
51			if r.match(fname):
52			fnames.append(fname)
53			count += 1
54
55			# do a conversion if necessary
56			if format != "pnm":
57			print "Converting frames to pnm"
58			r = re.compile("%s\\d+\\." % (fnameStem))
59			for fname in fnames:
60			# strip off the bit after the last dot
61			front = r.findall(fname)
62			front = front[0]
63			# make the conversion string to pass to 'convert'
64			convString = "convert %s%s %spnm" % (front, format, front)
65			retVal = os.system(convString)
66			if retVal != 0:
67			raise SystemError, "Conversion of %s%s failed" % (front, format)
68
69			# use ppmtompeg to convert the pnms to a movie
70			print "Making movie"
71
72			# now automatically generate the ppmtompeg params file
73			print "Generating params file..."
74
75			# old pattern
76			# PATTERN IBBBPBBBPBBBPBBBPBBB
77
78			paramsFileString = """REFERENCE_FRAME DECODED
79			FRAME_RATE 24
80			OUTPUT %s
81			PATTERN IBBPBBPBB
82			FORCE_ENCODE_LAST_FRAME
83			GOP_SIZE 20
84			BSEARCH_ALG CROSS2
85			PSEARCH_ALG TWOLEVEL
86			IQSCALE 10
87			PQSCALE 11
88			BQSCALE 16
89			RANGE 8
90			SLICES_PER_FRAME 1
91			BASE_FILE_FORMAT PNM
92			INPUT_DIR .
93			INPUT_CONVERT *
94			INPUT
95			""" % (mpegName)
96
97			# need to determine the first number and last number of the list of files
98			# sort the files first just in case
99			fnames.sort()
100			firstFile = fnames[0]
101			lastFile = fnames[-1]
102
103			r = re.compile("([a-zA-Z])(\\d+)(\\.\\w+)")
104			firstNum = r.findall(firstFile)
105			firstNum = firstNum[0][1]
106			lastNum = r.findall(lastFile)
107			lastNum = lastNum[0][1]
108
109			# finish off the params file string
110			paramsFileString += "%s*.pnm [%s-%s]\n" % (fnameStem, firstNum, lastNum)
111
112			paramsFileString += """END_INPUT
113			PIXEL HALF
114			ASPECT_RATIO 1
115			"""
116
117			# write the string to file
118			fp = open( "%s.params" % (fnameStem,), "w" )
119			fp.write(paramsFileString + '\n')
120			fp.close()
121			print "Done params file generation"
122
123			# now do the conversion to mpeg
124			print "Performing conversion to mpeg"
125			convertString = "ppmtompeg %s.params" % (fnameStem,)
126			result = os.system(convertString)
127			if result != 0:
128			print "An error occurred in mpeg conversion"
129
130			# now clean up a bit
131			#for i in range(count):
132			# rmFname = "%s_%04d.ppm" % (fnameStem,i)
133			# os.unlink(rmFname)
134
135			print "\nDone!"
136
Name	Value
svn:eol-style	native
svn:executable	*
svn:keywords	Author Date Id Revision