Add feature and format support query functions

Author: hypernewbie

src/vrhi_device.cpp

@@ -499,6 +499,28 @@ std::string vhGetDeviceInfo()
     return std::string( buffer );
 }
 
+bool vhQueryFeatureSupport( nvrhi::Feature feature, void* pInfo, size_t infoSize )
+{
+    if ( !g_vhDevice )
+    {
+        VRHI_ERR( "vhQueryFeatureSupport(): Device not initialised.\n" );
+        return false;
+    }
+    std::lock_guard<std::mutex> lock( g_nvRHIStateMutex );
+    return g_vhDevice->queryFeatureSupport( feature, pInfo, infoSize );
+}
+
+nvrhi::FormatSupport vhQueryFormatSupport( nvrhi::Format format )
+{
+    if ( !g_vhDevice )
+    {
+        VRHI_ERR( "vhQueryFormatSupport(): Device not initialised.\n" );
+        return nvrhi::FormatSupport::None;
+    }
+    std::lock_guard<std::mutex> lock( g_nvRHIStateMutex );
+    return g_vhDevice->queryFormatSupport( format );
+}
+
 void vhDispatch( vhStateId stateID, glm::uvec3 workGroupCount )
 {
     VIDL_vhDispatch* cmd = vhCmdAlloc<VIDL_vhDispatch>( stateID, workGroupCount );

test/test_device.cpp

@@ -287,4 +287,49 @@ UTEST( Device, HashingReflectionMembers )
     emptyName.push_back( m4 );
     uint64_t hash7 = vhHashReflectionMembers( emptyName );
     EXPECT_NE( hash7, 0u );
+}
+
+UTEST( Device, QueryFeatureSupport )
+{
+    if ( !g_testInit )
+    {
+        vhInit( g_testInitQuiet );
+        g_testInit = true;
+    }
+
+    // Test basic feature queries that should return deterministic values
+    EXPECT_TRUE( vhQueryFeatureSupport( nvrhi::Feature::ComputeQueue ) ); // Vulkan always has this
+    // Test any bool is valid for this feature
+    bool deferredSupport = vhQueryFeatureSupport( nvrhi::Feature::DeferredCommandLists );
+    // Just verify it returns a valid bool - don't assume specific support
+
+    // Test feature with info struct
+    nvrhi::VariableRateShadingFeatureInfo vrsInfo = {};
+    bool vrsSupported = vhQueryFeatureSupport( nvrhi::Feature::VariableRateShading, &vrsInfo, sizeof(vrsInfo) );
+    if ( vrsSupported )
+    {
+        EXPECT_GT( vrsInfo.shadingRateImageTileSize, 0 ); // Should have valid tile size if supported
+    }
+}
+
+UTEST( Device, QueryFormatSupport )
+{
+    if ( !g_testInit )
+    {
+        vhInit( g_testInitQuiet );
+        g_testInit = true;
+    }
+
+    // Test common format queries
+    nvrhi::FormatSupport rgba8Support = vhQueryFormatSupport( nvrhi::Format::RGBA8_UNORM );
+    EXPECT_NE( rgba8Support, nvrhi::FormatSupport::None ); // Common format should be supported
+    EXPECT_TRUE( ( rgba8Support & nvrhi::FormatSupport::Texture ) != nvrhi::FormatSupport::None );
+    EXPECT_TRUE( ( rgba8Support & nvrhi::FormatSupport::ShaderSample ) != nvrhi::FormatSupport::None );
+
+    // Test format support flags
+    nvrhi::FormatSupport depthSupport = vhQueryFormatSupport( nvrhi::Format::D24S8 );
+    EXPECT_TRUE( ( depthSupport & nvrhi::FormatSupport::DepthStencil ) != nvrhi::FormatSupport::None );
+
+    nvrhi::FormatSupport floatSupport = vhQueryFormatSupport( nvrhi::Format::R32_FLOAT );
+    EXPECT_TRUE( ( floatSupport & nvrhi::FormatSupport::ShaderLoad ) != nvrhi::FormatSupport::None );
 }

vrhi.h

@@ -116,6 +116,18 @@ void vhShutdown( bool quiet = false );
 // Returns a string containing information about the selected physical device and queues.
 std::string vhGetDeviceInfo();
 
+// Queries whether a specific device feature is supported.
+// Returns true if the feature is supported on the current device.
+//
+// For features that require additional information (e.g. VRS tile size), pass an optional
+// output struct via |pInfo| with size |infoSize|. See nvrhi::Feature for available features.
+bool vhQueryFeatureSupport( nvrhi::Feature feature, void* pInfo = nullptr, size_t infoSize = 0 );
+
+// Queries the supported operations for a specific texture/buffer format.
+// Returns a bitmask of nvrhi::FormatSupport flags indicating what operations
+// the format can be used for (e.g. RenderTarget, ShaderSample, UAV, etc).
+nvrhi::FormatSupport vhQueryFormatSupport( nvrhi::Format format );
+
 // Flushes the command queue to the backend.
 //
 // If |wait| is true, blocks until the backend has processed all queued commands.