banan-os/BAN/include/BAN/Iterators.h

159 lines
2.3 KiB
C++

#pragma once
#include <BAN/Assert.h>
namespace BAN
{
template<typename T, typename Container>
class IteratorSimple
{
public:
IteratorSimple() = default;
const T& operator*() const
{
ASSERT(m_pointer);
return *m_pointer;
}
T& operator*()
{
ASSERT(m_pointer);
return *m_pointer;
}
const T* operator->() const
{
ASSERT(m_pointer);
return m_pointer;
}
T* operator->()
{
ASSERT(m_pointer);
return m_pointer;
}
IteratorSimple& operator++()
{
ASSERT(m_pointer);
++m_pointer;
return *this;
}
IteratorSimple operator++(int)
{
auto temp = *this;
++(*this);
return temp;
}
IteratorSimple& operator--()
{
ASSERT(m_pointer);
return --m_pointer;
}
IteratorSimple operator--(int)
{
auto temp = *this;
--(*this);
return temp;
}
bool operator==(const IteratorSimple& other) const
{
return m_pointer == other.m_pointer;
}
bool operator!=(const IteratorSimple& other) const
{
return !(*this == other);
}
operator bool() const
{
return m_pointer;
}
private:
IteratorSimple(T* pointer)
: m_pointer(pointer)
{
}
private:
T* m_pointer = nullptr;
friend Container;
};
template<typename T, typename Container>
class ConstIteratorSimple
{
public:
ConstIteratorSimple() = default;
ConstIteratorSimple(IteratorSimple<T, Container> other)
: m_pointer(other.m_pointer)
{
}
const T& operator*() const
{
ASSERT(m_pointer);
return *m_pointer;
}
const T* operator->() const
{
ASSERT(m_pointer);
return m_pointer;
}
ConstIteratorSimple& operator++()
{
ASSERT(m_pointer);
++m_pointer;
return *this;
}
ConstIteratorSimple operator++(int)
{
auto temp = *this;
++(*this);
return temp;
}
ConstIteratorSimple& operator--()
{
ASSERT(m_pointer);
return --m_pointer;
}
ConstIteratorSimple operator--(int)
{
auto temp = *this;
--(*this);
return temp;
}
bool operator==(const ConstIteratorSimple& other) const
{
return m_pointer == other.m_pointer;
}
bool operator!=(const ConstIteratorSimple& other) const
{
return !(*this == other);
}
operator bool() const
{
return !!m_pointer;
}
private:
ConstIteratorSimple(const T* pointer)
: m_pointer(pointer)
{
}
private:
const T* m_pointer = nullptr;
friend Container;
};
}