CMeshFast.h

Go to the documentation of this file.

/* +------------------------------------------------------------------------+
   |                     Mobile Robot Programming Toolkit (MRPT)            |
   |                          http://www.mrpt.org/                          |
   |                                                                        |
   | Copyright (c) 2005-2017, Individual contributors, see AUTHORS file     |
   | See: http://www.mrpt.org/Authors - All rights reserved.                |
   | Released under BSD License. See details in http://www.mrpt.org/License |
   +------------------------------------------------------------------------+ */

#ifndef opengl_CMeshFast_H
#define opengl_CMeshFast_H

#include <mrpt/opengl/CRenderizableDisplayList.h>
#include <mrpt/math/CMatrix.h>
#include <mrpt/utils/CImage.h>
#include <mrpt/utils/color_maps.h>

namespace mrpt
{
namespace opengl
{
/** A planar (XY) grid where each cell has an associated height and, optionally,
 * a texture map.
  * To make it faster to render, instead of drawing lines and triangles it draws
 * a point at each
  * gridcell.
  *  A typical usage example would be an elevation map or a 3D model of a
 * terrain.
  *  \sa opengl::COpenGLScene
  *
  *  <div align="center">
  *  <table border="0" cellspan="4" cellspacing="4" style="border-width: 1px;
 * border-style: solid;">
  *   <tr> <td> mrpt::opengl::CMeshFast </td> <td> \image html
 * preview_CMeshFast.png </td> </tr>
  *  </table>
  *  </div>
  *
  * \ingroup mrpt_opengl_grp
  */
class CMeshFast : public CRenderizableDisplayList
{
        DEFINE_SERIALIZABLE(CMeshFast)

   protected:
        mrpt::utils::CImage m_textureImage;

        bool m_enableTransparency;
        bool m_colorFromZ;
        bool m_isImage;

        /** X(x,y): X-coordinate of the point (x,y) */
        mutable math::CMatrix X;
        /** Y(x,y): Y-coordinate of the point (x,y) */
        mutable math::CMatrix Y;
        /** Z(x,y): Z-coordinate of the point (x,y) */
        mutable math::CMatrix Z;

        /** Grayscale Color [0,1] for each cell, updated by updateColorsMatrix */
        mutable math::CMatrix C;
        /** Red Component of the Color [0,1] for each cell, updated by
         * updateColorsMatrix */
        mutable math::CMatrix C_r;
        /** Green Component of the  Color [0,1] for each cell, updated by
         * updateColorsMatrix */
        mutable math::CMatrix C_g;
        /** Blue Component of the  Color [0,1] for each cell, updated by
         * updateColorsMatrix */
        mutable math::CMatrix C_b;

        /** Used when m_colorFromZ is true */
        mrpt::utils::TColormap m_colorMap;
        /** By default is 1.0 */
        float m_pointSize;
        /** Default: false */
        bool m_pointSmooth;

        /** Whether C is not up-to-date wrt to Z */
        mutable bool m_modified_Z;
        /** Whether C is not up-to-date wrt to the texture image */
        mutable bool m_modified_Image;

        /** Called internally to assure C is updated. */
        void updateColorsMatrix() const;
        void updatePoints() const;

        /** Mesh bounds */
        float xMin, xMax, yMin, yMax;

        /**Whether the coordinates of the points needs to be recalculated */
        mutable bool pointsUpToDate;

   public:
        /** By default is 1.0 */
        inline void setPointSize(float p) { m_pointSize = p; }
        inline float getPointSize() const { return m_pointSize; }
        inline void enablePointSmooth(bool enable = true)
        {
                m_pointSmooth = enable;
        }
        inline void disablePointSmooth() { m_pointSmooth = false; }
        void setGridLimits(float xmin, float xmax, float ymin, float ymax)
        {
                xMin = xmin;
                xMax = xmax;
                yMin = ymin;
                yMax = ymax;
                CRenderizableDisplayList::notifyChange();
        }

        void getGridLimits(float& xmin, float& xmax, float& ymin, float& ymax) const
        {
                xmin = xMin;
                xmax = xMax;
                ymin = yMin;
                ymax = yMax;
        }

        void enableTransparency(bool v)
        {
                m_enableTransparency = v;
                CRenderizableDisplayList::notifyChange();
        }
        void enableColorFromZ(
                bool v, mrpt::utils::TColormap colorMap = mrpt::utils::cmJET)
        {
                m_colorFromZ = v;
                m_colorMap = colorMap;
                CRenderizableDisplayList::notifyChange();
        }

