forked from 1b-agent/opencadstudio-upstream
feat(plugin): support analytical GPU preview wires for circles and arcs
This commit is contained in:
1 parent
d218ec1451
commit
ffdde71057
2 files changed
+244
-42
No files matched your search
@@ -611,23 +611,88 @@ pub trait InteractiveCommand: Send {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// A preview wire/polyline rendered in real-time during interactive commands.
|
/// A preview wire rendered in real-time during interactive commands.
|
||||||
|
/// Supports straight polylines as well as analytical circles and arcs
|
||||||
|
/// that render via GPU shaders with infinite smoothness.
|
||||||
#[derive(Debug, Clone, PartialEq)]
|
#[derive(Debug, Clone, PartialEq)]
|
||||||
#[cfg_attr(feature = "host", derive(serde::Serialize, serde::Deserialize))]
|
#[cfg_attr(feature = "host", derive(serde::Serialize, serde::Deserialize))]
|
||||||
pub struct PreviewWire {
|
pub enum PreviewWire {
|
||||||
/// World-coordinate points along the polyline.
|
/// A connected sequence of straight line segments in world space.
|
||||||
pub points: Vec<[f64; 3]>,
|
Polyline {
|
||||||
/// Optional RGBA color (0.0 to 1.0). If `None`, the host's default rubber-band color (cyan) is used.
|
points: Vec<[f64; 3]>,
|
||||||
pub color: Option<[f32; 4]>,
|
color: Option<[f32; 4]>,
|
||||||
|
},
|
||||||
|
/// An analytical circle rendered with sub-pixel GPU anti-aliasing.
|
||||||
|
Circle {
|
||||||
|
center: [f64; 3],
|
||||||
|
radius: f64,
|
||||||
|
color: Option<[f32; 4]>,
|
||||||
|
},
|
||||||
|
/// An analytical circular arc swept counter-clockwise from `start_angle_rad` to `end_angle_rad`.
|
||||||
|
Arc {
|
||||||
|
center: [f64; 3],
|
||||||
|
radius: f64,
|
||||||
|
start_angle_rad: f64,
|
||||||
|
end_angle_rad: f64,
|
||||||
|
color: Option<[f32; 4]>,
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
impl PreviewWire {
|
impl PreviewWire {
|
||||||
|
/// Create a polyline preview wire with default host cyan color.
|
||||||
pub fn new(points: Vec<[f64; 3]>) -> Self {
|
pub fn new(points: Vec<[f64; 3]>) -> Self {
|
||||||
Self { points, color: None }
|
Self::Polyline { points, color: None }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Optional RGBA color (0.0 to 1.0) specified for this preview wire.
|
||||||
|
pub fn color(&self) -> Option<[f32; 4]> {
|
||||||
|
match self {
|
||||||
|
Self::Polyline { color, .. } => *color,
|
||||||
|
Self::Circle { color, .. } => *color,
|
||||||
|
Self::Arc { color, .. } => *color,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create a polyline preview wire with a custom RGBA color.
|
||||||
pub fn with_color(points: Vec<[f64; 3]>, color: [f32; 4]) -> Self {
|
pub fn with_color(points: Vec<[f64; 3]>, color: [f32; 4]) -> Self {
|
||||||
Self { points, color: Some(color) }
|
Self::Polyline {
|
||||||
|
points,
|
||||||
|
color: Some(color),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create a straight line segment preview between two points.
|
||||||
|
pub fn line(from: [f64; 3], to: [f64; 3], color: Option<[f32; 4]>) -> Self {
|
||||||
|
Self::Polyline {
|
||||||
|
points: vec![from, to],
|
||||||
|
color,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create an analytical circle preview.
|
||||||
|
pub fn circle(center: [f64; 3], radius: f64, color: Option<[f32; 4]>) -> Self {
|
||||||
|
Self::Circle {
|
||||||
|
center,
|
||||||
|
radius,
|
||||||
|
color,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create an analytical circular arc preview.
|
||||||
|
pub fn arc(
|
||||||
|
center: [f64; 3],
|
||||||
|
radius: f64,
|
||||||
|
start_angle_rad: f64,
|
||||||
|
end_angle_rad: f64,
|
||||||
|
color: Option<[f32; 4]>,
|
||||||
|
) -> Self {
|
||||||
|
Self::Arc {
|
||||||
|
center,
|
||||||
|
radius,
|
||||||
|
start_angle_rad,
|
||||||
|
end_angle_rad,
|
||||||
|
color,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
+171
-34
@@ -2715,20 +2715,7 @@ impl crate::command::CadCommand for PluginInteractiveAdapter {
|
|||||||
self.inner.on_cursor_move([pt.x as f64, pt.y as f64, pt.z as f64])
|
self.inner.on_cursor_move([pt.x as f64, pt.y as f64, pt.z as f64])
|
||||||
})
|
})
|
||||||
.unwrap_or_default();
|
.unwrap_or_default();
|
||||||
|
convert_preview_wires(raw_wires)
|
||||||
raw_wires
|
|
||||||
.into_iter()
|
|
||||||
.filter(|w| w.points.len() >= 2)
|
|
||||||
.map(|w| {
|
|
||||||
let color = w.color.unwrap_or(crate::scene::model::wire_model::WireModel::CYAN);
|
|
||||||
crate::scene::model::wire_model::WireModel::solid_f64(
|
|
||||||
"rubber_band".into(),
|
|
||||||
w.points,
|
|
||||||
color,
|
|
||||||
false,
|
|
||||||
)
|
|
||||||
})
|
|
||||||
.collect()
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -2862,23 +2849,87 @@ impl crate::command::CadCommand for PluginProcessInteractiveAdapter {
|
|||||||
.process
|
.process
|
||||||
.on_cursor_move(self.command_id, [pt.x as f64, pt.y as f64, pt.z as f64])
|
.on_cursor_move(self.command_id, [pt.x as f64, pt.y as f64, pt.z as f64])
|
||||||
.unwrap_or_default();
|
.unwrap_or_default();
|
||||||
|
convert_preview_wires(raw_wires)
|
||||||
raw_wires
|
|
||||||
.into_iter()
|
|
||||||
.filter(|w| w.points.len() >= 2)
|
|
||||||
.map(|w| {
|
|
||||||
let color = w.color.unwrap_or(crate::scene::model::wire_model::WireModel::CYAN);
|
|
||||||
crate::scene::model::wire_model::WireModel::solid_f64(
|
|
||||||
"rubber_band".into(),
|
|
||||||
w.points,
|
|
||||||
color,
|
|
||||||
false,
|
|
||||||
)
|
|
||||||
})
|
|
||||||
.collect()
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn convert_preview_wires(
|
||||||
|
raw_wires: Vec<ocs_plugin_api::host::PreviewWire>,
|
||||||
|
) -> Vec<crate::scene::model::wire_model::WireModel> {
|
||||||
|
use ocs_plugin_api::host::PreviewWire;
|
||||||
|
use crate::scene::model::wire_model::{TangentGeom, WireModel};
|
||||||
|
|
||||||
|
raw_wires
|
||||||
|
.into_iter()
|
||||||
|
.filter_map(|w| match w {
|
||||||
|
PreviewWire::Polyline { points, color } => {
|
||||||
|
if points.len() < 2 {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
let c = color.unwrap_or(WireModel::CYAN);
|
||||||
|
Some(WireModel::solid_f64("rubber_band".into(), points, c, false))
|
||||||
|
}
|
||||||
|
PreviewWire::Circle { center, radius, color } => {
|
||||||
|
if radius <= 0.0 || !radius.is_finite() {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
let c = color.unwrap_or(WireModel::CYAN);
|
||||||
|
// Two opposite diameter points ensure OCS's viewport frame-cache
|
||||||
|
// (render_signature) detects cursor motion and triggers 60-120 FPS
|
||||||
|
// redraws without requiring polar snap or artificial guide lines.
|
||||||
|
let points = vec![
|
||||||
|
[center[0] - radius, center[1], center[2]],
|
||||||
|
[center[0] + radius, center[1], center[2]],
|
||||||
|
];
|
||||||
|
let mut wire = WireModel::solid_f64("rubber_band".into(), points, c, false);
|
||||||
|
wire.tangent_geoms.push(TangentGeom::PlanarCircle {
|
||||||
|
center,
|
||||||
|
axis_x: [1.0, 0.0, 0.0],
|
||||||
|
axis_y: [0.0, 1.0, 0.0],
|
||||||
|
radius,
|
||||||
|
});
|
||||||
|
Some(wire)
|
||||||
|
}
|
||||||
|
PreviewWire::Arc {
|
||||||
|
center,
|
||||||
|
radius,
|
||||||
|
start_angle_rad,
|
||||||
|
end_angle_rad,
|
||||||
|
color,
|
||||||
|
} => {
|
||||||
|
if radius <= 0.0 || !radius.is_finite() {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
let c = color.unwrap_or(WireModel::CYAN);
|
||||||
|
// Start and end points along the circumference ensure OCS's viewport
|
||||||
|
// frame-cache (render_signature) detects cursor motion and triggers
|
||||||
|
// 60-120 FPS redraws without requiring polar snap or artificial guide lines.
|
||||||
|
let p_start = [
|
||||||
|
center[0] + radius * start_angle_rad.cos(),
|
||||||
|
center[1] + radius * start_angle_rad.sin(),
|
||||||
|
center[2],
|
||||||
|
];
|
||||||
|
let p_end = [
|
||||||
|
center[0] + radius * end_angle_rad.cos(),
|
||||||
|
center[1] + radius * end_angle_rad.sin(),
|
||||||
|
center[2],
|
||||||
|
];
|
||||||
|
let points = vec![p_start, p_end];
|
||||||
|
let mut wire = WireModel::solid_f64("rubber_band".into(), points, c, false);
|
||||||
|
wire.tangent_geoms.push(TangentGeom::Arc {
|
||||||
|
center,
|
||||||
|
axis_x: [1.0, 0.0, 0.0],
|
||||||
|
axis_y: [0.0, 1.0, 0.0],
|
||||||
|
radius,
|
||||||
|
start_angle: start_angle_rad,
|
||||||
|
end_angle: end_angle_rad,
|
||||||
|
});
|
||||||
|
Some(wire)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.collect()
|
||||||
|
}
|
||||||
|
|
||||||
fn plugin_step_to_result(step: ocs_plugin_api::host::CommandStep) -> crate::command::CmdResult {
|
fn plugin_step_to_result(step: ocs_plugin_api::host::CommandStep) -> crate::command::CmdResult {
|
||||||
use crate::command::CmdResult;
|
use crate::command::CmdResult;
|
||||||
use ocs_plugin_api::host::CommandStep;
|
use ocs_plugin_api::host::CommandStep;
|
||||||
@@ -9338,10 +9389,53 @@ step('undo_move', lambda: M.move([u], (0, 0, 0), (7, 0, 0)))
|
|||||||
ocs_plugin_api::host::CommandStep::Done
|
ocs_plugin_api::host::CommandStep::Done
|
||||||
}
|
}
|
||||||
fn on_cursor_move(&mut self, pt: [f64; 3]) -> Vec<ocs_plugin_api::host::PreviewWire> {
|
fn on_cursor_move(&mut self, pt: [f64; 3]) -> Vec<ocs_plugin_api::host::PreviewWire> {
|
||||||
vec![ocs_plugin_api::host::PreviewWire {
|
vec![ocs_plugin_api::host::PreviewWire::line(self.start, pt, self.color)]
|
||||||
points: vec![self.start, pt],
|
}
|
||||||
color: self.color,
|
}
|
||||||
}]
|
|
||||||
|
/// A plugin command that previews an analytical circle.
|
||||||
|
struct PreviewCircle {
|
||||||
|
center: [f64; 3],
|
||||||
|
color: Option<[f32; 4]>,
|
||||||
|
}
|
||||||
|
impl ocs_plugin_api::host::InteractiveCommand for PreviewCircle {
|
||||||
|
fn prompt(&self) -> String {
|
||||||
|
"Pick radius".to_string()
|
||||||
|
}
|
||||||
|
fn on_point(&mut self, _pt: [f64; 3]) -> ocs_plugin_api::host::CommandStep {
|
||||||
|
ocs_plugin_api::host::CommandStep::Done
|
||||||
|
}
|
||||||
|
fn on_cursor_move(&mut self, pt: [f64; 3]) -> Vec<ocs_plugin_api::host::PreviewWire> {
|
||||||
|
let dx = pt[0] - self.center[0];
|
||||||
|
let dy = pt[1] - self.center[1];
|
||||||
|
let r = (dx * dx + dy * dy).sqrt();
|
||||||
|
vec![ocs_plugin_api::host::PreviewWire::circle(self.center, r, self.color)]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// A plugin command that previews an analytical arc.
|
||||||
|
struct PreviewArc {
|
||||||
|
center: [f64; 3],
|
||||||
|
color: Option<[f32; 4]>,
|
||||||
|
}
|
||||||
|
impl ocs_plugin_api::host::InteractiveCommand for PreviewArc {
|
||||||
|
fn prompt(&self) -> String {
|
||||||
|
"Pick endpoint".to_string()
|
||||||
|
}
|
||||||
|
fn on_point(&mut self, _pt: [f64; 3]) -> ocs_plugin_api::host::CommandStep {
|
||||||
|
ocs_plugin_api::host::CommandStep::Done
|
||||||
|
}
|
||||||
|
fn on_cursor_move(&mut self, pt: [f64; 3]) -> Vec<ocs_plugin_api::host::PreviewWire> {
|
||||||
|
let dx = pt[0] - self.center[0];
|
||||||
|
let dy = pt[1] - self.center[1];
|
||||||
|
let r = (dx * dx + dy * dy).sqrt();
|
||||||
|
vec![ocs_plugin_api::host::PreviewWire::arc(
|
||||||
|
self.center,
|
||||||
|
r,
|
||||||
|
0.0,
|
||||||
|
std::f64::consts::PI,
|
||||||
|
self.color,
|
||||||
|
)]
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -9350,7 +9444,7 @@ step('undo_move', lambda: M.move([u], (0, 0, 0), (7, 0, 0)))
|
|||||||
let mut app = OpenCADStudio::new_for_test();
|
let mut app = OpenCADStudio::new_for_test();
|
||||||
app.tabs[0].is_start = false;
|
app.tabs[0].is_start = false;
|
||||||
|
|
||||||
// 1. Default color (None -> CYAN)
|
// 1. Polyline default color (None -> CYAN)
|
||||||
{
|
{
|
||||||
let mut host = HostSession::new(&mut app, 0);
|
let mut host = HostSession::new(&mut app, 0);
|
||||||
host.start_interactive(Box::new(PreviewLine { start: [0.0, 0.0, 0.0], color: None }));
|
host.start_interactive(Box::new(PreviewLine { start: [0.0, 0.0, 0.0], color: None }));
|
||||||
@@ -9362,7 +9456,7 @@ step('undo_move', lambda: M.move([u], (0, 0, 0), (7, 0, 0)))
|
|||||||
assert_eq!(wires[0].name, "rubber_band");
|
assert_eq!(wires[0].name, "rubber_band");
|
||||||
assert_eq!(wires[0].color, crate::scene::model::wire_model::WireModel::CYAN);
|
assert_eq!(wires[0].color, crate::scene::model::wire_model::WireModel::CYAN);
|
||||||
|
|
||||||
// 2. Custom color ([1.0, 0.0, 0.0, 1.0])
|
// 2. Polyline custom color ([1.0, 0.0, 0.0, 1.0])
|
||||||
{
|
{
|
||||||
let mut host = HostSession::new(&mut app, 0);
|
let mut host = HostSession::new(&mut app, 0);
|
||||||
host.start_interactive(Box::new(PreviewLine { start: [0.0, 0.0, 0.0], color: Some([1.0, 0.0, 0.0, 1.0]) }));
|
host.start_interactive(Box::new(PreviewLine { start: [0.0, 0.0, 0.0], color: Some([1.0, 0.0, 0.0, 1.0]) }));
|
||||||
@@ -9371,6 +9465,49 @@ step('undo_move', lambda: M.move([u], (0, 0, 0), (7, 0, 0)))
|
|||||||
let wires2 = cmd2.on_preview_wires(glam::DVec3::new(10.0, 20.0, 0.0));
|
let wires2 = cmd2.on_preview_wires(glam::DVec3::new(10.0, 20.0, 0.0));
|
||||||
assert_eq!(wires2.len(), 1);
|
assert_eq!(wires2.len(), 1);
|
||||||
assert_eq!(wires2[0].color, [1.0, 0.0, 0.0, 1.0]);
|
assert_eq!(wires2[0].color, [1.0, 0.0, 0.0, 1.0]);
|
||||||
|
|
||||||
|
// 3. Analytical Circle with custom color
|
||||||
|
{
|
||||||
|
let mut host = HostSession::new(&mut app, 0);
|
||||||
|
host.start_interactive(Box::new(PreviewCircle {
|
||||||
|
center: [5.0, 5.0, 0.0],
|
||||||
|
color: Some([0.0, 1.0, 0.0, 1.0]),
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
let cmd3 = app.tabs[0].active_cmd.as_mut().unwrap();
|
||||||
|
let circle_wires = cmd3.on_preview_wires(glam::DVec3::new(15.0, 5.0, 0.0));
|
||||||
|
assert_eq!(circle_wires.len(), 1);
|
||||||
|
assert_eq!(circle_wires[0].points.len(), 2);
|
||||||
|
assert_eq!(circle_wires[0].color, [0.0, 1.0, 0.0, 1.0]);
|
||||||
|
assert_eq!(circle_wires[0].tangent_geoms.len(), 1);
|
||||||
|
assert!(matches!(
|
||||||
|
circle_wires[0].tangent_geoms[0],
|
||||||
|
crate::scene::model::wire_model::TangentGeom::PlanarCircle { radius, .. } if (radius - 10.0).abs() < 1e-6
|
||||||
|
));
|
||||||
|
let instances = crate::scene::pipeline::circle_gpu::extract_circle_instances(&circle_wires[0], 0.0);
|
||||||
|
assert!(instances.is_some());
|
||||||
|
assert_eq!(instances.unwrap().len(), 1);
|
||||||
|
|
||||||
|
// 4. Analytical Arc
|
||||||
|
{
|
||||||
|
let mut host = HostSession::new(&mut app, 0);
|
||||||
|
host.start_interactive(Box::new(PreviewArc {
|
||||||
|
center: [0.0, 0.0, 0.0],
|
||||||
|
color: None,
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
let cmd4 = app.tabs[0].active_cmd.as_mut().unwrap();
|
||||||
|
let arc_wires = cmd4.on_preview_wires(glam::DVec3::new(10.0, 0.0, 0.0));
|
||||||
|
assert_eq!(arc_wires.len(), 1);
|
||||||
|
assert_eq!(arc_wires[0].points.len(), 2);
|
||||||
|
assert_eq!(arc_wires[0].tangent_geoms.len(), 1);
|
||||||
|
assert!(matches!(
|
||||||
|
arc_wires[0].tangent_geoms[0],
|
||||||
|
crate::scene::model::wire_model::TangentGeom::Arc { radius, .. } if (radius - 10.0).abs() < 1e-6
|
||||||
|
));
|
||||||
|
let arc_instances = crate::scene::pipeline::circle_gpu::extract_circle_instances(&arc_wires[0], 0.0);
|
||||||
|
assert!(arc_instances.is_some());
|
||||||
|
assert_eq!(arc_instances.unwrap().len(), 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// A plugin command that picks an existing object, then marks it.
|
/// A plugin command that picks an existing object, then marks it.
|
||||||
|
|||||||
Reference in new issue
Block a user