Added packages for linear grayscale colour varients.
This commit is contained in:
		
							
								
								
									
										82
									
								
								lgray/lgray.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										82
									
								
								lgray/lgray.go
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,82 @@ | ||||
| // Provides a [color.Color] type for dealing with linear grayscale colours without alpha. | ||||
| package lgray | ||||
|  | ||||
| import ( | ||||
| 	"image/color" | ||||
| 	"math" | ||||
|  | ||||
| 	"smariot.com/color/internal/helper" | ||||
| ) | ||||
|  | ||||
| // Color is a linear grayscale [color.Color]. | ||||
| type Color struct { | ||||
| 	Y float64 | ||||
| } | ||||
|  | ||||
| // DistanceSqr returns the euclidean distance squared between two colours. | ||||
| func DistanceSqr(a, b Color) float64 { | ||||
| 	dY := a.Y - b.Y | ||||
| 	return dY * dY | ||||
| } | ||||
|  | ||||
| // Distance returns the euclidean distance between two colours, | ||||
| // | ||||
| // If you just want to compare relative distances, use [DistanceSqr] instead. | ||||
| func Distance(a, b Color) float64 { | ||||
| 	return math.Abs(a.Y - b.Y) | ||||
| } | ||||
|  | ||||
| // RGBA converts to premultiplied RGBA, implementing the [color.Color] interface. | ||||
| func (c Color) RGBA() (r, g, b, a uint32) { | ||||
| 	_y := helper.Delinearize(c.Y) | ||||
| 	return _y, _y, _y, 0xffff | ||||
| } | ||||
|  | ||||
| // NRGBA converts to non-premultiplied RGBA. | ||||
| func (c Color) NRGBA() (r, g, b, a uint32) { | ||||
| 	_y := helper.Delinearize(c.Y) | ||||
| 	return _y, _y, _y, 0xffff | ||||
| } | ||||
|  | ||||
| // NLRGBA converts to non-premultiplied linear RGBA. | ||||
| func (c Color) NLRGBA() (r, g, b, a float64) { | ||||
| 	return c.Y, c.Y, c.Y, 1 | ||||
| } | ||||
|  | ||||
| // NXYZA converts to non-premultiplied XYZ+Alpha. | ||||
| func (c Color) NXYZA() (x, y, z, a float64) { | ||||
| 	x, y, z = helper.LRGBtoXYZ(c.Y, c.Y, c.Y) | ||||
| 	return x, y, z, 1 | ||||
| } | ||||
|  | ||||
| // NOkLabA converts to non-premultiplied OkLab+Alpha. | ||||
| func (c Color) NOkLabA() (lightness, chromaA, chromaB, a float64) { | ||||
| 	lightness, chromaA, chromaB = helper.LMStoOkLab(helper.LRGBtoLMS(c.Y, c.Y, c.Y)) | ||||
| 	return lightness, chromaA, chromaB, 1 | ||||
| } | ||||
|  | ||||
| // Convert converts an arbitrary colour type to a linear RGB [Color]. | ||||
| func Convert(c color.Color) Color { | ||||
| 	if c, ok := c.(Color); ok { | ||||
| 		return c | ||||
| 	} | ||||
|  | ||||
| 	r, g, b, _ := helper.ColorToNLRGBA(c) | ||||
|  | ||||
| 	// the color.Gray16Model documents that it uses the coefficients 0.299, 0.5867, and 0.114. | ||||
| 	// however, it does this using integer arithmetic, so the actual coefficients are effectively rounded to the nearest 1/0x10000. | ||||
|  | ||||
| 	return Color{Y: helper.DelinearizeF( | ||||
| 		helper.LinearizeF(r)*0x0.4c8bp0 + | ||||
| 			helper.LinearizeF(g)*0x0.9646p0 + | ||||
| 			helper.LinearizeF(b)*0x0.1d2fp0, | ||||
| 	)} | ||||
| } | ||||
|  | ||||
| // A [color.Model] for converting arbitrary colours to a linear RGB [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{} | ||||
							
								
								
									
										47
									
								
								lgray/lgray_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										47
									
								
								lgray/lgray_test.go
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,47 @@ | ||||
| package lgray | ||||
|  | ||||
| import ( | ||||
| 	"math" | ||||
| 	"testing" | ||||
|  | ||||
| 	"smariot.com/color/internal/helper" | ||||
| ) | ||||
|  | ||||
| func eq(c0, c1 Color) bool { | ||||
| 	return helper.EqFloat64Fuzzy(c0.Y, c1.Y) | ||||
| } | ||||
|  | ||||
| func midpoint(c0, c1 Color) Color { | ||||
| 	return Color{(c0.Y + c1.Y) / 2} | ||||
| } | ||||
|  | ||||
| func TestModel(t *testing.T) { | ||||
| 	helper.TestModel(t, false, false, Model, eq, []helper.ConvertTest[Color]{ | ||||
| 		{ | ||||
| 			Name: "passthrough +inf", | ||||
| 			In:   Color{math.Inf(1)}, | ||||
| 			Out:  Color{math.Inf(1)}, | ||||
| 		}, { | ||||
| 			Name: "passthrough -inf", | ||||
| 			In:   Color{math.Inf(-1)}, | ||||
| 			Out:  Color{math.Inf(-1)}, | ||||
| 		}, { | ||||
| 			Name: "passthrough nan", | ||||
| 			In:   Color{math.NaN()}, | ||||
| 			Out:  Color{math.NaN()}, | ||||
| 		}, | ||||
| 	}) | ||||
| } | ||||
|  | ||||
| func distance(a, b Color) float64 { | ||||
| 	d := Distance(a, b) | ||||
| 	dSqr := DistanceSqr(a, b) | ||||
| 	if !helper.EqFloat64Fuzzy(d*d, dSqr) { | ||||
| 		panic("Distance and DistanceSqr are not equivalent") | ||||
| 	} | ||||
| 	return d | ||||
| } | ||||
|  | ||||
| func TestDistance(t *testing.T) { | ||||
| 	helper.TestDistance(t, false, false, midpoint, distance, Model) | ||||
| } | ||||
		Reference in New Issue
	
	Block a user