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