diag(naga): clarify select built-in type mismatch messages

jimblandy · Dec 20, 2024 · 6c9f93b · 6c9f93b
1 parent 9ea464b
commit 6c9f93b
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 8 deletions.
diff --git a/naga/src/valid/expression.rs b/naga/src/valid/expression.rs
@@ -54,8 +54,13 @@ pub enum ExpressionError {
         rhs_expr: Handle<crate::Expression>,
         rhs_type: crate::TypeInner,
     },
-    #[error("Selecting is not possible")]
-    InvalidSelectTypes,
+    #[error("Expected selection argument types to match, but reject value of type {reject:?} does not match accept value of value {accept:?}")]
+    SelectValuesTypeMismatch {
+        accept: crate::TypeInner,
+        reject: crate::TypeInner,
+    },
+    #[error("Expected selection condition to be a boolean value, got {actual:?}")]
+    SelectConditionNotABool { actual: crate::TypeInner },
     #[error("Relational argument {0:?} is not a boolean vector")]
     InvalidBooleanVector(Handle<crate::Expression>),
     #[error("Relational argument {0:?} is not a float")]
@@ -901,7 +906,8 @@ impl super::Validator {
             } => {
                 let accept_inner = &resolver[accept];
                 let reject_inner = &resolver[reject];
-                let condition_good = match resolver[condition] {
+                let condition_ty = &resolver[condition];
+                let condition_good = match *condition_ty {
                     Ti::Scalar(Sc {
                         kind: Sk::Bool,
                         width: _,
@@ -928,8 +934,16 @@ impl super::Validator {
                     },
                     _ => false,
                 };
-                if !condition_good || accept_inner != reject_inner {
-                    return Err(ExpressionError::InvalidSelectTypes);
+                if accept_inner != reject_inner {
+                    return Err(ExpressionError::SelectValuesTypeMismatch {
+                        accept: accept_inner.clone(),
+                        reject: reject_inner.clone(),
+                    });
+                }
+                if !condition_good {
+                    return Err(ExpressionError::SelectConditionNotABool {
+                        actual: condition_ty.clone(),
+                    });
                 }
                 ShaderStages::all()
             }

diff --git a/naga/tests/validation.rs b/naga/tests/validation.rs
@@ -650,7 +650,7 @@ error: Entry point main at Compute is invalid
   │         ^^^^^^ naga::Expression [3]
   │
   = Expression [3] is invalid
-  = Selecting is not possible
+  = Expected selection condition to be a boolean value, got Scalar(Scalar { kind: Sint, width: 4 })
 
 ",
         ),
@@ -670,7 +670,7 @@ error: Entry point main at Compute is invalid
   │         ^^^^^^ naga::Expression [3]
   │
   = Expression [3] is invalid
-  = Selecting is not possible
+  = Expected selection argument types to match, but reject value of type Scalar(Scalar { kind: Bool, width: 1 }) does not match accept value of value Scalar(Scalar { kind: Sint, width: 4 })
 
 ",
         ),

diff --git a/naga/tests/wgsl_errors.rs b/naga/tests/wgsl_errors.rs
@@ -1543,7 +1543,7 @@ fn select() {
             naga::valid::ValidationError::Function {
                 name,
                 source: naga::valid::FunctionError::Expression {
-                    source: naga::valid::ExpressionError::InvalidSelectTypes,
+                    source: naga::valid::ExpressionError::SelectConditionNotABool { .. },
                     ..
                 },
                 ..