array_buffer.hh

Go to the documentation of this file.

/*
    @@@@@@@@  @@           @@@@@@   @@@@@@@@ @@
   /@@/////  /@@          @@////@@ @@////// /@@
   /@@       /@@  @@@@@  @@    // /@@       /@@
   /@@@@@@@  /@@ @@///@@/@@       /@@@@@@@@@/@@
   /@@////   /@@/@@@@@@@/@@       ////////@@/@@
   /@@       /@@/@@//// //@@    @@       /@@/@@
   /@@       @@@//@@@@@@ //@@@@@@  @@@@@@@@ /@@
   //       ///  //////   //////  ////////  //

   Copyright (c) 2016, Los Alamos National Security, LLC
   All rights reserved.
                                                                              */
#pragma once

#include "flecsi/util/id.hh"

namespace flecsi {
namespace topo {

template<size_t, typename>
class domain_entity;

template<typename item_t>
struct array_buffer_type {
  using type = item_t *;
};

template<size_t M, typename E>
struct array_buffer_type<domain_entity<M, E>> {
  using type = E *;
};

template<typename item_t>
struct array_buffer_type<item_t *> {
  using type = item_t *;
};

template<typename T>
struct array_buf_ref_type {
  using type = T &;
};

template<typename S>
struct array_buf_ref_type<S *> {
  using type = S *;
};

template<size_t M, class E>
struct array_buf_ref_type<domain_entity<M, E>> {
  using type = E *;
};

template<typename T, bool B>
struct array_buf_ref_get {
  static T get(T a, size_t i) {
    return &a[i];
  }
};

template<typename T>
struct array_buf_ref_get<T, false> {
  static auto get(T a, size_t i) -> decltype(a[i]) {
    return a[i];
  }
};

template<typename T>
class array_buffer
{
public:
  using item_t = typename array_buffer_type<T>::type;

  using iterator = item_t;

  using const_iterator = item_t;

  using ref_t = typename array_buf_ref_type<T>::type;

  array_buffer() : buf_(nullptr), size_(0), capacity_(0) {}

  ref_t operator[](size_t index) {
    return array_buf_ref_get<item_t, std::is_pointer<ref_t>::value>::get(
      buf_, index);
  }

  const ref_t operator[](size_t index) const {
    return array_buf_ref_get<const item_t, std::is_pointer<ref_t>::value>::get(
      buf_, index);
  }

  ref_t back() {
    return array_buf_ref_get<item_t, std::is_pointer<ref_t>::value>::get(
      buf_, size_ - 1);
  }

  const ref_t back() const {
    return array_buf_ref_get<item_t, std::is_pointer<ref_t>::value>::get(
      buf_, size_ - 1);
  }

  size_t size() const {
    return size_;
  } // size

  size_t capacity() const {
    return capacity_;
  } // capacity

  item_t begin() {
    return buf_;
  }

  item_t end() {
    return buf_ + size_;
  }

  const item_t begin() const {
    return buf_;
  }

  const item_t end() const {
    return buf_ + size_;
  }

  template<typename... Args>
  void insert(Args &&...) {}

  void push_back(const ref_t x) {
    assert(size_ < capacity_ && "array buffer capacity exceeded");
    buf_[size_++] = x;
  }

  void pushed() {
    ++size_;
  }

  void clear() {
    size_ = 0;
  }

  bool empty() const {
    return size_ == 0;
  }

  void resize(size_t n) {
    assert(n <= capacity_);
    size_ = n;
  }

  void set_buffer(item_t buf, size_t size) {
    buf_ = buf;
    size_ = size;
    capacity_ = size;
  }

  void set_buffer(item_t buf, size_t capacity, bool initialized) {
    buf_ = buf;
    capacity_ = capacity;
    size_ = initialized ? capacity_ : 0;
  }

  void set_buffer(item_t buf, size_t capacity, size_t size) {
    buf_ = buf;
    capacity_ = capacity;
    size_ = size;
  }

  item_t buffer() {
    return buf_;
  }

  const item_t buffer() const {
    return buf_;
  }

  item_t data() {
    return buf_;
  }

  const item_t data() const {
    return buf_;
  }

  template<typename... Args>
  void assign(Args &&...) {
    assert(false && "unimplemented");
  }

  void reserve(size_t) {
    assert(false && "unimplemented");
  }

private:
  item_t buf_;
  size_t size_;
  size_t capacity_;
};

} // namespace topo
} // namespace flecsi