from yt.mods import *

for n in range(0,28):
    pf = load("RD%04i/RedshiftOutput%04i" % (n,n))
    halos = HaloFinder(pf,padding=0.1)
    halos.write_out("%s_halos.txt" % pf)
    pc = PlotCollection(pf)
    p = pc.add_projection("Dark_Matter_Density",0)
    p.modify["hop_circles"](halos)
    p = pc.add_projection("Density",0)
    p.modify["hop_circles"](halos)
    p = pc.add_projection("Temperature",0)
    p.modify["hop_circles"](halos)
    p = pc.add_projection("Dark_Matter_Density",1)
    p.modify["hop_circles"](halos)
    p = pc.add_projection("Density",1)
    p.modify["hop_circles"](halos)
    p = pc.add_projection("Temperature",1)
    p.modify["hop_circles"](halos)
    p = pc.add_projection("Dark_Matter_Density",2)
    p.modify["hop_circles"](halos)
    p = pc.add_projection("Density",2)
    p.modify["hop_circles"](halos)
    p = pc.add_projection("Temperature",2)
    p.modify["hop_circles"](halos)
    for height in range(0,4):
        pos = height/4.0    
        p = pc.add_slice("Dark_Matter_Density",0,center=[pos,0.5,0.5])
        p.modify["hop_circles"](halos)
        p = pc.add_slice("Density",0,center=[pos,0.5,0.5])
        p.modify["hop_circles"](halos)
        p = pc.add_slice("Temperature",0,center=[pos,0.5,0.5])
        p.modify["hop_circles"](halos)
        p = pc.add_slice("Dark_Matter_Density",1,center=[0.5,pos,0.5])
        p.modify["hop_circles"](halos)
        p = pc.add_slice("Density",1,center=[0.5,pos,0.5])
        p.modify["hop_circles"](halos)
        p = pc.add_slice("Temperature",1,center=[0.5,pos,0.5])
        p.modify["hop_circles"](halos)
        p = pc.add_slice("Dark_Matter_Density",2,center=[0.5,0.5,pos])
        p.modify["hop_circles"](halos)
        p = pc.add_slice("Density",2,center=[0.5,0.5,pos])
        p.modify["hop_circles"](halos)
        p = pc.add_slice("Temperature",2,center=[0.5,0.5,pos])
        p.modify["hop_circles"](halos)
        pc.save("images/%s_%i" % (pf,height))
    del pf

# the end