|
8
|
1 import logging
|
|
|
2 from dataclasses import dataclass
|
|
4
|
3
|
|
8
|
4 from dataclasses_json import DataClassJsonMixin
|
|
|
5
|
|
|
6 log = logging.getLogger('colr')
|
|
|
7
|
|
|
8
|
|
|
9 def lerp(a, b, t):
|
|
|
10 return (1 - t) * a + (t) * b
|
|
|
11
|
|
|
12
|
|
|
13 @dataclass(frozen=True)
|
|
|
14 class Color(DataClassJsonMixin):
|
|
|
15 r: float
|
|
|
16 g: float
|
|
|
17 b: float
|
|
4
|
18
|
|
8
|
19 def __post_init__(self):
|
|
|
20 super().__setattr__('r', max(0, min(1, self.r)))
|
|
|
21 super().__setattr__('g', max(0, min(1, self.g)))
|
|
|
22 super().__setattr__('b', max(0, min(1, self.b)))
|
|
|
23
|
|
|
24 def __repr__(self):
|
|
|
25 return f'(Color(r={self.r:.3f}, g={self.g:.3f}, b={self.b:.3f}))'
|
|
|
26
|
|
|
27 def avg(self):
|
|
|
28 return (self.r + self.g + self.b) / 3
|
|
4
|
29
|
|
8
|
30 def mix(self, other, x):
|
|
|
31 return Color(
|
|
|
32 lerp(self.r, other.r, x),
|
|
|
33 lerp(self.g, other.g, x),
|
|
|
34 lerp(self.b, other.b, x),
|
|
|
35 )
|
|
|
36
|
|
|
37 def hex(self):
|
|
|
38 r, g, b = int(self.r * 255), int(self.g * 255), int(self.b * 255)
|
|
|
39 return '#%02x%02x%02x' % (r, g, b)
|
|
9
|
40
|
|
|
41 @classmethod
|
|
|
42 def fromHex(cls, h: str):
|
|
|
43 return cls(
|
|
|
44 r=int(h[1:3], 16) / 255,
|
|
|
45 g=int(h[3:5], 16) / 255,
|
|
|
46 b=int(h[5:7], 16) / 255,
|
|
|
47 )
|
