Subversion Repositories HelenOS

Compare Revisions

Ignore whitespace Rev 2415 → Rev 2416

/branches/rcu/kernel/test/extavltree/extavltree1.def
0,0 → 1,6
{
"extavltree1",
"Test Extended Avl tree operations",
&test_extavltree1,
true
},
/branches/rcu/kernel/test/extavltree/extavltree1.c
0,0 → 1,427
/*
* Copyright (c) 2007 Vojtech Mencl
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*
* - Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* - Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* - The name of the author may not be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
 
#include <test.h>
#include <print.h>
#include <adt/extavl.h>
#include <debug.h>
 
#include <panic.h>
 
 
#define NODE_COUNT 100
 
/*
* wrapper structure with a pointer to extended avl tree root
*/
static extavltree_t exttree;
 
/*
* extavltree nodes in array for faster allocating
*/
static extavltree_node_t extavltree_nodes[NODE_COUNT];
 
/*
* head of free nodes' list:
*/
static extavltree_node_t *first_free_node = NULL;
 
 
 
static int exttree_test_height(extavltree_node_t *node,bool timeout);
static extavltree_node_t *exttree_test_parents(extavltree_node_t *node);
static void print_exttree_structure_flat (extavltree_node_t *node, int level);
static bool exttree_test_link(int node_count);
static void print_exttree_link(int node_count);
static extavltree_node_t *alloc_extavltree_node(void);
 
 
 
//vraci hloubku stromu
static int exttree_test_height(extavltree_node_t *node,bool timeout)
{
int h1, h2;
 
if (!node) return -1;
 
h1 = exttree_test_height(node->lft,timeout) + 1;
if (!timeout && node->lft_height != h1) {
printf("Bad height: %d of LEFT subtree in node: %d with address: %p\n",h1,node->key,node);
}
h2 = exttree_test_height(node->rgt,0) + 1;
if (node->rgt_height != h2) {
printf("Bad height: %d of RIGHT subtree in node: %d with address: %p\n",h2,node->key,node);
}
if (!timeout && (((h1-h2)>0?h1-h2:h2-h1) > 1)) {
printf("Bad height: error in definition of avltree: %d with address: %p\n",node->key,node);
}
 
return (node->lft_height > node->rgt_height? node->lft_height: node->rgt_height);
}
 
 
/** Tests par atribute of every tree node
*
*/
static extavltree_node_t *exttree_test_parents(extavltree_node_t *node)
{
extavltree_node_t *tmp;
if (!node) return NULL;
 
if (node->lft) {
tmp = exttree_test_parents(node->lft);
if (tmp != node) {
printf("Bad parent pointer at node with key: %d, address: %p\n",tmp->key,node->lft);
}
}
if (node->rgt) {
tmp = exttree_test_parents(node->rgt);
if (tmp != node) {
printf("Bad parent pointer at node with key: %d, address: %p\n",tmp->key,node->rgt);
}
}
return node->par;
}
 
/** Checks list of nodes
*
*/
static bool exttree_test_link(int node_count)
{
extavltree_node_t *node;
int i;
bool test_link = true;
 
for (i = 0,node = exttree.head.next; node != &(exttree.head); i++,node = node->next) {
if ((node->next != &(exttree.head)) && (node->key > node->next->key)) {
printf("\nList is not sorted (forward direction) at key: %d\n",node->key);
test_link = false;
}
}
if (node_count && i != node_count) {
printf("\nBad node count!!! Counted: %d, right number: %d", i, node_count);
test_link = false;
}
for (i = 0,node = exttree.head.prev; node != &(exttree.head); i++,node = node->prev) {
if ((node->prev != &(exttree.head)) && (node->key < node->prev->key)) {
printf("\nList is not sorted (backward direction) at key: %d\n",node->key);
test_link = false;
}
}
if (node_count && i != node_count) {
printf("\nBad node count!!! Counted: %d, right number: %d", i, node_count);
test_link = false;
}
return test_link;
}
 
