mirror of https://github.com/PrimeDecomp/prime.git
78 lines
2.4 KiB
C++
78 lines
2.4 KiB
C++
#ifndef _RSTL_POINTER_ITERATOR_HPP
|
|
#define _RSTL_POINTER_ITERATOR_HPP
|
|
|
|
#include "types.h"
|
|
|
|
#include "rstl/construct.hpp"
|
|
|
|
namespace rstl {
|
|
struct input_iterator_tag {};
|
|
struct output_iterator_tag {};
|
|
struct forward_iterator_tag : public input_iterator_tag {};
|
|
struct bidirectional_iterator_tag : public forward_iterator_tag {};
|
|
struct random_access_iterator_tag : public bidirectional_iterator_tag {};
|
|
|
|
template < typename T, typename Vec, typename Alloc >
|
|
class const_pointer_iterator {
|
|
public:
|
|
typedef random_access_iterator_tag iterator_category;
|
|
|
|
const_pointer_iterator() : current(nullptr) {}
|
|
const_pointer_iterator(const T* begin) : current(const_cast< T* >(begin)) {}
|
|
const_pointer_iterator& operator++() {
|
|
++current;
|
|
return *this;
|
|
}
|
|
const_pointer_iterator& operator--() {
|
|
--current;
|
|
return *this;
|
|
}
|
|
const T& operator*() const { return *current; }
|
|
const T* operator->() const { return current; }
|
|
bool operator==(const const_pointer_iterator& other) { return current == other.current; }
|
|
bool operator!=(const const_pointer_iterator& other) { return current != other.current; }
|
|
|
|
friend const_pointer_iterator operator+(const const_pointer_iterator& x, int v) {
|
|
return const_pointer_iterator(x.current + v);
|
|
}
|
|
friend const_pointer_iterator operator-(const const_pointer_iterator& x, int v) {
|
|
return const_pointer_iterator(x.current - v);
|
|
}
|
|
|
|
protected:
|
|
T* current;
|
|
};
|
|
|
|
template < typename T, typename Vec, typename Alloc >
|
|
class pointer_iterator : public const_pointer_iterator< T, Vec, Alloc > {
|
|
public:
|
|
pointer_iterator() : const_pointer_iterator< T, Vec, Alloc >(nullptr) {}
|
|
pointer_iterator(T* begin) : const_pointer_iterator< T, Vec, Alloc >(begin) {}
|
|
void operator=(const T& other) { rstl::construct(this->current, other); }
|
|
T& operator*() { return *this->current; }
|
|
T* operator->() { return this->current; }
|
|
void destroy() const {
|
|
if (this->current != nullptr) {
|
|
rstl::destroy(this->current);
|
|
}
|
|
}
|
|
pointer_iterator& operator++() {
|
|
++this->current;
|
|
return *this;
|
|
}
|
|
pointer_iterator& operator--() {
|
|
--this->current;
|
|
return *this;
|
|
}
|
|
|
|
friend pointer_iterator operator+(const pointer_iterator& x, int v) {
|
|
return pointer_iterator(x.current + v);
|
|
}
|
|
friend pointer_iterator operator-(const pointer_iterator& x, int v) {
|
|
return pointer_iterator(x.current - v);
|
|
}
|
|
};
|
|
} // namespace rstl
|
|
|
|
#endif
|