glfw-surface-test.lua

local webgpu_cdefs = require("cdefs")
local ffi = require("ffi")

ffi.cdef(webgpu_cdefs)

local glfw_cdefs = [[
    // Platform-specific (don't care)
    typedef void* GLFWwindow;
    typedef void* GLFWmonitor;

    int glfwInit(void);
    void glfwWindowHint(int hint, int value);
    int glfwWindowShouldClose(GLFWwindow window);
    void glfwPollEvents(void);
    void glfwDestroyWindow(GLFWwindow window);
    void glfwTerminate(void);

    GLFWwindow glfwCreateWindow(int width, int height, const char* title, GLFWmonitor monitor, GLFWwindow share);
]]
ffi.cdef(glfw_cdefs)

local glfw_ext_cdefs = [[
	// Custom extension
	WGPUSurface glfwGetWGPUSurface(WGPUInstance instance, GLFWwindow* window);
]]
ffi.cdef(glfw_ext_cdefs)

local webgpu = ffi.load("wgpu_native")
local glfw = ffi.load("glfw3")
local glfwExt = ffi.load("glfw3webgpu")

print(webgpu, glfw, glfwExt)

local desc = ffi.new("WGPUInstanceDescriptor")
-- 	desc.nextInChain = nullptr

local instance = webgpu.wgpuCreateInstance(desc)
if not instance then
	error("Could not initialize WebGPU!")
end

if not glfw.glfwInit() then
	error("Could not initialize GLFW")
end

local GLFW_CLIENT_API = 0x00022001
local GLFW_NO_API = 0
glfw.glfwWindowHint(GLFW_CLIENT_API, GLFW_NO_API)

local window = glfw.glfwCreateWindow(640, 480, "Learn WebGPU", nil, nil)
if not window then
	error("Could not open window!")
end

print("Requesting adaptor...")

-- 	// Utility function provided by glfw3webgpu.h

-- TBD not actually part of glfw... easier if in the runtime :/
local surface = glfwExt.glfwGetWGPUSurface(instance, window)
-- 	WGPUSurface surface = glfwGetWGPUSurface(instance, window)
print(surface)

-- 	// Adapter options: we need the adapter to draw to the window's surface
local adapterOpts = ffi.new("WGPURequestAdapterOptions")
-- 	adapterOpts.nextInChain = nullptr
adapterOpts.compatibleSurface = surface

-- 	// Get the adapter, see the comments in the definition of the body of the
-- 	// requestAdapter function above.

local requestedAdapter
local function onAdapterRequested(status, adapter, message, pUserData)
	print("onAdapterRequested", status, adapter, message, pUserData)
	assert(status == webgpu.WGPURequestAdapterStatus_Success, "Failed to request adapter")
	requestedAdapter = adapter
	-- TBD
end
local userdata = nil -- TBD
webgpu.wgpuInstanceRequestAdapter(instance, adapterOpts, onAdapterRequested, userdata)
print("Got adapter: ", requestedAdapter)

