feat(field): expose ray marching to clients

This commit is contained in:
Nova
2022-11-01 07:59:49 -04:00
parent f19ba93958
commit cfa3584dda
2 changed files with 31 additions and 11 deletions

View File

@@ -13,7 +13,7 @@ use crate::nodes::spatial::find_reference_space;
use anyhow::Result;
use glam::{vec2, vec3a, Vec3, Vec3A};
use mint::Vector3;
use serde::Deserialize;
use serde::{Deserialize, Serialize};
use stardust_xr::schemas::flex::{deserialize, serialize};
use std::ops::Deref;
@@ -65,6 +65,7 @@ pub trait FieldTrait {
node.add_local_method("distance", field_distance_flex);
node.add_local_method("normal", field_normal_flex);
node.add_local_method("closest_point", field_closest_point_flex);
node.add_local_method("ray_march", field_ray_march_flex);
}
fn spatial_ref(&self) -> &Spatial;
@@ -124,6 +125,26 @@ fn field_closest_point_flex(
);
Ok(serialize(mint::Vector3::from(closest_point))?)
}
fn field_ray_march_flex(node: &Node, calling_client: Arc<Client>, data: &[u8]) -> Result<Vec<u8>> {
#[derive(Deserialize)]
struct FieldInfoArgs<'a> {
reference_space_path: &'a str,
ray_origin: Vector3<f32>,
ray_direction: Vector3<f32>,
}
let args: FieldInfoArgs = deserialize(data)?;
let reference_space = find_reference_space(&calling_client, args.reference_space_path)?;
let ray_march_result = ray_march(
Ray {
origin: args.ray_origin.into(),
direction: args.ray_direction.into(),
space: reference_space,
},
node.field.get().unwrap(),
);
Ok(serialize(ray_march_result)?)
}
pub enum Field {
Box(BoxField),
@@ -156,9 +177,9 @@ pub struct Ray {
pub space: Arc<Spatial>,
}
#[derive(Debug, Serialize)]
pub struct RayMarchResult {
pub ray: Ray,
pub distance: f32,
pub min_distance: f32,
pub deepest_point_distance: f32,
pub ray_length: f32,
pub ray_steps: u32,
@@ -175,17 +196,16 @@ const MAX_RAY_LENGTH: f32 = 1000_f32;
pub fn ray_march(ray: Ray, field: &Field) -> RayMarchResult {
let mut result = RayMarchResult {
ray,
distance: f32::MAX,
min_distance: f32::MAX,
deepest_point_distance: 0_f32,
ray_length: 0_f32,
ray_steps: 0,
};
let ray_to_field_matrix =
Spatial::space_to_space_matrix(Some(&result.ray.space), Some(field.spatial_ref()));
let mut ray_point = ray_to_field_matrix.transform_point3a(result.ray.origin.into());
let ray_direction = ray_to_field_matrix.transform_vector3a(result.ray.direction.into());
Spatial::space_to_space_matrix(Some(&ray.space), Some(field.spatial_ref()));
let mut ray_point = ray_to_field_matrix.transform_point3a(ray.origin.into());
let ray_direction = ray_to_field_matrix.transform_vector3a(ray.direction.into());
while result.ray_steps < MAX_RAY_STEPS && result.ray_length < MAX_RAY_LENGTH {
let distance = field.local_distance(ray_point);
@@ -194,10 +214,10 @@ pub fn ray_march(ray: Ray, field: &Field) -> RayMarchResult {
result.ray_length += march_distance;
ray_point += ray_direction * march_distance;
if result.distance > distance {
if result.min_distance > distance {
result.deepest_point_distance = result.ray_length;
}
result.distance = distance.min(result.distance);
result.min_distance = distance.min(result.min_distance);
result.ray_steps += 1;
}

View File

@@ -30,7 +30,7 @@ impl Pointer {
impl InputSpecialization for Pointer {
fn distance(&self, space: &Arc<Spatial>, field: &Field) -> f32 {
self.ray_march(space, field).distance
self.ray_march(space, field).min_distance
}
fn serialize(
&self,