path: root/cpukit/score/src/rbtreefind.c

/**
 * @file
 *
 * @brief Find the control structure of the tree containing the given node
 * @ingroup ScoreRBTree
 */

/*
 *  Copyright (c) 2010 Gedare Bloom.
 *
 *  The license and distribution terms for this file may be
 *  found in the file LICENSE in this distribution or at
 *  http://www.rtems.org/license/LICENSE.
 */

#if HAVE_CONFIG_H
#include "config.h"
#endif

#include <rtems/score/rbtreeimpl.h>

RBTree_Node *_RBTree_Find(
  const RBTree_Control *the_rbtree,
  const RBTree_Node    *the_node,
  RBTree_Compare        compare,
  bool                  is_unique
)
{
  RBTree_Node *iter_node = the_rbtree->root;
  RBTree_Node *found = NULL;

  while ( iter_node != NULL ) {
    RBTree_Compare_result compare_result = ( *compare )( the_node, iter_node );
    RBTree_Direction      dir;

    if ( _RBTree_Is_equal( compare_result ) ) {
      found = iter_node;

      if ( is_unique )
        break;
    }

    dir = (RBTree_Direction) _RBTree_Is_greater( compare_result );
    iter_node = iter_node->child[ dir ];
  }

  return found;
}