local function inspectAdapter(adapter)
	-- std::vector<WGPUFeatureName> features;
	local featureCount = webgpu.wgpuAdapterEnumerateFeatures(adapter, nil)
	local features = ffi.new("WGPUFeatureName[?]", featureCount)
	webgpu.wgpuAdapterEnumerateFeatures(adapter, features)

	print("Adapter features:")
	for index = 0, tonumber(featureCount) - 1 do
		local feature = features[index]
		print(index + 1, feature)
	end
	-- end

	-- 	WGPUSupportedLimits limits = {};

	local limits = ffi.new("WGPUSupportedLimits")
	-- 	limits.nextInChain = nullptr;
	local success = webgpu.wgpuAdapterGetLimits(adapter, limits)
	if not success then
		error("Failed to get adapter limits")
	end
	print("Adapter limits:")
	print("\tmaxTextureDimension1D: ", limits.limits.maxTextureDimension1D)
	print("\tmaxTextureDimension2D: ", limits.limits.maxTextureDimension2D)
	print("\tmaxTextureDimension3D: ", limits.limits.maxTextureDimension3D)
	print("\tmaxTextureArrayLayers: ", limits.limits.maxTextureArrayLayers)
	print("\tmaxBindGroups: ", limits.limits.maxBindGroups)
	print("\tmaxDynamicUniformBuffersPerPipelineLayout: ", limits.limits.maxDynamicUniformBuffersPerPipelineLayout)
	print("\tmaxDynamicStorageBuffersPerPipelineLayout: ", limits.limits.maxDynamicStorageBuffersPerPipelineLayout)
	print("\tmaxSampledTexturesPerShaderStage: ", limits.limits.maxSampledTexturesPerShaderStage)
	print("\tmaxSamplersPerShaderStage: ", limits.limits.maxSamplersPerShaderStage)
	print("\tmaxStorageBuffersPerShaderStage: ", limits.limits.maxStorageBuffersPerShaderStage)
	print("\tmaxStorageTexturesPerShaderStage: ", limits.limits.maxStorageTexturesPerShaderStage)
	print("\tmaxUniformBuffersPerShaderStage: ", limits.limits.maxUniformBuffersPerShaderStage)
	print("\tmaxUniformBufferBindingSize: ", limits.limits.maxUniformBufferBindingSize)
	print("\tmaxStorageBufferBindingSize: ", limits.limits.maxStorageBufferBindingSize)
	print("\tminUniformBufferOffsetAlignment: ", limits.limits.minUniformBufferOffsetAlignment)
	print("\tminStorageBufferOffsetAlignment: ", limits.limits.minStorageBufferOffsetAlignment)
	print("\tmaxVertexBuffers: ", limits.limits.maxVertexBuffers)
	print("\tmaxVertexAttributes: ", limits.limits.maxVertexAttributes)
	print("\tmaxVertexBufferArrayStride: ", limits.limits.maxVertexBufferArrayStride)
	print("\tmaxInterStageShaderComponents: ", limits.limits.maxInterStageShaderComponents)
	print("\tmaxComputeWorkgroupStorageSize: ", limits.limits.maxComputeWorkgroupStorageSize)
	print("\tmaxComputeInvocationsPerWorkgroup: ", limits.limits.maxComputeInvocationsPerWorkgroup)
	print("\tmaxComputeWorkgroupSizeX: ", limits.limits.maxComputeWorkgroupSizeX)
	print("\tmaxComputeWorkgroupSizeY: ", limits.limits.maxComputeWorkgroupSizeY)
	print("\tmaxComputeWorkgroupSizeZ: ", limits.limits.maxComputeWorkgroupSizeZ)
	print("\tmaxComputeWorkgroupsPerDimension: ", limits.limits.maxComputeWorkgroupsPerDimension)

	-- 	WGPUAdapterProperties properties = {};
	local properties = ffi.new("WGPUAdapterProperties")
	-- 	properties.nextInChain = nullptr;
	webgpu.wgpuAdapterGetProperties(adapter, properties)
	print("Adapter properties:")
	print("\tvendorID: ", properties.vendorID)
	print("\tdeviceID: ", properties.deviceID)
	print("\tname: ", properties.name)
	if properties.driverDescription then
		print("\tdriverDescription: ", properties.driverDescription)
	end
	print("\tadapterType: ", properties.adapterType)
	print("\tbackendType: ", properties.backendType)
end

inspectAdapter(requestedAdapter)

local function runMainLoop()
	local uv = require("uv")
	local timer = uv.new_timer()
	timer:start(0, 16, function() -- starts immediately, repeats every 16 milliseconds
		glfw.glfwPollEvents()
		if glfw.glfwWindowShouldClose(window) ~= 0 then
			timer:stop()
			uv.stop()
		end
	end)
end

runMainLoop()