use super::*;
#[derive(Default, Clone, Copy, PartialEq, Eq, Debug)]
pub enum GizmoMode {
#[default]
None,
Transform,
Dimension,
}
#[derive(Default)]
pub struct TransformArm {
pub(super) feature_id: Option<String>,
pub(super) mode: GizmoMode,
pub(super) drag: Option<TransformDrag>,
pub(super) anchor: Option<usize>,
}
#[derive(Clone, Copy)]
pub(super) struct TransformDrag {
handle: u32,
sx: f32,
sy: f32,
start: TransformPose,
}
#[derive(Clone, Copy, PartialEq, Debug)]
struct TransformPose {
position: [f64; 3],
rotation_deg: [f64; 3],
scale: [f64; 3],
}
#[derive(Clone, Copy, PartialEq, Debug)]
pub(super) enum TransformDelta {
Translate([f64; 3]),
Rotate { axis: [f64; 3], radians: f64 },
}
pub(super) fn parse_drag_delta(json: &str) -> Option<TransformDelta> {
let value: serde_json::Value = serde_json::from_str(json).ok()?;
match value.get("kind").and_then(|k| k.as_str()) {
Some("translate") => Some(TransformDelta::Translate(
crate::json_support::vec3_or(value.get("world"), [0.0; 3]),
)),
Some("rotate") => Some(TransformDelta::Rotate {
axis: crate::json_support::vec3_or(value.get("axisWorld"), [0.0; 3]),
radians: value.get("radians").and_then(|n| n.as_f64()).unwrap_or(0.0),
}),
_ => None,
}
}
impl EngineState {
pub fn transform_armed(&self) -> bool {
matches!(self.transform_gizmo.mode, GizmoMode::Transform)
}
pub fn transform_armed_feature(&self) -> String {
if self.transform_armed() {
self.transform_gizmo.feature_id.clone().unwrap_or_default()
} else {
String::new()
}
}
pub fn gizmo_state_json(&self) -> String {
let mode = match self.transform_gizmo.mode {
GizmoMode::None => "none",
GizmoMode::Transform => "transform",
GizmoMode::Dimension => "dimension",
};
serde_json::json!({
"mode": mode,
"armed": self.transform_armed(),
"feature": self.transform_gizmo.feature_id,
"anchor": self.transform_gizmo.anchor,
"origin": self.widgets.transform_origin(),
})
.to_string()
}
pub fn transform_armed_for(&self, feature_id: &str) -> bool {
self.transform_armed() && self.transform_gizmo.feature_id.as_deref() == Some(feature_id)
}
pub fn arm_transform(&mut self, feature_id: &str) {
let is_acomp = self
.history
.index_of(feature_id)
.and_then(|index| self.history.feature_type(index))
.is_some_and(|ty| super::components::is_acomp_feature_type(&ty));
if is_acomp {
self.component_move_arm(feature_id);
return;
}
self.component_move_reset();
self.transform_gizmo.feature_id = Some(feature_id.to_string());
self.transform_gizmo.mode = GizmoMode::Transform;
self.transform_gizmo.drag = None;
self.transform_gizmo.anchor = None;
self.clear_feature_dimension_overlay();
self.sync_transform_gizmo();
}
pub fn disarm_transform(&mut self) {
self.component_move_reset();
self.transform_gizmo.feature_id = None;
self.transform_gizmo.mode = GizmoMode::None;
self.transform_gizmo.drag = None;
self.transform_gizmo.anchor = None;
let _ = self.widgets.set_transform_json("null");
self.clear_feature_dimension_overlay();
self.dirty = true;
}
fn armed_pose(&self) -> Option<TransformPose> {
let id = self.transform_gizmo.feature_id.as_deref()?;
let index = self.history.index_of(id)?;
if let Some(anchor) = self.transform_gizmo.anchor {
let (position, rotation_deg) = self.spline_anchor_pose(index, anchor)?;
return Some(TransformPose {
position,
rotation_deg,
scale: [1.0, 1.0, 1.0],
});
}
let params = self.history.feature_params(index)?;
let transform = params.get("transform");
let pivot = if self.is_xform_feature(index) {
self.xform_pivot(index, ¶ms)
} else {
[0.0; 3]
};
let env = brep_kernel::Env::build(&self.history.expressions(), &self.history.configurator())
.ok();
let env = env.as_ref();
let position = read_pose_vec3(env, transform, "position", [0.0; 3]);
Some(TransformPose {
position: std::array::from_fn(|axis| position[axis] + pivot[axis]),
rotation_deg: read_pose_vec3(env, transform, "rotationEuler", [0.0, 0.0, 0.0]),
scale: read_pose_vec3(env, transform, "scale", [1.0, 1.0, 1.0]),
})
}
fn is_xform_feature(&self, index: usize) -> bool {
self.history.feature_type(index).is_some_and(|ty| {
ty.eq_ignore_ascii_case("XFORM") || ty.eq_ignore_ascii_case("TRANSFORM")
})
}
fn xform_pivot(&self, index: usize, params: &serde_json::Value) -> [f64; 3] {
if !params["pivot"]
.as_str()
.is_some_and(|p| p.eq_ignore_ascii_case("BBOX_CENTER"))
{
return [0.0; 3];
}
let Ok(request) = serde_json::from_value::<HistoryRequest>(self.history.prefix_request())
else {
return [0.0; 3];
};
let result = brep_kernel::execute_history(&request);
let mut handles = HashMap::new();
for feature in result.results.iter().take(index) {
for name in &feature.removed {
handles.remove(name);
}
for solid in &feature.added {
handles.insert(solid.name.clone(), solid.handle);
}
}
let Some(refs) = params["solids"].as_array() else {
return [0.0; 3];
};
for reference in refs {
let Some(name) = reference.as_str().or_else(|| reference["name"].as_str()) else {
continue;
};
let Some(handle) = handles.get(name.trim()) else {
continue;
};
if let Ok(center) = brep_kernel::transform_pivot_native(*handle) {
return center;
}
}
[0.0; 3]
}
pub fn sync_transform_gizmo(&mut self) {
if !matches!(self.transform_gizmo.mode, GizmoMode::Transform) {
return;
}
let Some(pose) = self.armed_pose() else {
self.disarm_transform();
return;
};
let _ = self
.widgets
.set_transform_json(&transform_frame_json(&pose));
self.dirty = true;
}
pub fn feature_has_transform(&self, feature_id: &str) -> bool {
self.history
.index_of(feature_id)
.and_then(|i| self.history.feature_type(i))
.and_then(|ty| crate::features::feature_schema(&ty))
.is_some_and(|schema| schema["inputParamsSchema"]["transform"]["type"] == "transform")
}
pub fn transform_gizmo_anchor(&self) -> Option<(f64, f64)> {
let pose = self.armed_pose()?;
let (sx, sy, _) = self.camera.project(pose.position);
self.camera.projectable(pose.position).then_some((sx, sy))
}
pub fn transform_axis_labels_json(&self) -> String {
if !matches!(self.transform_gizmo.mode, GizmoMode::Transform) {
return "[]".to_string();
}
let Some(pose) = self.armed_pose() else {
return "[]".to_string();
};
let euler = [
pose.rotation_deg[0].to_radians(),
pose.rotation_deg[1].to_radians(),
pose.rotation_deg[2].to_radians(),
];
let axes = [
normalize3(rotate_euler_xyz_f64([1.0, 0.0, 0.0], euler)),
normalize3(rotate_euler_xyz_f64([0.0, 1.0, 0.0], euler)),
normalize3(rotate_euler_xyz_f64([0.0, 0.0, 1.0], euler)),
];
let gap_px = 12.0_f64;
let dist =
(brep_gizmos::transform::PX_AXIS_LEN as f64 + gap_px) * self.camera.world_per_pixel();
let labels: [(&str, [f32; 3]); 3] = [
("XC", [0.92, 0.26, 0.28]), ("YC", [0.30, 0.78, 0.36]), ("ZC", [0.30, 0.52, 0.98]), ];
let out: Vec<serde_json::Value> = (0..3)
.map(|i| {
let o = pose.position;
let a = axes[i];
serde_json::json!({
"text": labels[i].0,
"rgb": labels[i].1,
"world": [o[0] + a[0] * dist, o[1] + a[1] * dist, o[2] + a[2] * dist],
})
})
.collect();
serde_json::to_string(&out).unwrap_or_else(|_| "[]".to_string())
}
pub fn transform_hit_areas_json(&self) -> String {
let cam = gizmo_camera(&self.camera);
let regions = self.widgets.transform_hit_regions(&cam);
super::camera_widgets::hit_shapes_json(regions.iter().map(|(_, shape)| shape))
}
pub fn transform_press(&mut self, x: f64, y: f64) -> bool {
if !matches!(self.transform_gizmo.mode, GizmoMode::Transform) {
return false;
}
let handle = self.transform_pick(x, y);
if handle == 0 {
return false;
}
let Some(pose) = self.armed_pose() else {
return false;
};
self.widgets.set_transform_active(handle);
self.transform_gizmo.drag = Some(TransformDrag {
handle,
sx: x as f32,
sy: y as f32,
start: pose,
});
self.dirty = true;
true
}
pub fn transform_dragging(&self) -> bool {
self.transform_gizmo.drag.is_some()
}
pub fn transform_drag_to(&mut self, cx: f64, cy: f64) {
let Some(drag) = self.transform_gizmo.drag else {
return;
};
let Some(delta) = self.resolve_transform_delta(&drag, cx, cy) else {
return;
};
let pose = apply_transform_delta(&drag.start, &delta);
self.write_armed_pose(&pose);
self.sync_transform_gizmo();
}
pub fn transform_release(&mut self) {
if self.transform_gizmo.drag.take().is_some() {
self.widgets.set_transform_active(0);
self.sync_transform_gizmo();
}
}
fn resolve_transform_delta(
&self,
drag: &TransformDrag,
cx: f64,
cy: f64,
) -> Option<TransformDelta> {
let cam = gizmo_camera(&self.camera);
let frame_json = transform_frame_json(&drag.start);
let json = self.widgets.transform_drag_json_with_frame(
&cam,
&frame_json,
drag.handle,
drag.sx,
drag.sy,
cx as f32,
cy as f32,
);
parse_drag_delta(&json)
}
fn write_armed_pose(&mut self, pose: &TransformPose) {
let Some(id) = self.transform_gizmo.feature_id.clone() else {
return;
};
let Some(index) = self.history.index_of(&id) else {
return;
};
if let Some(anchor) = self.transform_gizmo.anchor {
self.write_spline_anchor_pose(&id, anchor, pose.position, pose.rotation_deg);
return;
}
let mut params = self
.history
.feature_params(index)
.unwrap_or_else(|| serde_json::json!({}));
let pivot = if self.is_xform_feature(index) {
self.xform_pivot(index, ¶ms)
} else {
[0.0; 3]
};
let position: [f64; 3] = std::array::from_fn(|axis| pose.position[axis] - pivot[axis]);
if !params
.get("transform")
.map(|t| t.is_object())
.unwrap_or(false)
{
if let Some(object) = params.as_object_mut() {
object.insert("transform".into(), serde_json::json!({}));
}
}
if let Some(transform) = params.get_mut("transform").and_then(|t| t.as_object_mut()) {
transform.insert("position".into(), serde_json::json!(position));
transform.insert("rotationEuler".into(), serde_json::json!(pose.rotation_deg));
}
let _ = self.update_feature_params(&id, ¶ms.to_string());
}
}
fn apply_transform_delta(start: &TransformPose, delta: &TransformDelta) -> TransformPose {
match delta {
TransformDelta::Translate(world) => TransformPose {
position: [
start.position[0] + world[0],
start.position[1] + world[1],
start.position[2] + world[2],
],
..*start
},
TransformDelta::Rotate { axis, radians } => {
let q0 = quat_from_euler_xyz_deg(start.rotation_deg);
let dq = quat_from_axis_angle(*axis, *radians);
let nq = quat_mul(dq, q0);
TransformPose {
rotation_deg: euler_xyz_deg_from_quat(nq),
..*start
}
}
}
}
fn read_pose_vec3(
env: Option<&brep_kernel::Env>,
transform: Option<&serde_json::Value>,
key: &str,
default: [f64; 3],
) -> [f64; 3] {
let array = transform
.and_then(|t| t.get(key))
.and_then(|v| v.as_array());
let mut out = default;
if let Some(array) = array {
for (index, slot) in out.iter_mut().enumerate() {
match array.get(index) {
Some(serde_json::Value::Number(number)) => {
if let Some(number) = number.as_f64() {
*slot = number;
}
}
Some(serde_json::Value::String(source)) => {
if let Some(number) = env
.and_then(|env| env.eval(source).ok())
.filter(|number| number.is_finite())
{
*slot = number;
}
}
_ => {}
}
}
}
out
}
fn transform_frame_json(pose: &TransformPose) -> String {
let euler = [
pose.rotation_deg[0].to_radians(),
pose.rotation_deg[1].to_radians(),
pose.rotation_deg[2].to_radians(),
];
let x = normalize3(rotate_euler_xyz_f64([1.0, 0.0, 0.0], euler));
let y = normalize3(rotate_euler_xyz_f64([0.0, 1.0, 0.0], euler));
let z = normalize3(rotate_euler_xyz_f64([0.0, 0.0, 1.0], euler));
serde_json::json!({
"origin": pose.position,
"x": x,
"y": y,
"z": z,
"showCenter": true,
})
.to_string()
}
pub(super) fn normalize3(v: [f64; 3]) -> [f64; 3] {
let length = (v[0] * v[0] + v[1] * v[1] + v[2] * v[2]).sqrt();
if length < 1e-12 {
[0.0, 0.0, 1.0]
} else {
[v[0] / length, v[1] / length, v[2] / length]
}
}
pub(crate) fn rotate_euler_xyz_f64(v: [f64; 3], euler: [f64; 3]) -> [f64; 3] {
let (c1, s1) = (euler[0].cos(), euler[0].sin());
let (c2, s2) = (euler[1].cos(), euler[1].sin());
let (c3, s3) = (euler[2].cos(), euler[2].sin());
let m00 = c2 * c3;
let m01 = -c2 * s3;
let m02 = s2;
let m10 = c1 * s3 + c3 * s1 * s2;
let m11 = c1 * c3 - s1 * s2 * s3;
let m12 = -c2 * s1;
let m20 = s1 * s3 - c1 * c3 * s2;
let m21 = c3 * s1 + c1 * s2 * s3;
let m22 = c1 * c2;
[
m00 * v[0] + m01 * v[1] + m02 * v[2],
m10 * v[0] + m11 * v[1] + m12 * v[2],
m20 * v[0] + m21 * v[1] + m22 * v[2],
]
}
pub(super) type Quat = [f64; 4];
pub(super) fn quat_from_axis_angle(axis: [f64; 3], angle: f64) -> Quat {
let n = normalize3(axis);
let half = angle * 0.5;
let s = half.sin();
[n[0] * s, n[1] * s, n[2] * s, half.cos()]
}
pub(super) fn quat_from_euler_xyz_deg(deg: [f64; 3]) -> Quat {
let (c1, s1) = (
(deg[0].to_radians() * 0.5).cos(),
(deg[0].to_radians() * 0.5).sin(),
);
let (c2, s2) = (
(deg[1].to_radians() * 0.5).cos(),
(deg[1].to_radians() * 0.5).sin(),
);
let (c3, s3) = (
(deg[2].to_radians() * 0.5).cos(),
(deg[2].to_radians() * 0.5).sin(),
);
[
s1 * c2 * c3 + c1 * s2 * s3,
c1 * s2 * c3 - s1 * c2 * s3,
c1 * c2 * s3 + s1 * s2 * c3,
c1 * c2 * c3 - s1 * s2 * s3,
]
}
pub(super) fn quat_mul(a: Quat, b: Quat) -> Quat {
[
a[3] * b[0] + a[0] * b[3] + a[1] * b[2] - a[2] * b[1],
a[3] * b[1] - a[0] * b[2] + a[1] * b[3] + a[2] * b[0],
a[3] * b[2] + a[0] * b[1] - a[1] * b[0] + a[2] * b[3],
a[3] * b[3] - a[0] * b[0] - a[1] * b[1] - a[2] * b[2],
]
}
pub(super) fn euler_xyz_deg_from_quat(q: Quat) -> [f64; 3] {
let [x, y, z, w] = q;
let (x2, y2, z2) = (x + x, y + y, z + z);
let (xx, xy, xz) = (x * x2, x * y2, x * z2);
let (yy, yz, zz) = (y * y2, y * z2, z * z2);
let (wx, wy, wz) = (w * x2, w * y2, w * z2);
let m11 = 1.0 - (yy + zz);
let m12 = xy - wz;
let m13 = xz + wy;
let m22 = 1.0 - (xx + zz);
let m23 = yz - wx;
let m32 = yz + wx;
let m33 = 1.0 - (xx + yy);
let ey = m13.clamp(-1.0, 1.0).asin();
let (ex, ez) = if m13.abs() < 0.9999999 {
((-m23).atan2(m33), (-m12).atan2(m11))
} else {
(m32.atan2(m22), 0.0)
};
[ex.to_degrees(), ey.to_degrees(), ez.to_degrees()]
}