intrusive_list.h

Go to the documentation of this file.

/*
    Copyright (c) 2005-2020 Intel Corporation

    Licensed under the Apache License, Version 2.0 (the "License");
    you may not use this file except in compliance with the License.
    You may obtain a copy of the License at

        http://www.apache.org/licenses/LICENSE-2.0

    Unless required by applicable law or agreed to in writing, software
    distributed under the License is distributed on an "AS IS" BASIS,
    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    See the License for the specific language governing permissions and
    limitations under the License.
*/

#ifndef _TBB_intrusive_list_H
#define _TBB_intrusive_list_H

#include "tbb/tbb_stddef.h"
#include "tbb/internal/_template_helpers.h"

namespace tbb {
namespace internal {


struct intrusive_list_node {
    intrusive_list_node *my_prev_node,
                        *my_next_node;
#if TBB_USE_ASSERT
    intrusive_list_node () { my_prev_node = my_next_node = this; }
#endif /* TBB_USE_ASSERT */
};


template <class List, class T>
class intrusive_list_base {
    intrusive_list_node my_head;

    size_t my_size;

    static intrusive_list_node& node ( T& item ) { return List::node(item); }

    static T& item ( intrusive_list_node* node ) { return List::item(node); }

    static const T& item( const intrusive_list_node* node ) { return List::item(node); }

    template <typename DereferenceType>
    class iterator_impl {
        __TBB_STATIC_ASSERT((tbb::internal::is_same_type<DereferenceType, T>::value ||
                            tbb::internal::is_same_type<DereferenceType, const T>::value),
                            "Incorrect DereferenceType in iterator_impl");
        typedef typename tbb::internal::conditional<tbb::internal::is_same_type<DereferenceType, T>::value,
                                                    intrusive_list_node*,
                                                    const intrusive_list_node*>::type pointer_type;
    public:
        iterator_impl() : my_pos(NULL) {}

        iterator_impl( pointer_type pos ) : my_pos(pos) {}

        iterator_impl& operator=( const iterator_impl& other ) {
            if (this != &other) {
                my_pos = other.my_pos;
            }
            return *this;
        }

        iterator_impl& operator=( const T& val ) {
            my_pos = &node(val);
            return *this;
        }

        iterator_impl& operator++() {
            my_pos = my_pos->my_next_node;
            return *this;
        }

        iterator_impl operator++( int ) {
            iterator_impl it(*this);
            ++*this;
            return it;
        }

        iterator_impl& operator--() {
            my_pos = my_pos->my_prev_node;
            return *this;
        }

        iterator_impl operator--( int ) {
            iterator_impl it(*this);
            --*this;
            return it;
        }

        bool operator==( const iterator_impl& rhs ) const {
            return my_pos == rhs.my_pos;
        }

        bool operator!=( const iterator_impl& rhs ) const {
            return my_pos != rhs.my_pos;
        }

        DereferenceType& operator*() const {
            return intrusive_list_base::item(my_pos);
        }

        DereferenceType* operator->() const {
            return &intrusive_list_base::item(my_pos);
        }

    private:
        // Node the iterator points to at the moment
        pointer_type my_pos;
    }; // class iterator_impl

    void assert_ok () const {
        __TBB_ASSERT( (my_head.my_prev_node == &my_head && !my_size) ||
                      (my_head.my_next_node != &my_head && my_size >0), "intrusive_list_base corrupted" );
#if TBB_USE_ASSERT >= 2
        size_t i = 0;
        for ( intrusive_list_node *n = my_head.my_next_node; n != &my_head; n = n->my_next_node )
            ++i;
        __TBB_ASSERT( my_size == i, "Wrong size" );
#endif /* TBB_USE_ASSERT >= 2 */
    }

public:
    typedef iterator_impl<T> iterator;
    typedef iterator_impl<const T> const_iterator;

    intrusive_list_base () : my_size(0) {
        my_head.my_prev_node = &my_head;
        my_head.my_next_node = &my_head;
    }

    bool empty () const { return my_head.my_next_node == &my_head; }

    size_t size () const { return my_size; }

    iterator begin () { return iterator(my_head.my_next_node); }

    iterator end () { return iterator(&my_head); }

    const_iterator begin () const { return const_iterator(my_head.my_next_node); }

    const_iterator end () const { return const_iterator(&my_head); }

    void push_front ( T& val ) {
        __TBB_ASSERT( node(val).my_prev_node == &node(val) && node(val).my_next_node == &node(val),
                    "Object with intrusive list node can be part of only one intrusive list simultaneously" );
        // An object can be part of only one intrusive list at the given moment via the given node member
        node(val).my_prev_node = &my_head;
        node(val).my_next_node = my_head.my_next_node;
        my_head.my_next_node->my_prev_node = &node(val);
        my_head.my_next_node = &node(val);
        ++my_size;
        assert_ok();
    }

    void remove( T& val ) {
        __TBB_ASSERT( node(val).my_prev_node != &node(val) && node(val).my_next_node != &node(val), "Element to remove is not in the list" );
        __TBB_ASSERT( node(val).my_prev_node->my_next_node == &node(val) && node(val).my_next_node->my_prev_node == &node(val), "Element to remove is not in the list" );
        --my_size;
        node(val).my_next_node->my_prev_node = node(val).my_prev_node;
        node(val).my_prev_node->my_next_node = node(val).my_next_node;
#if TBB_USE_ASSERT
        node(val).my_prev_node = node(val).my_next_node = &node(val);
#endif
        assert_ok();
    }

    iterator erase ( iterator it ) {
        T& val = *it;
        ++it;
        remove( val );
        return it;
    }

}; // intrusive_list_base



template <class T, class U, intrusive_list_node U::*NodePtr>
class memptr_intrusive_list : public intrusive_list_base<memptr_intrusive_list<T, U, NodePtr>, T>
{
    friend class intrusive_list_base<memptr_intrusive_list<T, U, NodePtr>, T>;

    static intrusive_list_node& node ( T& val ) { return val.*NodePtr; }

    static T& item ( intrusive_list_node* node ) {
        // Cannot use __TBB_offsetof (and consequently __TBB_get_object_ref) macro
        // with *NodePtr argument because gcc refuses to interpret pasted "->" and "*"
        // as member pointer dereferencing operator, and explicit usage of ## in
        // __TBB_offsetof implementation breaks operations with normal member names.
        return *reinterpret_cast<T*>((char*)node - ((ptrdiff_t)&(reinterpret_cast<T*>(0x1000)->*NodePtr) - 0x1000));
    }

    static const T& item( const intrusive_list_node* node ) {
        return item(const_cast<intrusive_list_node*>(node));
    }
}; // intrusive_list<T, U, NodePtr>


template <class T>
class intrusive_list : public intrusive_list_base<intrusive_list<T>, T>
{
    friend class intrusive_list_base<intrusive_list<T>, T>;

    static intrusive_list_node& node ( T& val ) { return val; }

    static T& item ( intrusive_list_node* node ) { return *static_cast<T*>(node); }
    static const T& item( const intrusive_list_node* node ) { return *static_cast<const T*>(node); }
}; // intrusive_list<T>

} // namespace internal
} // namespace tbb

#endif /* _TBB_intrusive_list_H */