dlib C++ Library - binary_search_tree_kernel_abstract.h

    // Copyright (C) 2003  Davis E. King (davis@dlib.net)
    // License: Boost Software License   See LICENSE.txt for the full license.
    #undef DLIB_BINARY_SEARCH_TREE_KERNEl_ABSTRACT_
    #ifdef DLIB_BINARY_SEARCH_TREE_KERNEl_ABSTRACT_

    #include "../interfaces/map_pair.h"
    #include "../interfaces/enumerable.h"
    #include "../interfaces/remover.h"
    #include "../serialize.h"
    #include "../algs.h"
    #include <functional>

    namespace dlib
    {

        template <
            typename domain,
            typename range,
            typename mem_manager = default_memory_manager,
            typename compare = std::less<domain>
            >
        class binary_search_tree : public enumerable<map_pair<domain,range> >,
                                   public asc_pair_remover<domain,range,compare>
        {

            /*!
                REQUIREMENTS ON domain
                    domain must be comparable by compare where compare is a functor compatible with std::less and
                    domain is swappable by a global swap() and
                    domain must have a default constructor

                REQUIREMENTS ON range
                    range is swappable by a global swap() and
                    range must have a default constructor

                REQUIREMENTS ON mem_manager
                    must be an implementation of memory_manager/memory_manager_kernel_abstract.h or
                    must be an implementation of memory_manager_global/memory_manager_global_kernel_abstract.h or
                    must be an implementation of memory_manager_stateless/memory_manager_stateless_kernel_abstract.h
                    mem_manager::type can be set to anything.

                POINTERS AND REFERENCES TO INTERNAL DATA
                    swap(), count(), height(),  and operator[] functions
                    do not invalidate pointers or references to internal data.

                    position_enumerator() invalidates pointers or references to
                    data returned by element() and only by element() (i.e. pointers and
                    references returned by operator[] are still valid).

                    All other functions have no such guarantees.

                INITIAL VALUE
                    size() == 0
                    height() == 0

                ENUMERATION ORDER
                    The enumerator will iterate over the domain (and each associated
                    range element) elements in ascending order according to the compare functor.
                    (i.e. the elements are enumerated in sorted order)

                WHAT THIS OBJECT REPRESENTS
                    this object represents a data dictionary that is built on top of some
                    kind of binary search tree.  It maps objects of type domain to objects
                    of type range.

                    Also note that unless specified otherwise, no member functions
                    of this object throw exceptions.

                    NOTE:
                        definition of equivalent:
                        a is equivalent to b if
                        a < b == false and
                        b < a == false
            !*/

        public:

            typedef domain domain_type;
            typedef range range_type;
            typedef compare compare_type;
            typedef mem_manager mem_manager_type;

            binary_search_tree(
            );
            /*!
                ensures
                    - #*this is properly initialized
                throws
                    - std::bad_alloc or any exception thrown by domain's or range's
                      constructor.
            !*/

            virtual ~binary_search_tree(
            );
            /*!
                ensures
                    - all memory associated with *this has been released
            !*/

            void clear(
            );
            /*!
                ensures
                    - #*this has its initial value
                throws
                    - std::bad_alloc or any exception thrown by domain's or range's
                      constructor.
                        if this exception is thrown then *this is unusable
                        until clear() is called and succeeds
            !*/

            short height (
            ) const;
            /*!
                ensures
                    - returns the number of elements in the longest path from the root
                      of the tree to a leaf
            !*/

            unsigned long count (
                const domain& d
            ) const;
            /*!
                ensures
                    - returns the number of elements in the domain of *this that are
                      equivalent to d
            !*/

            void add (
                domain& d,
                range& r
            );
            /*!
                requires
                    - &d != &r (i.e. d and r cannot be the same variable)
                ensures
                    - adds a mapping between d and r to *this
                    - if (count(d) == 0) then
                        - #*(*this)[d] == r
                    - else
                        - #(*this)[d] != 0
                    - #d and #r have initial values for their types
                    - #count(d) == count(d) + 1
                    - #at_start() == true
                    - #size() == size() + 1
                throws
                    - std::bad_alloc or any exception thrown by domain's or range's
                      constructor.
                        if add() throws then it has no effect
            !*/

            void remove (
                const domain& d,
                domain& d_copy,
                range& r
            );
            /*!
                requires
                    - (*this)[d] != 0
                    - &d != &r (i.e. d and r cannot be the same variable)
                    - &d != &d_copy (i.e. d and d_copy cannot be the same variable)
                    - &r != &d_copy (i.e. r and d_copy cannot be the same variable)
                ensures
                    - some element in the domain of *this that is equivalent to d has
                      been removed and swapped into #d_copy.  Additionally, its
                      associated range element has been removed and swapped into #r.
                    - #count(d) == count(d) - 1
                    - #size() == size() - 1
                    - #at_start() == true
            !*/

            void destroy (
                const domain& d
            );
            /*!
                requires
                    - (*this)[d] != 0
                ensures
                    - an element in the domain of *this equivalent to d has been removed.
                      The element in the range of *this associated with d has also been
                      removed.
                    - #count(d) == count(d) - 1
                    - #size() == size() - 1
                    - #at_start() == true
            !*/

            void remove_last_in_order (
                domain& d,
                range& r
            );
            /*!
                requires
                    - size() > 0
                ensures
                    - the last/biggest (according to the compare functor) element in the domain of *this has
                      been removed and swapped into #d.  The element in the range of *this
                      associated with #d has also been removed and swapped into #r.
                    - #count(#d) == count(#d) - 1
                    - #size() == size() - 1
                    - #at_start() == true
            !*/

            void remove_current_element (
                domain& d,
                range& r
            );
            /*!
                requires
                    - current_element_valid() == true
                ensures
                    - the current element given by element() has been removed and swapped into d and r.
                    - #d == element().key()
                    - #r == element().value()
                    - #count(#d) == count(#d) - 1
                    - #size() == size() - 1
                    - moves the enumerator to the next element.  If element() was the last
                      element in enumeration order then #current_element_valid() == false
                      and #at_start() == false.
            !*/

            void position_enumerator (
                const domain& d
            ) const;
            /*!
                ensures
                    - #at_start() == false
                    - if (count(d) > 0) then
                        - #element().key() == d
                    - else if (there are any items in the domain of *this that are bigger than
                      d according to the compare functor) then
                        - #element().key() == the smallest item in the domain of *this that is
                          bigger than d according to the compare functor.
                    - else
                        - #current_element_valid() == false
            !*/

            const range* operator[] (
                const domain& d
            ) const;
            /*!
                ensures
                    - if (there is an element in the domain equivalent to d) then
                        - returns a pointer to an element in the range of *this that
                          is associated with an element in the domain of *this
                          equivalent to d.
                    - else
                        - returns 0
            !*/

            range* operator[] (
                const domain& d
            );
            /*!
                ensures
                    - if (there is an element in the domain equivalent to d) then
                        - returns a pointer to an element in the range of *this that
                          is associated with an element in the domain of *this
                          equivalent to d.
                    - else
                        - returns 0
            !*/

            void swap (
                binary_search_tree& item
            );
            /*!
                ensures
                    - swaps *this and item
            !*/

        private:

            // restricted functions
            binary_search_tree(binary_search_tree&);
            binary_search_tree& operator=(binary_search_tree&);

        };

        template <
            typename domain,
            typename range,
            typename mem_manager,
            typename compare
            >
        inline void swap (
            binary_search_tree<domain,range,mem_manager,compare>& a,
            binary_search_tree<domain,range,mem_manager,compare>& b
        ) { a.swap(b); }
        /*!
            provides a global swap function
        !*/

        template <
            typename domain,
            typename range,
            typename mem_manager,
            typename compare
            >
        void deserialize (
            binary_search_tree<domain,range,mem_manager,compare>& item,
            std::istream& in
        );
        /*!
            provides deserialization support
        !*/
    }

    #endif // DLIB_BINARY_SEARCH_TREE_KERNEl_ABSTRACT_