#!/usr/bin/env python3
"""Test extraction pipeline on existing OP2 file."""

import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).parent.parent))

from optimization_engine.extractors import extract_displacement, extract_solid_stress

op2_path = r"C:\Users\antoi\Atomizer\studies\_Other\Model_for_dev\support_arm_sim1-solution_1.op2"

print("Testing extractors on existing OP2 file...")
print(f"File: {op2_path}")
print(f"Exists: {Path(op2_path).exists()}")
print()

# Test displacement extraction
print("1. Displacement extraction:")
try:
    result = extract_displacement(op2_path)
    print(f"   Max magnitude: {result.get('max_magnitude', 'N/A')} mm")
    print(f"   Full result: {result}")
except Exception as e:
    print(f"   ERROR: {e}")

# Test stress extraction
print()
print("2. Stress extraction (CTETRA):")
try:
    result = extract_solid_stress(op2_path, element_type="ctetra")
    print(f"   Max von Mises: {result.get('max_von_mises', 'N/A')} MPa")
    print(f"   Full result: {result}")
except Exception as e:
    print(f"   ERROR: {e}")

print()
print("Done!")