Workshop o mikrokontrolérech na SKSP 2024.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

201 lines
6.2 KiB

/*
* Types used in table printer
*
* (c) 2014 Robert Kessl <robert.kessl@economia.cz>
*/
#include <ucw/lib.h>
#include <ucw/table.h>
#include <ucw/xtypes-extra.h>
#include <ucw/opt.h>
#include <stdio.h>
enum test_table_cols {
TEST_COL0_STR, TEST_COL1_INT, TEST_COL2_UINT, TEST_COL3_BOOL, TEST_COL4_DOUBLE, TEST_COL5_SIZE, TEST_COL6_TIME
};
static struct table_col_instance test_column_order[] = { TBL_COL(TEST_COL3_BOOL), TBL_COL(TEST_COL4_DOUBLE),
TBL_COL(TEST_COL2_UINT), TBL_COL(TEST_COL1_INT), TBL_COL(TEST_COL0_STR), TBL_COL_ORDER_END };
static struct table_template test_tbl = {
TBL_COLUMNS {
[TEST_COL0_STR] = TBL_COL_STR("col0_str", 20),
[TEST_COL1_INT] = TBL_COL_INT("col1_int", 8),
[TEST_COL2_UINT] = TBL_COL_UINT("col2_uint", 9),
[TEST_COL3_BOOL] = TBL_COL_BOOL_FMT("col3_bool", 9, XTYPE_FMT_PRETTY),
[TEST_COL4_DOUBLE] = TBL_COL_DOUBLE_FMT("col4_double", 11, XTYPE_FMT_PRETTY),
[TEST_COL5_SIZE] = TBL_COL_SIZE("col5_size", 11),
[TEST_COL6_TIME] = TBL_COL_TIMESTAMP("col6_timestamp", 20),
TBL_COL_END
},
TBL_COL_ORDER(test_column_order),
TBL_FMT_HUMAN_READABLE,
TBL_COL_DELIMITER("\t"),
};
enum test_default_order_cols {
TEST_DEFAULT_ORDER_COL0_INT, TEST_DEFAULT_ORDER_COL1_INT, TEST_DEFAULT_ORDER_COL2_INT
};
static struct table_template test_default_order_tbl = {
TBL_COLUMNS {
[TEST_DEFAULT_ORDER_COL0_INT] = TBL_COL_INT("col0_int", 8),
[TEST_DEFAULT_ORDER_COL1_INT] = TBL_COL_INT("col1_int", 9),
[TEST_DEFAULT_ORDER_COL2_INT] = TBL_COL_INT("col2_int", 9),
TBL_COL_END
},
TBL_FMT_HUMAN_READABLE,
TBL_COL_DELIMITER("\t"),
};
static void do_default_order_test(struct fastbuf *out)
{
struct table *tbl = table_init(&test_default_order_tbl);
table_start(tbl, out);
table_col_int(tbl, TEST_DEFAULT_ORDER_COL0_INT, 0);
table_col_int(tbl, TEST_DEFAULT_ORDER_COL1_INT, 1);
table_col_int(tbl, TEST_DEFAULT_ORDER_COL2_INT, 2);
table_end_row(tbl);
table_col_int(tbl, TEST_DEFAULT_ORDER_COL0_INT, 10);
table_col_int(tbl, TEST_DEFAULT_ORDER_COL1_INT, 11);
table_col_int(tbl, TEST_DEFAULT_ORDER_COL2_INT, 12);
table_end_row(tbl);
table_end(tbl);
table_cleanup(tbl);
}
/**
* tests: table_col_int, table_col_uint, table_col_bool, table_col_double, table_col_printf
**/
static void do_print1(struct table *test_tbl)
{
table_col_str(test_tbl, TEST_COL0_STR, "sdsdf");
table_col_int(test_tbl, TEST_COL1_INT, -10);
table_col_int(test_tbl, TEST_COL1_INT, 10000);
table_col_uint(test_tbl, TEST_COL2_UINT, 10);
table_col_printf(test_tbl, TEST_COL2_UINT, "XXX-%u", 22222);
table_col_bool(test_tbl, TEST_COL3_BOOL, 1);
table_col_double(test_tbl, TEST_COL4_DOUBLE, 1.5);
table_col_size(test_tbl, TEST_COL5_SIZE, (1024LLU*1024LLU*1024LLU*5LLU));
table_col_timestamp(test_tbl, TEST_COL6_TIME, 1404305876);
table_end_row(test_tbl);
table_col_str(test_tbl, TEST_COL0_STR, "test");
table_col_int(test_tbl, TEST_COL1_INT, -100);
table_col_uint(test_tbl, TEST_COL2_UINT, 100);
table_col_bool(test_tbl, TEST_COL3_BOOL, 0);
table_col_double(test_tbl, TEST_COL4_DOUBLE, 1.5);
table_col_size(test_tbl, TEST_COL5_SIZE, (1024LLU*1024LLU*1024LLU*2LLU));
table_col_timestamp(test_tbl, TEST_COL6_TIME, 1404305909);
table_end_row(test_tbl);
}
static char **cli_table_opts;
enum test_type_t {
TEST_DEFAULT_COLUMN_ORDER = 1,
TEST_INVALID_OPTION = 2,
TEST_INVALID_ORDER = 3
};
static int test_to_perform = -1;
static struct opt_section table_printer_opts = {
OPT_ITEMS {
OPT_HELP("Options:"),
OPT_STRING_MULTIPLE('T', "table", cli_table_opts, OPT_REQUIRED_VALUE, "\tSets options for the table."),
OPT_SWITCH('d', 0, test_to_perform, TEST_DEFAULT_COLUMN_ORDER, OPT_SINGLE, "\tRun the test that uses the default column order."),
OPT_SWITCH('i', 0, test_to_perform, TEST_INVALID_OPTION, OPT_SINGLE, "\tTest the output for invalid option."),
OPT_SWITCH('n', 0, test_to_perform, TEST_INVALID_ORDER, OPT_SINGLE, "\tTest the output for invalid names of columns for the 'cols' option."),
OPT_END
}
};
static void process_command_line_opts(char *argv[], struct table *tbl)
{
GARY_INIT(cli_table_opts, 0);
opt_parse(&table_printer_opts, argv+1);
const char *err = table_set_gary_options(tbl, cli_table_opts);
if(err) {
opt_failure("error while setting cmd line options: %s", err);
}
GARY_FREE(cli_table_opts);
}
static bool user_defined_option(struct table *tbl UNUSED, const char *key, const char *value, const char **err UNUSED)
{
if(value == NULL && strcmp(key, "novaluekey") == 0) {
printf("setting key: %s; value: (null)\n", key);
return 1;
}
if(value != NULL && strcmp(value, "value") == 0 &&
key != NULL && strcmp(key, "valuekey") == 0) {
printf("setting key: %s; value: %s\n", key, value);
return 1;
}
return 0;
}
static void test_option_parser(struct table *tbl)
{
struct table_formatter test_option_fmtr = table_fmt_human_readable;
test_option_fmtr.process_option = user_defined_option;
const struct table_formatter *tmp_fmtr = tbl->formatter;
tbl->formatter = &test_option_fmtr;
const char *rv = table_set_option(tbl, "invalid:option");
if(rv) printf("Tableprinter option parser returned error: \"%s\".\n", rv);
rv = table_set_option(tbl, "invalid");
if(rv) printf("Tableprinter option parser returned error: \"%s\".\n", rv);
rv = table_set_option(tbl, "novaluekey");
if(rv) printf("Tableprinter option parser returned error: \"%s\".\n", rv);
rv = table_set_option(tbl, "valuekey:value");
if(rv) printf("Tableprinter option parser returned error: \"%s\".\n", rv);
tbl->formatter = tmp_fmtr;
}
int main(int argc UNUSED, char **argv)
{
struct fastbuf *out;
out = bfdopen_shared(1, 4096);
struct table *tbl = table_init(&test_tbl);
process_command_line_opts(argv, tbl);
const char *rv = NULL;
switch(test_to_perform) {
case TEST_INVALID_ORDER:
rv = table_set_option(tbl, "cols:test_col0_str,test_col1_int,xxx");
if(rv) printf("Tableprinter option parser returned: '%s'.\n", rv);
return 0;
case TEST_DEFAULT_COLUMN_ORDER:
do_default_order_test(out);
bclose(out);
return 0;
case TEST_INVALID_OPTION:
test_option_parser(tbl);
bclose(out);
return 0;
};
table_start(tbl, out);
do_print1(tbl);
table_end(tbl);
table_cleanup(tbl);
bclose(out);
return 0;
}