getData.py

#!/usr/bin/python


from scipy.interpolate import interp1d
import sys
from parse import *
import yaml
import numpy as np
import cmath
from sympy import *

#This function is designed to return refractive index for specified lambda
#for file in "tabluated n " format
def getDataN(yamlFile,lamb,retcoeffs=False):
	yamlStream=open(yamlFile,'r')
	allData=yaml.load(yamlStream);	

	materialData=allData["DATA"][0]


	assert materialData["type"]=="tabulated n"

	matLambda=[]
	matN=[]
	matK=[]
	#in this type of material read data line by line
	for line in materialData["data"].split('\r\n'):
		parsed=parse("{l:g} {n:g} {k:g}",line)
		try:
			n=parsed["n"]+1j*parsed["k"]
			matLambda.append(parsed["l"]);
			matN.append(parsed["n"])
			matK.append(parsed["k"])
		except TypeError as e:
			sys.stderr.write("TypeError occured:"+str(e)+"\r\n")

	matLambda=np.array(matLambda)
	matN=np.array(matN)
	matK=np.array(matK)

	interN=interp1d(matLambda,matN)
	interK=interp1d(matLambda,matK)
	
	return [ cmath.sqrt(x) for x in np.nditer(interN(lamb))]


def getRangeN(yamlFile):
	yamlStream=open(yamlFile,'r')
	allData=yaml.load(yamlStream);	

	materialData=allData["DATA"][0]

	assert materialData["type"]=="tabulated n"
	#in this type of material read data line by line
	matLambda=[]
	for line in materialData["data"].split('\r\n'):
		parsed=parse("{l:g} {n:g}",line)
		try:
			matLambda.append(parsed["l"])
		except TypeError as e:
			sys.stderr.write("TypeError occured:"+str(e)+"\r\n")
	return (min(matLambda),max(matLambda))
		
		




#This function is designed to return refractive index for specified lambda
#for file in "tabluated nk " format
def getDataNK(yamlFile,lamb,retcoeffs=False):
	yamlStream=open(yamlFile,'r')
	allData=yaml.load(yamlStream);	

	materialData=allData["DATA"][0]


	assert materialData["type"]=="tabulated nk"

	matLambda=[]
	matN=[]
	matK=[]
	#in this type of material read data line by line
	for line in materialData["data"].split('\r\n'):
		parsed=parse("{l:g} {n:g} {k:g}",line)
		try:
			n=parsed["n"]+1j*parsed["k"]
			matLambda.append(parsed["l"]);
			matN.append(parsed["n"])
			matK.append(parsed["k"])
		except TypeError as e:
			sys.stderr.write("TypeError occured:"+str(e)+"\r\n")

	matLambda=np.array(matLambda)
	matN=np.array(matN)
	matK=np.array(matK)

	interN=interp1d(matLambda,matN)
	interK=interp1d(matLambda,matK)
	
	return [ x for x in np.nditer(interN(lamb)+1j*interK(lamb)) ]

def getRangeNK(yamlFile):
	yamlStream=open(yamlFile,'r')
	allData=yaml.load(yamlStream);	

	materialData=allData["DATA"][0]

	assert materialData["type"]=="tabulated nk"
	#in this type of material read data line by line
	matLambda=[]
	for line in materialData["data"].split('\r\n'):
		parsed=parse("{l:g} {n:g} {k:g}",line)
		try:
			matLambda.append(parsed["l"])
		except TypeError as e:
			sys.stderr.write("TypeError occured:"+str(e)+"\r\n")
	return (min(matLambda),max(matLambda))
		
		


##this function is desined to get data from files in formula1 format
def getDataF1(yamlFile,lamb,retcoeffs=False):
	yamlStream=open(yamlFile,'r')
	allData=yaml.load(yamlStream);	

	materialData=allData["DATA"][0]

	assert materialData["type"]=="formula 1"

	dataRange=np.array(map(float,materialData["range"].split()));
	coeff=np.array(map(float,materialData["coefficients"].split()));

	n=0
	L = Symbol('L')
	N=0
	if min(lamb)>=dataRange[0] or max(lamb)<=dataRange[1]:
		for i in reversed(range(1,np.size(coeff),2)):
			n=n+((coeff[i]*lamb**2)/(lamb**2-coeff[i+1]**2))
			N=N+((coeff[i]*L**2)/(L**2-coeff[i+1]**2))
		
	else:
		raise Exception("OutOfBands","No data for this material for this l")
	
	n=n+coeff[0]+1
	N=sqrt(N+coeff[0]+1)
	epsRes=[]
	for oneN in n:
		epsRes.append(cmath.sqrt(oneN))
	return [epsRes,N]


##this function is desined to get data from files in formula2 format
def getDataF2(yamlFile,lamb,retcoeffs=False):
	yamlStream=open(yamlFile,'r')
	allData=yaml.load(yamlStream);	

	materialData=allData["DATA"][0]

	assert materialData["type"]=="formula 2"

	dataRange=np.array(map(float,materialData["range"].split()));
	coeff=np.array(map(float,materialData["coefficients"].split()));

	n=0
	L = Symbol('L')
	N=0
	if min(lamb)>=dataRange[0] or max(lamb)<=dataRange[1]:
		for i in reversed(range(1,np.size(coeff),2)):
			n=n+((coeff[i]*lamb**2)/(lamb**2-coeff[i+1]))
			N=N+((coeff[i]*L**2)/(L**2-coeff[i+1]))
		
	else:
		raise Exception("OutOfBands","No data for this material for this l")
	
	n=n+coeff[0]+1
	N=sqrt(N+coeff[0]+1)
	nres=[]
	for oneN in n:
		nres.append(cmath.sqrt(oneN))
	return [nres,N]


