feat: Add E11 Part Mass extractor, document pyNastran mass accuracy issue
New E11 Part Mass Extractor: - Add nx_journals/extract_part_mass_material.py - NX journal using NXOpen.MeasureManager.NewMassProperties() for accurate geometry-based mass - Add optimization_engine/extractors/extract_part_mass_material.py - Python wrapper that reads JSON output from journal - Add E11 entry to extractors/catalog.json Documentation Updates: - SYS_12_EXTRACTOR_LIBRARY.md: Add mass accuracy warning noting pyNastran get_mass_breakdown() under-reports ~7% on hex-dominant meshes with tet/pyramid fill elements. E11 (geometry .prt) should be preferred over E4 (BDF) unless material is overridden at FEM level. - 01_CHEATSHEET.md: Add mass extraction tip V14 Config: - Expand design variable bounds (blank_backface_angle max 4.5°, whiffle_triangle_closeness max 80mm, whiffle_min max 60mm) Testing showed: - E11 from .prt: 97.66 kg (accurate - matches NX GUI) - E4 pyNastran get_mass_breakdown(): 90.73 kg (~7% under-reported) 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
@@ -49,6 +49,9 @@ requires_skills:
|
|||||||
| Zernike builder | E10 | `ZernikeObjectiveBuilder(op2_finder)` |
|
| Zernike builder | E10 | `ZernikeObjectiveBuilder(op2_finder)` |
|
||||||
| Part mass + material | E11 | `extract_part_mass_material(prt_file)` → mass, volume, material |
|
| Part mass + material | E11 | `extract_part_mass_material(prt_file)` → mass, volume, material |
|
||||||
|
|
||||||
|
> **Mass extraction tip**: Always use E11 (geometry .prt) over E4 (BDF) for accuracy.
|
||||||
|
> pyNastran under-reports mass ~7% on hex-dominant meshes with tet/pyramid fills.
|
||||||
|
|
||||||
**Full details**: See `SYS_12_EXTRACTOR_LIBRARY.md` or `modules/extractors-catalog.md`
|
**Full details**: See `SYS_12_EXTRACTOR_LIBRARY.md` or `modules/extractors-catalog.md`
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|||||||
@@ -195,6 +195,18 @@ material = extractor.material_name
|
|||||||
- `NXOpen.Body.GetBodies()`
|
- `NXOpen.Body.GetBodies()`
|
||||||
- `NXOpen.PhysicalMaterial`
|
- `NXOpen.PhysicalMaterial`
|
||||||
|
|
||||||
|
**IMPORTANT - Mass Accuracy Note**:
|
||||||
|
> **Always prefer E11 (geometry-based) over E4 (BDF-based) for mass extraction.**
|
||||||
|
>
|
||||||
|
> Testing on hex-dominant meshes with tet/pyramid fill elements revealed that:
|
||||||
|
> - **E11 from .prt**: 97.66 kg (accurate - matches NX GUI)
|
||||||
|
> - **E4 pyNastran get_mass_breakdown()**: 90.73 kg (~7% under-reported)
|
||||||
|
> - **E4 pyNastran sum(elem.Volume())*rho**: 100.16 kg (~2.5% over-reported)
|
||||||
|
>
|
||||||
|
> The `get_mass_breakdown()` function in pyNastran has known issues with mixed-element
|
||||||
|
> meshes (CHEXA + CPENTA + CPYRAM + CTETRA). Use E11 with the NX journal for reliable
|
||||||
|
> mass values. Only use E4 if material properties are overridden at FEM level.
|
||||||
|
|
||||||
### E6: Field Data Extraction
|
### E6: Field Data Extraction
|
||||||
|
|
||||||
**Module**: `optimization_engine.extractors.field_data_extractor`
|
**Module**: `optimization_engine.extractors.field_data_extractor`
|
||||||
|
|||||||
328
nx_journals/extract_part_mass_material.py
Normal file
328
nx_journals/extract_part_mass_material.py
Normal file
@@ -0,0 +1,328 @@
|
|||||||
|
"""
|
||||||
|
NX Journal Script to Extract Part Mass and Material Properties
|
||||||
|
|
||||||
|
This script extracts:
|
||||||
|
- Mass (kg)
|
||||||
|
- Volume (mm^3)
|
||||||
|
- Surface area (mm^2)
|
||||||
|
- Center of gravity (mm)
|
||||||
|
- Material name (if assigned)
|
||||||
|
- Density (kg/mm^3)
|
||||||
|
|
||||||
|
Results are written to _temp_part_properties.json in the working directory.
|
||||||
|
|
||||||
|
Usage:
|
||||||
|
run_journal.exe extract_part_mass_material.py <prt_file_path> [output_dir]
|
||||||
|
|
||||||
|
NX Open APIs Used:
|
||||||
|
- NXOpen.MeasureManager.NewMassProperties()
|
||||||
|
- NXOpen.MeasureBodies
|
||||||
|
- NXOpen.Body.GetBodies()
|
||||||
|
- NXOpen.PhysicalMaterial
|
||||||
|
|
||||||
|
Author: Atomizer
|
||||||
|
Created: 2025-12-05
|
||||||
|
Version: 1.0
|
||||||
|
"""
|
||||||
|
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
import json
|
||||||
|
import NXOpen
|
||||||
|
import NXOpen.UF
|
||||||
|
|
||||||
|
|
||||||
|
def get_all_solid_bodies(part):
|
||||||
|
"""Get all solid bodies from the part."""
|
||||||
|
bodies = []
|
||||||
|
try:
|
||||||
|
# Get bodies from the part
|
||||||
|
for body in part.Bodies:
|
||||||
|
if body.IsSolidBody:
|
||||||
|
bodies.append(body)
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[JOURNAL] Warning getting bodies: {e}")
|
||||||
|
return bodies
|
||||||
|
|
||||||
|
|
||||||
|
def get_material_info(part, body):
|
||||||
|
"""Extract material information from a body if assigned."""
|
||||||
|
material_info = {
|
||||||
|
'name': None,
|
||||||
|
'density': None,
|
||||||
|
'density_unit': 'kg/mm^3'
|
||||||
|
}
|
||||||
|
|
||||||
|
# Method 1: Try to get physical material directly from body
|
||||||
|
try:
|
||||||
|
phys_mat = body.GetPhysicalMaterial()
|
||||||
|
if phys_mat:
|
||||||
|
material_info['name'] = phys_mat.Name
|
||||||
|
print(f"[JOURNAL] Material from body: {phys_mat.Name}")
|
||||||
|
|
||||||
|
# Try to get density property
|
||||||
|
try:
|
||||||
|
density_prop = phys_mat.GetPropertyValue("Density")
|
||||||
|
if density_prop:
|
||||||
|
material_info['density'] = float(density_prop)
|
||||||
|
print(f"[JOURNAL] Density: {density_prop}")
|
||||||
|
except Exception as de:
|
||||||
|
print(f"[JOURNAL] Could not get density: {de}")
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[JOURNAL] GetPhysicalMaterial failed: {e}")
|
||||||
|
|
||||||
|
# Method 2: If no material from body, try part-level material assignment
|
||||||
|
if material_info['name'] is None:
|
||||||
|
try:
|
||||||
|
# Check if part has PhysicalMaterialManager
|
||||||
|
pmm = part.PhysicalMaterialManager
|
||||||
|
if pmm:
|
||||||
|
# Get all materials in the part
|
||||||
|
materials = pmm.GetAllPhysicalMaterials()
|
||||||
|
if materials and len(materials) > 0:
|
||||||
|
# Use the first material as default
|
||||||
|
mat = materials[0]
|
||||||
|
material_info['name'] = mat.Name
|
||||||
|
print(f"[JOURNAL] Material from PhysicalMaterialManager: {mat.Name}")
|
||||||
|
|
||||||
|
try:
|
||||||
|
density_prop = mat.GetPropertyValue("Density")
|
||||||
|
if density_prop:
|
||||||
|
material_info['density'] = float(density_prop)
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[JOURNAL] PhysicalMaterialManager failed: {e}")
|
||||||
|
|
||||||
|
# Method 3: Try using body attributes
|
||||||
|
if material_info['name'] is None:
|
||||||
|
try:
|
||||||
|
# Some NX versions store material as an attribute
|
||||||
|
attrs = body.GetUserAttributes()
|
||||||
|
for attr in attrs:
|
||||||
|
if 'material' in attr.Title.lower():
|
||||||
|
material_info['name'] = attr.StringValue
|
||||||
|
print(f"[JOURNAL] Material from attribute: {attr.StringValue}")
|
||||||
|
break
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[JOURNAL] Body attributes failed: {e}")
|
||||||
|
|
||||||
|
return material_info
|
||||||
|
|
||||||
|
|
||||||
|
def extract_mass_properties(theSession, part, bodies):
|
||||||
|
"""
|
||||||
|
Extract mass properties using MeasureManager.
|
||||||
|
|
||||||
|
API Reference: https://www.nxjournaling.com/content/mass-properties-using-python
|
||||||
|
|
||||||
|
Returns dict with mass, volume, area, center of gravity.
|
||||||
|
"""
|
||||||
|
results = {
|
||||||
|
'mass_kg': 0.0,
|
||||||
|
'volume_mm3': 0.0,
|
||||||
|
'surface_area_mm2': 0.0,
|
||||||
|
'center_of_gravity_mm': [0.0, 0.0, 0.0],
|
||||||
|
'moments_of_inertia': None,
|
||||||
|
'num_bodies': len(bodies)
|
||||||
|
}
|
||||||
|
|
||||||
|
if not bodies:
|
||||||
|
print("[JOURNAL] No solid bodies found in part")
|
||||||
|
return results
|
||||||
|
|
||||||
|
try:
|
||||||
|
# Get the measure manager
|
||||||
|
measureManager = part.MeasureManager
|
||||||
|
|
||||||
|
# Convert bodies list to array for NX API
|
||||||
|
bodyArray = bodies if isinstance(bodies, list) else list(bodies)
|
||||||
|
|
||||||
|
# Get unit collection and build mass_units array
|
||||||
|
# API requires: [Area, Volume, Mass, Length] base units
|
||||||
|
uc = part.UnitCollection
|
||||||
|
mass_units = [
|
||||||
|
uc.GetBase("Area"),
|
||||||
|
uc.GetBase("Volume"),
|
||||||
|
uc.GetBase("Mass"),
|
||||||
|
uc.GetBase("Length")
|
||||||
|
]
|
||||||
|
|
||||||
|
# Create mass properties measurement
|
||||||
|
# Signature: NewMassProperties(mass_units, accuracy, objects)
|
||||||
|
measureBodies = measureManager.NewMassProperties(mass_units, 0.99, bodyArray)
|
||||||
|
print("[JOURNAL] Using NewMassProperties(mass_units, accuracy, bodies) API")
|
||||||
|
|
||||||
|
# Get the results
|
||||||
|
if measureBodies:
|
||||||
|
# Mass
|
||||||
|
try:
|
||||||
|
results['mass_kg'] = measureBodies.Mass
|
||||||
|
print(f"[JOURNAL] Raw mass value: {measureBodies.Mass}")
|
||||||
|
except AttributeError as e:
|
||||||
|
print(f"[JOURNAL] Mass attribute error: {e}")
|
||||||
|
|
||||||
|
# Volume
|
||||||
|
try:
|
||||||
|
results['volume_mm3'] = measureBodies.Volume
|
||||||
|
except AttributeError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Surface area
|
||||||
|
try:
|
||||||
|
results['surface_area_mm2'] = measureBodies.Area
|
||||||
|
except AttributeError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Center of gravity
|
||||||
|
try:
|
||||||
|
cog = measureBodies.Centroid
|
||||||
|
if cog:
|
||||||
|
results['center_of_gravity_mm'] = [cog.X, cog.Y, cog.Z]
|
||||||
|
except AttributeError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
# Moments of inertia - NewMassProperties doesn't provide inertia tensors
|
||||||
|
# Would need different API for that
|
||||||
|
results['moments_of_inertia'] = None
|
||||||
|
|
||||||
|
# Dispose
|
||||||
|
try:
|
||||||
|
measureBodies.Dispose()
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[JOURNAL] Error in mass properties extraction: {e}")
|
||||||
|
import traceback
|
||||||
|
traceback.print_exc()
|
||||||
|
|
||||||
|
return results
|
||||||
|
|
||||||
|
|
||||||
|
def main(args):
|
||||||
|
"""
|
||||||
|
Main entry point for NX journal.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
args[0]: .prt file path
|
||||||
|
args[1]: output directory (optional, defaults to prt directory)
|
||||||
|
"""
|
||||||
|
if len(args) < 1:
|
||||||
|
print("ERROR: No .prt file path provided")
|
||||||
|
print("Usage: run_journal.exe extract_part_mass_material.py <prt_file> [output_dir]")
|
||||||
|
return False
|
||||||
|
|
||||||
|
prt_file_path = args[0]
|
||||||
|
output_dir = args[1] if len(args) > 1 else os.path.dirname(prt_file_path)
|
||||||
|
|
||||||
|
prt_filename = os.path.basename(prt_file_path)
|
||||||
|
|
||||||
|
print(f"[JOURNAL] " + "="*60)
|
||||||
|
print(f"[JOURNAL] NX PART MASS & MATERIAL EXTRACTOR")
|
||||||
|
print(f"[JOURNAL] " + "="*60)
|
||||||
|
print(f"[JOURNAL] Part file: {prt_filename}")
|
||||||
|
print(f"[JOURNAL] Output dir: {output_dir}")
|
||||||
|
|
||||||
|
results = {
|
||||||
|
'part_file': prt_filename,
|
||||||
|
'part_path': prt_file_path,
|
||||||
|
'mass_kg': 0.0,
|
||||||
|
'mass_g': 0.0,
|
||||||
|
'volume_mm3': 0.0,
|
||||||
|
'surface_area_mm2': 0.0,
|
||||||
|
'center_of_gravity_mm': [0.0, 0.0, 0.0],
|
||||||
|
'moments_of_inertia': None,
|
||||||
|
'material': {
|
||||||
|
'name': None,
|
||||||
|
'density': None,
|
||||||
|
'density_unit': 'kg/mm^3'
|
||||||
|
},
|
||||||
|
'num_bodies': 0,
|
||||||
|
'success': False,
|
||||||
|
'error': None
|
||||||
|
}
|
||||||
|
|
||||||
|
try:
|
||||||
|
theSession = NXOpen.Session.GetSession()
|
||||||
|
|
||||||
|
# Set load options
|
||||||
|
working_dir = os.path.dirname(prt_file_path)
|
||||||
|
theSession.Parts.LoadOptions.ComponentLoadMethod = NXOpen.LoadOptions.LoadMethod.FromDirectory
|
||||||
|
theSession.Parts.LoadOptions.SetSearchDirectories([working_dir], [True])
|
||||||
|
|
||||||
|
# Open the part file
|
||||||
|
print(f"[JOURNAL] Opening part file...")
|
||||||
|
basePart, partLoadStatus = theSession.Parts.OpenActiveDisplay(
|
||||||
|
prt_file_path,
|
||||||
|
NXOpen.DisplayPartOption.AllowAdditional
|
||||||
|
)
|
||||||
|
partLoadStatus.Dispose()
|
||||||
|
|
||||||
|
workPart = theSession.Parts.Work
|
||||||
|
print(f"[JOURNAL] Loaded part: {workPart.Name}")
|
||||||
|
|
||||||
|
# Get all solid bodies
|
||||||
|
bodies = get_all_solid_bodies(workPart)
|
||||||
|
print(f"[JOURNAL] Found {len(bodies)} solid bodies")
|
||||||
|
|
||||||
|
if bodies:
|
||||||
|
# Extract mass properties
|
||||||
|
mass_props = extract_mass_properties(theSession, workPart, bodies)
|
||||||
|
results.update(mass_props)
|
||||||
|
results['mass_g'] = results['mass_kg'] * 1000.0
|
||||||
|
|
||||||
|
# Get material from first body (typically all bodies have same material)
|
||||||
|
material_info = get_material_info(workPart, bodies[0])
|
||||||
|
results['material'] = material_info
|
||||||
|
|
||||||
|
results['success'] = True
|
||||||
|
|
||||||
|
print(f"[JOURNAL] ")
|
||||||
|
print(f"[JOURNAL] RESULTS:")
|
||||||
|
print(f"[JOURNAL] Mass: {results['mass_kg']:.6f} kg ({results['mass_g']:.2f} g)")
|
||||||
|
print(f"[JOURNAL] Volume: {results['volume_mm3']:.2f} mm^3")
|
||||||
|
print(f"[JOURNAL] Surface Area: {results['surface_area_mm2']:.2f} mm^2")
|
||||||
|
print(f"[JOURNAL] CoG: [{results['center_of_gravity_mm'][0]:.2f}, {results['center_of_gravity_mm'][1]:.2f}, {results['center_of_gravity_mm'][2]:.2f}] mm")
|
||||||
|
if material_info['name']:
|
||||||
|
print(f"[JOURNAL] Material: {material_info['name']}")
|
||||||
|
if material_info['density']:
|
||||||
|
print(f"[JOURNAL] Density: {material_info['density']} {material_info['density_unit']}")
|
||||||
|
else:
|
||||||
|
results['error'] = "No solid bodies found in part"
|
||||||
|
print(f"[JOURNAL] ERROR: No solid bodies found")
|
||||||
|
|
||||||
|
# Write results to JSON file
|
||||||
|
output_file = os.path.join(output_dir, "_temp_part_properties.json")
|
||||||
|
with open(output_file, 'w') as f:
|
||||||
|
json.dump(results, f, indent=2)
|
||||||
|
print(f"[JOURNAL] Results written to: {output_file}")
|
||||||
|
|
||||||
|
# Also write simple mass value for backward compatibility
|
||||||
|
mass_file = os.path.join(output_dir, "_temp_mass.txt")
|
||||||
|
with open(mass_file, 'w') as f:
|
||||||
|
f.write(str(results['mass_kg']))
|
||||||
|
print(f"[JOURNAL] Mass written to: {mass_file}")
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
results['error'] = str(e)
|
||||||
|
results['success'] = False
|
||||||
|
print(f"[JOURNAL] FATAL ERROR: {e}")
|
||||||
|
import traceback
|
||||||
|
traceback.print_exc()
|
||||||
|
|
||||||
|
# Still write results file even on error
|
||||||
|
output_file = os.path.join(output_dir, "_temp_part_properties.json")
|
||||||
|
try:
|
||||||
|
with open(output_file, 'w') as f:
|
||||||
|
json.dump(results, f, indent=2)
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main(sys.argv[1:])
|
||||||
@@ -34,5 +34,19 @@
|
|||||||
"metric": "total"
|
"metric": "total"
|
||||||
},
|
},
|
||||||
"signature": "2f58f241a96afb1f"
|
"signature": "2f58f241a96afb1f"
|
||||||
|
},
|
||||||
|
"e11_part_mass_material": {
|
||||||
|
"name": "extract_part_mass_material",
|
||||||
|
"filename": "extract_part_mass_material.py",
|
||||||
|
"action": "extract_part_mass_material",
|
||||||
|
"domain": "cad_extraction",
|
||||||
|
"description": "Extract mass, volume, surface area, CoG, and material from NX .prt files via NXOpen.MeasureManager",
|
||||||
|
"params": {
|
||||||
|
"result_type": "part_mass",
|
||||||
|
"metric": "total",
|
||||||
|
"requires_journal": "nx_journals/extract_part_mass_material.py"
|
||||||
|
},
|
||||||
|
"signature": "e11_part_mass_material",
|
||||||
|
"notes": "Requires running NX journal first: run_journal.exe nx_journals/extract_part_mass_material.py <prt_file> -args <prt_path> <output_dir>"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
276
optimization_engine/extractors/extract_part_mass_material.py
Normal file
276
optimization_engine/extractors/extract_part_mass_material.py
Normal file
@@ -0,0 +1,276 @@
|
|||||||
|
"""
|
||||||
|
Extract Part Mass and Material Properties from NX Part Files
|
||||||
|
|
||||||
|
This extractor reads mass and material data from a temp file written by
|
||||||
|
the NX journal: nx_journals/extract_part_mass_material.py
|
||||||
|
|
||||||
|
The journal uses NXOpen.MeasureManager to extract:
|
||||||
|
- Mass (kg)
|
||||||
|
- Volume (mm^3)
|
||||||
|
- Surface area (mm^2)
|
||||||
|
- Center of gravity (mm)
|
||||||
|
- Material name and density
|
||||||
|
|
||||||
|
NX Open APIs Used (by journal):
|
||||||
|
- NXOpen.MeasureManager.NewMassProperties()
|
||||||
|
- NXOpen.MeasureBodies
|
||||||
|
- NXOpen.Body.GetBodies()
|
||||||
|
- NXOpen.PhysicalMaterial
|
||||||
|
|
||||||
|
Author: Atomizer
|
||||||
|
Created: 2025-12-05
|
||||||
|
Version: 1.0
|
||||||
|
"""
|
||||||
|
|
||||||
|
from pathlib import Path
|
||||||
|
from typing import Dict, Any, Optional, Union
|
||||||
|
import json
|
||||||
|
|
||||||
|
|
||||||
|
def extract_part_mass_material(
|
||||||
|
prt_file: Union[str, Path],
|
||||||
|
properties_file: Optional[Union[str, Path]] = None
|
||||||
|
) -> Dict[str, Any]:
|
||||||
|
"""
|
||||||
|
Extract mass and material properties from NX part file.
|
||||||
|
|
||||||
|
This function reads from a temp JSON file that must be created by
|
||||||
|
running the NX journal: nx_journals/extract_part_mass_material.py
|
||||||
|
|
||||||
|
Args:
|
||||||
|
prt_file: Path to .prt file (used to locate temp file)
|
||||||
|
properties_file: Optional explicit path to _temp_part_properties.json
|
||||||
|
If not provided, looks in same directory as prt_file
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary containing:
|
||||||
|
- 'mass_kg': Mass in kilograms (float)
|
||||||
|
- 'mass_g': Mass in grams (float)
|
||||||
|
- 'volume_mm3': Volume in mm^3 (float)
|
||||||
|
- 'surface_area_mm2': Surface area in mm^2 (float)
|
||||||
|
- 'center_of_gravity_mm': [x, y, z] in mm (list)
|
||||||
|
- 'moments_of_inertia': {'Ixx', 'Iyy', 'Izz', 'unit'} or None
|
||||||
|
- 'material': {'name', 'density', 'density_unit'} (dict)
|
||||||
|
- 'num_bodies': Number of solid bodies (int)
|
||||||
|
|
||||||
|
Raises:
|
||||||
|
FileNotFoundError: If prt file or temp properties file not found
|
||||||
|
ValueError: If temp file has invalid format or extraction failed
|
||||||
|
|
||||||
|
Example:
|
||||||
|
>>> result = extract_part_mass_material('model.prt')
|
||||||
|
>>> print(f"Mass: {result['mass_kg']:.3f} kg")
|
||||||
|
Mass: 1.234 kg
|
||||||
|
>>> print(f"Material: {result['material']['name']}")
|
||||||
|
Material: Aluminum_6061
|
||||||
|
|
||||||
|
Note:
|
||||||
|
Before calling this function, you must run the NX journal to
|
||||||
|
create the temp file:
|
||||||
|
```
|
||||||
|
run_journal.exe extract_part_mass_material.py model.prt
|
||||||
|
```
|
||||||
|
"""
|
||||||
|
prt_file = Path(prt_file)
|
||||||
|
|
||||||
|
if not prt_file.exists():
|
||||||
|
raise FileNotFoundError(f"Part file not found: {prt_file}")
|
||||||
|
|
||||||
|
# Determine properties file location
|
||||||
|
if properties_file:
|
||||||
|
props_file = Path(properties_file)
|
||||||
|
else:
|
||||||
|
props_file = prt_file.parent / "_temp_part_properties.json"
|
||||||
|
|
||||||
|
if not props_file.exists():
|
||||||
|
raise FileNotFoundError(
|
||||||
|
f"Part properties temp file not found: {props_file}\n"
|
||||||
|
f"Run the NX journal first:\n"
|
||||||
|
f" run_journal.exe extract_part_mass_material.py {prt_file}"
|
||||||
|
)
|
||||||
|
|
||||||
|
# Read and parse JSON
|
||||||
|
try:
|
||||||
|
with open(props_file, 'r') as f:
|
||||||
|
data = json.load(f)
|
||||||
|
except json.JSONDecodeError as e:
|
||||||
|
raise ValueError(f"Invalid JSON in properties file: {e}")
|
||||||
|
|
||||||
|
# Check for extraction errors
|
||||||
|
if not data.get('success', False):
|
||||||
|
error_msg = data.get('error', 'Unknown error during extraction')
|
||||||
|
raise ValueError(f"NX extraction failed: {error_msg}")
|
||||||
|
|
||||||
|
# Build result dictionary
|
||||||
|
result = {
|
||||||
|
'mass_kg': float(data.get('mass_kg', 0.0)),
|
||||||
|
'mass_g': float(data.get('mass_g', 0.0)),
|
||||||
|
'volume_mm3': float(data.get('volume_mm3', 0.0)),
|
||||||
|
'surface_area_mm2': float(data.get('surface_area_mm2', 0.0)),
|
||||||
|
'center_of_gravity_mm': data.get('center_of_gravity_mm', [0.0, 0.0, 0.0]),
|
||||||
|
'moments_of_inertia': data.get('moments_of_inertia'),
|
||||||
|
'material': data.get('material', {'name': None, 'density': None, 'density_unit': 'kg/mm^3'}),
|
||||||
|
'num_bodies': int(data.get('num_bodies', 0)),
|
||||||
|
'part_file': data.get('part_file', prt_file.name),
|
||||||
|
}
|
||||||
|
|
||||||
|
print(f"[OK] Part mass: {result['mass_kg']:.6f} kg ({result['mass_g']:.2f} g)")
|
||||||
|
if result['material'].get('name'):
|
||||||
|
print(f"[OK] Material: {result['material']['name']}")
|
||||||
|
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
|
def extract_part_mass(
|
||||||
|
prt_file: Union[str, Path],
|
||||||
|
properties_file: Optional[Union[str, Path]] = None
|
||||||
|
) -> float:
|
||||||
|
"""
|
||||||
|
Convenience function to extract just the mass in kg.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
prt_file: Path to .prt file
|
||||||
|
properties_file: Optional explicit path to temp file
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Mass in kilograms (float)
|
||||||
|
|
||||||
|
Example:
|
||||||
|
>>> mass = extract_part_mass('model.prt')
|
||||||
|
>>> print(f"Mass: {mass:.3f} kg")
|
||||||
|
Mass: 1.234 kg
|
||||||
|
"""
|
||||||
|
result = extract_part_mass_material(prt_file, properties_file)
|
||||||
|
return result['mass_kg']
|
||||||
|
|
||||||
|
|
||||||
|
def extract_part_material(
|
||||||
|
prt_file: Union[str, Path],
|
||||||
|
properties_file: Optional[Union[str, Path]] = None
|
||||||
|
) -> Dict[str, Any]:
|
||||||
|
"""
|
||||||
|
Convenience function to extract just material info.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
prt_file: Path to .prt file
|
||||||
|
properties_file: Optional explicit path to temp file
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary with 'name', 'density', 'density_unit'
|
||||||
|
|
||||||
|
Example:
|
||||||
|
>>> mat = extract_part_material('model.prt')
|
||||||
|
>>> print(f"Material: {mat['name']}, Density: {mat['density']}")
|
||||||
|
Material: Steel_304, Density: 7.93e-06
|
||||||
|
"""
|
||||||
|
result = extract_part_mass_material(prt_file, properties_file)
|
||||||
|
return result['material']
|
||||||
|
|
||||||
|
|
||||||
|
class PartMassExtractor:
|
||||||
|
"""
|
||||||
|
Class-based extractor for part mass and material with caching.
|
||||||
|
|
||||||
|
Use this when you need to extract properties from multiple parts
|
||||||
|
or want to cache results.
|
||||||
|
|
||||||
|
Example:
|
||||||
|
>>> extractor = PartMassExtractor('model.prt')
|
||||||
|
>>> result = extractor.extract()
|
||||||
|
>>> print(result['mass_kg'])
|
||||||
|
1.234
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
prt_file: Union[str, Path],
|
||||||
|
properties_file: Optional[Union[str, Path]] = None
|
||||||
|
):
|
||||||
|
"""
|
||||||
|
Initialize the extractor.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
prt_file: Path to .prt file
|
||||||
|
properties_file: Optional explicit path to temp file
|
||||||
|
"""
|
||||||
|
self.prt_file = Path(prt_file)
|
||||||
|
self.properties_file = Path(properties_file) if properties_file else None
|
||||||
|
self._cached_result = None
|
||||||
|
|
||||||
|
def extract(self, use_cache: bool = True) -> Dict[str, Any]:
|
||||||
|
"""
|
||||||
|
Extract mass and material properties.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
use_cache: If True, returns cached result if available
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary with all extracted properties
|
||||||
|
"""
|
||||||
|
if use_cache and self._cached_result is not None:
|
||||||
|
return self._cached_result
|
||||||
|
|
||||||
|
self._cached_result = extract_part_mass_material(
|
||||||
|
self.prt_file,
|
||||||
|
self.properties_file
|
||||||
|
)
|
||||||
|
return self._cached_result
|
||||||
|
|
||||||
|
@property
|
||||||
|
def mass_kg(self) -> float:
|
||||||
|
"""Get mass in kg (extracts if needed)."""
|
||||||
|
return self.extract()['mass_kg']
|
||||||
|
|
||||||
|
@property
|
||||||
|
def mass_g(self) -> float:
|
||||||
|
"""Get mass in grams (extracts if needed)."""
|
||||||
|
return self.extract()['mass_g']
|
||||||
|
|
||||||
|
@property
|
||||||
|
def material_name(self) -> Optional[str]:
|
||||||
|
"""Get material name (extracts if needed)."""
|
||||||
|
return self.extract()['material'].get('name')
|
||||||
|
|
||||||
|
@property
|
||||||
|
def density(self) -> Optional[float]:
|
||||||
|
"""Get material density in kg/mm^3 (extracts if needed)."""
|
||||||
|
return self.extract()['material'].get('density')
|
||||||
|
|
||||||
|
def clear_cache(self):
|
||||||
|
"""Clear the cached result."""
|
||||||
|
self._cached_result = None
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
import sys
|
||||||
|
|
||||||
|
if len(sys.argv) < 2:
|
||||||
|
print(f"Usage: python {sys.argv[0]} <prt_file> [properties_file]")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
prt_file = Path(sys.argv[1])
|
||||||
|
props_file = Path(sys.argv[2]) if len(sys.argv) > 2 else None
|
||||||
|
|
||||||
|
try:
|
||||||
|
result = extract_part_mass_material(prt_file, props_file)
|
||||||
|
print("\n" + "="*50)
|
||||||
|
print("PART MASS & MATERIAL EXTRACTION RESULTS")
|
||||||
|
print("="*50)
|
||||||
|
print(f"Part File: {result['part_file']}")
|
||||||
|
print(f"Mass: {result['mass_kg']:.6f} kg ({result['mass_g']:.2f} g)")
|
||||||
|
print(f"Volume: {result['volume_mm3']:.2f} mm^3")
|
||||||
|
print(f"Surface Area: {result['surface_area_mm2']:.2f} mm^2")
|
||||||
|
print(f"Center of Gravity: {result['center_of_gravity_mm']} mm")
|
||||||
|
print(f"Num Bodies: {result['num_bodies']}")
|
||||||
|
if result['material']['name']:
|
||||||
|
print(f"Material: {result['material']['name']}")
|
||||||
|
if result['material']['density']:
|
||||||
|
print(f"Density: {result['material']['density']} {result['material']['density_unit']}")
|
||||||
|
if result['moments_of_inertia']:
|
||||||
|
moi = result['moments_of_inertia']
|
||||||
|
print(f"Moments of Inertia: Ixx={moi['Ixx']}, Iyy={moi['Iyy']}, Izz={moi['Izz']} {moi['unit']}")
|
||||||
|
except Exception as e:
|
||||||
|
print(f"\nERROR: {e}")
|
||||||
|
import traceback
|
||||||
|
traceback.print_exc()
|
||||||
|
sys.exit(1)
|
||||||
@@ -77,7 +77,7 @@
|
|||||||
"name": "whiffle_min",
|
"name": "whiffle_min",
|
||||||
"expression_name": "whiffle_min",
|
"expression_name": "whiffle_min",
|
||||||
"min": 30.0,
|
"min": 30.0,
|
||||||
"max": 55.0,
|
"max": 60.0,
|
||||||
"baseline": 40.55,
|
"baseline": 40.55,
|
||||||
"units": "mm",
|
"units": "mm",
|
||||||
"enabled": true
|
"enabled": true
|
||||||
@@ -95,7 +95,7 @@
|
|||||||
"name": "whiffle_triangle_closeness",
|
"name": "whiffle_triangle_closeness",
|
||||||
"expression_name": "whiffle_triangle_closeness",
|
"expression_name": "whiffle_triangle_closeness",
|
||||||
"min": 50.0,
|
"min": 50.0,
|
||||||
"max": 65.0,
|
"max": 80.0,
|
||||||
"baseline": 60.00,
|
"baseline": 60.00,
|
||||||
"units": "mm",
|
"units": "mm",
|
||||||
"enabled": true
|
"enabled": true
|
||||||
@@ -104,7 +104,7 @@
|
|||||||
"name": "blank_backface_angle",
|
"name": "blank_backface_angle",
|
||||||
"expression_name": "blank_backface_angle",
|
"expression_name": "blank_backface_angle",
|
||||||
"min": 4.1,
|
"min": 4.1,
|
||||||
"max": 4.2,
|
"max": 4.5,
|
||||||
"baseline": 4.15,
|
"baseline": 4.15,
|
||||||
"units": "degrees",
|
"units": "degrees",
|
||||||
"enabled": true
|
"enabled": true
|
||||||
|
|||||||
Reference in New Issue
Block a user