from shared import *
import crl, crl.tables

bounds = Bounds()

plot_setup(width=0.5, color_plot=True)
plot_grid()
crl.chromaticity_diagram(crl.UVstarNormedD65, locus=True, isotherms=True,
                         pures_labels=False)


for i in range(14):
	S = crl.Spectrum(crl.tables.tcs_ra[:, 0], crl.tables.tcs_ra[:, 1 + i])
	S = crl.CombinedSpectrum(S, crl.Illuminants.D65)

	u, v, _ = crl.UVstarNormedD65(S)
	label = "%d" % (i + 1)

	plt.plot(u, v, "k+", label="TCS" if not i else None)
	plt.annotate(label, xy=(u, v), xytext=(10, 0),
	             textcoords="offset points", fontsize=11,
	             verticalalignment="center",
	             horizontalalignment="left",
	             bbox=dict(boxstyle="square,pad=0.2", fc="white"))
	bounds.point(u, v)


plot_sRGB(crl.UVstarNormedD65)
bounds.set(margin=0.1)
plot_export()