GR_HeightFieldVK.h

Go to the documentation of this file.

/*
 * PROPRIETARY INFORMATION.  This software is proprietary to
 * Side Effects Software Inc., and is not to be reproduced,
 * transmitted, or disclosed in any way without written permission.
 * 
 * NAME:        GR_HeightFieldVK ( GR Library, C++)
 *
 * COMMENTS:
 *      Heightfield renderer.
 */
#ifndef GR_HeightFieldVK_h
#define GR_HeightFieldVK_h

#include "GR_API.h"
#include "GR_GeoRenderVK.h"
#include <UT/UT_VectorTypes.h>

class RV_Geometry;
class RV_Render;
class RV_ShaderVariableSet;
class RV_ShaderBlock;
class RV_ShaderProgram;
class RV_VKImage;
class GR_Material;
class GR_HeightfieldMaterial;
class GT_PrimVolume;

class GR_API GR_HeightFieldVK : public GR_GeoRenderVK
{
public:
             GR_HeightFieldVK(GR_Primitive *prim);
            ~GR_HeightFieldVK() override;

    const char         *className() const override { return "GR_HeightFieldVK"; }
    

    RV_Geometry        *getVkPickGeometry(unsigned int pick_mode) override;

    void                update(RE_RenderContext rc,
                               const GT_PrimitiveHandle &primh,
                               const GR_UpdateParms     &parms,
                               const char               *cache_name) override;
    
    void                draw(RE_RenderContext rc,
                             GR_RenderMode mode,
                             GR_RenderFlags flags,
                             GR_DrawParms dp) override;
    
    exint               getNumGLPrimitives() const override { return 1; }
    bool                hasTransparency() const override { return false; }
    
    static bool         initShaders(RV_Instance* inst);
    static void         cleanupShaders();

    bool                supportsRayTrace() const override { return true; }

    void updateAccelerationStructure(RV_Render* r,
                                     int tag,
                                     int light_id,
                                     const UT_Matrix4F &obj_mat) override;

    void retireGeometry() override;
    
private:
    void                updateHeight(RV_Render *r,
                                     const GT_PrimVolume *volume);
    void                updateColor(RV_Render *r,
                                    const GT_PrimVolume *volume,
                                    const GR_UpdateParms &p);
    void                updateColorRamp(RV_Render *r,
                                        const GT_PrimVolume *volume);
    void                updateBlendMasks(RV_Render *r,
                                         const GT_PrimVolume *volume);
    void                updateSurface(RV_Render *r,
                                      const GT_PrimVolume *volume,
                                      const GR_UpdateParms &p,
                                      const char *cache_name);
    void                updateMaterial(RV_Render *r,
                                       const GR_UpdateParms *p,
                                       GR_Material *material);
    GR_MaterialPtr      defaultMaterial(RV_Render *r);
    void                updateMaterial(RV_Render *r,
                                       GR_HeightfieldMaterial* mat,
                                       const GR_UpdateParms *p,
                                       GR_Material *material);

    int                 myMaterialID = -1;
    GR_MaterialPtr      myHFMat;
    GR_MaterialPtr      myHFDefaultMat;
    UT_Vector2i         myImageRes = {1,1};
    UT_Vector2i         myGridRes = {1,1};
    UT_Vector2i         myWireRes = {1,1};
    UT_Vector3F         myBaseColor = {1,1,1};
    bool                myOverrideToWireframe = false;
    bool                myColorRequiresRemap = false;
    fpreal              myColorMinRange = 0;
    fpreal              myColorInvRange = 1;
    int                 myColorChannels = 4;
    int                 myColorTypeInfo = 0;
    UT_UniquePtr<RV_VKImage> myHeightImage;
    UT_UniquePtr<RV_VKImage> myColorImage;
    UT_UniquePtr<RV_VKImage> myColorRampImage;
    UT_UniquePtr<RV_VKImage> myBlendImage;
    UT_UniquePtr<RV_ShaderBlock> myHeightGridUbo;
    UT_UniquePtr<RV_ShaderVariableSet> myImageSet;
    UT_UniquePtr<RV_ShaderVariableSet> myMeshIndicesSet;
    UT_UniquePtr<RV_ShaderVariableSet> myMeshAttribsSet;
    
    static UT_Map<exint, RV_ShaderProgram *> theProgramMap;
};

#endif