Initial commit.

oklaba/oklaba.go

@@ -0,0 +1,93 @@
+// Provides a [color.Color] type for dealing with premultiplied OkLab+Alpha colours.
+package oklaba
+
+import (
+	"image/color"
+	"math"
+
+	"smariot.com/color/internal/helper"
+)
+
+// Color is a premultiplied OkLab+Alpha [color.Color].
+type Color struct {
+	Lightness, ChromaA, ChromaB, A float64
+}
+
+func sqr(a float64) float64 {
+	return a * a
+}
+
+// DistanceSqr returns the maximum possible euclidean distance squared between two colours,
+// accounting for the possible backgrounds they might be composited over.
+func DistanceSqr(a, b Color) float64 {
+	dLightness := a.Lightness - b.Lightness
+	dChromaA := a.ChromaA - b.ChromaA
+	dChromaB := a.ChromaB - b.ChromaB
+	dA := a.A - b.A
+	return max(sqr(dLightness), sqr(dLightness+dA)) + max(sqr(dChromaA), sqr(dChromaA+dA)) + max(sqr(dChromaB), sqr(dChromaB+dA))
+}
+
+// Distance returns the maximum possible euclidean distance between two colours,
+// accounting for the possible backgrounds they might be composited over.
+//
+// If you just want to compare relative distances, use [DistanceSqr] instead.
+func Distance(a, b Color) float64 {
+	return math.Sqrt(DistanceSqr(a, b))
+}
+
+// RGBA converts to premultiplied RGBA, implementing the [color.Color] interface.
+func (c Color) RGBA() (r, g, b, a uint32) {
+	return helper.NLRGBAtoRGBA(c.NLRGBA())
+}
+
+// NRGBA converts to non-premultiplied RGBA.
+func (c Color) NRGBA() (r, g, b, a uint32) {
+	return helper.NLRGBAtoNRGBA(c.NLRGBA())
+}
+
+// NLRGBA converts to non-premultiplied linear RGBA.
+func (c Color) NLRGBA() (r, g, b, a float64) {
+	if c.A <= 0 {
+		return 0, 0, 0, 0
+	}
+
+	r, g, b = helper.LMStoLRGB(helper.OkLabToLMS(c.Lightness/c.A, c.ChromaA/c.A, c.ChromaB/c.A))
+	return r, g, b, c.A
+}
+
+// NXYZA converts to non-premultiplied XYZ+Alpha.
+func (c Color) NXYZA() (x, y, z, a float64) {
+	if c.A <= 0 {
+		return 0, 0, 0, 0
+	}
+
+	x, y, z = helper.LMStoXYZ(helper.OkLabToLMS(c.Lightness/c.A, c.ChromaA/c.A, c.ChromaB/c.A))
+	return x, y, z, c.A
+}
+
+// NOkLabA converts to non-premultiplied OkLab+Alpha.
+func (c Color) NOkLabA() (lightness, chromaA, chromaB, a float64) {
+	if c.A <= 0 {
+		return 0, 0, 0, 0
+	}
+
+	return c.Lightness / c.A, c.ChromaA / c.A, c.ChromaB / c.A, c.A
+}
+
+// Convert converts an arbitrary colour type to a pre-multiplied OkLab+Alpha [Color].
+func Convert(c color.Color) Color {
+	if c, ok := c.(Color); ok {
+		return c
+	}
+
+	lightness, chromaA, chromaB, a := helper.ColorToNOkLabA(c)
+	return Color{lightness * a, chromaA * a, chromaB * a, a}
+}
+
+// A [color.Model] for converting arbitrary colours to a premultiplied OkLab+Alpha [Color].
+//
+// Wraps the [Convert] function, returning a [color.Color] interface rather than the [Color] type.
+var Model = helper.Model(Convert)
+
+// Type assertion.
+var _ color.Color = Color{}

oklaba/oklaba_test.go

@@ -0,0 +1,36 @@
+package oklaba
+
+import (
+	"math"
+	"testing"
+
+	"smariot.com/color/internal/helper"
+)
+
+func eq(c0, c1 Color) bool {
+	return helper.EqFloat64SliceFuzzy(
+		[]float64{c0.Lightness, c0.ChromaA, c0.ChromaB, c0.A},
+		[]float64{c1.Lightness, c1.ChromaA, c1.ChromaB, c1.A},
+	)
+}
+
+func midpoint(c0, c1 Color) Color {
+	return Color{(c0.Lightness + c1.Lightness) / 2, (c0.ChromaA + c1.ChromaA) / 2, (c0.ChromaB + c1.ChromaB) / 2, (c0.A + c1.A) / 2}
+}
+
+func TestModel(t *testing.T) {
+	helper.TestModel(t, true, Model, eq, []helper.ConvertTest[Color]{
+		{
+			Name: "passthrough",
+			// These is a very illegal colour. If it makes it through
+			// unchanged, we can be reasonably confident no colour space conversions were
+			// attempted.
+			In:  Color{math.Inf(1), math.Inf(-1), math.NaN(), 0},
+			Out: Color{math.Inf(1), math.Inf(-1), math.NaN(), 0},
+		},
+	})
+}
+
+func TestDistance(t *testing.T) {
+	helper.TestDistance(t, true, midpoint, Distance, Model)
+}