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

#output = 23

f=open("star_count.txt",'a')

#fn = 'output_%04i'%(output)
#fn = 'RedshiftOutput%04i'%output
dlist = glob.glob('RD00??')
dlist.sort()
#dlist.reverse() 
#output = int(re.sub(r'\D',"",fn))
for folder in dlist:
  i = int(re.sub(r'\D',"",folder))
  if i < 20 :continue
  a=commands.getoutput('grep CurrentRedshift '+folder+'/RedshiftOutput00??')
  z=float(a.split()[2])
  a=commands.getoutput('grep InitialTime '+folder+'/RedshiftOutput00??')
  time=float(a.split()[2])
  a=commands.getoutput('tail -n 1 '+folder+'/PopII_list_%04i.txt'%i)
  st = a.split()[0]
  NPopII = int(st if st[0] != '#' else 0)
  a=commands.getoutput('tail -n 1 '+folder+'/PopIII_list_%04i_type1.txt'%i) 
  st = a.split()[0] 
  NPopIII1 = int(st if st[0] != '#' else 0)
  a=commands.getoutput('tail -n 1 '+folder+'/PopIII_list_%04i_type5.txt'%i)
  st = a.split()[0]
  NPopIII5 = int(st if st[0] != '#' else 0)
  f.write(str(z)+" "+str(time)+" "+str(NPopII)+" "+str(NPopIII1)+" "+str(NPopIII5)+" "+str(NPopIII1+NPopIII5)+"\n")

#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()