wgpu/wgpu-native/src/instance.rs

use hal::{self, Instance as _Instance, PhysicalDevice as _PhysicalDevice};

use registry::{HUB, Items, Registry};
use {AdapterId, Device, DeviceId, InstanceId};

#[repr(C)]
#[derive(Copy, Clone, Debug, Hash, Eq, PartialEq)]
pub enum PowerPreference {
    Default = 0,
    LowPower = 1,
    HighPerformance = 2,
}

#[repr(C)]
pub struct AdapterDescriptor {
    pub power_preference: PowerPreference,
}

#[repr(C)]
pub struct Extensions {
    pub anisotropic_filtering: bool,
}

#[repr(C)]
pub struct DeviceDescriptor {
    pub extensions: Extensions,
}

#[no_mangle]
pub extern "C" fn wgpu_create_instance() -> InstanceId {
    #[cfg(any(
        feature = "gfx-backend-vulkan",
        feature = "gfx-backend-dx12",
        feature = "gfx-backend-metal"
    ))]
    {
        let inst = ::back::Instance::create("wgpu", 1);
        HUB.instances.lock().register(inst)
    }
    #[cfg(not(any(
        feature = "gfx-backend-vulkan",
        feature = "gfx-backend-dx12",
        feature = "gfx-backend-metal"
    )))]
    {
        unimplemented!()
    }
}

#[no_mangle]
pub extern "C" fn wgpu_instance_get_adapter(
    instance_id: InstanceId,
    desc: &AdapterDescriptor,
) -> AdapterId {
    let instance_guard = HUB.instances.lock();
    let instance = instance_guard.get(instance_id);
    let (mut low, mut high, mut other) = (None, None, None);
    for adapter in instance.enumerate_adapters() {
        match adapter.info.device_type {
            hal::adapter::DeviceType::IntegratedGpu => low = Some(adapter),
            hal::adapter::DeviceType::DiscreteGpu => high = Some(adapter),
            _ => other = Some(adapter),
        }
    }

    let some = match desc.power_preference {
        PowerPreference::LowPower => low.or(high),
        PowerPreference::HighPerformance | PowerPreference::Default => high.or(low),
    };
    HUB.adapters
        .lock()
        .register(some.or(other).unwrap())
}

#[no_mangle]
pub extern "C" fn wgpu_adapter_create_device(
    adapter_id: AdapterId,
    _desc: &DeviceDescriptor,
) -> DeviceId {
    let mut adapter_guard = HUB.adapters.lock();
    let adapter = adapter_guard.get_mut(adapter_id);
    let (device, queue_group) = adapter.open_with::<_, hal::General>(1, |_qf| true).unwrap();
    let mem_props = adapter.physical_device.memory_properties();
    HUB.devices
        .lock()
        .register(Device::new(device, queue_group, mem_props))
}