import pylab
from pymses import RamsesOutput
from pymses.analysis.visualization import *

# RamsesOutput
ro = RamsesOutput("/data/Aquarius/output/", 193)

# AMR data source
amr = ro.amr_source(["rho"])

# Defining a Camera object
cam  = Camera(center=[0.5, 0.5, 0.5], line_of_sight_axis='z', region_size=[1., 1.], up_vector='y', map_max_size=256, log_sensitive=True)

# Density field access operator
rho_op = ScalarOperator(lambda dset: dset["rho"])

# Slice map computation
map = SliceMap(amr, cam, rho_op, z=0.4) # create a density slice map at z=0.4 depth position

map = apply_log_scale(map)

pylab.imshow(map)

#pylab.show()