path: root/lpg/libqr/qrgen.c

#include <assert.h>
#include <ctype.h>
#include <getopt.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <qr/bitmap.h>
#include <qr/bitstream.h>
#include <qr/code.h>
#include <qr/data.h>
#include <qr/version.h>

struct config {
        int               version;
        enum qr_ec_level  ec;
        enum qr_data_type dtype;
        int               ansi;
        const char *      file;
        const char *      outfile;
        const char *      input;
};

struct qr_code * create(int               version,
                        enum qr_ec_level  ec,
                        enum qr_data_type dtype,
                        const char *      input,
                        size_t            len)
{
        struct qr_data * data;
        struct qr_code * code;

        data = qr_data_create(version, ec, dtype, input, len);

        if (!data) {
                /* BUG: this could also indicate OOM or
                 * some other error.
                 */
                fprintf(stderr, "Invalid data\n");
                exit(1);
        }

        code = qr_code_create(data);
        qr_data_destroy(data);

        if (!code) {
                perror("Failed to create code");
                exit(2);
        }

        return code;
}

void output_pbm(FILE * file, const struct qr_bitmap * bmp, const char * comment)
{
        unsigned char * row;
        int x, y;

        fputs("P1\n", file);

        if (comment)
                fprintf(file, "# %s\n", comment);

        fprintf(file, "%u %u\n",
               (unsigned)bmp->width + 8,
               (unsigned)bmp->height + 8);

        row = bmp->bits;

        for (y = -4; y < (int)bmp->height + 4; ++y) {

                if (y < 0 || y >= (int)bmp->height) {
                        for (x = 0; x < (int)bmp->width + 8; ++x)
                                fputs("0 ", file);
                        fputc('\n', file);
                        continue;
                }

                fputs("0 0 0 0 ", file);

                for (x = 0; x < (int)bmp->width; ++x) {

                        int mask = 1 << x % CHAR_BIT;
                        int byte = row[x / CHAR_BIT];

                        fprintf(file, "%c ", (byte & mask) ? '1' : '0');
                }

                fputs("0 0 0 0\n", file);
                row += bmp->stride;
        }
}

void output_ansi(FILE * file, const struct qr_bitmap * bmp)
{
        const char * out[2] = {
                "  ",
                "\033[7m  \033[0m",
        };

        unsigned char * line;
        size_t x, y;

        line = bmp->bits;

        for (y = 0; y < bmp->height; ++y) {

                for (x = 0; x < bmp->width; ++x) {

                        int mask = 1 << (x % CHAR_BIT);
                        int byte = line[x / CHAR_BIT];

                        fprintf(file, "%s", out[!!(byte & mask)]);
                }

                fputc('\n', file);

                line += bmp->stride;
        }
}

void show_help() {
        fprintf(stderr,
                "Usage:\n\t%s [options] <data>\n\n"
                "\t-h         Display this help message\n"
                "\t-f <file>  File containing data to encode (- for stdin)\n"
                "\t-v <n>     Specify QR version (size) 1 <= n <= 40\n"
                "\t-e <type>  Specify EC type: L, M, Q, H\n"
                "\t-a         Output as ANSI graphics (default)\n"
                "\t-p         Output as PBM\n"
                "\t-o <file>  File to write (- for stdout)\n\n",
                "qrgen");
}

void set_default_config(struct config * conf)
{
        conf->version = 0;
        conf->ec = QR_EC_LEVEL_M;
        conf->dtype = QR_DATA_8BIT;
        conf->ansi = 1;
        conf->file = NULL;
        conf->outfile = NULL;
        conf->input = NULL;
}

void parse_options(int argc, char ** argv, struct config * conf)
{
        int c;

        for (;;) {
                c = getopt(argc, argv, ":hf:v:e:t:apo:");

                if (c == -1) /* no more options */
                        break;

                switch (c) {
                case 'h': /* help */
                        show_help();
                        exit(0);
                        break;
                case 'f': /* file */
                        conf->file = optarg;
                        break;
                case 'v': /* version */
                        conf->version = atoi(optarg);
                        if (conf->version < 1 || conf->version > 40) {
                                fprintf(stderr,
                                 "Version must be between 1 and 40\n");
                                exit(1);
                        }
                        break;
                case 'e': /* ec */
                        switch (tolower(optarg[0])) {
                        case 'l': conf->ec = QR_EC_LEVEL_L; break;
                        case 'm': conf->ec = QR_EC_LEVEL_M; break;
                        case 'q': conf->ec = QR_EC_LEVEL_Q; break;
                        case 'h': conf->ec = QR_EC_LEVEL_H; break;
                        default:
                                fprintf(stderr,
                                        "Invalid EC type (%c). Choose from"
                                        " L, M, Q or H.\n", optarg[0]);
                        }
                        break;
                case 't': /* type */
                        fprintf(stderr, "XXX: ignored \"type\"\n");
                        break;
                case 'a': /* ansi */
                        conf->ansi = 1; break;
                case 'p': /* pnm */
                        conf->ansi = 0; break;
                case 'o': /* output file */
                        conf->outfile = optarg; break;
                case ':':
                        fprintf(stderr,
                                "Argument \"%s\" missing parameter\n",
                                argv[optind-1]);
                        exit(1);
                        break;
                case '?': default:
                        fprintf(stderr,
                                "Invalid argument: \"%s\"\n"
                                "Try -h for help\n",
                                argv[optind-1]);
                        exit(1);
                        break;
                }
        }


        if (optind < argc)
                conf->input = argv[optind++];

        if (!conf->file && !conf->input) {
                fprintf(stderr, "No data (try -h for help)\n");
                exit(1);
        }
}

void slurp_file(const char * path, char ** data, size_t * len)
{
        const size_t chunk_size = 65536;
        FILE * file;
        char * tmpbuf;
        size_t count;

        if (strcmp(path, "-") == 0)
                file = stdin;
        else
                file = fopen(path, "rb");

        if (!file) {
                fprintf(stderr, "Failed to open %s\n", path);
                exit(2);
        }

        *data = NULL;
        *len = 0;

        do {
                tmpbuf = realloc(*data, *len + chunk_size);
                if (!tmpbuf) {
                        perror("realloc");
                        exit(2);
                }
                *data = tmpbuf;
                count = fread(*data + *len, 1, chunk_size, file);
                if (count == 0 && !feof(file)) {
                        perror("fread");
                        exit(2);
                }
                *len += count;
        } while (*len == chunk_size);

        fclose(file);
}

int main(int argc, char ** argv) {

        struct config conf;
        struct qr_code * code;
        char * file_data;
        size_t len;
        FILE * outfile;

        set_default_config(&conf);
        parse_options(argc, argv, &conf);

        if (conf.file)
                slurp_file(conf.file, &file_data, &len);
        else
                len = strlen(conf.input);

        if (!conf.outfile) {
                conf.outfile = conf.ansi ? "-" : "qr.pbm";
        }

        if (strcmp(conf.outfile, "-") == 0) {
                outfile = stdout;
        } else {
                outfile = fopen(conf.outfile, "wb");
                if (!outfile) {
                        perror("fopen");
                        exit(2);
                }
        }

        code = create(conf.version,
                      conf.ec,
                      conf.dtype,
                      conf.file ? file_data : conf.input,
                      len);

        if (conf.file)
                free(file_data);

        if (conf.ansi)
                output_ansi(outfile, code->modules);
        else
                output_pbm(outfile, code->modules, "libqr v" QR_VERSION);

        fclose(outfile);
        qr_code_destroy(code);

        return 0;
}