1224 lines
45 KiB
Python
1224 lines
45 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
Atomizer Optical Performance Report Generator
|
|||
|
|
===============================================
|
|||
|
|
|
|||
|
|
Generates a comprehensive, CDR-ready HTML report for the optical
|
|||
|
|
performance of an M1 mirror design from FEA results (OP2 file).
|
|||
|
|
|
|||
|
|
The report combines:
|
|||
|
|
1. Executive Summary with pass/fail vs design targets
|
|||
|
|
2. Per-Angle Wavefront Error Analysis (3D surface plots)
|
|||
|
|
3. Zernike Trajectory Analysis (mode-specific metrics across elevation)
|
|||
|
|
4. Sensitivity Matrix (axial vs lateral load response)
|
|||
|
|
5. Manufacturing Analysis (90° correction metrics)
|
|||
|
|
6. Full Zernike coefficient tables
|
|||
|
|
|
|||
|
|
Usage:
|
|||
|
|
conda activate atomizer
|
|||
|
|
python generate_optical_report.py "path/to/solution.op2"
|
|||
|
|
|
|||
|
|
# With annular aperture
|
|||
|
|
python generate_optical_report.py "path/to/solution.op2" --inner-radius 135.75
|
|||
|
|
|
|||
|
|
# Custom targets
|
|||
|
|
python generate_optical_report.py "path/to/solution.op2" --target-40 4.0 --target-60 10.0 --target-mfg 20.0
|
|||
|
|
|
|||
|
|
# Include design parameters from study database
|
|||
|
|
python generate_optical_report.py "path/to/solution.op2" --study-db "path/to/study.db" --trial 725
|
|||
|
|
|
|||
|
|
Output:
|
|||
|
|
Creates a single comprehensive HTML file:
|
|||
|
|
{basename}_OPTICAL_REPORT_{timestamp}.html
|
|||
|
|
|
|||
|
|
Author: Atomizer / Atomaste
|
|||
|
|
Created: 2026-01-29
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import sys
|
|||
|
|
import os
|
|||
|
|
import argparse
|
|||
|
|
import json
|
|||
|
|
from pathlib import Path
|
|||
|
|
from math import factorial
|
|||
|
|
from datetime import datetime
|
|||
|
|
import numpy as np
|
|||
|
|
from numpy.linalg import LinAlgError
|
|||
|
|
|
|||
|
|
# Add Atomizer root to path
|
|||
|
|
ATOMIZER_ROOT = Path(__file__).parent.parent
|
|||
|
|
if str(ATOMIZER_ROOT) not in sys.path:
|
|||
|
|
sys.path.insert(0, str(ATOMIZER_ROOT))
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
import plotly.graph_objects as go
|
|||
|
|
from plotly.subplots import make_subplots
|
|||
|
|
from matplotlib.tri import Triangulation
|
|||
|
|
from pyNastran.op2.op2 import OP2
|
|||
|
|
from pyNastran.bdf.bdf import BDF
|
|||
|
|
except ImportError as e:
|
|||
|
|
print(f"ERROR: Missing dependency: {e}")
|
|||
|
|
print("Run: conda activate atomizer")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
# Import Atomizer extractors
|
|||
|
|
from optimization_engine.extractors.extract_zernike import (
|
|||
|
|
compute_zernike_coefficients,
|
|||
|
|
compute_rms_metrics,
|
|||
|
|
compute_aberration_magnitudes,
|
|||
|
|
compute_rms_with_custom_filter,
|
|||
|
|
zernike_noll,
|
|||
|
|
zernike_label,
|
|||
|
|
zernike_name,
|
|||
|
|
noll_indices,
|
|||
|
|
read_node_geometry,
|
|||
|
|
find_geometry_file,
|
|||
|
|
extract_displacements_by_subcase,
|
|||
|
|
UNIT_TO_NM,
|
|||
|
|
DEFAULT_N_MODES,
|
|||
|
|
DEFAULT_FILTER_ORDERS,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
from optimization_engine.extractors.extract_zernike_trajectory import (
|
|||
|
|
ZernikeTrajectoryExtractor,
|
|||
|
|
MODE_GROUPS,
|
|||
|
|
MODE_NAMES,
|
|||
|
|
compute_trajectory_params,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ============================================================================
|
|||
|
|
# Configuration
|
|||
|
|
# ============================================================================
|
|||
|
|
|
|||
|
|
N_MODES = 50
|
|||
|
|
FILTER_LOW_ORDERS = 4
|
|||
|
|
PLOT_DOWNSAMPLE = 12000
|
|||
|
|
COLORSCALE = 'Turbo'
|
|||
|
|
|
|||
|
|
# Default design targets (nm)
|
|||
|
|
DEFAULT_TARGETS = {
|
|||
|
|
'wfe_40_20': 4.0,
|
|||
|
|
'wfe_60_20': 10.0,
|
|||
|
|
'mfg_90': 20.0,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# Default annular aperture for M1 (271.5mm central hole diameter)
|
|||
|
|
DEFAULT_INNER_RADIUS = 135.75 # mm
|
|||
|
|
|
|||
|
|
DISP_UNIT = 'mm'
|
|||
|
|
NM_SCALE = UNIT_TO_NM[DISP_UNIT]
|
|||
|
|
|
|||
|
|
# Subcase mapping: subcase_id -> angle
|
|||
|
|
SUBCASE_ANGLE_MAP = {
|
|||
|
|
'1': 90, '2': 20, '3': 40, '4': 60,
|
|||
|
|
'90': 90, '20': 20, '40': 40, '60': 60,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ============================================================================
|
|||
|
|
# Data Extraction Helpers
|
|||
|
|
# ============================================================================
|
|||
|
|
|
|||
|
|
def load_study_params(db_path: str, trial_id: int = None) -> dict:
|
|||
|
|
"""Load design parameters from study database."""
|
|||
|
|
import sqlite3
|
|||
|
|
conn = sqlite3.connect(db_path)
|
|||
|
|
c = conn.cursor()
|
|||
|
|
|
|||
|
|
if trial_id is None:
|
|||
|
|
# Find best trial by weighted sum
|
|||
|
|
c.execute('''
|
|||
|
|
SELECT t.trial_id, tua.key, tua.value_json
|
|||
|
|
FROM trials t
|
|||
|
|
JOIN trial_user_attributes tua ON t.trial_id = tua.trial_id
|
|||
|
|
WHERE t.state = 'COMPLETE' AND tua.key = 'weighted_sum'
|
|||
|
|
ORDER BY CAST(tua.value_json AS REAL) ASC
|
|||
|
|
LIMIT 1
|
|||
|
|
''')
|
|||
|
|
row = c.fetchone()
|
|||
|
|
if row:
|
|||
|
|
trial_id = row[0]
|
|||
|
|
else:
|
|||
|
|
conn.close()
|
|||
|
|
return {}
|
|||
|
|
|
|||
|
|
# Get all attributes for the trial
|
|||
|
|
c.execute('''
|
|||
|
|
SELECT key, value_json
|
|||
|
|
FROM trial_user_attributes
|
|||
|
|
WHERE trial_id = ?
|
|||
|
|
''', (trial_id,))
|
|||
|
|
attrs = {row[0]: json.loads(row[1]) for row in c.fetchall()}
|
|||
|
|
|
|||
|
|
# Get parameters
|
|||
|
|
c.execute('''
|
|||
|
|
SELECT tp.key, tp.value_json
|
|||
|
|
FROM trial_params tp
|
|||
|
|
WHERE tp.trial_id = ?
|
|||
|
|
''', (trial_id,))
|
|||
|
|
params = {row[0]: json.loads(row[1]) for row in c.fetchall()}
|
|||
|
|
|
|||
|
|
conn.close()
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
'trial_id': trial_id,
|
|||
|
|
'attributes': attrs,
|
|||
|
|
'parameters': params,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def build_wfe_arrays(node_ids, disp, node_geo):
|
|||
|
|
"""Build X, Y, WFE arrays from displacement data."""
|
|||
|
|
X, Y, WFE = [], [], []
|
|||
|
|
for nid, vec in zip(node_ids, disp):
|
|||
|
|
geo = node_geo.get(int(nid))
|
|||
|
|
if geo is None:
|
|||
|
|
continue
|
|||
|
|
X.append(geo[0])
|
|||
|
|
Y.append(geo[1])
|
|||
|
|
WFE.append(vec[2] * 2.0 * NM_SCALE)
|
|||
|
|
return np.array(X), np.array(Y), np.array(WFE)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def compute_relative_wfe(X1, Y1, WFE1, nids1, X2, Y2, WFE2, nids2):
|
|||
|
|
"""Compute WFE1 - WFE2 for common nodes."""
|
|||
|
|
ref_map = {int(n): w for n, w in zip(nids2, WFE2)}
|
|||
|
|
Xr, Yr, Wr = [], [], []
|
|||
|
|
for nid, x, y, w in zip(nids1, X1, Y1, WFE1):
|
|||
|
|
nid = int(nid)
|
|||
|
|
if nid in ref_map:
|
|||
|
|
Xr.append(x)
|
|||
|
|
Yr.append(y)
|
|||
|
|
Wr.append(w - ref_map[nid])
|
|||
|
|
return np.array(Xr), np.array(Yr), np.array(Wr)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def zernike_fit(X, Y, W, n_modes=N_MODES, inner_radius=None):
|
|||
|
|
"""Compute Zernike fit with optional annular masking."""
|
|||
|
|
Xc = X - np.mean(X)
|
|||
|
|
Yc = Y - np.mean(Y)
|
|||
|
|
R_outer = float(np.max(np.hypot(Xc, Yc)))
|
|||
|
|
r = np.hypot(Xc, Yc) / R_outer
|
|||
|
|
th = np.arctan2(Yc, Xc)
|
|||
|
|
|
|||
|
|
# Annular mask
|
|||
|
|
if inner_radius is not None:
|
|||
|
|
r_inner_norm = inner_radius / R_outer
|
|||
|
|
mask = (r >= r_inner_norm) & (r <= 1.0) & ~np.isnan(W)
|
|||
|
|
else:
|
|||
|
|
mask = (r <= 1.0) & ~np.isnan(W)
|
|||
|
|
|
|||
|
|
idx = np.nonzero(mask)[0]
|
|||
|
|
m = int(n_modes)
|
|||
|
|
G = np.zeros((m, m), dtype=np.float64)
|
|||
|
|
h = np.zeros((m,), dtype=np.float64)
|
|||
|
|
v = W.astype(np.float64)
|
|||
|
|
|
|||
|
|
for start in range(0, len(idx), 100000):
|
|||
|
|
sl = idx[start:start+100000]
|
|||
|
|
Zb = np.column_stack([zernike_noll(j, r[sl].astype(np.float32), th[sl].astype(np.float32)).astype(np.float32)
|
|||
|
|
for j in range(1, m+1)])
|
|||
|
|
G += (Zb.T @ Zb).astype(np.float64)
|
|||
|
|
h += (Zb.T @ v[sl]).astype(np.float64)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
coeffs = np.linalg.solve(G, h)
|
|||
|
|
except LinAlgError:
|
|||
|
|
coeffs = np.linalg.lstsq(G, h, rcond=None)[0]
|
|||
|
|
|
|||
|
|
# Compute residuals
|
|||
|
|
Z_all = np.column_stack([zernike_noll(j, r.astype(np.float32), th.astype(np.float32))
|
|||
|
|
for j in range(1, m+1)])
|
|||
|
|
W_low4 = Z_all[:, :FILTER_LOW_ORDERS].dot(coeffs[:FILTER_LOW_ORDERS])
|
|||
|
|
W_low3 = Z_all[:, :3].dot(coeffs[:3])
|
|||
|
|
W_res_j4 = W - W_low4 # J1-J4 removed
|
|||
|
|
W_res_j3 = W - W_low3 # J1-J3 removed
|
|||
|
|
|
|||
|
|
global_rms = float(np.sqrt(np.mean(W[mask]**2)))
|
|||
|
|
filtered_rms = float(np.sqrt(np.mean(W_res_j4[mask]**2)))
|
|||
|
|
rms_j1to3 = float(np.sqrt(np.mean(W_res_j3[mask]**2)))
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
'coefficients': coeffs,
|
|||
|
|
'R_outer': R_outer,
|
|||
|
|
'global_rms': global_rms,
|
|||
|
|
'filtered_rms': filtered_rms,
|
|||
|
|
'rms_j1to3': rms_j1to3,
|
|||
|
|
'W_res_filt': W_res_j4,
|
|||
|
|
'mask': mask,
|
|||
|
|
'n_masked': int(np.sum(mask)),
|
|||
|
|
'n_total': len(W),
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def aberration_magnitudes(coeffs):
|
|||
|
|
"""Get individual aberration magnitudes from Zernike coefficients."""
|
|||
|
|
defocus = float(abs(coeffs[3]))
|
|||
|
|
astig = float(np.sqrt(coeffs[4]**2 + coeffs[5]**2))
|
|||
|
|
coma = float(np.sqrt(coeffs[6]**2 + coeffs[7]**2))
|
|||
|
|
trefoil = float(np.sqrt(coeffs[8]**2 + coeffs[9]**2))
|
|||
|
|
spherical = float(abs(coeffs[10])) if len(coeffs) > 10 else 0.0
|
|||
|
|
return {
|
|||
|
|
'defocus': defocus, 'astigmatism': astig, 'coma': coma,
|
|||
|
|
'trefoil': trefoil, 'spherical': spherical,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ============================================================================
|
|||
|
|
# HTML Report Generation
|
|||
|
|
# ============================================================================
|
|||
|
|
|
|||
|
|
def status_badge(value, target, unit='nm'):
|
|||
|
|
"""Return pass/fail badge HTML."""
|
|||
|
|
if value <= target:
|
|||
|
|
return f'<span class="badge pass">✅ {value:.2f} {unit} ≤ {target:.1f}</span>'
|
|||
|
|
ratio = value / target
|
|||
|
|
if ratio < 1.5:
|
|||
|
|
return f'<span class="badge warn">⚠️ {value:.2f} {unit} ({ratio:.1f}× target)</span>'
|
|||
|
|
return f'<span class="badge fail">❌ {value:.2f} {unit} ({ratio:.1f}× target)</span>'
|
|||
|
|
|
|||
|
|
|
|||
|
|
def make_surface_plot(X, Y, W_res, mask, inner_radius=None, title="", amp=0.5, downsample=PLOT_DOWNSAMPLE):
|
|||
|
|
"""Create a 3D surface plot of residual WFE."""
|
|||
|
|
Xm, Ym, Wm = X[mask], Y[mask], W_res[mask]
|
|||
|
|
|
|||
|
|
n = len(Xm)
|
|||
|
|
if n > downsample:
|
|||
|
|
rng = np.random.default_rng(42)
|
|||
|
|
sel = rng.choice(n, size=downsample, replace=False)
|
|||
|
|
Xp, Yp, Wp = Xm[sel], Ym[sel], Wm[sel]
|
|||
|
|
else:
|
|||
|
|
Xp, Yp, Wp = Xm, Ym, Wm
|
|||
|
|
|
|||
|
|
res_amp = amp * Wp
|
|||
|
|
max_amp = float(np.max(np.abs(res_amp))) if res_amp.size else 1.0
|
|||
|
|
|
|||
|
|
traces = []
|
|||
|
|
try:
|
|||
|
|
tri = Triangulation(Xp, Yp)
|
|||
|
|
if tri.triangles is not None and len(tri.triangles) > 0:
|
|||
|
|
# Filter triangles spanning central hole
|
|||
|
|
if inner_radius is not None:
|
|||
|
|
cx, cy = np.mean(X), np.mean(Y)
|
|||
|
|
valid = []
|
|||
|
|
for t in tri.triangles:
|
|||
|
|
vx = Xp[t] - cx
|
|||
|
|
vy = Yp[t] - cy
|
|||
|
|
vr = np.hypot(vx, vy)
|
|||
|
|
if np.any(vr < inner_radius * 0.9):
|
|||
|
|
continue
|
|||
|
|
p0, p1, p2 = Xp[t] + 1j*Yp[t]
|
|||
|
|
if max(abs(p1-p0), abs(p2-p1), abs(p0-p2)) > 2*inner_radius:
|
|||
|
|
continue
|
|||
|
|
valid.append(t)
|
|||
|
|
if valid:
|
|||
|
|
tri_arr = np.array(valid)
|
|||
|
|
else:
|
|||
|
|
tri_arr = tri.triangles
|
|||
|
|
else:
|
|||
|
|
tri_arr = tri.triangles
|
|||
|
|
|
|||
|
|
i, j, k = tri_arr.T
|
|||
|
|
traces.append(go.Mesh3d(
|
|||
|
|
x=Xp, y=Yp, z=res_amp,
|
|||
|
|
i=i, j=j, k=k,
|
|||
|
|
intensity=res_amp,
|
|||
|
|
colorscale=COLORSCALE,
|
|||
|
|
opacity=1.0,
|
|||
|
|
flatshading=False,
|
|||
|
|
lighting=dict(ambient=0.4, diffuse=0.8, specular=0.3, roughness=0.5, fresnel=0.2),
|
|||
|
|
lightposition=dict(x=100, y=200, z=300),
|
|||
|
|
showscale=True,
|
|||
|
|
colorbar=dict(title=dict(text="nm", side="right"), thickness=12, len=0.5, tickformat=".1f"),
|
|||
|
|
hovertemplate="X: %{x:.1f}<br>Y: %{y:.1f}<br>Residual: %{z:.2f} nm<extra></extra>"
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
# Inner hole circle
|
|||
|
|
if inner_radius:
|
|||
|
|
theta_c = np.linspace(0, 2*np.pi, 80)
|
|||
|
|
traces.append(go.Scatter3d(
|
|||
|
|
x=cx + inner_radius*np.cos(theta_c),
|
|||
|
|
y=cy + inner_radius*np.sin(theta_c),
|
|||
|
|
z=np.zeros(80),
|
|||
|
|
mode='lines', line=dict(color='white', width=2),
|
|||
|
|
name='Central Hole', showlegend=False, hoverinfo='name'
|
|||
|
|
))
|
|||
|
|
except Exception:
|
|||
|
|
traces.append(go.Scatter3d(
|
|||
|
|
x=Xp, y=Yp, z=res_amp,
|
|||
|
|
mode='markers', marker=dict(size=2, color=res_amp, colorscale=COLORSCALE, showscale=True),
|
|||
|
|
showlegend=False
|
|||
|
|
))
|
|||
|
|
|
|||
|
|
fig = go.Figure(data=traces)
|
|||
|
|
fig.update_layout(
|
|||
|
|
scene=dict(
|
|||
|
|
camera=dict(eye=dict(x=1.2, y=1.2, z=0.8), up=dict(x=0, y=0, z=1)),
|
|||
|
|
xaxis=dict(title="X (mm)", showgrid=True, gridcolor='rgba(128,128,128,0.3)',
|
|||
|
|
showbackground=True, backgroundcolor='rgba(240,240,240,0.9)'),
|
|||
|
|
yaxis=dict(title="Y (mm)", showgrid=True, gridcolor='rgba(128,128,128,0.3)',
|
|||
|
|
showbackground=True, backgroundcolor='rgba(240,240,240,0.9)'),
|
|||
|
|
zaxis=dict(title="Residual (nm)", range=[-max_amp*3.0, max_amp*3.0],
|
|||
|
|
showgrid=True, gridcolor='rgba(128,128,128,0.3)',
|
|||
|
|
showbackground=True, backgroundcolor='rgba(230,230,250,0.9)'),
|
|||
|
|
aspectmode='manual',
|
|||
|
|
aspectratio=dict(x=1, y=1, z=0.4),
|
|||
|
|
),
|
|||
|
|
margin=dict(t=30, b=10, l=10, r=10),
|
|||
|
|
paper_bgcolor='rgba(0,0,0,0)',
|
|||
|
|
plot_bgcolor='rgba(0,0,0,0)',
|
|||
|
|
font=dict(color='#e0e0e0'),
|
|||
|
|
height=500,
|
|||
|
|
width=700,
|
|||
|
|
)
|
|||
|
|
return fig.to_html(include_plotlyjs=False, full_html=False, div_id=f"surface_{title.replace(' ','_')}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def make_bar_chart(coeffs, title="Zernike Coefficients", max_modes=30):
|
|||
|
|
"""Create horizontal bar chart of Zernike coefficient magnitudes."""
|
|||
|
|
n = min(len(coeffs), max_modes)
|
|||
|
|
labels = [zernike_label(j) for j in range(1, n+1)]
|
|||
|
|
vals = np.abs(coeffs[:n])
|
|||
|
|
|
|||
|
|
fig = go.Figure(go.Bar(
|
|||
|
|
x=vals, y=labels, orientation='h',
|
|||
|
|
marker_color='#6366f1',
|
|||
|
|
hovertemplate="%{y}<br>|c| = %{x:.3f} nm<extra></extra>",
|
|||
|
|
))
|
|||
|
|
fig.update_layout(
|
|||
|
|
height=max(400, n*22),
|
|||
|
|
margin=dict(t=30, b=10, l=200, r=20),
|
|||
|
|
paper_bgcolor='rgba(0,0,0,0)',
|
|||
|
|
plot_bgcolor='rgba(17,24,39,0.8)',
|
|||
|
|
font=dict(color='#e0e0e0', size=10),
|
|||
|
|
xaxis=dict(title="|Coefficient| (nm)", gridcolor='rgba(128,128,128,0.2)'),
|
|||
|
|
yaxis=dict(autorange='reversed'),
|
|||
|
|
)
|
|||
|
|
return fig.to_html(include_plotlyjs=False, full_html=False, div_id=f"bar_{title.replace(' ','_')}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def make_trajectory_plot(angles, coefficients_relative, mode_groups, sensitivity, title=""):
|
|||
|
|
"""Create trajectory visualization: Zernike modes vs elevation angle."""
|
|||
|
|
fig = go.Figure()
|
|||
|
|
|
|||
|
|
# Plot each mode group
|
|||
|
|
colors = ['#f59e0b', '#ef4444', '#10b981', '#6366f1', '#ec4899', '#14b8a6', '#f97316']
|
|||
|
|
color_idx = 0
|
|||
|
|
|
|||
|
|
for group_name, noll_indices in mode_groups.items():
|
|||
|
|
indices = [n - 5 for n in noll_indices if 5 <= n < 5 + coefficients_relative.shape[1]]
|
|||
|
|
if not indices:
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
# RSS of modes in this group at each angle
|
|||
|
|
rss = np.sqrt(np.sum(coefficients_relative[:, indices]**2, axis=1))
|
|||
|
|
color = colors[color_idx % len(colors)]
|
|||
|
|
|
|||
|
|
fig.add_trace(go.Scatter(
|
|||
|
|
x=angles, y=rss,
|
|||
|
|
mode='lines+markers',
|
|||
|
|
name=MODE_NAMES.get(group_name, group_name),
|
|||
|
|
line=dict(color=color, width=2),
|
|||
|
|
marker=dict(size=8),
|
|||
|
|
hovertemplate=f"{group_name}<br>%{{x}}°: %{{y:.2f}} nm<extra></extra>"
|
|||
|
|
))
|
|||
|
|
color_idx += 1
|
|||
|
|
|
|||
|
|
fig.update_layout(
|
|||
|
|
height=400,
|
|||
|
|
margin=dict(t=30, b=40, l=60, r=20),
|
|||
|
|
paper_bgcolor='rgba(0,0,0,0)',
|
|||
|
|
plot_bgcolor='rgba(17,24,39,0.8)',
|
|||
|
|
font=dict(color='#e0e0e0'),
|
|||
|
|
xaxis=dict(title="Elevation Angle (°)", gridcolor='rgba(128,128,128,0.2)',
|
|||
|
|
tickvals=angles, dtick=10),
|
|||
|
|
yaxis=dict(title="RMS (nm)", gridcolor='rgba(128,128,128,0.2)'),
|
|||
|
|
legend=dict(x=0.02, y=0.98, bgcolor='rgba(17,24,39,0.7)'),
|
|||
|
|
)
|
|||
|
|
return fig.to_html(include_plotlyjs=False, full_html=False, div_id="trajectory_plot")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def make_sensitivity_bar(sensitivity_dict):
|
|||
|
|
"""Create stacked bar chart of axial vs lateral sensitivity per mode."""
|
|||
|
|
modes = list(sensitivity_dict.keys())
|
|||
|
|
axial = [sensitivity_dict[m]['axial'] for m in modes]
|
|||
|
|
lateral = [sensitivity_dict[m]['lateral'] for m in modes]
|
|||
|
|
labels = [MODE_NAMES.get(m, m) for m in modes]
|
|||
|
|
|
|||
|
|
fig = go.Figure()
|
|||
|
|
fig.add_trace(go.Bar(
|
|||
|
|
y=labels, x=axial, orientation='h',
|
|||
|
|
name='Axial (sin θ)', marker_color='#f59e0b',
|
|||
|
|
hovertemplate="%{y}<br>Axial: %{x:.3f} nm/unit<extra></extra>"
|
|||
|
|
))
|
|||
|
|
fig.add_trace(go.Bar(
|
|||
|
|
y=labels, x=lateral, orientation='h',
|
|||
|
|
name='Lateral (cos θ)', marker_color='#6366f1',
|
|||
|
|
hovertemplate="%{y}<br>Lateral: %{x:.3f} nm/unit<extra></extra>"
|
|||
|
|
))
|
|||
|
|
fig.update_layout(
|
|||
|
|
barmode='group',
|
|||
|
|
height=max(300, len(modes)*40),
|
|||
|
|
margin=dict(t=30, b=40, l=200, r=20),
|
|||
|
|
paper_bgcolor='rgba(0,0,0,0)',
|
|||
|
|
plot_bgcolor='rgba(17,24,39,0.8)',
|
|||
|
|
font=dict(color='#e0e0e0', size=11),
|
|||
|
|
xaxis=dict(title="Sensitivity (nm per load fraction)", gridcolor='rgba(128,128,128,0.2)'),
|
|||
|
|
yaxis=dict(autorange='reversed'),
|
|||
|
|
legend=dict(x=0.6, y=0.98, bgcolor='rgba(17,24,39,0.7)'),
|
|||
|
|
)
|
|||
|
|
return fig.to_html(include_plotlyjs=False, full_html=False, div_id="sensitivity_bar")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def make_per_angle_rms_plot(angle_rms_data, ref_angle=20):
|
|||
|
|
"""Create bar chart of per-angle RMS relative to reference."""
|
|||
|
|
angles = sorted(angle_rms_data.keys())
|
|||
|
|
rms_vals = [angle_rms_data[a] for a in angles]
|
|||
|
|
labels = [f"{a}° vs {ref_angle}°" for a in angles]
|
|||
|
|
|
|||
|
|
fig = go.Figure(go.Bar(
|
|||
|
|
x=labels, y=rms_vals,
|
|||
|
|
marker_color=['#10b981' if v < 10 else '#f59e0b' if v < 20 else '#ef4444' for v in rms_vals],
|
|||
|
|
text=[f"{v:.2f} nm" for v in rms_vals],
|
|||
|
|
textposition='outside',
|
|||
|
|
hovertemplate="%{x}: %{y:.2f} nm<extra></extra>"
|
|||
|
|
))
|
|||
|
|
fig.update_layout(
|
|||
|
|
height=350,
|
|||
|
|
margin=dict(t=30, b=40, l=60, r=20),
|
|||
|
|
paper_bgcolor='rgba(0,0,0,0)',
|
|||
|
|
plot_bgcolor='rgba(17,24,39,0.8)',
|
|||
|
|
font=dict(color='#e0e0e0'),
|
|||
|
|
yaxis=dict(title="Filtered RMS WFE (nm)", gridcolor='rgba(128,128,128,0.2)'),
|
|||
|
|
)
|
|||
|
|
return fig.to_html(include_plotlyjs=False, full_html=False, div_id="per_angle_rms")
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ============================================================================
|
|||
|
|
# Main Report Builder
|
|||
|
|
# ============================================================================
|
|||
|
|
|
|||
|
|
def generate_report(
|
|||
|
|
op2_path: Path,
|
|||
|
|
inner_radius: float = None,
|
|||
|
|
targets: dict = None,
|
|||
|
|
study_db: str = None,
|
|||
|
|
trial_id: int = None,
|
|||
|
|
title: str = "M1 Mirror Optical Performance Report",
|
|||
|
|
study_name: str = None,
|
|||
|
|
) -> Path:
|
|||
|
|
"""Generate comprehensive optical performance HTML report."""
|
|||
|
|
|
|||
|
|
targets = targets or DEFAULT_TARGETS
|
|||
|
|
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M")
|
|||
|
|
ts_file = datetime.now().strftime("%Y%m%d_%H%M%S")
|
|||
|
|
|
|||
|
|
print("=" * 70)
|
|||
|
|
print(" ATOMIZER OPTICAL PERFORMANCE REPORT GENERATOR")
|
|||
|
|
print("=" * 70)
|
|||
|
|
print(f"\nOP2 File: {op2_path.name}")
|
|||
|
|
print(f"Inner Radius: {inner_radius} mm" if inner_radius else "Aperture: Full disk")
|
|||
|
|
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
# 1. Load geometry & displacement data
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
print("\n[1/5] Loading data...")
|
|||
|
|
geo_path = find_geometry_file(op2_path)
|
|||
|
|
node_geo = read_node_geometry(geo_path)
|
|||
|
|
print(f" Geometry: {geo_path.name} ({len(node_geo)} nodes)")
|
|||
|
|
|
|||
|
|
op2 = OP2()
|
|||
|
|
op2.read_op2(str(op2_path))
|
|||
|
|
displacements = extract_displacements_by_subcase(op2_path)
|
|||
|
|
print(f" Subcases: {list(displacements.keys())}")
|
|||
|
|
|
|||
|
|
# Map subcases to angles
|
|||
|
|
subcase_map = {}
|
|||
|
|
for angle in ['90', '20', '40', '60']:
|
|||
|
|
if angle in displacements:
|
|||
|
|
subcase_map[angle] = angle
|
|||
|
|
if len(subcase_map) < 4:
|
|||
|
|
if all(str(i) in displacements for i in range(1, 5)):
|
|||
|
|
subcase_map = {'90': '1', '20': '2', '40': '3', '60': '4'}
|
|||
|
|
print(f" Subcase map: {subcase_map}")
|
|||
|
|
|
|||
|
|
# Also detect intermediate angles (30, 50) if present
|
|||
|
|
extra_angles = []
|
|||
|
|
for a in ['30', '50']:
|
|||
|
|
if a in displacements:
|
|||
|
|
extra_angles.append(a)
|
|||
|
|
if extra_angles:
|
|||
|
|
print(f" Extra angles detected: {extra_angles}")
|
|||
|
|
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
# 2. Per-angle Zernike analysis
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
print("\n[2/5] Per-angle Zernike analysis...")
|
|||
|
|
|
|||
|
|
ref_label = subcase_map['20']
|
|||
|
|
ref_data = displacements[ref_label]
|
|||
|
|
X_ref, Y_ref, WFE_ref = build_wfe_arrays(ref_data['node_ids'], ref_data['disp'], node_geo)
|
|||
|
|
|
|||
|
|
# Analysis results storage
|
|||
|
|
angle_results = {} # angle -> {rms_data, X, Y, WFE, ...}
|
|||
|
|
|
|||
|
|
for angle_name, label in subcase_map.items():
|
|||
|
|
data = displacements[label]
|
|||
|
|
X, Y, WFE = build_wfe_arrays(data['node_ids'], data['disp'], node_geo)
|
|||
|
|
|
|||
|
|
# Absolute fit
|
|||
|
|
rms_abs = zernike_fit(X, Y, WFE, inner_radius=inner_radius)
|
|||
|
|
|
|||
|
|
# Relative fit (vs 20 deg reference)
|
|||
|
|
if angle_name != '20':
|
|||
|
|
Xr, Yr, Wr = compute_relative_wfe(
|
|||
|
|
X, Y, WFE, data['node_ids'],
|
|||
|
|
X_ref, Y_ref, WFE_ref, ref_data['node_ids']
|
|||
|
|
)
|
|||
|
|
rms_rel = zernike_fit(Xr, Yr, Wr, inner_radius=inner_radius)
|
|||
|
|
else:
|
|||
|
|
Xr, Yr, Wr = X, Y, np.zeros_like(WFE)
|
|||
|
|
rms_rel = {'filtered_rms': 0.0, 'rms_j1to3': 0.0, 'coefficients': np.zeros(N_MODES)}
|
|||
|
|
|
|||
|
|
angle_results[int(angle_name)] = {
|
|||
|
|
'X': X, 'Y': Y, 'WFE': WFE,
|
|||
|
|
'X_rel': Xr, 'Y_rel': Yr, 'WFE_rel': Wr,
|
|||
|
|
'rms_abs': rms_abs,
|
|||
|
|
'rms_rel': rms_rel,
|
|||
|
|
'aberrations_abs': aberration_magnitudes(rms_abs['coefficients']),
|
|||
|
|
'aberrations_rel': aberration_magnitudes(rms_rel['coefficients']) if angle_name != '20' else None,
|
|||
|
|
}
|
|||
|
|
print(f" {angle_name}° - Abs Filt: {rms_abs['filtered_rms']:.2f} nm, "
|
|||
|
|
f"Rel Filt: {rms_rel['filtered_rms']:.2f} nm")
|
|||
|
|
|
|||
|
|
# Extra angles (30, 50)
|
|||
|
|
for ea in extra_angles:
|
|||
|
|
data = displacements[ea]
|
|||
|
|
X, Y, WFE = build_wfe_arrays(data['node_ids'], data['disp'], node_geo)
|
|||
|
|
Xr, Yr, Wr = compute_relative_wfe(
|
|||
|
|
X, Y, WFE, data['node_ids'],
|
|||
|
|
X_ref, Y_ref, WFE_ref, ref_data['node_ids']
|
|||
|
|
)
|
|||
|
|
rms_abs = zernike_fit(X, Y, WFE, inner_radius=inner_radius)
|
|||
|
|
rms_rel = zernike_fit(Xr, Yr, Wr, inner_radius=inner_radius)
|
|||
|
|
angle_results[int(ea)] = {
|
|||
|
|
'X': X, 'Y': Y, 'WFE': WFE,
|
|||
|
|
'X_rel': Xr, 'Y_rel': Yr, 'WFE_rel': Wr,
|
|||
|
|
'rms_abs': rms_abs,
|
|||
|
|
'rms_rel': rms_rel,
|
|||
|
|
'aberrations_abs': aberration_magnitudes(rms_abs['coefficients']),
|
|||
|
|
'aberrations_rel': aberration_magnitudes(rms_rel['coefficients']),
|
|||
|
|
}
|
|||
|
|
print(f" {ea}° - Abs Filt: {rms_abs['filtered_rms']:.2f} nm, "
|
|||
|
|
f"Rel Filt: {rms_rel['filtered_rms']:.2f} nm")
|
|||
|
|
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
# 3. Trajectory analysis
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
print("\n[3/5] Trajectory analysis...")
|
|||
|
|
traj_result = None
|
|||
|
|
try:
|
|||
|
|
traj_extractor = ZernikeTrajectoryExtractor(
|
|||
|
|
op2_file=op2_path,
|
|||
|
|
bdf_file=geo_path,
|
|||
|
|
reference_angle=20.0,
|
|||
|
|
unit=DISP_UNIT,
|
|||
|
|
n_modes=N_MODES,
|
|||
|
|
inner_radius=inner_radius,
|
|||
|
|
)
|
|||
|
|
traj_result = traj_extractor.extract_trajectory(exclude_angles=[90.0])
|
|||
|
|
print(f" R² fit: {traj_result['linear_fit_r2']:.4f}")
|
|||
|
|
print(f" Dominant mode: {traj_result['dominant_mode']}")
|
|||
|
|
print(f" Total filtered RMS: {traj_result['total_filtered_rms_nm']:.2f} nm")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f" [WARN] Trajectory analysis failed: {e}")
|
|||
|
|
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
# 4. Manufacturing analysis
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
print("\n[4/5] Manufacturing analysis...")
|
|||
|
|
r90 = angle_results[90]
|
|||
|
|
mfg_abs_aberr = r90['aberrations_abs']
|
|||
|
|
mfg_correction = aberration_magnitudes(angle_results[90]['rms_rel']['coefficients'])
|
|||
|
|
mfg_rms_j1to3 = r90['rms_rel']['rms_j1to3']
|
|||
|
|
print(f" MFG 90 (J1-J3 filtered): {mfg_rms_j1to3:.2f} nm")
|
|||
|
|
print(f" Correction - Astigmatism: {mfg_correction['astigmatism']:.2f} nm, "
|
|||
|
|
f"Coma: {mfg_correction['coma']:.2f} nm")
|
|||
|
|
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
# 5. Load study params (optional)
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
study_params = None
|
|||
|
|
if study_db:
|
|||
|
|
print("\n[5/5] Loading study parameters...")
|
|||
|
|
try:
|
|||
|
|
study_params = load_study_params(study_db, trial_id)
|
|||
|
|
print(f" Trial #{study_params.get('trial_id', '?')}")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f" [WARN] Could not load study params: {e}")
|
|||
|
|
else:
|
|||
|
|
print("\n[5/5] No study database provided (skipping design parameters)")
|
|||
|
|
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
# Key metrics for executive summary
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
wfe_40_20 = angle_results[40]['rms_rel']['filtered_rms']
|
|||
|
|
wfe_60_20 = angle_results[60]['rms_rel']['filtered_rms']
|
|||
|
|
mfg_90 = mfg_rms_j1to3
|
|||
|
|
|
|||
|
|
# Weighted sum
|
|||
|
|
ws = 6*wfe_40_20 + 5*wfe_60_20 + 3*mfg_90
|
|||
|
|
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
# Generate HTML
|
|||
|
|
# ------------------------------------------------------------------
|
|||
|
|
print("\nGenerating HTML report...")
|
|||
|
|
|
|||
|
|
# Surface plots
|
|||
|
|
surf_40 = make_surface_plot(
|
|||
|
|
angle_results[40]['X_rel'], angle_results[40]['Y_rel'],
|
|||
|
|
angle_results[40]['rms_rel']['W_res_filt'], angle_results[40]['rms_rel']['mask'],
|
|||
|
|
inner_radius=inner_radius, title="40 vs 20"
|
|||
|
|
)
|
|||
|
|
surf_60 = make_surface_plot(
|
|||
|
|
angle_results[60]['X_rel'], angle_results[60]['Y_rel'],
|
|||
|
|
angle_results[60]['rms_rel']['W_res_filt'], angle_results[60]['rms_rel']['mask'],
|
|||
|
|
inner_radius=inner_radius, title="60 vs 20"
|
|||
|
|
)
|
|||
|
|
surf_90 = make_surface_plot(
|
|||
|
|
angle_results[90]['X'], angle_results[90]['Y'],
|
|||
|
|
angle_results[90]['rms_abs']['W_res_filt'], angle_results[90]['rms_abs']['mask'],
|
|||
|
|
inner_radius=inner_radius, title="90 abs"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# Bar charts
|
|||
|
|
bar_40 = make_bar_chart(angle_results[40]['rms_rel']['coefficients'], title="40v20 coeffs")
|
|||
|
|
bar_60 = make_bar_chart(angle_results[60]['rms_rel']['coefficients'], title="60v20 coeffs")
|
|||
|
|
bar_90 = make_bar_chart(angle_results[90]['rms_abs']['coefficients'], title="90abs coeffs")
|
|||
|
|
|
|||
|
|
# Per-angle RMS plot
|
|||
|
|
angle_rms_data = {}
|
|||
|
|
for ang in sorted(angle_results.keys()):
|
|||
|
|
if ang != 20:
|
|||
|
|
angle_rms_data[ang] = angle_results[ang]['rms_rel']['filtered_rms']
|
|||
|
|
per_angle_plot = make_per_angle_rms_plot(angle_rms_data)
|
|||
|
|
|
|||
|
|
# Trajectory & sensitivity plots
|
|||
|
|
traj_plot_html = ""
|
|||
|
|
sens_plot_html = ""
|
|||
|
|
if traj_result:
|
|||
|
|
coeffs_rel = np.array(traj_result['coefficients_relative'])
|
|||
|
|
traj_plot_html = make_trajectory_plot(
|
|||
|
|
traj_result['angles_deg'], coeffs_rel, MODE_GROUPS,
|
|||
|
|
traj_result['sensitivity_matrix']
|
|||
|
|
)
|
|||
|
|
sens_plot_html = make_sensitivity_bar(traj_result['sensitivity_matrix'])
|
|||
|
|
|
|||
|
|
# Design parameters table
|
|||
|
|
params_html = ""
|
|||
|
|
if study_params and study_params.get('parameters'):
|
|||
|
|
params = study_params['parameters']
|
|||
|
|
rows = ""
|
|||
|
|
for k, v in sorted(params.items()):
|
|||
|
|
unit = "°" if "angle" in k else "mm"
|
|||
|
|
rows += f"<tr><td>{k}</td><td>{v:.4f} {unit}</td></tr>\n"
|
|||
|
|
params_html = f"""
|
|||
|
|
<div class="section">
|
|||
|
|
<h2>🔧 Design Parameters (Trial #{study_params.get('trial_id', '?')})</h2>
|
|||
|
|
<table class="data-table"><thead><tr><th>Parameter</th><th>Value</th></tr></thead>
|
|||
|
|
<tbody>{rows}</tbody></table>
|
|||
|
|
</div>
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
# Per-angle detail table
|
|||
|
|
angle_detail_rows = ""
|
|||
|
|
for ang in sorted(angle_results.keys()):
|
|||
|
|
r = angle_results[ang]
|
|||
|
|
rel_filt = r['rms_rel']['filtered_rms']
|
|||
|
|
abs_filt = r['rms_abs']['filtered_rms']
|
|||
|
|
abs_glob = r['rms_abs']['global_rms']
|
|||
|
|
ab = r['aberrations_abs']
|
|||
|
|
angle_detail_rows += f"""<tr>
|
|||
|
|
<td><b>{ang}°</b></td>
|
|||
|
|
<td>{abs_glob:.2f}</td><td>{abs_filt:.2f}</td>
|
|||
|
|
<td>{rel_filt:.2f}</td>
|
|||
|
|
<td>{ab['astigmatism']:.2f}</td><td>{ab['coma']:.2f}</td>
|
|||
|
|
<td>{ab['trefoil']:.2f}</td><td>{ab['spherical']:.2f}</td>
|
|||
|
|
</tr>"""
|
|||
|
|
|
|||
|
|
# Trajectory metrics table
|
|||
|
|
traj_metrics_html = ""
|
|||
|
|
if traj_result:
|
|||
|
|
traj_metrics_html = f"""
|
|||
|
|
<div class="metrics-grid">
|
|||
|
|
<div class="metric-card">
|
|||
|
|
<div class="metric-label">Coma RMS</div>
|
|||
|
|
<div class="metric-value">{traj_result['coma_rms_nm']:.2f} nm</div>
|
|||
|
|
</div>
|
|||
|
|
<div class="metric-card">
|
|||
|
|
<div class="metric-label">Astigmatism RMS</div>
|
|||
|
|
<div class="metric-value">{traj_result['astigmatism_rms_nm']:.2f} nm</div>
|
|||
|
|
</div>
|
|||
|
|
<div class="metric-card">
|
|||
|
|
<div class="metric-label">Trefoil RMS</div>
|
|||
|
|
<div class="metric-value">{traj_result['trefoil_rms_nm']:.2f} nm</div>
|
|||
|
|
</div>
|
|||
|
|
<div class="metric-card">
|
|||
|
|
<div class="metric-label">Spherical RMS</div>
|
|||
|
|
<div class="metric-value">{traj_result['spherical_rms_nm']:.2f} nm</div>
|
|||
|
|
</div>
|
|||
|
|
<div class="metric-card">
|
|||
|
|
<div class="metric-label">Total Filtered RMS</div>
|
|||
|
|
<div class="metric-value">{traj_result['total_filtered_rms_nm']:.2f} nm</div>
|
|||
|
|
</div>
|
|||
|
|
<div class="metric-card">
|
|||
|
|
<div class="metric-label">Linear Fit R²</div>
|
|||
|
|
<div class="metric-value">{traj_result['linear_fit_r2']:.4f}</div>
|
|||
|
|
</div>
|
|||
|
|
</div>
|
|||
|
|
<p class="note">Dominant aberration mode: <b>{MODE_NAMES.get(traj_result['dominant_mode'], traj_result['dominant_mode'])}</b></p>
|
|||
|
|
<p class="note">Mode ranking: {' → '.join(traj_result['mode_ranking'][:5])}</p>
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
# Manufacturing details
|
|||
|
|
mfg_html = f"""
|
|||
|
|
<table class="data-table">
|
|||
|
|
<thead><tr><th>Metric</th><th>Absolute 90°</th><th>Correction (90°−20°)</th></tr></thead>
|
|||
|
|
<tbody>
|
|||
|
|
<tr><td>Defocus (J4)</td><td>{mfg_abs_aberr['defocus']:.2f} nm</td><td>{mfg_correction['defocus']:.2f} nm</td></tr>
|
|||
|
|
<tr><td>Astigmatism (J5+J6)</td><td>{mfg_abs_aberr['astigmatism']:.2f} nm</td><td>{mfg_correction['astigmatism']:.2f} nm</td></tr>
|
|||
|
|
<tr><td>Coma (J7+J8)</td><td>{mfg_abs_aberr['coma']:.2f} nm</td><td>{mfg_correction['coma']:.2f} nm</td></tr>
|
|||
|
|
<tr><td>Trefoil (J9+J10)</td><td>{mfg_abs_aberr['trefoil']:.2f} nm</td><td>{mfg_correction['trefoil']:.2f} nm</td></tr>
|
|||
|
|
<tr><td>Spherical (J11)</td><td>{mfg_abs_aberr['spherical']:.2f} nm</td><td>{mfg_correction['spherical']:.2f} nm</td></tr>
|
|||
|
|
<tr class="highlight"><td><b>J1−J3 Filtered RMS</b></td><td>{r90['rms_abs']['rms_j1to3']:.2f} nm</td><td><b>{mfg_rms_j1to3:.2f} nm</b></td></tr>
|
|||
|
|
</tbody>
|
|||
|
|
</table>
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
# Assemble full HTML
|
|||
|
|
html = f"""<!DOCTYPE html>
|
|||
|
|
<html lang="en">
|
|||
|
|
<head>
|
|||
|
|
<meta charset="UTF-8">
|
|||
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
|||
|
|
<title>{title}</title>
|
|||
|
|
<script src="https://cdn.plot.ly/plotly-2.27.0.min.js"></script>
|
|||
|
|
<style>
|
|||
|
|
:root {{
|
|||
|
|
--bg-primary: #0f172a;
|
|||
|
|
--bg-secondary: #1e293b;
|
|||
|
|
--bg-card: #1e293b;
|
|||
|
|
--text-primary: #f1f5f9;
|
|||
|
|
--text-secondary: #94a3b8;
|
|||
|
|
--accent: #6366f1;
|
|||
|
|
--accent-hover: #818cf8;
|
|||
|
|
--success: #10b981;
|
|||
|
|
--warning: #f59e0b;
|
|||
|
|
--danger: #ef4444;
|
|||
|
|
--border: #334155;
|
|||
|
|
}}
|
|||
|
|
* {{ margin: 0; padding: 0; box-sizing: border-box; }}
|
|||
|
|
body {{
|
|||
|
|
font-family: 'Inter', -apple-system, BlinkMacSystemFont, 'Segoe UI', system-ui, sans-serif;
|
|||
|
|
background: var(--bg-primary);
|
|||
|
|
color: var(--text-primary);
|
|||
|
|
line-height: 1.6;
|
|||
|
|
}}
|
|||
|
|
.container {{ max-width: 1400px; margin: 0 auto; padding: 2rem; }}
|
|||
|
|
|
|||
|
|
/* Header */
|
|||
|
|
.header {{
|
|||
|
|
background: linear-gradient(135deg, #1e293b 0%, #0f172a 100%);
|
|||
|
|
border: 1px solid var(--border);
|
|||
|
|
border-radius: 12px;
|
|||
|
|
padding: 2rem 3rem;
|
|||
|
|
margin-bottom: 2rem;
|
|||
|
|
display: flex;
|
|||
|
|
justify-content: space-between;
|
|||
|
|
align-items: center;
|
|||
|
|
}}
|
|||
|
|
.header h1 {{ font-size: 1.8rem; font-weight: 700; }}
|
|||
|
|
.header .subtitle {{ color: var(--text-secondary); font-size: 0.95rem; margin-top: 0.3rem; }}
|
|||
|
|
.header .branding {{
|
|||
|
|
text-align: right;
|
|||
|
|
font-size: 0.85rem;
|
|||
|
|
color: var(--text-secondary);
|
|||
|
|
}}
|
|||
|
|
.header .branding .logo {{ font-size: 1.4rem; font-weight: 700; color: var(--accent); }}
|
|||
|
|
|
|||
|
|
/* Sections */
|
|||
|
|
.section {{
|
|||
|
|
background: var(--bg-card);
|
|||
|
|
border: 1px solid var(--border);
|
|||
|
|
border-radius: 12px;
|
|||
|
|
padding: 1.5rem 2rem;
|
|||
|
|
margin-bottom: 1.5rem;
|
|||
|
|
}}
|
|||
|
|
.section h2 {{
|
|||
|
|
font-size: 1.3rem;
|
|||
|
|
margin-bottom: 1rem;
|
|||
|
|
padding-bottom: 0.5rem;
|
|||
|
|
border-bottom: 1px solid var(--border);
|
|||
|
|
}}
|
|||
|
|
|
|||
|
|
/* Executive Summary */
|
|||
|
|
.exec-grid {{
|
|||
|
|
display: grid;
|
|||
|
|
grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
|
|||
|
|
gap: 1rem;
|
|||
|
|
margin: 1rem 0;
|
|||
|
|
}}
|
|||
|
|
.exec-card {{
|
|||
|
|
background: var(--bg-primary);
|
|||
|
|
border: 1px solid var(--border);
|
|||
|
|
border-radius: 8px;
|
|||
|
|
padding: 1.2rem;
|
|||
|
|
}}
|
|||
|
|
.exec-card .label {{ font-size: 0.85rem; color: var(--text-secondary); margin-bottom: 0.3rem; }}
|
|||
|
|
.exec-card .value {{ font-size: 1.8rem; font-weight: 700; }}
|
|||
|
|
.exec-card .target {{ font-size: 0.8rem; color: var(--text-secondary); margin-top: 0.3rem; }}
|
|||
|
|
|
|||
|
|
/* Badges */
|
|||
|
|
.badge {{
|
|||
|
|
display: inline-block;
|
|||
|
|
padding: 0.2rem 0.6rem;
|
|||
|
|
border-radius: 4px;
|
|||
|
|
font-size: 0.85rem;
|
|||
|
|
font-weight: 600;
|
|||
|
|
}}
|
|||
|
|
.badge.pass {{ background: rgba(16,185,129,0.15); color: var(--success); }}
|
|||
|
|
.badge.warn {{ background: rgba(245,158,11,0.15); color: var(--warning); }}
|
|||
|
|
.badge.fail {{ background: rgba(239,68,68,0.15); color: var(--danger); }}
|
|||
|
|
|
|||
|
|
/* Tables */
|
|||
|
|
.data-table {{
|
|||
|
|
width: 100%;
|
|||
|
|
border-collapse: collapse;
|
|||
|
|
margin: 0.5rem 0;
|
|||
|
|
}}
|
|||
|
|
.data-table th, .data-table td {{
|
|||
|
|
padding: 0.6rem 1rem;
|
|||
|
|
text-align: left;
|
|||
|
|
border-bottom: 1px solid var(--border);
|
|||
|
|
}}
|
|||
|
|
.data-table th {{
|
|||
|
|
background: var(--bg-primary);
|
|||
|
|
font-weight: 600;
|
|||
|
|
font-size: 0.85rem;
|
|||
|
|
text-transform: uppercase;
|
|||
|
|
letter-spacing: 0.05em;
|
|||
|
|
color: var(--text-secondary);
|
|||
|
|
}}
|
|||
|
|
.data-table tr.highlight td {{
|
|||
|
|
background: rgba(99,102,241,0.08);
|
|||
|
|
font-weight: 600;
|
|||
|
|
}}
|
|||
|
|
|
|||
|
|
/* Metrics Grid */
|
|||
|
|
.metrics-grid {{
|
|||
|
|
display: grid;
|
|||
|
|
grid-template-columns: repeat(auto-fit, minmax(180px, 1fr));
|
|||
|
|
gap: 0.8rem;
|
|||
|
|
margin: 1rem 0;
|
|||
|
|
}}
|
|||
|
|
.metric-card {{
|
|||
|
|
background: var(--bg-primary);
|
|||
|
|
border: 1px solid var(--border);
|
|||
|
|
border-radius: 8px;
|
|||
|
|
padding: 1rem;
|
|||
|
|
text-align: center;
|
|||
|
|
}}
|
|||
|
|
.metric-label {{ font-size: 0.8rem; color: var(--text-secondary); margin-bottom: 0.3rem; }}
|
|||
|
|
.metric-value {{ font-size: 1.3rem; font-weight: 700; color: var(--accent); }}
|
|||
|
|
|
|||
|
|
/* Plots */
|
|||
|
|
.plot-grid {{
|
|||
|
|
display: grid;
|
|||
|
|
grid-template-columns: repeat(auto-fit, minmax(650px, 1fr));
|
|||
|
|
gap: 1rem;
|
|||
|
|
}}
|
|||
|
|
.plot-container {{
|
|||
|
|
background: var(--bg-primary);
|
|||
|
|
border: 1px solid var(--border);
|
|||
|
|
border-radius: 8px;
|
|||
|
|
padding: 1rem;
|
|||
|
|
}}
|
|||
|
|
.plot-container h3 {{
|
|||
|
|
font-size: 1rem;
|
|||
|
|
margin-bottom: 0.5rem;
|
|||
|
|
color: var(--text-secondary);
|
|||
|
|
}}
|
|||
|
|
|
|||
|
|
/* Collapsible */
|
|||
|
|
details {{ margin: 0.5rem 0; }}
|
|||
|
|
summary {{
|
|||
|
|
cursor: pointer;
|
|||
|
|
font-weight: 600;
|
|||
|
|
padding: 0.5rem;
|
|||
|
|
background: var(--bg-primary);
|
|||
|
|
border-radius: 6px;
|
|||
|
|
border: 1px solid var(--border);
|
|||
|
|
}}
|
|||
|
|
summary:hover {{ background: rgba(99,102,241,0.1); }}
|
|||
|
|
details > div {{ padding: 1rem; }}
|
|||
|
|
|
|||
|
|
.note {{ color: var(--text-secondary); font-size: 0.9rem; margin: 0.5rem 0; }}
|
|||
|
|
|
|||
|
|
/* Print styles */
|
|||
|
|
@media print {{
|
|||
|
|
body {{ background: white; color: black; }}
|
|||
|
|
.section {{ border: 1px solid #ccc; page-break-inside: avoid; }}
|
|||
|
|
}}
|
|||
|
|
|
|||
|
|
.two-col {{ display: grid; grid-template-columns: 1fr 1fr; gap: 1.5rem; }}
|
|||
|
|
@media (max-width: 900px) {{ .two-col {{ grid-template-columns: 1fr; }} }}
|
|||
|
|
</style>
|
|||
|
|
</head>
|
|||
|
|
<body>
|
|||
|
|
<div class="container">
|
|||
|
|
|
|||
|
|
<!-- Header -->
|
|||
|
|
<div class="header">
|
|||
|
|
<div>
|
|||
|
|
<h1>🔭 {title}</h1>
|
|||
|
|
<div class="subtitle">Generated {timestamp} | OP2: {op2_path.name}</div>
|
|||
|
|
{'<div class="subtitle">Study: ' + study_name + '</div>' if study_name else ''}
|
|||
|
|
</div>
|
|||
|
|
<div class="branding">
|
|||
|
|
<div class="logo">ATOMIZER</div>
|
|||
|
|
<div>by Atomaste</div>
|
|||
|
|
<div style="margin-top:0.3rem">FEA Optimization Platform</div>
|
|||
|
|
</div>
|
|||
|
|
</div>
|
|||
|
|
|
|||
|
|
<!-- Executive Summary -->
|
|||
|
|
<div class="section">
|
|||
|
|
<h2>📋 Executive Summary</h2>
|
|||
|
|
<div class="exec-grid">
|
|||
|
|
<div class="exec-card">
|
|||
|
|
<div class="label">WFE 40° vs 20° (Tracking)</div>
|
|||
|
|
<div class="value">{wfe_40_20:.2f} <small>nm</small></div>
|
|||
|
|
<div class="target">{status_badge(wfe_40_20, targets['wfe_40_20'])}</div>
|
|||
|
|
</div>
|
|||
|
|
<div class="exec-card">
|
|||
|
|
<div class="label">WFE 60° vs 20° (Tracking)</div>
|
|||
|
|
<div class="value">{wfe_60_20:.2f} <small>nm</small></div>
|
|||
|
|
<div class="target">{status_badge(wfe_60_20, targets['wfe_60_20'])}</div>
|
|||
|
|
</div>
|
|||
|
|
<div class="exec-card">
|
|||
|
|
<div class="label">MFG 90° (J1−J3 Filtered)</div>
|
|||
|
|
<div class="value">{mfg_90:.2f} <small>nm</small></div>
|
|||
|
|
<div class="target">{status_badge(mfg_90, targets['mfg_90'])}</div>
|
|||
|
|
</div>
|
|||
|
|
<div class="exec-card">
|
|||
|
|
<div class="label">Weighted Sum (6·W40 + 5·W60 + 3·MFG)</div>
|
|||
|
|
<div class="value" style="color: var(--accent)">{ws:.1f}</div>
|
|||
|
|
<div class="target">Lower is better</div>
|
|||
|
|
</div>
|
|||
|
|
</div>
|
|||
|
|
{'<p class="note">Annular aperture: inner radius = ' + f'{inner_radius:.1f} mm (ø{2*inner_radius:.1f} mm central hole)' + '</p>' if inner_radius else ''}
|
|||
|
|
</div>
|
|||
|
|
|
|||
|
|
<!-- Per-Angle Summary -->
|
|||
|
|
<div class="section">
|
|||
|
|
<h2>📊 Per-Angle RMS Summary</h2>
|
|||
|
|
{per_angle_plot}
|
|||
|
|
<table class="data-table" style="margin-top:1rem">
|
|||
|
|
<thead>
|
|||
|
|
<tr>
|
|||
|
|
<th>Angle</th><th>Abs Global RMS</th><th>Abs Filtered RMS</th>
|
|||
|
|
<th>Rel Filtered RMS</th>
|
|||
|
|
<th>Astigmatism</th><th>Coma</th><th>Trefoil</th><th>Spherical</th>
|
|||
|
|
</tr>
|
|||
|
|
</thead>
|
|||
|
|
<tbody>{angle_detail_rows}</tbody>
|
|||
|
|
</table>
|
|||
|
|
<p class="note">All values in nm. Filtered = J1−J4 removed. Relative = vs 20° reference. Aberrations are absolute.</p>
|
|||
|
|
</div>
|
|||
|
|
|
|||
|
|
<!-- Surface Plots -->
|
|||
|
|
<div class="section">
|
|||
|
|
<h2>🌊 Wavefront Error Surface Maps</h2>
|
|||
|
|
<p class="note">3D residual surfaces after removing piston, tip, tilt, and defocus (J1−J4). Interactive — drag to rotate.</p>
|
|||
|
|
<div class="plot-grid">
|
|||
|
|
<div class="plot-container">
|
|||
|
|
<h3>40° vs 20° (Relative)</h3>
|
|||
|
|
{surf_40}
|
|||
|
|
</div>
|
|||
|
|
<div class="plot-container">
|
|||
|
|
<h3>60° vs 20° (Relative)</h3>
|
|||
|
|
{surf_60}
|
|||
|
|
</div>
|
|||
|
|
</div>
|
|||
|
|
<div class="plot-container" style="margin-top:1rem">
|
|||
|
|
<h3>90° Manufacturing (Absolute)</h3>
|
|||
|
|
{surf_90}
|
|||
|
|
</div>
|
|||
|
|
</div>
|
|||
|
|
|
|||
|
|
<!-- Trajectory Analysis -->
|
|||
|
|
{'<div class="section"><h2>📈 Zernike Trajectory Analysis</h2>' +
|
|||
|
|
'<p class="note">Mode-specific integrated RMS across the operating elevation range. ' +
|
|||
|
|
'The linear model c<sub>j</sub>(θ) = a<sub>j</sub>·Δsinθ + b<sub>j</sub>·Δcosθ decomposes gravity into axial and lateral components.</p>' +
|
|||
|
|
traj_metrics_html +
|
|||
|
|
'<div class="two-col" style="margin-top:1rem">' +
|
|||
|
|
'<div class="plot-container"><h3>Mode RMS vs Elevation Angle</h3>' + traj_plot_html + '</div>' +
|
|||
|
|
'<div class="plot-container"><h3>Axial vs Lateral Sensitivity</h3>' + sens_plot_html + '</div>' +
|
|||
|
|
'</div></div>' if traj_result else ''}
|
|||
|
|
|
|||
|
|
<!-- Manufacturing Analysis -->
|
|||
|
|
<div class="section">
|
|||
|
|
<h2>🏭 Manufacturing Analysis (90° Orientation)</h2>
|
|||
|
|
<p class="note">
|
|||
|
|
The mirror is manufactured (polished) at 90° orientation. The "Correction" column shows the
|
|||
|
|
aberrations that must be polished out to achieve the 20° operational figure.
|
|||
|
|
</p>
|
|||
|
|
{mfg_html}
|
|||
|
|
</div>
|
|||
|
|
|
|||
|
|
<!-- Design Parameters -->
|
|||
|
|
{params_html}
|
|||
|
|
|
|||
|
|
<!-- Zernike Coefficient Details -->
|
|||
|
|
<div class="section">
|
|||
|
|
<h2>🔬 Zernike Coefficient Details</h2>
|
|||
|
|
<details>
|
|||
|
|
<summary>40° vs 20° — Relative Coefficients</summary>
|
|||
|
|
<div>{bar_40}</div>
|
|||
|
|
</details>
|
|||
|
|
<details>
|
|||
|
|
<summary>60° vs 20° — Relative Coefficients</summary>
|
|||
|
|
<div>{bar_60}</div>
|
|||
|
|
</details>
|
|||
|
|
<details>
|
|||
|
|
<summary>90° — Absolute Coefficients</summary>
|
|||
|
|
<div>{bar_90}</div>
|
|||
|
|
</details>
|
|||
|
|
</div>
|
|||
|
|
|
|||
|
|
<!-- Methodology -->
|
|||
|
|
<div class="section" style="opacity:0.85">
|
|||
|
|
<h2>📝 Methodology</h2>
|
|||
|
|
<table class="data-table">
|
|||
|
|
<tbody>
|
|||
|
|
<tr><td><b>Zernike Modes</b></td><td>{N_MODES} (Noll convention)</td></tr>
|
|||
|
|
<tr><td><b>Filtered Modes</b></td><td>J1−J4 (Piston, Tip, Tilt, Defocus)</td></tr>
|
|||
|
|
<tr><td><b>WFE Calculation</b></td><td>WFE = 2 × Surface Error (reflective)</td></tr>
|
|||
|
|
<tr><td><b>Displacement Unit</b></td><td>{DISP_UNIT} → nm ({NM_SCALE:.0e}×)</td></tr>
|
|||
|
|
<tr><td><b>Aperture</b></td><td>{'Annular (inner R = ' + f'{inner_radius:.1f} mm)' if inner_radius else 'Full disk'}</td></tr>
|
|||
|
|
<tr><td><b>Reference Angle</b></td><td>20° (polishing/measurement orientation)</td></tr>
|
|||
|
|
<tr><td><b>MFG Objective</b></td><td>90°−20° relative, J1−J3 filtered (optician workload)</td></tr>
|
|||
|
|
<tr><td><b>Weighted Sum</b></td><td>6×WFE(40−20) + 5×WFE(60−20) + 3×MFG(90)</td></tr>
|
|||
|
|
{'<tr><td><b>Trajectory R²</b></td><td>' + f'{traj_result["linear_fit_r2"]:.6f}' + '</td></tr>' if traj_result else ''}
|
|||
|
|
</tbody>
|
|||
|
|
</table>
|
|||
|
|
</div>
|
|||
|
|
|
|||
|
|
<!-- Footer -->
|
|||
|
|
<div style="text-align:center; padding:2rem; color:var(--text-secondary); font-size:0.8rem;">
|
|||
|
|
Generated by <b>Atomizer</b> Optical Report Generator | {timestamp}<br>
|
|||
|
|
© Atomaste | atomaste.ca
|
|||
|
|
</div>
|
|||
|
|
|
|||
|
|
</div>
|
|||
|
|
</body>
|
|||
|
|
</html>"""
|
|||
|
|
|
|||
|
|
# Write output
|
|||
|
|
output_path = op2_path.parent / f"{op2_path.stem}_OPTICAL_REPORT_{ts_file}.html"
|
|||
|
|
output_path.write_text(html, encoding='utf-8')
|
|||
|
|
|
|||
|
|
print(f"\n{'=' * 70}")
|
|||
|
|
print(f"REPORT GENERATED: {output_path.name}")
|
|||
|
|
print(f"{'=' * 70}")
|
|||
|
|
print(f"\nLocation: {output_path}")
|
|||
|
|
print(f"Size: {output_path.stat().st_size / 1024:.0f} KB")
|
|||
|
|
|
|||
|
|
return output_path
|
|||
|
|
|
|||
|
|
|
|||
|
|
# ============================================================================
|
|||
|
|
# CLI
|
|||
|
|
# ============================================================================
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
parser = argparse.ArgumentParser(
|
|||
|
|
description='Atomizer Optical Performance Report Generator',
|
|||
|
|
epilog='Generates a comprehensive CDR-ready HTML report from FEA results.'
|
|||
|
|
)
|
|||
|
|
parser.add_argument('op2_file', nargs='?', help='Path to OP2 results file')
|
|||
|
|
parser.add_argument('--inner-radius', '-r', type=float, default=None,
|
|||
|
|
help=f'Inner radius of central hole in mm (default: {DEFAULT_INNER_RADIUS}mm for M1)')
|
|||
|
|
parser.add_argument('--inner-diameter', '-d', type=float, default=None,
|
|||
|
|
help='Inner diameter of central hole in mm')
|
|||
|
|
parser.add_argument('--no-annular', action='store_true',
|
|||
|
|
help='Disable annular aperture (treat as full disk)')
|
|||
|
|
parser.add_argument('--target-40', type=float, default=DEFAULT_TARGETS['wfe_40_20'],
|
|||
|
|
help=f'WFE 40-20 target in nm (default: {DEFAULT_TARGETS["wfe_40_20"]})')
|
|||
|
|
parser.add_argument('--target-60', type=float, default=DEFAULT_TARGETS['wfe_60_20'],
|
|||
|
|
help=f'WFE 60-20 target in nm (default: {DEFAULT_TARGETS["wfe_60_20"]})')
|
|||
|
|
parser.add_argument('--target-mfg', type=float, default=DEFAULT_TARGETS['mfg_90'],
|
|||
|
|
help=f'MFG 90 target in nm (default: {DEFAULT_TARGETS["mfg_90"]})')
|
|||
|
|
parser.add_argument('--study-db', type=str, default=None,
|
|||
|
|
help='Path to study.db for design parameters')
|
|||
|
|
parser.add_argument('--trial', type=int, default=None,
|
|||
|
|
help='Trial ID (default: best trial)')
|
|||
|
|
parser.add_argument('--title', type=str, default="M1 Mirror Optical Performance Report",
|
|||
|
|
help='Report title')
|
|||
|
|
parser.add_argument('--study-name', type=str, default=None,
|
|||
|
|
help='Study name for report header')
|
|||
|
|
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
|
|||
|
|
# Find OP2 file
|
|||
|
|
if args.op2_file:
|
|||
|
|
op2_path = Path(args.op2_file)
|
|||
|
|
if not op2_path.exists():
|
|||
|
|
print(f"ERROR: File not found: {op2_path}")
|
|||
|
|
sys.exit(1)
|
|||
|
|
else:
|
|||
|
|
# Search current directory
|
|||
|
|
cwd = Path.cwd()
|
|||
|
|
candidates = list(cwd.glob("*solution*.op2")) + list(cwd.glob("*.op2"))
|
|||
|
|
if not candidates:
|
|||
|
|
print("ERROR: No OP2 file found. Specify path as argument.")
|
|||
|
|
sys.exit(1)
|
|||
|
|
op2_path = max(candidates, key=lambda p: p.stat().st_mtime)
|
|||
|
|
print(f"Found: {op2_path}")
|
|||
|
|
|
|||
|
|
# Handle inner radius
|
|||
|
|
inner_radius = DEFAULT_INNER_RADIUS # Default to M1 annular
|
|||
|
|
if args.no_annular:
|
|||
|
|
inner_radius = None
|
|||
|
|
elif args.inner_diameter is not None:
|
|||
|
|
inner_radius = args.inner_diameter / 2.0
|
|||
|
|
elif args.inner_radius is not None:
|
|||
|
|
inner_radius = args.inner_radius
|
|||
|
|
|
|||
|
|
targets = {
|
|||
|
|
'wfe_40_20': args.target_40,
|
|||
|
|
'wfe_60_20': args.target_60,
|
|||
|
|
'mfg_90': args.target_mfg,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
generate_report(
|
|||
|
|
op2_path=op2_path,
|
|||
|
|
inner_radius=inner_radius,
|
|||
|
|
targets=targets,
|
|||
|
|
study_db=args.study_db,
|
|||
|
|
trial_id=args.trial,
|
|||
|
|
title=args.title,
|
|||
|
|
study_name=args.study_name,
|
|||
|
|
)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"\nERROR: {e}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
main()
|