#!/usr/bin/env python
#asdf test
from __future__ import division
from numpy import *
from koko.fab.asdf import ASDF
from matplotlib import pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import csv
import sys

def grab_boxes(asdf,lb,fb,eb,z,depth=0,max_depth=100):
	for b in asdf.branches:
		if b is None: continue
		#print b.zmin,z,b.zmax
		if b.state == 'FILLED' and b.zmin<z<b.zmax:
			fb[depth].append( [[b.xmin,b.ymin],[b.xmax,b.ymax]] )
		if b.state == 'LEAF'   and b.zmin<z<b.zmax:
			lb[depth].append( [[b.xmin,b.ymin],[b.xmax,b.ymax]] )            
		if b.state == 'EMPTY'  and b.zmin<z<b.zmax:
			eb[depth].append( [[b.xmin,b.ymin],[b.xmax,b.ymax]] )
		if b.state == 'BRANCH':
			if depth<max_depth:
				grab_boxes(b,lb,fb,eb,z,depth+1,max_depth=max_depth)

def draw_rect(minp,maxp,col,linewidth=1):
    x = [minp[0],maxp[0],maxp[0],minp[0],minp[0]]
    y = [minp[1],minp[1],maxp[1],maxp[1],minp[1]]
    plt.plot(x,y,c=col,linewidth=linewidth)
def error(near_pi):
	return absolute(pi-near_pi)/pi

'''
filename = 'circle/circle-2.asdf'
print 'opening asdf'
asdf = ASDF.load(filename)
print 'opened asdf of depth %d and %d cells'%(asdf.depth, asdf.cell_count)
ni=2
vol = asdf.volume(ni)/25.4**3
print "%d point rule, volume=%f, error=%e"%(ni,vol,error(vol))
sys.exit(0)
'''
'''
import csv
with open('sphere/fill-face-edge-vert.csv', 'w') as csvfile:
	writer = csv.writer(csvfile, delimiter=',',
                            quotechar='|', quoting=csv.QUOTE_MINIMAL)
	writer.writerow(["res","gauss_pts","volume"])
	for ni in [1,2,3]:
		#for res in [1,2,3,4,5,6,7,8,9,10,11,12,14,15]:
		#for res in [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,20,25]:
		for res in [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15]:
		#for res in [1,2,3,4,5]:
			filename = 'sphere/sphere-'+str(res)+'.asdf'
			#print 'opening asdf'
			asdf = ASDF.load(filename)
			#print 'opened asdf of depth %d and %d cells'%(asdf.depth, asdf.cell_count)
			#print "%d res, %d point rule, volume=%.4f"%(res,ni,asdf.volume(ni)/25.4**3)
			ppi = .75*asdf.volume(ni)/10.**3
			writer.writerow([res,ni,ppi])
sys.exit(0)

#cont_f = 'detail/detail-31.asdf'
cont_f = 'circle/circle-10.asdf'
cont_asdf = ASDF.load(cont_f)
paths = cont_asdf.slice(.01).contour()
'''
N=1
filename = 'sphere/sphere-2.asdf'
asdf = ASDF.load(filename)
asdf.save_integration_points(N,6)

fb = [[] for x in range(asdf.depth)]
eb = [[] for x in range(asdf.depth)]
lb = [[] for x in range(asdf.depth)]
grab_boxes(asdf,lb,fb,eb,.5,max_depth=20);

with open('int_pts_test_filled', 'rU') as csvfile:
		reader = csv.reader(csvfile, delimiter=',', quotechar='|')
		cols = reader.next() #get column headings (in first row)
		units = reader.next() #get column units (in second row)
		data_f = array([ map(float,row) for row in reader])
with open('int_pts_test_leaf', 'rU') as csvfile:
		reader = csv.reader(csvfile, delimiter=',', quotechar='|')
		cols = reader.next() #get column headings (in first row)
		units = reader.next() #get column units (in second row)
		data_l = array([ map(float,row) for row in reader])

c_leaf = (.85,.5,.5)
fig = plt.figure(figsize=(12,8))

ax = fig.add_subplot(111, projection='3d')
ax.plot(data_l[:,0],data_l[:,1],data_l[:,2],linestyle='',marker='.',c=c_leaf)
ax.plot(data_f[:,0],data_f[:,1],data_f[:,2],linestyle='',marker='.')


#2D
'''
zmin = amin(data_f[:,2])
#data_f = data_f[where(data_f[:,2]==zmin)]
#data_l = data_l[where(data_l[:,2]==amin(data_l[:,2]))]
plt.gca().set_aspect(1)
for i in range(0,asdf.depth):
	print len(fb[i])
	for c in lb[i]:
		draw_rect(c[0],c[1],c_leaf,linewidth=1)
	for c in fb[i]:
		draw_rect(c[0],c[1],'b',linewidth=1)


plt.plot(data_f[:,0],data_f[:,1],c='b',linestyle='',marker='.')
plt.plot(data_l[:,0],data_l[:,1],c=c_leaf,linestyle='',marker='.')

plt.gca().get_xaxis().set_visible(False)
plt.gca().get_yaxis().set_visible(False)

for p in paths:
	plt.plot(p.points[:,0],p.points[:,1],c='g')
'''


plt.show()