/Users/jab/Documents/Teaching/02585/GEL2_and_demos/GEL/src/Geometry/RGrid.h

#ifndef __GEOMETRY_RGRID_H
#define __GEOMETRY_RGRID_H

#include <vector>
#include "CGLA/Vec3i.h"
#include "AncestorGrid.h"

namespace Geometry 
{

        template<class T>
        class RGrid: public AncestorGrid<T,RGrid<T> >
        {
        public:
                typedef T DataType;

        private:
                int x_dim;
                
                int xy_dim;

                std::vector<T> data;

                int grid_idx(const CGLA::Vec3i& idx) const
                {
                        return  idx[2] * xy_dim + idx[1] * x_dim + idx[0];
                }

                DataType default_val;

        public:
        
                RGrid(CGLA::Vec3i _dims, const T& val = T()):
                        AncestorGrid<T,RGrid<T> >(_dims), 
                        x_dim(_dims[0]), xy_dim(_dims[0]*_dims[1]),
                        data(_dims[0]*_dims[1]*_dims[2],val),
                        default_val(val)
                {}      

                RGrid(): AncestorGrid<T,RGrid<T> >(CGLA::Vec3i(0)), 
                        x_dim(0), xy_dim(0),
                        data(0,0), default_val(0)
                {}      

                void store(const CGLA::Vec3i& p, const T& t) 
                {
                        assert(this->in_domain(p));
                        data[grid_idx(p)] = t;
                }

                T& operator[](const CGLA::Vec3i& p) 
                {
                        assert(this->in_domain(p));
                        return data[grid_idx(p)];
                }

                const T& operator[](const CGLA::Vec3i& p) const 
                {
                        assert(this->in_domain(p));
                        return data[grid_idx(p)];
                }

                const T* get() const {return &data[0];}

                T* get() {return &data[0];}

                int get_x_dim() const { return x_dim;}

                int get_xy_dim() const { return xy_dim;}

                int get_size() const { return data.size();}

                void clear()
                {
                        int N = data.size();
                        for(int i=0;i<N;++i)
                                data[i] = default_val;
                }


        };

}

#endif