options.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <getopt.h>

#include "ascii.h"
#include "options.h"


// Default weights; must add up to 1.0
static const float weight_red   = 0.2989f;
static const float weight_green = 0.5866f;
static const float weight_blue  = 0.1145f;

unsigned short int opt_width   = 80,
                   opt_height  = 0,

                   auto_width  = 0,
                   auto_height = 1;

char opt_address[OPTIONS_BUFF_SIZE] = "127.0.0.1",
     opt_port   [OPTIONS_BUFF_SIZE] = "9001";

char ascii_palette[ASCII_PALETTE_SIZE + 1] =
    "   ...',;:clodxkO0KXNWM";

unsigned short int RED  [ASCII_PALETTE_SIZE],
                   GREEN[ASCII_PALETTE_SIZE],
                   BLUE [ASCII_PALETTE_SIZE],
                   GRAY [ASCII_PALETTE_SIZE];

static struct option long_options[] = {
    {"address", optional_argument, NULL, 'a'},
    {"port",    optional_argument, NULL, 'p'},
    {"width",   optional_argument, NULL, 'w'},
    {"height",  optional_argument, NULL, 'h'},
    {0, 0, 0, 0}
};


void options_init(int argc, char** argv) {
    precalc_rgb(weight_red, weight_green, weight_blue);
    while (1) {
        int index  = 0,
            option = getopt_long(argc, argv, "p:a:w:h:", long_options, &index);
        if (option == -1)
            break;

        char argbuf[1024];
        switch (option) {
            case 0:
                break;

            case 'a':
                snprintf(opt_address, OPTIONS_BUFF_SIZE, "%s", optarg);
                break;

            case 'p':
                snprintf(opt_port, OPTIONS_BUFF_SIZE, "%s", optarg);
                break;

            case 'w':
                snprintf(argbuf, OPTIONS_BUFF_SIZE, "%s", optarg);
                opt_width = strtoint(argbuf);
                break;

            case 'h':
                snprintf(argbuf, OPTIONS_BUFF_SIZE, "%s", optarg);
                opt_height = strtoint(argbuf);
                break;

            case '?':
                break;

            default:
                abort();
        }
    }
}

void precalc_rgb(const float red, const float green, const float blue) {
    for (int n = 0; n < ASCII_PALETTE_SIZE; ++n) {
        RED[n]   = ((float) n) * red;
        GREEN[n] = ((float) n) * green;
        BLUE[n]  = ((float) n) * blue;
        GRAY[n]  = ((float) n);
    }
}