from gaepsi.cosmology import WMAP7 from sys import argv from nishitab import subhalodtype as nishitabdtype from matplotlib.figure import Figure from matplotlib import cm snapid = int(argv[1]) snappath = '/physics/yfeng1/mb2/snapdir/snapdir_%03d/snapshot_%03d.%%d' %(snapid, snapid) nishitabpath = '/physics/yfeng1/mb2/galaxy_props/%03d/subhalo_tab_%03d.bin' \ % (snapid, snapid) N = numpy.fromfile(nishitabpath, 'i4', 1)[0] nishitab = \ numpy.memmap('/physics/yfeng1/mb2/galaxy_props/%03d/subhalo_tab_%03d.bin' \ % (snapid, snapid), mode='r', offset=4, dtype=nishitabdtype, shape=N) ct = GaplotContext() ct.use(snappath, 'cmugadget', periodic=True) print ct.C['Ntot'] print ct.C['redshift'] halomass = numpy.array([100, 200, 500, 800, 1000, 2000, 5000, 10000, 20000, 50000], dtype='f8') haloarg = [numpy.abs(nishitab['mass'] / mass - 1.0).argmin()\ for mass in halomass] center = nishitab['pos'][haloarg] vel = nishitab['vel'][haloarg] mass = nishitab['mass'][haloarg] #radius = WMAP7.Rvir(nishitab['mass'][haloarg], z=5.0) * 1.5 radius = WMAP7.Rvir(nishitab['mass'][haloarg], z=ct.C['redshift']) print center, mass, radius raise