CuboidGridObjectPopulation.h

Go to the documentation of this file.

#ifndef CUBOIDGRIDOBJECTPOPULATION_H_
#define CUBOIDGRIDOBJECTPOPULATION_H_

#include "SpatialSimObjectPopulation.h"


class CuboidGridObjectPopulation : public SpatialSimObjectPopulation {
    public:

        typedef enum { X = 0, Y = 1, Z = 2 } CoordinateIndex;

        CuboidGridObjectPopulation(SimNetwork &net, GridPoint3D origin, Volume3DSize dims, SimObjectFactory &objFactory);

        CuboidGridObjectPopulation(SimNetwork &net, GridPoint3D origin, Volume3DSize dims, SimObject::ID::Vector vec);

        GridPoint3D & getOrigin() {
            return origin;
        };


        virtual Point3D<double> getLocation( size_t index ) const {
            GridPoint3D p = origin + getRelativeLocation(index);
            return Point3D<double>( p.x(), p.y(), p.z() );
        };

        GridPoint3D getRelativeLocation(size_t idx) const {
            div_t xy = div(idx, dim[Y] * dim[Z]);
            div_t yz = div(xy.rem, dim[Z]);
            return GridPoint3D(xy.quot, yz.quot, yz.rem);
        };

        SimObject::ID operator() (int x, int y, int z) const {
            return (SimObject::ID)(*id_vec)[dim[Y]*dim[Z]*x + dim[Z]*y + z];
        };

        SimObject::ID operator() (const GridPoint3D &loc) const {
            return (SimObject::ID)(*id_vec)[dim[Y]*dim[Z]*loc.x() + dim[Z]*loc.y() + loc.z()];
        };

        int getIndex(int x, int y, int z) const {
            return dim[Y]*dim[Z]*x + dim[Z]*y + z;
        };

        int getIndex(const GridPoint3D &loc) const {
            return  dim[Y]*dim[Z]*loc.x() + dim[Z]*loc.y() + loc.z();
        }

        SimObject::ID operator() (int idx) const {
            return (SimObject::ID)((*id_vec)[idx]);
        };

        SimObject::ID atRelative(int x, int y, int z) const {
            return (SimObject::ID)(*id_vec)[dim[Y]*dim[Z]*x + dim[Z]*y + z];
        };

        SimObject::ID at(int x, int y, int z) const {
            return at(x - origin.x(), y - origin.y(), z - origin.z());
        };

        CuboidGridObjectPopulation subVolume(int origX, int origY, int origZ,
                                             int Xdim, int Ydim, int Zdim) const;

        CuboidGridObjectPopulation subVolume(const GridPoint3D &origin, const Volume3DSize & size) const;

        const Volume3DSize & getDimensions() const {
            return dim;
        }


    protected:
        GridPoint3D origin;
        Volume3DSize dim;

};


#endif /* CUBOIDGRIDOBJECTPOPULATION_H_ */

---