Module:Sandbox/AlexNB/nmColor

local p = {}

local redvalue = 0
local greenvalue = 0
local bluevalue = 0
local textforegroundcolor = 0
 
local function nm2RGB(wavelength)
-- calculating RGB color components
    if (wavelength >= 380) and (wavelength < 440) then
        redvalue = (440 - wavelength) / 90
        greenvalue = 0
        bluevalue = 1
    elseif (wavelength >= 440) and (wavelength < 490) then
        redvalue = 0
        greenvalue = (wavelength - 440) / 50
        bluevalue = 1
    elseif (wavelength >= 490) and (wavelength < 510) then
        redvalue = 0
        greenvalue = 1
        bluevalue = (510 - wavelength) / 20
    elseif (wavelength >= 510) and (wavelength < 580) then
        redvalue = (wavelength - 510) / 70
        greenvalue = 1
        bluevalue = 0
    elseif (wavelength >= 580) and (wavelength < 645) then
        redvalue = 1
        greenvalue = (645 - wavelength) / 65
        bluevalue = 0
    elseif (wavelength >= 645) and (wavelength < 780) then
        redvalue = 1
        greenvalue = 0
        bluevalue = 0
    end
-- calculating intensity correction factor
    if (wavelength >= 380) and (wavelength < 420) then
        intensityfactor = 0.3 + 0.7*(wavelength - 350) / 70
    elseif (wavelength >= 420) and (wavelength <= 700) then
        intensityfactor = 1
    elseif (wavelength > 700) and (wavelength <= 780) then
        intensityfactor = 0.3 + 0.7*(780 - wavelength) / 80
    else
        intensityfactor = 0
    end
    redvalue = redvalue * intensityfactor
    greenvalue = greenvalue * intensityfactor
    bluevalue = bluevalue * intensityfactor
end

local function luminance(r,g,b)
	return 0.3*r+0.59*g+0.11*b
end

function p.emission(frame)
    local wavelength = tonumber(frame.args[1])
    nm2RGB(wavelength)
    local result='#' .. string.format("%.2X%.2X%.2X", 255*redvalue, 255*greenvalue, 255*bluevalue) 
    return result
end

function p.absorption(frame)
    local wavelength = tonumber(frame.args[1])
    nm2RGB(wavelength)
    local result='#' .. string.format("%.2X%.2X%.2X", 255*(1-redvalue), 255*(1-greenvalue), 255*(1-bluevalue))
    return result
end

function p.emission_label(frame)
    local wavelength = tonumber(frame.args[1])
    nm2RGB(wavelength)
    if luminance(redvalue, greenvalue, bluevalue)>0.5 then
        local result="#000000" --|#" .. string.format("%.2X%.2X%.2X", 255*redvalue, 255*greenvalue, 255*bluevalue)
    else
    	local result="#FFFFFF" --|#" .. string.format("%.2X%.2X%.2X", 255*redvalue, 255*greenvalue, 255*bluevalue)
    end
    return result
end

function p.absorption_label(frame)
    local wavelength = tonumber(frame.args[1])
    nm2RGB(wavelength)
    if luminance(1-redvalue, 1-greenvalue, 1-bluevalue)>0.5 then
        local result="#000000" --|#" .. string.format("%.2X%.2X%.2X", 255*(1-redvalue), 255*(1-greenvalue), 255*(1-bluevalue))
    else
    	local result="#FFFFFF" --|#" .. string.format("%.2X%.2X%.2X", 255*(1-redvalue), 255*(1-greenvalue), 255*(1-bluevalue))
    end
    return result
end

return p