perception.nvim/lua/perception/builtins.lua at master · moyiz/perception.nvim · 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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
local normalize = require("perception.utils").normalize
local truncate = require("perception.utils").truncate

local M = {}

---@param color perception.Color
---@param value number
---@return perception.Color
M.dim = function(color, value)
  local factor = (value / 255) + 1
  return color:apply_rgb(function(c)
    return truncate(c * factor)
  end)
end

---@param color perception.Color
---@param value number
---@return perception.Color
M.brightness = function(color, value)
  return color:apply_rgb(function(c)
    return truncate(c + value)
  end)
end

---@param color perception.Color
---@param value number
---@return perception.Color
M.contrast = function(color, value)
  local contrast = (259 * (255 + value)) / (255 * (259 - value))
  return color:apply_rgb(function(c)
    return truncate(contrast * (c - 128) + 128)
  end)
end

---@param color perception.Color
---@param value number
---@return perception.Color
M.red = function(color, value)
  local r, g, b = unpack(color:as_rgb())
  return color:from_rgb(truncate(r + value, 255), g, b)
end

---@param color perception.Color
---@param value number
---@return perception.Color
M.green = function(color, value)
  local r, g, b = unpack(color:as_rgb())
  return color:from_rgb(r, truncate(g + value, 255), b)
end

---@param color perception.Color
---@param value number
---@return perception.Color
M.blue = function(color, value)
  local r, g, b = unpack(color:as_rgb())
  return color:from_rgb(r, g, truncate(b + value, 255))
end

---@param color perception.Color
---@param value number
---@return perception.Color
M.hue = function(color, value)
  local h, s, l = unpack(color:as_hsl())
  return color:from_hsl(truncate(h * (value / 360 + 1), 360), s, l)
end

---@param color perception.Color
---@param value number
---@return perception.Color
M.hsl_saturation = function(color, value)
  local h, s, l = unpack(color:as_hsl())
  return color:from_hsl(h, truncate(s * (value / 100 + 1), 100), l)
end

---@param color perception.Color
---@param value number
---@return perception.Color
M.hsv_saturation = function(color, value)
  local h, s, v = unpack(color:as_hsv())
  return color:from_hsv(h, truncate(s * (value / 100 + 1), 100), v)
end

---@param color perception.Color
---@param value number
---@return perception.Color
M.luminance = function(color, value)
  local h, s, l = unpack(color:as_hsl())
  return color:from_hsl(h, s, truncate(l * (value / 100 + 1), 100))
end

---@param color perception.Color
---@param value number
---@return perception.Color
M.value = function(color, value)
  local h, s, v = unpack(color:as_hsv())
  return color:from_hsv(h, s, truncate(v * (value / 100 + 1), 100))
end

---@param color perception.Color
---@param value number
---@return perception.Color
M.temperature = function(color, value)
  local r, g, b = unpack(color:as_rgb())
  return color:from_rgb(truncate(r + value, 255), g, truncate(b - value))
end

M.tint = M.green

-- ---@param color perception.Color
-- ---@param value number
-- M.invert = function(color, value)
--   return color:apply_rgb(function(c)
--     return value % 2 == 0 and c or 255 - c
--   end)
-- end

---@param color perception.Color
---@param value number
---@return perception.Color
M.inversion = function(color, value)
  return color:apply_rgb(function(c)
    return normalize(value, 0, 100, c, 255 - c)
  end)
end

---@param color perception.Color
---@param value number
---@return perception.Color
M.grayscale = function(color, value)
  if value % 2 == 0 then
    return color
  end
  local r, g, b = unpack(color:as_rgb())
  local gray = math.floor((r + g + b) / 3)
  return color:from_rgb(gray, gray, gray)
end

---@param color perception.Color
---@param value number
---@return perception.Color
M.sepia = function(color, value)
  if value ~= 0 then
    color = M.grayscale(color, 1)
    local g = color:as_rgb()[1]
    return color:from_rgb(g + (value * 2), g + value, g)
  end
  return color
end

---@param color perception.Color
---@param value number
---@return perception.Color
M.sepia_intensity = function(color, value)
  return M.blue(color, -value)
end

return M