Use container and C++20 range algorithms for allocator.

1 files changed, 40 insertions, 4 deletions

diff --git a/arch/x86_64/src/memory/multiboot/memory_map.cpp b/arch/x86_64/src/memory/multiboot/memory_map.cpp
index da7f05d..b5e2759 100644
--- a/arch/x86_64/src/memory/multiboot/memory_map.cpp
+++ b/arch/x86_64/src/memory/multiboot/memory_map.cpp
@@ -4,18 +4,21 @@
 
 namespace teachos::arch::memory::multiboot
 {
-  memory_area_iterator::memory_area_iterator(multiboot::memory_area * p)
+  memory_area_iterator::memory_area_iterator(value_type * p)
       : ptr(p)
   {
     exception_handling::assert(ptr, "[Memory Area] Attempted to pass nullptr as iterator");
   }
 
-  multiboot::memory_area & memory_area_iterator::operator*() const { return *ptr; }
+  auto memory_area_iterator::operator*() const -> value_type & { return *ptr; }
+
+  auto memory_area_iterator::operator->() const -> value_type * { return ptr; }
 
   auto memory_area_iterator::operator++(int) -> memory_area_iterator
   {
-    ++(*this);
-    return *this;
+    memory_area_iterator old_value = *this;
+    ++ptr;
+    return old_value;
   }
 
   auto memory_area_iterator::operator++() -> memory_area_iterator &
@@ -24,6 +27,35 @@ namespace teachos::arch::memory::multiboot
     return *this;
   }
 
+  auto memory_area_iterator::operator+=(difference_type value) -> memory_area_iterator &
+  {
+    ptr += value;
+    return *this;
+  }
+
+  auto memory_area_iterator::operator-=(difference_type value) -> memory_area_iterator &
+  {
+    ptr -= value;
+    return *this;
+  }
+
+  auto memory_area_iterator::operator+(difference_type value) const -> memory_area_iterator
+  {
+    return memory_area_iterator{ptr + value};
+  }
+
+  auto memory_area_iterator::operator-(difference_type value) const -> memory_area_iterator
+  {
+    return memory_area_iterator{ptr - value};
+  }
+
+  auto memory_area_iterator::operator-(const memory_area_iterator & other) const -> difference_type
+  {
+    return ptr - other.ptr;
+  }
+
+  auto memory_area_iterator::operator[](difference_type index) const -> value_type & { return *(ptr + index); }
+
   memory_area_container::memory_area_container(memory_area * begin, std::size_t size)
       : area_begin(begin)
       , area_end(begin + size)
@@ -34,4 +66,8 @@ namespace teachos::arch::memory::multiboot
   auto memory_area_container::begin() const -> multiboot::memory_area_iterator { return area_begin; }
 
   auto memory_area_container::end() const -> multiboot::memory_area_iterator { return area_end; }
+
+  auto memory_area_container::size() const -> size_type { return std::distance(begin(), end()); }
+
+  auto memory_area_container::empty() const -> bool { return size() == 0; }
 }  // namespace teachos::arch::memory::multiboot