Add support for VkVideoSession

vulkano-rs · Oct 3, 2023 · cd80a32 · cd80a32
1 parent 70b5e6b
commit cd80a32
Show file tree

Hide file tree

Showing 3 changed files with 407 additions and 4 deletions.
diff --git a/examples/src/bin/video.rs b/examples/src/bin/video.rs
@@ -3,10 +3,11 @@ use std::sync::Arc;
 use vulkano::{
     device::{Device, DeviceCreateInfo, DeviceExtensions, QueueCreateInfo, QueueFlags},
     image::ImageUsage,
-    instance::{Instance, InstanceCreateFlags, InstanceCreateInfo, InstanceExtensions},
+    instance::{Instance, InstanceCreateFlags, InstanceCreateInfo},
     video::{
         VideoDecodeCapabilityFlags, VideoDecodeH264PictureLayoutFlags, VideoDecodeH264ProfileInfo,
-        VideoFormatInfo, VideoProfileInfo, VideoProfileListInfo,
+        VideoFormatInfo, VideoProfileInfo, VideoProfileListInfo, VideoSession,
+        VideoSessionCreateInfo,
     },
     VulkanLibrary,
 };
@@ -151,9 +152,31 @@ fn main() {
         },
     };
 
-    let formats = physical_device
+    let mut formats = physical_device
         .video_format_properties(video_format_info)
         .unwrap();
 
     println!("video formats: {:#?}", formats);
+
+    let format = formats.pop().unwrap();
+    let h264_extension_properties = physical_device
+        .extension_properties()
+        .iter()
+        .find(|&p| p.extension_name == "VK_STD_vulkan_video_codec_h264_decode")
+        .expect("h264 extension not found");
+
+    let video_session_create_info = VideoSessionCreateInfo {
+        queue_family_index: video_queue_family_index,
+        video_profile: profile_info,
+        picture_format: format.format,
+        max_coded_extent: video_caps.max_coded_extent,
+        reference_picture_format: format.format,
+        max_dpb_slots: video_caps.max_dpb_slots,
+        max_active_reference_pictures: video_caps.max_active_reference_pictures,
+        std_header_version: h264_extension_properties.clone(),
+        ..Default::default()
+    };
+
+    let video_session = VideoSession::new(Arc::clone(&device), video_session_create_info).unwrap();
+    println!("video session: {:#?}", video_session);
 }
diff --git a/vulkano/src/device/mod.rs b/vulkano/src/device/mod.rs
@@ -119,6 +119,7 @@ use crate::{
     instance::{Instance, InstanceOwned, InstanceOwnedDebugWrapper},
     macros::{impl_id_counter, vulkan_bitflags},
     memory::ExternalMemoryHandleType,
+    video::{VideoSession, VideoSessionCreateInfo},
     Requires, RequiresAllOf, RequiresOneOf, Validated, ValidationError, Version, VulkanError,
     VulkanObject,
 };