        /** This method sets the matrix of heights for each position (cell) in the
         * mesh grid */
        void setZ(const mrpt::math::CMatrixTemplateNumeric<float>& in_Z);

        /** Returns a reference to the internal Z matrix, allowing changing it
         * efficiently */
        inline void getZ(mrpt::math::CMatrixFloat& out) const { out = Z; }
        inline float getXMin() const { return xMin; }
        inline float getXMax() const { return xMax; }
        inline float getYMin() const { return yMin; }
        inline float getYMax() const { return yMax; }
        inline void setXMin(const float& nxm)
        {
                xMin = nxm;
                pointsUpToDate = false;
                CRenderizableDisplayList::notifyChange();
        }
        inline void setXMax(const float& nxm)
        {
                xMax = nxm;
                pointsUpToDate = false;
                CRenderizableDisplayList::notifyChange();
        }
        inline void setYMin(const float& nym)
        {
                yMin = nym;
                pointsUpToDate = false;
                CRenderizableDisplayList::notifyChange();
        }
        inline void setYMax(const float& nym)
        {
                yMax = nym;
                pointsUpToDate = false;
                CRenderizableDisplayList::notifyChange();
        }
        inline void getXBounds(float& min, float& max) const
        {
                min = xMin;
                max = xMax;
        }
        inline void getYBounds(float& min, float& max) const
        {
                min = yMin;
                max = yMax;
        }
        inline void setXBounds(const float& min, const float& max)
        {
                xMin = min;
                xMax = max;
                pointsUpToDate = false;
                CRenderizableDisplayList::notifyChange();
        }
        inline void setYBounds(const float& min, const float& max)
        {
                yMin = min;
                yMax = max;
                pointsUpToDate = false;
                CRenderizableDisplayList::notifyChange();
        }

        /** Class factory  */
        static CMeshFast::Ptr Create(
                bool enableTransparency, float xMin = -1.0f, float xMax = 1.0f,
                float yMin = -1.0f, float yMax = 1.0f);

        /** Render
          */
        void render_dl() const override;

        /** Evaluates the bounding box of this object (including possible children)
         * in the coordinate frame of the object parent. */
        void getBoundingBox(
                mrpt::math::TPoint3D& bb_min,
                mrpt::math::TPoint3D& bb_max) const override;

        /** Assigns a texture image, and disable transparency.
          */
        void assignImage(const mrpt::utils::CImage& img);

        /** Assigns a texture image and Z simultaneously, and disable transparency.
          */
        void assignImageAndZ(
                const mrpt::utils::CImage& img,
                const mrpt::math::CMatrixTemplateNumeric<float>& in_Z);

        /** Adjust grid limits according to the image aspect ratio, maintaining the
         * X limits and resizing in the Y direction.
          */
        inline void adjustGridToImageAR()
        {
                ASSERT_(m_isImage);
                const float ycenter = 0.5 * (yMin + yMax);
                const float xwidth = xMax - xMin;
                const float newratio = float(m_textureImage.getWidth()) /
                                                           float(m_textureImage.getHeight());
                yMax = ycenter + 0.5 * newratio * xwidth;
                yMin = ycenter - 0.5 * newratio * xwidth;
                CRenderizableDisplayList::notifyChange();
        }

        /** Constructor
          */
        CMeshFast(
                bool enableTransparency = false, float xMin = -1.0f, float xMax = 1.0f,
                float yMin = -1.0f, float yMax = 1.0f)
                : m_textureImage(0, 0),
                  m_enableTransparency(enableTransparency),
                  m_colorFromZ(false),
                  m_isImage(false),
                  X(0, 0),
                  Y(0, 0),
                  Z(0, 0),
                  C(0, 0),
                  C_r(0, 0),
                  C_g(0, 0),
                  C_b(0, 0),
                  m_colorMap(mrpt::utils::cmJET),
                  m_modified_Z(true),
                  m_modified_Image(false),
                  xMin(xMin),
                  xMax(xMax),
                  yMin(yMin),
                  yMax(yMax),
                  pointsUpToDate(false)
        {
                m_color.A = 255;
                m_color.R = 0;
                m_color.G = 0;
                m_color.B = 150;
        }
        /** Private, virtual destructor: only can be deleted from smart pointers */
        virtual ~CMeshFast() {}
};

}  // end namespace

}  // End of namespace

#endif