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)
def bmaxCPF(i):
"""
Smoothly varying brightness with time so that the movie looks good
This version keeps the R band constant, while G and B bands get brighter with time
"""
return [cmd_args.brightscale, cmd_args.brightscale/(1.0+(float(i)/70)**2), cmd_args.brightscale/(1.0+(float(i)/50)**2)]
if cmd_args.mode == "tumble":
movie.camera.set_steps(360.0/movie.nframes, 360.0/movie.nframes)
elif cmd_args.mode == "evo":
movie.camera.set_steps(0.0, 0.0)
movie.dtime = 1
if cmd_args.emshort == "NHO":
movie.brightmaxfunc = bmaxNHO
elif cmd_args.emshort == "CPF":
movie.brightmaxfunc = bmaxCPF
else:
raise ValueError, "Unknown mode: {}".format(cmd_args.mode)
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()
