Array2D.h

Go to the documentation of this file.

#ifndef ARRAY_2D_H
#define ARRAY_2D_H
 
#include "Element.h"
#include "Array.h"
 
namespace bridges {
    namespace datastructure {
        template <typename E>
        class Array2D : public Array<E> {
 
            private:
                int num_rows;
                int num_cols;
 
                Array2D()
                    : num_rows(0), num_cols(0) {
                }
            public:
 
                virtual ~Array2D() {
                }
 
 
                Array2D(int rows, int cols)
                    : num_rows(rows), num_cols(cols) {
                    int dims[] = {cols, rows, 1};
                    this->setSize (2, dims);
                }
                int getNumRows () {
                    return num_rows;
                }
                int getNumColumns () {
                    return num_cols;
                }
 
                Element<E>& getElement(int row, int col) {
                    return Array<E>::getElement(row * num_cols + col);
                }
 
                void setElement(int row, int col, Element<E> el) {
                    setElement(row * num_cols + col, el);
                }
 
                struct Bracket_helper {
                    Array2D<E>& arr;
                    int row;
                    Bracket_helper(Array2D<E>& a, int r) : arr(a), row(r) {}
 
                    E& operator[] (int col) {
                        return arr.getElement(row, col).getValue();
                    }
                };
 
                Bracket_helper operator[] (int index) {
                    return Bracket_helper(*this, index);
                }
 
                struct Bracket_helper_const {
                    Array2D<E> const& arr;
                    int row;
                    Bracket_helper_const(Array2D<E>& a, int r) : arr(a), row(r) {}
 
                    E const & operator[] (int col) const {
                        return arr.getElement(row, col).getValue();
                    }
                };
 
                Bracket_helper_const operator[] (int index) const {
                    return Bracket_helper_const(*this, index);
                }
 
        }; // Array2D
    }
}// end namespace bridges
 
#endif