ShaderGeometry.h

Go to the documentation of this file.

/*
Copyright 2007-2025. Algoryx Simulation AB.
 
All AGX source code, intellectual property, documentation, sample code,
tutorials, scene files and technical white papers, are copyrighted, proprietary
and confidential material of Algoryx Simulation AB. You may not download, read,
store, distribute, publish, copy or otherwise disseminate, use or expose this
material unless having a written signed agreement with Algoryx Simulation AB, or having been
advised so by Algoryx Simulation AB for a time limited evaluation, or having purchased a
valid commercial license from Algoryx Simulation AB.
 
Algoryx Simulation AB disclaims all responsibilities for loss or damage caused
from using this software, unless otherwise stated in written agreements with
Algoryx Simulation AB.
*/
 
#pragma once
 
#include <osg/MatrixTransform>
#include <osg/Geometry>
 
#include <agxOSG/export.h>
 
#include <agxRender/Color.h>
 
namespace agxOSG
{
  struct ShaderGeometry;
  using OsgShaderGeometryRef = osg::ref_ptr<ShaderGeometry>;
  using OsgShaderGeometryRefVector = std::vector<OsgShaderGeometryRef>;
 
  struct AGXOSG_EXPORT ShaderGeometry : public osg::Geometry
  {
    using OsgVec3ArrayRef = osg::ref_ptr<osg::Vec3Array>;
    using OsgVec4ArrayRef = osg::ref_ptr<osg::Vec4Array>;
    using OsgShaderRef = osg::ref_ptr<osg::Shader>;
 
    static OsgShaderGeometryRefVector getSorted( const OsgShaderGeometryRefVector& in );
 
    static void insertSorted( OsgShaderGeometryRefVector& v, OsgShaderGeometryRef shaderGeometry );
 
    bool loadShaders( std::string_view sourcePath );
 
    void createVertexBuffer( size_t size, int renderMode = -1 );
 
    bool bind( size_t location, osg::Array* array, std::string name = "" );
 
    template<typename VertexArrayType>
    VertexArrayType* getVertexAttributeArray( std::string name = "" ) const;
 
    virtual bool initialize();
 
    virtual void update();
 
    OsgVec3ArrayRef vertices{};
 
    protected:
      struct RenderOrder
      {
        enum Preset : agx::UInt16
        {
          EARLY = 1 << 4,
          MIDDLE = 1 << 8,
          LATE = 1 << 12,
        };
      };
 
    protected:
      ShaderGeometry( agx::UInt16 renderOrder );
      virtual ~ShaderGeometry() = default;
 
    private:
      agx::UInt16 m_renderOrder = (agx::UInt16)RenderOrder::MIDDLE;
  };
 
  template<typename VertexArrayType>
  VertexArrayType* ShaderGeometry::getVertexAttributeArray( std::string name /*= ""*/ ) const
  {
    // 'vertices' is at index 0 and is a member of this instance so that
    // should be used explicitly.
    for ( unsigned i = 1u; i < getNumVertexAttribArrays(); ++i ) {
      auto array = getVertexAttribArray( i );
      if ( auto arrayOfType = dynamic_cast<const VertexArrayType*>( array ) )
        if ( arrayOfType->getName() == name )
          return const_cast<VertexArrayType*>( arrayOfType );
    }
 
    return nullptr;
  }
}