Mbyja-Wind-Project/temp_rgb.py at main · SebasMP14/Mbyja-Wind-Project · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
# Note:
# Red values below 6600 K are always 255.
# Blue values below 2000 K are always 0.
# Blue values above 6500 K are always 255.
# Using the formulas, the squared error is approx. 0.98.
import math
from matplotlib import pyplot as plt
from class_datos import *


def conversion_temp_rgb(temp_k):
    ### 1. Algorithm scope: Between 1000 and 40000 (is accurate).
    norm_temp = temp_k / 100.0 # Normalize data.

    if norm_temp <= 66:
        red = 255
    else:
        temp_red = 329.698727446 * math.pow(norm_temp - 60, -0.1332047592) # From Vendian.
        if temp_red < 0:
            red = 0
        elif temp_red > 255:
            red = 255
        else:
            red = temp_red

    if norm_temp <= 66: # For purposes of fitting a curve to the data (squared error).
        temp_green = 99.4708025861 * math.log(norm_temp) - 161.1195681661 # From Vendian.
        if temp_green < 0:
            green = 0
        elif temp_green > 255:
            green = 255
        else:
            green = temp_green
    else:
        temp_green = 288.1221695283 * math.pow(norm_temp - 60, -0.0755148492) # From Vendian.
        if temp_green < 0:
            green = 0
        elif temp_green > 255:
            green = 255
        else:
            green = temp_green

    if norm_temp >= 66:
        blue = 255
    elif norm_temp <= 19:
        blue = 0
    else:
        temp_blue = 138.5177312231 * math.log(norm_temp - 10) - 305.0447927307 # From Vendian.
        if temp_blue < 0:
            blue = 0
        elif temp_blue > 255:
            blue = 255
        else:
            blue = temp_blue

    return red, green, blue