summaryrefslogtreecommitdiffstats
path: root/src/tacmap
diff options
context:
space:
mode:
authorJon Santmyer <jon@jonsantmyer.com>2026-05-05 09:12:50 -0400
committerJon Santmyer <jon@jonsantmyer.com>2026-05-05 09:12:50 -0400
commitbe2cf936ca48f3d638c3ef01f4e338dfc904c5e3 (patch)
tree9fbdbfbf3e0d427f3a6c4bd4c74901e421ef0ad6 /src/tacmap
parentb35d6cf47ed154697fb45e10586206295148eb35 (diff)
downloadsystemic4x-be2cf936ca48f3d638c3ef01f4e338dfc904c5e3.tar.gz
systemic4x-be2cf936ca48f3d638c3ef01f4e338dfc904c5e3.tar.bz2
systemic4x-be2cf936ca48f3d638c3ef01f4e338dfc904c5e3.zip
seperate orbit logic from body
Diffstat (limited to 'src/tacmap')
-rw-r--r--src/tacmap/body_render.rs193
-rw-r--r--src/tacmap/orbit_render.rs240
-rw-r--r--src/tacmap/render.rs420
3 files changed, 433 insertions, 420 deletions
diff --git a/src/tacmap/body_render.rs b/src/tacmap/body_render.rs
new file mode 100644
index 0000000..8b32532
--- /dev/null
+++ b/src/tacmap/body_render.rs
@@ -0,0 +1,193 @@
+use std::error::Error;
+
+use crate::solar_system::Kilometers;
+use crate::timeman::Second;
+use crate::wgpuctx::{WgpuCtx, pipeline::RenderPipelineBuilder};
+
+use super::*;
+
+pub struct BodyRenderer
+{
+ needs_rebuild: bool,
+ last_time: Option<Second>,
+
+ pipeline: wgpu::RenderPipeline,
+
+ body_instance_buffer: Option<(usize, wgpu::Buffer)>
+}
+
+impl BodyRenderer
+{
+ pub fn new(
+ wgpuctx: &WgpuCtx)
+ -> Self {
+ let shader = wgpuctx.create_shader(
+ wgpu::include_wgsl!(concat!(
+ env!("CARGO_MANIFEST_DIR"),
+ "/assets/shaders/tacmap/body.wgsl")
+ ));
+
+ let render_pipeline = RenderPipelineBuilder::new(&shader)
+ .add_bindgroup(&Camera::bindgroup_layout(wgpuctx))
+ .add_vertex_layout(BodyInstanceRaw::descr())
+ .cull_mode(None)
+ .build(Some("Tactical map render pipeline"), wgpuctx);
+
+ Self {
+ needs_rebuild: true,
+ last_time: None,
+ pipeline: render_pipeline,
+ body_instance_buffer: None
+ }
+ }
+
+ pub fn mark_to_rebuild(&mut self)
+ { self.needs_rebuild = true; }
+
+ pub fn rebuild(
+ &mut self,
+ wgpuctx: &WgpuCtx,
+ solar_system: &SolarSystem)
+ {
+ if self.body_instance_buffer.is_some() && !self.needs_rebuild {
+ return;
+ }
+ self.needs_rebuild = false;
+
+ match self.body_instance_buffer.as_mut() {
+ Some(buffer) => {
+ buffer.1.destroy();
+ self.body_instance_buffer = None;
+ }
+ None => {}
+ }
+
+ let bodies = solar_system.bodies();
+ let buffer_len = bodies.len() * size_of::<BodyInstanceRaw>();
+
+ let buffer = wgpuctx.create_buffer(
+ &wgpu::BufferDescriptor {
+ label: Some("Tactical map bodies instance buffer"),
+ size: buffer_len as u64,
+ usage: wgpu::BufferUsages::COPY_DST |
+ wgpu::BufferUsages::VERTEX,
+ mapped_at_creation: false
+ }
+ );
+
+ self.last_time = None;
+ self.body_instance_buffer = Some((bodies.len(), buffer));
+ }
+
+ pub fn update(
+ &mut self,
+ wgpuctx: &WgpuCtx,
+ solar_system: &SolarSystem,
+ time: Second)
+ -> Result<(), Box<dyn Error>>
+ {
+ //If the last updated time is the same, we don't need to update
+ //the positions of all the bodies.
+ if self.last_time.is_some_and(|last_time| { last_time == time }) {
+ return Ok(());
+ }
+
+ self.last_time = Some(time);
+
+ let (_, bodies_buffer) = match &self.body_instance_buffer {
+ Some(tuple) => tuple,
+ None => return Err(Box::new(NeedsRebuildError))
+ };
+
+ let bodies = solar_system.bodies();
+ let body_instances = bodies.iter().map(|body| {
+ let position = body.position();
+ let origin = match body.get_orbits() {
+ Some(origin_id) => {
+ let origin_body = &bodies[origin_id];
+ origin_body.position()
+ },
+ None => cgmath::Vector3::new(0.0, 0.0, 0.0)
+ };
+ BodyInstance {
+ position: position,
+ origin: origin,
+ radius: body.radius()
+ }.raw()
+ }).collect::<Vec<_>>();
+
+ wgpuctx.queue().write_buffer(bodies_buffer, 0, bytemuck::cast_slice(&body_instances));
+
+ Ok(())
+ }
+
+ pub fn render(
+ &self,
+ pass: &mut wgpu::RenderPass,
+ camera: &Camera)
+ {
+ let (num_bodies, bodies_buffer) = match &self.body_instance_buffer {
+ Some(tuple) => tuple,
+ None => return
+ };
+
+ pass.set_pipeline(&self.pipeline);
+ pass.set_bind_group(0, camera.bindgroup(), &[]);
+
+ pass.set_vertex_buffer(0, bodies_buffer.slice(..));
+
+ pass.draw(0..6, 0..num_bodies.clone() as _);
+ }
+} // impl RenderState
+
+struct BodyInstance
+{
+ position: cgmath::Vector3<Kilometers>,
+ origin: cgmath::Vector3<Kilometers>,
+ radius: f32
+}
+
+#[repr(C)]
+#[derive(Copy, Clone, bytemuck::Pod, bytemuck::Zeroable)]
+struct BodyInstanceRaw
+{
+ position: [f32;3],
+ origin: [f32;3],
+ radius: f32
+}
+
+impl BodyInstance
+{
+ fn raw(&self) -> BodyInstanceRaw
+ {
+ BodyInstanceRaw {
+ position: [
+ self.position.x as f32,
+ self.position.y as f32,
+ self.position.z as f32 ],
+ origin: [
+ self.origin.x as f32,
+ self.origin.y as f32,
+ self.origin.z as f32 ],
+ radius: self.radius
+ }
+ }
+} // impl BodyInstance
+
+impl BodyInstanceRaw
+{
+ const ATTRIBS: [wgpu::VertexAttribute;3] =
+ wgpu::vertex_attr_array![0 => Float32x3, 1 => Float32x3, 2 => Float32];
+
+ pub fn descr()
+ -> wgpu::VertexBufferLayout<'static> {
+ use std::mem;
+
+ wgpu::VertexBufferLayout {
+ array_stride: mem::size_of::<Self>() as wgpu::BufferAddress,
+ step_mode: wgpu::VertexStepMode::Instance,
+ attributes: &Self::ATTRIBS
+ }
+ }
+}
+
diff --git a/src/tacmap/orbit_render.rs b/src/tacmap/orbit_render.rs
new file mode 100644
index 0000000..d79294e
--- /dev/null
+++ b/src/tacmap/orbit_render.rs
@@ -0,0 +1,240 @@
+use super::*;
+
+use crate::solar_system::BodyId;
+use crate::timeman::{SYSTEM_TICK_INTERVAL, Second};
+use crate::wgpuctx::{WgpuCtx, pipeline::RenderPipelineBuilder};
+
+pub struct OrbitRenderer
+{
+ needs_rebuild: bool,
+ last_time: Option<Second>,
+
+ pipeline: wgpu::RenderPipeline,
+
+ orbit_vertex_buffers: Option<Vec<(BodyId, Option<wgpu::Buffer>)>>,
+ origin_buffer: Option<wgpu::Buffer>,
+ origin_bind_group: Option<wgpu::BindGroup>,
+ origin_bind_group_layout: wgpu::BindGroupLayout
+}
+
+impl OrbitRenderer
+{
+ pub fn new(
+ wgpuctx: &WgpuCtx)
+ -> Self {
+ let shader = wgpuctx.create_shader(
+ wgpu::include_wgsl!(concat!(
+ env!("CARGO_MANIFEST_DIR"),
+ "/assets/shaders/tacmap/orbit.wgsl")
+ ));
+
+ let origin_bind_group_layout = wgpuctx.device().create_bind_group_layout(
+ &wgpu::BindGroupLayoutDescriptor {
+ label: None,
+ entries: &[
+ wgpu::BindGroupLayoutEntry {
+ binding: 0,
+ visibility: wgpu::ShaderStages::VERTEX,
+ ty: wgpu::BindingType::Buffer {
+ ty: wgpu::BufferBindingType::Storage { read_only: true },
+ has_dynamic_offset: false,
+ min_binding_size: None
+ },
+ count: None
+ }
+ ]
+ }
+ );
+
+ let render_pipeline = RenderPipelineBuilder::new(&shader)
+ .add_bindgroup(&Camera::bindgroup_layout(wgpuctx))
+ .add_bindgroup(&origin_bind_group_layout)
+ .add_vertex_layout(OrbitVertex::descr())
+ .primitive_topology(wgpu::PrimitiveTopology::TriangleStrip)
+ .cull_mode(None)
+ .build(Some("Tactical map orbit render pipeline"), wgpuctx);
+
+ Self {
+ needs_rebuild: true,
+ last_time: None,
+ pipeline: render_pipeline,
+ orbit_vertex_buffers: None,
+ origin_buffer: None,
+ origin_bind_group: None,
+ origin_bind_group_layout: origin_bind_group_layout
+ }
+ }
+
+ pub fn mark_to_rebuild(&mut self)
+ { self.needs_rebuild = true; }
+
+ pub fn rebuild(
+ &mut self,
+ wgpuctx: &WgpuCtx,
+ solar_system: &SolarSystem)
+ {
+ if self.orbit_vertex_buffers.is_some() && !self.needs_rebuild {
+ return;
+ }
+ self.needs_rebuild = false;
+
+ match &self.orbit_vertex_buffers {
+ Some(buffers) => {
+ for buffer in buffers {
+ match &buffer.1 {
+ Some(v) => v.destroy(),
+ None => {}
+ }
+ }
+ self.orbit_vertex_buffers = None;
+ },
+ None => {}
+ };
+
+ //From the solar system's bodies, calculate the orbits.
+ let bodies = solar_system.bodies();
+
+ match &self.origin_buffer {
+ Some(buffer) => {
+ buffer.destroy();
+ }
+ None => {}
+ };
+ let origin_buffer = wgpuctx.create_buffer(
+ &wgpu::BufferDescriptor {
+ label: None,
+ size: (std::mem::size_of::<[f32;3]>() * bodies.len()) as _,
+ usage: wgpu::BufferUsages::COPY_DST
+ | wgpu::BufferUsages::STORAGE,
+ mapped_at_creation: false
+ }
+ );
+
+ self.origin_bind_group = Some(wgpuctx.device().create_bind_group(
+ &wgpu::BindGroupDescriptor {
+ label: None,
+ layout: &self.origin_bind_group_layout,
+ entries: &[
+ wgpu::BindGroupEntry {
+ binding: 0,
+ resource: origin_buffer.as_entire_binding()
+ }
+ ]
+ }
+ ));
+ self.origin_buffer = Some(origin_buffer);
+
+ self.orbit_vertex_buffers = Some(bodies.iter().map(|body| {
+ if !body.does_orbit() { return (0, None); }
+
+ let period = body.orbital_period();
+ let num_points = ((period / (SYSTEM_TICK_INTERVAL)) as usize).clamp(180, 360*4);
+ let period_interval = period as f64 / num_points as f64;
+
+ let mut points = vec![
+ OrbitVertex::default();(num_points+1)*2];
+ for i in 0..num_points {
+ let position = body.calculate_orbit_at(
+ (i as f64 * period_interval) as u64);
+
+ points[i*2].origin_id = body.get_orbits().unwrap() as _;
+ points[i*2].position = [
+ position.x as f32,
+ position.y as f32,
+ position.z as f32
+ ];
+
+ points[(i*2)+1] = points[i*2].clone();
+ }
+ points[num_points*2] = points[0].clone();
+ points[num_points*2+1] = points[1].clone();
+
+ let buffer = wgpuctx.create_buffer_init(
+ &wgpu::util::BufferInitDescriptor {
+ label: None,
+ usage: wgpu::BufferUsages::VERTEX,
+ contents: bytemuck::cast_slice(&points)
+ }
+ );
+ ((num_points+1)*2, Some(buffer))
+ }).collect::<Vec<_>>());
+ }
+
+ pub fn update(
+ &mut self,
+ wgpuctx: &WgpuCtx,
+ solar_system: &SolarSystem,
+ time: Second)
+ {
+ if self.last_time.is_some_and(|t| { t == time }) {
+ return;
+ }
+
+ self.last_time = Some(time);
+ let positions = solar_system.bodies().iter().map(|body| {
+ let pos = solar_system.body_position(body);
+ [ pos.x as f32, pos.y as f32, pos.z as f32 ]
+ }).collect::<Vec<_>>();
+
+ match &self.origin_buffer {
+ Some(buffer) => {
+ wgpuctx.queue().write_buffer(buffer, 0, bytemuck::cast_slice(&positions));
+ },
+ None => { return; }
+ };
+ }
+
+ pub fn render(
+ &self,
+ pass: &mut wgpu::RenderPass,
+ camera: &Camera)
+ {
+ let buffers = match &self.orbit_vertex_buffers {
+ Some(v) => v,
+ None => return
+ };
+
+ pass.set_pipeline(&self.pipeline);
+ pass.set_bind_group(0, camera.bindgroup(), &[]);
+
+ match &self.origin_bind_group {
+ Some(bind_group) => { pass.set_bind_group(1, bind_group, &[]); },
+ None => { return; }
+ }
+
+ for (num_points, buffer) in buffers {
+ match &buffer {
+ Some(buffer) => {
+ pass.set_vertex_buffer(0, buffer.slice(..));
+ pass.draw(0..(*num_points as u32), 0..1);
+ },
+ None => {}
+ }
+ }
+ }
+}
+
+#[repr(C)]
+#[derive(Copy, Clone, Default, Debug, bytemuck::Pod, bytemuck::Zeroable)]
+pub struct OrbitVertex
+{
+ pub origin_id: u32,
+ pub position: [f32;3]
+}
+
+impl OrbitVertex
+{
+ const ATTRIBS: [wgpu::VertexAttribute;2] =
+ wgpu::vertex_attr_array![0 => Uint32, 1 => Float32x3];
+
+ pub fn descr()
+ -> wgpu::VertexBufferLayout<'static> {
+ use std::mem;
+
+ wgpu::VertexBufferLayout {
+ array_stride: mem::size_of::<Self>() as wgpu::BufferAddress,
+ step_mode: wgpu::VertexStepMode::Vertex,
+ attributes: &Self::ATTRIBS
+ }
+ }
+}
diff --git a/src/tacmap/render.rs b/src/tacmap/render.rs
index e6944d0..fd93941 100644
--- a/src/tacmap/render.rs
+++ b/src/tacmap/render.rs
@@ -19,191 +19,6 @@ impl Display for NeedsRebuildError {
impl Error for NeedsRebuildError {}
-pub struct BodyRenderer
-{
- needs_rebuild: bool,
- last_time: Option<Second>,
-
- pipeline: wgpu::RenderPipeline,
-
- body_instance_buffer: Option<(usize, wgpu::Buffer)>
-}
-
-impl BodyRenderer
-{
- pub fn new(
- wgpuctx: &WgpuCtx)
- -> Self {
- let shader = wgpuctx.create_shader(
- wgpu::include_wgsl!(concat!(
- env!("CARGO_MANIFEST_DIR"),
- "/assets/shaders/tacmap/body.wgsl")
- ));
-
- let render_pipeline = RenderPipelineBuilder::new(&shader)
- .add_bindgroup(&Camera::bindgroup_layout(wgpuctx))
- .add_vertex_layout(BodyInstanceRaw::descr())
- .cull_mode(None)
- .build(Some("Tactical map render pipeline"), wgpuctx);
-
- Self {
- needs_rebuild: true,
- last_time: None,
- pipeline: render_pipeline,
- body_instance_buffer: None
- }
- }
-
- pub fn mark_to_rebuild(&mut self)
- { self.needs_rebuild = true; }
-
- pub fn rebuild(
- &mut self,
- wgpuctx: &WgpuCtx,
- solar_system: &SolarSystem)
- {
- if self.body_instance_buffer.is_some() && !self.needs_rebuild {
- return;
- }
- self.needs_rebuild = false;
-
- match self.body_instance_buffer.as_mut() {
- Some(buffer) => {
- buffer.1.destroy();
- self.body_instance_buffer = None;
- }
- None => {}
- }
-
- let bodies = solar_system.bodies();
- let buffer_len = bodies.len() * size_of::<BodyInstanceRaw>();
-
- let buffer = wgpuctx.create_buffer(
- &wgpu::BufferDescriptor {
- label: Some("Tactical map bodies instance buffer"),
- size: buffer_len as u64,
- usage: wgpu::BufferUsages::COPY_DST |
- wgpu::BufferUsages::VERTEX,
- mapped_at_creation: false
- }
- );
-
- self.last_time = None;
- self.body_instance_buffer = Some((bodies.len(), buffer));
- }
-
- pub fn update(
- &mut self,
- wgpuctx: &WgpuCtx,
- solar_system: &SolarSystem,
- time: Second)
- -> Result<(), Box<dyn Error>>
- {
- //If the last updated time is the same, we don't need to update
- //the positions of all the bodies.
- if self.last_time.is_some_and(|last_time| { last_time == time }) {
- return Ok(());
- }
-
- self.last_time = Some(time);
-
- let (_, bodies_buffer) = match &self.body_instance_buffer {
- Some(tuple) => tuple,
- None => return Err(Box::new(NeedsRebuildError))
- };
-
- let bodies = solar_system.bodies();
- let body_instances = bodies.iter().map(|body| {
- let position = body.position();
- let origin = match body.get_orbits() {
- Some(origin_id) => {
- let origin_body = &bodies[origin_id];
- origin_body.position()
- },
- None => cgmath::Vector3::new(0.0, 0.0, 0.0)
- };
- BodyInstance {
- position: position,
- origin: origin,
- radius: body.radius()
- }.raw()
- }).collect::<Vec<_>>();
-
- wgpuctx.queue().write_buffer(bodies_buffer, 0, bytemuck::cast_slice(&body_instances));
-
- Ok(())
- }
-
- pub fn render(
- &self,
- pass: &mut RenderPass,
- camera: &Camera)
- {
- let (num_bodies, bodies_buffer) = match &self.body_instance_buffer {
- Some(tuple) => tuple,
- None => return
- };
-
- pass.set_pipeline(&self.pipeline);
- pass.set_bind_group(0, camera.bindgroup(), &[]);
-
- pass.set_vertex_buffer(0, bodies_buffer.slice(..));
-
- pass.draw(0..6, 0..num_bodies.clone() as _);
- }
-} // impl RenderState
-
-struct BodyInstance
-{
- position: cgmath::Vector3<Kilometers>,
- origin: cgmath::Vector3<Kilometers>,
- radius: f32
-}
-
-#[repr(C)]
-#[derive(Copy, Clone, bytemuck::Pod, bytemuck::Zeroable)]
-struct BodyInstanceRaw
-{
- position: [f32;3],
- origin: [f32;3],
- radius: f32
-}
-
-impl BodyInstance
-{
- fn raw(&self) -> BodyInstanceRaw
- {
- BodyInstanceRaw {
- position: [
- self.position.x as f32,
- self.position.y as f32,
- self.position.z as f32 ],
- origin: [
- self.origin.x as f32,
- self.origin.y as f32,
- self.origin.z as f32 ],
- radius: self.radius
- }
- }
-} // impl BodyInstance
-
-impl BodyInstanceRaw
-{
- const ATTRIBS: [wgpu::VertexAttribute;3] =
- wgpu::vertex_attr_array![0 => Float32x3, 1 => Float32x3, 2 => Float32];
-
- pub fn descr()
- -> wgpu::VertexBufferLayout<'static> {
- use std::mem;
-
- wgpu::VertexBufferLayout {
- array_stride: mem::size_of::<Self>() as wgpu::BufferAddress,
- step_mode: wgpu::VertexStepMode::Instance,
- attributes: &Self::ATTRIBS
- }
- }
-}
-
pub struct GridRenderer
{
pipeline: wgpu::RenderPipeline,
@@ -242,238 +57,3 @@ impl GridRenderer
pass.draw(0..6, 0..1);
}
} //impl GridRenderer
-
-pub struct OrbitRenderer
-{
- needs_rebuild: bool,
- last_time: Option<Second>,
-
- pipeline: wgpu::RenderPipeline,
-
- orbit_vertex_buffers: Option<Vec<(usize, Option<wgpu::Buffer>)>>,
- origin_buffer: Option<wgpu::Buffer>,
- origin_bind_group: Option<wgpu::BindGroup>,
- origin_bind_group_layout: wgpu::BindGroupLayout
-}
-
-impl OrbitRenderer
-{
- pub fn new(
- wgpuctx: &WgpuCtx)
- -> Self {
- let shader = wgpuctx.create_shader(
- wgpu::include_wgsl!(concat!(
- env!("CARGO_MANIFEST_DIR"),
- "/assets/shaders/tacmap/orbit.wgsl")
- ));
-
- let origin_bind_group_layout = wgpuctx.device().create_bind_group_layout(
- &wgpu::BindGroupLayoutDescriptor {
- label: None,
- entries: &[
- wgpu::BindGroupLayoutEntry {
- binding: 0,
- visibility: wgpu::ShaderStages::VERTEX,
- ty: wgpu::BindingType::Buffer {
- ty: wgpu::BufferBindingType::Storage { read_only: true },
- has_dynamic_offset: false,
- min_binding_size: None
- },
- count: None
- }
- ]
- }
- );
-
- let render_pipeline = RenderPipelineBuilder::new(&shader)
- .add_bindgroup(&Camera::bindgroup_layout(wgpuctx))
- .add_bindgroup(&origin_bind_group_layout)
- .add_vertex_layout(OrbitVertex::descr())
- .primitive_topology(wgpu::PrimitiveTopology::TriangleStrip)
- .cull_mode(None)
- .build(Some("Tactical map orbit render pipeline"), wgpuctx);
-
- Self {
- needs_rebuild: true,
- last_time: None,
- pipeline: render_pipeline,
- orbit_vertex_buffers: None,
- origin_buffer: None,
- origin_bind_group: None,
- origin_bind_group_layout: origin_bind_group_layout
- }
- }
-
- pub fn mark_to_rebuild(&mut self)
- { self.needs_rebuild = true; }
-
- pub fn rebuild(
- &mut self,
- wgpuctx: &WgpuCtx,
- solar_system: &SolarSystem)
- {
- if self.orbit_vertex_buffers.is_some() && !self.needs_rebuild {
- return;
- }
- self.needs_rebuild = false;
-
- match &self.orbit_vertex_buffers {
- Some(buffers) => {
- for buffer in buffers {
- match &buffer.1 {
- Some(v) => v.destroy(),
- None => {}
- }
- }
- self.orbit_vertex_buffers = None;
- },
- None => {}
- };
-
- //From the solar system's bodies, calculate the orbits.
- let bodies = solar_system.bodies();
-
- match &self.origin_buffer {
- Some(buffer) => {
- buffer.destroy();
- }
- None => {}
- };
- let origin_buffer = wgpuctx.create_buffer(
- &wgpu::BufferDescriptor {
- label: None,
- size: (std::mem::size_of::<[f32;3]>() * bodies.len()) as _,
- usage: wgpu::BufferUsages::COPY_DST
- | wgpu::BufferUsages::STORAGE,
- mapped_at_creation: false
- }
- );
-
- self.origin_bind_group = Some(wgpuctx.device().create_bind_group(
- &wgpu::BindGroupDescriptor {
- label: None,
- layout: &self.origin_bind_group_layout,
- entries: &[
- wgpu::BindGroupEntry {
- binding: 0,
- resource: origin_buffer.as_entire_binding()
- }
- ]
- }
- ));
- self.origin_buffer = Some(origin_buffer);
-
- self.orbit_vertex_buffers = Some(bodies.iter().map(|body| {
- if !body.does_orbit() { return (0, None); }
-
- let period = body.orbital_period();
- let num_points = ((period / (SYSTEM_TICK_INTERVAL)) as usize).clamp(180, 360*4);
- let period_interval = period as f64 / num_points as f64;
-
- let mut points = vec![
- OrbitVertex::default();(num_points+1)*2];
- for i in 0..num_points {
- let position = body.calculate_orbit_at(
- (i as f64 * period_interval) as u64);
-
- points[i*2].origin_id = body.get_orbits().unwrap() as _;
- points[i*2].position = [
- position.x as f32,
- position.y as f32,
- position.z as f32
- ];
-
- points[(i*2)+1] = points[i*2].clone();
- }
- points[num_points*2] = points[0].clone();
- points[num_points*2+1] = points[1].clone();
-
- let buffer = wgpuctx.create_buffer_init(
- &wgpu::util::BufferInitDescriptor {
- label: None,
- usage: wgpu::BufferUsages::VERTEX,
- contents: bytemuck::cast_slice(&points)
- }
- );
- ((num_points+1)*2, Some(buffer))
- }).collect::<Vec<_>>());
- }
-
- pub fn update(
- &mut self,
- wgpuctx: &WgpuCtx,
- solar_system: &SolarSystem,
- time: Second)
- {
- if self.last_time.is_some_and(|t| { t == time }) {
- return;
- }
-
- self.last_time = Some(time);
- let positions = solar_system.bodies().iter().map(|body| {
- let pos = solar_system.body_position(body);
- [ pos.x as f32, pos.y as f32, pos.z as f32 ]
- }).collect::<Vec<_>>();
-
- match &self.origin_buffer {
- Some(buffer) => {
- wgpuctx.queue().write_buffer(buffer, 0, bytemuck::cast_slice(&positions));
- },
- None => { return; }
- };
- }
-
- pub fn render(
- &self,
- pass: &mut RenderPass,
- camera: &Camera)
- {
- let buffers = match &self.orbit_vertex_buffers {
- Some(v) => v,
- None => return
- };
-
- pass.set_pipeline(&self.pipeline);
- pass.set_bind_group(0, camera.bindgroup(), &[]);
-
- match &self.origin_bind_group {
- Some(bind_group) => { pass.set_bind_group(1, bind_group, &[]); },
- None => { return; }
- }
-
- for (num_points, buffer) in buffers {
- match &buffer {
- Some(buffer) => {
- pass.set_vertex_buffer(0, buffer.slice(..));
- pass.draw(0..(*num_points as u32), 0..1);
- },
- None => {}
- }
- }
- }
-}
-
-#[repr(C)]
-#[derive(Copy, Clone, Default, Debug, bytemuck::Pod, bytemuck::Zeroable)]
-pub struct OrbitVertex
-{
- pub origin_id: u32,
- pub position: [f32;3]
-}
-
-impl OrbitVertex
-{
- const ATTRIBS: [wgpu::VertexAttribute;2] =
- wgpu::vertex_attr_array![0 => Uint32, 1 => Float32x3];
-
- pub fn descr()
- -> wgpu::VertexBufferLayout<'static> {
- use std::mem;
-
- wgpu::VertexBufferLayout {
- array_stride: mem::size_of::<Self>() as wgpu::BufferAddress,
- step_mode: wgpu::VertexStepMode::Vertex,
- attributes: &Self::ATTRIBS
- }
- }
-}