#!/usr/bin/python                                 

# prevent old yt files being looked at
from yt.config import ytcfg; ytcfg["yt","serialize"] = "False"

#import yt.extensions.EnzoSimulation as ES
#import yt.extensions.enzo_simulation as ES
from yt.analysis_modules.api import EnzoSimulation as ES
from yt.mods import *

#es = ES.EnzoSimulation("CollapseTest", initial_time=0.0,final_time=20.0) 
es = ES("CollapseTest", initial_time=0.0,final_time=2.0)

#fn="clouddrop_0010"
# pf = load(fn)

# String manipulation because old enzo doesn't store outputs in DD000X files.
# Note, 'output' is a dict object, not a normal string
for output in es.allOutputs:
    print output['filename']
    output2 = output['filename'].split('/')
    fn = output2[2]
    pf = load(fn)
    pc = PlotCollection(pf, center=[100,100,100])
    p = pc.add_slice("Density", 2)                 
    p.modify["grids"]()                                                
    p = pc.add_projection("Density", 2)
    p.modify["grids"]()                        
    pc.save(fn)