import sys
from yt.mods import *
import numpy as np
from yt.analysis_modules.halo_finding.api import *

fields = ['DarkMatterDensity']#"Density", "Temperature", "Metallicity3", "Metallicity", "Electron_Fraction"]

#fn = sys.argv[-3]
#field = sys.argv[-2]
#dim = int(sys.argv[-1])
fn = 'RedshiftOutput0003'

pf = load(fn)
center = [0.5]*3
#sv = pf.h.region([0.5]*3, [0.4, .4, .4], [.6, .6, .6])
#halo_list=parallelHF(pf,subvolume = sv)
#center=np.array(halo_list[0].center_of_mass())
#halo_list.write_out("DD0119/sv.out")
#center=np.array([4.881974554e-01,4.666467383e-01,4.661217068e-01])
pc = PlotCollection(pf, center=center)
field = 'Density'
#pc.add_slice(field, 0 )
#pc.plots[-1].modify["grids"]()
#pc.add_slice(field, 1 )
#pc.plots[-1].modify["grids"]()
#pc.add_slice(field, 2 )
#pc.plots[-1].modify["grids"]()
pc.add_projection(field,0)
pc.add_projection(field,1)
pc.add_projection(field,2)
#pc.add_particles(0,0.1)
#field = 'Density'
#pc.add_slice(field, 0 )
#pc.add_slice(field, 1 )
#pc.add_slice(field, 2 )
#pc.set_width(0.2, "1")
pc.save()