42 lines
893 B
Python
42 lines
893 B
Python
from PIL import Image
|
|
|
|
kelvin_table = {
|
|
1000: (255, 56, 0),
|
|
1500: (255, 109, 0),
|
|
2000: (255, 137, 18),
|
|
2500: (255, 161, 72),
|
|
3000: (255, 180, 107),
|
|
3500: (255, 196, 137),
|
|
4000: (255, 209, 163),
|
|
4500: (255, 219, 186),
|
|
5000: (255, 228, 206),
|
|
5500: (255, 236, 224),
|
|
6000: (255, 243, 239),
|
|
6500: (255, 249, 253),
|
|
7000: (245, 243, 255),
|
|
7500: (235, 238, 255),
|
|
8000: (227, 233, 255),
|
|
8500: (220, 229, 255),
|
|
9000: (214, 225, 255),
|
|
9500: (208, 222, 255),
|
|
10000: (204, 219, 255),
|
|
}
|
|
|
|
|
|
def change_temperature(image: Image, temperature: float) -> Image:
|
|
r, g, b = kelvin_table[temperature]
|
|
matrix = (
|
|
r / 255.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
g / 255.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
0.0,
|
|
b / 255.0,
|
|
0.0,
|
|
)
|
|
return image.convert("RGB", matrix)
|