Rewrite scheme in a more sensible order

1 files changed, 100 insertions, 106 deletions

diff --git a/bin/scheme.c b/bin/scheme.c
index fd1d8286..273dd2dc 100644
--- a/bin/scheme.c
+++ b/bin/scheme.c
@@ -25,9 +25,80 @@
 #include <unistd.h>
 #include <zlib.h>
 
-struct Hsv { double h, s, v; };
-struct Rgb { uint8_t r, g, b; };
+static const struct Hsv { double h, s, v; }
+    R = {   0.0, 1.0, 1.0 },
+    Y = {  60.0, 1.0, 1.0 },
+    G = { 120.0, 1.0, 1.0 },
+    C = { 180.0, 1.0, 1.0 },
+    B = { 240.0, 1.0, 1.0 },
+    M = { 300.0, 1.0, 1.0 };
+
+static struct Hsv x(struct Hsv o, double hd, double sf, double vf) {
+    return (struct Hsv) { o.h + hd, o.s * sf, o.v * vf };
+}
 
+struct Ansi {
+    enum { BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE };
+    struct Hsv dark[8];
+    struct Hsv light[8];
+};
+
+struct Terminal {
+    struct Ansi ansi;
+    struct Hsv background, text, bold, selection, cursor;
+};
+
+#define HSV_LEN(x) (sizeof(x) / sizeof(struct Hsv))
+struct Scheme {
+    size_t len;
+    union {
+        struct Hsv hsv;
+        struct Ansi ansi;
+        struct Terminal terminal;
+    };
+};
+
+static struct Scheme ansi(void) {
+    struct Ansi a = {
+        .light = {
+            [BLACK]   = x(R, +45.0, 0.3, 0.3),
+            [RED]     = x(R, +10.0, 0.9, 0.8),
+            [GREEN]   = x(G, -55.0, 0.8, 0.6),
+            [YELLOW]  = x(Y, -20.0, 0.8, 0.8),
+            [BLUE]    = x(B, -55.0, 0.4, 0.5),
+            [MAGENTA] = x(M, +45.0, 0.4, 0.6),
+            [CYAN]    = x(C, -60.0, 0.3, 0.6),
+            [WHITE]   = x(R, +45.0, 0.3, 0.8),
+        },
+    };
+    a.dark[BLACK] = x(a.light[BLACK], 0.0, 1.0, 0.3);
+    a.dark[WHITE] = x(a.light[WHITE], 0.0, 1.0, 0.6);
+    for (int i = RED; i < WHITE; ++i) {
+        a.dark[i] = x(a.light[i], 0.0, 1.0, 0.8);
+    }
+    return (struct Scheme) { .len = HSV_LEN(a), .ansi = a };
+}
+
+static struct Scheme terminal(void) {
+    struct Ansi a = ansi().ansi;
+    struct Terminal t = {
+        .ansi       = a,
+        .background = x(a.dark[BLACK],    0.0, 1.0, 0.9),
+        .text       = x(a.light[WHITE],   0.0, 1.0, 0.9),
+        .bold       = x(a.light[WHITE],   0.0, 1.0, 1.0),
+        .selection  = x(a.light[RED],   +10.0, 1.0, 0.8),
+        .cursor     = x(a.dark[WHITE],    0.0, 1.0, 0.8),
+    };
+    return (struct Scheme) { .len = HSV_LEN(t), .terminal = t };
+}
+
+static void hsv(const struct Hsv *hsv, size_t len) {
+    for (size_t i = 0; i < len; ++i) {
+        printf("%g,%g,%g\n", hsv[i].h, hsv[i].s, hsv[i].v);
+    }
+}
+
+struct Rgb { uint8_t r, g, b; };
 static struct Rgb toRgb(struct Hsv hsv) {
     double c = hsv.v * hsv.s;
     double h = hsv.h / 60.0;
@@ -43,6 +114,21 @@ static struct Rgb toRgb(struct Hsv hsv) {
     return (struct Rgb) { r * 255.0, g * 255.0, b * 255.0 };
 }
 
+static void hex(const struct Hsv *hsv, size_t len) {
+    for (size_t i = 0; i < len; ++i) {
+        struct Rgb rgb = toRgb(hsv[i]);
+        printf("%02x%02x%02x\n", rgb.r, rgb.g, rgb.b);
+    }
+}
+
+static void linux(const struct Hsv *hsv, size_t len) {
+    if (len > 16) len = 16;
+    for (size_t i = 0; i < len; ++i) {
+        struct Rgb rgb = toRgb(hsv[i]);
+        printf("\x1B]P%zx%02x%02x%02x", i, rgb.r, rgb.g, rgb.b);
+    }
+}
+
 static uint32_t crc;
 static void pngWrite(const void *ptr, size_t size) {
     fwrite(ptr, size, 1, stdout);
@@ -58,9 +144,9 @@ static void pngChunk(const char *type, uint32_t size) {
     crc = crc32(0, Z_NULL, 0);
     pngWrite(type, 4);
 }
-enum { NONE, SUB, UP, AVERAGE, PAETH };
 
-static void outPng(const struct Hsv *scheme, uint32_t len) {
+static void png(const struct Hsv *hsv, size_t len) {
+    if (len > 256) len = 256;
     uint32_t swatchWidth = 64;
     uint32_t swatchHeight = 64;
     uint32_t columns = 8;
@@ -78,7 +164,7 @@ static void outPng(const struct Hsv *scheme, uint32_t len) {
 
     pngChunk("PLTE", 3 * len);
     for (uint32_t i = 0; i < len; ++i) {
-        struct Rgb rgb = toRgb(scheme[i]);
+        struct Rgb rgb = toRgb(hsv[i]);
         pngWrite(&rgb, 3);
     }
     pngInt(crc);
@@ -86,6 +172,7 @@ static void outPng(const struct Hsv *scheme, uint32_t len) {
     uint8_t data[height][1 + width];
     memset(data, 0, sizeof(data));
     for (uint32_t y = 0; y < height; ++y) {
+        enum { NONE, SUB, UP, AVERAGE, PAETH };
         data[y][0] = (y % swatchHeight) ? UP : SUB;
     }
     for (uint32_t i = 0; i < len; ++i) {
@@ -107,115 +194,22 @@ static void outPng(const struct Hsv *scheme, uint32_t len) {
     pngInt(crc);
 }
 
-static void outHsv(const struct Hsv *scheme, uint32_t len) {
-    for (uint32_t i = 0; i < len; ++i) {
-        printf("%g,%g,%g\n", scheme[i].h, scheme[i].s, scheme[i].v);
-    }
-}
-
-static void outHex(const struct Hsv *scheme, uint32_t len) {
-    for (uint32_t i = 0; i < len; ++i) {
-        struct Rgb rgb = toRgb(scheme[i]);
-        printf("%02x%02x%02x\n", rgb.r, rgb.g, rgb.b);
-    }
-}
-
-static void outLinux(const struct Hsv *scheme, uint32_t len) {
-    if (len > 16) len = 16;
-    for (uint32_t i = 0; i < len; ++i) {
-        struct Rgb rgb = toRgb(scheme[i]);
-        printf("\x1B]P%x%02x%02x%02x", i, rgb.r, rgb.g, rgb.b);
-    }
-}
-
-static const struct Hsv R = {   0.0, 1.0, 1.0 };
-static const struct Hsv Y = {  60.0, 1.0, 1.0 };
-static const struct Hsv G = { 120.0, 1.0, 1.0 };
-static const struct Hsv C = { 180.0, 1.0, 1.0 };
-static const struct Hsv B = { 240.0, 1.0, 1.0 };
-static const struct Hsv M = { 300.0, 1.0, 1.0 };
-
-static struct Hsv p(struct Hsv o, double hd, double sf, double vf) {
-    return (struct Hsv) { o.h + hd, o.s * sf, o.v * vf };
-}
-
-struct Ansi {
-    enum { BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE };
-    struct Hsv dark[8];
-    struct Hsv light[8];
-};
-
-struct Terminal {
-    struct Ansi ansi;
-    struct Hsv background, text, bold, selection, cursor;
-};
-
-#define HSV_LEN(x) (sizeof(x) / sizeof(struct Hsv))
-struct Scheme {
-    uint32_t len;
-    union {
-        struct Hsv hsv;
-        struct Ansi ansi;
-        struct Terminal terminal;
-    };
-};
-
-static struct Scheme genAnsi(void) {
-    struct Ansi ansi = {
-        .light = {
-            [BLACK]   = p(R, +45.0, 0.3, 0.3),
-            [RED]     = p(R, +10.0, 0.9, 0.8),
-            [GREEN]   = p(G, -55.0, 0.8, 0.6),
-            [YELLOW]  = p(Y, -20.0, 0.8, 0.8),
-            [BLUE]    = p(B, -55.0, 0.4, 0.5),
-            [MAGENTA] = p(M, +45.0, 0.4, 0.6),
-            [CYAN]    = p(C, -60.0, 0.3, 0.6),
-            [WHITE]   = p(R, +45.0, 0.3, 0.8),
-        },
-    };
-    ansi.dark[BLACK] = p(ansi.light[BLACK], 0.0, 1.0, 0.3);
-    ansi.dark[WHITE] = p(ansi.light[WHITE], 0.0, 1.0, 0.6);
-    for (int i = RED; i < WHITE; ++i) {
-        ansi.dark[i] = p(ansi.light[i], 0.0, 1.0, 0.8);
-    }
-    return (struct Scheme) { .len = HSV_LEN(ansi), .ansi = ansi };
-}
-
-static struct Scheme genTerminal(void) {
-    struct Ansi ansi = genAnsi().ansi;
-    struct Terminal terminal = {
-        .ansi       = ansi,
-        .background = p(ansi.dark[BLACK],    0.0, 1.0, 0.9),
-        .text       = p(ansi.light[WHITE],   0.0, 1.0, 0.9),
-        .bold       = p(ansi.light[WHITE],   0.0, 1.0, 1.0),
-        .selection  = p(ansi.light[RED],   +10.0, 1.0, 0.8),
-        .cursor     = p(ansi.dark[WHITE],    0.0, 1.0, 0.8),
-    };
-    return (struct Scheme) {
-        .len = HSV_LEN(terminal),
-        .terminal = terminal,
-    };
-}
-
 int main(int argc, char *argv[]) {
-    struct Scheme (*gen)(void) = genAnsi;
-    void (*out)(const struct Hsv *, uint32_t len) = outHex;
-
+    struct Scheme (*gen)(void) = ansi;
+    void (*out)(const struct Hsv *, size_t len) = hex;
     int opt;
     while (0 < (opt = getopt(argc, argv, "aghltx"))) {
         switch (opt) {
-            case 'a': gen = genAnsi; break;
-            case 'g': out = outPng; break;
-            case 'h': out = outHsv; break;
-            case 'l': out = outLinux; break;
-            case 't': gen = genTerminal; break;
-            case 'x': out = outHex; break;
+            case 'a': gen = ansi; break;
+            case 'g': out = png; break;
+            case 'h': out = hsv; break;
+            case 'l': out = linux; break;
+            case 't': gen = terminal; break;
+            case 'x': out = hex; break;
             default: return EX_USAGE;
         }
     }
-
     struct Scheme scheme = gen();
     out(&scheme.hsv, scheme.len);
-
     return EX_OK;
 }