use wgt::{Backends, PowerPreference, RequestAdapterOptions};
use crate::{Adapter, Instance};
pub fn backend_bits_from_env() -> Option<Backends> {
Some(
match std::env::var("WGPU_BACKEND")
.as_deref()
.map(str::to_lowercase)
.as_deref()
{
Ok("vulkan") => Backends::VULKAN,
Ok("dx12") => Backends::DX12,
Ok("dx11") => Backends::DX11,
Ok("metal") => Backends::METAL,
Ok("gl") => Backends::GL,
Ok("webgpu") => Backends::BROWSER_WEBGPU,
_ => return None,
},
)
}
pub fn power_preference_from_env() -> Option<PowerPreference> {
Some(
match std::env::var("WGPU_POWER_PREF")
.as_deref()
.map(str::to_lowercase)
.as_deref()
{
Ok("low") => PowerPreference::LowPower,
Ok("high") => PowerPreference::HighPerformance,
_ => return None,
},
)
}
#[cfg(not(target_arch = "wasm32"))]
pub fn initialize_adapter_from_env(instance: &Instance, backend_bits: Backends) -> Option<Adapter> {
let desired_adapter_name = std::env::var("WGPU_ADAPTER_NAME")
.as_deref()
.map(str::to_lowercase)
.ok()?;
let adapters = instance.enumerate_adapters(backend_bits);
let mut chosen_adapter = None;
for adapter in adapters {
let info = adapter.get_info();
if info.name.to_lowercase().contains(&desired_adapter_name) {
chosen_adapter = Some(adapter);
break;
}
}
Some(chosen_adapter.expect("WGPU_ADAPTER_NAME set but no matching adapter found!"))
}
#[cfg(target_arch = "wasm32")]
pub fn initialize_adapter_from_env(
_instance: &Instance,
_backend_bits: Backends,
) -> Option<Adapter> {
None
}
pub async fn initialize_adapter_from_env_or_default(
instance: &Instance,
backend_bits: wgt::Backends,
) -> Option<Adapter> {
match initialize_adapter_from_env(instance, backend_bits) {
Some(a) => Some(a),
None => {
instance
.request_adapter(&RequestAdapterOptions {
power_preference: power_preference_from_env()
.unwrap_or_else(PowerPreference::default),
compatible_surface: None,
})
.await
}
}
}