mirror of
https://gitee.com/Lamdonn/varch.git
synced 2025-12-06 16:56:42 +08:00
583 lines
15 KiB
C
583 lines
15 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#if defined(TEST_TARGET_tree)
|
|
#include <varch/command.h>
|
|
#include <varch/unitt.h>
|
|
#include <varch/tree.h>
|
|
#else
|
|
#include "init.h"
|
|
#include "command.h"
|
|
#include "unitt.h"
|
|
#include "kern.h"
|
|
#include "tree.h"
|
|
#endif
|
|
|
|
/************************************************************************************/
|
|
/************************************* Unit Test ************************************/
|
|
/************************************************************************************/
|
|
|
|
// #define EXIT_TEST
|
|
extern uint64_t unitt_clock(void);
|
|
|
|
static int test_0(void)
|
|
{
|
|
for (int i = 0; i < 100; i++)
|
|
{
|
|
if (0)
|
|
{
|
|
|
|
#if defined (EXIT_TEST)
|
|
exit(0);
|
|
#endif
|
|
return UNITT_E_FAIL;
|
|
}
|
|
}
|
|
|
|
return UNITT_E_OK;
|
|
}
|
|
|
|
static void unitt_task(void)
|
|
{
|
|
static UNITT_TCASE rand_tests[] = {
|
|
UNITT_TCASE(test_0),
|
|
// UNITT_TCASE(test_1),
|
|
// UNITT_TCASE(test_2),
|
|
};
|
|
|
|
static UNITT suites[] = {
|
|
{ "xxx suite", rand_tests, sizeof(rand_tests) / sizeof(rand_tests[0]) , unitt_clock },
|
|
};
|
|
|
|
UNITT_EXE(suites);
|
|
}
|
|
|
|
/************************************************************************************/
|
|
/************************************* Base Test ************************************/
|
|
/************************************************************************************/
|
|
|
|
static void print_int(tree_t tree)
|
|
{
|
|
if (tree_data(tree)) printf("%d", *(int *)tree_data(tree));
|
|
}
|
|
|
|
static void print_string(tree_t tree)
|
|
{
|
|
if (tree_data(tree)) printf("%s", (char *)tree_data(tree));
|
|
}
|
|
|
|
static void print_dir(tree_t tree)
|
|
{
|
|
if (tree_data(tree)) printf("%s", (char *)tree_data(tree));
|
|
}
|
|
|
|
static void print_ini(tree_t tree)
|
|
{
|
|
if (!tree_asize(tree)) return;
|
|
printf("%s", (char *)tree_attribute(tree));
|
|
if (tree_dsize(tree)) printf(" = %s", (char *)tree_data(tree));
|
|
}
|
|
|
|
static void print_bitree(tree_t tree)
|
|
{
|
|
if (tree_dsize(tree)) printf("%s", (char *)tree_data(tree));
|
|
}
|
|
|
|
#if 0
|
|
#include "ini.h"
|
|
static tree_t tree_from_ini(const char *filename)
|
|
{
|
|
tree_t tree = NULL, section, pair;
|
|
ini_t ini;
|
|
char *section_name, *key, *value;
|
|
int i, j;
|
|
ini = ini_file_load(filename);
|
|
if (!ini) goto FAIL;
|
|
tree = tree_create();
|
|
if (!tree) goto FAIL;
|
|
for (i = 0; i < ini_section_count(ini); i++)
|
|
{
|
|
section = tree_create();
|
|
if (!section) goto FAIL;
|
|
tree_insert(tree, tree_csize(tree));
|
|
tree_attach(tree, tree_csize(tree) - 1, section);
|
|
section_name = (char*)ini_section_name(ini, i);
|
|
tree_set_attribute(section, section_name, strlen(section_name) + 1);
|
|
for (j = 0; j < ini_pair_count(ini, section_name); j++)
|
|
{
|
|
pair = tree_create();
|
|
if (!pair) goto FAIL;
|
|
tree_insert(section, tree_csize(section));
|
|
tree_attach(section, tree_csize(section) - 1, pair);
|
|
key = (char *)ini_key_name(ini, section_name, j);
|
|
tree_set_attribute(pair, key, strlen(key) + 1);
|
|
value = (char *)ini_get_value(ini, section_name, key);
|
|
tree_set_data(pair, value, strlen(value) + 1);
|
|
}
|
|
}
|
|
ini_delete(ini);
|
|
return tree;
|
|
FAIL:
|
|
if (ini) ini_delete(ini);
|
|
if (tree) tree_delete(tree, NULL);
|
|
return NULL;
|
|
}
|
|
|
|
static void test_ini(void)
|
|
{
|
|
tree_t tree, n;
|
|
tree = tree_from_ini("../source/application/test/file/read.ini");
|
|
if (!tree)
|
|
{
|
|
printf("transfer fail!\r\n");
|
|
}
|
|
tree_print(tree_child(tree, 0), 0, print_ini);
|
|
tree_delete(tree, NULL);
|
|
}
|
|
#endif
|
|
|
|
#include <dirent.h>
|
|
#include <sys/stat.h>
|
|
static tree_t tree_directory(const char *directory, int depth)
|
|
{
|
|
tree_t tree = NULL, n;
|
|
DIR *dirptr;
|
|
struct dirent* dircontent;
|
|
if (depth < 0) return NULL;
|
|
tree = tree_create();
|
|
if (!tree) return NULL;
|
|
dirptr = opendir(directory);
|
|
if (!dirptr) goto FAIL;
|
|
while (1)
|
|
{
|
|
dircontent = readdir(dirptr);
|
|
if (dircontent == NULL) break;
|
|
if (strcmp(dircontent->d_name,".") == 0 || strcmp(dircontent->d_name,"..") == 0) continue;
|
|
struct stat statinfo;
|
|
char fullname[1024];
|
|
sprintf(fullname, "%s/%s", directory, dircontent->d_name);
|
|
if (stat(fullname, &statinfo)) continue;
|
|
if (S_ISDIR(statinfo.st_mode) && depth != 1) n = tree_directory(fullname, depth ? depth - 1 : 0);
|
|
else if (S_ISREG(statinfo.st_mode)) n = tree_create();
|
|
else continue;
|
|
if (n)
|
|
{
|
|
if (!tree_insert(tree, tree_csize(tree))) goto FAIL;
|
|
tree_attach(tree, tree_csize(tree) - 1, n);
|
|
}
|
|
tree_set_data(n, dircontent->d_name, strlen(dircontent->d_name) + 1);
|
|
}
|
|
if (!tree_data(tree)) tree_set_data(tree, directory, strlen(directory) + 1);
|
|
return tree;
|
|
FAIL:
|
|
if (tree) tree_delete(tree, NULL);
|
|
return NULL;
|
|
}
|
|
|
|
static void test_dir(void)
|
|
{
|
|
tree_t tree;
|
|
tree = tree_directory("./", 0);
|
|
if (!tree)
|
|
{
|
|
printf("load fail!\r\n");
|
|
}
|
|
// tree_insert(tree, 1);
|
|
// printf("name %s, %d\r\n", tree_data(tree), tree_size(tree));
|
|
tree_print(tree, 0, print_dir);
|
|
tree_delete(tree, NULL);
|
|
}
|
|
|
|
static void test_create(void)
|
|
{
|
|
tree_t tree = tree_create();
|
|
|
|
if (tree)
|
|
{
|
|
printf("tree create success!!!\r\n");
|
|
}
|
|
else
|
|
{
|
|
printf("[ERROR] tree create fail!!!\r\n");
|
|
}
|
|
|
|
tree_delete(tree, NULL);
|
|
}
|
|
|
|
static void test_set(void)
|
|
{
|
|
tree_t tree = tree_create();
|
|
int set = 100;
|
|
int get = 0;
|
|
|
|
tree_set_data(tree, &set, sizeof(set));
|
|
|
|
tree_get_data(tree, &get, sizeof(get));
|
|
|
|
printf("get %d\r\n", get);
|
|
printf("data %d\r\n", *(int *)tree_data(tree));
|
|
|
|
tree_delete(tree, NULL);
|
|
}
|
|
|
|
static void test_insert(void)
|
|
{
|
|
tree_t tree = tree_create();
|
|
|
|
tree_insert(tree, 0);
|
|
tree_insert(tree, 1);
|
|
tree_insert(tree, 2);
|
|
tree_insert(tree, 3);
|
|
|
|
printf("tree child size %d\r\n", tree_csize(tree));
|
|
|
|
tree_erase(tree, 2);
|
|
|
|
printf("tree child size %d\r\n", tree_csize(tree));
|
|
|
|
tree_delete(tree, NULL);
|
|
}
|
|
|
|
static void test_attach(void)
|
|
{
|
|
tree_t tree = tree_create();
|
|
tree_t node = NULL;
|
|
int data = 0;
|
|
|
|
tree_insert(tree, 0);
|
|
tree_insert(tree, 1);
|
|
tree_insert(tree, 2);
|
|
tree_insert(tree, 3);
|
|
|
|
for (int i = 0; i < tree_csize(tree); i++)
|
|
{
|
|
node = tree_create();
|
|
data = 1000 + i;
|
|
tree_attach(tree, i, node);
|
|
tree_set_data(node, &data, sizeof(data));
|
|
}
|
|
|
|
tree_print(tree, 0, print_int);
|
|
|
|
node = tree_detach(tree, 1);
|
|
tree_delete(node, NULL);
|
|
|
|
tree_print(tree, 0, print_int);
|
|
|
|
tree_delete(tree, NULL);
|
|
}
|
|
|
|
static void test_parent(void)
|
|
{
|
|
tree_t tree = tree_create();
|
|
tree_t node = NULL;
|
|
tree_t parent, child;
|
|
|
|
tree_insert(tree, 0);
|
|
tree_insert(tree, 1);
|
|
tree_insert(tree, 2);
|
|
tree_insert(tree, 3);
|
|
|
|
tree_set_data(tree, "Parent", 7);
|
|
|
|
node = tree_create(); tree_set_data(node, "Child0", 7); tree_attach(tree, 0, node);
|
|
node = tree_create(); tree_set_data(node, "Child1", 7); tree_attach(tree, 1, node);
|
|
node = tree_create(); tree_set_data(node, "Child2", 7); tree_attach(tree, 2, node);
|
|
node = tree_create(); tree_set_data(node, "Child3", 7); tree_attach(tree, 3, node);
|
|
|
|
tree_print(tree, 0, print_string);
|
|
|
|
parent = tree_parent(node); print_string(parent); printf("\r\n");
|
|
|
|
child = tree_child(tree, 0); print_string(child); printf("\r\n");
|
|
child = tree_child(tree, 1); print_string(child); printf("\r\n");
|
|
child = tree_child(tree, 2); print_string(child); printf("\r\n");
|
|
child = tree_child(tree, 3); print_string(child); printf("\r\n");
|
|
|
|
tree_delete(tree, NULL);
|
|
}
|
|
|
|
static void test_attr(void)
|
|
{
|
|
tree_t tree = tree_create();
|
|
char attr[20];
|
|
|
|
tree_set_attribute(tree, "This attribute!!!", 18);
|
|
tree_get_attribute(tree, attr, sizeof(attr));
|
|
|
|
printf("attr %s\r\n", attr);
|
|
|
|
tree_delete(tree, NULL);
|
|
}
|
|
|
|
static void test_depth(void)
|
|
{
|
|
tree_t root = tree_create();
|
|
tree_t node, temp;
|
|
|
|
tree_insert(root, 0);
|
|
tree_insert(root, 1);
|
|
tree_insert(root, 2);
|
|
|
|
tree_attach(root, 0, tree_create());
|
|
tree_attach(root, 1, tree_create());
|
|
|
|
node = tree_child(root, 0);
|
|
tree_insert(node, 0);
|
|
tree_insert(node, 1);
|
|
|
|
node = tree_child(root, 1);
|
|
tree_insert(node, 0);
|
|
tree_insert(node, 1);
|
|
tree_attach(node, 0, tree_create());
|
|
tree_attach(node, 1, tree_create());
|
|
|
|
tree_print(root, 0, NULL);
|
|
|
|
printf("depth %d\r\n", tree_depth(root));
|
|
|
|
tree_delete(root, NULL);
|
|
}
|
|
|
|
static void test_to(void)
|
|
{
|
|
tree_t root = tree_create();
|
|
tree_t node, temp;
|
|
|
|
tree_insert(root, 0);
|
|
tree_insert(root, 1);
|
|
tree_insert(root, 2);
|
|
|
|
tree_attach(root, 0, tree_create());
|
|
tree_attach(root, 1, tree_create());
|
|
|
|
node = tree_child(root, 0);
|
|
tree_insert(node, 0);
|
|
tree_insert(node, 1);
|
|
|
|
node = tree_child(root, 1);
|
|
tree_insert(node, 0);
|
|
tree_insert(node, 1);
|
|
tree_attach(node, 0, tree_create());
|
|
tree_attach(node, 1, tree_create());
|
|
|
|
temp = tree_to(root, 1, 1);
|
|
tree_set_data(temp, "To function!!!", 15);
|
|
|
|
tree_print(root, 0, print_string);
|
|
|
|
tree_delete(root, NULL);
|
|
}
|
|
|
|
static void test_common(void)
|
|
{
|
|
tree_t root, n, t;
|
|
int i, data;
|
|
|
|
root = tree_create();
|
|
data = 1024; tree_set_data(root, &data, sizeof(data));
|
|
|
|
for (i = 0; i < 12; i++)
|
|
{
|
|
n = tree_create();
|
|
if (!n) break;
|
|
tree_insert(root, tree_csize(root));
|
|
tree_attach(root, tree_csize(root) - 1, n);
|
|
data = i; tree_set_data(n, &data, sizeof(data));
|
|
}
|
|
|
|
t = tree_to(root, 3);
|
|
for (i = 0; i < 8; i++)
|
|
{
|
|
n = tree_create();
|
|
if (!n) break;
|
|
tree_insert(t, tree_csize(t));
|
|
tree_attach(t, tree_csize(t) - 1, n);
|
|
data = i + 30; tree_set_data(n, &data, sizeof(data));
|
|
}
|
|
|
|
t = tree_to(root, 3, 6);
|
|
for (i = 0; i < 4; i++)
|
|
{
|
|
n = tree_create();
|
|
if (!n) break;
|
|
tree_insert(t, tree_csize(t));
|
|
tree_attach(t, tree_csize(t) - 1, n);
|
|
data = i + 360; tree_set_data(n, &data, sizeof(data));
|
|
}
|
|
|
|
printf("tree size %d\r\n", tree_size(root));
|
|
printf("tree depth %d\r\n", tree_depth(root));
|
|
/* show */
|
|
tree_print(root, 0, print_int);
|
|
tree_delete(root, NULL);
|
|
}
|
|
|
|
static tree_t bitree_gen(int depth)
|
|
{
|
|
tree_t tree;
|
|
if (depth <= 0) return NULL;
|
|
tree = tree_create();
|
|
tree_insert(tree, 0);
|
|
tree_insert(tree, 0);
|
|
tree_attach(tree, 0, bitree_gen(depth - 1));
|
|
tree_attach(tree, 1, bitree_gen(depth - 1));
|
|
return tree;
|
|
}
|
|
|
|
static void test_bitree(void)
|
|
{
|
|
tree_t tree = bitree_gen(5);
|
|
tree_set_data(tree_to(tree, 0, 0, 0), "123", strlen("123") + 1);
|
|
tree_print(tree, 0, print_bitree);
|
|
tree_delete(tree, NULL);
|
|
}
|
|
|
|
static void test_base(void)
|
|
{
|
|
// test_common();
|
|
test_dir();
|
|
// test_ini();
|
|
// test_bitree();
|
|
}
|
|
|
|
/************************************************************************************/
|
|
/************************************* Command ************************************/
|
|
/************************************************************************************/
|
|
|
|
static void usage(void)
|
|
{
|
|
printf(
|
|
"Usage: tree [opt] [arg] ...\n"
|
|
"\n"
|
|
"options:\n"
|
|
" -e <execute> Specifies the function to execute, the default is the <base> test\n"
|
|
" <base> Test base function\n"
|
|
" <ut> Unit test\n"
|
|
" <create> Test create and delete functions\n"
|
|
" <set> Test set and get data function\n"
|
|
" <insert> Test insert and erase child function\n"
|
|
" <attach> Test attach and dettach child function\n"
|
|
" <parent> Test get parent and child node function\n"
|
|
" <attr> Test set and get attrubute function\n"
|
|
" <depth> Test get tree depth function\n"
|
|
" <to> Test locate to specify node function\n"
|
|
" -h Print help\n"
|
|
" -v Print version\n"
|
|
" -u [<period>] Unit test period, unit ms, the default is 1000ms\n"
|
|
"\n"
|
|
|
|
);
|
|
}
|
|
|
|
static int test(int argc, char *argv[])
|
|
{
|
|
char *execute = NULL;
|
|
int ut_period = 1000;
|
|
|
|
/* reset getopt */
|
|
command_opt_init();
|
|
|
|
while (1)
|
|
{
|
|
int opt = command_getopt(argc, argv, "e:hvu::");
|
|
if (opt == -1) break;
|
|
|
|
switch (opt)
|
|
{
|
|
case 'u' :
|
|
if (command_optarg) ut_period = atoi(command_optarg);
|
|
break;
|
|
case 'e' :
|
|
execute = command_optarg;
|
|
break;
|
|
case 'v' :
|
|
printf("tree version %d.%d.%d\r\n", TREE_V_MAJOR, TREE_V_MINOR, TREE_V_PATCH);
|
|
return 0;
|
|
case '?':
|
|
printf("Unknown option `%c`\r\n", command_optopt);
|
|
return -1;
|
|
case 'h' :
|
|
default:
|
|
usage();
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
if (execute)
|
|
{
|
|
if (!strcmp(execute, "base"))
|
|
{
|
|
test_base();
|
|
}
|
|
else if (!strcmp(execute, "ut"))
|
|
{
|
|
srand((unsigned int)time(NULL));
|
|
#if defined(TEST_TARGET_tree)
|
|
while (1)
|
|
{
|
|
unitt_task();
|
|
usleep(1000 * ut_period);
|
|
}
|
|
#else
|
|
printf("create task %d\r\n", task_create(ut_period, unitt_task));
|
|
#endif
|
|
}
|
|
else if (!strcmp(execute, "create"))
|
|
{
|
|
test_create();
|
|
}
|
|
else if (!strcmp(execute, "set"))
|
|
{
|
|
test_set();
|
|
}
|
|
else if (!strcmp(execute, "insert"))
|
|
{
|
|
test_insert();
|
|
}
|
|
else if (!strcmp(execute, "attach"))
|
|
{
|
|
test_attach();
|
|
}
|
|
else if (!strcmp(execute, "parent"))
|
|
{
|
|
test_parent();
|
|
}
|
|
else if (!strcmp(execute, "attr"))
|
|
{
|
|
test_attr();
|
|
}
|
|
else if (!strcmp(execute, "depth"))
|
|
{
|
|
test_depth();
|
|
}
|
|
else if (!strcmp(execute, "to"))
|
|
{
|
|
test_to();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
test_base();
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/************************************************************************************/
|
|
/************************************ Test entry ************************************/
|
|
/************************************************************************************/
|
|
|
|
#if defined(TEST_TARGET_tree)
|
|
int main(int argc, char *argv[])
|
|
{
|
|
return test(argc, argv);
|
|
}
|
|
#else
|
|
void test_tree(void)
|
|
{
|
|
command_export("tree", test);
|
|
}
|
|
init_export_app(test_tree);
|
|
#endif
|