/** Prints the structure of node, which is level levels from the top of the tree.
*
*/
static void print_exttree_structure_flat (extavltree_node_t *node, int level)
{
extavltree_node_t *tmp;
int i;
 
if (level > 16)
{
printf ("[...]");
return;
}
 
if (node == NULL)
return;
for (tmp = node,i = 0; tmp->key == node->key; tmp = tmp->next,i++)
;
 
printf ("%d[%d,%d,(%d)]", node->key,node->lft_height,node->rgt_height,i);
if (node->lft != NULL || node->rgt != NULL)
{
printf("(");
 
print_exttree_structure_flat (node->lft, level + 1);
if (node->rgt != NULL)
{
printf(",");
print_exttree_structure_flat (node->rgt, level + 1);
}
 
printf(")");
}
}
 
/** Prints list of nodes
*
*/
static void print_exttree_link(int node_count)
{
extavltree_node_t *node;
printf("\n");
for (node = exttree.head.next; node != &(exttree.head); node = node->next) {
printf(" %d,",node->key);
}
for (node = exttree.head.prev; node != &(exttree.head); node = node->prev) {
printf(" %d,",node->key);
}
}
 
//****************************************************************
static void alloc_extavltree_node_prepare(void)
{
int i;
 
for (i = 0; i < NODE_COUNT - 1; i++) {
extavltree_nodes[i].next = &(extavltree_nodes[i+1]);
}
/*
* Node keys which will be used for insertion. Up to NODE_COUNT size of array.
*/
 
// First tree node and same key
extavltree_nodes[0].key = 60;
extavltree_nodes[1].key = 60;
extavltree_nodes[2].key = 60;
//LL rotation
extavltree_nodes[3].key = 50;
extavltree_nodes[4].key = 40;
extavltree_nodes[5].key = 30;
//LR rotation
extavltree_nodes[6].key = 20;
extavltree_nodes[7].key = 20;
extavltree_nodes[8].key = 25;
extavltree_nodes[9].key = 25;
//LL rotation in lower floor
extavltree_nodes[10].key = 35;
//RR rotation
extavltree_nodes[11].key = 70;
extavltree_nodes[12].key = 80;
//RL rotation
extavltree_nodes[13].key = 90;
extavltree_nodes[14].key = 85;
extavltree_nodes[15].key = 100;
extavltree_nodes[16].key = 200;
extavltree_nodes[17].key = 300;
extavltree_nodes[18].key = 400;
extavltree_nodes[19].key = 500;
extavltree_nodes[20].key = 600;
 
for (i = 21; i < NODE_COUNT; i++)
extavltree_nodes[i].key = i * 3;
extavltree_nodes[i].next = NULL;
first_free_node = &(extavltree_nodes[0]);
}
 
static extavltree_node_t *alloc_extavltree_node(void)
{
extavltree_node_t *node;
 
node = first_free_node;
first_free_node = first_free_node->next;
 
return node;
}
//****************************************************************
 
static void test_exttree_insert(extavltree_t *tree, unsigned int node_count, int quiet)
{
unsigned int i;
extavltree_node_t *newnode;
 
/*
* Initialize tree before using.
*/
extavltree_create(tree);
if (!quiet) printf("\n\nInserting %d nodes ...\n", node_count);
 
for (i = 0; i < node_count; i++) {
newnode = alloc_extavltree_node();
//if (!quiet) printf("[[[%d]]]\n",newnode->key);
extavltree_insert(tree, newnode);
if (!quiet) {
if (!exttree_test_link(i+1)) {
print_exttree_link(i+1);
printf("\n");
}
exttree_test_parents(tree->root);
exttree_test_height(tree->root,1);
}
}
if (!quiet) printf("Inserting was finished\n");
}
 
/*
static extavltree_node_t *exttree_random_delete_node(extavltree_t *tree, int node_count, int r, bool quiet)
{
extavltree_node_t *delnode;
int i;
 
for (i = 0,delnode = tree->head.next; i < (r-1); i++)
delnode = delnode->next;
if (delnode == &tree->head) {
if (!quiet) printf("Try to delete head! Node count: %d, number of deleted node: %d\n",node_count,r);
return NULL;
}
 
extavltree_delete(tree, delnode);
 
return delnode;
}
*/
 
