StretchedGrid.h

Go to the documentation of this file.

#ifndef _StretchedGrid_
#define _StretchedGrid_
#include <vapor/common.h>
#include <vapor/Grid.h>
#include <vapor/StructuredGrid.h>
 
namespace VAPoR {
//
class VDF_API StretchedGrid : public StructuredGrid {
public:
    //
    StretchedGrid(const DimsType &dims, const DimsType &bs, const std::vector<float *> &blks, const std::vector<double> &xcoords, const std::vector<double> &ycoords,
                  const std::vector<double> &zcoords);
    StretchedGrid(const std::vector<size_t> &dims, const std::vector<size_t> &bs, const std::vector<float *> &blks, const std::vector<double> &xcoords, const std::vector<double> &ycoords,
                  const std::vector<double> &zcoords);
 
    StretchedGrid() = default;
    virtual ~StretchedGrid() = default;
 
    virtual size_t GetGeometryDim() const override;
 
    virtual DimsType GetCoordDimensions(size_t dim) const override;
 
    static std::string GetClassType() { return ("Stretched"); }
    std::string        GetType() const override { return (GetClassType()); }
 
    // \copydoc GetGrid::GetBoundingBox()
    //
    virtual void GetBoundingBox(const DimsType &min, const DimsType &max, CoordType &minu, CoordType &maxu) const override;
 
    // \copydoc GetGrid::GetUserCoordinates()
    //
    virtual void GetUserCoordinates(const DimsType &indices, CoordType &coords) const override;
 
    //
    virtual bool GetIndicesCell(const CoordType &coords, DimsType &indices, double wgts[3]) const;
 
    virtual bool GetIndicesCell(const CoordType &coords, DimsType &indices) const override
    {
        double dummy[3];
        return (GetIndicesCell(coords, indices, dummy));
    };
 
    // \copydoc GetGrid::InsideGrid()
    //
    virtual bool InsideGrid(const CoordType &coords) const override;
 
    const std::vector<double> &GetXCoords() const { return (_xcoords); };
 
    const std::vector<double> &GetYCoords() const { return (_ycoords); };
 
    const std::vector<double> &GetZCoords() const { return (_zcoords); };
 
    class ConstCoordItrSG : public Grid::ConstCoordItrAbstract {
    public:
        ConstCoordItrSG(const StretchedGrid *cg, bool begin);
        ConstCoordItrSG(const ConstCoordItrSG &rhs);
 
        ConstCoordItrSG();
        virtual ~ConstCoordItrSG() {}
 
        virtual void            next();
        virtual void            next(const long &offset);
        virtual ConstCoordType &deref() const { return (_coords); }
        virtual const void *    address() const { return this; };
 
        virtual bool equal(const void *rhs) const
        {
            const ConstCoordItrSG *itrptr = static_cast<const ConstCoordItrSG *>(rhs);
 
            return (_index == itrptr->_index);
        }
 
        virtual std::unique_ptr<ConstCoordItrAbstract> clone() const { return std::unique_ptr<ConstCoordItrAbstract>(new ConstCoordItrSG(*this)); };
 
    private:
        const StretchedGrid *_sg;
        DimsType             _index;
        CoordType            _coords;
    };
 
    virtual ConstCoordItr ConstCoordBegin() const override { return ConstCoordItr(std::unique_ptr<ConstCoordItrAbstract>(new ConstCoordItrSG(this, true))); }
    virtual ConstCoordItr ConstCoordEnd() const override { return ConstCoordItr(std::unique_ptr<ConstCoordItrAbstract>(new ConstCoordItrSG(this, false))); }
 
protected:
    virtual float GetValueNearestNeighbor(const CoordType &coords) const override;
 
    virtual float GetValueLinear(const CoordType &coords) const override;
 
    void GetUserExtentsHelper(CoordType &minu, CoordType &maxu) const override;
 
private:
    std::vector<double> _xcoords;
    std::vector<double> _ycoords;
    std::vector<double> _zcoords;
    CoordType           _minu = {{0.0, 0.0, 0.0}};
    CoordType           _maxu = {{0.0, 0.0, 0.0}};
 
    void _stretchedGrid(const std::vector<double> &xcoords, const std::vector<double> &ycoords, const std::vector<double> &zcoords);
 
    bool _insideGrid(double x, double y, double z, size_t &i, size_t &j, size_t &k, double &xwgt, double &ywgt, double &zwgt) const;
};
};    // namespace VAPoR
#endif