##this function is desined to get data from files in formula3 format
def getDataF3(yamlFile,lamb,retcoeffs=False):
	yamlStream=open(yamlFile,'r')
	allData=yaml.load(yamlStream);	

	materialData=allData["DATA"][0]

	assert materialData["type"]=="formula 3"

	dataRange=np.array(map(float,materialData["range"].split()));
	coeff=np.array(map(float,materialData["coefficients"].split()));




	n=coeff[0]
	L = Symbol('L');
	N=coeff[0];
	if min(lamb)>=dataRange[0] and max(lamb)<=dataRange[1]:
		for i in range(1,np.size(coeff),2):
			n=n+coeff[i]*lamb**coeff[i+1]
			N=N+coeff[i]*L**coeff[i+1]
	else:
		raise Exception("OutOfBands","No data for this material for lambda="+str(lamb))
	
	nres=[]
	N=sqrt(N);
	for oneN in n:
		nres.append(cmath.sqrt(oneN))
	return nres

def getDataF4(yamlFile,lamb,retcoeffs=False):
	yamlStream=open(yamlFile,'r')
	allData=yaml.load(yamlStream);	

	materialData=allData["DATA"][0]

	assert materialData["type"]=="formula 4"

	dataRange=np.array(map(float,materialData["range"].split()));
	coeff=np.zeros(17)
	coeff2=map(float,materialData["coefficients"].split());

	for i in range(0,len(coeff2)):
		coeff[i]=coeff2[i]

	print(coeff)




	n=coeff[0]
	L = Symbol('L');
	N = coeff[0];
	if min(lamb)>=dataRange[0] and max(lamb)<=dataRange[1]:
			n=n+coeff[1]*lamb**coeff[2]/(lamb**2-coeff[3]**coeff[4])
			n=n+coeff[5]*lamb**coeff[6]/(lamb**2-coeff[7]**coeff[8])
			n=n+coeff[9]*lamb**coeff[10]
			n=n+coeff[11]*lamb**coeff[12]
			n=n+coeff[13]*lamb**coeff[14]
			n=n+coeff[15]*lamb**coeff[16]

			N=N+coeff[1]*L**coeff[2]/(L**2-coeff[3]**coeff[4])
			N=N+coeff[5]*L**coeff[6]/(L**2-coeff[7]**coeff[8])
			N=N+coeff[9]*L**coeff[10]
			N=N+coeff[11]*L**coeff[12]
			N=N+coeff[13]*L**coeff[14]
			N=N+coeff[15]*L**coeff[16]
	else:
		raise Exception("OutOfBands","No data for this material("+yamlFile+" )for lambda="+str(lamb))
	
	nres=[]
	N = sqrt(N)
	for oneN in n:
		nres.append(cmath.sqrt(oneN))
	return [nres,N]








def Error(BaseException):
	pass

###this is general function to check data type, and run appropriate actions
def getData(yamlFile,lamb,retcoeffs=False):
	yamlStream=open(yamlFile,'r')
	allData=yaml.load(yamlStream)
	materialData=allData["DATA"][0]

	if materialData["type"]=="tabulated nk":
		return	getDataNK(yamlFile,lamb)
	elif materialData["type"]=="tabulated n":
		return	getDataN(yamlFile,lamb)
	elif materialData["type"]=="formula 1":
		return getDataF1(yamlFile,lamb)
	elif materialData["type"]=="formula 2":
		return getDataF2(yamlFile,lamb)
	elif materialData["type"]=="formula 3":
		return getDataF3(yamlFile,lamb)
	elif materialData["type"]=="formula 4":
		return getDataF4(yamlFile,lamb)

	else:
	#	raise Error("UnsupportedDataType:This data type is currnetly not supported");
		return np.zeros((1,0))

def getRange(yamlFile):
	yamlStream=open(yamlFile,'r')
	allData=yaml.load(yamlStream)
	materialData=allData["DATA"][0]

	if materialData["type"]=="tabulated nk":
		return	getRangeNK(yamlFile)
	elif materialData["type"]=="tabulated n":
		return	getRangeN(yamlFile)
	elif materialData["type"]=="formula 1":
		return np.array(map(float,materialData["range"].split())) 
	elif materialData["type"]=="formula 2":
		return np.array(map(float,materialData["range"].split())) 
	elif materialData["type"]=="formula 3" or materialData["type"]=="formula 4":
		return np.array(map(float,materialData["range"].split()))
	else:
	#	raise Error("UnsupportedDataType:This data type "+materialData["type"] +" is currnetly not supported");
		return (0,0)




if __name__ == "__main__":
	print("Direct tests:")
	print("MyData(Rakic)="+str(getDataNK("database/main/Ag/Rakic.yml",1))+" ,test ~  0.23 + 6.41j")
	print("MyData(Moutzouris-e)="+str(getDataF3("database/other/doped crystals/Mg-LiTaO3/Moutzouris-e.yml",np.array([0.7473])))+" test ~")
	print("MyData(Ghosh-o)="+str(getDataF2("./database/main/CaCO3/Ghosh-o.yml",np.array([0.204])))+", test ~ 1.88")
	print("Tests through general getData function ")
	print("MyData(Rakic)="+str(getData("database/main/Ag/Rakic.yml",1)))
	print("MyData(Moutzouris-e)="+str(getData("database/other/doped crystals/Mg-LiTaO3/Moutzouris-e.yml",np.array([0.7473]))))
	print("MyData(Ghosh-o)="+str(getData("./database/main/CaCO3/Ghosh-o.yml",np.array([0.204]))))