refactor: impl IndexMut<Handle<Expression>> for FunctionInfo

gfx-rs · Oct 25, 2022 · 6b3dab0 · 6b3dab0
1 parent e6792b1
commit 6b3dab0
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/valid/analyzer.rs b/src/valid/analyzer.rs
@@ -288,6 +288,12 @@ impl ops::Index<Handle<crate::Expression>> for FunctionInfo {
     }
 }
 
+impl ops::IndexMut<Handle<crate::Expression>> for FunctionInfo {
+    fn index_mut(&mut self, handle: Handle<crate::Expression>) -> &mut ExpressionInfo {
+        &mut self.expressions[handle.index()]
+    }
+}
+
 /// Disruptor of the uniform control flow.
 #[derive(Clone, Copy, Debug, thiserror::Error)]
 #[cfg_attr(test, derive(PartialEq))]
@@ -707,7 +713,7 @@ impl FunctionInfo {
         };
 
         let ty = resolve_context.resolve(expression, |h| Ok(&self[h].ty))?;
-        self.expressions[handle.index()] = ExpressionInfo {
+        self[handle] = ExpressionInfo {
             uniformity,
             ref_count: 0,
             assignable_global,