static void initialise_rbt(GtHuffman *huffman,
const void *distr,
GtDistrFunc distrfunc)
{
GtHuffmanTree *huffptr, GT_UNUSED *huffptr2;
unsigned long i;
bool nodecreated = false;
huffman->numofsymbols = 0;
huffman->rbt_root = gt_rbtree_new(huffman_tree_cmp,
huffman_tree_delete,
NULL);
gt_assert(huffman->rbt_root);
for (i = 0; i < huffman->totalnumofsymbols; i++) {
if (distrfunc(distr, i) > 0) {
huffptr = huffman_tree_new(i, distrfunc(distr, i));
huffptr2 = (GtHuffmanTree*)gt_rbtree_search(huffman->rbt_root,
huffptr,
&nodecreated);
gt_assert(nodecreated && huffptr2);
huffman->numofsymbols++;
}
}
}
static int make_huffman_tree(GtHuffman *huffman)
{
GtHuffmanTree *n1 = NULL,
*n2 = NULL,
GT_UNUSED *n3 = NULL,
*newnode = NULL;
unsigned long i,
symbol;
unsigned long long freq;
int GT_UNUSED deleted;
bool nodecreated = false;
if (huffman->numofsymbols == 0)
huffman->roothuffmantree = NULL;
else if (huffman->numofsymbols == 1) {
huffman->roothuffmantree =
gt_rbtree_root_key(huffman->rbt_root);
huffman->roothuffmantree->code.code = 0;
huffman->roothuffmantree->code.numofbits = 1;
}
else {
for (i = 0; i < huffman->numofsymbols - 1; i++) {
n1 = gt_rbtree_minimum_key(huffman->rbt_root);
n1 = huffman_tree_ref(n1);
deleted = gt_rbtree_erase(huffman->rbt_root,
n1);
gt_assert(deleted == 0);
n2 = gt_rbtree_minimum_key(huffman->rbt_root);
n2 = huffman_tree_ref(n2);
deleted = gt_rbtree_erase(huffman->rbt_root,
n2);
gt_assert(deleted == 0);
symbol = n1->symbol.symbol
< n2->symbol.symbol ? n2->symbol.symbol : n1->symbol.symbol;
freq = n1->symbol.freq + n2->symbol.freq;
newnode = huffman_tree_new(symbol, freq);
if (n1->symbol.freq < n2->symbol.freq) {
newnode->leftchild = n2;
newnode->rightchild = n1;
}
else {
newnode->leftchild = n1;
newnode->rightchild = n2;
}
gt_assert(huffman->rbt_root);
n3 = gt_rbtree_search(huffman->rbt_root, newnode, &nodecreated);
gt_assert(nodecreated && n3);
}
huffman->roothuffmantree = (GtHuffmanTree*)newnode;
huffman->roothuffmantree->code.code = 0;
huffman->roothuffmantree->code.numofbits = 0;
}
return 0;
}
static GtRBTree *seq_decoder_init_file_info(FastqFileInfo *fileinfos,
GtUword num_off_files)
{
GtRBTree *tree;
bool nodecreated = false;
GtUword i;
tree = gt_rbtree_new(hcr_cmp_FastqFileInfo, NULL, NULL);
for (i = 0; i < num_off_files; i++) {
(void) gt_rbtree_search(tree, &fileinfos[i], &nodecreated);
}
gt_assert(nodecreated);
return tree;
}
static int hmmsearch_process_seq(void *data,
GtUword seqnum,
GT_UNUSED GtError *err)
{
int had_err = 0;
bool nodecreated = false;
GtUword *seqnum_p = gt_malloc(sizeof (*seqnum_p));
GtRBTree *tree = data;
*seqnum_p = seqnum;
(void) gt_rbtree_search(tree,
seqnum_p,
&nodecreated);
if (!nodecreated)
gt_free(seqnum_p);
return had_err;
}
static bool nrbt_mangle_tree(GtRBTreeSearchOrder how, GtRBTreeDoAction what,
GtRBTree *tree, GtUword lag, GtError *err)
{
GtUword i;
bool nodecreated, haserr = false;
if (how == GT_RBTREE_RANDOMORDER) {
for (i = 0; i < (GtUword) GT_RBTREE_SIZE; ++i) {
gt_rbtree_ytab[i] = i;
}
gt_rbtree_permuteintarray (gt_rbtree_ytab);
}
for (i = 0; i < GT_RBTREE_SIZE + lag; ++i) {
void *elem;
GtUword j, k;
switch (how) {
case GT_RBTREE_RANDOMORDER:
if (i >= lag) {
k = gt_rbtree_ytab[i - lag];
} else {
k = gt_rbtree_ytab[GT_RBTREE_SIZE - i - 1 + lag];
}
j = gt_rbtree_ytab[i];
break;
case GT_RBTREE_ASCENDING:
k = i - lag;
j = i;
break;
case GT_RBTREE_DESCENDING:
k = GT_RBTREE_SIZE - i - 1 + lag;
j = GT_RBTREE_SIZE - i - 1;
break;
default:
abort ();
}
switch (what) {
case GtRBTreeBuildDelete:
case GtRBTreeBuild:
if (i < (GtUword) GT_RBTREE_SIZE) {
if (gt_rbtree_find (tree, gt_rbtree_xtab + j) != NULL) {
gt_error_set(err,"Found element which is not in tree yet");
haserr = true;
}
if (!haserr) {
elem = gt_rbtree_search(tree, gt_rbtree_xtab + j, &nodecreated);
if (elem == NULL || gt_rbtree_find(tree,
gt_rbtree_xtab + j) == NULL) {
gt_error_set(err,"Couldn't find element after it was added");
haserr = true;
}
}
}
if (haserr || what ==GtRBTreeBuild || i < lag)
break;
j = k;
/*@fallthrough@*/
case GtRBTreeDelete:
elem = gt_rbtree_find (tree, gt_rbtree_xtab + j);
if (elem == NULL || gt_rbtree_erase (tree, gt_rbtree_xtab + j) != 0) {
gt_error_set(err,"Error deleting element");
haserr = true;
}
break;
case GtRBTreeFind:
if (gt_rbtree_find (tree, gt_rbtree_xtab + j) == NULL) {
gt_error_set(err,"Couldn't find element after it was added");
haserr = true;
}
break;
}
}
return haserr;
}
