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

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

dlist = glob.glob('RD00??')
dlist.sort()
#output = int(re.sub(r'\D',"",fn))
for folder in dlist:
  n = int(re.sub(r'\D',"",folder))
  #if n > 5 :continue
  if glob.glob(folder+'/PopII_haloes.txt') !=[]: continue
  f=open(folder+'/PopII_haloes_%04i.txt'%n,'r')
  f2=open(folder+'/Overlaped_haloes_list.txt','r')
  overlaped=[]
  line=f2.readline()
  while line !='':
   overlaped.append(int(line))
   line=f2.readline()
  fo = open(folder+'/PopII_haloes.txt','w')
  line = f.readline()
  j=0
  while line !='':
   mass = line.split()[1]
   line = f.readline()
   pn = line.split()[0] 
   m = []
   for i in xrange(int(pn)):
     line=f.readline()
     starmass = float(line.split()[1])
     m.append(starmass)
   if len(m) == 0: 
      pm = 0; pms =0
   else:
      pm=np.array(m).mean()
      pms=np.array(m).std()
   if j not in overlaped:
      fo.write(str(mass)+" "+str(len(m))+" "+str(pm)+" "+str(pms)+"\n")
   f.readline();f.readline()
   line = f.readline()
   j=j+1
  f.close()
  fo.close()