array2.h

Go to the documentation of this file.

#ifndef ION_BASE_ARRAY2_H_
#define ION_BASE_ARRAY2_H_

#include "ion/base/allocatable.h"
#include "ion/base/invalid.h"
#include "ion/base/logging.h"
#include "ion/base/stlalloc/allocvector.h"

namespace ion {
namespace base {

template <typename T> class Array2 : public Allocatable {
 public:
  Array2() : width_(0), height_(0), data_(GetAllocator()) {}

  Array2(size_t width, size_t height)
      : width_(width), height_(height), data_(GetAllocator()) {
    data_.resize(width * height);
  }

  Array2(size_t width, size_t height, const T& initial_value)
      : width_(width),
        height_(height),
        data_(GetAllocator(), width * height, initial_value) {
  }

  ~Array2() override {}

  size_t GetWidth() const { return width_; }
  size_t GetHeight() const { return height_; }
  size_t GetSize() const { return data_.size(); }

  bool Set(size_t column, size_t row, const T& val) {
    const size_t index = GetIndex(column, row);
    if (index == base::kInvalidIndex) {
      return false;
    } else {
      data_[index] = val;
      return true;
    }
  }

  const T& Get(size_t column, size_t row) const {
    const size_t index = GetIndex(column, row);
    return index == base::kInvalidIndex ?
        base::InvalidReference<T>() : data_[index];
  }

  T* GetMutable(size_t column, size_t row) {
    const size_t index = GetIndex(column, row);
    return index == base::kInvalidIndex ? NULL : &data_[index];
  }

 private:
  size_t GetIndex(size_t column, size_t row) const {
    if (column >= width_ || row >= height_) {
      LOG(ERROR) << "Bad indices (" << column << ", " << row
                 << ") for Array2 of size " << width_ << " x " << height_;
      return base::kInvalidIndex;
    }
    return row * width_ + column;
  }

  size_t width_;
  size_t height_;
  AllocVector<T> data_;
};

}  // namespace base
}  // namespace ion

#endif  // ION_BASE_ARRAY2_H_