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()