Skip to content

Commit

Permalink
move object wrap back to ext root
Browse files Browse the repository at this point in the history
  • Loading branch information
crowlKats committed Jan 14, 2025
1 parent c26512f commit eb21c76
Show file tree
Hide file tree
Showing 22 changed files with 191 additions and 197 deletions.
4 changes: 2 additions & 2 deletions ext/webgpu/wrap/adapter.rs → ext/webgpu/adapter.rs
Original file line number Diff line number Diff line change
Expand Up @@ -12,8 +12,8 @@ use deno_core::WebIDL;
use tokio::sync::Mutex;

use super::device::GPUDevice;
use crate::wrap::webidl::features_to_feature_names;
use crate::wrap::webidl::GPUFeatureName;
use crate::webidl::features_to_feature_names;
use crate::webidl::GPUFeatureName;
use crate::Instance;

#[derive(WebIDL)]
Expand Down
6 changes: 3 additions & 3 deletions ext/webgpu/wrap/bind_group.rs → ext/webgpu/bind_group.rs
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,9 @@ use deno_core::webidl::WebIdlInterfaceConverter;
use deno_core::GarbageCollected;
use deno_core::WebIDL;

use crate::wrap::buffer::GPUBuffer;
use crate::wrap::sampler::GPUSampler;
use crate::wrap::texture::GPUTextureView;
use crate::buffer::GPUBuffer;
use crate::sampler::GPUSampler;
use crate::texture::GPUTextureView;
use crate::Instance;

