Pastebin

#!/usr/bin/python

import numpy as np
import sys

# import matplotlib
# matplotlib.use('Agg')
# import matplotlib.pyplot as plt
# import matplotlib.cm as cm
from  yt.mods import *
from yt.frontends.stream.api import load_uniform_grid
import yt.visualization.eps_writer as eps

savepath = './'
fileformat = 'png'

loadpath = './'
res = 256

dmdens = np.fromfile(loadpath+'snapshot_cut_5_175_dmMass', dtype=np.dtype(np.float64), count=-1, sep='')
dmdens = dmdens.reshape((res,res,res))


Mpc_cgs = 3.08567758e+24
Mpc_si = 3.08567758E22
G_si = 6.67384E-11
kpc_cgs = 3.08567758e+21
Msol_cgs = 1.98892e+33
h = 0.6711
cFactor = 1E10*Msol_cgs/h/(Mpc_cgs/h)**3
dmdens = dmdens*cFactor

print 'Mean dm density', np.float64(dmdens).mean()


data = dict(dm_density=dmdens)
bbox = np.array([[0.0, 5/h], [0.0, 5/h], [0.0, 5/h]])

pf = load_uniform_grid(data, dmdens.shape, 3.08568025e24, bbox=bbox)

axis = 2
# dark matter
pro=ProjectionPlot(pf,axis,'dm_density')
pro.set_log('dm_density',True)
pro.set_zlim('dm_density',1E-35,1E-31)
pro.set_cmap('dm_density','gist_heat')
eps_fig = eps.single_plot(pro)
eps_fig.save_fig('test',format='pdf')
#pro.save('./gadget_dmdensity_dtfe_20_512.png')