Antoine/Atomizer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Add NX file dependency tree to introspection panel

Browse Source 
Backend:
- Add scan_nx_file_dependencies() function to parse NX file chain
- Uses naming conventions to build dependency tree (.sim -> .afm -> .fem -> _i.prt -> .prt)
- Include file_dependencies in introspection response
- Works without NX (pure file-based analysis)

Frontend:
- Add FileDependencies interface for typed API response
- Add collapsible 'File Dependencies' section with tree visualization
- Color-coded file types (purple=sim, blue=afm, green=fem, yellow=idealized, orange=prt)
- Shows orphan geometry files that aren't in the dependency chain
This commit is contained in:
Anto01
2026-01-20 15:33:04 -05:00
parent abbc7b1b50
commit 14354a2606
2 changed files with 286 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

131
atomizer-dashboard/backend/api/routes/optimization.py
View File

@@ -4485,6 +4485,120 @@ async def export_study_data(study_id: str, format: str):
# ============================================================================
def scan_nx_file_dependencies(model_dir: Path) -> dict:
"""
Scan a model directory for NX files and build a dependency tree.
This uses file naming conventions to determine relationships:
- .sim files reference .afm (assembly FEM) or .fem files
- .afm files reference multiple .fem files
- .fem files reference _i.prt (idealized) files
- _i.prt files are derived from .prt (geometry) files
Returns:
Dictionary with file lists and dependency tree
"""
result = {
"files": {
"sim": [], # Simulation files
"afm": [], # Assembly FEM files
"fem": [], # FEM files
"prt": [], # Part/geometry files
"idealized": [], # Idealized parts (*_i.prt)
},
"dependencies": [], # List of {source, target, type} edges
"root_sim": None,
}
# Collect all files by type
for f in model_dir.iterdir():
if not f.is_file():
continue
name = f.name.lower()
if name.endswith(".sim"):
result["files"]["sim"].append(f.name)
elif name.endswith(".afm"):
result["files"]["afm"].append(f.name)
elif name.endswith(".fem"):
result["files"]["fem"].append(f.name)
elif name.endswith("_i.prt"):
result["files"]["idealized"].append(f.name)
elif name.endswith(".prt"):
result["files"]["prt"].append(f.name)
# Build dependency edges based on NX naming conventions
# SIM -> AFM: SIM file basename matches AFM (e.g., ASSY_M1_assyfem1_sim1.sim -> ASSY_M1_assyfem1.afm)
for sim in result["files"]["sim"]:
sim_base = sim.lower().replace(".sim", "")
# Remove _sim1, _sim2 suffix to get AFM name
afm_base = sim_base.rsplit("_sim", 1)[0] if "_sim" in sim_base else sim_base
# Check for matching AFM
for afm in result["files"]["afm"]:
if afm.lower().replace(".afm", "") == afm_base:
result["dependencies"].append({"source": sim, "target": afm, "type": "sim_to_afm"})
break
else:
# No AFM, check for direct FEM reference
for fem in result["files"]["fem"]:
fem_base = fem.lower().replace(".fem", "")
if afm_base.endswith(fem_base) or fem_base in afm_base:
result["dependencies"].append(
{"source": sim, "target": fem, "type": "sim_to_fem"}
)
break
# Set root sim (first one found)
if result["root_sim"] is None:
result["root_sim"] = sim
# AFM -> FEM: Assembly FEM references component FEMs
# Convention: AFM basename often contains assembly name, FEMs have component names
for afm in result["files"]["afm"]:
afm_base = afm.lower().replace(".afm", "")
# Get the assembly prefix (e.g., ASSY_M1_assyfem1 -> ASSY_M1)
parts = afm_base.split("_")
for fem in result["files"]["fem"]:
fem_base = fem.lower().replace(".fem", "")
# FEM is likely a component if it's not the assembly FEM itself
if fem_base != afm_base:
result["dependencies"].append({"source": afm, "target": fem, "type": "afm_to_fem"})
# FEM -> Idealized PRT: FEM references idealized geometry
# Convention: Model_fem1.fem -> Model_fem1_i.prt
for fem in result["files"]["fem"]:
fem_base = fem.lower().replace(".fem", "")
for idealized in result["files"]["idealized"]:
idealized_base = idealized.lower().replace("_i.prt", "")
if idealized_base == fem_base:
result["dependencies"].append(
{"source": fem, "target": idealized, "type": "fem_to_idealized"}
)
break
# Idealized PRT -> PRT: Idealized part derives from geometry
# Convention: Model_fem1_i.prt -> Model.prt
for idealized in result["files"]["idealized"]:
idealized_base = idealized.lower().replace("_i.prt", "")
# Remove fem suffix to get geometry name (Model_fem1_i -> Model)
geom_base = (
idealized_base.rsplit("_fem", 1)[0] if "_fem" in idealized_base else idealized_base
)
for prt in result["files"]["prt"]:
prt_base = prt.lower().replace(".prt", "")
if prt_base == geom_base:
result["dependencies"].append(
{"source": idealized, "target": prt, "type": "idealized_to_prt"}
)
break
return result
@router.get("/studies/{study_id}/nx/introspect")
async def introspect_nx_model(study_id: str, force: bool = False):
"""
@@ -4554,12 +4668,18 @@ async def introspect_nx_model(study_id: str, force: bool = False):
except:
pass # Invalid cache, re-run
# Scan file dependencies (fast, doesn't need NX)
file_deps = scan_nx_file_dependencies(model_dir)
# Run introspection
try:
from optimization_engine.extractors.introspect_part import introspect_part
result = introspect_part(str(prt_file), str(model_dir), verbose=False)
# Add file dependencies to result
result["file_dependencies"] = file_deps
# Cache results
with open(cache_file, "w") as f:
json.dump(result, f, indent=2)
@@ -4567,7 +4687,16 @@ async def introspect_nx_model(study_id: str, force: bool = False):
return {"study_id": study_id, "cached": False, "introspection": result}
except ImportError:
raise HTTPException(status_code=500, detail="introspect_part module not available")
# If introspect_part not available, return just file dependencies
return {
"study_id": study_id,
"cached": False,
"introspection": {
"success": False,
"error": "introspect_part module not available",
"file_dependencies": file_deps,
},
}
except Exception as e:
raise HTTPException(status_code=500, detail=f"Introspection failed: {str(e)}")