from math import log10
import math
import numpy as na
#nullfmt = NullFormatter()
#from mpi4py import MPI
import re
import glob
import os
import time

#comm = MPI.COMM_WORLD
#rank = comm.Get_rank()
#size=comm.size

redshift = 15.0
#redshift_end = 23.5
#tfinal = 4345107   #4615649   # in years
nx = 512
Ex = 300  # photon energy (eV)

for redshift in [24,23.5,23,22,21,20,19.5,19,18.5,17.91,17.45,16.99,16.6,15.99,15.4,15.0]:
 filelist = glob.glob("jXRAY_*%6.3f_dat"%redshift)
 Xray_intensity = 0
 for i,name in enumerate(filelist):
  xrayzstart = float(re.findall(r"[-+]?\d*\.\d+|\d+",name)[0])
  xrayzend = float(re.findall(r"[-+]?\d*\.\d+|\d+",name)[1])
  xrayenergy = Ex*(1+redshift)/(1+(xrayzstart+xrayzend)/2)
  f = open(name,'r')
  Xray=(na.fromfile(f,dtype='float64').reshape(nx,nx,nx))*(1+redshift)/(1+(xrayzstart+xrayzend)/2)
  print "Reading %s"%name, xrayzstart, xrayzend, xrayenergy
  f.close()
  Xray_intensity +=Xray.mean()

 print Xray_intensity/4/math.pi
#del rho,Xray_r
#del Tem, ef