brep-io-kernel 1.0.21 → 1.0.23

package/src/UI/SelectionFilter.js

@@ -1,11 +1,19 @@
-import { CADmaterials } from "./CADmaterials.js";
+import { SelectionState } from "./SelectionState.js";
 import {BREP} from '../BREP/BREP.js';
+
+const debugMode = false;
+
+
+
+
 export class SelectionFilter {
     static SOLID = "SOLID";
     static COMPONENT = "COMPONENT";
     static FACE = "FACE";
     static PLANE = "PLANE";
     static SKETCH = "SKETCH";
+    static DATUM = "DATUM";
+    static HELIX = "HELIX";
     static EDGE = "EDGE";
     static LOOP = "LOOP";
     static VERTEX = "VERTEX";
@@ -16,6 +24,7 @@ export class SelectionFilter {
     static viewer = null;
     static previouseAllowedSelectionTypes = null;
     static _hovered = new Set(); // objects currently hover-highlighted
+    static _hoveredSourceMap = new Map(); // key -> source object for hover
     static hoverColor = '#fbff00'; // default hover tint
     static _selectionActions = new Map();
     static _selectionActionOrder = [];
@@ -33,6 +42,10 @@ export class SelectionFilter {
     static _selectionFilterTypes = null;
     static _selectionFilterOutsideBound = false;
     static _selectionFilterTintBtn = null;
+    static _clickWatcherTimer = null;
+    static _missingClickLogged = new Map();
+    static _clickWatcherIntervalMs = 2000;
+    static _onClickWatcherSeq = 1;
     static _selectableTintState = {
         active: false,
         activeColor: null,
@@ -45,7 +58,7 @@ export class SelectionFilter {
         throw new Error("SelectionFilter is static and cannot be instantiated.");
     }
 
-    static get TYPES() { return [this.SOLID, this.COMPONENT, this.FACE, this.PLANE, this.SKETCH, this.EDGE, this.LOOP, this.VERTEX, this.ALL]; }
+    static get TYPES() { return [this.SOLID, this.COMPONENT, this.FACE, this.PLANE, this.SKETCH, this.DATUM, this.HELIX, this.EDGE, this.LOOP, this.VERTEX, this.ALL]; }
 
     // Convenience: return the list of selectable types for the dropdown (excludes ALL)
     static getAvailableTypes() {
@@ -61,6 +74,144 @@ export class SelectionFilter {
         return null;
     }
 
+    static _withSilentOnClick(target, fn) {
+        if (!target || typeof fn !== 'function') return;
+        try { target.__brepOnClickSilent = true; } catch { }
+        try { fn(); } catch { } finally {
+            try { target.__brepOnClickSilent = false; } catch { }
+        }
+    }
+
+    static _installOnClickWatcher(target) {
+        if (!target || typeof target !== 'object') return;
+        const existingDesc = Object.getOwnPropertyDescriptor(target, 'onClick');
+        if (existingDesc?.get && existingDesc?.get.__brepOnClickWatcher) return;
+        let current = typeof target.onClick !== 'undefined' ? target.onClick : undefined;
+        const getter = function () { return current; };
+        getter.__brepOnClickWatcher = true;
+        const setter = function (v) {
+            const prev = current;
+            current = v;
+            try {
+                target.__brepOnClickLastSetAt = Date.now();
+                target.__brepOnClickLastSetStack = new Error('[SelectionFilter] onClick set').stack;
+            } catch { }
+            const silent = !!target.__brepOnClickSilent;
+            const prevFn = typeof prev === 'function';
+            const nextFn = typeof v === 'function';
+            if (!silent && prev !== v) {
+                if (!nextFn || (prevFn && !nextFn)) {
+                    if (debugMode) {
+                        try {
+                            console.log('[SelectionFilter] onClick removed/overwritten', {
+                                name: target?.name,
+                                type: target?.type,
+                                uuid: target?.uuid,
+                                prev,
+                                next: v,
+                                target,
+                            });
+                            console.trace('[SelectionFilter] onClick change stack');
+                        } catch { }
+                    }
+                } else if (!v?.__brepSelectionHandler) {
+                    if (debugMode) {
+                        try {
+                            console.log('[SelectionFilter] onClick replaced', {
+                                name: target?.name,
+                                type: target?.type,
+                                uuid: target?.uuid,
+                                prev,
+                                next: v,
+                                target,
+                            });
+                            console.trace('[SelectionFilter] onClick set stack');
+                        } catch { }
+                    }
+                }
+            }
+        };
+        setter.__brepOnClickWatcher = true;
+        try {
+            Object.defineProperty(target, 'onClick', {
+                get: getter,
+                set: setter,
+                configurable: true,
+                enumerable: true,
+            });
+        } catch {
+            try { target.onClick = current; } catch { }
+        }
+    }
+
+    static startClickWatcher(viewer = null, { intervalMs = 2000 } = {}) {
+        const v = viewer || SelectionFilter.viewer;
+        SelectionFilter._clickWatcherIntervalMs = Math.max(250, Number(intervalMs) || 2000);
+        if (SelectionFilter._clickWatcherTimer) return;
+        const scan = () => {
+            try {
+                const scene = v?.partHistory?.scene || v?.scene || SelectionFilter.viewer?.partHistory?.scene || SelectionFilter.viewer?.scene || null;
+                if (!scene) return;
+                const selectionTypes = new Set(SelectionFilter.TYPES.filter(t => t && t !== SelectionFilter.ALL));
+                const missingNow = new Set();
+                const stack = Array.isArray(scene.children) ? [...scene.children] : [];
+                while (stack.length) {
+                    const current = stack.pop();
+                    if (!current) continue;
+                    const kids = Array.isArray(current?.children) ? current.children : [];
+                    for (const child of kids) stack.push(child);
+
+                    const type = String(current.type || '').toUpperCase();
+                    if (!selectionTypes.has(type)) continue;
+
+                    SelectionFilter._installOnClickWatcher(current);
+                    let hasClick = typeof current.onClick === 'function';
+                    if (!hasClick) {
+                        try {
+                            SelectionFilter.ensureSelectionHandlers(current, { deep: false });
+                        } catch { }
+                        hasClick = typeof current.onClick === 'function';
+                    }
+                    if (!hasClick) {
+                        missingNow.add(current.uuid);
+                        const last = SelectionFilter._missingClickLogged.get(current.uuid);
+                        if (!last) {
+                            SelectionFilter._missingClickLogged.set(current.uuid, Date.now());
+                            if (debugMode) {
+                                try {
+                                    console.log('[SelectionFilter] Missing onClick', {
+                                        name: current?.name,
+                                        type: current?.type,
+                                        uuid: current?.uuid,
+                                        parentName: current?.parent?.name,
+                                        parentType: current?.parent?.type,
+                                        lastSetAt: current?.__brepOnClickLastSetAt || null,
+                                        lastSetStack: current?.__brepOnClickLastSetStack || null,
+                                        object: current,
+                                    });
+                                } catch { }
+                            }
+                        }
+                    }
+                }
+                // Clear recovered entries
+                for (const key of SelectionFilter._missingClickLogged.keys()) {
+                    if (!missingNow.has(key)) SelectionFilter._missingClickLogged.delete(key);
+                }
+            } catch { }
+        };
+        try { scan(); } catch { }
+        SelectionFilter._clickWatcherTimer = setInterval(scan, SelectionFilter._clickWatcherIntervalMs);
+    }
+
+    static stopClickWatcher() {
+        if (SelectionFilter._clickWatcherTimer) {
+            clearInterval(SelectionFilter._clickWatcherTimer);
+            SelectionFilter._clickWatcherTimer = null;
+        }
+        SelectionFilter._missingClickLogged.clear();
+    }
+
     static setCurrentType(_type) {
         // No-op: current type is no longer tracked.
         void _type;
@@ -72,7 +223,6 @@ export class SelectionFilter {
         if (types === SelectionFilter.ALL) {
             SelectionFilter.allowedSelectionTypes = SelectionFilter.ALL;
             SelectionFilter.triggerUI();
-            SelectionFilter._ensureSceneSelectionHandlers();
             SelectionFilter.#logAllowedTypesChange(SelectionFilter.allowedSelectionTypes, 'SetSelectionTypes');
             return;
         }
@@ -81,7 +231,6 @@ export class SelectionFilter {
         if (invalid.length) throw new Error(`Unknown selection type(s): ${invalid.join(", ")}`);
         SelectionFilter.allowedSelectionTypes = new Set(list);
         SelectionFilter.triggerUI();
-        SelectionFilter._ensureSceneSelectionHandlers();
         SelectionFilter.#logAllowedTypesChange(SelectionFilter.allowedSelectionTypes, 'SetSelectionTypes');
     }
 
@@ -94,7 +243,6 @@ export class SelectionFilter {
             SelectionFilter.allowedSelectionTypes = SelectionFilter.previouseAllowedSelectionTypes;
             SelectionFilter.previouseAllowedSelectionTypes = null;
             SelectionFilter.triggerUI();
-            SelectionFilter._ensureSceneSelectionHandlers();
             SelectionFilter.#logAllowedTypesChange(SelectionFilter.allowedSelectionTypes, 'RestoreSelectionTypes');
         }
     }
@@ -104,19 +252,25 @@ export class SelectionFilter {
         let changed = false;
         const attach = (target) => {
             if (!target || typeof target !== 'object') return;
+            SelectionState.attach(target);
+            SelectionFilter._installOnClickWatcher(target);
             if (typeof target.onClick === 'function') return;
-            target.onClick = () => {
-                try {
-                    if (target.type === SelectionFilter.SOLID && target.parent && target.parent.type === SelectionFilter.COMPONENT) {
-                        const handledByParent = SelectionFilter.toggleSelection(target.parent);
-                        if (!handledByParent) SelectionFilter.toggleSelection(target);
-                        return;
+            SelectionFilter._withSilentOnClick(target, () => {
+                target.onClick = () => {
+                    try {
+                        if (target.type === SelectionFilter.SOLID && target.parent && target.parent.type === SelectionFilter.COMPONENT) {
+                            const handledByParent = SelectionFilter.toggleSelection(target.parent);
+                            if (!handledByParent) SelectionFilter.toggleSelection(target);
+                            return;
+                        }
+                        SelectionFilter.toggleSelection(target);
+                    } catch (error) {
+                        if (debugMode) {
+                            try { console.warn('[SelectionFilter] toggleSelection failed:', error); } catch (_) { /* ignore */ }
+                        }
                     }
-                    SelectionFilter.toggleSelection(target);
-                } catch (error) {
-                    try { console.warn('[SelectionFilter] toggleSelection failed:', error); } catch (_) { /* ignore */ }
-                }
-            };
+                };
+            });
             try { target.onClick.__brepSelectionHandler = true; } catch (_) { /* ignore */ }
             changed = true;
         };
@@ -138,21 +292,6 @@ export class SelectionFilter {
         return changed;
     }
 
-    static _ensureSceneSelectionHandlers() {
-        try {
-            const scene = SelectionFilter.viewer?.partHistory?.scene
-                || SelectionFilter.viewer?.scene
-                || null;
-            if (!scene || !Array.isArray(scene.children)) return;
-            for (const child of scene.children) {
-                if (!child || child.type !== SelectionFilter.SOLID) continue;
-                SelectionFilter.ensureSelectionHandlers(child, { deep: true });
-            }
-        } catch { }
-    }
-
-
-
     static allowType(type) {
         // Legacy support: expand available set; does not change currentType
         if (type === SelectionFilter.ALL) { SelectionFilter.allowedSelectionTypes = SelectionFilter.ALL; SelectionFilter.triggerUI(); return; }
@@ -196,37 +335,67 @@ export class SelectionFilter {
         SelectionFilter.#logAllowedTypesChange(SelectionFilter.allowedSelectionTypes, 'Reset');
     }
 
-    static #getBaseMaterial(obj) {
-        if (!obj) return null;
-        const ud = obj.userData || {};
-        if (ud.__baseMaterial) return ud.__baseMaterial;
-        if (obj.type === SelectionFilter.FACE) return CADmaterials.FACE?.BASE ?? CADmaterials.FACE ?? obj.material;
-        if (obj.type === SelectionFilter.PLANE) return CADmaterials.PLANE?.BASE ?? CADmaterials.FACE?.BASE ?? obj.material;
-        if (obj.type === SelectionFilter.EDGE) return CADmaterials.EDGE?.BASE ?? CADmaterials.EDGE ?? obj.material;
-        if (obj.type === SelectionFilter.SOLID || obj.type === SelectionFilter.COMPONENT) return obj.material;
-        return obj.material;
-    }
-
     // ---------------- Hover Highlighting ----------------
-    static getHoverColor() { return SelectionFilter.hoverColor; }
+    static getHoverColor() { return SelectionState.hoverColor || SelectionFilter.hoverColor; }
     static setHoverColor(hex) {
         if (!hex) return;
         try { SelectionFilter.hoverColor = String(hex); } catch (_) { }
+        SelectionState.setHoverColor(SelectionFilter.hoverColor);
         // Update current hovered objects live
         for (const o of Array.from(SelectionFilter._hovered)) {
-            if (o && o.material && o.material.color && typeof o.material.color.set === 'function') {
-                try { o.material.color.set(SelectionFilter.hoverColor); } catch (_) { }
-            }
+            if (!o) continue;
+            try {
+                SelectionState.attach(o);
+                o.hovered = false;
+                o.hovered = true;
+            } catch { }
         }
     }
 
     static setHoverObject(obj, options = {}) {
-        const { ignoreFilter = false } = options;
-        // Clear existing hover first
-        SelectionFilter.clearHover();
-        if (!obj) return;
-        // Highlight depending on type
-        SelectionFilter.#applyHover(obj, { ignoreFilter });
+        SelectionFilter.setHoverObjects(obj ? [obj] : [], options);
+    }
+
+    static setHoverObjects(objs, options = {}) {
+        const { ignoreFilter = false, append = false } = options;
+        const prevKeys = new Set(SelectionFilter._hoveredSourceMap.keys());
+        if (!append) {
+            SelectionFilter._clearHoverState({ emit: false });
+        }
+        if (!objs) return;
+        const list = Array.isArray(objs) ? objs : [objs];
+        const seen = new Set();
+        const keyFor = (obj) => obj?.uuid || obj?.id || obj?.name || obj;
+        for (const obj of list) {
+            if (!obj) continue;
+            const key = keyFor(obj);
+            if (seen.has(key)) continue;
+            seen.add(key);
+            const allowed = ignoreFilter || SelectionFilter.IsAllowed(obj.type);
+            if (!allowed) continue;
+            if (key && !SelectionFilter._hoveredSourceMap.has(key)) {
+                SelectionFilter._hoveredSourceMap.set(key, obj);
+            }
+            const targets = SelectionState.getHoverTargets(obj);
+            for (const t of targets) {
+                if (!t) continue;
+                try {
+                    SelectionState.attach(t);
+                    t.hovered = true;
+                    SelectionFilter._hovered.add(t);
+                } catch { }
+            }
+        }
+        const nextKeys = new Set(SelectionFilter._hoveredSourceMap.keys());
+        let changed = prevKeys.size !== nextKeys.size;
+        if (!changed) {
+            for (const k of nextKeys) {
+                if (!prevKeys.has(k)) { changed = true; break; }
+            }
+        }
+        if (changed) {
+            SelectionFilter._emitHoverChanged(Array.from(SelectionFilter._hoveredSourceMap.values()));
+        }
     }
 
     static setHoverByName(scene, name) {
@@ -237,157 +406,38 @@ export class SelectionFilter {
     }
 
     static clearHover() {
-        if (!SelectionFilter._hovered || SelectionFilter._hovered.size === 0) return;
-        for (const o of Array.from(SelectionFilter._hovered)) {
-            SelectionFilter.#restoreHover(o);
-        }
-        SelectionFilter._hovered.clear();
+        SelectionFilter._clearHoverState({ emit: true });
     }
 
-    static #applyHover(obj, options = {}) {
-        const { ignoreFilter = false } = options;
-        if (!obj) return;
-        // Respect selection filter: skip if disallowed
-        const allowed = ignoreFilter
-            || SelectionFilter.IsAllowed(obj.type)
-            || SelectionFilter.matchesAllowedType(obj.type);
-        if (!allowed) return;
-
-        // Only ever highlight one object: the exact object provided, if it has a color
-        const target = obj;
-        if (!target) return;
-
-        const applyToOne = (t) => {
-            if (!t) return;
-            if (!t.userData) t.userData = {};
-            const origMat = t.material;
-            if (!origMat) return;
-            if (t.userData.__hoverMatApplied) { SelectionFilter._hovered.add(t); return; }
-            let clone;
-            try { clone = typeof origMat.clone === 'function' ? origMat.clone() : origMat; } catch { clone = origMat; }
-            try { if (clone && clone.color && typeof clone.color.set === 'function') clone.color.set(SelectionFilter.hoverColor); } catch { }
-            try {
-                if (origMat && clone && origMat.resolution && clone.resolution && typeof clone.resolution.copy === 'function') {
-                    clone.resolution.copy(origMat.resolution);
-                }
-            } catch { }
-            try {
-                if (origMat && clone && typeof origMat.dashed !== 'undefined' && typeof clone.dashed !== 'undefined') {
-                    clone.dashed = origMat.dashed;
-                }
-                if (origMat && clone && typeof origMat.dashSize !== 'undefined' && typeof clone.dashSize !== 'undefined') {
-                    clone.dashSize = origMat.dashSize;
-                }
-                if (origMat && clone && typeof origMat.gapSize !== 'undefined' && typeof clone.gapSize !== 'undefined') {
-                    clone.gapSize = origMat.gapSize;
-                }
-                if (origMat && clone && typeof origMat.dashScale !== 'undefined' && typeof clone.dashScale !== 'undefined') {
-                    clone.dashScale = origMat.dashScale;
-                }
-            } catch { }
-            try {
-                t.userData.__hoverOrigMat = origMat;
-                t.userData.__hoverMatApplied = true;
-                if (clone !== origMat) t.material = clone;
-                t.userData.__hoverMat = clone;
-            } catch { }
-            SelectionFilter._hovered.add(t);
-        };
-
-        if (target.type === SelectionFilter.SOLID || target.type === SelectionFilter.COMPONENT) {
-            // Highlight all immediate child faces/edges for SOLID or COMPONENT children
-            if (Array.isArray(target.children)) {
-                for (const ch of target.children) {
-                    if (!ch) continue;
-                    if (ch.type === SelectionFilter.SOLID || ch.type === SelectionFilter.COMPONENT) {
-                        if (Array.isArray(ch.children)) {
-                            for (const nested of ch.children) {
-                                if (nested && (nested.type === SelectionFilter.FACE || nested.type === SelectionFilter.EDGE)) applyToOne(nested);
-                            }
-                        }
-                    } else if (ch.type === SelectionFilter.FACE || ch.type === SelectionFilter.EDGE) {
-                        applyToOne(ch);
-                    }
-                }
-            }
-            // Track the solid as a logical hovered root to clear later
-            SelectionFilter._hovered.add(target);
-            return;
-        }
-
-        if (target.type === SelectionFilter.VERTEX) {
-            // Apply to the vertex object and any drawable children (e.g., Points)
-            applyToOne(target);
-            if (Array.isArray(target.children)) {
-                for (const ch of target.children) {
-                    applyToOne(ch);
-                }
+    static _emitHoverChanged(objs = []) {
+        try {
+            const list = Array.isArray(objs) ? objs : [];
+            const uuids = [];
+            for (const obj of list) {
+                if (obj && obj.uuid) uuids.push(obj.uuid);
             }
-            return;
-        }
-
-        applyToOne(target);
+            const ev = new CustomEvent('hover-changed', { detail: { objects: list, uuids } });
+            window.dispatchEvent(ev);
+        } catch { /* ignore */ }
     }
 
-    static #restoreHover(obj) {
-        if (!obj) return;
-        const restoreOne = (t) => {
-            if (!t) return;
-            const ud = t.userData || {};
-            if (!ud.__hoverMatApplied) return;
-            const cloneWithColor = (mat, colorHex) => {
-                if (!mat) return mat;
-                let c = mat;
+    static _clearHoverState({ emit = true } = {}) {
+        const hadHover = (SelectionFilter._hovered && SelectionFilter._hovered.size) || SelectionFilter._hoveredSourceMap.size;
+        if (SelectionFilter._hovered && SelectionFilter._hovered.size) {
+            for (const o of Array.from(SelectionFilter._hovered)) {
                 try {
-                    c = typeof mat.clone === 'function' ? mat.clone() : mat;
-                    if (c && c.color && typeof c.color.set === 'function' && colorHex) c.color.set(colorHex);
-                } catch { c = mat; }
-                return c;
-            };
-            const applySelectionMaterial = () => {
-                if (t.type === SelectionFilter.FACE) return CADmaterials.FACE?.SELECTED ?? CADmaterials.FACE ?? ud.__hoverOrigMat ?? t.material;
-                if (t.type === SelectionFilter.PLANE) return CADmaterials.PLANE?.SELECTED ?? CADmaterials.FACE?.SELECTED ?? ud.__hoverOrigMat ?? t.material;
-                if (t.type === SelectionFilter.EDGE) {
-                    const base = ud.__hoverOrigMat ?? t.material;
-                    const selColor = CADmaterials.EDGE?.SELECTED?.color || CADmaterials.EDGE?.SELECTED?.color?.getHexString?.();
-                    return cloneWithColor(base, selColor || '#ff00ff');
-                }
-                if (t.type === SelectionFilter.SOLID || t.type === SelectionFilter.COMPONENT) return CADmaterials.SOLID?.SELECTED ?? ud.__hoverOrigMat ?? t.material;
-                return ud.__hoverOrigMat ?? t.material;
-            };
-            const applyDeselectedMaterial = () => {
-                if (t.type === SelectionFilter.FACE) return ud.__hoverOrigMat ?? SelectionFilter.#getBaseMaterial(t) ?? t.material;
-                if (t.type === SelectionFilter.PLANE) return ud.__hoverOrigMat ?? SelectionFilter.#getBaseMaterial(t) ?? t.material;
-                if (t.type === SelectionFilter.EDGE) return ud.__hoverOrigMat ?? SelectionFilter.#getBaseMaterial(t) ?? t.material;
-                if (t.type === SelectionFilter.SOLID || t.type === SelectionFilter.COMPONENT) return ud.__hoverOrigMat ?? t.material;
-                return ud.__hoverOrigMat ?? t.material;
-            };
-            try {
-                if (t.selected) {
-                    const selMat = applySelectionMaterial();
-                    if (selMat) t.material = selMat;
-                } else {
-                    const baseMat = applyDeselectedMaterial();
-                    if (baseMat) t.material = baseMat;
-                }
-                if (ud.__hoverMat && ud.__hoverMat !== ud.__hoverOrigMat && typeof ud.__hoverMat.dispose === 'function') ud.__hoverMat.dispose();
-            } catch { }
-            try { delete t.userData.__hoverMatApplied; } catch { }
-            try { delete t.userData.__hoverOrigMat; } catch { }
-            try { delete t.userData.__hoverMat; } catch { }
-        };
-
-        if (obj.type === SelectionFilter.SOLID || obj.type === SelectionFilter.COMPONENT) {
-            if (Array.isArray(obj.children)) {
-                for (const ch of obj.children) restoreOne(ch);
+                    SelectionState.attach(o);
+                    o.hovered = false;
+                } catch { }
             }
+            SelectionFilter._hovered.clear();
         }
-        if (obj.type === SelectionFilter.VERTEX) {
-            if (Array.isArray(obj.children)) {
-                for (const ch of obj.children) restoreOne(ch);
-            }
+        if (SelectionFilter._hoveredSourceMap.size) {
+            SelectionFilter._hoveredSourceMap.clear();
+        }
+        if (emit && hadHover) {
+            SelectionFilter._emitHoverChanged([]);
         }
-        restoreOne(obj);
     }
 
     static #isInputUsable(el) {
@@ -415,246 +465,208 @@ export class SelectionFilter {
         return true;
     }
 
-    static toggleSelection(objectToToggleSelectionOn) {
-        // get the type of the object
-        const type = objectToToggleSelectionOn.type;
-        if (!type) throw new Error("Object to toggle selection on must have a type.");
-
-        // Check if a reference selection is active
+    static #handleReferenceSelection(objectToToggleSelectionOn) {
         try {
             let activeRefInput = document.querySelector('[active-reference-selection="true"],[active-reference-selection=true]');
             if (!activeRefInput) {
                 try { activeRefInput = window.__BREP_activeRefInput || null; } catch (_) { /* ignore */ }
             }
-            if (activeRefInput) {
-                const usable = SelectionFilter.#isInputUsable(activeRefInput);
-                if (!usable) {
-                    try { activeRefInput.removeAttribute('active-reference-selection'); } catch (_) { }
-                    try { activeRefInput.style.filter = 'none'; } catch (_) { }
-                    try { if (window.__BREP_activeRefInput === activeRefInput) window.__BREP_activeRefInput = null; } catch (_) { }
-                    SelectionFilter.restoreAllowedSelectionTypes();
-                    return false;
-                }
+            if (!activeRefInput) return false;
+
+            const usable = SelectionFilter.#isInputUsable(activeRefInput);
+            if (!usable) {
+                try { activeRefInput.removeAttribute('active-reference-selection'); } catch (_) { }
+                try { activeRefInput.style.filter = 'none'; } catch (_) { }
+                try { if (window.__BREP_activeRefInput === activeRefInput) window.__BREP_activeRefInput = null; } catch (_) { }
+                SelectionFilter.restoreAllowedSelectionTypes();
+                return false;
+            }
 
-                const dataset = activeRefInput.dataset || {};
-                const isMultiRef = dataset.multiple === 'true';
-                const maxSelections = Number(dataset.maxSelections);
-                const hasMax = Number.isFinite(maxSelections) && maxSelections > 0;
-                if (isMultiRef) {
-                    let currentCount = 0;
+            const dataset = activeRefInput.dataset || {};
+            const isMultiRef = dataset.multiple === 'true';
+            const maxSelections = Number(dataset.maxSelections);
+            const hasMax = Number.isFinite(maxSelections) && maxSelections > 0;
+            if (isMultiRef) {
+                let currentCount = 0;
+                try {
+                    if (typeof activeRefInput.__getSelectionList === 'function') {
+                        const list = activeRefInput.__getSelectionList();
+                        if (Array.isArray(list)) currentCount = list.length;
+                    } else if (dataset.selectedCount !== undefined) {
+                        const parsed = Number(dataset.selectedCount);
+                        if (Number.isFinite(parsed)) currentCount = parsed;
+                    }
+                } catch (_) { /* ignore */ }
+                if (hasMax && currentCount >= maxSelections) {
                     try {
-                        if (typeof activeRefInput.__getSelectionList === 'function') {
-                            const list = activeRefInput.__getSelectionList();
-                            if (Array.isArray(list)) currentCount = list.length;
-                        } else if (dataset.selectedCount !== undefined) {
-                            const parsed = Number(dataset.selectedCount);
-                            if (Number.isFinite(parsed)) currentCount = parsed;
+                        const wrap = activeRefInput.closest('.ref-single-wrap, .ref-multi-wrap');
+                        if (wrap) {
+                            wrap.classList.add('ref-limit-reached');
+                            setTimeout(() => {
+                                try { wrap.classList.remove('ref-limit-reached'); } catch (_) { }
+                            }, 480);
                         }
-                    } catch (_) { /* ignore */ }
-                    if (hasMax && currentCount >= maxSelections) {
-                        try {
-                            const wrap = activeRefInput.closest('.ref-single-wrap, .ref-multi-wrap');
-                            if (wrap) {
-                                wrap.classList.add('ref-limit-reached');
-                                setTimeout(() => {
-                                    try { wrap.classList.remove('ref-limit-reached'); } catch (_) { }
-                                }, 480);
-                            }
-                        } catch (_) { }
-                        return true;
-                    }
+                    } catch (_) { }
+                    return true;
                 }
-                const allowed = SelectionFilter.allowedSelectionTypes;
-                const allowAll = allowed === SelectionFilter.ALL;
-                const priorityOrder = [
-                    SelectionFilter.VERTEX,
-                    SelectionFilter.EDGE,
-                    SelectionFilter.FACE,
-                    SelectionFilter.PLANE,
-                    SelectionFilter.SOLID,
-                    SelectionFilter.COMPONENT,
-                ];
-                const allowedHas = (t) => !!(allowed && typeof allowed.has === 'function' && allowed.has(t));
-                const allowedPriority = allowAll ? priorityOrder : priorityOrder.filter(t => allowedHas(t));
-
-                // Helper: find first descendant matching a type
-                const findDescendantOfType = (root, desired) => {
-                    if (!root || !desired) return null;
-                    let found = null;
-                    try {
-                        root.traverse?.((ch) => {
-                            if (!found && ch && ch.type === desired) found = ch;
-                        });
-                    } catch (_) { /* ignore */ }
-                    return found;
-                };
+            }
 
-                const findAncestorOfType = (obj, desired) => {
-                    let cur = obj;
-                    while (cur && cur.parent) {
-                        if (cur.type === desired) return cur;
-                        cur = cur.parent;
-                    }
-                    return null;
-                };
-                const pickByTypeList = (typeList) => {
-                    if (!Array.isArray(typeList)) return null;
-                    for (const desired of typeList) {
-                        if (!desired) continue;
-                        if (objectToToggleSelectionOn?.type === desired) return objectToToggleSelectionOn;
-                        let picked = findDescendantOfType(objectToToggleSelectionOn, desired);
-                        if (!picked) picked = findAncestorOfType(objectToToggleSelectionOn, desired);
-                        if (picked) return picked;
-                    }
-                    return null;
-                };
+            const allowed = SelectionFilter.allowedSelectionTypes;
+            const allowAll = allowed === SelectionFilter.ALL;
+            const priorityOrder = [
+                SelectionFilter.VERTEX,
+                SelectionFilter.EDGE,
+                SelectionFilter.FACE,
+                SelectionFilter.PLANE,
+                SelectionFilter.SKETCH,
+                SelectionFilter.LOOP,
+                SelectionFilter.SOLID,
+                SelectionFilter.COMPONENT,
+            ];
+            const allowedHas = (t) => !!(allowed && typeof allowed.has === 'function' && allowed.has(t));
+            const allowedPriority = allowAll ? priorityOrder : priorityOrder.filter(t => allowedHas(t));
+
+            const findDescendantOfType = (root, desired) => {
+                if (!root || !desired) return null;
+                let found = null;
+                try {
+                    root.traverse?.((ch) => {
+                        if (!found && ch && ch.type === desired) found = ch;
+                    });
+                } catch (_) { /* ignore */ }
+                return found;
+            };
 
-                let targetObj = null;
-                // Prefer the exact object the user clicked if it is allowed.
-                if (allowAll || allowedHas(objectToToggleSelectionOn?.type)) {
-                    targetObj = objectToToggleSelectionOn;
-                }
-                if (!targetObj) {
-                    targetObj = pickByTypeList(allowedPriority);
+            const findAncestorOfType = (obj, desired) => {
+                let cur = obj;
+                while (cur && cur.parent) {
+                    if (cur.type === desired) return cur;
+                    cur = cur.parent;
                 }
-                if (!targetObj && !allowAll && allowed && typeof allowed[Symbol.iterator] === 'function') {
-                    targetObj = pickByTypeList(Array.from(allowed));
-                }
-                if (!targetObj && allowAll) {
-                    targetObj = objectToToggleSelectionOn;
+                return null;
+            };
+
+            const pickByTypeList = (typeList) => {
+                if (!Array.isArray(typeList)) return null;
+                for (const desired of typeList) {
+                    if (!desired) continue;
+                    if (objectToToggleSelectionOn?.type === desired) return objectToToggleSelectionOn;
+                    let picked = findDescendantOfType(objectToToggleSelectionOn, desired);
+                    if (!picked) picked = findAncestorOfType(objectToToggleSelectionOn, desired);
+                    if (picked) return picked;
                 }
-                if (!targetObj) return false;
+                return null;
+            };
 
-                // Update the reference input with the chosen object
-                try {
-                    if (activeRefInput && typeof activeRefInput.__captureReferencePreview === 'function') {
-                        activeRefInput.__captureReferencePreview(targetObj);
-                    }
-                } catch (_) { /* ignore preview capture errors */ }
-                const objType = targetObj.type;
-                const objectName = targetObj.name || `${objType}(${targetObj.position?.x || 0},${targetObj.position?.y || 0},${targetObj.position?.z || 0})`;
-
-                const snapshotSelections = (inputEl) => {
-                    const data = inputEl?.dataset || {};
-                    let values = null;
-                    if (data.selectedValues) {
-                        try {
-                            const parsed = JSON.parse(data.selectedValues);
-                            if (Array.isArray(parsed)) values = parsed;
-                        } catch (_) { /* ignore */ }
-                    }
-                    if (!Array.isArray(values) && typeof inputEl?.__getSelectionList === 'function') {
-                        try {
-                            const list = inputEl.__getSelectionList();
-                            if (Array.isArray(list)) values = list.slice();
-                        } catch (_) { /* ignore */ }
-                    }
-                    let count = 0;
-                    if (Array.isArray(values)) {
-                        count = values.length;
-                    } else if (data.selectedCount !== undefined) {
-                        const parsed = Number(data.selectedCount);
-                        if (Number.isFinite(parsed) && parsed >= 0) count = parsed;
-                    }
-                    return count;
-                };
+            let targetObj = null;
+            if (allowAll || allowedHas(objectToToggleSelectionOn?.type)) {
+                targetObj = objectToToggleSelectionOn;
+            }
+            if (!targetObj) {
+                targetObj = pickByTypeList(allowedPriority);
+            }
+            if (!targetObj && !allowAll && allowed && typeof allowed[Symbol.iterator] === 'function') {
+                targetObj = pickByTypeList(Array.from(allowed));
+            }
+            if (!targetObj && allowAll) {
+                targetObj = objectToToggleSelectionOn;
+            }
+            if (!targetObj) return false;
 
-                activeRefInput.value = objectName;
-                activeRefInput.dispatchEvent(new Event('change'));
-                const afterSelectionCount = snapshotSelections(activeRefInput);
+            try {
+                if (activeRefInput && typeof activeRefInput.__captureReferencePreview === 'function') {
+                    activeRefInput.__captureReferencePreview(targetObj);
+                }
+            } catch (_) { /* ignore preview capture errors */ }
 
-                const didReachLimit = isMultiRef && hasMax && afterSelectionCount >= maxSelections;
-                const keepActive = isMultiRef && !didReachLimit;
+            const objType = targetObj.type;
+            const objectName = targetObj.name || `${objType}(${targetObj.position?.x || 0},${targetObj.position?.y || 0},${targetObj.position?.z || 0})`;
 
-                if (!keepActive) {
-                    // Clean up the reference selection state
-                    activeRefInput.removeAttribute('active-reference-selection');
-                    activeRefInput.style.filter = 'none';
+            const snapshotSelections = (inputEl) => {
+                const data = inputEl?.dataset || {};
+                let values = null;
+                if (data.selectedValues) {
                     try {
-                        const wrap = activeRefInput.closest('.ref-single-wrap, .ref-multi-wrap');
-                        if (wrap) wrap.classList.remove('ref-active');
-                    } catch (_) { }
-                    // Restore selection filter
-                    SelectionFilter.restoreAllowedSelectionTypes();
-                    try { if (window.__BREP_activeRefInput === activeRefInput) window.__BREP_activeRefInput = null; } catch (_) { }
-                } else {
-                    activeRefInput.setAttribute('active-reference-selection', 'true');
-                    activeRefInput.style.filter = 'invert(1)';
+                        const parsed = JSON.parse(data.selectedValues);
+                        if (Array.isArray(parsed)) values = parsed;
+                    } catch (_) { /* ignore */ }
+                }
+                if (!Array.isArray(values) && typeof inputEl?.__getSelectionList === 'function') {
                     try {
-                        const wrap = activeRefInput.closest('.ref-single-wrap, .ref-multi-wrap');
-                        if (wrap) wrap.classList.add('ref-active');
-                    } catch (_) { }
-                    try { window.__BREP_activeRefInput = activeRefInput; } catch (_) { }
+                        const list = inputEl.__getSelectionList();
+                        if (Array.isArray(list)) values = list.slice();
+                    } catch (_) { /* ignore */ }
+                }
+                let count = 0;
+                if (Array.isArray(values)) {
+                    count = values.length;
+                } else if (data.selectedCount !== undefined) {
+                    const parsed = Number(data.selectedCount);
+                    if (Number.isFinite(parsed) && parsed >= 0) count = parsed;
                 }
-                return true; // handled as a reference selection
+                return count;
+            };
+
+            activeRefInput.value = objectName;
+            activeRefInput.dispatchEvent(new Event('change'));
+            const afterSelectionCount = snapshotSelections(activeRefInput);
+
+            const didReachLimit = isMultiRef && hasMax && afterSelectionCount >= maxSelections;
+            const keepActive = isMultiRef && !didReachLimit;
+
+            if (!keepActive) {
+                activeRefInput.removeAttribute('active-reference-selection');
+                activeRefInput.style.filter = 'none';
+                try {
+                    const wrap = activeRefInput.closest('.ref-single-wrap, .ref-multi-wrap');
+                    if (wrap) wrap.classList.remove('ref-active');
+                } catch (_) { }
+                SelectionFilter.restoreAllowedSelectionTypes();
+                try { if (window.__BREP_activeRefInput === activeRefInput) window.__BREP_activeRefInput = null; } catch (_) { }
+            } else {
+                activeRefInput.setAttribute('active-reference-selection', 'true');
+                activeRefInput.style.filter = 'invert(1)';
+                try {
+                    const wrap = activeRefInput.closest('.ref-single-wrap, .ref-multi-wrap');
+                    if (wrap) wrap.classList.add('ref-active');
+                } catch (_) { }
+                try { window.__BREP_activeRefInput = activeRefInput; } catch (_) { }
             }
+            return true;
         } catch (error) {
-            console.warn("Error handling reference selection:", error);
+            if (debugMode) {
+                console.warn("Error handling reference selection:", error);
+            }
+            return false;
         }
+    }
 
+    static #toggleStandardSelection(objectToToggleSelectionOn) {
+        const type = objectToToggleSelectionOn.type;
         let parentSelectedAction = false;
-        // check if the object is selectable and if it is toggle the .selected atribute on the object. 
-        // Allow toggling off even if type is currently disallowed; only block new selections
         if (SelectionFilter.IsAllowed(type) || objectToToggleSelectionOn.selected === true) {
+            SelectionState.attach(objectToToggleSelectionOn);
             objectToToggleSelectionOn.selected = !objectToToggleSelectionOn.selected;
-            // change the material on the object to indicate it is selected or not.
-            if (objectToToggleSelectionOn.selected) {
-                if (objectToToggleSelectionOn.type === SelectionFilter.FACE) {
-                    objectToToggleSelectionOn.material = CADmaterials.FACE?.SELECTED ?? CADmaterials.FACE;
-                } else if (objectToToggleSelectionOn.type === SelectionFilter.PLANE) {
-                    objectToToggleSelectionOn.material = CADmaterials.PLANE?.SELECTED ?? CADmaterials.FACE?.SELECTED ?? objectToToggleSelectionOn.material;
-                } else if (objectToToggleSelectionOn.type === SelectionFilter.EDGE) {
-                    objectToToggleSelectionOn.material = CADmaterials.EDGE?.SELECTED ?? CADmaterials.EDGE;
-                } else if (objectToToggleSelectionOn.type === SelectionFilter.VERTEX) {
-                    // Vertex visuals are handled by its selected accessor (point + sphere)
-                } else if (objectToToggleSelectionOn.type === SelectionFilter.SOLID || objectToToggleSelectionOn.type === SelectionFilter.COMPONENT) {
-                    parentSelectedAction = true;
-                    objectToToggleSelectionOn.children.forEach(child => {
-                        // apply selected material based on object type for faces and edges
-                        if (child.type === SelectionFilter.FACE) child.material = CADmaterials.FACE?.SELECTED ?? CADmaterials.FACE;
-                        if (child.type === SelectionFilter.PLANE) child.material = CADmaterials.PLANE?.SELECTED ?? CADmaterials.FACE?.SELECTED ?? child.material;
-                        if (child.type === SelectionFilter.EDGE) child.material = CADmaterials.EDGE?.SELECTED ?? CADmaterials.EDGE;
-                    });
-                }
-
-            } else {
-                if (objectToToggleSelectionOn.type === SelectionFilter.FACE) {
-                    objectToToggleSelectionOn.material = SelectionFilter.#getBaseMaterial(objectToToggleSelectionOn) ?? objectToToggleSelectionOn.material;
-                } else if (objectToToggleSelectionOn.type === SelectionFilter.PLANE) {
-                    objectToToggleSelectionOn.material = SelectionFilter.#getBaseMaterial(objectToToggleSelectionOn) ?? objectToToggleSelectionOn.material;
-                } else if (objectToToggleSelectionOn.type === SelectionFilter.EDGE) {
-                    objectToToggleSelectionOn.material = SelectionFilter.#getBaseMaterial(objectToToggleSelectionOn) ?? objectToToggleSelectionOn.material;
-                } else if (objectToToggleSelectionOn.type === SelectionFilter.VERTEX) {
-                    // Vertex accessor handles its own visual reset
-                } else if (objectToToggleSelectionOn.type === SelectionFilter.SOLID || objectToToggleSelectionOn.type === SelectionFilter.COMPONENT) {
-                    parentSelectedAction = true;
-                    objectToToggleSelectionOn.children.forEach(child => {
-                        // apply selected material based on object type for faces and edges
-                        if (child.type === SelectionFilter.FACE) child.material = child.selected ? CADmaterials.FACE?.SELECTED ?? CADmaterials.FACE : (SelectionFilter.#getBaseMaterial(child) ?? child.material);
-                        if (child.type === SelectionFilter.PLANE) child.material = child.selected ? (CADmaterials.PLANE?.SELECTED ?? CADmaterials.FACE?.SELECTED ?? child.material) : (SelectionFilter.#getBaseMaterial(child) ?? child.material);
-                        if (child.type === SelectionFilter.EDGE) child.material = child.selected ? CADmaterials.EDGE?.SELECTED ?? CADmaterials.EDGE : (SelectionFilter.#getBaseMaterial(child) ?? child.material);
-                    });
-                }
+            if (type === SelectionFilter.SOLID || type === SelectionFilter.COMPONENT) {
+                parentSelectedAction = true;
             }
             SelectionFilter._emitSelectionChanged();
         }
-
         return parentSelectedAction;
     }
 
+    static toggleSelection(objectToToggleSelectionOn) {
+        const type = objectToToggleSelectionOn.type;
+        if (!type) throw new Error("Object to toggle selection on must have a type.");
+        if (SelectionFilter.#handleReferenceSelection(objectToToggleSelectionOn)) return true;
+        return SelectionFilter.#toggleStandardSelection(objectToToggleSelectionOn);
+    }
+
     static unselectAll(scene) {
         // itterate over all children and nested children of the scene and set the .selected atribute to false. 
         scene.traverse((child) => {
+            SelectionState.attach(child);
             child.selected = false;
-            // reset material to base
-            if (child.type === SelectionFilter.FACE) {
-                child.material = SelectionFilter.#getBaseMaterial(child) ?? child.material;
-            } else if (child.type === SelectionFilter.PLANE) {
-                child.material = SelectionFilter.#getBaseMaterial(child) ?? child.material;
-            } else if (child.type === SelectionFilter.EDGE) {
-                child.material = SelectionFilter.#getBaseMaterial(child) ?? child.material;
-            }
-
         });
         SelectionFilter._emitSelectionChanged();
     }
@@ -662,17 +674,8 @@ export class SelectionFilter {
     static selectItem(scene, itemName) {
         scene.traverse((child) => {
             if (child && child.name === itemName) {
+                SelectionState.attach(child);
                 child.selected = true;
-                // change material to selected
-                if (child.type === SelectionFilter.FACE) {
-                    child.material = CADmaterials.FACE?.SELECTED ?? CADmaterials.FACE;
-                } else if (child.type === SelectionFilter.PLANE) {
-                    child.material = CADmaterials.PLANE?.SELECTED ?? CADmaterials.FACE?.SELECTED ?? child.material;
-                } else if (child.type === SelectionFilter.EDGE) {
-                    child.material = CADmaterials.EDGE?.SELECTED ?? CADmaterials.EDGE;
-                } else if (child.type === SelectionFilter.SOLID || child.type === SelectionFilter.COMPONENT) {
-                    child.material = CADmaterials.SOLID?.SELECTED ?? CADmaterials.SOLID;
-                }
             }
         });
         SelectionFilter._emitSelectionChanged();
@@ -682,27 +685,8 @@ export class SelectionFilter {
         // Traverse scene and deselect a single item by name, updating materials appropriately
         scene.traverse((child) => {
             if (child.name === itemName) {
+                SelectionState.attach(child);
                 child.selected = false;
-                if (child.type === SelectionFilter.FACE) {
-                    child.material = SelectionFilter.#getBaseMaterial(child) ?? child.material;
-                } else if (child.type === SelectionFilter.PLANE) {
-                    child.material = SelectionFilter.#getBaseMaterial(child) ?? child.material;
-                } else if (child.type === SelectionFilter.EDGE) {
-                    child.material = SelectionFilter.#getBaseMaterial(child) ?? child.material;
-                } else if (child.type === SelectionFilter.SOLID || child.type === SelectionFilter.COMPONENT) {
-                    // For solids, keep children materials consistent with their own selected flags
-                    child.children.forEach(grandchild => {
-                        if (grandchild.type === SelectionFilter.FACE) {
-                            grandchild.material = grandchild.selected ? (CADmaterials.FACE?.SELECTED ?? CADmaterials.FACE) : (SelectionFilter.#getBaseMaterial(grandchild) ?? grandchild.material);
-                        }
-                        if (grandchild.type === SelectionFilter.PLANE) {
-                            grandchild.material = grandchild.selected ? (CADmaterials.PLANE?.SELECTED ?? CADmaterials.FACE?.SELECTED ?? grandchild.material) : (SelectionFilter.#getBaseMaterial(grandchild) ?? grandchild.material);
-                        }
-                        if (grandchild.type === SelectionFilter.EDGE) {
-                            grandchild.material = grandchild.selected ? (CADmaterials.EDGE?.SELECTED ?? CADmaterials.EDGE) : (SelectionFilter.#getBaseMaterial(grandchild) ?? grandchild.material);
-                        }
-                    });
-                }
             }
         });
         SelectionFilter._emitSelectionChanged();
@@ -716,6 +700,21 @@ export class SelectionFilter {
 
     // Emit a global event so UI can react without polling
     static _emitSelectionChanged() {
+        try {
+            const selection = SelectionFilter.getSelectedObjects();
+            const names = selection.map((obj) => (
+                obj?.name
+                || obj?.userData?.faceName
+                || obj?.userData?.edgeName
+                || obj?.userData?.vertexName
+                || obj?.userData?.solidName
+                || obj?.userData?.name
+                || obj?.type
+                || 'Object'
+            ));
+            const desc = names.length ? names.join(', ') : '(none)';
+            console.log(`[SelectionFilter] selection changed -> ${desc}`);
+        } catch { /* noop */ }
         try {
             const ev = new CustomEvent('selection-changed');
             window.dispatchEvent(ev);
@@ -874,6 +873,7 @@ export class SelectionFilter {
     static _getHistoryContextActionSpecs(selection, viewer) {
         const out = [];
         const items = Array.isArray(selection) ? selection : [];
+        const suppressFeatureButtons = SelectionFilter._hasAssemblyComponentSelection(items, viewer);
         const safeId = (prefix, key) => {
             const raw = String(key || '').toLowerCase().replace(/[^a-z0-9_-]+/g, '-');
             return `${prefix}-${raw || 'item'}`;
@@ -887,21 +887,23 @@ export class SelectionFilter {
         if (!pmiActive) {
             const featureRegistry = viewer?.partHistory?.featureRegistry || null;
             const features = Array.isArray(featureRegistry?.features) ? featureRegistry.features : [];
-            for (const FeatureClass of features) {
-                if (!FeatureClass) continue;
-                let result = null;
-                try { result = FeatureClass.showContexButton?.(items); } catch { result = null; }
-                if (!result) continue;
-                if (result && typeof result === 'object' && result.show === false) continue;
-                const label = (result && typeof result === 'object' && result.label) || FeatureClass.longName || FeatureClass.shortName || FeatureClass.name || 'Feature';
-                const typeKey = FeatureClass.shortName || FeatureClass.type || FeatureClass.name || label;
-                const params = SelectionFilter._extractContextParams(result);
-                addSpec({
-                    id: safeId('ctx-feature', typeKey),
-                    label,
-                    title: `Create ${label}`,
-                    onClick: () => SelectionFilter._createFeatureFromContext(viewer, typeKey, params),
-                });
+            if (!suppressFeatureButtons) {
+                for (const FeatureClass of features) {
+                    if (!FeatureClass) continue;
+                    let result = null;
+                    try { result = FeatureClass.showContexButton?.(items); } catch { result = null; }
+                    if (!result) continue;
+                    if (result && typeof result === 'object' && result.show === false) continue;
+                    const label = (result && typeof result === 'object' && result.label) || FeatureClass.longName || FeatureClass.shortName || FeatureClass.name || 'Feature';
+                    const typeKey = FeatureClass.shortName || FeatureClass.type || FeatureClass.name || label;
+                    const params = SelectionFilter._extractContextParams(result);
+                    addSpec({
+                        id: safeId('ctx-feature', typeKey),
+                        label,
+                        title: `Create ${label}`,
+                        onClick: () => SelectionFilter._createFeatureFromContext(viewer, typeKey, params),
+                    });
+                }
             }
 
             const constraintRegistry = viewer?.partHistory?.assemblyConstraintRegistry || null;
@@ -952,6 +954,28 @@ export class SelectionFilter {
         return out;
     }
 
+    static _hasAssemblyComponentSelection(items, viewer) {
+        if (!Array.isArray(items) || !items.length) return false;
+        const findComponent = (obj) => {
+            if (!obj) return null;
+            if (viewer && typeof viewer._findOwningComponent === 'function') {
+                try { return viewer._findOwningComponent(obj); } catch { /* ignore */ }
+            }
+            let cur = obj;
+            while (cur) {
+                if (cur.isAssemblyComponent || cur.type === SelectionFilter.COMPONENT || cur.type === 'COMPONENT') return cur;
+                cur = cur.parent || null;
+            }
+            return null;
+        };
+        for (const item of items) {
+            const obj = item?.object || item?.target || item;
+            if (!obj) continue;
+            if (findComponent(obj)) return true;
+        }
+        return false;
+    }
+
     static async _createFeatureFromContext(viewer, typeKey, params = null) {
         if (!viewer || !typeKey) return;
         SelectionFilter.setContextBarSuppressed('context-create', true);
@@ -1127,6 +1151,8 @@ export class SelectionFilter {
             FACE: 'Face',
             PLANE: 'Plane',
             SKETCH: 'Sketch',
+            DATUM: 'Datum',
+            HELIX: 'Helix',
             EDGE: 'Edge',
             LOOP: 'Loop',
             VERTEX: 'Vertex',
@@ -1494,6 +1520,7 @@ export class SelectionFilter {
     }
 
     static #logAllowedTypesChange(next, reason = '') {
+        if (!debugMode) return;
         try {
             const desc = next === SelectionFilter.ALL
                 ? 'ALL'