libs/array.h

   1 /* Quarn OS
   2  *
   3  * Pointer
   4  *
   5  * Copyright (C) 2009 Pawel Dziepak
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License along
  18  * with this program; if not, write to the Free Software Foundation, Inc.,
  19  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  */
  22
  23 #ifndef _ARRAY_H_
  24 #define _ARRAY_H_
  25
  26 #include "buffer.h"
  27 #include "manes/exception.h"
  28
  29 template<typename T>
  30 class array {
  31 private:
  32         T *data;
  33         int size;
  34 public:
  35         array(int count) : data(new T[count]), size(count * sizeof(T)) { }
  36         array(T* _data, int _size) : data(_data), size(_size) { }
  37         ~array() {
  38                 delete data;
  39         }
  40
  41         T &operator[](int index) {
  42                 if (index >= 0 && index < size / sizeof(T))
  43                         return data[index];
  44
  45                 throw new index_out_of_range_exception();
  46         }
  47
  48         template<typename U>
  49         array<U> cast() {
  50                 return array<U>(reinterpret_cast<U*>(data), size);
  51         }
  52
  53         static array<T> from_buffer(const buffer &x) {
  54                 return array<T>((T*)x.get_address(), x.get_size());
  55         }
  56
  57         buffer to_buffer() const {
  58                 return buffer(data, size);
  59         }
  60 };
  61
  62
  63 template<typename T, int X>
  64 class array_stack {
  65 private:
  66         T data[X];
  67         const int size;
  68 public:
  69         array_stack() : size(X) { }
  70
  71         T &operator[](int index) {
  72                 if (index >= 0 && index < size)
  73                         return data[index];
  74
  75                 throw new index_out_of_range_exception();
  76         }
  77 };
  78 #endif