1use super::conv;
2use crate::auxil::map_naga_stage;
3use glow::HasContext;
4use std::{
5 convert::TryInto,
6 ptr,
7 sync::{Arc, Mutex},
8};
9
10use arrayvec::ArrayVec;
11#[cfg(not(target_arch = "wasm32"))]
12use std::mem;
13use std::sync::atomic::Ordering;
14
15type ShaderStage<'a> = (
16 naga::ShaderStage,
17 &'a crate::ProgrammableStage<'a, super::Api>,
18);
19type NameBindingMap = rustc_hash::FxHashMap<String, (super::BindingRegister, u8)>;
20
21struct CompilationContext<'a> {
22 layout: &'a super::PipelineLayout,
23 sampler_map: &'a mut super::SamplerBindMap,
24 name_binding_map: &'a mut NameBindingMap,
25 multiview: Option<std::num::NonZeroU32>,
26}
27
28impl CompilationContext<'_> {
29 fn consume_reflection(
30 self,
31 module: &naga::Module,
32 ep_info: &naga::valid::FunctionInfo,
33 reflection_info: naga::back::glsl::ReflectionInfo,
34 ) {
35 for (handle, var) in module.global_variables.iter() {
36 if ep_info[handle].is_empty() {
37 continue;
38 }
39 let register = match var.space {
40 naga::AddressSpace::Uniform => super::BindingRegister::UniformBuffers,
41 naga::AddressSpace::Storage { .. } => super::BindingRegister::StorageBuffers,
42 _ => continue,
43 };
44
45 let br = var.binding.as_ref().unwrap();
46 let slot = self.layout.get_slot(br);
47
48 let name = match reflection_info.uniforms.get(&handle) {
49 Some(name) => name.clone(),
50 None => continue,
51 };
52 log::debug!(
53 "Rebind buffer: {:?} -> {}, register={:?}, slot={}",
54 var.name.as_ref(),
55 &name,
56 register,
57 slot
58 );
59 self.name_binding_map.insert(name, (register, slot));
60 }
61
62 for (name, mapping) in reflection_info.texture_mapping {
63 let var = &module.global_variables[mapping.texture];
64 let register = match module.types[var.ty].inner {
65 naga::TypeInner::Image {
66 class: naga::ImageClass::Storage { .. },
67 ..
68 } => super::BindingRegister::Images,
69 _ => super::BindingRegister::Textures,
70 };
71
72 let tex_br = var.binding.as_ref().unwrap();
73 let texture_linear_index = self.layout.get_slot(tex_br);
74
75 self.name_binding_map
76 .insert(name, (register, texture_linear_index));
77 if let Some(sampler_handle) = mapping.sampler {
78 let sam_br = module.global_variables[sampler_handle]
79 .binding
80 .as_ref()
81 .unwrap();
82 let sampler_linear_index = self.layout.get_slot(sam_br);
83 self.sampler_map[texture_linear_index as usize] = Some(sampler_linear_index);
84 }
85 }
86 }
87}
88
89impl super::Device {
90 #[cfg(any(not(target_arch = "wasm32"), target_os = "emscripten"))]
98 pub unsafe fn texture_from_raw(
99 &self,
100 name: std::num::NonZeroU32,
101 desc: &crate::TextureDescriptor,
102 drop_guard: Option<crate::DropGuard>,
103 ) -> super::Texture {
104 let (target, _, is_cubemap) = super::Texture::get_info_from_desc(desc);
105
106 super::Texture {
107 inner: super::TextureInner::Texture {
108 raw: glow::NativeTexture(name),
109 target,
110 },
111 drop_guard,
112 mip_level_count: desc.mip_level_count,
113 array_layer_count: desc.array_layer_count(),
114 format: desc.format,
115 format_desc: self.shared.describe_texture_format(desc.format),
116 copy_size: desc.copy_extent(),
117 is_cubemap,
118 }
119 }
120
121 #[cfg(any(not(target_arch = "wasm32"), target_os = "emscripten"))]
129 pub unsafe fn texture_from_raw_renderbuffer(
130 &self,
131 name: std::num::NonZeroU32,
132 desc: &crate::TextureDescriptor,
133 drop_guard: Option<crate::DropGuard>,
134 ) -> super::Texture {
135 super::Texture {
136 inner: super::TextureInner::Renderbuffer {
137 raw: glow::NativeRenderbuffer(name),
138 },
139 drop_guard,
140 mip_level_count: desc.mip_level_count,
141 array_layer_count: desc.array_layer_count(),
142 format: desc.format,
143 format_desc: self.shared.describe_texture_format(desc.format),
144 copy_size: desc.copy_extent(),
145 is_cubemap: false,
146 }
147 }
148
149 unsafe fn compile_shader(
150 gl: &glow::Context,
151 shader: &str,
152 naga_stage: naga::ShaderStage,
153 #[cfg_attr(target_arch = "wasm32", allow(unused))] label: Option<&str>,
154 ) -> Result<glow::Shader, crate::PipelineError> {
155 let target = match naga_stage {
156 naga::ShaderStage::Vertex => glow::VERTEX_SHADER,
157 naga::ShaderStage::Fragment => glow::FRAGMENT_SHADER,
158 naga::ShaderStage::Compute => glow::COMPUTE_SHADER,
159 };
160
161 let raw = unsafe { gl.create_shader(target) }.unwrap();
162 #[cfg(not(target_arch = "wasm32"))]
163 if gl.supports_debug() {
164 let name = unsafe { mem::transmute(raw) };
167 unsafe { gl.object_label(glow::SHADER, name, label) };
168 }
169
170 unsafe { gl.shader_source(raw, shader) };
171 unsafe { gl.compile_shader(raw) };
172
173 log::info!("\tCompiled shader {:?}", raw);
174
175 let compiled_ok = unsafe { gl.get_shader_compile_status(raw) };
176 let msg = unsafe { gl.get_shader_info_log(raw) };
177 if compiled_ok {
178 if !msg.is_empty() {
179 log::warn!("\tCompile: {}", msg);
180 }
181 Ok(raw)
182 } else {
183 Err(crate::PipelineError::Linkage(
184 map_naga_stage(naga_stage),
185 msg,
186 ))
187 }
188 }
189
190 fn create_shader(
191 gl: &glow::Context,
192 naga_stage: naga::ShaderStage,
193 stage: &crate::ProgrammableStage<super::Api>,
194 context: CompilationContext,
195 ) -> Result<glow::Shader, crate::PipelineError> {
196 use naga::back::glsl;
197 let pipeline_options = glsl::PipelineOptions {
198 shader_stage: naga_stage,
199 entry_point: stage.entry_point.to_string(),
200 multiview: context.multiview,
201 };
202
203 let shader = &stage.module.naga;
204 let entry_point_index = shader
205 .module
206 .entry_points
207 .iter()
208 .position(|ep| ep.name.as_str() == stage.entry_point)
209 .ok_or(crate::PipelineError::EntryPoint(naga_stage))?;
210
211 use naga::proc::BoundsCheckPolicy;
212 let version = gl.version();
214 let image_check = if !version.is_embedded && (version.major, version.minor) >= (1, 3) {
215 BoundsCheckPolicy::ReadZeroSkipWrite
216 } else {
217 BoundsCheckPolicy::Unchecked
218 };
219
220 let policies = naga::proc::BoundsCheckPolicies {
222 index: BoundsCheckPolicy::Unchecked,
223 buffer: BoundsCheckPolicy::Unchecked,
224 image_load: image_check,
225 image_store: image_check,
226 binding_array: BoundsCheckPolicy::Unchecked,
227 };
228
229 let mut output = String::new();
230 let mut writer = glsl::Writer::new(
231 &mut output,
232 &shader.module,
233 &shader.info,
234 &context.layout.naga_options,
235 &pipeline_options,
236 policies,
237 )
238 .map_err(|e| {
239 let msg = format!("{e}");
240 crate::PipelineError::Linkage(map_naga_stage(naga_stage), msg)
241 })?;
242
243 let reflection_info = writer.write().map_err(|e| {
244 let msg = format!("{e}");
245 crate::PipelineError::Linkage(map_naga_stage(naga_stage), msg)
246 })?;
247
248 log::debug!("Naga generated shader:\n{}", output);
249
250 context.consume_reflection(
251 &shader.module,
252 shader.info.get_entry_point(entry_point_index),
253 reflection_info,
254 );
255
256 unsafe { Self::compile_shader(gl, &output, naga_stage, stage.module.label.as_deref()) }
257 }
258
259 unsafe fn create_pipeline<'a>(
260 &self,
261 gl: &glow::Context,
262 shaders: ArrayVec<ShaderStage<'a>, 3>,
263 layout: &super::PipelineLayout,
264 #[cfg_attr(target_arch = "wasm32", allow(unused))] label: Option<&str>,
265 multiview: Option<std::num::NonZeroU32>,
266 ) -> Result<Arc<super::PipelineInner>, crate::PipelineError> {
267 let mut program_stages = ArrayVec::new();
268 let mut group_to_binding_to_slot = Vec::with_capacity(layout.group_infos.len());
269 for group in &*layout.group_infos {
270 group_to_binding_to_slot.push(group.binding_to_slot.clone());
271 }
272 for &(naga_stage, stage) in &shaders {
273 program_stages.push(super::ProgramStage {
274 naga_stage: naga_stage.to_owned(),
275 shader_id: stage.module.id,
276 entry_point: stage.entry_point.to_owned(),
277 });
278 }
279 let glsl_version = match self.shared.shading_language_version {
280 naga::back::glsl::Version::Embedded { version, .. } => version,
281 naga::back::glsl::Version::Desktop(_) => unreachable!(),
282 };
283 let mut guard = self
284 .shared
285 .program_cache
286 .try_lock()
287 .expect("Couldn't acquire program_cache lock");
288 let program = guard
291 .entry(super::ProgramCacheKey {
292 stages: program_stages,
293 group_to_binding_to_slot: group_to_binding_to_slot.into_boxed_slice(),
294 })
295 .or_insert_with(|| unsafe {
296 Self::create_program(
297 gl,
298 shaders,
299 layout,
300 label,
301 multiview,
302 glsl_version,
303 self.shared.private_caps,
304 )
305 })
306 .to_owned()?;
307 drop(guard);
308
309 Ok(program)
310 }
311
312 unsafe fn create_program<'a>(
313 gl: &glow::Context,
314 shaders: ArrayVec<ShaderStage<'a>, 3>,
315 layout: &super::PipelineLayout,
316 #[cfg_attr(target_arch = "wasm32", allow(unused))] label: Option<&str>,
317 multiview: Option<std::num::NonZeroU32>,
318 glsl_version: u16,
319 private_caps: super::PrivateCapabilities,
320 ) -> Result<Arc<super::PipelineInner>, crate::PipelineError> {
321 let program = unsafe { gl.create_program() }.unwrap();
322 #[cfg(not(target_arch = "wasm32"))]
323 if let Some(label) = label {
324 if gl.supports_debug() {
325 let name = unsafe { mem::transmute(program) };
326 unsafe { gl.object_label(glow::PROGRAM, name, Some(label)) };
327 }
328 }
329
330 let mut name_binding_map = NameBindingMap::default();
331 let mut sampler_map = [None; super::MAX_TEXTURE_SLOTS];
332 let mut has_stages = wgt::ShaderStages::empty();
333 let mut shaders_to_delete = arrayvec::ArrayVec::<_, 3>::new();
334
335 for (naga_stage, stage) in shaders {
336 has_stages |= map_naga_stage(naga_stage);
337 let context = CompilationContext {
338 layout,
339 sampler_map: &mut sampler_map,
340 name_binding_map: &mut name_binding_map,
341 multiview,
342 };
343
344 let shader = Self::create_shader(gl, naga_stage, stage, context)?;
345 shaders_to_delete.push(shader);
346 }
347
348 if has_stages == wgt::ShaderStages::VERTEX {
350 let shader_src = format!("#version {glsl_version} es \n void main(void) {{}}",);
351 log::info!("Only vertex shader is present. Creating an empty fragment shader",);
352 let shader = unsafe {
353 Self::compile_shader(
354 gl,
355 &shader_src,
356 naga::ShaderStage::Fragment,
357 Some("(wgpu internal) dummy fragment shader"),
358 )
359 }?;
360 shaders_to_delete.push(shader);
361 }
362
363 for &shader in shaders_to_delete.iter() {
364 unsafe { gl.attach_shader(program, shader) };
365 }
366 unsafe { gl.link_program(program) };
367
368 for shader in shaders_to_delete {
369 unsafe { gl.delete_shader(shader) };
370 }
371
372 log::info!("\tLinked program {:?}", program);
373
374 let linked_ok = unsafe { gl.get_program_link_status(program) };
375 let msg = unsafe { gl.get_program_info_log(program) };
376 if !linked_ok {
377 return Err(crate::PipelineError::Linkage(has_stages, msg));
378 }
379 if !msg.is_empty() {
380 log::warn!("\tLink: {}", msg);
381 }
382
383 if !private_caps.contains(super::PrivateCapabilities::SHADER_BINDING_LAYOUT) {
384 unsafe { gl.use_program(Some(program)) };
387 for (ref name, (register, slot)) in name_binding_map {
388 log::trace!("Get binding {:?} from program {:?}", name, program);
389 match register {
390 super::BindingRegister::UniformBuffers => {
391 let index = unsafe { gl.get_uniform_block_index(program, name) }.unwrap();
392 unsafe { gl.uniform_block_binding(program, index, slot as _) };
393 }
394 super::BindingRegister::StorageBuffers => {
395 let index =
396 unsafe { gl.get_shader_storage_block_index(program, name) }.unwrap();
397 log::error!(
398 "Unable to re-map shader storage block {} to {}",
399 name,
400 index
401 );
402 return Err(crate::DeviceError::Lost.into());
403 }
404 super::BindingRegister::Textures | super::BindingRegister::Images => {
405 let location = unsafe { gl.get_uniform_location(program, name) };
406 unsafe { gl.uniform_1_i32(location.as_ref(), slot as _) };
407 }
408 }
409 }
410 }
411
412 let mut uniforms: [super::UniformDesc; super::MAX_PUSH_CONSTANTS] =
413 [None; super::MAX_PUSH_CONSTANTS].map(|_: Option<()>| Default::default());
414 let count = unsafe { gl.get_active_uniforms(program) };
415 let mut offset = 0;
416
417 for uniform in 0..count {
418 let glow::ActiveUniform { utype, name, .. } =
419 unsafe { gl.get_active_uniform(program, uniform) }.unwrap();
420
421 if conv::is_opaque_type(utype) {
422 continue;
423 }
424
425 if let Some(location) = unsafe { gl.get_uniform_location(program, &name) } {
426 if uniforms[offset / 4].location.is_some() {
427 panic!("Offset already occupied")
428 }
429
430 let uniform_size = conv::uniform_byte_size(utype);
432
433 uniforms[offset / 4] = super::UniformDesc {
434 location: Some(location),
435 size: uniform_size,
436 utype,
437 };
438
439 offset += uniform_size as usize;
440 }
441 }
442
443 Ok(Arc::new(super::PipelineInner {
444 program,
445 sampler_map,
446 uniforms,
447 }))
448 }
449}
450
451impl crate::Device<super::Api> for super::Device {
452 unsafe fn exit(self, queue: super::Queue) {
453 let gl = &self.shared.context.lock();
454 unsafe { gl.delete_vertex_array(self.main_vao) };
455 unsafe { gl.delete_framebuffer(queue.draw_fbo) };
456 unsafe { gl.delete_framebuffer(queue.copy_fbo) };
457 unsafe { gl.delete_buffer(queue.zero_buffer) };
458 }
459
460 unsafe fn create_buffer(
461 &self,
462 desc: &crate::BufferDescriptor,
463 ) -> Result<super::Buffer, crate::DeviceError> {
464 let target = if desc.usage.contains(crate::BufferUses::INDEX) {
465 glow::ELEMENT_ARRAY_BUFFER
466 } else {
467 glow::ARRAY_BUFFER
468 };
469
470 let emulate_map = self
471 .shared
472 .workarounds
473 .contains(super::Workarounds::EMULATE_BUFFER_MAP)
474 || !self
475 .shared
476 .private_caps
477 .contains(super::PrivateCapabilities::BUFFER_ALLOCATION);
478
479 if emulate_map && desc.usage.intersects(crate::BufferUses::MAP_WRITE) {
480 return Ok(super::Buffer {
481 raw: None,
482 target,
483 size: desc.size,
484 map_flags: 0,
485 data: Some(Arc::new(Mutex::new(vec![0; desc.size as usize]))),
486 });
487 }
488
489 let gl = &self.shared.context.lock();
490
491 let target = if desc.usage.contains(crate::BufferUses::INDEX) {
492 glow::ELEMENT_ARRAY_BUFFER
493 } else {
494 glow::ARRAY_BUFFER
495 };
496
497 let is_host_visible = desc
498 .usage
499 .intersects(crate::BufferUses::MAP_READ | crate::BufferUses::MAP_WRITE);
500 let is_coherent = desc
501 .memory_flags
502 .contains(crate::MemoryFlags::PREFER_COHERENT);
503
504 let mut map_flags = 0;
505 if desc.usage.contains(crate::BufferUses::MAP_READ) {
506 map_flags |= glow::MAP_READ_BIT;
507 }
508 if desc.usage.contains(crate::BufferUses::MAP_WRITE) {
509 map_flags |= glow::MAP_WRITE_BIT;
510 }
511
512 let raw = Some(unsafe { gl.create_buffer() }.map_err(|_| crate::DeviceError::OutOfMemory)?);
513 unsafe { gl.bind_buffer(target, raw) };
514 let raw_size = desc
515 .size
516 .try_into()
517 .map_err(|_| crate::DeviceError::OutOfMemory)?;
518
519 if self
520 .shared
521 .private_caps
522 .contains(super::PrivateCapabilities::BUFFER_ALLOCATION)
523 {
524 if is_host_visible {
525 map_flags |= glow::MAP_PERSISTENT_BIT;
526 if is_coherent {
527 map_flags |= glow::MAP_COHERENT_BIT;
528 }
529 }
530 unsafe { gl.buffer_storage(target, raw_size, None, map_flags) };
531 } else {
532 assert!(!is_coherent);
533 let usage = if is_host_visible {
534 if desc.usage.contains(crate::BufferUses::MAP_READ) {
535 glow::STREAM_READ
536 } else {
537 glow::DYNAMIC_DRAW
538 }
539 } else {
540 glow::DYNAMIC_DRAW
544 };
545 unsafe { gl.buffer_data_size(target, raw_size, usage) };
546 }
547
548 unsafe { gl.bind_buffer(target, None) };
549
550 if !is_coherent && desc.usage.contains(crate::BufferUses::MAP_WRITE) {
551 map_flags |= glow::MAP_FLUSH_EXPLICIT_BIT;
552 }
553 #[cfg(not(target_arch = "wasm32"))]
556 if let Some(label) = desc.label {
557 if gl.supports_debug() {
558 let name = unsafe { mem::transmute(raw) };
559 unsafe { gl.object_label(glow::BUFFER, name, Some(label)) };
560 }
561 }
562
563 let data = if emulate_map && desc.usage.contains(crate::BufferUses::MAP_READ) {
564 Some(Arc::new(Mutex::new(vec![0; desc.size as usize])))
565 } else {
566 None
567 };
568
569 Ok(super::Buffer {
570 raw,
571 target,
572 size: desc.size,
573 map_flags,
574 data,
575 })
576 }
577 unsafe fn destroy_buffer(&self, buffer: super::Buffer) {
578 if let Some(raw) = buffer.raw {
579 let gl = &self.shared.context.lock();
580 unsafe { gl.delete_buffer(raw) };
581 }
582 }
583
584 unsafe fn map_buffer(
585 &self,
586 buffer: &super::Buffer,
587 range: crate::MemoryRange,
588 ) -> Result<crate::BufferMapping, crate::DeviceError> {
589 let is_coherent = buffer.map_flags & glow::MAP_COHERENT_BIT != 0;
590 let ptr = match buffer.raw {
591 None => {
592 let mut vec = buffer.data.as_ref().unwrap().lock().unwrap();
593 let slice = &mut vec.as_mut_slice()[range.start as usize..range.end as usize];
594 slice.as_mut_ptr()
595 }
596 Some(raw) => {
597 let gl = &self.shared.context.lock();
598 unsafe { gl.bind_buffer(buffer.target, Some(raw)) };
599 let ptr = if let Some(ref map_read_allocation) = buffer.data {
600 let mut guard = map_read_allocation.lock().unwrap();
601 let slice = guard.as_mut_slice();
602 unsafe { self.shared.get_buffer_sub_data(gl, buffer.target, 0, slice) };
603 slice.as_mut_ptr()
604 } else {
605 unsafe {
606 gl.map_buffer_range(
607 buffer.target,
608 range.start as i32,
609 (range.end - range.start) as i32,
610 buffer.map_flags,
611 )
612 }
613 };
614 unsafe { gl.bind_buffer(buffer.target, None) };
615 ptr
616 }
617 };
618 Ok(crate::BufferMapping {
619 ptr: ptr::NonNull::new(ptr).ok_or(crate::DeviceError::Lost)?,
620 is_coherent,
621 })
622 }
623 unsafe fn unmap_buffer(&self, buffer: &super::Buffer) -> Result<(), crate::DeviceError> {
624 if let Some(raw) = buffer.raw {
625 if buffer.data.is_none() {
626 let gl = &self.shared.context.lock();
627 unsafe { gl.bind_buffer(buffer.target, Some(raw)) };
628 unsafe { gl.unmap_buffer(buffer.target) };
629 unsafe { gl.bind_buffer(buffer.target, None) };
630 }
631 }
632 Ok(())
633 }
634 unsafe fn flush_mapped_ranges<I>(&self, buffer: &super::Buffer, ranges: I)
635 where
636 I: Iterator<Item = crate::MemoryRange>,
637 {
638 if let Some(raw) = buffer.raw {
639 let gl = &self.shared.context.lock();
640 unsafe { gl.bind_buffer(buffer.target, Some(raw)) };
641 for range in ranges {
642 unsafe {
643 gl.flush_mapped_buffer_range(
644 buffer.target,
645 range.start as i32,
646 (range.end - range.start) as i32,
647 )
648 };
649 }
650 }
651 }
652 unsafe fn invalidate_mapped_ranges<I>(&self, _buffer: &super::Buffer, _ranges: I) {
653 }
655
656 unsafe fn create_texture(
657 &self,
658 desc: &crate::TextureDescriptor,
659 ) -> Result<super::Texture, crate::DeviceError> {
660 let gl = &self.shared.context.lock();
661
662 let render_usage = crate::TextureUses::COLOR_TARGET
663 | crate::TextureUses::DEPTH_STENCIL_WRITE
664 | crate::TextureUses::DEPTH_STENCIL_READ;
665 let format_desc = self.shared.describe_texture_format(desc.format);
666
667 let (inner, is_cubemap) = if render_usage.contains(desc.usage)
668 && desc.dimension == wgt::TextureDimension::D2
669 && desc.size.depth_or_array_layers == 1
670 {
671 let raw = unsafe { gl.create_renderbuffer().unwrap() };
672 unsafe { gl.bind_renderbuffer(glow::RENDERBUFFER, Some(raw)) };
673 if desc.sample_count > 1 {
674 unsafe {
675 gl.renderbuffer_storage_multisample(
676 glow::RENDERBUFFER,
677 desc.sample_count as i32,
678 format_desc.internal,
679 desc.size.width as i32,
680 desc.size.height as i32,
681 )
682 };
683 } else {
684 unsafe {
685 gl.renderbuffer_storage(
686 glow::RENDERBUFFER,
687 format_desc.internal,
688 desc.size.width as i32,
689 desc.size.height as i32,
690 )
691 };
692 }
693
694 #[cfg(not(target_arch = "wasm32"))]
695 if let Some(label) = desc.label {
696 if gl.supports_debug() {
697 let name = unsafe { mem::transmute(raw) };
698 unsafe { gl.object_label(glow::RENDERBUFFER, name, Some(label)) };
699 }
700 }
701
702 unsafe { gl.bind_renderbuffer(glow::RENDERBUFFER, None) };
703 (super::TextureInner::Renderbuffer { raw }, false)
704 } else {
705 let raw = unsafe { gl.create_texture().unwrap() };
706 let (target, is_3d, is_cubemap) = super::Texture::get_info_from_desc(desc);
707
708 unsafe { gl.bind_texture(target, Some(raw)) };
709 match desc.format.sample_type(None) {
711 Some(
712 wgt::TextureSampleType::Float { filterable: false }
713 | wgt::TextureSampleType::Uint
714 | wgt::TextureSampleType::Sint,
715 ) => {
716 unsafe {
718 gl.tex_parameter_i32(target, glow::TEXTURE_MIN_FILTER, glow::NEAREST as i32)
719 };
720 unsafe {
721 gl.tex_parameter_i32(target, glow::TEXTURE_MAG_FILTER, glow::NEAREST as i32)
722 };
723 }
724 _ => {}
725 }
726
727 if is_3d {
728 unsafe {
729 gl.tex_storage_3d(
730 target,
731 desc.mip_level_count as i32,
732 format_desc.internal,
733 desc.size.width as i32,
734 desc.size.height as i32,
735 desc.size.depth_or_array_layers as i32,
736 )
737 };
738 } else if desc.sample_count > 1 {
739 unsafe {
740 gl.tex_storage_2d_multisample(
741 target,
742 desc.sample_count as i32,
743 format_desc.internal,
744 desc.size.width as i32,
745 desc.size.height as i32,
746 true,
747 )
748 };
749 } else {
750 unsafe {
751 gl.tex_storage_2d(
752 target,
753 desc.mip_level_count as i32,
754 format_desc.internal,
755 desc.size.width as i32,
756 desc.size.height as i32,
757 )
758 };
759 }
760
761 #[cfg(not(target_arch = "wasm32"))]
762 if let Some(label) = desc.label {
763 if gl.supports_debug() {
764 let name = unsafe { mem::transmute(raw) };
765 unsafe { gl.object_label(glow::TEXTURE, name, Some(label)) };
766 }
767 }
768
769 unsafe { gl.bind_texture(target, None) };
770 (super::TextureInner::Texture { raw, target }, is_cubemap)
771 };
772
773 Ok(super::Texture {
774 inner,
775 drop_guard: None,
776 mip_level_count: desc.mip_level_count,
777 array_layer_count: desc.array_layer_count(),
778 format: desc.format,
779 format_desc,
780 copy_size: desc.copy_extent(),
781 is_cubemap,
782 })
783 }
784 unsafe fn destroy_texture(&self, texture: super::Texture) {
785 if texture.drop_guard.is_none() {
786 let gl = &self.shared.context.lock();
787 match texture.inner {
788 super::TextureInner::Renderbuffer { raw, .. } => {
789 unsafe { gl.delete_renderbuffer(raw) };
790 }
791 super::TextureInner::DefaultRenderbuffer => {}
792 super::TextureInner::Texture { raw, .. } => {
793 unsafe { gl.delete_texture(raw) };
794 }
795 #[cfg(all(target_arch = "wasm32", not(target_os = "emscripten")))]
796 super::TextureInner::ExternalFramebuffer { .. } => {}
797 }
798 }
799
800 drop(texture.drop_guard);
803 }
804
805 unsafe fn create_texture_view(
806 &self,
807 texture: &super::Texture,
808 desc: &crate::TextureViewDescriptor,
809 ) -> Result<super::TextureView, crate::DeviceError> {
810 Ok(super::TextureView {
811 inner: texture.inner.clone(),
813 aspects: crate::FormatAspects::new(texture.format, desc.range.aspect),
814 mip_levels: desc.range.mip_range(texture.mip_level_count),
815 array_layers: desc.range.layer_range(texture.array_layer_count),
816 format: texture.format,
817 })
818 }
819 unsafe fn destroy_texture_view(&self, _view: super::TextureView) {}
820
821 unsafe fn create_sampler(
822 &self,
823 desc: &crate::SamplerDescriptor,
824 ) -> Result<super::Sampler, crate::DeviceError> {
825 let gl = &self.shared.context.lock();
826
827 let raw = unsafe { gl.create_sampler().unwrap() };
828
829 let (min, mag) =
830 conv::map_filter_modes(desc.min_filter, desc.mag_filter, desc.mipmap_filter);
831
832 unsafe { gl.sampler_parameter_i32(raw, glow::TEXTURE_MIN_FILTER, min as i32) };
833 unsafe { gl.sampler_parameter_i32(raw, glow::TEXTURE_MAG_FILTER, mag as i32) };
834
835 unsafe {
836 gl.sampler_parameter_i32(
837 raw,
838 glow::TEXTURE_WRAP_S,
839 conv::map_address_mode(desc.address_modes[0]) as i32,
840 )
841 };
842 unsafe {
843 gl.sampler_parameter_i32(
844 raw,
845 glow::TEXTURE_WRAP_T,
846 conv::map_address_mode(desc.address_modes[1]) as i32,
847 )
848 };
849 unsafe {
850 gl.sampler_parameter_i32(
851 raw,
852 glow::TEXTURE_WRAP_R,
853 conv::map_address_mode(desc.address_modes[2]) as i32,
854 )
855 };
856
857 if let Some(border_color) = desc.border_color {
858 let border = match border_color {
859 wgt::SamplerBorderColor::TransparentBlack | wgt::SamplerBorderColor::Zero => {
860 [0.0; 4]
861 }
862 wgt::SamplerBorderColor::OpaqueBlack => [0.0, 0.0, 0.0, 1.0],
863 wgt::SamplerBorderColor::OpaqueWhite => [1.0; 4],
864 };
865 unsafe { gl.sampler_parameter_f32_slice(raw, glow::TEXTURE_BORDER_COLOR, &border) };
866 }
867
868 unsafe { gl.sampler_parameter_f32(raw, glow::TEXTURE_MIN_LOD, desc.lod_clamp.start) };
869 unsafe { gl.sampler_parameter_f32(raw, glow::TEXTURE_MAX_LOD, desc.lod_clamp.end) };
870
871 if desc.anisotropy_clamp != 1 {
873 unsafe {
874 gl.sampler_parameter_i32(
875 raw,
876 glow::TEXTURE_MAX_ANISOTROPY,
877 desc.anisotropy_clamp as i32,
878 )
879 };
880 }
881
882 if let Some(compare) = desc.compare {
885 unsafe {
886 gl.sampler_parameter_i32(
887 raw,
888 glow::TEXTURE_COMPARE_MODE,
889 glow::COMPARE_REF_TO_TEXTURE as i32,
890 )
891 };
892 unsafe {
893 gl.sampler_parameter_i32(
894 raw,
895 glow::TEXTURE_COMPARE_FUNC,
896 conv::map_compare_func(compare) as i32,
897 )
898 };
899 }
900
901 #[cfg(not(target_arch = "wasm32"))]
902 if let Some(label) = desc.label {
903 if gl.supports_debug() {
904 let name = unsafe { mem::transmute(raw) };
905 unsafe { gl.object_label(glow::SAMPLER, name, Some(label)) };
906 }
907 }
908
909 Ok(super::Sampler { raw })
910 }
911 unsafe fn destroy_sampler(&self, sampler: super::Sampler) {
912 let gl = &self.shared.context.lock();
913 unsafe { gl.delete_sampler(sampler.raw) };
914 }
915
916 unsafe fn create_command_encoder(
917 &self,
918 _desc: &crate::CommandEncoderDescriptor<super::Api>,
919 ) -> Result<super::CommandEncoder, crate::DeviceError> {
920 Ok(super::CommandEncoder {
921 cmd_buffer: super::CommandBuffer::default(),
922 state: Default::default(),
923 private_caps: self.shared.private_caps,
924 })
925 }
926 unsafe fn destroy_command_encoder(&self, _encoder: super::CommandEncoder) {}
927
928 unsafe fn create_bind_group_layout(
929 &self,
930 desc: &crate::BindGroupLayoutDescriptor,
931 ) -> Result<super::BindGroupLayout, crate::DeviceError> {
932 Ok(super::BindGroupLayout {
933 entries: Arc::from(desc.entries),
934 })
935 }
936 unsafe fn destroy_bind_group_layout(&self, _bg_layout: super::BindGroupLayout) {}
937
938 unsafe fn create_pipeline_layout(
939 &self,
940 desc: &crate::PipelineLayoutDescriptor<super::Api>,
941 ) -> Result<super::PipelineLayout, crate::DeviceError> {
942 use naga::back::glsl;
943
944 let mut group_infos = Vec::with_capacity(desc.bind_group_layouts.len());
945 let mut num_samplers = 0u8;
946 let mut num_textures = 0u8;
947 let mut num_images = 0u8;
948 let mut num_uniform_buffers = 0u8;
949 let mut num_storage_buffers = 0u8;
950
951 let mut writer_flags = glsl::WriterFlags::ADJUST_COORDINATE_SPACE;
952 writer_flags.set(
953 glsl::WriterFlags::TEXTURE_SHADOW_LOD,
954 self.shared
955 .private_caps
956 .contains(super::PrivateCapabilities::SHADER_TEXTURE_SHADOW_LOD),
957 );
958 writer_flags.set(glsl::WriterFlags::FORCE_POINT_SIZE, true);
961 let mut binding_map = glsl::BindingMap::default();
962
963 for (group_index, bg_layout) in desc.bind_group_layouts.iter().enumerate() {
964 let mut binding_to_slot = vec![
966 !0;
967 bg_layout
968 .entries
969 .last()
970 .map_or(0, |b| b.binding as usize + 1)
971 ]
972 .into_boxed_slice();
973
974 for entry in bg_layout.entries.iter() {
975 let counter = match entry.ty {
976 wgt::BindingType::Sampler { .. } => &mut num_samplers,
977 wgt::BindingType::Texture { .. } => &mut num_textures,
978 wgt::BindingType::StorageTexture { .. } => &mut num_images,
979 wgt::BindingType::Buffer {
980 ty: wgt::BufferBindingType::Uniform,
981 ..
982 } => &mut num_uniform_buffers,
983 wgt::BindingType::Buffer {
984 ty: wgt::BufferBindingType::Storage { .. },
985 ..
986 } => &mut num_storage_buffers,
987 };
988
989 binding_to_slot[entry.binding as usize] = *counter;
990 let br = naga::ResourceBinding {
991 group: group_index as u32,
992 binding: entry.binding,
993 };
994 binding_map.insert(br, *counter);
995 *counter += entry.count.map_or(1, |c| c.get() as u8);
996 }
997
998 group_infos.push(super::BindGroupLayoutInfo {
999 entries: Arc::clone(&bg_layout.entries),
1000 binding_to_slot,
1001 });
1002 }
1003
1004 Ok(super::PipelineLayout {
1005 group_infos: group_infos.into_boxed_slice(),
1006 naga_options: glsl::Options {
1007 version: self.shared.shading_language_version,
1008 writer_flags,
1009 binding_map,
1010 zero_initialize_workgroup_memory: true,
1011 },
1012 })
1013 }
1014 unsafe fn destroy_pipeline_layout(&self, _pipeline_layout: super::PipelineLayout) {}
1015
1016 unsafe fn create_bind_group(
1017 &self,
1018 desc: &crate::BindGroupDescriptor<super::Api>,
1019 ) -> Result<super::BindGroup, crate::DeviceError> {
1020 let mut contents = Vec::new();
1021
1022 for (entry, layout) in desc.entries.iter().zip(desc.layout.entries.iter()) {
1023 let binding = match layout.ty {
1024 wgt::BindingType::Buffer { .. } => {
1025 let bb = &desc.buffers[entry.resource_index as usize];
1026 super::RawBinding::Buffer {
1027 raw: bb.buffer.raw.unwrap(),
1028 offset: bb.offset as i32,
1029 size: match bb.size {
1030 Some(s) => s.get() as i32,
1031 None => (bb.buffer.size - bb.offset) as i32,
1032 },
1033 }
1034 }
1035 wgt::BindingType::Sampler { .. } => {
1036 let sampler = desc.samplers[entry.resource_index as usize];
1037 super::RawBinding::Sampler(sampler.raw)
1038 }
1039 wgt::BindingType::Texture { .. } => {
1040 let view = desc.textures[entry.resource_index as usize].view;
1041 if view.mip_levels.start != 0 || view.array_layers.start != 0 {
1042 log::error!("Unable to create a sampled texture binding for non-zero mipmap level or array layer.\n{}",
1043 "This is an implementation problem of wgpu-hal/gles backend.")
1044 }
1045 let (raw, target) = view.inner.as_native();
1046 super::RawBinding::Texture {
1047 raw,
1048 target,
1049 aspects: view.aspects,
1050 }
1051 }
1052 wgt::BindingType::StorageTexture {
1053 access,
1054 format,
1055 view_dimension,
1056 } => {
1057 let view = desc.textures[entry.resource_index as usize].view;
1058 let format_desc = self.shared.describe_texture_format(format);
1059 let (raw, _target) = view.inner.as_native();
1060 super::RawBinding::Image(super::ImageBinding {
1061 raw,
1062 mip_level: view.mip_levels.start,
1063 array_layer: match view_dimension {
1064 wgt::TextureViewDimension::D2Array
1065 | wgt::TextureViewDimension::CubeArray => None,
1066 _ => Some(view.array_layers.start),
1067 },
1068 access: conv::map_storage_access(access),
1069 format: format_desc.internal,
1070 })
1071 }
1072 };
1073 contents.push(binding);
1074 }
1075
1076 Ok(super::BindGroup {
1077 contents: contents.into_boxed_slice(),
1078 })
1079 }
1080 unsafe fn destroy_bind_group(&self, _group: super::BindGroup) {}
1081
1082 unsafe fn create_shader_module(
1083 &self,
1084 desc: &crate::ShaderModuleDescriptor,
1085 shader: crate::ShaderInput,
1086 ) -> Result<super::ShaderModule, crate::ShaderError> {
1087 Ok(super::ShaderModule {
1088 naga: match shader {
1089 crate::ShaderInput::SpirV(_) => {
1090 panic!("`Features::SPIRV_SHADER_PASSTHROUGH` is not enabled")
1091 }
1092 crate::ShaderInput::Naga(naga) => naga,
1093 },
1094 label: desc.label.map(|str| str.to_string()),
1095 id: self.shared.next_shader_id.fetch_add(1, Ordering::Relaxed),
1096 })
1097 }
1098 unsafe fn destroy_shader_module(&self, _module: super::ShaderModule) {}
1099
1100 unsafe fn create_render_pipeline(
1101 &self,
1102 desc: &crate::RenderPipelineDescriptor<super::Api>,
1103 ) -> Result<super::RenderPipeline, crate::PipelineError> {
1104 let gl = &self.shared.context.lock();
1105 let mut shaders = ArrayVec::new();
1106 shaders.push((naga::ShaderStage::Vertex, &desc.vertex_stage));
1107 if let Some(ref fs) = desc.fragment_stage {
1108 shaders.push((naga::ShaderStage::Fragment, fs));
1109 }
1110 let inner =
1111 unsafe { self.create_pipeline(gl, shaders, desc.layout, desc.label, desc.multiview) }?;
1112
1113 let (vertex_buffers, vertex_attributes) = {
1114 let mut buffers = Vec::new();
1115 let mut attributes = Vec::new();
1116 for (index, vb_layout) in desc.vertex_buffers.iter().enumerate() {
1117 buffers.push(super::VertexBufferDesc {
1118 step: vb_layout.step_mode,
1119 stride: vb_layout.array_stride as u32,
1120 });
1121 for vat in vb_layout.attributes.iter() {
1122 let format_desc = conv::describe_vertex_format(vat.format);
1123 attributes.push(super::AttributeDesc {
1124 location: vat.shader_location,
1125 offset: vat.offset as u32,
1126 buffer_index: index as u32,
1127 format_desc,
1128 });
1129 }
1130 }
1131 (buffers.into_boxed_slice(), attributes.into_boxed_slice())
1132 };
1133
1134 let color_targets = {
1135 let mut targets = Vec::new();
1136 for ct in desc.color_targets.iter().filter_map(|at| at.as_ref()) {
1137 targets.push(super::ColorTargetDesc {
1138 mask: ct.write_mask,
1139 blend: ct.blend.as_ref().map(conv::map_blend),
1140 });
1141 }
1142 targets.into_boxed_slice()
1145 };
1146
1147 Ok(super::RenderPipeline {
1148 inner,
1149 primitive: desc.primitive,
1150 vertex_buffers,
1151 vertex_attributes,
1152 color_targets,
1153 depth: desc.depth_stencil.as_ref().map(|ds| super::DepthState {
1154 function: conv::map_compare_func(ds.depth_compare),
1155 mask: ds.depth_write_enabled,
1156 }),
1157 depth_bias: desc
1158 .depth_stencil
1159 .as_ref()
1160 .map(|ds| ds.bias)
1161 .unwrap_or_default(),
1162 stencil: desc
1163 .depth_stencil
1164 .as_ref()
1165 .map(|ds| conv::map_stencil(&ds.stencil)),
1166 alpha_to_coverage_enabled: desc.multisample.alpha_to_coverage_enabled,
1167 })
1168 }
1169 unsafe fn destroy_render_pipeline(&self, pipeline: super::RenderPipeline) {
1170 let mut program_cache = self.shared.program_cache.lock();
1171 if Arc::strong_count(&pipeline.inner) == 2 {
1176 program_cache.retain(|_, v| match *v {
1177 Ok(ref p) => p.program != pipeline.inner.program,
1178 Err(_) => false,
1179 });
1180 let gl = &self.shared.context.lock();
1181 unsafe { gl.delete_program(pipeline.inner.program) };
1182 }
1183 }
1184
1185 unsafe fn create_compute_pipeline(
1186 &self,
1187 desc: &crate::ComputePipelineDescriptor<super::Api>,
1188 ) -> Result<super::ComputePipeline, crate::PipelineError> {
1189 let gl = &self.shared.context.lock();
1190 let mut shaders = ArrayVec::new();
1191 shaders.push((naga::ShaderStage::Compute, &desc.stage));
1192 let inner = unsafe { self.create_pipeline(gl, shaders, desc.layout, desc.label, None) }?;
1193
1194 Ok(super::ComputePipeline { inner })
1195 }
1196 unsafe fn destroy_compute_pipeline(&self, pipeline: super::ComputePipeline) {
1197 let mut program_cache = self.shared.program_cache.lock();
1198 if Arc::strong_count(&pipeline.inner) == 2 {
1203 program_cache.retain(|_, v| match *v {
1204 Ok(ref p) => p.program != pipeline.inner.program,
1205 Err(_) => false,
1206 });
1207 let gl = &self.shared.context.lock();
1208 unsafe { gl.delete_program(pipeline.inner.program) };
1209 }
1210 }
1211
1212 #[cfg_attr(target_arch = "wasm32", allow(unused))]
1213 unsafe fn create_query_set(
1214 &self,
1215 desc: &wgt::QuerySetDescriptor<crate::Label>,
1216 ) -> Result<super::QuerySet, crate::DeviceError> {
1217 let gl = &self.shared.context.lock();
1218 let mut temp_string = String::new();
1219
1220 let mut queries = Vec::with_capacity(desc.count as usize);
1221 for i in 0..desc.count {
1222 let query =
1223 unsafe { gl.create_query() }.map_err(|_| crate::DeviceError::OutOfMemory)?;
1224 #[cfg(not(target_arch = "wasm32"))]
1225 if gl.supports_debug() {
1226 use std::fmt::Write;
1227
1228 if let Some(label) = desc.label {
1229 temp_string.clear();
1230 let _ = write!(temp_string, "{label}[{i}]");
1231 let name = unsafe { mem::transmute(query) };
1232 unsafe { gl.object_label(glow::QUERY, name, Some(&temp_string)) };
1233 }
1234 }
1235 queries.push(query);
1236 }
1237
1238 Ok(super::QuerySet {
1239 queries: queries.into_boxed_slice(),
1240 target: match desc.ty {
1241 wgt::QueryType::Occlusion => glow::ANY_SAMPLES_PASSED,
1242 _ => unimplemented!(),
1243 },
1244 })
1245 }
1246 unsafe fn destroy_query_set(&self, set: super::QuerySet) {
1247 let gl = &self.shared.context.lock();
1248 for &query in set.queries.iter() {
1249 unsafe { gl.delete_query(query) };
1250 }
1251 }
1252 unsafe fn create_fence(&self) -> Result<super::Fence, crate::DeviceError> {
1253 Ok(super::Fence {
1254 last_completed: 0,
1255 pending: Vec::new(),
1256 })
1257 }
1258 unsafe fn destroy_fence(&self, fence: super::Fence) {
1259 let gl = &self.shared.context.lock();
1260 for (_, sync) in fence.pending {
1261 unsafe { gl.delete_sync(sync) };
1262 }
1263 }
1264 unsafe fn get_fence_value(
1265 &self,
1266 fence: &super::Fence,
1267 ) -> Result<crate::FenceValue, crate::DeviceError> {
1268 #[cfg_attr(target_arch = "wasm32", allow(clippy::needless_borrow))]
1269 Ok(fence.get_latest(&self.shared.context.lock()))
1270 }
1271 unsafe fn wait(
1272 &self,
1273 fence: &super::Fence,
1274 wait_value: crate::FenceValue,
1275 timeout_ms: u32,
1276 ) -> Result<bool, crate::DeviceError> {
1277 if fence.last_completed < wait_value {
1278 let gl = &self.shared.context.lock();
1279 let timeout_ns = if cfg!(target_arch = "wasm32") {
1280 0
1281 } else {
1282 (timeout_ms as u64 * 1_000_000).min(!0u32 as u64)
1283 };
1284 let &(_, sync) = fence
1285 .pending
1286 .iter()
1287 .find(|&&(value, _)| value >= wait_value)
1288 .unwrap();
1289 match unsafe {
1290 gl.client_wait_sync(sync, glow::SYNC_FLUSH_COMMANDS_BIT, timeout_ns as i32)
1291 } {
1292 #[cfg(target_arch = "wasm32")]
1294 glow::WAIT_FAILED => {
1295 log::warn!("wait failed!");
1296 Ok(false)
1297 }
1298 glow::TIMEOUT_EXPIRED => Ok(false),
1299 glow::CONDITION_SATISFIED | glow::ALREADY_SIGNALED => Ok(true),
1300 _ => Err(crate::DeviceError::Lost),
1301 }
1302 } else {
1303 Ok(true)
1304 }
1305 }
1306
1307 unsafe fn start_capture(&self) -> bool {
1308 #[cfg(all(not(target_arch = "wasm32"), feature = "renderdoc"))]
1309 return unsafe {
1310 self.render_doc
1311 .start_frame_capture(self.shared.context.raw_context(), ptr::null_mut())
1312 };
1313 #[allow(unreachable_code)]
1314 false
1315 }
1316 unsafe fn stop_capture(&self) {
1317 #[cfg(all(not(target_arch = "wasm32"), feature = "renderdoc"))]
1318 unsafe {
1319 self.render_doc
1320 .end_frame_capture(ptr::null_mut(), ptr::null_mut())
1321 }
1322 }
1323}
1324
1325#[cfg(all(
1326 target_arch = "wasm32",
1327 feature = "fragile-send-sync-non-atomic-wasm",
1328 not(target_feature = "atomics")
1329))]
1330unsafe impl Sync for super::Device {}
1331#[cfg(all(
1332 target_arch = "wasm32",
1333 feature = "fragile-send-sync-non-atomic-wasm",
1334 not(target_feature = "atomics")
1335))]
1336unsafe impl Send for super::Device {}