mirror-github-bspeice/cuburn
1
0
Fork 0
mirror of https://github.com/stevenrobertson/cuburn.git synced 2025-02-05 11:40:04 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add raw output for direct x264 encoding

Browse Source
This commit is contained in:
Steven Robertson 2011-10-15 14:31:56 -04:00
parent 63483480d0
commit 92708b32ba
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
main.py
View File

@ -28,6 +28,8 @@ from cuburn.code.mwc import MWCTest
np.set_printoptions(precision=5, edgeitems=20) np.set_printoptions(precision=5, edgeitems=20)
real_stdout = sys.stdout
def fmt_time(time): def fmt_time(time):
# Format time in a lexically-ordered way that doesn't interfere with the # Format time in a lexically-ordered way that doesn't interfere with the
# typical case of ascending natural numbers # typical case of ascending natural numbers
@ -39,8 +41,12 @@ def fmt_time(time):
return dcml return dcml
def save(args, time, raw): def save(args, time, raw):
name = os.path.join(args.dir, '%s_%s' % (args.name, fmt_time(time)))
noalpha = raw[:,:,:3] noalpha = raw[:,:,:3]
if args.raw:
real_stdout.write(buffer(np.uint8(noalpha * 255.0)))
return
name = os.path.join(args.dir, '%s_%s' % (args.name, fmt_time(time)))
img = scipy.misc.toimage(noalpha, cmin=0, cmax=1) img = scipy.misc.toimage(noalpha, cmin=0, cmax=1)
img.save(name+'.png') img.save(name+'.png')
@ -57,6 +63,9 @@ def main(args):
MWCTest.test_mwc() MWCTest.test_mwc()
return return
if args.raw:
sys.stdout = sys.stderr
genome_ptr, ngenomes = pyflam3.Genome.from_string(args.flame.read()) genome_ptr, ngenomes = pyflam3.Genome.from_string(args.flame.read())
genomes = cast(genome_ptr, POINTER(pyflam3.Genome*ngenomes)).contents genomes = cast(genome_ptr, POINTER(pyflam3.Genome*ngenomes)).contents
@ -181,6 +190,8 @@ if __name__ == "__main__":
help="Prefix to use when saving files (default is basename of input)") help="Prefix to use when saving files (default is basename of input)")
parser.add_argument('-o', metavar='DIR', type=str, dest='dir', parser.add_argument('-o', metavar='DIR', type=str, dest='dir',
help="Output directory", default='.') help="Output directory", default='.')
parser.add_argument('--raw', action='store_true', dest='raw',
help="Do not write files; instead, send raw RGBA data to stdout.")
time = parser.add_argument_group('Sequence options', description=""" time = parser.add_argument_group('Sequence options', description="""
Control which frames are rendered from a genome sequence. If '-k' is Control which frames are rendered from a genome sequence. If '-k' is