import time, copy, pickle

import os, os.path

import sys

import numpy, pylab

sys.path.append('C:/Users/Gregoire/Documents/PythonCode/JCAP')

from readplatemap import *

#modelpath='C:/Users/Gregoire/Documents/CaltechWork/platemaps/Quaternarysingleplate/plate333_1map_full.txt'

#newpath='C:/Users/Gregoire/Documents/CaltechWork/platemaps/Quaternarysingleplate/plate20intervwbin.txt'

#writelines=[]

#f=open(modelpath, mode='r')

#ls=f.readlines()[:2]

#writelines+=[l.strip() for l in ls]

#f.close()

#dlist=readsingleplatemaptxt(modelpath, returnfiducials=False)

#dlistsrc=readplatemaptxt(codes=[0, 1, 2, 3])

#

#smpsrc=numpy.array([d['Sample'] for d in dlistsrc])

#codesrc=numpy.array([d['code'] for d in dlistsrc])

intervs=20

cutmiddleinterv=1

cmi=cutmiddleinterv

compsint=[[b, c, (intervs-a-b-c), a] for a in numpy.arange(0,intervs+1)[::-1] for b in numpy.arange(0,intervs+1-a) for c in numpy.arange(0, intervs+1-a-b)][::-1]

print len(compsint)

#compsint2=[[a, b, c, d] for (a, b, c, d) in compsint if numpy.any((numpy.array([a, b, c, d])<cmi)) or numpy.sum(numpy.array([a, b, c, d])%2)==0]# or (numpy.sum(numpy.array([a, b, c, d])%2)==0) )]#+(numpy.array([a, b, c, d])>(intervs-cmi)))]

#comps=numpy.float32(compsint2)/intervs

compsint2=[[b, c, (intervs-a-b-c), a] for a in numpy.arange(0,intervs+1, 2)[::-1] for b in numpy.arange(a,intervs+1-2*a) for c in numpy.arange(a, intervs+1-2*a-b)][::-1]

compsint2+=[[d, a, b, c] for a, b, c, d in compsint2 if not [d, a, b, c] in compsint2]

compsint2+=[[c, d, a, b] for a, b, c, d in compsint2 if not [c, d, a, b] in compsint2]

compsint2+=[[b, c, d, a] for a, b, c, d in compsint2 if not [b, c, d, a] in compsint2]

compsint2+=[cc for cc in compsint if 1 in cc and not (cc in compsint2)]

comps=numpy.float32(compsint2)/intervs

#comps=numpy.array([comps.T[1], comps.T[2], comps.T[3], comps.T[0]]).T

print len(comps)

def genbinarycomps(intervs, elind1, elind2, ndim=4):

return c

#comps2=comps

#codes=[0]*len(comps)

#binintervs=5

#for i, j in [(0, 1), (0, 2), (0, 3), (1, 2), (1, 3), (2, 3)]:

# comps2+=list(genbinarycomps(binintervs, i, j))+[numpy.zeros(4, dtype='float64')] #add 6 compositions in binary line and then zeros

# codes+=[4]*6+[1]

#comps2+=[numpy.zeros(4, dtype='float64')]*6 #6 more zeros to round out the 1819 code0 samples in a standard platemap

#codes+=[1]*6

#comps2=[numpy.array(c) for c in comps2]

#

#comps2pop=copy.copy(comps2)

#codespop=copy.copy(codes)

#

#for d in dlist:

# if d['code']==0:

# c=comps2pop.pop(0)

# cd=codespop.pop(0)

# for k, v in zip(['A', 'B', 'C', 'D'], c):

# d[k]=v

# d['code']=cd

#

#k_f=[\

#('Sample','%04d'),\

#('x','%.2f'),\

#('y','%.2f'),\

#('dx','%.2f'),\

#('dx','%.2f'),\

#('A','%.3f'),\

#('B','%.3f'),\

#('C','%.3f'),\

#('D','%.3f'),\

#('E','%.3f'),\

#('F','%.3f'),\

#('G','%.3f'),\

#('H','%.3f'),\

#('code','%d'),\

#]

#

##writelines+=[', '.join([f %d[k] for k, f in k_f]) for d in dlist]

#

##f=open(newpath, mode='w')

##f.write('\n'.join(writelines))

##f.close()

#

sys.path.append('C:/Users/Gregoire/Documents/PythonCode/ternaryplot')

from myquaternaryutility import QuaternaryPlot

from quaternary_FOM_stackedtern20 import *

axl, stpl=make20ternaxes()

#

#for d in dlist:

# c=numpy.array([d[el] for el in ['A', 'B', 'C', 'D']])

# if c.sum()>0:

# c/=c.sum()

# d['compositions']=c

#carr=numpy.array([d['compositions'] for d in dlist])

carr=numpy.array(comps)

pylab.figure()

stpq=QuaternaryPlot(111)

#stpq.scatter(carr)

cols=stpq.rgb_comp(carr)

stpq.scatter(carr, c=cols, s=20, edgecolors='none')

scatter_20axes(carr, cols, stpl, s=20, edgecolors='none', cb=False)

stpq.label()

axl[0].figure.savefig('C:/Users/Gregoire/Documents/CaltechWork/platemaps/nestedtetr/test2.png')

pylab.show()