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 --print to main.py

Browse Source
This commit is contained in:
Steven Robertson 2012-07-05 00:14:44 -07:00
parent bb852ff255
commit 3619d96926
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
main.py
View File

@ -32,11 +32,11 @@ def save(out):
print out.idx, out.gpu_time print out.idx, out.gpu_time
def main(args, prof): def main(args, prof):
import pycuda.autoinit
gdb = db.connect(args.genomedb) gdb = db.connect(args.genomedb)
gnm, basename = gdb.get_anim(args.flame, args.half) gnm, basename = gdb.get_anim(args.flame, args.half)
if getattr(args, 'print'):
print convert.to_json(gnm)
return
gprof = profile.wrap(prof, gnm) gprof = profile.wrap(prof, gnm)
if args.name is not None: if args.name is not None:
@ -55,6 +55,7 @@ def main(args, prof):
frames = (f for f in frames frames = (f for f in frames
if not os.path.isfile(f[0]) or m > os.path.getmtime(f[0])) if not os.path.isfile(f[0]) or m > os.path.getmtime(f[0]))
import pycuda.autoinit
rmgr = render.RenderManager() rmgr = render.RenderManager()
gen = rmgr.render(gnm, gprof, frames) gen = rmgr.render(gnm, gprof, frames)
@ -135,6 +136,8 @@ if __name__ == "__main__":
help="Use basename as subdirectory of out dir, instead of prefix") help="Use basename as subdirectory of out dir, instead of prefix")
parser.add_argument('--half', action='store_true', parser.add_argument('--half', action='store_true',
help='Use half-loops when converting nodes to animations') help='Use half-loops when converting nodes to animations')
parser.add_argument('--print', action='store_true',
help="Print the blended animation and exit.")
profile.add_args(parser) profile.add_args(parser)
args = parser.parse_args() args = parser.parse_args()