import numpy as np

f=open('PopIII_list_0031_type5.txt','r')
f1=open('PopIII_list_0031_type1.txt','r')

fo = open('PopIII_evolution.txt','w')

currenttime = 11.65522741145 

t_min = 0
t_max = 8.0

bins = 100
dt = (t_max-t_min) /bins

line = f.readline()
line1 = f1.readline()
line = f.readline()
line1 =f.readline()

time_bins = np.zeros(bins)
mass_bins = np.zeros(bins)

while line !='':
   mass = float(line.split()[1])
   if mass > 1e20: mass = mass/1e20
   if mass < 5.0: 
      line = f.readline()
      continue   
   ct = float(line.split()[2])
   index = int((currenttime-ct)/dt) 
   time_bins[index] = time_bins[index]+1
   mass_bins[index] = mass_bins[index]+mass
   line = f.readline()

while line1 !='':
   mass = float(line1.split()[1])
   if mass > 1.0e20: mass = mass/1.0e20
   ct = float(line1.split()[2])
   index = int((currenttime-ct)/dt)
   time_bins[index] = time_bins[index]+1
   mass_bins[index] = mass_bins[index]+mass
   line1 = f1.readline()
  
total_stars = time_bins.sum()
total_mass = mass_bins.sum()
for i in xrange(bins):
   fo.write(str(i*dt)+" "+str(currenttime-i*dt)+" "+str(time_bins[i])+" "+str(total_stars) \
     +" "+str(mass_bins[i])+" "+str(total_mass)+"\n")
   total_stars = total_stars - time_bins[i]
   total_mass = total_mass -mass_bins[i]
   if total_mass < 1: total_mass = 0.0

f.close()
f1.close()
fo.close()

