aboutsummaryrefslogtreecommitdiff
path: root/lib/color-conversion/default.nix
blob: 437b51126d985381b61212e6f036d530677970de (plain)
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
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
{ lib }:

let
  hexToDecMap = {
    "0" = 0;
    "1" = 1;
    "2" = 2;
    "3" = 3;
    "4" = 4;
    "5" = 5;
    "6" = 6;
    "7" = 7;
    "8" = 8;
    "9" = 9;
    "a" = 10;
    "b" = 11;
    "c" = 12;
    "d" = 13;
    "e" = 14;
    "f" = 15;
  };

  /* Base raised to the power of the exponent.

     Type: pow :: int or float -> int -> int

     Args:
       base: The base.
       exponent: The exponent.

     Example:
       pow 0 1000
       => 0
       pow 1000 0
       => 1
       pow 2 30
       => 1073741824
       pow 3 3
       => 27
       pow (-5) 3
       => -125
  */
  pow = base: exponent:
    if exponent > 1 then
      let
        x = pow base (exponent / 2);
        odd_exp = lib.mod exponent 2 == 1;
      in
      x * x * (if odd_exp then base else 1)
    else if exponent == 1 then
      base
    else if exponent == 0 && base == 0 then
      throw "undefined"
    else if exponent == 0 then
      1
    else
      throw "undefined";

  /* Conversion from base 16 to base 10 with a exponent. Is of the form
     scalar * 16 ** exponent.

     Type: base16To10 :: int -> int -> int

     Args:
       exponent: The exponent for the base, 16.
       scalar: The value to multiple to the exponentiated base.

     Example:
       base16To10 0 11
       => 11
       base16To10 1 3
       => 48
       base16To10 2 7
       1792
       base16To10 3 14
       57344
  */
  base16To10 = exponent: scalar: scalar * pow 16 exponent;

  /* Converts a hexadecimal character to decimal.
     Only takes a string of length 1.

     Type: hexCharToDec :: string -> int

     Args:
       hex: A hexadecimal character.

     Example:
       hexCharToDec "5"
       => 5
       hexCharToDec "e"
       => 14
       hexCharToDec "A"
       => 10
  */
  hexCharToDec = hex:
    let
      lowerHex = lib.toLower hex;
    in
    if builtins.stringLength hex != 1 then
      throw "Function only accepts a single character."
    else if hexToDecMap ? ${lowerHex} then
      hexToDecMap."${lowerHex}"
    else
      throw "Character ${hex} is not a hexadecimal value.";
in rec {
  /* Converts from hexadecimal to decimal.

     Type: hexToDec :: string -> int

     Args:
       hex: A hexadecimal string.

     Example:
       hexadecimal "12"
       => 18
       hexadecimal "FF"
       => 255
       hexadecimal "abcdef"
       => 11259375
  */
  hexToDec = hex:
    let
      decimals = builtins.map hexCharToDec (lib.stringToCharacters hex);
      decimalsAscending = lib.reverseList decimals;
      decimalsPowered = lib.imap0 base16To10 decimalsAscending;
    in
    lib.foldl builtins.add 0 decimalsPowered;

  /* Converts a 6 character hexadecimal string to RGB values.

     Type: hexToRGB :: string => [int]

     Args:
       hex: A hexadecimal string of length 6.

     Example:
       hexToRGB "012345"
       => [ 1 35 69 ]
       hexToRGB "abcdef"
       => [171 205 239 ]
       hexToRGB "000FFF"
       => [ 0 15 255 ]
  */
  hexToRGB = hex:
    let
      rgbStartIndex = [ 0 2 4 ];
      hexList = builtins.map (x: builtins.substring x 2 hex) rgbStartIndex;
      hexLength = builtins.stringLength hex;
    in
    if hexLength != 6 then
      throw ''
        Unsupported hex string length of ${builtins.toString hexLength}.
        Length must be exactly 6.
      ''
    else
      builtins.map hexToDec hexList;

  /* Converts a 6 character hexadecimal string to an RGB string seperated by a
     delimiter.

     Type: hexToRGBString :: string -> string

     Args:
       sep: The delimiter or seperator.
       hex: A hexadecimal string of length 6.
  */
  hexToRGBString = sep: hex:
    let
      inherit (builtins) map toString;
      hexInRGB = hexToRGB hex;
      hexInRGBString = map toString hexInRGB;
    in
    lib.concatStringsSep sep hexInRGBString;
}