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

#output = 23

#fn = 'output_%04i'%(output)
#fn = 'RedshiftOutput%04i'%(output)
fn, = glob.glob('RedshiftOutput????') 
output = int(re.sub(r'\D',"",fn))

pf = load(fn)
vl=pf["RefineRegionLeftEdge"]
vr=pf["RefineRegionRightEdge"]
sv = pf.h.region((vl+vr)/2, vl, vr)
halo_list=parallelHF(pf,dm_only=False,subvolume = sv,threshold=80)
#halo_list=HaloFinder(pf,dm_only=True,subvolume = sv)
#center=np.array(halo_list[0].center_of_mass())
halo_list.write_out("sv.out")
halo_list.dump("RD%04i"%output)