RedBlackTreeSet.hpp

Go to the documentation of this file.

// ======================================================================
// \file   RedBlackTreeSet.hpp
// \author bocchino
// \brief  An array-based set with internal storage
// ======================================================================

#ifndef Fw_RedBlackTreeSet_HPP
#define Fw_RedBlackTreeSet_HPP

#include "Fw/DataStructures/ExternalRedBlackTreeSet.hpp"

namespace Fw {

template <typename T, FwSizeType C>
class RedBlackTreeSet final : public SetBase<T> {
    // ----------------------------------------------------------------------
    // Static assertions
    // ----------------------------------------------------------------------

    static_assert(C > 0, "capacity must be greater than zero");

    // ----------------------------------------------------------------------
    // Friend class for testing
    // ----------------------------------------------------------------------

    template <typename TT, FwSizeType CC>
    friend class RedBlackTreeSetTester;

  public:
    // ----------------------------------------------------------------------
    // Public types
    // ----------------------------------------------------------------------

    using ConstIterator = SetConstIterator<T>;

    using Node = typename RedBlackTreeSetOrMapImpl<T, Nil>::Node;

    using Nodes = Node[C];

    using Index = typename RedBlackTreeSetOrMapImpl<T, Nil>::Index;

    using FreeNodes = Index[C];

  public:
    // ----------------------------------------------------------------------
    // Public constructors and destructors
    // ----------------------------------------------------------------------

    RedBlackTreeSet() : SetBase<T>(), m_extSet(m_nodes, m_freeNodes, C) {}

    RedBlackTreeSet(const RedBlackTreeSet<T, C>& set) : SetBase<T>(), m_extSet(m_nodes, m_freeNodes, C) { *this = set; }

    ~RedBlackTreeSet() override = default;

  public:
    // ----------------------------------------------------------------------
    // Public member functions
    // ----------------------------------------------------------------------

    RedBlackTreeSet<T, C>& operator=(const RedBlackTreeSet<T, C>& set) {
        this->m_extSet.copyDataFrom(set);
        return *this;
    }

    ConstIterator begin() const override { return this->m_extSet.begin(); }

    void clear() override { this->m_extSet.clear(); }

    ConstIterator end() const override { return this->m_extSet.end(); }

    Success find(const T& element  
    ) const override {
        return this->m_extSet.find(element);
    }

    FwSizeType getCapacity() const override { return this->m_extSet.getCapacity(); }

    FwSizeType getSize() const override { return this->m_extSet.getSize(); }

    Success insert(const T& element  
                   ) override {
        return this->m_extSet.insert(element);
    }

    Success remove(const T& element  
                   ) override {
        return this->m_extSet.remove(element);
    }

  private:
    // ----------------------------------------------------------------------
    // Private member variables
    // ----------------------------------------------------------------------

    Nodes m_nodes = {};

    FreeNodes m_freeNodes = {};

    ExternalRedBlackTreeSet<T> m_extSet = {};
};

}  // namespace Fw

#endif