From 0e523e13ba2c6a01685e1d13e7b9100514ed0340 Mon Sep 17 00:00:00 2001
From: Stephen Seo <seo.disparate@gmail.com>
Date: Mon, 4 Mar 2024 12:25:12 +0900
Subject: [PATCH] WIP validation layers implementation

TODO: Rest of https://vulkan-tutorial.com/Drawing_a_triangle/Setup/Validation_layers
---
 src/main.rs | 59 ++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 58 insertions(+), 1 deletion(-)

diff --git a/src/main.rs b/src/main.rs
index acbbfcb..afddcaa 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -15,11 +15,19 @@ mod ffi {
     }
 }
 
-use std::ffi::CString;
+use std::ffi::{CStr, CString};
 
 const WINDOW_WIDTH: i32 = 800;
 const WINDOW_HEIGHT: i32 = 600;
 
+#[cfg(debug_assertions)]
+const ENABLE_VALIDATION_LAYERS: bool = true;
+#[cfg(not(debug_assertions))]
+const ENABLE_VALIDATION_LAYERS: bool = false;
+
+const VALIDATION_LAYER_STR_0: &str = "VK_LAYER_KHRONOS_validation\x00";
+const VALIDATION_LAYERS: [&str; 1] = [VALIDATION_LAYER_STR_0];
+
 struct VulkanApp {
     window: *mut ffi::GLFWwindow,
     vk_instance: ffi::VkInstance,
@@ -56,6 +64,11 @@ impl VulkanApp {
     }
 
     fn init_vulkan(&mut self) {
+        // Check validation layers before creating instance.
+        if ENABLE_VALIDATION_LAYERS && !Self::check_validation_layer_support() {
+            panic!("Validation layers requested, but not available!");
+        }
+
         // Create instance.
         let app_name = CString::new("Vulkan Triangle").unwrap();
         let engine_name = CString::new("No Engine").unwrap();
@@ -103,6 +116,50 @@ impl VulkanApp {
         }
     }
 
+    fn check_validation_layer_support() -> bool {
+        let mut layer_count: u32 = 0;
+        unsafe {
+            ffi::vkEnumerateInstanceLayerProperties(
+                std::ptr::addr_of_mut!(layer_count),
+                std::ptr::null_mut(),
+            );
+        }
+
+        let mut layers: Vec<ffi::VkLayerProperties> = Vec::with_capacity(layer_count as usize);
+        layers.resize(
+            layer_count as usize,
+            ffi::VkLayerProperties {
+                layerName: [0; 256],
+                specVersion: 0,
+                implementationVersion: 0,
+                description: [0; 256],
+            },
+        );
+
+        unsafe {
+            ffi::vkEnumerateInstanceLayerProperties(std::ptr::addr_of_mut!(layer_count), layers.as_mut_ptr());
+        }
+
+        for layer_name in VALIDATION_LAYERS {
+            let mut layer_found = false;
+            let ln_cstr = unsafe { CStr::from_ptr(layer_name.as_ptr() as *const i8) };
+            for layer_prop in &layers {
+                let lp_cstr: &CStr = unsafe { CStr::from_ptr(layer_prop.layerName.as_ptr()) };
+                if ln_cstr == lp_cstr {
+                    layer_found = true;
+                    break;
+                }
+            }
+
+            if !layer_found {
+                return false;
+            }
+        }
+
+        println!("Validation layers available");
+        true
+    }
+
     fn main_loop(&mut self) {
         if self.window.is_null() {
             panic!("ERROR: Cannot execute main loop if window is null!");