strafe-project/src/graphics_context.rs

use crate::instruction::TimedInstruction;

fn optional_features() -> wgpu::Features {
	wgpu::Features::empty()
}
fn required_features() -> wgpu::Features {
	wgpu::Features::empty()
}
fn required_downlevel_capabilities() -> wgpu::DownlevelCapabilities {
	wgpu::DownlevelCapabilities {
		flags: wgpu::DownlevelFlags::empty(),
		shader_model: wgpu::ShaderModel::Sm5,
		..wgpu::DownlevelCapabilities::default()
	}
}
pub fn required_limits() -> wgpu::Limits {
	wgpu::Limits::downlevel_webgl2_defaults() // These downlevel limits will allow the code to run on all possible hardware
}

struct GraphicsContextPartial1{
	backends:wgpu::Backends,
	instance:wgpu::Instance,
}
fn create_instance()->GraphicsContextPartial1{
	let backends=wgpu::util::backend_bits_from_env().unwrap_or_else(wgpu::Backends::all);
	let dx12_shader_compiler=wgpu::util::dx12_shader_compiler_from_env().unwrap_or_default();
	GraphicsContextPartial1{
		backends,
		instance:wgpu::Instance::new(wgpu::InstanceDescriptor{
			backends,
			dx12_shader_compiler,
		}),
	}
}
impl GraphicsContextPartial1{
	fn create_surface(self,window:&winit::window::Window)->Result<GraphicsContextPartial2,wgpu::CreateSurfaceError>{
		Ok(GraphicsContextPartial2{
			backends:self.backends,
			instance:self.instance,
			surface:unsafe{self.instance.create_surface(window)}?
		})
	}
}
struct GraphicsContextPartial2{
	backends:wgpu::Backends,
	instance:wgpu::Instance,
	surface:wgpu::Surface,
}
impl GraphicsContextPartial2{
	fn pick_adapter(self)->GraphicsContextPartial3{
		let adapter;

		let optional_features=optional_features();
		let required_features=required_features();

		//no helper function smh gotta write it myself
		let adapters=self.instance.enumerate_adapters(self.backends);

		let mut chosen_adapter=None;
		let mut chosen_adapter_score=0;
		for adapter in adapters {
			if !adapter.is_surface_supported(&self.surface) {
				continue;
			}

			let score=match adapter.get_info().device_type{
				wgpu::DeviceType::IntegratedGpu=>3,
				wgpu::DeviceType::DiscreteGpu=>4,
				wgpu::DeviceType::VirtualGpu=>2,
				wgpu::DeviceType::Other|wgpu::DeviceType::Cpu=>1,
			};

			let adapter_features=adapter.features();
			if chosen_adapter_score<score&&adapter_features.contains(required_features) {
				chosen_adapter_score=score;
				chosen_adapter=Some(adapter);
			}
		}

		if let Some(maybe_chosen_adapter)=chosen_adapter{
			adapter=maybe_chosen_adapter;
		}else{
			panic!("No suitable GPU adapters found on the system!");
		}


		let adapter_info=adapter.get_info();
		println!("Using {} ({:?})", adapter_info.name, adapter_info.backend);

		let required_downlevel_capabilities=required_downlevel_capabilities();
		let downlevel_capabilities=adapter.get_downlevel_capabilities();
		assert!(
			downlevel_capabilities.shader_model >= required_downlevel_capabilities.shader_model,
			"Adapter does not support the minimum shader model required to run this example: {:?}",
			required_downlevel_capabilities.shader_model
		);
		assert!(
			downlevel_capabilities
				.flags
				.contains(required_downlevel_capabilities.flags),
			"Adapter does not support the downlevel capabilities required to run this example: {:?}",
			required_downlevel_capabilities.flags - downlevel_capabilities.flags
		);
		GraphicsContextPartial3{
			instance:self.instance,
			surface:self.surface,
			adapter,
		}
	}
}
struct GraphicsContextPartial3{
	instance:wgpu::Instance,
	surface:wgpu::Surface,
	adapter:wgpu::Adapter,
}
impl GraphicsContextPartial3{
	fn request_device(self)->GraphicsContextPartial4{
		let optional_features=optional_features();
		let required_features=required_features();

		// Make sure we use the texture resolution limits from the adapter, so we can support images the size of the surface.
		let needed_limits=required_limits().using_resolution(self.adapter.limits());

		let trace_dir=std::env::var("WGPU_TRACE");
		let (device, queue)=pollster::block_on(self.adapter
			.request_device(
				&wgpu::DeviceDescriptor {
					label: None,
					features: (optional_features & self.adapter.features()) | required_features,
					limits: needed_limits,
				},
				trace_dir.ok().as_ref().map(std::path::Path::new),
			))
			.expect("Unable to find a suitable GPU adapter!");

		GraphicsContextPartial4{
			instance:self.instance,
			surface:self.surface,
			adapter:self.adapter,
			device,
			queue,
		}
	}
}
struct GraphicsContextPartial4{
	instance:wgpu::Instance,
	surface:wgpu::Surface,
	adapter:wgpu::Adapter,
	device:wgpu::Device,
	queue:wgpu::Queue,
}
impl GraphicsContextPartial4{
	fn configure_surface(self,size:&winit::dpi::PhysicalSize<u32>)->GraphicsContext{
		let mut config=self.surface
			.get_default_config(&self.adapter, size.width, size.height)
			.expect("Surface isn't supported by the adapter.");
		let surface_view_format=config.format.add_srgb_suffix();
		config.view_formats.push(surface_view_format);
		self.surface.configure(&self.device, &config);

		GraphicsContext{
			instance:self.instance,
			surface:self.surface,
			adapter:self.adapter,
			device:self.device,
			queue:self.queue,
			config,
		}
	}
}
pub struct GraphicsContext{
	pub instance:wgpu::Instance,
	pub surface:wgpu::Surface,
	pub adapter:wgpu::Adapter,
	pub device:wgpu::Device,
	pub queue:wgpu::Queue,
	pub config:wgpu::SurfaceConfiguration,
}

