#!/usr/bin/python
from astroML.correlation import two_point
import numpy as np
import pylab as plt
np.random.seed(0)
# X = np.random.random((50000, 3))
file1=raw_input('Enter file name : ')
file2=raw_input('Enter output file name : ')
inp1=raw_input('Enter range : ')
Nlimit=int(inp1)
bins = np.linspace(0, Nlimit, 50)
print 'No of bins are : ',len(bins)
fp = open(file1, 'rt')
lines=fp.readlines()
fp.close()
lv=[]
ev=[]
xv=[]
yv=[]
zv=[]
for line in lines:
p=line.split()
lv.append(float(p[0]))
ev.append(float(p[1]))
xv.append(float(p[2]))
yv.append(float(p[3]))
zv.append(float(p[4]))
Xtemp = [np.array(xv), np.array(yv), np.array(zv)]
X=np.array(Xtemp)
X=X.T
print 'shape of array is : ', X.shape
# print X
# fig, axes = plt.subplots(1, 3, figsize=(12,4))
# axes[0].plot(xv, yv)
# axes[1].plot(xv, zv)
corr = two_point(X, bins)
xx=[]
for ii in range(0, len(bins)-1):
xx.append(0.5*(bins[ii]+bins[ii+1]))
fp = open(file2, 'wt')
for ii in range(0, len(xx)):
fp.write('%f\t%f\n' %(xx[ii], corr[ii]));
fp.close()
print len(xx), len(corr)
plt.plot(xx, corr, color="blue", linewidth=1, ls='-', marker='o', markersize=2)
# axes[2].plot(xx, corr);
plt.show()