dlib C++ Library - set_utils.h

    // Copyright (C) 2007  Davis E. King (davis@dlib.net)
    // License: Boost Software License   See LICENSE.txt for the full license.
    #ifndef DLIB_SET_UTILs_
    #define DLIB_SET_UTILs_

    #include "../algs.h"
    #include "set_utils_abstract.h"

    namespace dlib
    {

    // ----------------------------------------------------------------------------------------

        template <
            typename T,
            typename U
            >
        unsigned long set_intersection_size (
            const T& a,
            const U& b
        )
        {
            if (is_same_object(a,b))
                return a.size();

            unsigned long num = 0;

            if (a.size() < b.size())
            {
                a.reset();
                while (a.move_next())
                {
                    if (b.is_member(a.element()))
                        ++num;
                }
            }
            else
            {
                b.reset();
                while (b.move_next())
                {
                    if (a.is_member(b.element()))
                        ++num;
                }
            }

            return num;
        }

    // ----------------------------------------------------------------------------------------

        template <
            typename T,
            typename U,
            typename V
            >
        void set_union (
            const T& a,
            const U& b,
            V& u
        )
        {
            typedef typename T::type type;
            if (is_same_object(a,u) || is_same_object(b,u))
            {
                V local_u;
                type temp;
                a.reset();
                while (a.move_next())
                {
                    temp = a.element();
                    local_u.add(temp);
                }

                b.reset();
                while (b.move_next())
                {
                    if (a.is_member(b.element()) == false)
                    {
                        temp = b.element();
                        local_u.add(temp);
                    }
                }

                local_u.swap(u);
            }
            else
            {
                u.clear();

                type temp;
                a.reset();
                while (a.move_next())
                {
                    temp = a.element();
                    u.add(temp);
                }

                b.reset();
                while (b.move_next())
                {
                    if (a.is_member(b.element()) == false)
                    {
                        temp = b.element();
                        u.add(temp);
                    }
                }
            }
        }

    // ----------------------------------------------------------------------------------------

        template <
            typename T,
            typename U,
            typename V
            >
        void set_intersection (
            const T& a,
            const U& b,
            V& i
        )
        {
            typedef typename T::type type;
            if (is_same_object(a,i) || is_same_object(b,i))
            {
                V local_i;

                type temp;

                if (a.size() < b.size())
                {
                    a.reset();
                    while (a.move_next())
                    {
                        if (b.is_member(a.element()))
                        {
                            temp = a.element();
                            local_i.add(temp);
                        }
                    }
                }
                else
                {
                    b.reset();
                    while (b.move_next())
                    {
                        if (a.is_member(b.element()))
                        {
                            temp = b.element();
                            local_i.add(temp);
                        }
                    }
                }

                local_i.swap(i);
            }
            else
            {
                i.clear();
                type temp;

                if (a.size() < b.size())
                {
                    a.reset();
                    while (a.move_next())
                    {
                        if (b.is_member(a.element()))
                        {
                            temp = a.element();
                            i.add(temp);
                        }
                    }
                }
                else
                {
                    b.reset();
                    while (b.move_next())
                    {
                        if (a.is_member(b.element()))
                        {
                            temp = b.element();
                            i.add(temp);
                        }
                    }
                }
            }
        }

    // ----------------------------------------------------------------------------------------

        template <
            typename T,
            typename U,
            typename V
            >
        void set_difference (
            const T& a,
            const U& b,
            V& d
        )
        {
            typedef typename T::type type;
            if (is_same_object(a,d) || is_same_object(b,d))
            {
                V local_d;

                type temp;

                a.reset();
                while (a.move_next())
                {
                    if (b.is_member(a.element()) == false)
                    {
                        temp = a.element();
                        local_d.add(temp);
                    }
                }

                local_d.swap(d);
            }
            else
            {
                d.clear();
                type temp;

                a.reset();
                while (a.move_next())
                {
                    if (b.is_member(a.element()) == false)
                    {
                        temp = a.element();
                        d.add(temp);
                    }
                }
            }
        }

    // ----------------------------------------------------------------------------------------

    }

    #endif // DLIB_SET_UTILs_