static void test_exttree_delete(extavltree_t *tree, int node_count, int node_position, bool quiet)
{
extavltree_node_t *delnode;
unsigned int i;
//aktualni pocet tiku:
if (!quiet) printf("Deleting tree...\n");
 
switch(node_position) {
case 0: //mazani vzdy korene
if (!quiet) printf("\n\nDelete root nodes\n");
i = node_count;
while(tree->root != NULL) {
delnode = tree->root;
extavltree_delete(tree,delnode);
if (!quiet) {
if (!exttree_test_link(i)) {
print_exttree_link(i);
printf("\n");
}
exttree_test_parents(tree->root);
exttree_test_height(tree->root,1);
}
i--;
}
break;
case 1:
if (!quiet) printf("\n\nDelete nodes according to their time of origin\n");
for (i = 0; i < node_count; i++) {
extavltree_delete(tree,&(extavltree_nodes[i]));
if (!quiet) {
if (!exttree_test_link(i+1)) {
print_exttree_link(i+1);
printf("\n");
}
exttree_test_parents(tree->root);
exttree_test_height(tree->root,1);
}
}
 
break;
}
if (!quiet) printf("Deletion was finished\n");
}
 
static void timeout_exttree(extavltree_t *tree, int node_count, bool quiet)
{
int i = node_count;
if (!quiet) printf("Timeout tree ...\n");
while(tree->head.next != &(tree->head)) {
extavltree_delete_min(tree);
if (!quiet) {
if (!exttree_test_link(i)) {
print_exttree_link(i);
printf("\n");
}
exttree_test_parents(tree->root);
exttree_test_height(tree->root,1);
}
i--;
}
 
if (!quiet && (i != 0)) printf("Bad node count. Some nodes have been lost!");
 
if (!quiet) printf("Timeout tree finished\n");
}
 
/*
void timeout_exttree_run(extavltree_t *tree, int operation_count, int verbal)
{
int i;
extavltree_node_t *node;
int r;
int count;
//inicializace stromu:
extavltree_create(tree);
for(i = 0, count = 0; i < operation_count; i++) {
if (tree->count && ((rand() % NODE_COUNT) <= tree->count)) {
if ((r = rand()) % DELETE_PROB == 1) { //mazu nahodne
node = exttree_random_delete_node(tree,(r % tree->count));
//printf("DELETE key: %d, number: %d,address: %p\n",node->key,r % (tree->count+1),node);
node->next = first_free_node;
first_free_node = node;
} else {
node = tree->head.next;
extavltree_delete_min(tree);
//printf("TIMEOUT key: %d, address: %p\n",node->key,node);
node->next = first_free_node;
first_free_node = node;
}
} else {
node = alloc_extavltree_node_random();
//printf("INSERT key: %d, address: %p\n",node->key + tree->basetime,node);
extavltree_insert(tree, node);
}
//test_exttree_height(tree->root,1);
//exttree_test_parents(tree->root);
//print_exttree_link(tree->count);
//print_exttree_structure_flat(tree->root,0); putchar('\n'); putchar('\n');
}
}
*/
 
char * test_extavltree1(bool quiet)
{
alloc_extavltree_node_prepare();
test_exttree_insert(&exttree, NODE_COUNT, quiet);
test_exttree_delete(&exttree, NODE_COUNT, 0, quiet);
 
alloc_extavltree_node_prepare();
test_exttree_insert(&exttree, NODE_COUNT, quiet);
test_exttree_delete(&exttree, NODE_COUNT, 1, quiet);
 
alloc_extavltree_node_prepare();
test_exttree_insert(&exttree, NODE_COUNT, quiet);
timeout_exttree(&exttree, NODE_COUNT, quiet);
 
return NULL;
}