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)

sm = sv["ParticleMassMsun"]
ct = sv["creation_time"]
pt = sv["particle_type"]
px = sv["particle_position_x"]
py = sv["particle_position_y"]
pz = sv["particle_position_z"]
# First pick out only PopII stars.
stars = np.logical_and(ct > 0, pt == 7)
ct = ct[stars]
sm = sm[stars]
px = px[stars]
py = py[stars]
pz = pz[stars]


f=open('./PopII_list_%04i.txt'%output,'a')
f.write('#  Mass CreationTime Position (x,y,z) \n')
for i in xrange(ct.size):
    f.write(str(i)+"  "+str(sm[i])+"  "+str(ct[i])+"  "+str(px[i])+"  "+str(py[i])+"  "+str(pz[i])+"\n")

f.close()