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

#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:
  n = int(re.sub(r'\D',"",folder))
  if glob.glob(folder+'/Overlaped_haloes_list.txt') !=[]: continue
  f=open(folder+'/PopIII_haloes_%04i_type5.txt'%n,'r')
  fo = open(folder+'/Overlaped_haloes_list.txt','w')
  line = f.readlines()
  radius = []
  cx = []
  cy = []
  cz = []
  lengthlines = len(line)-1
  i=0
  while i<=lengthlines:
   radius.append(float(line[i].split()[2]))
   cx.append(float(line[i].split()[3]))
   cy.append(float(line[i].split()[4]))
   cz.append(float(line[i].split()[5].replace(":","")))
   i=i+1
   pn = line[i].split()[0]
   i=i+int(pn)
   k=len(cx)-1
   for j in xrange(k):
     if ((cx[k]-cx[j])**2+(cy[k]-cy[j])**2+(cz[k]-cz[j])**2)**0.5<radius[j]:
          fo.write(str(k)+"\n")
          break
   i=i+3

  f.close()
  fo.close()