EdgeArray.h

Go to the documentation of this file.

 
#pragma once
 
#include <ogdf/basic/Graph_d.h>
 
namespace ogdf {
 
 
 
class EdgeArrayBase {
    ListIterator<EdgeArrayBase*> m_it;
 
public:
    const Graph* m_pGraph; 
 
    EdgeArrayBase() : m_pGraph(nullptr) { }
 
    explicit EdgeArrayBase(const Graph* pG) : m_pGraph(pG) {
        if (pG) {
            m_it = pG->registerArray(this);
        }
    }
 
    EdgeArrayBase(EdgeArrayBase& base) : m_it(base.m_it), m_pGraph(base.m_pGraph) {
        if (m_pGraph) {
            m_pGraph->moveRegisterArray(m_it, this);
        }
        base.m_pGraph = nullptr;
        base.m_it = ListIterator<EdgeArrayBase*>();
    }
 
    // destructor, unregisters the array
    virtual ~EdgeArrayBase() {
        if (m_pGraph) {
            m_pGraph->unregisterArray(m_it);
        }
    }
 
    // event interface used by Graph
    virtual void enlargeTable(int newTableSize) = 0;
    virtual void reinit(int initTableSize) = 0;
    virtual void disconnect() = 0;
 
    void reregister(const Graph* pG) {
        if (m_pGraph) {
            m_pGraph->unregisterArray(m_it);
        }
        if ((m_pGraph = pG) != nullptr) {
            m_it = pG->registerArray(this);
        }
    }
 
    void moveRegister(EdgeArrayBase& base) {
        if (m_pGraph) {
            m_pGraph->unregisterArray(m_it);
        }
        m_pGraph = base.m_pGraph;
        m_it = base.m_it;
        base.m_pGraph = nullptr;
        base.m_it = ListIterator<EdgeArrayBase*>();
        if (m_pGraph != nullptr) {
            m_pGraph->moveRegisterArray(m_it, this);
        }
    }
};
 
 
template<class T>
class EdgeArray : private Array<T>, protected EdgeArrayBase {
    T m_x; 
 
public:
    using key_type = edge;
    using value_type = T;
 
    using iterator = internal::GraphArrayIterator<EdgeArray<T>>;
    using const_iterator = internal::GraphArrayConstIterator<EdgeArray<T>>;
 
    EdgeArray() : Array<T>(), EdgeArrayBase() { }
 
    explicit EdgeArray(const Graph& G) : Array<T>(G.edgeArrayTableSize()), EdgeArrayBase(&G) { }
 
 
    EdgeArray(const Graph& G, const T& x)
        : Array<T>(0, G.edgeArrayTableSize() - 1, x), EdgeArrayBase(&G), m_x(x) { }
 
 
    EdgeArray(const EdgeArray<T>& A) : Array<T>(A), EdgeArrayBase(A.m_pGraph), m_x(A.m_x) { }
 
 
    EdgeArray(EdgeArray<T>&& A) : Array<T>(std::move(A)), EdgeArrayBase(A), m_x(A.m_x) { }
 
 
    bool valid() const { return Array<T>::low() <= Array<T>::high(); }
 
    const Graph* graphOf() const { return m_pGraph; }
 
    const T& operator[](edge e) const {
        OGDF_ASSERT(e != nullptr);
        OGDF_ASSERT(e->graphOf() == m_pGraph);
        return Array<T>::operator[](e->index());
    }
 
    T& operator[](edge e) {
        OGDF_ASSERT(e != nullptr);
        OGDF_ASSERT(e->graphOf() == m_pGraph);
        return Array<T>::operator[](e->index());
    }
 
    const T& operator()(edge e) const {
        OGDF_ASSERT(e != nullptr);
        OGDF_ASSERT(e->graphOf() == m_pGraph);
        return Array<T>::operator[](e->index());
    }
 
    T& operator()(edge e) {
        OGDF_ASSERT(e != nullptr);
        OGDF_ASSERT(e->graphOf() == m_pGraph);
        return Array<T>::operator[](e->index());
    }
 
    const T& operator[](adjEntry adj) const {
        OGDF_ASSERT(adj != nullptr);
        return Array<T>::operator[](adj->index() >> 1);
    }
 
    T& operator[](adjEntry adj) {
        OGDF_ASSERT(adj != nullptr);
        return Array<T>::operator[](adj->index() >> 1);
    }
 
    const T& operator()(adjEntry adj) const {
        OGDF_ASSERT(adj != nullptr);
        return Array<T>::operator[](adj->index() >> 1);
    }
 
    T& operator()(adjEntry adj) {
        OGDF_ASSERT(adj != nullptr);
        return Array<T>::operator[](adj->index() >> 1);
    }
 
    OGDF_DEPRECATED("Edge arrays should be indexed by an edge, not an integer index.")
 
    const T& operator[](int index) const { return Array<T>::operator[](index); }
 
    OGDF_DEPRECATED("Edge arrays should be indexed by an edge, not an integer index.")
 
    T& operator[](int index) { return Array<T>::operator[](index); }
 
 
 
 
    iterator begin() { return iterator(m_pGraph->firstEdge(), this); }
 
 
    const_iterator begin() const { return const_iterator(m_pGraph->firstEdge(), this); }
 
 
    const_iterator cbegin() const { return const_iterator(m_pGraph->firstEdge(), this); }
 
 
    iterator end() { return iterator(nullptr, this); }
 
 
    const_iterator end() const { return const_iterator(nullptr, this); }
 
 
    const_iterator cend() const { return const_iterator(nullptr, this); }
 
 
 
    void init() {
        Array<T>::init();
        reregister(nullptr);
    }
 
    void init(const Graph& G) {
        Array<T>::init(G.edgeArrayTableSize());
        reregister(&G);
    }
 
 
    void init(const Graph& G, const T& x) {
        Array<T>::init(0, G.edgeArrayTableSize() - 1, m_x = x);
        reregister(&G);
    }
 
    void fill(const T& x) {
        int high = m_pGraph->maxEdgeIndex();
        if (high >= 0) {
            Array<T>::fill(0, high, x);
        }
    }
 
    EdgeArray<T>& operator=(const EdgeArray<T>& a) {
        Array<T>::operator=(a);
        m_x = a.m_x;
        reregister(a.m_pGraph);
        return *this;
    }
 
 
    EdgeArray<T>& operator=(EdgeArray<T>&& a) {
        Array<T>::operator=(std::move(a));
        m_x = a.m_x;
        moveRegister(a);
        return *this;
    }
 
 
 
    static key_type findSuccKey(key_type key) { return key->succ(); }
 
    static key_type findPredKey(key_type key) { return key->pred(); }
 
 
private:
    virtual void enlargeTable(int newTableSize) { Array<T>::resize(newTableSize, m_x); }
 
    virtual void reinit(int initTableSize) { Array<T>::init(0, initTableSize - 1, m_x); }
 
    virtual void disconnect() {
        Array<T>::init();
        m_pGraph = nullptr;
    }
 
    OGDF_NEW_DELETE
};
 
 
class OGDF_EXPORT BucketEdgeArray : public BucketFunc<edge> {
    const EdgeArray<int>* m_pEdgeArray; 
 
public:
 
    explicit BucketEdgeArray(const EdgeArray<int>& edgeArray) : m_pEdgeArray(&edgeArray) { }
 
    int getBucket(const edge& e) override { return (*m_pEdgeArray)[e]; }
};
 
}