#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <SDL.h>
#include <getopt.h>
#include <errno.h>
#include "../qrspec.h"
#include "../qrinput.h"
#include "../qrencode_inner.h"
#include "../split.h"
static SDL_Surface *screen = NULL;
static int casesensitive = 1;
static int eightbit = 0;
static int version = 1;
static int size = 4;
static int margin = 4;
static int structured = 0;
static QRecLevel level = QR_ECLEVEL_L;
static QRencodeMode hint = QR_MODE_8;
static char **textv;
static int textc;
enum {
O_HELP,
O_SIZE,
O_VERSION,
O_LEVEL,
O_MARGIN,
O_KANJI,
O_CASE,
O_IGNORECASE,
O_8BIT,
O_STRUCTURED,
};
static const struct option options[] = {
{"h", no_argument , NULL, O_HELP},
{"l", required_argument, NULL, O_LEVEL},
{"s", required_argument, NULL, O_SIZE},
{"v", required_argument, NULL, O_VERSION},
{"m", required_argument, NULL, O_MARGIN},
{"k", no_argument , NULL, O_KANJI},
{"c", no_argument , NULL, O_CASE},
{"i", no_argument , NULL, O_IGNORECASE},
{"8", no_argument , NULL, O_8BIT},
{"S", no_argument , NULL, O_STRUCTURED},
{NULL, 0, NULL, 0}
};
static char levelChar[4] = {'L', 'M', 'Q', 'H'};
static void usage(void)
{
fprintf(stderr,
"view_qrcode version %s\n"
"Copyright (C) 2008 Kentaro Fukuchi\n"
"Usage: view_qrcode [OPTION]... [STRING]\n"
"Encode input data in a QR Code and save as a PNG image.\n\n"
" -h display this message.\n"
" -s NUMBER specify the size of dot (pixel). (default=4)\n"
" -l {LMQH} specify error collectin level from L (lowest) to H (highest).\n"
" (default=L)\n"
" -v NUMBER specify the version of the symbol. (default=1)\n"
" -m NUMBER specify the width of margin. (default=4)\n"
" -S make structured symbols. Version must be specified.\n"
" -k assume that the input text contains kanji (shift-jis).\n"
" -c encode lower-case alphabet characters in 8-bit mode. (default)\n"
" -i ignore case distinctions and use only upper-case characters.\n"
" -8 encode entire data in 8-bit mode. -k, -c and -i will be ignored.\n"
" [STRING] input data. If it is not specified, data will be taken from\n"
" standard input.\n",
VERSION);
}
#define MAX_DATA_SIZE (7090 * 16) /* from the specification */
static char *readStdin(void)
{
char *buffer;
int ret;
buffer = (char *)malloc(MAX_DATA_SIZE);
ret = fread(buffer, 1, MAX_DATA_SIZE, stdin);
if(ret == 0) {
fprintf(stderr, "No input data.\n");
exit(1);
}
if(!feof(stdin)) {
fprintf(stderr, "Input data is too large.\n");
exit(1);
}
buffer[ret] = '\0';
return buffer;
}
static void draw_QRcode(QRcode *qrcode, int ox, int oy)
{
int pitch;
int x, y, width;
unsigned char *p;
SDL_Rect rect;
ox += margin * size;
oy += margin * size;
width = qrcode->width;
p = qrcode->data;
pitch = screen->pitch;
for(y=0; y<width; y++) {
for(x=0; x<width; x++) {
rect.x = ox + x * size;
rect.y = oy + y * size;
rect.w = size;
rect.h = size;
SDL_FillRect(screen, &rect, (*p&1)?0:0xffffff);
p++;
}
}
}
void draw_singleQRcode(QRinput *stream, int mask)
{
QRcode *qrcode;
int width;
QRinput_setVersion(stream, version);
QRinput_setErrorCorrectionLevel(stream, level);
qrcode = QRcode_encodeMask(stream, mask);
if(qrcode == NULL) return;
version = qrcode->version;
width = (qrcode->width + margin * 2) * size;
screen = SDL_SetVideoMode(width, width, 32, 0);
SDL_FillRect(screen, NULL, 0xffffff);
draw_QRcode(qrcode, 0, 0);
SDL_Flip(screen);
QRcode_free(qrcode);
}
void draw_structuredQRcode(QRinput_Struct *s, int mask)
{
int i, w, h, n, x, y;
int swidth;
QRcode_List *qrcodes, *p;
qrcodes = QRcode_encodeInputStructured(s);
if(qrcodes == NULL) return;
swidth = (qrcodes->code->width + margin * 2) * size;
n = QRcode_List_size(qrcodes);
w = (n < 4)?n:4;
h = (n - 1) / 4 + 1;
screen = SDL_SetVideoMode(swidth * w, swidth * h, 32, 0);
SDL_FillRect(screen, NULL, 0xffffff);
p = qrcodes;
for(i=0; i<n; i++) {
x = (i % 4) * swidth;
y = (i / 4) * swidth;
draw_QRcode(p->code, x, y);
p = p->next;
}
SDL_Flip(screen);
QRcode_List_free(qrcodes);
}
void draw_structuredQRcodeFromText(int argc, char **argv, int mask)
{
QRinput_Struct *s;
QRinput *input;
int i, ret;
s = QRinput_Struct_new();
if(s == NULL) {
fprintf(stderr, "Failed to allocate memory.\n");
exit(1);
}
for(i=0; i<argc; i++) {
input = QRinput_new2(version, level);
if(input == NULL) {
fprintf(stderr, "Failed to allocate memory.\n");
exit(1);
}
if(eightbit) {
ret = QRinput_append(input, QR_MODE_8, strlen(argv[i]), (unsigned char *)argv[i]);
} else {
ret = Split_splitStringToQRinput(argv[i], input, hint, casesensitive);
}
if(ret < 0) {
perror("Encoding the input string");
exit(1);
}
ret = QRinput_Struct_appendInput(s, input);
if(ret < 0) {
perror("Encoding the input string");
exit(1);
}
}
ret = QRinput_Struct_insertStructuredAppendHeaders(s);
if(ret < 0) {
fprintf(stderr, "Too many inputs.\n");
}
draw_structuredQRcode(s, mask);
QRinput_Struct_free(s);
}
void draw_structuredQRcodeFromQRinput(QRinput *stream, int mask)
{
QRinput_Struct *s;
QRinput_setVersion(stream, version);
QRinput_setErrorCorrectionLevel(stream, level);
s = QRinput_splitQRinputToStruct(stream);
if(s != NULL) {
draw_structuredQRcode(s, mask);
QRinput_Struct_free(s);
} else {
fprintf(stderr, "Input data is too large for this setting.\n");
}
}
void view(int mode, QRinput *input)
{
int flag = 1;
int mask = -1;
SDL_Event event;
int loop;
while(flag) {
if(mode) {
draw_structuredQRcodeFromText(textc, textv, mask);
} else {
if(structured) {
draw_structuredQRcodeFromQRinput(input, mask);
} else {
draw_singleQRcode(input, mask);
}
}
printf("Version %d, Level %c, Mask %d.\n", version, levelChar[level], mask);
loop = 1;
while(loop) {
usleep(10000);
while(SDL_PollEvent(&event)) {
if(event.type == SDL_KEYDOWN) {
switch(event.key.keysym.sym) {
case SDLK_RIGHT:
version++;
if(version > QRSPEC_VERSION_MAX)
version = QRSPEC_VERSION_MAX;
loop = 0;
break;
case SDLK_LEFT:
version--;
if(version < 1)
version = 1;
loop = 0;
break;
case SDLK_UP:
size++;
loop = 0;
break;
case SDLK_DOWN:
size--;
if(size < 1) size = 1;
loop = 0;
break;
case SDLK_0:
case SDLK_1:
case SDLK_2:
case SDLK_3:
case SDLK_4:
case SDLK_5:
case SDLK_6:
case SDLK_7:
mask = (event.key.keysym.sym - SDLK_0);
loop = 0;
break;
case SDLK_8:
mask = -1;
loop = 0;
break;
case SDLK_l:
level = QR_ECLEVEL_L;
loop = 0;
break;
case SDLK_m:
level = QR_ECLEVEL_M;
loop = 0;
break;
case SDLK_h:
level = QR_ECLEVEL_H;
loop = 0;
break;
case SDLK_q:
level = QR_ECLEVEL_Q;
loop = 0;
break;
case SDLK_ESCAPE:
loop = 0;
flag = 0;
break;
default:
break;
}
}
if(event.type == SDL_QUIT) {
loop = 0;
flag = 0;
}
}
}
}
}
void view_simple(const char *str)
{
QRinput *input;
int ret;
input = QRinput_new2(version, level);
if(input == NULL) {
fprintf(stderr, "Memory allocation error.\n");
exit(1);
}
if(eightbit) {
ret = QRinput_append(input, QR_MODE_8, strlen(str), (unsigned char *)str);
} else {
ret = Split_splitStringToQRinput(str, input, hint, casesensitive);
}
if(ret < 0) {
perror("Encoding the input string");
exit(1);
}
view(0, input);
QRinput_free(input);
}
void view_multiText(char **argv, int argc)
{
textc = argc;
textv = argv;
view(1, NULL);
}
int main(int argc, char **argv)
{
int opt;
char *intext = NULL;
while((opt = getopt_long_only(argc, argv, "", options, NULL)) != -1) {
switch(opt) {
case O_HELP:
usage();
exit(0);
break;
case O_SIZE:
size = atoi(optarg);
if(size <= 0) {
fprintf(stderr, "Invalid size: %d\n", size);
exit(1);
}
break;
case O_VERSION:
version = atoi(optarg);
if(version < 0) {
fprintf(stderr, "Invalid version: %d\n", version);
exit(1);
}
break;
case O_LEVEL:
switch(*optarg) {
case 'l':
case 'L':
level = QR_ECLEVEL_L;
break;
case 'm':
case 'M':
level = QR_ECLEVEL_M;
break;
case 'q':
case 'Q':
level = QR_ECLEVEL_Q;
break;
case 'h':
case 'H':
level = QR_ECLEVEL_H;
break;
default:
fprintf(stderr, "Invalid level: %s\n", optarg);
exit(1);
break;
}
break;
case O_MARGIN:
margin = atoi(optarg);
if(margin < 0) {
fprintf(stderr, "Invalid margin: %d\n", margin);
exit(1);
}
break;
case O_STRUCTURED:
structured = 1;
case O_KANJI:
hint = QR_MODE_KANJI;
break;
case O_CASE:
casesensitive = 1;
break;
case O_IGNORECASE:
casesensitive = 0;
break;
case O_8BIT:
eightbit = 1;
break;
default:
usage();
exit(1);
break;
}
}
if(argc == 1) {
usage();
exit(0);
}
if(optind < argc) {
intext = argv[optind];
}
if(intext == NULL) {
intext = readStdin();
}
if(SDL_Init(SDL_INIT_VIDEO) < 0) {
fprintf(stderr, "Failed initializing SDL: %s\n", SDL_GetError());
return -1;
}
if(structured && version < 1) {
fprintf(stderr, "Version number must be larger than 0.\n");
exit(1);
}
if(structured && (argc - optind > 1)) {
view_multiText(argv + optind, argc - optind);
} else {
view_simple(intext);
}
SDL_Quit();
return 0;
}
