"""
The following recipe will run the HaloProfiler (see :ref:`halo_profiling`) on
all the datasets in one simulation between z = 10 and 0.
"""
import yt.extensions.EnzoSimulation as ES
import yt.extensions.HaloProfiler as HP

es = ES.EnzoSimulation("simulation_parameter_file", initial_redshift=10, final_redshift=0)

# Loop over all dataset in the requested time interval.
for output in es.allOutputs:

    # Instantiate HaloProfiler for this dataset.
    hp = HP.HaloProfiler(output['filename'])
    
    # Add a virialization filter.
    hp.add_halo_filter(HP.VirialFilter,must_be_virialized=True,
                       overdensity_field='ActualOverdensity',
                       virial_overdensity=200,
                       virial_filters=[['TotalMassMsun','>=','1e14']],
                       virial_quantities=['TotalMassMsun','RadiusMpc'])
    
    # Add profile fields.
    hp.add_profile('CellVolume',weight_field=None,accumulation=True)
    hp.add_profile('TotalMassMsun',weight_field=None,accumulation=True)
    hp.add_profile('Density',weight_field="CellMassMsun",accumulation=False)
    hp.add_profile('Temperature',weight_field='CellMassMsun',accumulation=False)
    # Make profiles and output filtered halo list to FilteredQuantities.out.
    hp.make_profiles(filename="FilteredQuantities.out")
    
    # Add projection fields.
    hp.add_projection('Density',weight_field=None)
    hp.add_projection('Temperature',weight_field='Density')
    hp.add_projection('Metallicity',weight_field='Density')
    # Make projections for all three axes using the filtered halo list and 
    # save data to hdf5 files.
    hp.make_projections(save_cube=True,save_images=True,
                        halo_list='filtered',axes=[0,1,2])
    
    del hp