import numpy as np
import crl, crl.data
from matplotlib import pyplot as plt
from shared import *

lamps = [Lamp(k, *v) for k, v in labdb.items()]

with TexWriter(os.path.join(os.path.dirname(__file__), "../build/lab1.tex")) as fd:
	for lamp in lamps:
		fd.write("%s & %g & %g & %.4g & $%s$ & %.1f & %.1f & %.1f & %.1f & %.1f \\\\\n"
		         % (lamp.lid, lamp.CCT_spec, lamp.ref_2deg["CCT"],
		            lamp.CCT, lamp.Ra_spec, lamp.ref_2deg["CRI"],
		            lamp.Ra.CRI, lamp.Ra.ECRI, lamp.GAI, lamp.FSCI))

with TexWriter(os.path.join(os.path.dirname(__file__), "../build/lab2a.tex")) as fd:
	for lamp in lamps:
		fd.write("%s" % lamp.lid)
		for i in range(8):
			fd.write(" & %.1f & %.1f"
			         % (lamp.ref_2deg["SCRI"][i], lamp.Ra.SCRIs[i]))
		fd.write("\\\\\n")

with TexWriter(os.path.join(os.path.dirname(__file__), "../build/lab2b.tex")) as fd:
	for lamp in lamps:
		fd.write("%s" % lamp.lid)
		for i in range(8, 14):
			fd.write(" & %.1f & %.1f"
			         % (lamp.ref_2deg["SCRI"][i], lamp.Ra.SCRIs[i]))
		fd.write("\\\\\n")