pub fn setup(title:&str)->GraphicsContextSetup{
	let event_loop=winit::event_loop::EventLoop::new();

	let window=crate::window::WindowState::create_window(title,&event_loop).unwrap();

	println!("Initializing the surface...");

	let partial_1=create_instance();

	let partial_2=partial_1.create_surface(&window).unwrap();

	let partial_3=partial_2.pick_adapter();

	let partial_4=partial_3.request_device();

	GraphicsContextSetup{
		window,
		event_loop,
		partial_graphics_context:partial_4,
	}
}

enum RunInstruction{
	Resize(winit::dpi::PhysicalSize<u32>),
	WindowEvent(winit::event::WindowEvent),
	DeviceEvent(winit::event::DeviceEvent),
	Render,
}

impl GraphicsContext{
	fn into_worker(self,mut global_state:crate::GlobalState)->crate::worker::QNWorker<TimedInstruction<RunInstruction>>{
		crate::worker::QNWorker::new(move |ins:TimedInstruction<RunInstruction>|{
			match ins.instruction{
				RunInstruction::WindowEvent(window_event)=>{
					global_state.window_event(window_event);
				},
				RunInstruction::DeviceEvent(device_event)=>{
					global_state.device_event(device_event);
				},
				RunInstruction::Resize(size)=>{
					self.config.width=size.width.max(1);
					self.config.height=size.height.max(1);
					global_state.graphics.resize(&self.device,&self.config);
					self.surface.configure(&self.device,&self.config);
				}
				RunInstruction::Render=>{
					let frame=match self.surface.get_current_texture(){
						Ok(frame)=>frame,
						Err(_)=>{
							self.surface.configure(&self.device,&self.config);
							self.surface
								.get_current_texture()
								.expect("Failed to acquire next surface texture!")
						}
					};
					let view=frame.texture.create_view(&wgpu::TextureViewDescriptor{
						format:Some(self.config.view_formats[0]),
						..wgpu::TextureViewDescriptor::default()
					});

					global_state.graphics.render(&view,&self.device,&self.queue);

					frame.present();
				}
			}
		})
	}
}

struct GraphicsContextSetup{
	window:winit::window::Window,
	event_loop:winit::event_loop::EventLoop<()>,
	partial_graphics_context:GraphicsContextPartial4,
}

impl GraphicsContextSetup{
	fn into_split(self)->(winit::window::Window,winit::event_loop::EventLoop<()>,GraphicsContext){
		let size=self.window.inner_size();
		//Steal values and drop self
		(
			self.window,
			self.event_loop,
			self.partial_graphics_context.configure_surface(&size),
		)
	}
	pub fn start(self,mut global_state:crate::GlobalState){
		let (window,event_loop,graphics_context)=self.into_split();

		//dedicated thread to pigh request redraw back and resize the window doesn't seem logical

		//physics and graphics render thread
		let run_thread=graphics_context.into_worker(global_state);

		println!("Entering render loop...");
		let root_time=std::time::Instant::now();
		event_loop.run(move |event,_,control_flow|{
			let time=crate::integer::Time::from_nanos(root_time.elapsed().as_nanos() as i64);
			*control_flow=if cfg!(feature="metal-auto-capture"){
				winit::event_loop::ControlFlow::Exit
			}else{
				winit::event_loop::ControlFlow::Poll
			};
			match event{
				winit::event::Event::RedrawEventsCleared=>{
					window.request_redraw();
				}
				winit::event::Event::WindowEvent {
					event:
						winit::event::WindowEvent::Resized(size)
						| winit::event::WindowEvent::ScaleFactorChanged {
							new_inner_size:&mut size,
							scale_factor:_,
						},
					window_id:_,
				} => {
					// Once winit is fixed, the detection conditions here can be removed.
					// https://github.com/rust-windowing/winit/issues/2876
					// this has been fixed if I update winit (remove the if statement and only use the else case)
					//drop adapter when you delete this
					let max_dimension=graphics_context.adapter.limits().max_texture_dimension_2d;
					if max_dimension<size.width||max_dimension<size.height{
						println!(
							"The resizing size {:?} exceeds the limit of {}.",
							size,
							max_dimension
						);
					}else{
						println!("Resizing to {:?}",size);
						run_thread.send(TimedInstruction{time,instruction:RunInstruction::Resize(size)});
					}
				}
				winit::event::Event::WindowEvent{event,..}=>match event{
					winit::event::WindowEvent::KeyboardInput{
						input:
							winit::event::KeyboardInput{
								virtual_keycode:Some(winit::event::VirtualKeyCode::Escape),
								state: winit::event::ElementState::Pressed,
								..
							},
						..
					}
					|winit::event::WindowEvent::CloseRequested=>{
						*control_flow=winit::event_loop::ControlFlow::Exit;
					}
					_=>{
						run_thread.send(TimedInstruction{time,instruction:RunInstruction::WindowEvent(event)});
					}
				},
				winit::event::Event::DeviceEvent{
					event,
					..
				} => {
					run_thread.send(TimedInstruction{time,instruction:RunInstruction::DeviceEvent(event)});
				},
				winit::event::Event::RedrawRequested(_)=>{
					//send
				}
				_=>{}
			}
		});
	}
}