Annotation of src/usr.bin/tmux/colour.c, Revision 1.18
1.18 ! nicm 1: /* $OpenBSD: colour.c,v 1.17 2019/03/13 21:39:21 nicm Exp $ */
1.1 nicm 2:
3: /*
1.12 nicm 4: * Copyright (c) 2008 Nicholas Marriott <nicholas.marriott@gmail.com>
1.13 nicm 5: * Copyright (c) 2016 Avi Halachmi <avihpit@yahoo.com>
1.1 nicm 6: *
7: * Permission to use, copy, modify, and distribute this software for any
8: * purpose with or without fee is hereby granted, provided that the above
9: * copyright notice and this permission notice appear in all copies.
10: *
11: * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
12: * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13: * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
14: * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15: * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
16: * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
17: * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18: */
19:
20: #include <sys/types.h>
21:
1.3 nicm 22: #include <ctype.h>
1.2 nicm 23: #include <stdlib.h>
1.1 nicm 24: #include <string.h>
25:
26: #include "tmux.h"
27:
1.13 nicm 28: static int
29: colour_dist_sq(int R, int G, int B, int r, int g, int b)
30: {
31: return ((R - r) * (R - r) + (G - g) * (G - g) + (B - b) * (B - b));
32: }
1.3 nicm 33:
1.13 nicm 34: static int
35: colour_to_6cube(int v)
1.3 nicm 36: {
1.13 nicm 37: if (v < 48)
38: return (0);
39: if (v < 114)
1.9 nicm 40: return (1);
1.13 nicm 41: return ((v - 35) / 40);
1.3 nicm 42: }
43:
1.13 nicm 44: /*
45: * Convert an RGB triplet to the xterm(1) 256 colour palette.
46: *
47: * xterm provides a 6x6x6 colour cube (16 - 231) and 24 greys (232 - 255). We
48: * map our RGB colour to the closest in the cube, also work out the closest
49: * grey, and use the nearest of the two.
50: *
1.15 nicm 51: * Note that the xterm has much lower resolution for darker colours (they are
1.13 nicm 52: * not evenly spread out), so our 6 levels are not evenly spread: 0x0, 0x5f
53: * (95), 0x87 (135), 0xaf (175), 0xd7 (215) and 0xff (255). Greys are more
54: * evenly spread (8, 18, 28 ... 238).
55: */
1.3 nicm 56: int
1.10 nicm 57: colour_find_rgb(u_char r, u_char g, u_char b)
1.3 nicm 58: {
1.13 nicm 59: static const int q2c[6] = { 0x00, 0x5f, 0x87, 0xaf, 0xd7, 0xff };
60: int qr, qg, qb, cr, cg, cb, d, idx;
61: int grey_avg, grey_idx, grey;
62:
63: /* Map RGB to 6x6x6 cube. */
64: qr = colour_to_6cube(r); cr = q2c[qr];
65: qg = colour_to_6cube(g); cg = q2c[qg];
66: qb = colour_to_6cube(b); cb = q2c[qb];
67:
68: /* If we have hit the colour exactly, return early. */
69: if (cr == r && cg == g && cb == b)
1.14 nicm 70: return ((16 + (36 * qr) + (6 * qg) + qb) | COLOUR_FLAG_256);
1.13 nicm 71:
72: /* Work out the closest grey (average of RGB). */
73: grey_avg = (r + g + b) / 3;
74: if (grey_avg > 238)
75: grey_idx = 23;
76: else
77: grey_idx = (grey_avg - 3) / 10;
78: grey = 8 + (10 * grey_idx);
79:
80: /* Is grey or 6x6x6 colour closest? */
81: d = colour_dist_sq(cr, cg, cb, r, g, b);
82: if (colour_dist_sq(grey, grey, grey, r, g, b) < d)
83: idx = 232 + grey_idx;
84: else
85: idx = 16 + (36 * qr) + (6 * qg) + qb;
1.14 nicm 86: return (idx | COLOUR_FLAG_256);
1.3 nicm 87: }
88:
1.14 nicm 89: /* Join RGB into a colour. */
90: int
91: colour_join_rgb(u_char r, u_char g, u_char b)
1.2 nicm 92: {
1.14 nicm 93: return ((((int)((r) & 0xff)) << 16) |
94: (((int)((g) & 0xff)) << 8) |
95: (((int)((b) & 0xff))) | COLOUR_FLAG_RGB);
1.2 nicm 96: }
97:
1.14 nicm 98: /* Split colour into RGB. */
1.2 nicm 99: void
1.14 nicm 100: colour_split_rgb(int c, u_char *r, u_char *g, u_char *b)
1.2 nicm 101: {
1.14 nicm 102: *r = (c >> 16) & 0xff;
103: *g = (c >> 8) & 0xff;
104: *b = c & 0xff;
1.2 nicm 105: }
106:
1.3 nicm 107: /* Convert colour to a string. */
1.1 nicm 108: const char *
1.2 nicm 109: colour_tostring(int c)
1.1 nicm 110: {
1.2 nicm 111: static char s[32];
1.14 nicm 112: u_char r, g, b;
113:
114: if (c & COLOUR_FLAG_RGB) {
115: colour_split_rgb(c, &r, &g, &b);
116: xsnprintf(s, sizeof s, "#%02x%02x%02x", r, g, b);
117: return (s);
118: }
1.2 nicm 119:
1.14 nicm 120: if (c & COLOUR_FLAG_256) {
121: xsnprintf(s, sizeof s, "colour%u", c & 0xff);
1.2 nicm 122: return (s);
123: }
124:
1.1 nicm 125: switch (c) {
126: case 0:
127: return ("black");
128: case 1:
129: return ("red");
130: case 2:
131: return ("green");
132: case 3:
133: return ("yellow");
134: case 4:
135: return ("blue");
136: case 5:
137: return ("magenta");
138: case 6:
139: return ("cyan");
140: case 7:
141: return ("white");
142: case 8:
143: return ("default");
1.16 nicm 144: case 9:
145: return ("terminal");
1.5 nicm 146: case 90:
147: return ("brightblack");
148: case 91:
149: return ("brightred");
150: case 92:
151: return ("brightgreen");
152: case 93:
153: return ("brightyellow");
154: case 94:
155: return ("brightblue");
156: case 95:
157: return ("brightmagenta");
158: case 96:
159: return ("brightcyan");
160: case 97:
161: return ("brightwhite");
1.1 nicm 162: }
1.17 nicm 163: return ("invalid");
1.1 nicm 164: }
165:
1.3 nicm 166: /* Convert colour from string. */
1.1 nicm 167: int
168: colour_fromstring(const char *s)
169: {
1.10 nicm 170: const char *errstr;
171: const char *cp;
172: int n;
173: u_char r, g, b;
1.3 nicm 174:
175: if (*s == '#' && strlen(s) == 7) {
176: for (cp = s + 1; isxdigit((u_char) *cp); cp++)
177: ;
178: if (*cp != '\0')
179: return (-1);
1.10 nicm 180: n = sscanf(s + 1, "%2hhx%2hhx%2hhx", &r, &g, &b);
1.3 nicm 181: if (n != 3)
182: return (-1);
1.14 nicm 183: return (colour_join_rgb(r, g, b));
1.3 nicm 184: }
1.2 nicm 185:
186: if (strncasecmp(s, "colour", (sizeof "colour") - 1) == 0) {
187: n = strtonum(s + (sizeof "colour") - 1, 0, 255, &errstr);
188: if (errstr != NULL)
189: return (-1);
1.14 nicm 190: return (n | COLOUR_FLAG_256);
1.2 nicm 191: }
192:
1.16 nicm 193: if (strcasecmp(s, "default") == 0)
194: return (8);
195: if (strcasecmp(s, "terminal") == 0)
196: return (9);
197:
1.9 nicm 198: if (strcasecmp(s, "black") == 0 || strcmp(s, "0") == 0)
1.1 nicm 199: return (0);
1.9 nicm 200: if (strcasecmp(s, "red") == 0 || strcmp(s, "1") == 0)
1.1 nicm 201: return (1);
1.9 nicm 202: if (strcasecmp(s, "green") == 0 || strcmp(s, "2") == 0)
1.1 nicm 203: return (2);
1.9 nicm 204: if (strcasecmp(s, "yellow") == 0 || strcmp(s, "3") == 0)
1.1 nicm 205: return (3);
1.9 nicm 206: if (strcasecmp(s, "blue") == 0 || strcmp(s, "4") == 0)
1.1 nicm 207: return (4);
1.9 nicm 208: if (strcasecmp(s, "magenta") == 0 || strcmp(s, "5") == 0)
1.1 nicm 209: return (5);
1.9 nicm 210: if (strcasecmp(s, "cyan") == 0 || strcmp(s, "6") == 0)
1.1 nicm 211: return (6);
1.9 nicm 212: if (strcasecmp(s, "white") == 0 || strcmp(s, "7") == 0)
1.1 nicm 213: return (7);
1.9 nicm 214: if (strcasecmp(s, "brightblack") == 0 || strcmp(s, "90") == 0)
1.5 nicm 215: return (90);
1.9 nicm 216: if (strcasecmp(s, "brightred") == 0 || strcmp(s, "91") == 0)
1.5 nicm 217: return (91);
1.9 nicm 218: if (strcasecmp(s, "brightgreen") == 0 || strcmp(s, "92") == 0)
1.5 nicm 219: return (92);
1.9 nicm 220: if (strcasecmp(s, "brightyellow") == 0 || strcmp(s, "93") == 0)
1.5 nicm 221: return (93);
1.9 nicm 222: if (strcasecmp(s, "brightblue") == 0 || strcmp(s, "94") == 0)
1.5 nicm 223: return (94);
1.9 nicm 224: if (strcasecmp(s, "brightmagenta") == 0 || strcmp(s, "95") == 0)
1.5 nicm 225: return (95);
1.9 nicm 226: if (strcasecmp(s, "brightcyan") == 0 || strcmp(s, "96") == 0)
1.5 nicm 227: return (96);
1.9 nicm 228: if (strcasecmp(s, "brightwhite") == 0 || strcmp(s, "97") == 0)
1.5 nicm 229: return (97);
1.1 nicm 230: return (-1);
231: }
232:
1.18 ! nicm 233: /* Convert 256 colour to RGB colour. */
! 234: int
! 235: colour_256toRGB(int c)
! 236: {
! 237: static const int table[256] = {
! 238: 0x000000, 0x800000, 0x008000, 0x808000,
! 239: 0x000080, 0x800080, 0x008080, 0xc0c0c0,
! 240: 0x808080, 0xff0000, 0x00ff00, 0xffff00,
! 241: 0x0000ff, 0xff00ff, 0x00ffff, 0xffffff,
! 242: 0x000000, 0x00005f, 0x000087, 0x0000af,
! 243: 0x0000d7, 0x0000ff, 0x005f00, 0x005f5f,
! 244: 0x005f87, 0x005faf, 0x005fd7, 0x005fff,
! 245: 0x008700, 0x00875f, 0x008787, 0x0087af,
! 246: 0x0087d7, 0x0087ff, 0x00af00, 0x00af5f,
! 247: 0x00af87, 0x00afaf, 0x00afd7, 0x00afff,
! 248: 0x00d700, 0x00d75f, 0x00d787, 0x00d7af,
! 249: 0x00d7d7, 0x00d7ff, 0x00ff00, 0x00ff5f,
! 250: 0x00ff87, 0x00ffaf, 0x00ffd7, 0x00ffff,
! 251: 0x5f0000, 0x5f005f, 0x5f0087, 0x5f00af,
! 252: 0x5f00d7, 0x5f00ff, 0x5f5f00, 0x5f5f5f,
! 253: 0x5f5f87, 0x5f5faf, 0x5f5fd7, 0x5f5fff,
! 254: 0x5f8700, 0x5f875f, 0x5f8787, 0x5f87af,
! 255: 0x5f87d7, 0x5f87ff, 0x5faf00, 0x5faf5f,
! 256: 0x5faf87, 0x5fafaf, 0x5fafd7, 0x5fafff,
! 257: 0x5fd700, 0x5fd75f, 0x5fd787, 0x5fd7af,
! 258: 0x5fd7d7, 0x5fd7ff, 0x5fff00, 0x5fff5f,
! 259: 0x5fff87, 0x5fffaf, 0x5fffd7, 0x5fffff,
! 260: 0x870000, 0x87005f, 0x870087, 0x8700af,
! 261: 0x8700d7, 0x8700ff, 0x875f00, 0x875f5f,
! 262: 0x875f87, 0x875faf, 0x875fd7, 0x875fff,
! 263: 0x878700, 0x87875f, 0x878787, 0x8787af,
! 264: 0x8787d7, 0x8787ff, 0x87af00, 0x87af5f,
! 265: 0x87af87, 0x87afaf, 0x87afd7, 0x87afff,
! 266: 0x87d700, 0x87d75f, 0x87d787, 0x87d7af,
! 267: 0x87d7d7, 0x87d7ff, 0x87ff00, 0x87ff5f,
! 268: 0x87ff87, 0x87ffaf, 0x87ffd7, 0x87ffff,
! 269: 0xaf0000, 0xaf005f, 0xaf0087, 0xaf00af,
! 270: 0xaf00d7, 0xaf00ff, 0xaf5f00, 0xaf5f5f,
! 271: 0xaf5f87, 0xaf5faf, 0xaf5fd7, 0xaf5fff,
! 272: 0xaf8700, 0xaf875f, 0xaf8787, 0xaf87af,
! 273: 0xaf87d7, 0xaf87ff, 0xafaf00, 0xafaf5f,
! 274: 0xafaf87, 0xafafaf, 0xafafd7, 0xafafff,
! 275: 0xafd700, 0xafd75f, 0xafd787, 0xafd7af,
! 276: 0xafd7d7, 0xafd7ff, 0xafff00, 0xafff5f,
! 277: 0xafff87, 0xafffaf, 0xafffd7, 0xafffff,
! 278: 0xd70000, 0xd7005f, 0xd70087, 0xd700af,
! 279: 0xd700d7, 0xd700ff, 0xd75f00, 0xd75f5f,
! 280: 0xd75f87, 0xd75faf, 0xd75fd7, 0xd75fff,
! 281: 0xd78700, 0xd7875f, 0xd78787, 0xd787af,
! 282: 0xd787d7, 0xd787ff, 0xd7af00, 0xd7af5f,
! 283: 0xd7af87, 0xd7afaf, 0xd7afd7, 0xd7afff,
! 284: 0xd7d700, 0xd7d75f, 0xd7d787, 0xd7d7af,
! 285: 0xd7d7d7, 0xd7d7ff, 0xd7ff00, 0xd7ff5f,
! 286: 0xd7ff87, 0xd7ffaf, 0xd7ffd7, 0xd7ffff,
! 287: 0xff0000, 0xff005f, 0xff0087, 0xff00af,
! 288: 0xff00d7, 0xff00ff, 0xff5f00, 0xff5f5f,
! 289: 0xff5f87, 0xff5faf, 0xff5fd7, 0xff5fff,
! 290: 0xff8700, 0xff875f, 0xff8787, 0xff87af,
! 291: 0xff87d7, 0xff87ff, 0xffaf00, 0xffaf5f,
! 292: 0xffaf87, 0xffafaf, 0xffafd7, 0xffafff,
! 293: 0xffd700, 0xffd75f, 0xffd787, 0xffd7af,
! 294: 0xffd7d7, 0xffd7ff, 0xffff00, 0xffff5f,
! 295: 0xffff87, 0xffffaf, 0xffffd7, 0xffffff,
! 296: 0x080808, 0x121212, 0x1c1c1c, 0x262626,
! 297: 0x303030, 0x3a3a3a, 0x444444, 0x4e4e4e,
! 298: 0x585858, 0x626262, 0x6c6c6c, 0x767676,
! 299: 0x808080, 0x8a8a8a, 0x949494, 0x9e9e9e,
! 300: 0xa8a8a8, 0xb2b2b2, 0xbcbcbc, 0xc6c6c6,
! 301: 0xd0d0d0, 0xdadada, 0xe4e4e4, 0xeeeeee
! 302: };
! 303:
! 304: return (table[c & 0xff] | COLOUR_FLAG_RGB);
! 305: }
! 306:
! 307: /* Convert 256 colour to 16 colour. */
! 308: int
! 309: colour_256to16(int c)
1.1 nicm 310: {
1.18 ! nicm 311: static const char table[256] = {
1.1 nicm 312: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
313: 0, 4, 4, 4, 12, 12, 2, 6, 4, 4, 12, 12, 2, 2, 6, 4,
314: 12, 12, 2, 2, 2, 6, 12, 12, 10, 10, 10, 10, 14, 12, 10, 10,
315: 10, 10, 10, 14, 1, 5, 4, 4, 12, 12, 3, 8, 4, 4, 12, 12,
316: 2, 2, 6, 4, 12, 12, 2, 2, 2, 6, 12, 12, 10, 10, 10, 10,
317: 14, 12, 10, 10, 10, 10, 10, 14, 1, 1, 5, 4, 12, 12, 1, 1,
318: 5, 4, 12, 12, 3, 3, 8, 4, 12, 12, 2, 2, 2, 6, 12, 12,
319: 10, 10, 10, 10, 14, 12, 10, 10, 10, 10, 10, 14, 1, 1, 1, 5,
320: 12, 12, 1, 1, 1, 5, 12, 12, 1, 1, 1, 5, 12, 12, 3, 3,
321: 3, 7, 12, 12, 10, 10, 10, 10, 14, 12, 10, 10, 10, 10, 10, 14,
322: 9, 9, 9, 9, 13, 12, 9, 9, 9, 9, 13, 12, 9, 9, 9, 9,
323: 13, 12, 9, 9, 9, 9, 13, 12, 11, 11, 11, 11, 7, 12, 10, 10,
324: 10, 10, 10, 14, 9, 9, 9, 9, 9, 13, 9, 9, 9, 9, 9, 13,
325: 9, 9, 9, 9, 9, 13, 9, 9, 9, 9, 9, 13, 9, 9, 9, 9,
326: 9, 13, 11, 11, 11, 11, 11, 15, 0, 0, 0, 0, 0, 0, 8, 8,
327: 8, 8, 8, 8, 7, 7, 7, 7, 7, 7, 15, 15, 15, 15, 15, 15
328: };
329:
1.18 ! nicm 330: return (table[c & 0xff]);
1.1 nicm 331: }