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

output = 108

fn = 'output_%04i'%(output)

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)
#center=np.array(halo_list[0].center_of_mass())
#halo_list.write_out("DD%04i/sv.out"%output)
halo_list.dump("DD%04i"%output)

f=open('./halo_list.txt'%output,'a')
f.write('Halo Mass    Halo Radius virial Max   Halo center of mass ')
for i in xrange(20):
    mass =  halo_list[i].total_mass()
    R_v = halo_list[i].virial_radius()
    R = halo_list[i].maximum_radius()
    position = halo_list[i].center_of_mass()
    f.write(str(mass)+"  "+str(R_v)+"  "+str(R)+"  "+str(position[0])+"  "+str(position[1])+"  "+str(position[2])+"\n")

f.close()