working types and random
This commit is contained in:
+112
-86
@@ -1,16 +1,18 @@
|
||||
#include "args.h"
|
||||
#include "rand.h"
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <time.h>
|
||||
|
||||
void print_help(int status_code)
|
||||
{
|
||||
void print_help(int status_code) {
|
||||
puts("usage: margen "
|
||||
"[--help] "
|
||||
"[-q] "
|
||||
"[-w=WIDTH] "
|
||||
"[-h=HEIGHT] "
|
||||
"[-o=PATH] "
|
||||
"[--seed=SEED]"
|
||||
"[-p=PIXEL_SIZE] "
|
||||
"[-s=SLOPE] "
|
||||
"[-c=R,G,B] "
|
||||
@@ -22,6 +24,7 @@ void print_help(int status_code)
|
||||
" -w, --width image width (default: 1920)\n"
|
||||
" -h, --height image height (default: 1080)\n"
|
||||
" -o, --output output file (default: output.bmp)\n"
|
||||
" -seed random seed (default: time based)\n"
|
||||
" -p, --pixel pixel size (default: random)\n"
|
||||
" -s, --slope slope [0-255] (default: random)\n"
|
||||
" -c, --color base color [0-255,0-255,0-255] (default: random)\n"
|
||||
@@ -30,20 +33,15 @@ void print_help(int status_code)
|
||||
exit(status_code);
|
||||
}
|
||||
|
||||
void invalid_arg(char *arg)
|
||||
{
|
||||
void invalid_arg(char *arg) {
|
||||
fprintf(stderr, "invalid argument: '%s'\n\n", arg);
|
||||
print_help(1);
|
||||
}
|
||||
|
||||
void invalid_value(char *arg, char *subarg, char *value)
|
||||
{
|
||||
if (subarg == NULL)
|
||||
{
|
||||
void invalid_value(char *arg, char *subarg, char *value) {
|
||||
if (subarg == NULL) {
|
||||
fprintf(stderr, "invalid value for argument '%s': '%s'\n\n", arg, value);
|
||||
}
|
||||
else
|
||||
{
|
||||
} else {
|
||||
fprintf(stderr, "invalid value for argument '%s' %s: '%s'\n\n", arg, subarg,
|
||||
value);
|
||||
}
|
||||
@@ -52,129 +50,157 @@ void invalid_value(char *arg, char *subarg, char *value)
|
||||
|
||||
bool is_arg(char *arg, char *ref) { return strcoll(arg, ref) == 0; }
|
||||
|
||||
char *split_arg_value(char *arg)
|
||||
{
|
||||
char *split_arg_value(char *arg) {
|
||||
strtok(arg, "=");
|
||||
return strtok(NULL, "=");
|
||||
}
|
||||
|
||||
bool is_digit(char c) { return c >= '0' && c <= '9'; }
|
||||
|
||||
bool is_number(char *value)
|
||||
{
|
||||
if (value == NULL)
|
||||
{
|
||||
bool is_number(char *value, char *max) {
|
||||
if (value == NULL) {
|
||||
return false;
|
||||
}
|
||||
unsigned long value_len = strlen(value);
|
||||
int i;
|
||||
for (i = 0; i < value_len; i++)
|
||||
{
|
||||
if (!is_digit(value[i]))
|
||||
{
|
||||
for (i = 0; i < value_len; i++) {
|
||||
if (!is_digit(value[i])) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
return value;
|
||||
return strcmp(value, max) <= 0;
|
||||
}
|
||||
|
||||
long parse_number(char *arg, char *value)
|
||||
{
|
||||
if (!is_number(value))
|
||||
{
|
||||
unsigned char parse_char(char *arg, char *value) {
|
||||
if (!is_number(value, "255")) {
|
||||
invalid_value(arg, NULL, value);
|
||||
}
|
||||
return atol(value);
|
||||
return (char)atoi(value);
|
||||
}
|
||||
|
||||
void parse_color(char *arg, char *value, unsigned char color[3])
|
||||
{
|
||||
unsigned short parse_ushort(char *arg, char *value) {
|
||||
if (!is_number(value, "65535")) {
|
||||
invalid_value(arg, NULL, value);
|
||||
}
|
||||
return (unsigned short)atoi(value);
|
||||
}
|
||||
|
||||
unsigned short parse_ulong(char *arg, char *value) {
|
||||
if (!is_number(value, "18446744073709551615")) {
|
||||
invalid_value(arg, NULL, value);
|
||||
}
|
||||
return (unsigned long)atoll(value);
|
||||
}
|
||||
|
||||
void parse_color(char *arg, char *value, unsigned char color[3]) {
|
||||
char *tmp;
|
||||
tmp = strtok(value, ",");
|
||||
if (!is_number(tmp))
|
||||
{
|
||||
if (!is_number(tmp, "255")) {
|
||||
invalid_value(arg, "(R)", tmp);
|
||||
}
|
||||
color[0] = (unsigned char)atoi(tmp);
|
||||
color[0] = (char)atoi(tmp);
|
||||
tmp = strtok(NULL, ",");
|
||||
if (!is_number(tmp))
|
||||
{
|
||||
if (!is_number(tmp, "255")) {
|
||||
invalid_value(arg, "(G)", tmp);
|
||||
}
|
||||
color[1] = (unsigned char)atoi(tmp);
|
||||
color[1] = (char)atoi(tmp);
|
||||
tmp = strtok(NULL, ",");
|
||||
if (!is_number(tmp))
|
||||
{
|
||||
if (!is_number(tmp, "255")) {
|
||||
invalid_value(arg, "(B)", tmp);
|
||||
}
|
||||
color[2] = (unsigned char)atoi(tmp);
|
||||
color[2] = (char)atoi(tmp);
|
||||
}
|
||||
|
||||
parameters parse_args(int argc, char **argv)
|
||||
{
|
||||
parameters parse_args(int argc, char **argv) {
|
||||
parameters params;
|
||||
|
||||
params.quiet = false;
|
||||
params.width = 1920;
|
||||
params.height = 1080;
|
||||
params.seed = (unsigned long)time(NULL);
|
||||
params.width = 0;
|
||||
params.height = 0;
|
||||
params.file_path = "output.bmp";
|
||||
params.size = 1; // TODO random
|
||||
params.slope = 128; // TODO random
|
||||
params.start[0] = 128; // TODO random
|
||||
params.start[1] = 128; // TODO random
|
||||
params.start[2] = 128; // TODO random
|
||||
params.var[0] = 20; // TODO random
|
||||
params.var[1] = 20; // TODO random
|
||||
params.var[2] = 20; // TODO random
|
||||
|
||||
bool size_set = false;
|
||||
bool slope_set = false;
|
||||
bool start_set = false;
|
||||
bool var_set = false;
|
||||
|
||||
int i;
|
||||
char *arg;
|
||||
char *value;
|
||||
for (i = 1; i < argc; i++)
|
||||
{
|
||||
for (i = 1; i < argc; i++) {
|
||||
arg = argv[i];
|
||||
value = split_arg_value(arg);
|
||||
if (is_arg(arg, "--help"))
|
||||
{
|
||||
if (is_arg(arg, "--help")) {
|
||||
print_help(0);
|
||||
}
|
||||
else if (is_arg(arg, "-q") || is_arg(arg, "--quiet"))
|
||||
{
|
||||
} else if (is_arg(arg, "-q") || is_arg(arg, "--quiet")) {
|
||||
params.quiet = true;
|
||||
}
|
||||
else if (is_arg(arg, "-w") || is_arg(arg, "--width"))
|
||||
{
|
||||
params.width = (unsigned long)parse_number(arg, value);
|
||||
}
|
||||
else if (is_arg(arg, "-h") || is_arg(arg, "--height"))
|
||||
{
|
||||
params.height = (unsigned long)parse_number(arg, value);
|
||||
}
|
||||
else if (is_arg(arg, "-o") || is_arg(arg, "--output"))
|
||||
{
|
||||
} else if (is_arg(arg, "-w") || is_arg(arg, "--width")) {
|
||||
params.width = parse_ushort(arg, value);
|
||||
if (params.width == 0) {
|
||||
invalid_value(arg, NULL, value);
|
||||
}
|
||||
if (params.height == 0) {
|
||||
params.height = params.width;
|
||||
}
|
||||
} else if (is_arg(arg, "-h") || is_arg(arg, "--height")) {
|
||||
params.height = parse_ushort(arg, value);
|
||||
if (params.height == 0) {
|
||||
invalid_value(arg, NULL, value);
|
||||
}
|
||||
if (params.width == 0) {
|
||||
params.width = params.height;
|
||||
}
|
||||
} else if (is_arg(arg, "--seed")) {
|
||||
params.seed = parse_ulong(arg, value);
|
||||
} else if (is_arg(arg, "-o") || is_arg(arg, "--output")) {
|
||||
params.file_path = value;
|
||||
}
|
||||
else if (is_arg(arg, "-p") || is_arg(arg, "--pixel"))
|
||||
{
|
||||
params.size = (unsigned char)parse_number(arg, value);
|
||||
}
|
||||
else if (is_arg(arg, "-s") || is_arg(arg, "--slope"))
|
||||
{
|
||||
params.size = (unsigned char)parse_number(arg, value);
|
||||
}
|
||||
else if (is_arg(arg, "-c") || is_arg(arg, "--color"))
|
||||
{
|
||||
} else if (is_arg(arg, "-p") || is_arg(arg, "--pixel")) {
|
||||
params.size = parse_ushort(arg, value);
|
||||
if (params.size == 0) {
|
||||
invalid_value(arg, NULL, value);
|
||||
}
|
||||
size_set = true;
|
||||
} else if (is_arg(arg, "-s") || is_arg(arg, "--slope")) {
|
||||
params.slope = parse_char(arg, value);
|
||||
slope_set = true;
|
||||
} else if (is_arg(arg, "-c") || is_arg(arg, "--color")) {
|
||||
parse_color(arg, value, params.start);
|
||||
}
|
||||
else if (is_arg(arg, "-v") || is_arg(arg, "--variation"))
|
||||
{
|
||||
parse_color(arg, value, params.start);
|
||||
}
|
||||
else
|
||||
{
|
||||
start_set = true;
|
||||
} else if (is_arg(arg, "-v") || is_arg(arg, "--variation")) {
|
||||
parse_color(arg, value, params.var);
|
||||
var_set = true;
|
||||
} else {
|
||||
invalid_arg(arg);
|
||||
}
|
||||
}
|
||||
|
||||
if (params.width == 0 && params.height == 0) {
|
||||
params.width = 1920;
|
||||
params.height = 1080;
|
||||
}
|
||||
|
||||
set_seed(params.seed);
|
||||
|
||||
if (!size_set) {
|
||||
params.size = rand_ushort(10) + 3;
|
||||
}
|
||||
|
||||
if (!slope_set) {
|
||||
params.slope = rand_uchar(50) + 100;
|
||||
}
|
||||
|
||||
if (!start_set) {
|
||||
params.start[0] = rand_uchar(256);
|
||||
params.start[1] = rand_uchar(256);
|
||||
params.start[2] = rand_uchar(256);
|
||||
}
|
||||
|
||||
if (!var_set) {
|
||||
params.var[0] = rand_uchar(30);
|
||||
params.var[1] = rand_uchar(30);
|
||||
params.var[2] = rand_uchar(30);
|
||||
}
|
||||
|
||||
return params;
|
||||
}
|
||||
Reference in New Issue
Block a user