Fix uses of dynamic_cast on types in reflection API (#731)

The `Type` infrastructure uses a class hierarchy, but blindly `dynamic_cast`ing to a desired case doesn't always give the expected result, because a `Type` could represent a `typedef` (a `NamedExpressionType`) that itself resolves to, e.g, a vector type (a `VectorExpressionType`). In that case a `dynamic_cast<VectorExpressionType*>(someType)` would fail, even though the type logically represents a vector. The `Type::As<T>()` method is designed to handle this case, by "looking through" simple `typedef`s to get at the real definition of a type.

The fix in this case is to use `Type::As<T>()` at various points in the reflection code (`reflection.cpp`) instead of `dynamic_cast`.

This problem surfaced with a `StructuredBuffer<float2>` not reflecting correctly, because the element type (`float2`) is actually a `typedef` (for `vector<float,2>`), so I've included a test case that stresses that case. Getting the right output in the test required tweaking the `slang-reflection-test` tool to produce additional output for resource types (currently narrowed down to only affect structured buffers to avoid large diffs in expected test outputs).

1 files changed, 17 insertions, 17 deletions

diff --git a/source/slang/reflection.cpp b/source/slang/reflection.cpp
index 73b616377..44920fb9f 100644
--- a/source/slang/reflection.cpp
+++ b/source/slang/reflection.cpp
@@ -189,7 +189,7 @@ SLANG_API unsigned int spReflectionType_GetFieldCount(SlangReflectionType* inTyp
 
     // TODO: maybe filter based on kind
 
-    if(auto declRefType = dynamic_cast<DeclRefType*>(type))
+    if(auto declRefType = type->As<DeclRefType>())
     {
         auto declRef = declRefType->declRef;
         if( auto structDeclRef = declRef.As<StructDecl>())
@@ -208,7 +208,7 @@ SLANG_API SlangReflectionVariable* spReflectionType_GetFieldByIndex(SlangReflect
 
     // TODO: maybe filter based on kind
 
-    if(auto declRefType = dynamic_cast<DeclRefType*>(type))
+    if(auto declRefType = type->As<DeclRefType>())
     {
         auto declRef = declRefType->declRef;
         if( auto structDeclRef = declRef.As<StructDecl>())
@@ -226,11 +226,11 @@ SLANG_API size_t spReflectionType_GetElementCount(SlangReflectionType* inType)
     auto type = convert(inType);
     if(!type) return 0;
 
-    if(auto arrayType = dynamic_cast<ArrayExpressionType*>(type))
+    if(auto arrayType = type->As<ArrayExpressionType>())
     {
         return arrayType->ArrayLength ? (size_t) GetIntVal(arrayType->ArrayLength) : 0;
     }
-    else if( auto vectorType = dynamic_cast<VectorExpressionType*>(type))
+    else if( auto vectorType = type->As<VectorExpressionType>())
     {
         return (size_t) GetIntVal(vectorType->elementCount);
     }
@@ -243,19 +243,19 @@ SLANG_API SlangReflectionType* spReflectionType_GetElementType(SlangReflectionTy
     auto type = convert(inType);
     if(!type) return nullptr;
 
-    if(auto arrayType = dynamic_cast<ArrayExpressionType*>(type))
+    if(auto arrayType = type->As<ArrayExpressionType>())
     {
         return (SlangReflectionType*) arrayType->baseType.Ptr();
     }
-    else if( auto constantBufferType = dynamic_cast<ConstantBufferType*>(type))
+    else if( auto constantBufferType = type->As<ConstantBufferType>())
     {
         return convert(constantBufferType->elementType.Ptr());
     }
-    else if( auto vectorType = dynamic_cast<VectorExpressionType*>(type))
+    else if( auto vectorType = type->As<VectorExpressionType>())
     {
         return convert(vectorType->elementType.Ptr());
     }
-    else if( auto matrixType = dynamic_cast<MatrixExpressionType*>(type))
+    else if( auto matrixType = type->As<MatrixExpressionType>())
     {
         return convert(matrixType->getElementType());
     }
@@ -268,15 +268,15 @@ SLANG_API unsigned int spReflectionType_GetRowCount(SlangReflectionType* inType)
     auto type = convert(inType);
     if(!type) return 0;
 
-    if(auto matrixType = dynamic_cast<MatrixExpressionType*>(type))
+    if(auto matrixType = type->As<MatrixExpressionType>())
     {
         return (unsigned int) GetIntVal(matrixType->getRowCount());
     }
-    else if(auto vectorType = dynamic_cast<VectorExpressionType*>(type))
+    else if(auto vectorType = type->As<VectorExpressionType>())
     {
         return 1;
     }
-    else if( auto basicType = dynamic_cast<BasicExpressionType*>(type) )
+    else if( auto basicType = type->As<BasicExpressionType>() )
     {
         return 1;
     }
@@ -289,15 +289,15 @@ SLANG_API unsigned int spReflectionType_GetColumnCount(SlangReflectionType* inTy
     auto type = convert(inType);
     if(!type) return 0;
 
-    if(auto matrixType = dynamic_cast<MatrixExpressionType*>(type))
+    if(auto matrixType = type->As<MatrixExpressionType>())
     {
         return (unsigned int) GetIntVal(matrixType->getColumnCount());
     }
-    else if(auto vectorType = dynamic_cast<VectorExpressionType*>(type))
+    else if(auto vectorType = type->As<VectorExpressionType>())
     {
         return (unsigned int) GetIntVal(vectorType->elementCount);
     }
-    else if( auto basicType = dynamic_cast<BasicExpressionType*>(type) )
+    else if( auto basicType = type->As<BasicExpressionType>() )
     {
         return 1;
     }
@@ -310,16 +310,16 @@ SLANG_API SlangScalarType spReflectionType_GetScalarType(SlangReflectionType* in
     auto type = convert(inType);
     if(!type) return 0;
 
-    if(auto matrixType = dynamic_cast<MatrixExpressionType*>(type))
+    if(auto matrixType = type->As<MatrixExpressionType>())
     {
         type = matrixType->getElementType();
     }
-    else if(auto vectorType = dynamic_cast<VectorExpressionType*>(type))
+    else if(auto vectorType = type->As<VectorExpressionType>())
     {
         type = vectorType->elementType.Ptr();
     }
 
-    if(auto basicType = dynamic_cast<BasicExpressionType*>(type))
+    if(auto basicType = type->As<BasicExpressionType>())
     {
         switch (basicType->baseType)
         {