pub struct GPUBindGroup {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ use deno_core::op2;
use deno_core::GarbageCollected;
use deno_core::WebIDL;

use crate::wrap::texture::GPUTextureViewDimension;
use crate::texture::GPUTextureViewDimension;
use crate::Instance;

pub struct GPUBindGroupLayout {
Expand Down
File renamed without changes.
File renamed without changes.
Original file line number Diff line number Diff line change
Expand Up @@ -11,14 +11,14 @@ use deno_error::JsErrorBox;
use wgpu_core::command::ImageCopyBuffer;
use wgpu_core::command::PassChannel;

use crate::wrap::buffer::GPUBuffer;
use crate::wrap::command_buffer::GPUCommandBuffer;
use crate::wrap::compute_pass::GPUComputePassEncoder;
use crate::wrap::queue::GPUTexelCopyTextureInfo;
use crate::wrap::render_pass::GPULoadOp;
use crate::wrap::render_pass::GPURenderPassEncoder;
use crate::wrap::render_pass::GPUStoreOp;
use crate::wrap::webidl::GPUExtent3D;
use crate::buffer::GPUBuffer;
use crate::command_buffer::GPUCommandBuffer;
use crate::compute_pass::GPUComputePassEncoder;
use crate::queue::GPUTexelCopyTextureInfo;
use crate::render_pass::GPULoadOp;
use crate::render_pass::GPURenderPassEncoder;
use crate::render_pass::GPUStoreOp;
use crate::webidl::GPUExtent3D;
use crate::Instance;

pub struct GPUCommandEncoder {
Expand Down Expand Up @@ -54,7 +54,7 @@ impl GPUCommandEncoder {
#[cppgc]
fn begin_render_pass(
&self,
#[webidl] descriptor: crate::wrap::render_pass::GPURenderPassDescriptor,
#[webidl] descriptor: crate::render_pass::GPURenderPassDescriptor,
) -> Result<GPURenderPassEncoder, JsErrorBox> {
let color_attachments = Cow::Owned(
descriptor
Expand Down Expand Up @@ -149,7 +149,7 @@ impl GPUCommandEncoder {
#[cppgc]
fn begin_compute_pass(
&self,
#[webidl] descriptor: crate::wrap::compute_pass::GPUComputePassDescriptor,
#[webidl] descriptor: crate::compute_pass::GPUComputePassDescriptor,
) -> GPUComputePassEncoder {
let timestamp_writes =
descriptor.timestamp_writes.map(|timestamp_writes| {
Expand Down Expand Up @@ -348,8 +348,7 @@ impl GPUCommandEncoder {
#[cppgc]
fn finish(
&self,
#[webidl]
descriptor: crate::wrap::command_buffer::GPUCommandBufferDescriptor,
#[webidl] descriptor: crate::command_buffer::GPUCommandBufferDescriptor,
) -> GPUCommandBuffer {
let wgpu_descriptor = wgpu_types::CommandBufferDescriptor {
label: Some(Cow::Owned(descriptor.label.clone())),
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ impl GPUComputePassEncoder {

fn set_pipeline(
&self,
#[webidl] pipeline: Ptr<crate::wrap::compute_pipeline::GPUComputePipeline>,
#[webidl] pipeline: Ptr<crate::compute_pipeline::GPUComputePipeline>,
) {
let err = self
.instance
Expand Down Expand Up @@ -74,7 +74,7 @@ impl GPUComputePassEncoder {

fn dispatch_workgroups_indirect(
&self,
#[webidl] indirect_buffer: Ptr<crate::wrap::buffer::GPUBuffer>,
#[webidl] indirect_buffer: Ptr<crate::buffer::GPUBuffer>,
#[webidl(options(enforce_range = true))] indirect_offset: u64,
) {
let err = self
Expand Down Expand Up @@ -134,7 +134,7 @@ impl GPUComputePassEncoder {
&self,
scope: &mut v8::HandleScope<'a>,
#[webidl(options(enforce_range = true))] index: u32,
#[webidl] bind_group: Nullable<Ptr<crate::wrap::bind_group::GPUBindGroup>>,
#[webidl] bind_group: Nullable<Ptr<crate::bind_group::GPUBindGroup>>,
dynamic_offsets: v8::Local<'a, v8::Value>,
dynamic_offsets_data_start: v8::Local<'a, v8::Value>,
dynamic_offsets_data_length: v8::Local<'a, v8::Value>,
Expand Down Expand Up @@ -225,7 +225,7 @@ pub(crate) struct GPUComputePassDescriptor {
#[derive(WebIDL)]
#[webidl(dictionary)]
pub(crate) struct GPUComputePassTimestampWrites {
pub query_set: Ptr<crate::wrap::query_set::GPUQuerySet>,
pub query_set: Ptr<crate::query_set::GPUQuerySet>,
#[options(enforce_range = true)]
pub beginning_of_pass_write_index: Option<u32>,
#[options(enforce_range = true)]
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,9 +7,9 @@ use deno_core::GarbageCollected;
use deno_core::WebIDL;
use indexmap::IndexMap;

use crate::wrap::bind_group_layout::GPUBindGroupLayout;
use crate::wrap::shader::GPUShaderModule;
use crate::wrap::webidl::GPUPipelineLayoutOrGPUAutoLayoutMode;
use crate::bind_group_layout::GPUBindGroupLayout;
use crate::shader::GPUShaderModule;
use crate::webidl::GPUPipelineLayoutOrGPUAutoLayoutMode;
use crate::Instance;

pub struct GPUComputePipeline {
Expand Down
18 changes: 9 additions & 9 deletions ext/webgpu/wrap/device.rs → ext/webgpu/device.rs
Original file line number Diff line number Diff line change
Expand Up @@ -22,14 +22,14 @@ use super::queue::GPUQueue;
use super::sampler::GPUSampler;
use super::shader::GPUShaderModule;
use super::texture::GPUTexture;
use crate::wrap::adapter::GPUAdapterInfo;
use crate::wrap::adapter::GPUSupportedFeatures;
use crate::wrap::adapter::GPUSupportedLimits;
use crate::wrap::command_encoder::GPUCommandEncoder;
use crate::wrap::query_set::GPUQuerySet;
use crate::wrap::render_bundle::GPURenderBundleEncoder;
use crate::wrap::render_pipeline::GPURenderPipeline;
use crate::wrap::webidl::features_to_feature_names;
use crate::adapter::GPUAdapterInfo;
use crate::adapter::GPUSupportedFeatures;
use crate::adapter::GPUSupportedLimits;
use crate::command_encoder::GPUCommandEncoder;
use crate::query_set::GPUQuerySet;
use crate::render_bundle::GPURenderBundleEncoder;
use crate::render_pipeline::GPURenderPipeline;
use crate::webidl::features_to_feature_names;
use crate::Instance;

pub struct GPUDevice {
Expand Down Expand Up @@ -554,7 +554,7 @@ impl GPUDevice {
#[cppgc]
fn create_query_set(
&self,
#[webidl] descriptor: crate::wrap::query_set::GPUQuerySetDescriptor,
#[webidl] descriptor: crate::query_set::GPUQuerySetDescriptor,
) -> GPUQuerySet {
let wgpu_descriptor = wgpu_core::resource::QuerySetDescriptor {
label: Some(Cow::Owned(descriptor.label.clone())),
Expand Down
File renamed without changes.
160 changes: 132 additions & 28 deletions ext/webgpu/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,48 +2,152 @@
#![cfg(not(target_arch = "wasm32"))]
#![warn(unsafe_op_in_unsafe_fn)]

use std::cell::RefCell;
use std::rc::Rc;
use std::sync::Arc;

use deno_core::cppgc::SameObject;
use deno_core::op2;
use deno_core::GarbageCollected;
use deno_core::OpState;
pub use wgpu_core;
pub use wgpu_types;
use wgpu_types::PowerPreference;

mod adapter;
mod bind_group;
mod bind_group_layout;
mod buffer;
mod command_buffer;
mod command_encoder;
mod compute_pass;
mod compute_pipeline;
mod device;
mod error;
mod pipeline_layout;
mod query_set;
mod queue;
mod render_bundle;
mod render_pass;
mod render_pipeline;
mod sampler;
mod shader;
mod texture;
mod webidl;

pub const UNSTABLE_FEATURE_NAME: &str = "webgpu";

//pub mod byow;
//pub mod surface;
mod wrap;

pub type Instance = std::sync::Arc<wgpu_core::global::Global>;
pub type Instance = Arc<wgpu_core::global::Global>;

deno_core::extension!(
deno_webgpu,
deps = [deno_webidl, deno_web],
ops = [wrap::create_gpu],
ops = [create_gpu],
objects = [
wrap::GPU,
wrap::adapter::GPUAdapter,
wrap::adapter::GPUAdapterInfo,
wrap::bind_group::GPUBindGroup,
wrap::bind_group_layout::GPUBindGroupLayout,
wrap::buffer::GPUBuffer,
wrap::command_buffer::GPUCommandBuffer,
wrap::command_encoder::GPUCommandEncoder,
wrap::compute_pass::GPUComputePassEncoder,
wrap::compute_pipeline::GPUComputePipeline,
wrap::device::GPUDevice,
wrap::device::GPUDeviceLostInfo,
wrap::pipeline_layout::GPUPipelineLayout,
wrap::query_set::GPUQuerySet,
wrap::queue::GPUQueue,
wrap::render_bundle::GPURenderBundle,
wrap::render_bundle::GPURenderBundleEncoder,
wrap::render_pass::GPURenderPassEncoder,
wrap::render_pipeline::GPURenderPipeline,
wrap::sampler::GPUSampler,
wrap::shader::GPUShaderModule,
wrap::adapter::GPUSupportedFeatures,
wrap::adapter::GPUSupportedLimits,
wrap::texture::GPUTexture,
wrap::texture::GPUTextureView,
GPU,
adapter::GPUAdapter,
adapter::GPUAdapterInfo,
bind_group::GPUBindGroup,
bind_group_layout::GPUBindGroupLayout,
buffer::GPUBuffer,
command_buffer::GPUCommandBuffer,
command_encoder::GPUCommandEncoder,
compute_pass::GPUComputePassEncoder,
compute_pipeline::GPUComputePipeline,
device::GPUDevice,
device::GPUDeviceLostInfo,
pipeline_layout::GPUPipelineLayout,
query_set::GPUQuerySet,
queue::GPUQueue,
render_bundle::GPURenderBundle,
render_bundle::GPURenderBundleEncoder,
render_pass::GPURenderPassEncoder,
render_pipeline::GPURenderPipeline,
sampler::GPUSampler,
shader::GPUShaderModule,
adapter::GPUSupportedFeatures,
adapter::GPUSupportedLimits,
texture::GPUTexture,
texture::GPUTextureView,
],
esm = ["00_init.js"],
lazy_loaded_esm = ["01_webgpu.js"],
);

#[op2]
#[cppgc]
pub fn create_gpu() -> GPU {
GPU
}

pub struct GPU;

impl GarbageCollected for GPU {}

#[op2]
impl GPU {
#[async_method]
#[cppgc]
async fn request_adapter(
&self,
state: Rc<RefCell<OpState>>,
#[webidl] options: adapter::GPURequestAdapterOptions,
) -> Option<adapter::GPUAdapter> {
let mut state = state.borrow_mut();

let backends = std::env::var("DENO_WEBGPU_BACKEND").map_or_else(
|_| wgpu_types::Backends::all(),
|s| wgpu_core::instance::parse_backends_from_comma_list(&s),
);
let instance = if let Some(instance) = state.try_borrow::<Instance>() {
instance
} else {
state.put(Arc::new(wgpu_core::global::Global::new(
"webgpu",
wgpu_types::InstanceDescriptor {
backends,
flags: wgpu_types::InstanceFlags::from_build_config(),
dx12_shader_compiler: wgpu_types::Dx12Compiler::Fxc,
gles_minor_version: wgpu_types::Gles3MinorVersion::default(),
},
)));
state.borrow::<Instance>()
};

let descriptor = wgpu_core::instance::RequestAdapterOptions {
power_preference: options
.power_preference
.map(|pp| match pp {
adapter::GPUPowerPreference::LowPower => PowerPreference::LowPower,
adapter::GPUPowerPreference::HighPerformance => {
PowerPreference::HighPerformance
}
})
.unwrap_or_default(),
force_fallback_adapter: options.force_fallback_adapter,
compatible_surface: None, // windowless
};
let id = instance.request_adapter(&descriptor, backends, None).ok()?;

Some(adapter::GPUAdapter {
instance: instance.clone(),
features: SameObject::new(),
limits: SameObject::new(),
info: Rc::new(SameObject::new()),
id,
})
}

#[string]
fn getPreferredCanvasFormat(&self) -> &'static str {
// https://github.com/mozilla/gecko-dev/blob/b75080bb8b11844d18cb5f9ac6e68a866ef8e243/dom/webgpu/Instance.h#L42-L47
if cfg!(target_os = "android") {
texture::GPUTextureFormat::Rgba8unorm.as_str()
} else {
texture::GPUTextureFormat::Bgra8unorm.as_str()
}
}
}
File renamed without changes.
File renamed without changes.
12 changes: 6 additions & 6 deletions ext/webgpu/wrap/queue.rs → ext/webgpu/queue.rs
Original file line number Diff line number Diff line change
Expand Up @@ -6,12 +6,12 @@ use deno_core::GarbageCollected;
use deno_core::WebIDL;
use deno_error::JsErrorBox;

use crate::wrap::buffer::GPUBuffer;
use crate::wrap::command_buffer::GPUCommandBuffer;
use crate::wrap::texture::GPUTexture;
use crate::wrap::texture::GPUTextureAspect;
use crate::wrap::webidl::GPUExtent3D;
use crate::wrap::webidl::GPUOrigin3D;
use crate::buffer::GPUBuffer;
use crate::command_buffer::GPUCommandBuffer;
use crate::texture::GPUTexture;
use crate::texture::GPUTextureAspect;
use crate::webidl::GPUExtent3D;
use crate::webidl::GPUOrigin3D;
use crate::Instance;

pub struct GPUQueue {
Expand Down
Loading

0 comments on commit eb21c76

Please sign in to comment.