"--encoder", type=str, choices=("mencoder", "ffmpeg"), default="ffmpeg", help="Video encoding program to use") cmd_args = parser.parse_args() movieid = "{}-t{:04}-th{}-ph{}-n{}".format(cmd_args.mode, cmd_args.time, cmd_args.orient[0], cmd_args.orient[1], cmd_args.frames) movie = Movie( runid=cmd_args.runid, movieid=movieid, time=cmd_args.time, nframes=cmd_args.frames, verbose=1, force=0, ) movie.imageprefix = "rgb-{}-{}-{}".format(cmd_args.emshort, movie.runid, movie.movieid) movie.emtypes = cmd_args.emtypes movie.brightmax = cmd_args.brightscale movie.bandscales = cmd_args.bandscales movie.vcodec = cmd_args.vcodec movie.containerformat = cmd_args.containerformat movie.encoder = cmd_args.encoder movie.boxsize = 4.0 movie.camera.set_angles(*cmd_args.orient) def bmaxNHO(i): """ Smoothly varying brightness with time so that the movie looks good """ return cmd_args.brightscale/(1.0+(float(i)/10)**2)
ph0 = float(sys.argv[4]) except (IndexError, ValueError): th0, ph0 = 0.0, 0.0 # Optional brightness arguments # set defaults brightmax0 = 2.e8 bandscales = Movie.bandscales try: # 5th arg is scalar: use to set brightmax0 brightmax0 = float(sys.argv[5]) except IndexError: pass except ValueError: try: # 5th arg is triplet: use to set bandscales bandscales = [float(x) for x in sys.argv[5].split()] except ValueError: # otherwise, just use defaults pass Movie.bandscales = bandscales movieid="tumble-%4.4i" % itime movie = Movie(runid=runid, movieid=movieid, time=itime, nframes=72, verbose=1, force=0) movie.imageprefix = "rgb-NHO-%s-%s" % (runid, movieid) movie.brightmaxfunc = brightmax movie.boxsize = 4.0 movie.camera.set_steps(5.0, 5.0) movie.makemovie()
# Optional brightness arguments # set defaults brightmax0 = 1.e7 bandscales = [0.2, 1.0e-6, 1.0e-9] try: # 7th arg is scalar: use to set brightmax0 brightmax0 = float(sys.argv[7]) except IndexError: pass except ValueError: try: # 7th arg is triplet: use to set bandscales bandscales = [float(x) for x in sys.argv[7].split()] except ValueError: # otherwise, just use defaults pass Movie.emtypes = ["neut00", "PAH000", "FF06cm"] Movie.bandscales = bandscales movieid = "evo%+3.3i%+3.3i" % (th0, ph0) movie = Movie(runid=runid, movieid=movieid, time=it1, dtime=idelta, frame=it1/idelta, nframes=1+(it2-it1)/idelta, verbose=1, force=0) movie.imageprefix = "rgb-CPF-%s-%s" % (runid, movieid) movie.brightmaxfunc = brightmax movie.camera.set_steps(0.0, 0.0) movie.camera.set_angles(th0, ph0) movie.makemovie()
except IndexError, ValueError: th0, ph0 = 0.0, 0.0 # Optional brightness arguments try: brightmax0 = float(sys.argv[7]) except IndexError, ValueError: print "No brightmax given - using default value" brightmax0 = 1.0e6 Movie.emtypes = ["S26731", "N26584", "Halpha"] Movie.bandscales = [0.15, 0.3, 0.8] movieid = "evo%+3.3i%+3.3i" % (th0, ph0) movie = Movie( runid=runid, movieid=movieid, time=it1, dtime=idelta, frame=it1 / idelta, nframes=1 + (it2 - it1) / idelta, verbose=1, force=0, ) movie.imageprefix = "rgb-SNH-%s-%s" % (runid, movieid) movie.brightmaxfunc = brightmax movie.camera.set_steps(0.0, 0.0) movie.camera.set_angles(th0, ph0) movie.boxsize = 4.0 movie.makemovie()