Vertex.h

#pragma once


#include <libutl/slist.h>


UTL_NS_BEGIN;



class Vertex : public Object
{
    UTL_CLASS_DECL(Vertex, Object);

public:
    Vertex(const Object* object)
    {
        init(object);
    }

    virtual void copy(const Object& object);

    virtual const Object&
    getKey() const
    {
        if (_object == nullptr)
            return super::getKey();
        else
            return *_object;
    }

    void
    addDirectedEdge(const Vertex* vertex)
    {
        slist_add(_edgesHead, &_edgesTail, vertex);
    }

    void
    addUndirectedEdge(Vertex* vertex)
    {
        slist_add(_edgesHead, &_edgesTail, vertex);
        slist_add(vertex->_edgesHead, &vertex->_edgesTail, this);
    }

    void
    clear()
    {
        slist_clear(_edgesHead, &_edgesTail, false);
        setObject(nullptr);
    }

    const SlistNode*
    edges() const
    {
        return _edgesHead;
    }

    Vertex*
    find(const Object& key) const
    {
        return utl::cast<Vertex>(slist_find(_edgesHead, key));
    }

    Object*
    getObject() const
    {
        return _object;
    }

    bool
    isOwner() const
    {
        return ((_flags & flg_owner) != 0);
    }

    void
    setOwner(bool owner)
    {
        if (owner)
            _flags |= flg_owner;
        else
            _flags &= ~flg_owner;
    }

    bool
    isVisited() const
    {
        return ((_flags & flg_visited) != 0);
    }

    void
    setVisited(bool visited)
    {
        if (visited)
            _flags |= flg_visited;
        else
            _flags &= ~flg_visited;
    }

    bool
    removeDirectedEdge(const Vertex* vertex)
    {
        return slist_remove(_edgesHead, &_edgesTail, *vertex, false);
    }

    bool
    removeUndirectedEdge(Vertex* vertex)
    {
        return slist_remove(_edgesHead, &_edgesTail, *vertex, false, true)
               && slist_remove(vertex->_edgesHead, &vertex->_edgesTail, self, false, true);
    }

    void
    setObject(const Object* object)
    {
        if (isOwner()) delete _object;
        _object = const_cast<Object*>(object);
    }

private:
    enum flg_t
    {
        flg_visited = 1,
        flg_owner = 2
    };

private:
    void
    init(const Object* object = nullptr)
    {
        _object = const_cast<Object*>(object);
        _edgesHead = _edgesTail = nullptr;
        _flags = flg_owner;
    }
    void
    deInit()
    {
        clear();
    }

private:
    Object* _object;
    SlistNode* _edgesHead;
    SlistNode* _edgesTail;
    byte_t _flags;
};


UTL_NS_END;