import numpy as np

n=38

fi1 = open('sv.out','r')
fi2 = open('PopII_list_%04i.txt'%n,'r')

fo1=open('PopII_haloes_%04i.txt'%n,'w')

line = fi1.readline()
line = fi1.readline()
line = fi1.readline()

i=0
while line !='':
   hmass = float(line.split()[1])
   hradius=float(line.split()[13])
   hcx=float(line.split()[7])
   hcy=float(line.split()[8])
   hcz=float(line.split()[9])
   fo1.write(str(i)+" "+str(hmass)+" "+str(hradius)+" "+str(hcx)+" "+str(hcy)+" "+str(hcz)+"\n")
   ct = [];mass=[];cx=[];cy=[];cz=[]
   fi2.seek(0)
   line = fi2.readline()
   line = fi2.readline()
   while line !="":
     sx=float(line.split()[3])
     sy=float(line.split()[4])
     sz=float(line.split()[5]) 
     if ((hcx-sx)**2+(hcy-sy)**2+(hcz-sz)**2)**0.5<hradius:
         m=float(line.split()[1])
         if m > 1e20: m=m/1e20
         mass.append(m)
         ct.append(float(line.split()[2]))
         cx.append(sx);cy.append(sy),cz.append(sz)
     line = fi2.readline()
   fo1.write("   "+str(len(mass))+"   "+str(np.mean(mass))+"\n")
   for j in xrange(len(mass)):
     fo1.write("   "+str(j)+" "+str(mass[j])+" "+str(ct[j])+" "+str(0)+" "+str(cx[j])+" "+str(cy[j])+" "+str(cz[j])+"\n")
   fo1.write("\n\n")
   line = fi1.readline()
   i=i+1

fi1.close()
fi2.close()
fo1.close()