ShaderModule::single_entry_point()

vulkano/src/shader/mod.rs

@@ -400,6 +400,43 @@ impl ShaderModule {
             })
         })
     }
+
+    /// check for *exactly* one item in the Iterator
+    #[inline]
+    fn single_entry_point_filter<P>(self: &Arc<Self>, mut filter: P) -> Option<EntryPoint>
+    where
+        P: FnMut(&EntryPointInfo) -> bool,
+    {
+        let mut iter = self
+            .entry_point_infos
+            .iter()
+            .enumerate()
+            .filter(|(_, infos)| filter(infos))
+            .map(|(x, _)| x);
+        let info_index = iter.next()?;
+        iter.next().is_none().then(|| EntryPoint {
+            module: self.clone(),
+            info_index,
+        })
+    }
+
+    /// Returns information about the entry point if self only contains a single entry point,
+    /// `None` otherwise.
+    #[inline]
+    pub fn single_entry_point(self: &Arc<Self>) -> Option<EntryPoint> {
+        self.single_entry_point_filter(|_| true)
+    }
+
+    /// Returns information about the entry point if self only contains a single entry point
+    /// with the provided ExecutionModel. Returns `None` if no entry point was found or multiple
+    /// entry points have been found matching the provided ExecutionModel.
+    #[inline]
+    pub fn single_entry_point_of_execution(
+        self: &Arc<Self>,
+        execution: ExecutionModel,
+    ) -> Option<EntryPoint> {
+        self.single_entry_point_filter(|info| ExecutionModel::from(&info.execution) == execution)
+    }
 }
 
 impl Drop for ShaderModule {