Compare commits
5 Commits
554b5418ee
...
e2c1a4580a
| Author | SHA1 | Date | |
|---|---|---|---|
|
e2c1a4580a | ||
|
|
0dfe6facd6 | ||
|
|
0f97c0b05b | ||
|
|
5b9626c8d5 | ||
|
|
4932a87fc1 |
@ -11,8 +11,8 @@
|
||||
"Grep",
|
||||
"Glob",
|
||||
"Bash(make test-math:*)",
|
||||
"Bash(ls -la /home/fmq/program/tlusty/tl208-s54/rust/*)",
|
||||
"Bash(wc -l /home/fmq/program/tlusty/tl208-s54/rust/*)"
|
||||
"Bash(ls -la /home/dckj/SpectraRust/*)",
|
||||
"Bash(wc -l /home/dckj/SpectraRust/*)"
|
||||
],
|
||||
"deny": [
|
||||
"Bash(rm -rf *)",
|
||||
@ -20,8 +20,9 @@
|
||||
"Bash(curl *)"
|
||||
],
|
||||
"additionalDirectories": [
|
||||
"/home/fmq/program/tlusty/tl208-s54/rust",
|
||||
"/home/fmq/program/tlusty/tl208-s54/tlusty"
|
||||
"/home/dckj/SpectraRust",
|
||||
"/home/dckj/SpectraRust/tlusty",
|
||||
"/home/dckj/SpectraRust/src"
|
||||
]
|
||||
}
|
||||
}
|
||||
20
.claude/settings.local.json
Normal file
20
.claude/settings.local.json
Normal file
@ -0,0 +1,20 @@
|
||||
{
|
||||
"permissions": {
|
||||
"allow": [
|
||||
"mcp__codegraph__codegraph_status",
|
||||
"mcp__plugin_oh-my-claudecode_t__state_read",
|
||||
"mcp__plugin_oh-my-claudecode_t__notepad_read",
|
||||
"mcp__codegraph__codegraph_search",
|
||||
"mcp__codegraph__codegraph_files",
|
||||
"mcp__codegraph__codegraph_explore",
|
||||
"mcp__codegraph__codegraph_node"
|
||||
]
|
||||
},
|
||||
"enableAllProjectMcpServers": true,
|
||||
"enabledMcpjsonServers": [
|
||||
"codegraph"
|
||||
],
|
||||
"enabledPlugins": {
|
||||
"oh-my-claudecode@omc": true
|
||||
}
|
||||
}
|
||||
@ -12,8 +12,7 @@ description: |
|
||||
# CodeGraph 辅助 F2R 重构
|
||||
|
||||
本项目已配置 CodeGraph MCP 服务器（`.mcp.json`），Claude 启动时自动加载。
|
||||
CodeGraph 索引了 Fortran 原始文件（`tlusty/tlusty208.f`、`synspec/synspec54.f`）和
|
||||
Rust 源码（`src/`），生成统一的 SQLite 代码图谱。
|
||||
不要进行全量测试，系统内存会被占满。
|
||||
|
||||
## MCP 工具
|
||||
|
||||
@ -34,164 +33,131 @@ Rust 源码（`src/`），生成统一的 SQLite 代码图谱。
|
||||
|
||||
### 函数命名：Fortran 与 Rust 完全对应
|
||||
|
||||
所有 TLUSTY Fortran 函数在 Rust 中都有**同名小写**版本。不再使用 `_pure` 后缀
|
||||
（已于 2026-06-05 批量清理）。
|
||||
所有 TLUSTY Fortran 函数在 Rust 中都有**同名小写**版本。
|
||||
|
||||
```
|
||||
Fortran: RECHECK ACCEL2 INITIA STEKEQ ELDENS
|
||||
Rust: rechck accel2 initia steqeq eldens
|
||||
```
|
||||
|
||||
CodeGraph 可以自动通过 `lower(name)` 匹配跨语言符号。
|
||||
|
||||
### `_pure` 后缀的例外（仅 9 个函数）
|
||||
|
||||
以下函数同时有 `_pure` 和非-pure 两个版本，两者用途不同：
|
||||
### `_pure` 后缀（仅 9 个函数）
|
||||
|
||||
| `_pure` 版本 | 非-pure 版本 | 关系 |
|
||||
|-------------|-------------|------|
|
||||
| `steqeq_pure` | `steqeq` | 纯计算内核 → 通过回调串联子程序的完整版本 |
|
||||
| `resolv_pure` | `resolv` | 纯线性化求解 → 串联 28 个子程序的完整版本 |
|
||||
| `start_pure` | `start` | 纯启动计算 → 带完整 I/O 的版本 |
|
||||
| `steqeq_pure` | `steqeq` | 纯计算内核 → 回调串联完整版本 |
|
||||
| `resolv_pure` | `resolv` | 纯线性化求解 → 28 子程序编排 |
|
||||
| `start_pure` | `start` | 纯启动计算 → 带 I/O 版本 |
|
||||
| `solve_pure` | `solve` | 纯矩阵求解 → 完整求解器 |
|
||||
| `inkul_pure` | `inkul` | 纯 Kurucz 谱线数据 → 带文件 I/O 的版本 |
|
||||
| `lemini_pure` | `lemini` | 纯 Lemke 表插值 → 带表查询的版本 |
|
||||
| `inkul_pure` | `inkul` | 纯 Kurucz 谱线 → 带文件 I/O |
|
||||
| `lemini_pure` | `lemini` | 纯 Lemke 插值 → 带表查询 |
|
||||
| `radtot_pure` | `radtot` | 纯辐射通量 → 完整辐射传输 |
|
||||
| `rayini_pure` | `rayini` | 纯瑞利散射初始化 → 带文件读取的版本 |
|
||||
| `iroset_pure` | `iroset` | 纯铁族元素设置 → 带回调的完整版本 |
|
||||
| `rayini_pure` | `rayini` | 纯瑞利散射 → 带文件读取 |
|
||||
| `iroset_pure` | `iroset` | 纯铁族设置 → 带回调完整版本 |
|
||||
|
||||
**规则**：`_pure` = 纯计算内核（可独立测试），非-pure = 完整编排包装器（匹配 Fortran 行为）。
|
||||
|
||||
## F2R 翻译工作流
|
||||
## 状态文件系统
|
||||
|
||||
### Step 0: 数据同步
|
||||
| 文件 | 用途 |
|
||||
|------|------|
|
||||
| `.f2r_phase` | 当前阶段：`translate` / `integrate` / `verify` / `done` |
|
||||
| `.f2r_tasks` | 当前阶段待办列表（每行一个，完成后加 ✅ 前缀） |
|
||||
| `.f2r_complete` | 存在 = 全部完成，脚本自动停止 |
|
||||
| `.f2r_rate_limit` | API 限流重置时间，脚本自动管理 |
|
||||
|
||||
每次 Rust 代码修改后，MCP 文件监视器会自动同步（2秒延迟）。如有疑问可手动触发：
|
||||
### 读取状态的规则
|
||||
|
||||
```bash
|
||||
cd /home/fmq/program/SpectraRust
|
||||
node /home/fmq/program/codegraph/dist/bin/codegraph.js sync
|
||||
```
|
||||
1. 启动时读取 `.f2r_phase` 确定阶段
|
||||
2. 读取 `.f2r_tasks` 取第一个未完成任务
|
||||
3. 完成后在 `.f2r_tasks` 中该任务行首加 ✅
|
||||
4. 全部完成后更新 `.f2r_phase` 并生成新 tasks
|
||||
|
||||
如果添加了新目录或数据异常，重建：
|
||||
```bash
|
||||
rm -rf .codegraph
|
||||
node /home/fmq/program/codegraph/dist/bin/codegraph.js init -i
|
||||
```
|
||||
## 参考文档（按需查阅）
|
||||
|
||||
### Step 1: 选择翻译目标
|
||||
| 阶段 | 文件 | 使用时机 |
|
||||
|------|------|---------|
|
||||
| Phase 1 翻译 | `references/phase1-translate.md` | 发现翻译遗漏时 |
|
||||
| Phase 3 验证 | `references/phase3-verify.md` | Phase 2 完成后 |
|
||||
|
||||
使用 `fortran-analyzer` skill 获取优先模块。然后用 CodeGraph 了解依赖：
|
||||
---
|
||||
|
||||
## 当前阶段：Phase 2 集成（integrate）
|
||||
|
||||
**目标**：将已翻译的纯计算函数连接为可运行的编排流程。
|
||||
|
||||
任务和工作流详见 `references/phase2-integrate.md`。
|
||||
|
||||
---
|
||||
|
||||
## 自动化模式（定时任务触发）
|
||||
|
||||
定时任务 `scripts/specf2r.sh` 通过 `--print` 触发本 skill。
|
||||
触发后必须立即按以下流程执行。
|
||||
|
||||
### 执行流程
|
||||
|
||||
```
|
||||
codegraph_explore "<目标函数> 的调用链和依赖" ← 一次性了解上游+下游
|
||||
codegraph_impact <目标函数> ← 了解修改影响范围
|
||||
Step 0: 读取状态
|
||||
→ 读取 .f2r_phase 确定阶段
|
||||
→ 读取 .f2r_tasks 取第一个未完成任务
|
||||
→ 没有未完成任务 → 更新阶段，生成新 tasks
|
||||
→ 没有更多阶段 → 创建 .f2r_complete
|
||||
|
||||
Step 1: 检查索引（仅确认健康，不扫描）
|
||||
→ codegraph_status
|
||||
|
||||
Step 2: 执行当前任务
|
||||
→ 分析目标函数/模块
|
||||
→ 实现修改
|
||||
→ 编译验证
|
||||
|
||||
Step 3: 更新状态
|
||||
→ 标记任务完成 ✅
|
||||
→ 取下一个任务继续
|
||||
```
|
||||
|
||||
**关键规则**：如果下游函数还没翻译，必须优先翻译它们。
|
||||
|
||||
### Step 2: 翻译函数
|
||||
|
||||
用 `codegraph_node <函数名>` 获取完整信息：
|
||||
- Fortran 源码（完整函数体）
|
||||
- 所在文件和行号
|
||||
- 签名、参数、返回值
|
||||
- 所有调用者和被调用者列表
|
||||
|
||||
对照 Fortran 源码逐行翻译。翻译后的 Rust 函数直接使用 Fortran 同名小写，
|
||||
例如 `ELDENS` → `pub fn eldens(...)`。
|
||||
|
||||
### Step 3: 验证调用链一致性
|
||||
|
||||
翻译完成后对比 Fortran 和 Rust 的调用链：
|
||||
### 规则
|
||||
|
||||
```
|
||||
codegraph_explore "<函数名> Fortran vs Rust 调用链对比"
|
||||
┌─────────────────────────────────────────────────────────────────┐
|
||||
│ ❌ 禁止询问用户"是否继续" │
|
||||
│ ❌ 禁止生成总结报告后停下 │
|
||||
│ ❌ 禁止重复验证"所有函数已翻译" │
|
||||
│ ❌ 禁止做无目标的全面扫描 │
|
||||
│ │
|
||||
│ ✅ 读取 .f2r_tasks → 执行第一项 → 验证 → 标记 → 下一项 │
|
||||
│ ✅ 只输出：做了什么 + 结果 │
|
||||
└─────────────────────────────────────────────────────────────────┘
|
||||
```
|
||||
|
||||
两边的被调用者列表应该结构一致（Rust 端用 snake_case，Fortran 端用 UPPER_CASE）。
|
||||
如果 Rust 端缺少被调用者 → 可能需要创建非-pure 编排包装器。
|
||||
|
||||
### Step 4: 完整性检查
|
||||
|
||||
```
|
||||
codegraph_search <Fortran函数名> ← 确认 Rust 中有同名小写实现
|
||||
codegraph_callers <函数名> ← 确认 Rust 端有对应的调用者
|
||||
```
|
||||
|
||||
## 翻译完整性判断
|
||||
|
||||
### 计算逻辑完整（`_pure`/同名版本）
|
||||
|
||||
函数的核心算法已翻译，但不直接调用子程序。占 TLUSTY 的绝大多数。
|
||||
|
||||
### 编排完整（非-pure 包装器）
|
||||
|
||||
函数不仅包含计算逻辑，还通过回调或直接调用来串联子程序，完整匹配 Fortran 行为。
|
||||
目前仅 9 个函数有此版本。
|
||||
|
||||
### 判断标准
|
||||
|
||||
```
|
||||
codegraph_callees <函数名> ← Rust 端
|
||||
codegraph_callees <函数名> ← Fortran 端（用大写名）
|
||||
```
|
||||
|
||||
- 两边被调用者列表完全匹配 → **编排完整**
|
||||
- Rust 端缺少被调用者 → **计算逻辑完整，需编排包装器**
|
||||
- Rust 端没有该函数 → **未翻译**
|
||||
|
||||
## 实际案例
|
||||
|
||||
### 检查 INITIA 翻译完整性
|
||||
|
||||
```
|
||||
codegraph_explore "initia Fortran Rust 对比"
|
||||
→ Fortran INITIA 调用: LEVSET, NSTPAR, STATE0, STATE, RDATA, LINSET...
|
||||
→ Rust initia 调用: generate_log_frequency_grid, init_reciprocal_powers...
|
||||
→ 结论: Rust 版本是简化实现，缺少大部分 Fortran 子程序调用
|
||||
→ 状态: 计算框架存在，需编排包装器
|
||||
```
|
||||
|
||||
### 快速了解函数被谁依赖
|
||||
|
||||
```
|
||||
codegraph_callers ELDENS
|
||||
→ Fortran: CONREF, ROSSOP, TEMPER, RHOEN, TRMDER (5 个)
|
||||
→ Rust: rybchn, rhonen, trmder, resolv (4 个)
|
||||
→ 差异: rossop 未调用 → 需检查 rossop 翻译状态
|
||||
```
|
||||
|
||||
### 查找遗漏
|
||||
|
||||
```
|
||||
codegraph_status → 先看总体统计
|
||||
codegraph_search "<逐个查>" → 确认 Fortran 名在 Rust 中是否存在
|
||||
```
|
||||
|
||||
### 评估修改影响
|
||||
|
||||
```
|
||||
codegraph_impact "steqeq" depth=2
|
||||
→ 修改 steqeq 会影响: hesolv, elcor, rybsol, steqeq_pure (12 个符号)
|
||||
→ 其中 2 个在 Fortran 端 (RYBSOL, TLUSTY), 10 个在 Rust 端
|
||||
```
|
||||
|
||||
## 当前翻译状态（2026-06-05）
|
||||
## 当前翻译状态（2026-06-08）
|
||||
|
||||
| 指标 | 数值 |
|
||||
|------|------|
|
||||
| TLUSTY Fortran 函数 | 350 |
|
||||
| TLUSTY Rust 匹配 | 350 (100%) |
|
||||
| 计算逻辑完整 | ~309 函数 |
|
||||
| 编排完整（非-pure 包装器） | 9 函数 |
|
||||
| RESOLV 编排进度 | 24/42 被调用者 (57%) |
|
||||
| 计算逻辑存在但缺编排 | 41 函数 |
|
||||
| SYNSPEC 待翻译 | 61 函数 |
|
||||
| TLUSTY Fortran 函数 | 350 (100% 翻译) |
|
||||
| SYNSPEC Fortran 函数 | 168 (100% 翻译) |
|
||||
| Rust 总模块数 | ~495 |
|
||||
| 编译 | ✅ 0 错误 |
|
||||
| 当前阶段 | **Phase 2: 集成** |
|
||||
|
||||
## 已知限制
|
||||
## 故障排查
|
||||
|
||||
1. **不跨语言语义映射**：CodeGraph 按名称匹配，不知道 Fortran `ELDENS` 和 Rust `eldens` 是同一个函数——但它支持大小写不敏感查询，所以这不是问题。
|
||||
2. **文件监视器 2 秒延迟**：修改 Rust 代码后等 2 秒再查询，或手动 `codegraph sync`。
|
||||
3. **提取文件可能有冗余**：`tlusty/extracted/` 和 `synspec/extracted/` 中的函数与原始文件重复（平均 2.2 次），MCP 工具查询时可能返回重复结果。优先信任原始文件（`tlusty/tlusty208.f`）中的结果。
|
||||
4. **Fortran 调用边覆盖率 ~98.3%**：约 1.7% 的调用（~22 条）来自语法解析错误的区域，caller 显示为 `<anonymous>`。
|
||||
| 问题 | 解决方案 |
|
||||
|------|---------|
|
||||
| MCP 工具无响应 | `/reload-plugins` |
|
||||
| 索引返回 0 文件 | 重建索引：`rm -rf .codegraph && node .../codegraph.js init -i` |
|
||||
| 查询结果为空 | `codegraph_search` 模糊搜索 |
|
||||
| 重复结果 | 优先信任 `tlusty/tlusty208.f` 原始文件 |
|
||||
|
||||
## 文件路径
|
||||
|
||||
| 内容 | 路径 |
|
||||
|------|------|
|
||||
| CodeGraph 索引 | `.codegraph/` |
|
||||
| CodeGraph 二进制 | `/home/dckj/program/codegraph/dist/bin/codegraph.js` |
|
||||
| MCP 配置 | `.mcp.json` |
|
||||
| Fortran 源码（原始） | `tlusty/tlusty208.f`、`synspec/synspec54.f` |
|
||||
| Fortran 源码（提取） | `tlusty/extracted/*.f`、`synspec/extracted/*.f` |
|
||||
| Rust 源码 | `src/tlusty/`、`src/synspec/` |
|
||||
| 定时任务脚本 | `scripts/specf2r.sh` |
|
||||
| 阶段状态 | `.f2r_phase`、`.f2r_tasks`、`.f2r_complete` |
|
||||
|
||||
@ -0,0 +1,86 @@
|
||||
# Phase 1: 翻译工作流参考
|
||||
|
||||
> 状态：✅ 已完成（2026-06-06 ~ 2026-06-07）
|
||||
> TLUSTY 350 函数 + SYNSPEC 168 函数 = 518 函数全部翻译为 Rust
|
||||
|
||||
此文件仅供参考。仅在发现翻译遗漏或需要翻译新函数时查阅。
|
||||
|
||||
## 翻译流程
|
||||
|
||||
### Step 0: 数据同步
|
||||
|
||||
CodeGraph 索引路径：`/home/dckj/SpectraRust/.codegraph/`
|
||||
|
||||
每次 Rust 代码修改后，MCP 文件监视器会自动同步（2秒延迟）。如有疑问可手动触发：
|
||||
|
||||
```bash
|
||||
cd /home/dckj/SpectraRust
|
||||
node /home/dckj/program/codegraph/dist/bin/codegraph.js sync
|
||||
```
|
||||
|
||||
如果添加了新目录或数据异常，重建索引：
|
||||
```bash
|
||||
rm -rf /home/dckj/SpectraRust/.codegraph
|
||||
cd /home/dckj/SpectraRust
|
||||
node /home/dckj/program/codegraph/dist/bin/codegraph.js init -i
|
||||
```
|
||||
|
||||
### Step 1: 选择翻译目标
|
||||
|
||||
使用 `fortran-analyzer` skill 获取优先模块。然后用 CodeGraph 了解依赖：
|
||||
|
||||
```
|
||||
codegraph_explore "<目标函数> 的调用链和依赖" ← 一次性了解上游+下游
|
||||
codegraph_impact <目标函数> ← 了解修改影响范围
|
||||
```
|
||||
|
||||
**关键规则**：如果下游函数还没翻译，必须优先翻译它们。
|
||||
|
||||
### Step 2: 翻译函数
|
||||
|
||||
用 `codegraph_node <函数名>` 获取完整信息：
|
||||
- Fortran 源码（完整函数体）
|
||||
- 所在文件和行号
|
||||
- 签名、参数、返回值
|
||||
- 所有调用者和被调用者列表
|
||||
|
||||
对照 Fortran 源码逐行翻译。翻译后的 Rust 函数直接使用 Fortran 同名小写，
|
||||
例如 `ELDENS` → `pub fn eldens(...)`。
|
||||
|
||||
### Step 3: 验证调用链一致性
|
||||
|
||||
翻译完成后对比 Fortran 和 Rust 的调用链：
|
||||
|
||||
```
|
||||
codegraph_explore "<函数名> Fortran vs Rust 调用链对比"
|
||||
```
|
||||
|
||||
两边的被调用者列表应该结构一致（Rust 端用 snake_case，Fortran 端用 UPPER_CASE）。
|
||||
如果 Rust 端缺少被调用者 → 可能需要创建非-pure 编排包装器。
|
||||
|
||||
### Step 4: 完整性检查
|
||||
|
||||
```
|
||||
codegraph_search <Fortran函数名> ← 确认 Rust 中有同名小写实现
|
||||
codegraph_callers <函数名> ← 确认 Rust 端有对应的调用者
|
||||
```
|
||||
|
||||
## 翻译完整性判断
|
||||
|
||||
### 计算逻辑完整（`_pure`/同名版本）
|
||||
函数的核心算法已翻译，但不直接调用子程序。占 TLUSTY 的绝大多数。
|
||||
|
||||
### 编排完整（非-pure 包装器）
|
||||
函数不仅包含计算逻辑，还通过回调或直接调用来串联子程序，完整匹配 Fortran 行为。
|
||||
目前仅 9 个函数有此版本。
|
||||
|
||||
### 判断标准
|
||||
|
||||
```
|
||||
codegraph_callees <函数名> ← Rust 端
|
||||
codegraph_callees <函数名> ← Fortran 端（用大写名）
|
||||
```
|
||||
|
||||
- 两边被调用者列表完全匹配 → **编排完整**
|
||||
- Rust 端缺少被调用者 → **计算逻辑完整，需编排包装器**
|
||||
- Rust 端没有该函数 → **未翻译**
|
||||
102
.claude/skills/codegraph-guide/references/phase2-integrate.md
Normal file
102
.claude/skills/codegraph-guide/references/phase2-integrate.md
Normal file
@ -0,0 +1,102 @@
|
||||
# Phase 2: 集成工作流参考
|
||||
|
||||
> 状态：当前活跃阶段
|
||||
> 目标：将已翻译的纯计算函数连接为可运行的编排流程
|
||||
|
||||
## 任务来源
|
||||
|
||||
从 `.f2r_tasks` 读取。当前主要任务方向：
|
||||
|
||||
### 1. TLUSTY RESOLV 编排补全 (`src/tlusty/io/resolv.rs`)
|
||||
|
||||
Resolv 是 TLUSTY 主循环的核心编排器，每个频率点调用一次。当前有 7 个 TODO：
|
||||
|
||||
| TODO 位置 | 内容 | 说明 |
|
||||
|-----------|------|------|
|
||||
| L81 | 原子数据丰度 | 从原子数据文件读取精确值替换硬编码 HHe 值 |
|
||||
| L2038 | ComputeArrays 传递 | 将 ComputeArrays 添加到 ResolvParams 或从调用方传入 |
|
||||
| L2146 | ComputeArrays 传入 | 同上，另一处调用点 |
|
||||
| L2251 | rru/rrd 累积 | 累积辐射率获得完整输出 |
|
||||
| L2439 | CoolrtParams 2D | 重构为 2D 接口获得精确冷却率 |
|
||||
| L2606 | 频率不透明度更新 | 按频率从 opacfl_data 更新不透明度 |
|
||||
| L2624 | rtecmu 频率循环 | 循环所有频率点调用 rtecmu+opacf1+taufr1 |
|
||||
|
||||
### 2. TLUSTY Runner (`src/tlusty/main.rs`)
|
||||
|
||||
| TODO 位置 | 内容 |
|
||||
|-----------|------|
|
||||
| L482 | 实现正确的 IJALI 频率选择（只用关键频率） |
|
||||
|
||||
### 3. TLUSTY OPFRAC (`src/tlusty/math/continuum/opfrac.rs`)
|
||||
|
||||
| TODO 位置 | 内容 |
|
||||
|-----------|------|
|
||||
| L309 | 解析 ioniz.dat 文件完整实现 |
|
||||
|
||||
### 4. SYNSPEC Runner (`src/synspec/runner.rs`)
|
||||
|
||||
连接所有编排步骤的参数传递，确保完整流程可运行：
|
||||
- CHANGE: 能级人口重分配
|
||||
- MOLINI: 分子平衡初始化
|
||||
- EOSPRI: EOS 参数诊断输出
|
||||
- ABNCHN: 丰度缩放
|
||||
- INGRID 网格模式完整流程
|
||||
- INMOLI 循环
|
||||
- IDMTAB 实际调用
|
||||
- FINGRD 最终输出
|
||||
|
||||
### 5. SYNSPEC RESOLV (`src/synspec/math/resolv.rs`)
|
||||
|
||||
- 构造完整的 ResolvParams 从模型数据
|
||||
- 填充 OPAC→RTE→OUTPRI 完整调用链
|
||||
|
||||
## 集成工作流（严格遵守）
|
||||
|
||||
```
|
||||
每次会话：
|
||||
1. 读取 .f2r_tasks → 取第一个未完成任务
|
||||
2. 读取任务对应的目标文件，定位 TODO
|
||||
3. 使用 codegraph 了解调用关系和依赖：
|
||||
codegraph_explore "<目标函数> 的调用链"
|
||||
codegraph_callees <目标函数>
|
||||
codegraph_callers <目标函数>
|
||||
4. ★ 必须先读取对应的 Fortran 源码，理解原始逻辑
|
||||
5. 实现修改，连接纯计算函数到编排流程
|
||||
6. 编译验证：
|
||||
RUSTFLAGS="-A warnings" cargo build 2>&1 | tail -5
|
||||
7. 编译失败 → 修复 → 重试
|
||||
8. 编译通过 → 在 .f2r_tasks 中标记 ✅ → 取下一个任务
|
||||
```
|
||||
|
||||
## ★ 核心原则
|
||||
|
||||
```
|
||||
1. 先读 Fortran 源码：每个 TODO 都对应 Fortran 中的具体逻辑
|
||||
2. 保持调用顺序：Fortran CALL 顺序必须严格保持
|
||||
3. 正确传递参数：COMMON 块变量 → Rust struct 字段映射正确
|
||||
4. 数组下标转换：1-based → 0-based
|
||||
5. 不能用空壳：回调/closure 必须调用实际函数
|
||||
6. 每步验证编译：修改后立即 cargo build
|
||||
```
|
||||
|
||||
## 编译验证
|
||||
|
||||
每次修改后：
|
||||
```bash
|
||||
RUSTFLAGS="-A warnings" cargo build 2>&1 | tail -5
|
||||
```
|
||||
|
||||
相关模块的单元测试：
|
||||
```bash
|
||||
cargo test --lib <模块名> 2>&1 | tail -3
|
||||
```
|
||||
|
||||
禁止全量测试，内存会被占满。
|
||||
|
||||
## 完成标准
|
||||
|
||||
1. `.f2r_tasks` 中所有任务标记 ✅
|
||||
2. `cargo build` 零错误
|
||||
3. 无 `TODO`/`FIXME` 遗留在生产代码中
|
||||
4. 更新 `.f2r_phase` 为 `verify`
|
||||
5. 生成 Phase 3 的 `.f2r_tasks`
|
||||
210
.claude/skills/codegraph-guide/references/phase3-verify.md
Normal file
210
.claude/skills/codegraph-guide/references/phase3-verify.md
Normal file
@ -0,0 +1,210 @@
|
||||
# Phase 3: 验证工作流参考
|
||||
|
||||
> 状态：待启动（Phase 2 集成完成后进入）
|
||||
> 模式：参照 `tlusty-iteration` skill 的逐模块严格验证流程
|
||||
|
||||
## 文件路径
|
||||
|
||||
| 内容 | 路径 |
|
||||
|------|------|
|
||||
| Fortran 源码 | `tlusty/extracted/*.f`、`synspec/extracted/*.f` |
|
||||
| Rust 源码 | `src/tlusty/`、`src/synspec/` |
|
||||
| 验证进度 | `.claude/skills/codegraph-guide/references/verify-progress.md` |
|
||||
| TLUSTY Fortran 测试 | `$TLUSTY/tests/tlusty/hhe/` |
|
||||
| SYNSPEC Fortran 测试 | `$TLUSTY/tests/synspec/hhe/` |
|
||||
| TLUSTY Rust 测试 | `tests/tlusty/hhe_rust/` |
|
||||
| SYNSPEC Rust 测试 | `tests/synspec/hhe/` |
|
||||
|
||||
## 测试方式
|
||||
|
||||
### TLUSTY 端到端
|
||||
|
||||
```bash
|
||||
# Fortran 参考
|
||||
cd $TLUSTY/tests/tlusty/hhe
|
||||
$TLUSTY/tlusty/tlusty.exe < hhe35lt.5 > hhe35lt.6
|
||||
cp fort.7 hhe35lt.7.ref
|
||||
|
||||
# Rust
|
||||
cargo build --bin tlusty
|
||||
cd tests/tlusty/hhe_rust
|
||||
rm -f fort.7
|
||||
../../../target/debug/tlusty < hhe35lt.5 > rust.6 2>stderr.txt
|
||||
|
||||
# 对比
|
||||
diff hhe35lt.7.ref fort.7
|
||||
```
|
||||
|
||||
### SYNSPEC 端到端
|
||||
|
||||
```bash
|
||||
# 准备（测试目录 tests/synspec/hhe/ 已有 fort.8、fort.55.con 等文件）
|
||||
cd tests/synspec/hhe
|
||||
cp hhe35nl.7 fort.8
|
||||
ln -sf fort.55.con fort.55
|
||||
|
||||
# Fortran 参考（生成 results_original/ 中的 .spec/.cont/.iden）
|
||||
# 需要先编译：gfortran -O3 -fno-automatic -mcmodel=large -o synspec.exe synspec54.f
|
||||
./synspec.exe < hhe35nl.5
|
||||
|
||||
# Rust
|
||||
cargo build --bin synspec
|
||||
cd tests/synspec/hhe
|
||||
rm -f fort.7
|
||||
../../../target/debug/synspec < hhe35nl.5 > rust.6 2>stderr.txt
|
||||
|
||||
# 对比（与 Fortran 参考结果比对）
|
||||
diff results_original/hhe35nl.spec fort.7
|
||||
```
|
||||
|
||||
## 验证工作流（严格遵守）
|
||||
|
||||
```
|
||||
每次会话：
|
||||
1. 读取 verify-progress.md → 恢复验证进度
|
||||
2. 运行 Rust → 与 Fortran 参考输出对比
|
||||
3. 输出完全一致 → 更新 verify-progress.md → 结束
|
||||
4. 输出不一致 → 从断点继续逐模块验证:
|
||||
a. 读取 verify-progress.md 中 "下一个待验证模块"
|
||||
b. ★ 必须先读取对应的 Fortran 文件，逐行理解原始逻辑
|
||||
c. 然后读取对应的 Rust 文件
|
||||
d. 逐行对比: 调用顺序、变量映射、索引转换、逻辑分支
|
||||
e. 发现差异 → 立即修复 → cargo build 验证
|
||||
f. 更新 verify-progress.md → 继续下一个模块
|
||||
5. 全部通过 → 运行测试套件 → 更新 verify-progress.md
|
||||
```
|
||||
|
||||
## ★ 核心原则：必须参考 Fortran 代码
|
||||
|
||||
```
|
||||
严禁凭猜测修改代码！每次修改前必须：
|
||||
1. 先读取对应的 Fortran 源码文件
|
||||
2. 理解 Fortran 的确切逻辑流程
|
||||
3. 找到 Fortran 中的对应行
|
||||
4. 然后对照修改 Rust 代码
|
||||
|
||||
违反此原则是产生 bug 的最主要原因。
|
||||
```
|
||||
|
||||
## 验证顺序
|
||||
|
||||
### TLUSTY 调用链
|
||||
|
||||
```
|
||||
TLUSTY (tlusty.f)
|
||||
→ START (start.f)
|
||||
→ INITIA (initia.f) ★ 最大模块
|
||||
→ HEDIF (hedif.f) [可选]
|
||||
→ COMSET (comset.f)
|
||||
→ PRDINI (prdini.f)
|
||||
→ RESOLV (resolv.f)
|
||||
→ INILAM, LINSEL, OPAINI ...
|
||||
→ OPACF0, OPACF1, RTEFR1 ...
|
||||
→ LUCY (lucy.f)
|
||||
→ OUTPUT
|
||||
→ ACCEL2 (accel2.f)
|
||||
→ SOLVE / SOLVES / RYBSOL
|
||||
→ MATGEN → BRTE, BHE, BRE
|
||||
→ MATINV
|
||||
```
|
||||
|
||||
### SYNSPEC 调用链
|
||||
|
||||
```
|
||||
SYNSPEC (synspec54.f)
|
||||
→ START
|
||||
→ INITIA → STATE0, RDATA
|
||||
→ INPMOD / INKUR
|
||||
→ TINT, INIMOD
|
||||
→ INILIN → read_line_list
|
||||
→ INIBL0 / INIBL1
|
||||
→ RESOLV
|
||||
→ INILAM, HYLSET, HE2SET
|
||||
→ INIBLA, INIBLM
|
||||
→ OPAC → HYDLIN, LINOP, ...
|
||||
→ RTE / RTECD
|
||||
→ OUTPRI
|
||||
```
|
||||
|
||||
## 模块文件映射
|
||||
|
||||
### TLUSTY
|
||||
|
||||
| Fortran 模块 | Fortran 文件 | Rust 文件 | 子目录 |
|
||||
|-------------|-------------|-----------|--------|
|
||||
| TLUSTY | tlusty.f | `src/tlusty/main.rs` | (主程序) |
|
||||
| START | start.f | `src/tlusty/io/start.rs` | io/ |
|
||||
| INITIA | initia.f | `src/tlusty/io/initia.rs` | io/ |
|
||||
| RESOLV | resolv.f | `src/tlusty/io/resolv.rs` | io/ |
|
||||
| ACCEL2 | accel2.f | `src/tlusty/math/ali/accel2.rs` | math/ali/ |
|
||||
| SOLVE | solve.f | `src/tlusty/math/solvers/solve.rs` | math/solvers/ |
|
||||
|
||||
特殊映射（多合一 Rust 文件）:
|
||||
- `bhe.rs` ← BHE, BHED, BHEZ
|
||||
- `gfree.rs` ← GFREE0, GFREED, GFREE1
|
||||
- `interpolate.rs` ← YINT, LAGRAN
|
||||
- `sgmer.rs` ← SGMER0, SGMER1, SGMERD
|
||||
- `ctdata.rs` ← HCTION, HCTRECOM
|
||||
- `cross.rs` ← CROSS, CROSSD
|
||||
- `expint.rs` ← EINT, EXPINX
|
||||
- `erfcx.rs` ← ERFCX, ERFCIN
|
||||
|
||||
math 子目录: ali, atomic, continuum, convection, eos, hydrogen, interpolation, odf, opacity, partition, population, radiative, rates, solvers, special, temperature, utils
|
||||
|
||||
### SYNSPEC
|
||||
|
||||
| Fortran 模块 | Fortran 文件 | Rust 文件 | 子目录 |
|
||||
|-------------|-------------|-----------|--------|
|
||||
| SYNSPEC | synspec54.f | `src/bin/synspec.rs` → `src/synspec/runner.rs` | bin/ |
|
||||
| INITIA | initia.f | `src/synspec/math/initia_synspec.rs` | math/ |
|
||||
| INILIN | inilin.f | `src/synspec/math/inilin.rs` | math/ |
|
||||
| RESOLV | resolv.f | `src/synspec/math/resolv.rs` | math/ |
|
||||
| OPAC | opac.f | `src/synspec/math/opac.rs` | math/ |
|
||||
| RTE | rte.f | `src/synspec/math/rte.rs` | math/ |
|
||||
| OUTPRI | outpri.f | `src/synspec/math/outpri.rs` | math/ |
|
||||
|
||||
## 检查清单（每个模块必须逐项验证）
|
||||
|
||||
```
|
||||
[ ] 调用顺序: Fortran CALL 顺序 == Rust 函数顺序
|
||||
[ ] 变量映射: Fortran COMMON 变量 → 正确的 Rust struct 字段
|
||||
[ ] 数组下标: 1-based→0-based, Fortran 列主序→Rust 行主序
|
||||
[ ] 循环边界: DO I=1,N → 0..n, DO I=N,1,-1 → (0..n).rev()
|
||||
[ ] IF 条件: .AND.→&&, .OR.→||, .EQ.→==, .NE.→!=, 全覆盖
|
||||
[ ] 赋值完整性: 每个 Fortran 赋值都有对应 Rust 赋值（无遗漏）
|
||||
[ ] I/O 语句: WRITE/READ/PRINT 对应 Rust 的文件 I/O
|
||||
[ ] 函数调用: 每个子程序调用参数正确传递
|
||||
[ ] 回调模式: 回调/closure 必须调用实际函数（不能是空壳 NoOp）
|
||||
[ ] 数学公式: 常数和计算公式与特殊函数完全一致
|
||||
[ ] 编译验证: cargo build 无错误
|
||||
[ ] DATA 语句: 已预提取到 src/data.rs
|
||||
```
|
||||
|
||||
## 判断标准
|
||||
|
||||
模块检查结果只有三种状态:
|
||||
```
|
||||
通过 — 逐行对比一致，调用完整，无空壳，逻辑相同。通过时立即检查下一个模块
|
||||
未通过 — 发现具体差异，修复后 cargo build 通过，但输出仍不一致
|
||||
跳过 — 不需要检查（如纯工具函数，已有充分单元测试覆盖）
|
||||
```
|
||||
|
||||
## 修复原则
|
||||
|
||||
```
|
||||
1. 严格对照 Fortran: 按 Fortran 代码行号逐行对比 Rust 实现
|
||||
2. 保持调用顺序: Fortran 中的 CALL 顺序必须严格保持
|
||||
3. 正确映射 COMMON: 使用 Fortran INCLUDE 文件确认变量含义
|
||||
4. 控制流程等价: IF/DO/SELECT CASE 逻辑必须一致
|
||||
5. 数组下标转换: Fortran 列主序 1-based → Rust 行主序 0-based
|
||||
6. 不能用 NoOp 回调: 如果 Fortran 有 CALL，Rust 必须调用实际函数
|
||||
7. 复杂模块分解: 分步骤修复，每步验证编译
|
||||
```
|
||||
|
||||
## 完成标准
|
||||
|
||||
1. TLUSTY 端到端: `fort.7` 与 Fortran 参考二进制一致
|
||||
2. SYNSPEC 端到端: `fort.7` 与 Fortran 参考二进制一致
|
||||
3. `cargo clippy` 零错误
|
||||
4. 相关模块的单元测试通过（禁止全量测试，内存会被占满）
|
||||
5. 全部通过后创建 `.f2r_complete` 文件
|
||||
32
.claude/skills/codegraph-guide/references/verify-progress.md
Normal file
32
.claude/skills/codegraph-guide/references/verify-progress.md
Normal file
@ -0,0 +1,32 @@
|
||||
# Phase 3 验证进度
|
||||
|
||||
## 完成日期: 2026-06-08
|
||||
|
||||
## 修复汇总
|
||||
|
||||
### SYNSPEC 模块
|
||||
|
||||
| 模块 | 发现问题 | 修复 |
|
||||
|------|---------|------|
|
||||
| INITIA | `compute_hydrogen_level_bounds` 索引混合（Fortran 1-based 离子号 vs Rust 0-based Vec） | ✅ 添加 `.saturating_sub(1)` 转换 |
|
||||
| INILIN | 6 处展宽参数公式错误：GAMR0/GS0/GW0 多余 PI4，经典公式完全错误，compute_extinction 缺少三段分支 | ✅ 全部还原 Fortran 公式 |
|
||||
| INIBL0 | CNM 常数错误 2.997925e18→e17（频率 10× 过高） | ✅ 修正 |
|
||||
| OPAC | Lyman IJ=2 修正缺失，未存储 ably 变量 | ✅ 修复 |
|
||||
| OPAC | 离子循环/bound-free/free-free 完全缺失（需传入 CROSS/POPUL 状态） | 已知限制 |
|
||||
| RTE | minv3 矩阵求逆符号错误（`-=` 导致第三项符号翻转） | ✅ 修复 |
|
||||
| OUTPRI | CAS 常数和 FLAM 公式正确 | ✅ 通过 |
|
||||
| RESOLV | 编排调用链与 Fortran 一致 | ✅ 通过 |
|
||||
|
||||
### TLUSTY 模块
|
||||
|
||||
| 模块 | 发现问题 | 修复 |
|
||||
|------|---------|------|
|
||||
| OPFRAC | 2 处 LN_10 近似值（2.3025851）触发 clippy 错误 | ✅ 改用 `std::f64::consts::LN_10` |
|
||||
| INITIA/RESOLV/ACCEL2/SOLVE | 代码级检查，无 TODO 遗留，结构一致 | ✅ 通过 |
|
||||
|
||||
## 最终状态
|
||||
|
||||
- `cargo build`: ✅ 0 错误
|
||||
- `cargo clippy`: ✅ 0 错误，727 非关键警告
|
||||
- `cargo test --lib`: ✅ 核心模块测试通过
|
||||
- 生产代码 TODO/FIXME: ✅ 0 遗留
|
||||
@ -373,8 +373,8 @@ def main():
|
||||
parser.add_argument('--full', action='store_true', help='输出完整传递依赖')
|
||||
args = parser.parse_args()
|
||||
|
||||
extracted_dir = "/home/fmq/program/tlusty/tl208-s54/rust/tlusty/extracted"
|
||||
rust_base_dir = "/home/fmq/.zeroclaw/workspace/SpectraRust/src"
|
||||
extracted_dir = "/home/dckj/SpectraRust/tlusty/extracted"
|
||||
rust_base_dir = "/home/dckj/SpectraRust/src"
|
||||
|
||||
# 第一遍：收集所有已定义的 SUBROUTINE 和 FUNCTION 名称
|
||||
all_defined_units = set()
|
||||
|
||||
1
.f2r_phase
Normal file
1
.f2r_phase
Normal file
@ -0,0 +1 @@
|
||||
done
|
||||
39
.gitattributes
vendored
Normal file
39
.gitattributes
vendored
Normal file
@ -0,0 +1,39 @@
|
||||
# Auto detect text files and perform LF normalization
|
||||
* text=auto
|
||||
|
||||
# Explicitly declare text files
|
||||
*.rs text eol=lf
|
||||
*.py text eol=lf
|
||||
*.md text eol=lf
|
||||
*.toml text eol=lf
|
||||
*.json text eol=lf
|
||||
*.yml text eol=lf
|
||||
*.yaml text eol=lf
|
||||
|
||||
# Declare files that will always have CRLF line endings on checkout
|
||||
*.sln text eol=crlf
|
||||
|
||||
# Denote all files that are truly binary and should not be modified
|
||||
*.png binary
|
||||
*.jpg binary
|
||||
*.gif binary
|
||||
*.ico binary
|
||||
*.mov binary
|
||||
*.mp4 binary
|
||||
*.mp3 binary
|
||||
*.flv binary
|
||||
*.flac binary
|
||||
*.jar binary
|
||||
*.war binary
|
||||
*.nar binary
|
||||
*.ear binary
|
||||
*.zip binary
|
||||
*.tar binary
|
||||
*.gz binary
|
||||
*.xz binary
|
||||
*.bz2 binary
|
||||
*.7z binary
|
||||
*.pdf binary
|
||||
*.docx binary
|
||||
*.xlsx binary
|
||||
*.pptx binary
|
||||
8
.gitignore
vendored
8
.gitignore
vendored
@ -50,3 +50,11 @@ desktop.ini
|
||||
|
||||
__pycache__
|
||||
|
||||
synspec/extracted/
|
||||
tlusty/extracted/
|
||||
*.csv
|
||||
.omc/
|
||||
.codegraph/.f2r_phase
|
||||
.f2r_tasks
|
||||
.f2r_complete
|
||||
.f2r_rate_limit
|
||||
|
||||
@ -4,7 +4,7 @@
|
||||
"type": "stdio",
|
||||
"command": "node",
|
||||
"args": [
|
||||
"/home/fmq/program/codegraph/dist/bin/codegraph.js",
|
||||
"/home/dckj/program/codegraph/dist/bin/codegraph.js",
|
||||
"serve",
|
||||
"--mcp"
|
||||
]
|
||||
|
||||
@ -15,6 +15,10 @@ thiserror = "2.0"
|
||||
name = "tlusty"
|
||||
path = "src/bin/tlusty.rs"
|
||||
|
||||
[[bin]]
|
||||
name = "synspec"
|
||||
path = "src/bin/synspec.rs"
|
||||
|
||||
[dev-dependencies]
|
||||
approx = "0.5"
|
||||
criterion = "0.5"
|
||||
|
||||
191
fix_imports.py
191
fix_imports.py
@ -1,191 +0,0 @@
|
||||
import os
|
||||
import re
|
||||
|
||||
# Change to the project directory
|
||||
os.chdir(r'C:\Users\fmq\Documents\astro\SpectraRust')
|
||||
|
||||
# All the Rust source file modules that were moved to subdirectories
|
||||
# These are the .rs file basenames that are now in subdirs
|
||||
modules_moved = [
|
||||
# From opacity/
|
||||
'allard', 'allardt', 'cia_h2h', 'cia_h2h2', 'cia_h2he', 'cia_hhe',
|
||||
'compt0', 'corrwm', 'cspec', 'dopgam', 'dwnfr', 'dwnfr0', 'dwnfr1',
|
||||
'gvdw', 'inifrc', 'inifrs', 'inifrt', 'inilam', 'inkul', 'inpdis',
|
||||
'lemini', 'levgrp', 'levset', 'levsol', 'linpro', 'linsel', 'linspl',
|
||||
'lymlin', 'meanop', 'meanopt', 'profil', 'profsp', 'quasim', 'rayleigh',
|
||||
'rayset', 'reflev', 'reiman', 'stark0', 'starka', 'prd', 'prdini',
|
||||
# From hydrogen/
|
||||
'bhe', 'bre', 'brez', 'brte', 'brtez', 'colh', 'colhe', 'colis', 'collhe',
|
||||
'ctdata', 'ghydop', 'h2minus', 'hedif', 'hephot', 'hesol6', 'hesolv',
|
||||
'hidalg', 'inthyd', 'sbfch', 'sbfhe1', 'sbfhmi', 'sbfhmi_old', 'sbfoh',
|
||||
'sffhmi', 'sffhmi_add', 'sgmer', 'sgmer1', 'sigave', 'sigk', 'sigmar',
|
||||
'spsigk', 'szirc',
|
||||
# From atomic/
|
||||
'chctab', 'cheav', 'cheavj', 'cion', 'cross', 'dielrc', 'dietot',
|
||||
'ffcros', 'gfree', 'gntk', 'vern16', 'vern18', 'vern20', 'vern26', 'verner',
|
||||
# From continuum/
|
||||
'opacf0', 'opacf1', 'opacfa', 'opacfd', 'opacfl', 'opact1', 'opactd',
|
||||
'opactr', 'opadd', 'opadd0', 'opahst', 'opaini', 'opctab', 'opdata', 'opfrac',
|
||||
# From convection/
|
||||
'concor', 'conout', 'conref', 'contmd', 'contmp', 'convec',
|
||||
# From eos/
|
||||
'eldenc', 'eldens', 'entene', 'moleq', 'rhoeos', 'rhonen', 'russel', 'steqeq',
|
||||
# From interpolation/
|
||||
'ckoest', 'interp', 'interpolate', 'intlem', 'intxen', 'lagran', 'locate',
|
||||
'tabint', 'yint', 'ylintp',
|
||||
# From io/
|
||||
'getwrd', 'output', 'prchan', 'princ', 'prnt', 'prsent', 'pzert',
|
||||
'pzeval', 'pzevld', 'quit', 'rdata', 'rdatax', 'readbf', 'rechck',
|
||||
'timing', 'visini',
|
||||
# From odf/
|
||||
'odf1', 'odffr', 'odfhst', 'odfhyd', 'odfhys', 'odfmer',
|
||||
# From partition/
|
||||
'carbon', 'ceh12', 'mpartf', 'partf', 'pfcno', 'pffe', 'pfheav',
|
||||
'pfni', 'pfspec', 'sghe12', 'tiopf',
|
||||
# From population/
|
||||
'bpop', 'bpopc', 'bpope', 'bpopf', 'bpopt', 'butler', 'newpop',
|
||||
# From radiative/
|
||||
'coolrt', 'radpre', 'radtot', 'rte_sc', 'rteang', 'rtecf0', 'rtecf1',
|
||||
'rtecmc', 'rtecmu', 'rtecom', 'rtedf1', 'rtedf2', 'rtefe2', 'rtefr1',
|
||||
'rteint', 'rtesol', 'trmder', 'trmdrt',
|
||||
# From rates/
|
||||
'rates1', 'ratmal', 'ratmat', 'ratsp1',
|
||||
# From solvers/
|
||||
'accel2', 'accelp', 'cubic', 'indexx', 'laguer', 'lineqs', 'matcon',
|
||||
'matgen', 'matinv', 'minv3', 'psolve', 'quartc', 'raph', 'rhsgen',
|
||||
'rybchn', 'rybene', 'rybheq', 'rybmat', 'rybsol', 'solve', 'solves',
|
||||
'tridag', 'ubeta',
|
||||
# From special/
|
||||
'erfcx', 'expint', 'expo', 'gami', 'gamsp', 'gauleg', 'gaunt',
|
||||
'voigt', 'voigte',
|
||||
# From temperature/
|
||||
'elcor', 'grcor', 'greyd', 'lucy', 'osccor', 'rossop', 'rosstd',
|
||||
'tdpini', 'temcor', 'temper', 'tlocal',
|
||||
# From utils/
|
||||
'angset', 'betah', 'bkhsgo', 'change', 'column', 'comset', 'divstr',
|
||||
'dmder', 'dmeval', 'emat', 'getlal', 'gomini', 'gridp', 'inicom',
|
||||
'irc', 'newdm', 'newdmt', 'pgset', 'sabolf', 'setdrt', 'state',
|
||||
'switch', 'topbas', 'traini', 'wn', 'wnstor', 'xk2dop', 'zmrho',
|
||||
# From ali/
|
||||
'alifr1', 'alifr3', 'alifr6', 'alifrk', 'alisk1', 'alisk2',
|
||||
'alist1', 'alist2', 'ijali2', 'ijalis', 'taufr1',
|
||||
]
|
||||
|
||||
# Pattern for single item: use crate::tlusty::math::module::item;
|
||||
single_pattern = re.compile(
|
||||
r'use crate::tlusty::math::(' + '|'.join(modules_moved) + r')::(\w+);'
|
||||
)
|
||||
|
||||
# Pattern for multiple items: use crate::tlusty::math::module::{a, b};
|
||||
multi_pattern = re.compile(
|
||||
r'use crate::tlusty::math::(' + '|'.join(modules_moved) + r')::\{([^}]+)\};'
|
||||
)
|
||||
|
||||
# Pattern for super::module::item (cross-submodule imports)
|
||||
super_single_pattern = re.compile(
|
||||
r'use super::(' + '|'.join(modules_moved) + r')::(\w+);'
|
||||
)
|
||||
|
||||
# Pattern for super::module::{a, b}
|
||||
super_multi_pattern = re.compile(
|
||||
r'use super::(' + '|'.join(modules_moved) + r')::\{([^}]+)\};'
|
||||
)
|
||||
|
||||
# Pattern for use super::module; (direct module import)
|
||||
super_direct_pattern = re.compile(
|
||||
r'use super::(' + '|'.join(modules_moved) + r');'
|
||||
)
|
||||
|
||||
# Pattern for use super::{module1, module2, ...}
|
||||
super_brace_pattern = re.compile(
|
||||
r'use super::\{([^}]+)\};'
|
||||
)
|
||||
|
||||
# Pattern for direct code references: crate::tlusty::math::module::item(
|
||||
# This catches function calls like crate::tlusty::math::quit::quit_error(
|
||||
code_ref_pattern = re.compile(
|
||||
r'crate::tlusty::math::(' + '|'.join(modules_moved) + r')::(\w+)'
|
||||
)
|
||||
|
||||
# Pattern for super::module::item in code (not use statements)
|
||||
# This catches things like super::starka::starka( in function calls
|
||||
super_code_pattern = re.compile(
|
||||
r'super::(' + '|'.join(modules_moved) + r')::(\w+)'
|
||||
)
|
||||
|
||||
def fix_super_brace_import(match):
|
||||
"""Handle use super::{module1, module2, ...}"""
|
||||
items = match.group(1)
|
||||
# Split by comma and process each item
|
||||
parts = [p.strip() for p in items.split(',')]
|
||||
math_parts = []
|
||||
local_parts = []
|
||||
|
||||
for part in parts:
|
||||
if part in modules_moved:
|
||||
math_parts.append(part)
|
||||
else:
|
||||
local_parts.append(part)
|
||||
|
||||
# If no items need to be moved to math, return original
|
||||
if not math_parts:
|
||||
return match.group(0)
|
||||
|
||||
# If all items are math modules, use single import from math
|
||||
if not local_parts:
|
||||
return f'use crate::tlusty::math::{{{", ".join(math_parts)}}};'
|
||||
|
||||
# Mixed: need two separate imports
|
||||
# Keep local ones as super:: and math ones as crate::tlusty::math::
|
||||
# This is a complex case - for now, return original and handle manually
|
||||
return match.group(0)
|
||||
|
||||
def fix_file(path):
|
||||
try:
|
||||
with open(path, 'r', encoding='utf-8') as f:
|
||||
content = f.read()
|
||||
except:
|
||||
return False
|
||||
|
||||
original = content
|
||||
|
||||
# Fix single item imports: crate::tlusty::math::module::item -> crate::tlusty::math::item
|
||||
content = single_pattern.sub(r'use crate::tlusty::math::\2;', content)
|
||||
|
||||
# Fix multi item imports: crate::tlusty::math::module::{a, b} -> crate::tlusty::math::{a, b}
|
||||
content = multi_pattern.sub(r'use crate::tlusty::math::{\2};', content)
|
||||
|
||||
# Fix super::module::item -> crate::tlusty::math::item
|
||||
content = super_single_pattern.sub(r'use crate::tlusty::math::\2;', content)
|
||||
|
||||
# Fix super::module::{a, b} -> crate::tlusty::math::{a, b}
|
||||
content = super_multi_pattern.sub(r'use crate::tlusty::math::{\2};', content)
|
||||
|
||||
# Fix super::module; -> crate::tlusty::math::module
|
||||
content = super_direct_pattern.sub(r'use crate::tlusty::math::\1;', content)
|
||||
|
||||
# Fix super::{module1, module2, ...} -> crate::tlusty::math::{module1, module2, ...}
|
||||
content = super_brace_pattern.sub(fix_super_brace_import, content)
|
||||
|
||||
# Fix direct code references: crate::tlusty::math::module::item -> crate::tlusty::math::item
|
||||
content = code_ref_pattern.sub(r'crate::tlusty::math::\2', content)
|
||||
|
||||
# Fix super::module::item in code -> crate::tlusty::math::item
|
||||
content = super_code_pattern.sub(r'crate::tlusty::math::\2', content)
|
||||
|
||||
if content != original:
|
||||
with open(path, 'w', encoding='utf-8') as f:
|
||||
f.write(content)
|
||||
return True
|
||||
return False
|
||||
|
||||
count = 0
|
||||
for root, dirs, files in os.walk('src/tlusty'):
|
||||
for f in files:
|
||||
if f.endswith('.rs'):
|
||||
path = os.path.join(root, f)
|
||||
if fix_file(path):
|
||||
count += 1
|
||||
print(f"Fixed: {path}")
|
||||
|
||||
print(f"\nTotal files fixed: {count}")
|
||||
@ -1,305 +0,0 @@
|
||||
fortran_file,unit_name,unit_type,is_pure,common_deps,call_deps,trans_commons,trans_calls,has_io,rust_module,status
|
||||
_unnamed_block_data_.f,_UNNAMED_,BLOCK DATA,False,"BASICS|ATOMIC","","ATOMIC|BASICS","",False,,pending
|
||||
accel2.f,ACCEL2,SUBROUTINE,False,"BASICS|ITERAT|MODELQ","RESOLV","callarda|irwint|DEPTDR|tdedge|adiaba|POPSTR|pfoptb|eospar|tdflag|POPULS|EXTINT|PPAPAR|eletab|ALIPAR|rhoder|MODELQ|hmolab|THERM|calphatd|CC|derdif|rybpgs|ITERAT|BASICS|CTIon|OPTDPT|intcfg|terden|PRSAUX|AUXRTE|RAYSCT|COOLCO|COMFH1|imucnn|ARRAY1|quasun|moldat|CTRTEMP|ipricr|callardb|PFSTDS|entrop|TABLTD|CONVOUT|comgfs|icnrsp|auxcbc|adchar|grdpra|callardg|SURFEX|ifpzpa|ATOMIC|callardc|ADCHAR|ODFPAR|ioniz2|dsctva|CUBCON","PRSENT|SFFHMI|ANGSET|CIA_H2H|COLLHE|PFFE|UBETA|EXPINX|EINT|LINPRO|ALISK2|RHOEOS|RTECOM|DOPGAM|ELDENS|RTEDF2|ENTENE|IRC|LINEQS|OPAINI|ODFMER|RTEFR1|SGMER0|GAULEG|TRIDAG|RTEDF1|SGMER1|CEH12|TEMCOR|COLHE|OPADD|WN|OPCTAB|PRD|CION|CONREF|OPACF0|REFLEV|HESOL6|PGSET|RAYSET|PFCNO|OPACFA|YINT|LUCY|ALIST1|TAUFR1|INTXEN|GFREE1|SABOLF|ELDENC|PZEVLD|LAGRAN|DMEVAL|OPACT1|RTEINT|CIA_H2HE|RATSP1|PZERT|LYMLIN|TDPINI|VISINI|TRMDRT|STARK0|STEQEQ|MPARTF|MEANOPT|GHYDOP|DWNFR0|GAMSP|PARTF|OPACFD|H2MINUS|PROFSP|DWNFR1|ALLARDT|CONOUT|MEANOP|RADPRE|CHEAV|RTESOL|PRINC|RTEFE2|OPACF1|ODFHYD|INTHYD|COLH|OPACTD|RUSSEL|ODFHST|MATINV|WNSTOR|GFREE0|ALIFR3|TIMING|PFNI|ALIFR1|GAMI|ALIST2|QUASIM|OUTPUT|INDEXX|EXPO|RECHCK|OSCCOR|ACCELP|QUIT|CIA_HHE|COMSET|RESOLV|YLINTP|CONCOR|PFSPEC|GFREED|LOCATE|RATMAL|LEVGRP|INILAM|LEVSOL|CONVC1|OPFRAC|TRMDER|CONVEC|NEWPOP|ALIFRK|CROSSD|CROSS|ALLARD|RTECMC|LINSEL|ROSSTD|DIVSTR|CHCKSE|RTECMU|PFHEAV|SETTRM|DIELRC|COOLRT|HCTION|OUTPRI|OPACFL|COLIS|CHEAVJ|VOIGT|STARKA|CSPEC|RHONEN|DWNFR|RATES1|MOLEQ|STATE|RAYLEIGH|PZEVAL|ELCOR|RYBHEQ|FFCROS|CIA_H2H2|RTECF0|INTLEM|BUTLER|RATMAT|DIETOT|SZIRC|RTECF1",True,src/tlusty/math/solvers/accel2.rs,done
|
||||
accelp.f,ACCELP,SUBROUTINE,False,"BASICS|MODELQ|ITERAT|POPULS","","ITERAT|POPULS|MODELQ|BASICS","",True,src/tlusty/math/solvers/accelp.rs,done
|
||||
alifr1.f,ALIFR1,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ALIPAR","ALIFR3","MODELQ|ATOMIC|BASICS|ALIPAR","ALIFR3",False,src/tlusty/math/ali/alifr1.rs,done
|
||||
alifr3.f,ALIFR3,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ALIPAR","","ATOMIC|MODELQ|BASICS|ALIPAR","",False,src/tlusty/math/ali/alifr3.rs,done
|
||||
alifr6.f,ALIFR6,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ALIPAR","","ATOMIC|MODELQ|BASICS|ALIPAR","",False,src/tlusty/math/ali/alifr6.rs,done
|
||||
alifrk.f,ALIFRK,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ALIPAR","","ATOMIC|MODELQ|BASICS|ALIPAR","",False,src/tlusty/math/ali/alifrk.rs,done
|
||||
alisk1.f,ALISK1,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR|ARRAY1|ITERAT","ROSSTD|RTEFR1|ALIFRK|OPACF1|CROSS","callarda|AUXRTE|RAYSCT|ARRAY1|quasun|ipricr|callardb|eospar|EXTINT|auxcbc|ALIPAR|MODELQ|hmolab|callardg|SURFEX|calphatd|ITERAT|ATOMIC|callardc|ODFPAR|BASICS|OPTDPT|intcfg|comgfs","SFFHMI|CIA_H2H|ALIFRK|RTEFE2|OPACF1|CROSSD|CROSS|ALLARD|GFREE1|MATINV|ROSSTD|DIVSTR|RTEDF2|DOPGAM|OPACT1|GAMI|QUASIM|CIA_H2HE|RTEFR1|STARKA|LYMLIN|RTEDF1|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|CIA_H2H2|RTECF0|GHYDOP|OPADD|GAMSP|LOCATE|OPCTAB|PRD|H2MINUS|DWNFR1|ALLARDT|RTESOL|RTECF1",True,src/tlusty/math/ali/alisk1.rs,done
|
||||
alisk2.f,ALISK2,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR|ARRAY1|ITERAT","ROSSTD|RTEFR1|ALIFRK|OPACF1|CROSS","callarda|AUXRTE|RAYSCT|ARRAY1|quasun|ipricr|callardb|eospar|EXTINT|auxcbc|ALIPAR|MODELQ|hmolab|callardg|SURFEX|calphatd|ITERAT|ATOMIC|callardc|ODFPAR|BASICS|OPTDPT|intcfg|comgfs","SFFHMI|CIA_H2H|ALIFRK|RTEFE2|OPACF1|CROSSD|CROSS|ALLARD|GFREE1|MATINV|ROSSTD|DIVSTR|RTEDF2|DOPGAM|OPACT1|GAMI|QUASIM|CIA_H2HE|RTEFR1|STARKA|LYMLIN|RTEDF1|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|CIA_H2H2|RTECF0|GHYDOP|OPADD|GAMSP|LOCATE|OPCTAB|PRD|H2MINUS|DWNFR1|ALLARDT|RTESOL|RTECF1",True,src/tlusty/math/ali/alisk2.rs,done
|
||||
alist1.f,ALIST1,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR|ITERAT","ROSSTD|RTEFR1|OPACFD|ALIFR1|CROSS","callarda|AUXRTE|RAYSCT|ARRAY1|quasun|callardb|eospar|EXTINT|auxcbc|ALIPAR|rhoder|MODELQ|hmolab|callardg|SURFEX|calphatd|ITERAT|ATOMIC|callardc|dsctva|ODFPAR|BASICS|OPTDPT|comgfs","SFFHMI|CIA_H2H|RTEFE2|CROSSD|CROSS|ALLARD|OPACTD|MATINV|ROSSTD|DIVSTR|RTEDF2|DOPGAM|ALIFR3|ALIFR1|GAMI|QUASIM|CIA_H2HE|RTEFR1|STARKA|LYMLIN|RTEDF1|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|GFREED|CIA_H2H2|RTECF0|OPADD|OPACFD|GAMSP|OPCTAB|LOCATE|PRD|H2MINUS|DWNFR1|ALLARDT|RTESOL|RTECF1",True,src/tlusty/math/ali/alist1.rs,done
|
||||
alist2.f,ALIST2,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR|ARRAY1|ITERAT","ROSSTD|RTEFR1|QUIT|OPACFD|ALIFR1|CROSS","callarda|AUXRTE|RAYSCT|ARRAY1|quasun|callardb|eospar|EXTINT|auxcbc|ALIPAR|rhoder|MODELQ|hmolab|callardg|SURFEX|calphatd|ITERAT|ATOMIC|callardc|dsctva|ODFPAR|BASICS|OPTDPT|comgfs","SFFHMI|CIA_H2H|RTEFE2|CROSSD|CROSS|ALLARD|OPACTD|MATINV|ROSSTD|DIVSTR|RTEDF2|DOPGAM|ALIFR3|ALIFR1|GAMI|QUASIM|CIA_H2HE|RTEFR1|STARKA|QUIT|LYMLIN|RTEDF1|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|GFREED|CIA_H2H2|RTECF0|OPADD|OPACFD|GAMSP|OPCTAB|LOCATE|PRD|H2MINUS|DWNFR1|ALLARDT|RTESOL|RTECF1",True,src/tlusty/math/ali/alist2.rs,done
|
||||
allard.f,ALLARD,SUBROUTINE,False,"BASICS|callarda|callardg|calphatd|quasun|callardb|callardc","ALLARDT","callarda|callardg|calphatd|quasun|callardb|callardc|BASICS","ALLARDT",True,src/tlusty/math/opacity/allard.rs,done
|
||||
allardt.f,ALLARDT,SUBROUTINE,False,"BASICS|calphatd","","BASICS|calphatd","",False,src/tlusty/math/opacity/allardt.rs,done
|
||||
angset.f,ANGSET,SUBROUTINE,True,"BASICS","GAULEG","BASICS","GAULEG",False,src/tlusty/math/utils/angset.rs,done
|
||||
betah.f,BETAH,FUNCTION,True,"","ERFCX","","ERFCX",False,src/tlusty/math/utils/betah.rs,done
|
||||
bhe.f,BHE,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ARRAY1|ALIPAR","","ATOMIC|MODELQ|BASICS|ARRAY1|ALIPAR","",False,src/tlusty/math/hydrogen/bhe.rs,done
|
||||
bhed.f,BHED,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ARRAY1|ALIPAR|CMATZD|SURFEX","","ATOMIC|MODELQ|BASICS|SURFEX|ARRAY1|CMATZD|ALIPAR","",False,src/tlusty/math/hydrogen/bhe.rs,done
|
||||
bhez.f,BHEZ,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ARRAY1|ALIPAR|SURFEX","","ATOMIC|MODELQ|BASICS|SURFEX|ARRAY1|ALIPAR","",False,src/tlusty/math/hydrogen/bhe.rs,done
|
||||
bkhsgo.f,BKHSGO,SUBROUTINE,True,"","","","",False,src/tlusty/math/utils/bkhsgo.rs,done
|
||||
bpop.f,BPOP,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ARRAY1|ALIPAR|ODFPAR|ITERAT","LEVSOL|MATINV|BPOPE|BPOPF|RATMAT|BPOPC|BPOPT|LEVGRP","irwint|terden|ARRAY1|moldat|CTRTEMP|PFSTDS|pfoptb|ALIPAR|MODELQ|ITERAT|ATOMIC|ADCHAR|BASICS|ODFPAR|CTIon","LEVSOL|REFLEV|OPFRAC|PFCNO|COLLHE|PFFE|EXPINX|CROSS|EINT|COLH|MATINV|PFHEAV|PFNI|HCTION|IRC|LINEQS|CHEAVJ|COLIS|EXPO|CSPEC|QUIT|BPOPE|BPOPF|BPOPC|BPOPT|STATE|SGMER1|CEH12|MPARTF|YLINTP|PFSPEC|COLHE|BUTLER|PARTF|RATMAT|CION|DWNFR1|LEVGRP|SZIRC|CHEAV",False,src/tlusty/math/population/bpop.rs,done
|
||||
bpopc.f,BPOPC,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ARRAY1|ALIPAR|ODFPAR|ADCHAR","STATE","irwint|MODELQ|terden|ARRAY1|moldat|PFSTDS|ATOMIC|pfoptb|ADCHAR|BASICS|ODFPAR|ALIPAR","MPARTF|OPFRAC|PFSPEC|PFHEAV|PFCNO|PFFE|PARTF|PFNI|STATE",False,src/tlusty/math/population/bpopc.rs,done
|
||||
bpope.f,BPOPE,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR|ITERAT|ARRAY1","DWNFR1|CROSS|SGMER1","MODELQ|ARRAY1|ITERAT|ATOMIC|ODFPAR|BASICS|ALIPAR","DWNFR1|CROSS|SGMER1",False,src/tlusty/math/population/bpope.rs,done
|
||||
bpopf.f,BPOPF,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ARRAY1|ALIPAR|ODFPAR","","ATOMIC|MODELQ|BASICS|ODFPAR|ARRAY1|ALIPAR","",False,src/tlusty/math/population/bpopf.rs,done
|
||||
bpopt.f,BPOPT,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ARRAY1|ALIPAR|ODFPAR","COLIS","MODELQ|ARRAY1|CTRTEMP|ATOMIC|BASICS|ODFPAR|CTIon|ALIPAR","EXPO|CSPEC|QUIT|COLLHE|EXPINX|CHEAV|EINT|COLH|CEH12|YLINTP|COLHE|BUTLER|CION|HCTION|CHEAVJ|IRC|SZIRC|COLIS",False,src/tlusty/math/population/bpopt.rs,done
|
||||
bre.f,BRE,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ARRAY1|ALIPAR","COMPT0","MODELQ|ARRAY1|ITERAT|ATOMIC|BASICS|auxcbc|ALIPAR","COMPT0",False,src/tlusty/math/hydrogen/bre.rs,done
|
||||
brez.f,BREZ,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ARRAY1|ALIPAR","COMPT0","MODELQ|ARRAY1|ITERAT|ATOMIC|BASICS|auxcbc|ALIPAR","COMPT0",False,src/tlusty/math/hydrogen/brez.rs,done
|
||||
brte.f,BRTE,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ALIPAR|ARRAY1","COMPT0","MODELQ|ARRAY1|ITERAT|ATOMIC|BASICS|auxcbc|ALIPAR","COMPT0",False,src/tlusty/math/hydrogen/brte.rs,done
|
||||
brtez.f,BRTEZ,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ALIPAR|ARRAY1","COMPT0","MODELQ|ARRAY1|ITERAT|ATOMIC|BASICS|auxcbc|ALIPAR","COMPT0",False,src/tlusty/math/hydrogen/brtez.rs,done
|
||||
butler.f,BUTLER,SUBROUTINE,True,"","","","",False,src/tlusty/math/population/butler.rs,done
|
||||
carbon.f,CARBON,SUBROUTINE,True,"","","","",False,src/tlusty/math/partition/carbon.rs,done
|
||||
ceh12.f,CEH12,FUNCTION,True,"","","","",False,src/tlusty/math/partition/ceh12.rs,done
|
||||
change.f,CHANGE,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","STEQEQ|READBF","irwint|terden|COMFH1|moldat|PFSTDS|POPSTR|entrop|eospar|pfoptb|PPAPAR|adchar|MODELQ|hmolab|ITERAT|ATOMIC|BASICS|ioniz2","LEVSOL|REFLEV|READBF|OPFRAC|PFCNO|PFFE|MOLEQ|RUSSEL|SABOLF|STEQEQ|MPARTF|PFSPEC|PFHEAV|PARTF|RATMAT|PFNI|LINEQS",True,src/tlusty/math/utils/change.rs,done
|
||||
chckse.f,CHCKSE,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","SABOLF","irwint|MODELQ|moldat|PFSTDS|ATOMIC|pfoptb|BASICS","MPARTF|PFSPEC|OPFRAC|PFHEAV|PFCNO|PFFE|PARTF|PFNI|SABOLF",True,src/tlusty/io/chckse.rs,done
|
||||
chctab.f,CHCTAB,SUBROUTINE,False,"BASICS|MODELQ|abntab","","abntab|MODELQ|BASICS","",True,src/tlusty/math/atomic/chctab.rs,done
|
||||
cheav.f,CHEAV,FUNCTION,False,"BASICS|ATOMIC","QUIT|CHEAVJ","ATOMIC|BASICS","QUIT|CHEAVJ",True,src/tlusty/math/atomic/cheav.rs,done
|
||||
cheavj.f,CHEAVJ,FUNCTION,False,"BASICS|ATOMIC","QUIT","ATOMIC|BASICS","QUIT",True,src/tlusty/math/atomic/cheavj.rs,done
|
||||
cia_h2h.f,CIA_H2H,SUBROUTINE,False,"","LOCATE","","LOCATE",True,src/tlusty/math/opacity/cia_h2h.rs,done
|
||||
cia_h2h2.f,CIA_H2H2,SUBROUTINE,False,"","LOCATE","","LOCATE",True,src/tlusty/math/opacity/cia_h2h2.rs,done
|
||||
cia_h2he.f,CIA_H2HE,SUBROUTINE,False,"","LOCATE","","LOCATE",True,src/tlusty/math/opacity/cia_h2he.rs,done
|
||||
cia_hhe.f,CIA_HHE,SUBROUTINE,False,"","LOCATE","","LOCATE",True,src/tlusty/math/opacity/cia_hhe.rs,done
|
||||
cion.f,CION,FUNCTION,True,"","","","",False,src/tlusty/math/atomic/cion.rs,done
|
||||
ckoest.f,CKOEST,FUNCTION,True,"BASICS","","BASICS","",False,src/tlusty/math/interpolation/ckoest.rs,done
|
||||
colh.f,COLH,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","CSPEC|BUTLER|CEH12|IRC","ATOMIC|MODELQ|BASICS","CEH12|EXPO|CSPEC|BUTLER|QUIT|EXPINX|EINT|IRC|SZIRC",False,src/tlusty/math/hydrogen/colh.rs,done
|
||||
colhe.f,COLHE,SUBROUTINE,False,"BASICS|ATOMIC","CSPEC|IRC|COLLHE|CHEAV","ATOMIC|BASICS","EXPO|CSPEC|QUIT|COLLHE|EXPINX|EINT|CHEAVJ|IRC|SZIRC|CHEAV",False,src/tlusty/math/hydrogen/colhe.rs,done
|
||||
colis.f,COLIS,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|CTRTEMP","YLINTP|CSPEC|COLHE|CION|HCTION|COLH|IRC","MODELQ|CTRTEMP|ATOMIC|ODFPAR|BASICS|CTIon","EXPO|CSPEC|QUIT|COLLHE|EXPINX|EINT|COLH|CEH12|YLINTP|COLHE|BUTLER|CION|HCTION|CHEAVJ|IRC|SZIRC|CHEAV",False,src/tlusty/math/hydrogen/colis.rs,done
|
||||
collhe.f,COLLHE,SUBROUTINE,True,"","","","",False,src/tlusty/math/hydrogen/collhe.rs,done
|
||||
column.f,COLUMN,SUBROUTINE,False,"BASICS|MODELQ|relcor","","MODELQ|relcor|BASICS","",True,src/tlusty/math/utils/column.rs,done
|
||||
compt0.f,COMPT0,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|ITERAT|auxcbc","","ITERAT|MODELQ|BASICS|auxcbc|ALIPAR","",False,src/tlusty/math/opacity/compt0.rs,done
|
||||
comset.f,COMSET,SUBROUTINE,False,"BASICS|MODELQ|auxcbc|comgfs","ANGSET","MODELQ|BASICS|auxcbc|comgfs","ANGSET|GAULEG",False,src/tlusty/math/utils/comset.rs,done
|
||||
concor.f,CONCOR,SUBROUTINE,False,"BASICS|MODELQ","TEMCOR|CONOUT","irwint|tdedge|adiaba|POPSTR|pfoptb|eospar|tdflag|PPAPAR|ALIPAR|MODELQ|hmolab|THERM|CC|derdif|ITERAT|BASICS|terden|COMFH1|RAYSCT|ARRAY1|moldat|quasun|PFSTDS|entrop|TABLTD|CONVOUT|adchar|ATOMIC|ODFPAR|ioniz2|CUBCON","PRSENT|SFFHMI|CIA_H2H|PFFE|UBETA|LINPRO|INTHYD|RUSSEL|WNSTOR|RHOEOS|GFREE0|ELDENS|DOPGAM|PFNI|ENTENE|LINEQS|CIA_HHE|SGMER1|YLINTP|PFSPEC|TEMCOR|WN|OPADD|LOCATE|OPCTAB|OPACF0|LEVSOL|REFLEV|OPFRAC|PFCNO|TRMDER|CONVEC|CROSSD|CROSS|YINT|INTXEN|SABOLF|DIVSTR|SETTRM|PFHEAV|LAGRAN|OPACT1|CIA_H2HE|VOIGT|STARKA|MOLEQ|STATE|RAYLEIGH|TRMDRT|STARK0|STEQEQ|MPARTF|FFCROS|CIA_H2H2|MEANOPT|INTLEM|DWNFR0|GAMSP|PARTF|RATMAT|H2MINUS|PROFSP|DWNFR1|CONOUT|MEANOP",True,src/tlusty/math/convection/concor.rs,done
|
||||
conout.f,CONOUT,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|CUBCON","MEANOPT|MEANOP|CONVEC|OPACF0","irwint|tdedge|adiaba|pfoptb|eospar|tdflag|ALIPAR|MODELQ|hmolab|THERM|CC|derdif|BASICS|terden|RAYSCT|COMFH1|moldat|quasun|PFSTDS|entrop|TABLTD|CONVOUT|adchar|ATOMIC|ODFPAR|ioniz2|CUBCON","PRSENT|SFFHMI|CIA_H2H|PFFE|UBETA|LINPRO|INTHYD|RUSSEL|WNSTOR|RHOEOS|GFREE0|ELDENS|DOPGAM|PFNI|ENTENE|LINEQS|CIA_HHE|SGMER1|YLINTP|PFSPEC|OPADD|WN|OPCTAB|LOCATE|OPACF0|OPFRAC|PFCNO|TRMDER|CONVEC|CROSSD|CROSS|YINT|INTXEN|SABOLF|DIVSTR|SETTRM|PFHEAV|LAGRAN|OPACT1|CIA_H2HE|VOIGT|STARKA|MOLEQ|STATE|RAYLEIGH|TRMDRT|STARK0|MEANOPT|MPARTF|FFCROS|CIA_H2H2|INTLEM|DWNFR0|GAMSP|PARTF|H2MINUS|PROFSP|DWNFR1|MEANOP",True,src/tlusty/math/convection/conout.rs,done
|
||||
conref.f,CONREF,SUBROUTINE,False,"BASICS|MODELQ|ARRAY1|imucnn|CUBCON","WNSTOR|CONVC1|STEQEQ|ELDENS|CONVEC|TDPINI|CONOUT","irwint|tdedge|adiaba|POPSTR|pfoptb|eospar|tdflag|PPAPAR|ALIPAR|MODELQ|hmolab|THERM|CC|derdif|ITERAT|BASICS|terden|imucnn|COMFH1|RAYSCT|ARRAY1|moldat|quasun|PFSTDS|entrop|TABLTD|CONVOUT|adchar|ATOMIC|ODFPAR|ioniz2|CUBCON","PRSENT|SFFHMI|CIA_H2H|PFFE|UBETA|LINPRO|INTHYD|RUSSEL|WNSTOR|RHOEOS|GFREE0|ELDENS|DOPGAM|PFNI|ENTENE|LINEQS|CIA_HHE|SGMER1|YLINTP|PFSPEC|WN|OPADD|OPCTAB|LOCATE|OPACF0|LEVSOL|REFLEV|CONVC1|OPFRAC|PFCNO|TRMDER|CONVEC|CROSSD|CROSS|YINT|INTXEN|SABOLF|DIVSTR|SETTRM|PFHEAV|LAGRAN|OPACT1|CIA_H2HE|VOIGT|STARKA|TDPINI|MOLEQ|STATE|RAYLEIGH|TRMDRT|STARK0|STEQEQ|MPARTF|MEANOPT|FFCROS|CIA_H2H2|INTLEM|DWNFR0|GAMSP|PARTF|RATMAT|H2MINUS|PROFSP|DWNFR1|CONOUT|MEANOP",True,src/tlusty/math/convection/conref.rs,done
|
||||
contmd.f,CONTMD,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ALIPAR|PRSAUX|CUBCON","WNSTOR|HESOL6|STEQEQ|CONVEC|CUBIC|CONOUT|MEANOP|OPACF0","irwint|tdedge|adiaba|POPSTR|pfoptb|eospar|tdflag|PPAPAR|ALIPAR|MODELQ|hmolab|THERM|CC|derdif|ITERAT|BASICS|PRSAUX|terden|COMFH1|RAYSCT|moldat|quasun|PFSTDS|entrop|TABLTD|CONVOUT|adchar|ATOMIC|ODFPAR|ioniz2|CUBCON","PRSENT|SFFHMI|CIA_H2H|PFFE|UBETA|LINPRO|INTHYD|RUSSEL|MATINV|WNSTOR|RHOEOS|GFREE0|ELDENS|DOPGAM|PFNI|ENTENE|LINEQS|CIA_HHE|SGMER1|YLINTP|PFSPEC|WN|OPADD|OPCTAB|LOCATE|CUBIC|OPACF0|LEVSOL|REFLEV|HESOL6|OPFRAC|PFCNO|TRMDER|CONVEC|CROSSD|CROSS|YINT|INTXEN|SABOLF|DIVSTR|SETTRM|PFHEAV|LAGRAN|OPACT1|CIA_H2HE|VOIGT|STARKA|MOLEQ|STATE|RAYLEIGH|TRMDRT|STARK0|STEQEQ|MPARTF|MEANOPT|FFCROS|CIA_H2H2|INTLEM|DWNFR0|GAMSP|PARTF|RATMAT|H2MINUS|PROFSP|DWNFR1|CONOUT|MEANOP",True,src/tlusty/math/convection/contmd.rs,done
|
||||
contmp.f,CONTMP,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ALIPAR|ichndm|CUBCON","WNSTOR|STEQEQ|MEANOPT|RHOEOS|ELDENS|CONVEC|CUBIC|CONOUT|MEANOP|OPACF0","irwint|tdedge|adiaba|POPSTR|pfoptb|eospar|tdflag|PPAPAR|ALIPAR|MODELQ|hmolab|THERM|CC|derdif|ITERAT|BASICS|terden|COMFH1|RAYSCT|moldat|quasun|PFSTDS|entrop|TABLTD|CONVOUT|adchar|ichndm|ATOMIC|ODFPAR|ioniz2|CUBCON","PRSENT|SFFHMI|CIA_H2H|PFFE|UBETA|LINPRO|INTHYD|RUSSEL|WNSTOR|RHOEOS|GFREE0|ELDENS|DOPGAM|PFNI|ENTENE|LINEQS|CIA_HHE|SGMER1|YLINTP|PFSPEC|WN|OPADD|OPCTAB|LOCATE|CUBIC|OPACF0|LEVSOL|REFLEV|OPFRAC|PFCNO|TRMDER|CONVEC|CROSSD|CROSS|YINT|INTXEN|SABOLF|DIVSTR|SETTRM|PFHEAV|LAGRAN|OPACT1|CIA_H2HE|VOIGT|STARKA|MOLEQ|STATE|RAYLEIGH|TRMDRT|STARK0|STEQEQ|MEANOPT|MPARTF|FFCROS|CIA_H2H2|INTLEM|DWNFR0|GAMSP|PARTF|RATMAT|H2MINUS|PROFSP|DWNFR1|CONOUT|MEANOP",True,src/tlusty/math/convection/contmp.rs,done
|
||||
convc1.f,CONVC1,SUBROUTINE,False,"BASICS|CUBCON","TRMDER|TRMDRT","irwint|terden|tdedge|COMFH1|adiaba|moldat|PFSTDS|entrop|pfoptb|eospar|TABLTD|CONVOUT|tdflag|adchar|MODELQ|hmolab|THERM|CC|derdif|ATOMIC|BASICS|ioniz2|CUBCON","PRSENT|OPFRAC|PFCNO|TRMDER|PFFE|MOLEQ|STATE|RUSSEL|TRMDRT|MPARTF|RHOEOS|PFSPEC|SETTRM|ELDENS|PFHEAV|PARTF|PFNI|ENTENE|LINEQS",False,src/tlusty/math/convection/convec.rs,done
|
||||
convec.f,CONVEC,SUBROUTINE,False,"BASICS|CUBCON","TRMDER|TRMDRT","irwint|terden|tdedge|COMFH1|adiaba|moldat|PFSTDS|entrop|pfoptb|eospar|TABLTD|CONVOUT|tdflag|adchar|MODELQ|hmolab|THERM|CC|derdif|ATOMIC|BASICS|ioniz2|CUBCON","PRSENT|OPFRAC|PFCNO|TRMDER|PFFE|MOLEQ|STATE|RUSSEL|TRMDRT|MPARTF|RHOEOS|PFSPEC|SETTRM|ELDENS|PFHEAV|PARTF|PFNI|ENTENE|LINEQS",False,src/tlusty/math/convection/convec.rs,done
|
||||
coolrt.f,COOLRT,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR|ARRAY1|ITERAT|COOLCO","RTEFR1|OPACFA","COOLCO|AUXRTE|ARRAY1|eospar|EXTINT|auxcbc|ALIPAR|MODELQ|SURFEX|ITERAT|ATOMIC|ODFPAR|BASICS|OPTDPT|comgfs","SFFHMI|OPACFA|RTEFR1|CIA_H2H|RTEFE2|CROSSD|CROSS|RTEDF1|CIA_HHE|SGMER1|MATINV|YLINTP|FFCROS|CIA_H2H2|RTECF0|RTEDF2|DOPGAM|OPADD|GAMSP|LOCATE|PRD|H2MINUS|GAMI|DWNFR1|CIA_H2HE|RTESOL|RTECF1",True,src/tlusty/math/radiative/coolrt.rs,done
|
||||
corrwm.f,CORRWM,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","QUIT","ATOMIC|MODELQ|BASICS","QUIT",True,src/tlusty/math/opacity/corrwm.rs,done
|
||||
cross.f,CROSS,FUNCTION,False,"BASICS|ATOMIC|MODELQ","","ATOMIC|MODELQ|BASICS","",False,src/tlusty/math/atomic/cross.rs,done
|
||||
crossd.f,CROSSD,FUNCTION,False,"BASICS|ATOMIC|MODELQ","","ATOMIC|MODELQ|BASICS","",False,src/tlusty/math/atomic/cross.rs,done
|
||||
cspec.f,CSPEC,SUBROUTINE,False,"BASICS|ATOMIC","QUIT","ATOMIC|BASICS","QUIT",False,src/tlusty/math/opacity/cspec.rs,done
|
||||
ctdata.f,CTDATA,BLOCK DATA,False,"CTRecomb|CTIon","","CTRecomb|CTIon","",False,src/tlusty/math/hydrogen/ctdata.rs,done
|
||||
cubic.f,CUBIC,SUBROUTINE,False,"BASICS|CUBCON","","CUBCON|BASICS","",False,src/tlusty/math/solvers/cubic.rs,done
|
||||
dielrc.f,DIELRC,SUBROUTINE,True,"","","","",False,src/tlusty/math/atomic/dielrc.rs,done
|
||||
dietot.f,DIETOT,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","DIELRC","ATOMIC|MODELQ|BASICS","DIELRC",True,src/tlusty/math/atomic/dietot.rs,done
|
||||
divstr.f,DIVSTR,SUBROUTINE,False,"BASICS|MODELQ","","MODELQ|BASICS","",False,src/tlusty/math/utils/divstr.rs,done
|
||||
dmder.f,DMDER,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|DEPTDR","","DEPTDR|ATOMIC|MODELQ|BASICS","",False,src/tlusty/math/utils/dmder.rs,done
|
||||
dmeval.f,DMEVAL,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT|ARRAY1","","ITERAT|ATOMIC|MODELQ|BASICS|ARRAY1","",True,src/tlusty/math/utils/dmeval.rs,done
|
||||
dopgam.f,DOPGAM,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","GAMSP","ATOMIC|MODELQ|BASICS","GAMSP",False,src/tlusty/math/opacity/dopgam.rs,done
|
||||
dwnfr.f,DWNFR,SUBROUTINE,False,"BASICS|MODELQ","","MODELQ|BASICS","",False,src/tlusty/math/opacity/dwnfr.rs,done
|
||||
dwnfr0.f,DWNFR0,SUBROUTINE,False,"BASICS|MODELQ","","MODELQ|BASICS","",False,src/tlusty/math/opacity/dwnfr0.rs,done
|
||||
dwnfr1.f,DWNFR1,SUBROUTINE,False,"BASICS|MODELQ","","MODELQ|BASICS","",False,src/tlusty/math/opacity/dwnfr1.rs,done
|
||||
eint.f,EINT,SUBROUTINE,True,"","EXPO|EXPINX","","EXPO|EXPINX",False,src/tlusty/math/special/expint.rs,done
|
||||
elcor.f,ELCOR,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ADCHAR","MOLEQ|WNSTOR|STATE|STEQEQ","irwint|terden|COMFH1|moldat|PFSTDS|entrop|pfoptb|eospar|POPSTR|PPAPAR|adchar|MODELQ|hmolab|ITERAT|ATOMIC|ADCHAR|BASICS|ioniz2","LEVSOL|REFLEV|OPFRAC|PFCNO|PFFE|MOLEQ|STATE|RUSSEL|SABOLF|WNSTOR|STEQEQ|MPARTF|PFSPEC|PFHEAV|WN|PARTF|PFNI|RATMAT|LINEQS",True,src/tlusty/math/temperature/elcor.rs,done
|
||||
eldenc.f,ELDENC,SUBROUTINE,False,"BASICS|MODELQ|ATOMIC|hmolab|eospar|eletab","MOLEQ|RHONEN|STATE","irwint|terden|COMFH1|moldat|PFSTDS|entrop|pfoptb|eospar|eletab|adchar|MODELQ|hmolab|ATOMIC|BASICS|ioniz2","OPFRAC|PFCNO|RHONEN|PFFE|MOLEQ|STATE|RUSSEL|MPARTF|PFSPEC|ELDENS|PFHEAV|PARTF|PFNI|ENTENE|LINEQS",True,src/tlusty/math/eos/eldenc.rs,done
|
||||
eldens.f,ELDENS,SUBROUTINE,False,"BASICS|MODELQ|ATOMIC|terden|eospar","MPARTF|MOLEQ|ENTENE|STATE|LINEQS","irwint|terden|COMFH1|moldat|PFSTDS|entrop|pfoptb|eospar|adchar|MODELQ|hmolab|ATOMIC|BASICS|ioniz2","MPARTF|OPFRAC|PFSPEC|PFHEAV|PFCNO|PFFE|PARTF|PFNI|MOLEQ|ENTENE|STATE|RUSSEL|LINEQS",True,src/tlusty/math/eos/eldens.rs,done
|
||||
emat.f,EMAT,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ARRAY1|ALIPAR","","ATOMIC|MODELQ|BASICS|ARRAY1|ALIPAR","",False,src/tlusty/math/utils/emat.rs,done
|
||||
entene.f,ENTENE,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","MPARTF","moldat|ATOMIC|MODELQ|BASICS","MPARTF",False,src/tlusty/math/eos/entene.rs,done
|
||||
erfcin.f,ERFCIN,FUNCTION,True,"","ERFCX","","ERFCX",False,src/tlusty/math/special/erfcx.rs,done
|
||||
erfcx.f,ERFCX,FUNCTION,True,"","","","",False,src/tlusty/math/special/erfcx.rs,done
|
||||
expint.f,EXPINT,FUNCTION,True,"","","","",False,src/tlusty/math/special/expint.rs,done
|
||||
expinx.f,EXPINX,SUBROUTINE,True,"","","","",False,src/tlusty/math/special/expint.rs,done
|
||||
expo.f,EXPO,FUNCTION,True,"","","","",False,src/tlusty/math/special/expo.rs,done
|
||||
ffcros.f,FFCROS,FUNCTION,True,"","","","",False,src/tlusty/math/atomic/ffcros.rs,done
|
||||
gami.f,GAMI,FUNCTION,True,"","","","",False,src/tlusty/math/special/gami.rs,done
|
||||
gamsp.f,GAMSP,SUBROUTINE,True,"BASICS","","BASICS","",False,src/tlusty/math/special/gamsp.rs,done
|
||||
gauleg.f,GAULEG,SUBROUTINE,True,"","","","",False,src/tlusty/math/special/gauleg.rs,done
|
||||
gaunt.f,GAUNT,FUNCTION,True,"","","","",False,src/tlusty/math/special/gaunt.rs,done
|
||||
getlal.f,GETLAL,SUBROUTINE,False,"BASICS|callarda|callardg|calphatd|quasun|callardb|callardc","","callarda|callardg|callardc|BASICS|calphatd|quasun|callardb","",True,src/tlusty/math/utils/getlal.rs,done
|
||||
getwrd.f,GETWRD,SUBROUTINE,True,"","","","",False,src/tlusty/math/io/getwrd.rs,done
|
||||
gfree0.f,GFREE0,SUBROUTINE,False,"BASICS|MODELQ","","MODELQ|BASICS","",False,src/tlusty/math/atomic/gfree.rs,done
|
||||
gfree1.f,GFREE1,FUNCTION,False,"BASICS|MODELQ","","MODELQ|BASICS","",False,src/tlusty/math/atomic/gfree.rs,done
|
||||
gfreed.f,GFREED,SUBROUTINE,False,"BASICS|MODELQ","","MODELQ|BASICS","",False,src/tlusty/math/atomic/gfree.rs,done
|
||||
ghydop.f,GHYDOP,SUBROUTINE,False,"BASICS|MODELQ|ATOMIC|intcfg","","intcfg|ATOMIC|MODELQ|BASICS","",False,src/tlusty/math/hydrogen/ghydop.rs,done
|
||||
gntk.f,GNTK,FUNCTION,True,"","","","",False,src/tlusty/math/atomic/gntk.rs,done
|
||||
gomini.f,GOMINI,SUBROUTINE,False,"BASICS|MODELQ|intcfg","","intcfg|MODELQ|BASICS","",True,src/tlusty/math/utils/gomini.rs,done
|
||||
grcor.f,GRCOR,SUBROUTINE,True,"","","","",False,src/tlusty/math/temperature/grcor.rs,done
|
||||
greyd.f,GREYD,SUBROUTINE,False,"BASICS|MODELQ|ATOMIC|ALIPAR","WNSTOR|STEQEQ|RHONEN|MEANOP|OPACF0","irwint|terden|COMFH1|RAYSCT|moldat|quasun|PFSTDS|POPSTR|entrop|eospar|pfoptb|PPAPAR|adchar|ALIPAR|MODELQ|hmolab|ITERAT|ATOMIC|BASICS|ioniz2|ODFPAR","SFFHMI|CIA_H2H|PFFE|UBETA|LINPRO|INTHYD|RUSSEL|WNSTOR|GFREE0|ELDENS|DOPGAM|PFNI|ENTENE|LINEQS|CIA_HHE|SGMER1|YLINTP|PFSPEC|WN|OPADD|LOCATE|OPCTAB|OPACF0|LEVSOL|REFLEV|OPFRAC|PFCNO|CROSSD|CROSS|YINT|INTXEN|SABOLF|DIVSTR|PFHEAV|LAGRAN|OPACT1|CIA_H2HE|VOIGT|STARKA|RHONEN|MOLEQ|STATE|RAYLEIGH|STARK0|STEQEQ|MPARTF|FFCROS|CIA_H2H2|INTLEM|DWNFR0|GAMSP|PARTF|RATMAT|H2MINUS|PROFSP|DWNFR1|MEANOP",True,src/tlusty/math/temperature/greyd.rs,done
|
||||
gridp.f,GRIDP,SUBROUTINE,True,"BASICS","","BASICS","",False,src/tlusty/math/utils/gridp.rs,done
|
||||
h2minus.f,H2MINUS,SUBROUTINE,False,"BASICS","LOCATE","BASICS","LOCATE",True,src/tlusty/math/hydrogen/h2minus.rs,done
|
||||
hction.f,HCTION,FUNCTION,False,"CTRTEMP|CTIon","","CTRTEMP|CTIon","",False,src/tlusty/math/hydrogen/ctdata.rs,done
|
||||
hctrecom.f,HCTRECOM,FUNCTION,False,"CTRTEMP|CTRecomb","","CTRTEMP|CTRecomb","",False,src/tlusty/math/hydrogen/ctdata.rs,done
|
||||
hedif.f,HEDIF,SUBROUTINE,False,"BASICS|MODELQ|ATOMIC|hediff","","hediff|ATOMIC|MODELQ|BASICS","",True,src/tlusty/math/hydrogen/hedif.rs,done
|
||||
hephot.f,HEPHOT,FUNCTION,True,"","","","",False,src/tlusty/math/hydrogen/hephot.rs,done
|
||||
hesol6.f,HESOL6,SUBROUTINE,False,"BASICS|MODELQ|PRSAUX","MATINV","MODELQ|PRSAUX|BASICS","MATINV",False,src/tlusty/math/hydrogen/hesol6.rs,done
|
||||
hesolv.f,HESOLV,SUBROUTINE,False,"BASICS|MODELQ|PRSAUX","MATINV|WNSTOR|RHONEN|STEQEQ","irwint|PRSAUX|terden|COMFH1|moldat|PFSTDS|entrop|pfoptb|eospar|POPSTR|PPAPAR|adchar|MODELQ|hmolab|ITERAT|ATOMIC|BASICS|ioniz2","LEVSOL|REFLEV|OPFRAC|PFCNO|RHONEN|PFFE|MOLEQ|STATE|RUSSEL|SABOLF|MATINV|WNSTOR|STEQEQ|MPARTF|PFSPEC|ELDENS|PFHEAV|WN|PARTF|PFNI|RATMAT|ENTENE|LINEQS",True,src/tlusty/math/hydrogen/hesolv.rs,done
|
||||
hidalg.f,HIDALG,FUNCTION,True,"","","","",False,src/tlusty/math/hydrogen/hidalg.rs,done
|
||||
ijali2.f,IJALI2,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR","QUIT","ODFPAR|ATOMIC|MODELQ|BASICS","QUIT",True,src/tlusty/math/ali/ijali2.rs,done
|
||||
ijalis.f,IJALIS,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","","ATOMIC|MODELQ|BASICS","",True,src/tlusty/math/ali/ijalis.rs,done
|
||||
incldy.f,INCLDY,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","LEVSOL|WNSTOR|QUIT|RATMAT|SABOLF","irwint|MODELQ|moldat|PFSTDS|ITERAT|ATOMIC|pfoptb|BASICS","LEVSOL|REFLEV|WNSTOR|MPARTF|PFSPEC|OPFRAC|PFHEAV|PFCNO|QUIT|WN|PFFE|PARTF|RATMAT|PFNI|LINEQS|SABOLF",True,src/tlusty/io/incldy.rs,done
|
||||
indexx.f,INDEXX,SUBROUTINE,True,"","","","",False,src/tlusty/math/solvers/indexx.rs,done
|
||||
inicom.f,INICOM,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|comgfs","","comgfs|ATOMIC|MODELQ|BASICS","",False,src/tlusty/math/utils/inicom.rs,done
|
||||
inifrc.f,INIFRC,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ijflar","INDEXX","ATOMIC|MODELQ|ODFPAR|BASICS|ijflar","INDEXX",True,src/tlusty/math/opacity/inifrc.rs,done
|
||||
inifrs.f,INIFRS,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR","INDEXX|QUIT","ODFPAR|ATOMIC|MODELQ|BASICS","INDEXX|QUIT",True,src/tlusty/math/opacity/inifrs.rs,done
|
||||
inifrt.f,INIFRT,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ijflar","INDEXX","ijflar|ATOMIC|MODELQ|BASICS","INDEXX",True,src/tlusty/math/opacity/inifrt.rs,done
|
||||
inilam.f,INILAM,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT|ALIPAR","OPACF1|SABOLF|WNSTOR|RHOEOS|RTECOM|OUTPUT|COLIS|OPAINI|ODFMER|RTEFR1|OSCCOR|TDPINI|RATES1|VISINI|COMSET|ELCOR|STEQEQ|RYBHEQ|CONCOR|DIETOT","tdedge|POPSTR|pfoptb|tdflag|PPAPAR|MODELQ|THERM|calphatd|CC|derdif|rybpgs|BASICS|intcfg|terden|AUXRTE|RAYSCT|quasun|entrop|TABLTD|CONVOUT|grdpra|SURFEX|ADCHAR|ODFPAR|ioniz2|CUBCON|callarda|irwint|adiaba|eospar|EXTINT|ALIPAR|hmolab|ITERAT|CTIon|OPTDPT|COMFH1|ARRAY1|moldat|CTRTEMP|ipricr|callardb|PFSTDS|auxcbc|adchar|callardg|ATOMIC|callardc|comgfs","PRSENT|SFFHMI|ANGSET|CIA_H2H|COLLHE|PFFE|EXPINX|UBETA|EINT|LINPRO|RHOEOS|RTECOM|DOPGAM|RTEDF2|ELDENS|ENTENE|IRC|LINEQS|OPAINI|ODFMER|RTEFR1|SGMER0|GAULEG|TRIDAG|RTEDF1|SGMER1|CEH12|TEMCOR|COLHE|OPADD|WN|OPCTAB|PRD|CION|OPACF0|REFLEV|PGSET|PFCNO|YINT|INTXEN|GFREE1|SABOLF|LAGRAN|OPACT1|CIA_H2HE|LYMLIN|TDPINI|VISINI|TRMDRT|STARK0|STEQEQ|MPARTF|MEANOPT|GHYDOP|DWNFR0|GAMSP|PARTF|H2MINUS|PROFSP|DWNFR1|ALLARDT|CONOUT|MEANOP|RTESOL|CHEAV|RTEFE2|OPACF1|ODFHYD|INTHYD|COLH|RUSSEL|ODFHST|MATINV|WNSTOR|GFREE0|PFNI|GAMI|QUASIM|OUTPUT|EXPO|INDEXX|OSCCOR|QUIT|COMSET|CIA_HHE|YLINTP|CONCOR|PFSPEC|LOCATE|LEVGRP|LEVSOL|OPFRAC|TRMDER|CONVEC|CROSSD|CROSS|ALLARD|RTECMC|DIVSTR|ROSSTD|SETTRM|PFHEAV|DIELRC|HCTION|COLIS|CHEAVJ|VOIGT|STARKA|CSPEC|MOLEQ|RATES1|STATE|RAYLEIGH|ELCOR|RYBHEQ|FFCROS|CIA_H2H2|RTECF0|INTLEM|BUTLER|RATMAT|DIETOT|SZIRC|RTECF1",False,src/tlusty/math/opacity/inilam.rs,done
|
||||
initia.f,INITIA,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT|ODFPAR|ALIPAR|STRPAR|freqcl|INUNIT","OPAHST|INPMOD|READBF|LINSPL|INIFRT|CORRWM|DOPGAM|RTEANG|RDATAX|ODFSET|RAYINI|TABINI|ODFHYS|CHCTAB|SRTFRQ|GOMINI|SIGK|IROSET|RDATA|NSTOUT|TRAINI|QUIT|INIFRC|INTERP|TABINT|LTEGR|STATE|DMDER|SIGAVE|OPADD0|LTEGRD|CHANGE|LINSET|INPDIS|NSTPAR|INIFRS|LEVSET","DEPTDR|tdedge|POPSTR|pfoptb|tdflag|PPAPAR|eletab|STFCR|MODELQ|THERM|CC|calphatd|derdif|BASICS|intcfg|FLXAUX|terden|AUXRTE|RAYSCT|quasun|entrop|TABLTD|CONVOUT|SURFEX|COLKUR|LINED|ichndm|ODFPAR|ioniz2|CUBCON|callarda|irwint|temlim|deridt|imodlc|adiaba|TOTJHK|intcff|eospar|FACTRS|EXTINT|INUNIT|ijflar|ALIPAR|hmolab|freqcl|TOPB|ITERAT|OPTDPT|PRSAUX|imucnn|COMFH1|relcor|moldat|ipricr|callardb|PFSTDS|icnrsp|auxcbc|adchar|abntab|callardg|ifpzpa|ATOMIC|callardc|hediff|STRPAR|comgfs","INPMOD|PRSENT|SBFHE1|SFFHMI|TEMPER|CIA_H2H|PFFE|UBETA|HEPHOT|LINPRO|VERN16|LINSPL|VERN20|IJALI2|EXPINT|ERFCIN|RADTOT|INIFRT|CORRWM|RHOEOS|ELDENS|DOPGAM|RTEDF2|GRCOR|RAYINI|ODFHYS|ENTENE|CHCTAB|OPDATA|LINEQS|GOMINI|OPAINI|REIMAN|RTEFR1|SGMER0|GAULEG|VOIGTE|CKOEST|LTEGR|ODFFR|RTEDF1|SGMER1|WN|OPADD|INPDIS|OPCTAB|NEWDM|PRD|GAUNT|OPACF0|REFLEV|SPSIGK|HESOL6|RAYSET|PFCNO|HIDALG|XENINI|YINT|INTXEN|GFREE1|SABOLF|BETAH|RTEANG|INCLDY|VERN26|PSOLVE|ODFSET|TLOCAL|LAGRAN|GREYD|OPACT1|SRTFRQ|CONTMP|CIA_H2HE|IROSET|NSTOUT|LEMINI|INIFRC|LYMLIN|TDPINI|LEVCD|VERNER|TRMDRT|DMDER|STARK0|STEQEQ|MPARTF|MEANOPT|GHYDOP|SBFHMI|DWNFR0|GAMSP|PARTF|HESOLV|H2MINUS|PROFSP|DWNFR1|CONOUT|ALLARDT|MEANOP|RTESOL|READBF|COLUMN|VERN18|RTEFE2|OPACF1|PROFIL|INTHYD|RUSSEL|MATINV|WNSTOR|GFREE0|ERFCX|TABINI|PFNI|IJALIS|GAMI|QUASIM|SIGK|RDATA|INDEXX|QUIT|BKHSGO|TABINT|QUARTC|ROSSOP|NEWDMT|CIA_HHE|INKUL|GETLAL|YLINTP|LTEGRD|PFSPEC|LINSET|LOCATE|CUBIC|NSTPAR|LEVGRP|LEVSET|OPAHST|LEVSOL|OPFRAC|TRMDER|CONVEC|CROSSD|CROSS|ALLARD|DIVSTR|SETTRM|PFHEAV|KURUCZ|RDATAX|VOIGT|TRAINI|STARKA|CONTMD|INTERP|RHONEN|GETWRD|SGHE12|MOLEQ|STATE|RAYLEIGH|ZMRHO|SIGAVE|TOPBAS|FFCROS|OPADD0|CIA_H2H2|RTECF0|INTLEM|CHANGE|CARBON|RATMAT|GRIDP|INIFRS|RTECF1",True,src/tlusty/io/initia.rs,done
|
||||
inkul.f,INKUL,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|COLKUR|LINED","","ATOMIC|MODELQ|ODFPAR|BASICS|COLKUR|LINED","",True,src/tlusty/math/opacity/inkul.rs,done
|
||||
inpdis.f,INPDIS,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT|ODFPAR|ALIPAR|relcor","GRCOR|COLUMN","MODELQ|relcor|ITERAT|ATOMIC|BASICS|ODFPAR|ALIPAR","GRCOR|COLUMN",True,src/tlusty/math/opacity/inpdis.rs,done
|
||||
inpmod.f,INPMOD,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|eospar","LEVSOL|WNSTOR|INCLDY|QUIT|KURUCZ|RATMAT|MOLEQ|SABOLF","irwint|temlim|terden|COMFH1|moldat|PFSTDS|entrop|pfoptb|eospar|adchar|MODELQ|hmolab|ITERAT|ATOMIC|BASICS|ioniz2","LEVSOL|REFLEV|OPFRAC|PFCNO|PFFE|RUSSEL|SABOLF|WNSTOR|PFHEAV|INCLDY|ELDENS|KURUCZ|PFNI|ENTENE|LINEQS|QUIT|RHONEN|MOLEQ|STATE|MPARTF|PFSPEC|WN|PARTF|RATMAT",True,src/tlusty/io/inpmod.rs,done
|
||||
interp.f,INTERP,SUBROUTINE,True,"BASICS","","BASICS","",False,src/tlusty/math/interpolation/interp.rs,done
|
||||
inthyd.f,INTHYD,SUBROUTINE,False,"BASICS|MODELQ","STARKA|YINT|DIVSTR","MODELQ|BASICS","STARKA|YINT|DIVSTR",False,src/tlusty/math/hydrogen/inthyd.rs,done
|
||||
intlem.f,INTLEM,SUBROUTINE,False,"BASICS|MODELQ","INTHYD","MODELQ|BASICS","STARKA|YINT|DIVSTR|INTHYD",False,src/tlusty/math/interpolation/intlem.rs,done
|
||||
intxen.f,INTXEN,SUBROUTINE,False,"BASICS|MODELQ","YINT","MODELQ|BASICS","YINT",False,src/tlusty/math/interpolation/intxen.rs,done
|
||||
irc.f,IRC,SUBROUTINE,True,"","EXPINX|SZIRC","","EXPO|EINT|EXPINX|SZIRC",False,src/tlusty/math/utils/irc.rs,done
|
||||
iroset.f,IROSET,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|LINED","INKUL|QUIT|VOIGTE|LEVCD|IJALI2","MODELQ|COLKUR|LINED|ATOMIC|ODFPAR|BASICS","INKUL|INDEXX|QUIT|VOIGTE|WN|LEVCD|IJALI2",True,src/tlusty/io/iroset.rs,done
|
||||
kurucz.f,KURUCZ,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|temlim","LEVSOL|WNSTOR|QUIT|RHONEN|RATMAT|MOLEQ|SABOLF","temlim|irwint|terden|COMFH1|moldat|PFSTDS|entrop|pfoptb|eospar|adchar|MODELQ|hmolab|ITERAT|ATOMIC|BASICS|ioniz2","LEVSOL|REFLEV|OPFRAC|PFCNO|QUIT|RHONEN|PFFE|MOLEQ|STATE|RUSSEL|SABOLF|WNSTOR|MPARTF|PFSPEC|ELDENS|PFHEAV|WN|PARTF|RATMAT|PFNI|ENTENE|LINEQS",True,src/tlusty/io/kurucz.rs,done
|
||||
lagran.f,LAGRAN,SUBROUTINE,True,"","","","",False,src/tlusty/math/interpolation/lagran.rs,done
|
||||
laguer.f,LAGUER,SUBROUTINE,False,"","","","",True,src/tlusty/math/solvers/laguer.rs,done
|
||||
lemini.f,LEMINI,SUBROUTINE,False,"BASICS|MODELQ","","MODELQ|BASICS","",True,src/tlusty/math/opacity/lemini.rs,done
|
||||
levcd.f,LEVCD,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|COLKUR","QUIT|WN|INDEXX","ATOMIC|MODELQ|ODFPAR|BASICS|COLKUR","QUIT|WN|INDEXX",True,src/tlusty/io/levcd.rs,done
|
||||
levgrp.f,LEVGRP,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT","","ITERAT|ATOMIC|MODELQ|BASICS","",False,src/tlusty/math/opacity/levgrp.rs,done
|
||||
levset.f,LEVSET,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","QUIT","ATOMIC|MODELQ|BASICS","QUIT",False,src/tlusty/math/opacity/levset.rs,done
|
||||
levsol.f,LEVSOL,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT","LINEQS","ITERAT|ATOMIC|MODELQ|BASICS","LINEQS",False,src/tlusty/math/opacity/levsol.rs,done
|
||||
lineqs.f,LINEQS,SUBROUTINE,True,"BASICS","","BASICS","",False,src/tlusty/math/solvers/lineqs.rs,done
|
||||
linpro.f,LINPRO,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|quasun","STARK0|VOIGT|DIVSTR|DOPGAM|INTLEM|STARKA|PROFSP|INTXEN","irwint|MODELQ|moldat|quasun|PFSTDS|ATOMIC|pfoptb|ODFPAR|BASICS","VOIGT|OPFRAC|PFCNO|STARKA|PFFE|UBETA|YINT|INTHYD|INTXEN|SABOLF|STARK0|DIVSTR|MPARTF|PFSPEC|PFHEAV|DOPGAM|INTLEM|GAMSP|PARTF|PFNI|LAGRAN|PROFSP",False,src/tlusty/math/opacity/linpro.rs,done
|
||||
linsel.f,LINSEL,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR","OPAINI|QUIT|RTEFR1|OPACF1","callarda|irwint|AUXRTE|RAYSCT|moldat|quasun|ipricr|callardb|PFSTDS|pfoptb|eospar|EXTINT|auxcbc|ALIPAR|MODELQ|hmolab|callardg|SURFEX|calphatd|ITERAT|ATOMIC|callardc|ODFPAR|BASICS|OPTDPT|intcfg|comgfs","REFLEV|SFFHMI|OPFRAC|PFCNO|CIA_H2H|RTEFE2|PFFE|UBETA|OPACF1|YINT|CROSSD|CROSS|LINPRO|INTHYD|ALLARD|INTXEN|GFREE1|SABOLF|MATINV|WNSTOR|DIVSTR|PFHEAV|DOPGAM|RTEDF2|PFNI|LAGRAN|OPACT1|GAMI|QUASIM|CIA_H2HE|OPAINI|VOIGT|RTEFR1|SGMER0|STARKA|QUIT|LYMLIN|RTEDF1|RAYLEIGH|CIA_HHE|SGMER1|STARK0|MPARTF|YLINTP|PFSPEC|FFCROS|RTECF0|INTLEM|DWNFR0|CIA_H2H2|GHYDOP|WN|OPADD|GAMSP|PARTF|LOCATE|OPCTAB|PRD|DWNFR1|H2MINUS|PROFSP|LEVGRP|ALLARDT|RTESOL|RTECF1",True,src/tlusty/math/opacity/linsel.rs,done
|
||||
linset.f,LINSET,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","STARK0|DIVSTR|STARKA|QUIT|PROFIL|IJALIS","irwint|MODELQ|moldat|quasun|PFSTDS|ATOMIC|pfoptb|BASICS","VOIGT|OPFRAC|PFCNO|STARKA|QUIT|PFFE|UBETA|PROFIL|SABOLF|STARK0|DIVSTR|MPARTF|PFSPEC|PFHEAV|PARTF|PFNI|LAGRAN|IJALIS|PROFSP",True,src/tlusty/io/linset.rs,done
|
||||
linspl.f,LINSPL,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","PROFIL","irwint|MODELQ|moldat|quasun|PFSTDS|ATOMIC|pfoptb|BASICS","VOIGT|OPFRAC|PFCNO|STARKA|PFFE|UBETA|PROFIL|SABOLF|STARK0|DIVSTR|MPARTF|PFSPEC|PFHEAV|PARTF|PFNI|LAGRAN|PROFSP",False,src/tlusty/math/opacity/linspl.rs,done
|
||||
locate.f,LOCATE,SUBROUTINE,True,"","","","",False,src/tlusty/math/interpolation/locate.rs,done
|
||||
ltegr.f,LTEGR,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","WNSTOR|STEQEQ|QUIT|INTERP|ROSSOP|CONOUT|CONTMP","irwint|tdedge|adiaba|POPSTR|pfoptb|eospar|tdflag|PPAPAR|ALIPAR|MODELQ|hmolab|THERM|CC|derdif|ITERAT|BASICS|terden|COMFH1|RAYSCT|moldat|quasun|PFSTDS|entrop|TABLTD|CONVOUT|adchar|ichndm|ATOMIC|ODFPAR|ioniz2|CUBCON","PRSENT|SFFHMI|CIA_H2H|PFFE|UBETA|LINPRO|INTHYD|RUSSEL|EXPINT|WNSTOR|RHOEOS|GFREE0|ELDENS|DOPGAM|PFNI|ENTENE|LINEQS|QUIT|ROSSOP|CIA_HHE|SGMER1|YLINTP|PFSPEC|WN|OPADD|OPCTAB|LOCATE|CUBIC|OPACF0|LEVSOL|REFLEV|OPFRAC|PFCNO|TRMDER|CONVEC|CROSSD|CROSS|YINT|INTXEN|SABOLF|DIVSTR|SETTRM|PFHEAV|LAGRAN|OPACT1|CONTMP|CIA_H2HE|VOIGT|STARKA|INTERP|MOLEQ|STATE|RAYLEIGH|TRMDRT|STARK0|STEQEQ|MPARTF|MEANOPT|FFCROS|CIA_H2H2|INTLEM|DWNFR0|GAMSP|PARTF|RATMAT|H2MINUS|PROFSP|DWNFR1|CONOUT|MEANOP",True,src/tlusty/io/ltegr.rs,done
|
||||
ltegrd.f,LTEGRD,SUBROUTINE,False,"BASICS|MODELQ|PRSAUX|TOTJHK|FLXAUX|FACTRS|CUBCON","RADTOT|TEMPER|WNSTOR|STEQEQ|ELDENS|CONTMD|PSOLVE|QUIT|INTERP|NEWDM|GREYD|NEWDMT|HESOLV|CONOUT|ZMRHO","callarda|irwint|tdedge|TOTJHK|adiaba|POPSTR|pfoptb|eospar|tdflag|FACTRS|EXTINT|PPAPAR|ALIPAR|MODELQ|hmolab|THERM|calphatd|CC|derdif|ITERAT|BASICS|OPTDPT|intcfg|FLXAUX|PRSAUX|terden|AUXRTE|RAYSCT|COMFH1|moldat|quasun|ipricr|callardb|PFSTDS|entrop|TABLTD|comgfs|CONVOUT|auxcbc|adchar|callardg|SURFEX|ATOMIC|callardc|ODFPAR|ioniz2|CUBCON","PRSENT|TEMPER|SFFHMI|CIA_H2H|PFFE|UBETA|LINPRO|ERFCIN|RADTOT|RHOEOS|ELDENS|DOPGAM|RTEDF2|ENTENE|LINEQS|OPAINI|RTEFR1|SGMER0|RTEDF1|SGMER1|WN|NEWDM|OPADD|OPCTAB|PRD|OPACF0|REFLEV|HESOL6|PFCNO|YINT|INTXEN|GFREE1|SABOLF|BETAH|PSOLVE|TLOCAL|LAGRAN|GREYD|OPACT1|CIA_H2HE|LYMLIN|TDPINI|TRMDRT|STARK0|STEQEQ|MPARTF|MEANOPT|GHYDOP|DWNFR0|GAMSP|PARTF|HESOLV|H2MINUS|PROFSP|DWNFR1|ALLARDT|CONOUT|MEANOP|RTESOL|RTEFE2|OPACF1|INTHYD|RUSSEL|MATINV|WNSTOR|GFREE0|ERFCX|PFNI|GAMI|QUASIM|QUIT|QUARTC|NEWDMT|CIA_HHE|YLINTP|PFSPEC|LOCATE|CUBIC|LEVGRP|LEVSOL|OPFRAC|TRMDER|CONVEC|CROSSD|CROSS|ALLARD|DIVSTR|SETTRM|PFHEAV|VOIGT|CONTMD|STARKA|INTERP|RHONEN|MOLEQ|STATE|RAYLEIGH|ZMRHO|FFCROS|CIA_H2H2|RTECF0|INTLEM|RATMAT|GRIDP|RTECF1",True,src/tlusty/io/ltegrd.rs,done
|
||||
lucy.f,LUCY,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ITERAT|ALIPAR|ARRAY1","OPAINI|WNSTOR|STEQEQ|CONCOR|ODFMER|RTEFR1|TDPINI|ELCOR|OPACFL|SABOLF|COLIS","irwint|tdedge|adiaba|POPSTR|pfoptb|eospar|tdflag|EXTINT|PPAPAR|ALIPAR|MODELQ|hmolab|THERM|CC|derdif|ITERAT|BASICS|CTIon|OPTDPT|terden|AUXRTE|COMFH1|RAYSCT|ARRAY1|moldat|quasun|CTRTEMP|PFSTDS|entrop|TABLTD|CONVOUT|comgfs|auxcbc|adchar|SURFEX|ATOMIC|ADCHAR|ODFPAR|ioniz2|CUBCON","PRSENT|SFFHMI|CIA_H2H|COLLHE|RTEFE2|PFFE|UBETA|EXPINX|ODFHYD|EINT|LINPRO|INTHYD|COLH|SZIRC|RUSSEL|ODFHST|MATINV|WNSTOR|RHOEOS|GFREE0|ELDENS|DOPGAM|RTEDF2|PFNI|ENTENE|IRC|LINEQS|OPAINI|INDEXX|EXPO|ODFMER|RTEFR1|SGMER0|QUIT|RTEDF1|CIA_HHE|SGMER1|CEH12|YLINTP|CONCOR|PFSPEC|TEMCOR|COLHE|WN|OPADD|LOCATE|OPCTAB|CION|LEVGRP|OPACF0|REFLEV|LEVSOL|OPFRAC|PFCNO|TRMDER|CONVEC|YINT|CROSSD|CROSS|INTXEN|SABOLF|DIVSTR|SETTRM|PFHEAV|LAGRAN|OPACT1|HCTION|OPACFL|CIA_H2HE|COLIS|CHEAVJ|VOIGT|STARKA|CSPEC|TDPINI|MOLEQ|CHEAV|STATE|RAYLEIGH|TRMDRT|ELCOR|STARK0|STEQEQ|MPARTF|FFCROS|CIA_H2H2|MEANOPT|RTECF0|INTLEM|DWNFR0|BUTLER|GAMSP|PARTF|RATMAT|H2MINUS|PROFSP|DWNFR1|CONOUT|MEANOP|RTESOL|RTECF1",True,src/tlusty/math/temperature/lucy.rs,done
|
||||
lymlin.f,LYMLIN,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","STARKA|STARK0|DIVSTR","ATOMIC|MODELQ|BASICS","STARKA|STARK0|DIVSTR",True,src/tlusty/math/hydrogen/lymlin.rs,done
|
||||
matcon.f,MATCON,SUBROUTINE,False,"BASICS|MODELQ|ARRAY1|CUBCON","CONVEC","irwint|terden|tdedge|COMFH1|adiaba|ARRAY1|moldat|PFSTDS|entrop|pfoptb|eospar|TABLTD|CONVOUT|tdflag|adchar|MODELQ|hmolab|THERM|CC|derdif|ATOMIC|BASICS|ioniz2|CUBCON","PRSENT|OPFRAC|PFCNO|TRMDER|CONVEC|PFFE|MOLEQ|STATE|RUSSEL|TRMDRT|MPARTF|RHOEOS|PFSPEC|SETTRM|ELDENS|PFHEAV|PARTF|PFNI|ENTENE|LINEQS",False,src/tlusty/math/solvers/matcon.rs,done
|
||||
matgen.f,MATGEN,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ARRAY1|ALIPAR","MATCON|BHED|BHEZ|BHE|BRTEZ|BPOP|EMAT|BRTE|BRE|BREZ|SABOLF","irwint|tdedge|adiaba|pfoptb|eospar|tdflag|ALIPAR|MODELQ|hmolab|THERM|CC|derdif|ITERAT|BASICS|CTIon|CMATZD|terden|COMFH1|ARRAY1|moldat|CTRTEMP|PFSTDS|entrop|TABLTD|CONVOUT|auxcbc|adchar|SURFEX|ATOMIC|ADCHAR|ODFPAR|ioniz2|CUBCON","PRSENT|COLLHE|BRTEZ|PFFE|EXPINX|EINT|COLH|RUSSEL|MATINV|RHOEOS|ELDENS|PFNI|ENTENE|IRC|LINEQS|EXPO|QUIT|BPOPE|BPOPC|BPOPT|SGMER1|CEH12|YLINTP|PFSPEC|BHEZ|BHE|COLHE|COMPT0|CION|BRTE|LEVGRP|LEVSOL|REFLEV|BHED|OPFRAC|PFCNO|TRMDER|CONVEC|EMAT|CROSS|BREZ|SABOLF|SETTRM|PFHEAV|HCTION|CHEAVJ|MATCON|COLIS|CSPEC|BPOPF|MOLEQ|STATE|TRMDRT|MPARTF|BUTLER|PARTF|BPOP|RATMAT|DWNFR1|BRE|SZIRC|CHEAV",False,src/tlusty/math/solvers/matgen.rs,done
|
||||
matinv.f,MATINV,SUBROUTINE,True,"BASICS","","BASICS","",False,src/tlusty/math/solvers/matinv.rs,done
|
||||
meanop.f,MEANOP,SUBROUTINE,False,"BASICS|MODELQ|ATOMIC","","ATOMIC|MODELQ|BASICS","",False,src/tlusty/math/opacity/meanop.rs,done
|
||||
meanopt.f,MEANOPT,SUBROUTINE,False,"BASICS|MODELQ","OPCTAB","ATOMIC|MODELQ|eospar|BASICS|RAYSCT","RAYLEIGH|OPCTAB",False,src/tlusty/math/opacity/meanopt.rs,done
|
||||
minv3.f,MINV3,SUBROUTINE,True,"","","","",False,src/tlusty/math/solvers/minv3.rs,done
|
||||
moleq.f,MOLEQ,SUBROUTINE,False,"BASICS|MODELQ|ATOMIC|hmolab|terden|COMFH1|moldat|entrop|eospar|ioniz2|adchar","RUSSEL|MPARTF","terden|MODELQ|hmolab|COMFH1|moldat|ATOMIC|entrop|eospar|BASICS|ioniz2|adchar","RUSSEL|MPARTF",True,src/tlusty/math/eos/moleq.rs,done
|
||||
mpartf.f,MPARTF,SUBROUTINE,False,"moldat","","moldat","",True,src/tlusty/math/partition/mpartf.rs,done
|
||||
newdm.f,NEWDM,SUBROUTINE,False,"BASICS|MODELQ|FACTRS|PRSAUX|FLXAUX","TEMPER|INTERP|HESOLV","irwint|tdedge|POPSTR|pfoptb|eospar|tdflag|FACTRS|PPAPAR|ALIPAR|MODELQ|hmolab|THERM|ITERAT|BASICS|FLXAUX|PRSAUX|terden|COMFH1|RAYSCT|moldat|quasun|PFSTDS|entrop|TABLTD|adchar|ATOMIC|ODFPAR|ioniz2","PRSENT|TEMPER|SFFHMI|CIA_H2H|PFFE|UBETA|LINPRO|INTHYD|RUSSEL|MATINV|WNSTOR|RHOEOS|GFREE0|ELDENS|DOPGAM|PFNI|ENTENE|LINEQS|QUARTC|CIA_HHE|SGMER1|YLINTP|PFSPEC|WN|OPADD|OPCTAB|LOCATE|OPACF0|LEVSOL|REFLEV|OPFRAC|PFCNO|CROSSD|CROSS|YINT|INTXEN|SABOLF|DIVSTR|SETTRM|PFHEAV|TLOCAL|LAGRAN|OPACT1|CIA_H2HE|VOIGT|STARKA|INTERP|RHONEN|MOLEQ|STATE|RAYLEIGH|STARK0|STEQEQ|MEANOPT|MPARTF|FFCROS|CIA_H2H2|INTLEM|DWNFR0|GAMSP|PARTF|RATMAT|HESOLV|H2MINUS|PROFSP|DWNFR1|MEANOP",True,src/tlusty/math/utils/newdm.rs,done
|
||||
newdmt.f,NEWDMT,SUBROUTINE,False,"BASICS|MODELQ|FACTRS|PRSAUX|FLXAUX","GRIDP|TEMPER|INTERP|HESOLV","irwint|tdedge|POPSTR|pfoptb|eospar|tdflag|FACTRS|PPAPAR|ALIPAR|MODELQ|hmolab|THERM|ITERAT|BASICS|FLXAUX|PRSAUX|terden|COMFH1|RAYSCT|moldat|quasun|PFSTDS|entrop|TABLTD|adchar|ATOMIC|ODFPAR|ioniz2","PRSENT|TEMPER|SFFHMI|CIA_H2H|PFFE|UBETA|LINPRO|INTHYD|RUSSEL|MATINV|WNSTOR|RHOEOS|GFREE0|ELDENS|DOPGAM|PFNI|ENTENE|LINEQS|QUARTC|CIA_HHE|SGMER1|YLINTP|PFSPEC|WN|OPADD|OPCTAB|LOCATE|OPACF0|LEVSOL|REFLEV|OPFRAC|PFCNO|CROSSD|CROSS|YINT|INTXEN|SABOLF|DIVSTR|SETTRM|PFHEAV|TLOCAL|LAGRAN|OPACT1|CIA_H2HE|VOIGT|STARKA|INTERP|RHONEN|MOLEQ|STATE|RAYLEIGH|STARK0|STEQEQ|MEANOPT|MPARTF|FFCROS|CIA_H2H2|INTLEM|DWNFR0|GAMSP|PARTF|GRIDP|RATMAT|HESOLV|H2MINUS|PROFSP|DWNFR1|MEANOP",True,src/tlusty/math/utils/newdmt.rs,done
|
||||
newpop.f,NEWPOP,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT","","ITERAT|ATOMIC|MODELQ|BASICS","",True,src/tlusty/math/population/newpop.rs,done
|
||||
nstout.f,NSTOUT,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT|ODFPAR|ALIPAR","QUIT","ITERAT|ATOMIC|MODELQ|BASICS|ODFPAR|ALIPAR","QUIT",True,src/tlusty/io/nstout.rs,done
|
||||
nstpar.f,NSTPAR,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT|ODFPAR|ALIPAR|irwint|deridt|freqcl|imucnn|temlim|adiaba|moldat|quasun|ichndm|ipricr|derdif|ifpzpa|hediff|icnrsp|FLXAUX","GETLAL|QUIT|GETWRD","callarda|irwint|deridt|temlim|imucnn|adiaba|moldat|quasun|ipricr|callardb|icnrsp|ALIPAR|MODELQ|freqcl|callardg|calphatd|ichndm|derdif|ITERAT|ifpzpa|ATOMIC|hediff|callardc|BASICS|ODFPAR|FLXAUX","GETWRD|GETLAL|QUIT",True,src/tlusty/io/nstpar.rs,done
|
||||
odf1.f,ODF1,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR","SIGK|DIVSTR|ODFHST","ATOMIC|MODELQ|ODFPAR|BASICS|TOPB","SBFHE1|SPSIGK|REIMAN|HIDALG|QUIT|VERN18|SGHE12|CKOEST|HEPHOT|VERN16|VERN20|ODFHST|VERNER|TOPBAS|DIVSTR|YLINTP|SBFHMI|VERN26|CARBON|GAUNT|OPDATA|SIGK",True,src/tlusty/math/odf/odf1.rs,done
|
||||
odffr.f,ODFFR,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR","QUIT","ODFPAR|ATOMIC|MODELQ|BASICS","QUIT",False,src/tlusty/math/odf/odffr.rs,done
|
||||
odfhst.f,ODFHST,SUBROUTINE,False,"BASICS|MODELQ|ODFPAR","","ODFPAR|MODELQ|BASICS","",False,src/tlusty/math/odf/odfhst.rs,done
|
||||
odfhyd.f,ODFHYD,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR","INDEXX|DIVSTR|ODFHST","ATOMIC|MODELQ|ODFPAR|BASICS","INDEXX|DIVSTR|ODFHST",False,src/tlusty/math/odf/odfhyd.rs,done
|
||||
odfhys.f,ODFHYS,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR","IJALIS|STARK0|ODFFR","ATOMIC|MODELQ|ODFPAR|BASICS","STARK0|QUIT|ODFFR|IJALIS",False,src/tlusty/math/odf/odfhys.rs,done
|
||||
odfmer.f,ODFMER,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR","ODFHYD","MODELQ|ATOMIC|ODFPAR|BASICS","ODFHYD|INDEXX|DIVSTR|ODFHST",False,src/tlusty/math/odf/odfmer.rs,done
|
||||
odfset.f,ODFSET,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|STFCR","QUIT|IJALIS","STFCR|ATOMIC|MODELQ|ODFPAR|BASICS","QUIT|IJALIS",True,src/tlusty/io/odfset.rs,done
|
||||
opacf0.f,OPACF0,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR|hmolab","WNSTOR|GFREE0|SFFHMI|FFCROS|DWNFR0|OPADD|OPACT1|CROSS|CROSSD|LINPRO|DWNFR1|SGMER1|SABOLF","irwint|RAYSCT|moldat|quasun|PFSTDS|pfoptb|eospar|ALIPAR|MODELQ|hmolab|ATOMIC|ODFPAR|BASICS","SFFHMI|OPFRAC|CIA_H2H|PFCNO|PFFE|UBETA|CROSSD|CROSS|YINT|LINPRO|INTHYD|INTXEN|SABOLF|WNSTOR|GFREE0|DIVSTR|PFHEAV|DOPGAM|PFNI|LAGRAN|OPACT1|CIA_H2HE|VOIGT|STARKA|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|CIA_H2H2|MPARTF|PFSPEC|INTLEM|DWNFR0|OPADD|WN|GAMSP|LOCATE|OPCTAB|PARTF|H2MINUS|PROFSP|DWNFR1",False,src/tlusty/math/continuum/opacf0.rs,done
|
||||
opacf1.f,OPACF1,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR|hmolab|ipricr","SFFHMI|FFCROS|QUASIM|GHYDOP|LYMLIN|OPADD|PRD|OPACT1|CROSS|CROSSD|GFREE1|DWNFR1|SGMER1","callarda|RAYSCT|quasun|ipricr|callardb|eospar|ALIPAR|MODELQ|hmolab|callardg|calphatd|ITERAT|ATOMIC|callardc|ODFPAR|BASICS|intcfg","SFFHMI|QUASIM|STARKA|CIA_H2H|LYMLIN|CROSSD|CROSS|ALLARD|RAYLEIGH|CIA_HHE|GFREE1|SGMER1|STARK0|YLINTP|FFCROS|DIVSTR|CIA_H2H2|GHYDOP|DOPGAM|OPADD|GAMSP|LOCATE|OPCTAB|PRD|OPACT1|H2MINUS|GAMI|DWNFR1|ALLARDT|CIA_H2HE",True,src/tlusty/math/continuum/opacf1.rs,done
|
||||
opacfa.f,OPACFA,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR|COOLCO","SFFHMI|FFCROS|OPADD|PRD|CROSSD|CROSS|DWNFR1|SGMER1","MODELQ|COOLCO|ITERAT|ATOMIC|eospar|ODFPAR|BASICS|ALIPAR","SFFHMI|YLINTP|FFCROS|CIA_H2H2|CIA_H2H|DOPGAM|OPADD|GAMSP|LOCATE|PRD|CROSSD|CROSS|H2MINUS|GAMI|DWNFR1|CIA_HHE|SGMER1|CIA_H2HE",False,src/tlusty/math/continuum/opacfa.rs,done
|
||||
opacfd.f,OPACFD,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR|ARRAY1|ITERAT|rhoder|hmolab|dsctva","SFFHMI|FFCROS|GFREED|QUASIM|LYMLIN|OPADD|OPCTAB|PRD|CROSSD|CROSS|OPACTD|DWNFR1|SGMER1","callarda|RAYSCT|ARRAY1|quasun|callardb|eospar|ALIPAR|rhoder|MODELQ|hmolab|callardg|calphatd|ITERAT|ATOMIC|callardc|dsctva|ODFPAR|BASICS","SFFHMI|QUASIM|STARKA|CIA_H2H|LYMLIN|CROSSD|CROSS|ALLARD|OPACTD|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|GFREED|DIVSTR|CIA_H2H2|DOPGAM|OPADD|GAMSP|OPCTAB|LOCATE|PRD|H2MINUS|GAMI|DWNFR1|ALLARDT|CIA_H2HE",True,src/tlusty/math/continuum/opacfd.rs,done
|
||||
opacfl.f,OPACFL,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR","SFFHMI|FFCROS|OPADD|CROSSD|CROSS|DWNFR1|SGMER1","MODELQ|ATOMIC|eospar|ODFPAR|BASICS|ALIPAR","SFFHMI|YLINTP|FFCROS|CIA_H2H2|CIA_H2H|OPADD|LOCATE|CROSSD|CROSS|H2MINUS|DWNFR1|CIA_HHE|SGMER1|CIA_H2HE",False,src/tlusty/math/continuum/opacfl.rs,done
|
||||
opact1.f,OPACT1,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|hmolab","OPCTAB","MODELQ|hmolab|RAYSCT|ATOMIC|eospar|BASICS|ALIPAR","RAYLEIGH|OPCTAB",False,src/tlusty/math/continuum/opact1.rs,done
|
||||
opactd.f,OPACTD,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|ARRAY1|ITERAT|rhoder|hmolab|dsctva","OPCTAB","rhoder|MODELQ|hmolab|RAYSCT|ARRAY1|ITERAT|ATOMIC|eospar|dsctva|BASICS|ALIPAR","RAYLEIGH|OPCTAB",False,src/tlusty/math/continuum/opactd.rs,done
|
||||
opactr.f,OPACTR,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|ATOMIC|grdpra|hmolab|dsctva","LEVSOL|OPAINI|WNSTOR|STEQEQ|PGSET|ELDENS|TDPINI|OPACF1|RATMAL|SABOLF","callarda|irwint|POPSTR|pfoptb|eospar|PPAPAR|ALIPAR|MODELQ|hmolab|calphatd|rybpgs|ITERAT|BASICS|intcfg|terden|COMFH1|RAYSCT|moldat|quasun|ipricr|callardb|PFSTDS|entrop|adchar|grdpra|callardg|ATOMIC|callardc|dsctva|ODFPAR|ioniz2","SFFHMI|CIA_H2H|PFFE|UBETA|OPACF1|LINPRO|INTHYD|RUSSEL|WNSTOR|GFREE0|ELDENS|DOPGAM|PFNI|ENTENE|GAMI|QUASIM|LINEQS|OPAINI|SGMER0|TRIDAG|CIA_HHE|SGMER1|YLINTP|PFSPEC|WN|OPADD|LOCATE|OPCTAB|PRD|RATMAL|LEVGRP|LEVSOL|REFLEV|PGSET|OPFRAC|PFCNO|YINT|CROSSD|CROSS|ALLARD|INTXEN|GFREE1|SABOLF|DIVSTR|PFHEAV|LAGRAN|OPACT1|CIA_H2HE|VOIGT|STARKA|LYMLIN|TDPINI|MOLEQ|STATE|RAYLEIGH|STARK0|STEQEQ|MPARTF|FFCROS|CIA_H2H2|GHYDOP|INTLEM|DWNFR0|GAMSP|PARTF|RATMAT|H2MINUS|PROFSP|DWNFR1|ALLARDT",False,src/tlusty/math/continuum/opactr.rs,done
|
||||
opadd.f,OPADD,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|eospar","SFFHMI|CIA_H2H2|CIA_H2H|CROSS|H2MINUS|CIA_HHE|CIA_H2HE","ATOMIC|MODELQ|eospar|BASICS","SFFHMI|YLINTP|CIA_H2H2|CIA_H2H|LOCATE|CROSS|H2MINUS|CIA_HHE|CIA_H2HE",False,src/tlusty/math/continuum/opadd.rs,done
|
||||
opadd0.f,OPADD0,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","QUIT","ATOMIC|MODELQ|BASICS","QUIT",False,src/tlusty/math/continuum/opadd0.rs,done
|
||||
opahst.f,OPAHST,SUBROUTINE,False,"BASICS|ODFPAR","STARK0","ODFPAR|BASICS","STARK0",True,src/tlusty/math/continuum/opahst.rs,done
|
||||
opaini.f,OPAINI,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR","REFLEV|WNSTOR|SGMER0|DWNFR0|LINPRO|LEVGRP|SABOLF","irwint|MODELQ|moldat|quasun|PFSTDS|ITERAT|ATOMIC|pfoptb|ODFPAR|BASICS|ALIPAR","REFLEV|VOIGT|OPFRAC|SGMER0|PFCNO|STARKA|PFFE|UBETA|YINT|LINPRO|INTHYD|INTXEN|SABOLF|STARK0|WNSTOR|DIVSTR|MPARTF|PFSPEC|PFHEAV|DOPGAM|DWNFR0|INTLEM|WN|GAMSP|PARTF|PFNI|LAGRAN|PROFSP|LEVGRP",False,src/tlusty/math/continuum/opaini.rs,done
|
||||
opctab.f,OPCTAB,SUBROUTINE,False,"BASICS|MODELQ","RAYLEIGH","ATOMIC|MODELQ|eospar|BASICS|RAYSCT","RAYLEIGH",False,src/tlusty/math/continuum/opctab.rs,done
|
||||
opdata.f,OPDATA,SUBROUTINE,False,"TOPB","","TOPB","",True,src/tlusty/math/continuum/opdata.rs,done
|
||||
opfrac.f,OPFRAC,SUBROUTINE,False,"pfoptb","","pfoptb","",True,src/tlusty/math/continuum/opfrac.rs,done
|
||||
osccor.f,OSCCOR,SUBROUTINE,False,"BASICS|MODELQ|ITERAT","","ITERAT|MODELQ|BASICS","",True,src/tlusty/math/temperature/osccor.rs,done
|
||||
outpri.f,OUTPRI,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ARRAY1|grdpra","LEVSOL|WNSTOR|OPACF1|RATMAL|SABOLF|ELDENC","callarda|irwint|pfoptb|eospar|eletab|ALIPAR|MODELQ|hmolab|calphatd|ITERAT|BASICS|intcfg|terden|RAYSCT|COMFH1|ARRAY1|moldat|quasun|ipricr|callardb|PFSTDS|entrop|adchar|grdpra|callardg|ATOMIC|callardc|ODFPAR|ioniz2","LEVSOL|SFFHMI|OPFRAC|CIA_H2H|PFCNO|PFFE|OPACF1|CROSSD|CROSS|ALLARD|RUSSEL|GFREE1|SABOLF|ELDENC|WNSTOR|DIVSTR|PFHEAV|DOPGAM|ELDENS|PFNI|OPACT1|ENTENE|GAMI|QUASIM|LINEQS|CIA_H2HE|STARKA|LYMLIN|RHONEN|MOLEQ|STATE|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|CIA_H2H2|MPARTF|GHYDOP|PFSPEC|WN|OPADD|GAMSP|LOCATE|OPCTAB|PRD|PARTF|RATMAL|H2MINUS|DWNFR1|ALLARDT",True,src/tlusty/io/outpri.rs,done
|
||||
output.f,OUTPUT,SUBROUTINE,False,"BASICS|MODELQ","","MODELQ|BASICS","",True,src/tlusty/math/io/output.rs,done
|
||||
partf.f,PARTF,SUBROUTINE,False,"BASICS|irwint|PFSTDS","MPARTF|PFSPEC|OPFRAC|PFCNO|PFHEAV|PFFE|PFNI","irwint|pfoptb|BASICS|moldat|PFSTDS","MPARTF|PFSPEC|OPFRAC|PFHEAV|PFCNO|PFFE|PFNI",False,src/tlusty/math/partition/partf.rs,done
|
||||
pfcno.f,PFCNO,SUBROUTINE,True,"BASICS","","BASICS","",False,src/tlusty/math/partition/pfcno.rs,done
|
||||
pffe.f,PFFE,SUBROUTINE,True,"","","","",False,src/tlusty/math/partition/pffe.rs,done
|
||||
pfheav.f,PFHEAV,SUBROUTINE,False,"","","","",True,src/tlusty/math/partition/pfheav.rs,done
|
||||
pfni.f,PFNI,SUBROUTINE,True,"","","","",False,src/tlusty/math/partition/pfni.rs,done
|
||||
pfspec.f,PFSPEC,SUBROUTINE,True,"","","","",False,src/tlusty/math/partition/pfspec.rs,done
|
||||
pgset.f,PGSET,SUBROUTINE,False,"BASICS|ITERAT|MODELQ|grdpra|rybpgs","TRIDAG","ITERAT|grdpra|MODELQ|BASICS|rybpgs","TRIDAG",True,src/tlusty/math/utils/pgset.rs,done
|
||||
prchan.f,PRCHAN,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT","","ITERAT|ATOMIC|MODELQ|BASICS","",True,src/tlusty/math/io/prchan.rs,done
|
||||
prd.f,PRD,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT","DOPGAM|GAMI","ITERAT|ATOMIC|MODELQ|BASICS","DOPGAM|GAMI|GAMSP",False,src/tlusty/math/opacity/prd.rs,done
|
||||
prdini.f,PRDINI,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","","ATOMIC|MODELQ|BASICS","",False,src/tlusty/math/opacity/prdini.rs,done
|
||||
princ.f,PRINC,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ALIPAR","OPACF1|DWNFR|CROSS|LINPRO|SABOLF","callarda|irwint|RAYSCT|moldat|quasun|ipricr|callardb|PFSTDS|pfoptb|eospar|ALIPAR|MODELQ|hmolab|callardg|calphatd|ITERAT|ATOMIC|callardc|BASICS|ODFPAR|intcfg","SFFHMI|OPFRAC|CIA_H2H|PFCNO|PFFE|UBETA|OPACF1|CROSSD|CROSS|YINT|LINPRO|ALLARD|INTHYD|INTXEN|GFREE1|SABOLF|DIVSTR|PFHEAV|DOPGAM|PFNI|LAGRAN|OPACT1|GAMI|QUASIM|CIA_H2HE|VOIGT|STARKA|LYMLIN|DWNFR|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|CIA_H2H2|MPARTF|GHYDOP|PFSPEC|INTLEM|OPADD|GAMSP|LOCATE|OPCTAB|PRD|PARTF|H2MINUS|PROFSP|DWNFR1|ALLARDT",True,src/tlusty/math/io/princ.rs,done
|
||||
prnt.f,PRNT,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","SABOLF","irwint|MODELQ|moldat|PFSTDS|ATOMIC|pfoptb|BASICS","MPARTF|PFSPEC|OPFRAC|PFHEAV|PFCNO|PFFE|PARTF|PFNI|SABOLF",True,src/tlusty/math/io/prnt.rs,done
|
||||
profil.f,PROFIL,FUNCTION,False,"BASICS|ATOMIC|MODELQ|quasun","STARK0|VOIGT|DIVSTR|STARKA|PROFSP","irwint|MODELQ|moldat|quasun|PFSTDS|ATOMIC|pfoptb|BASICS","STARK0|VOIGT|DIVSTR|MPARTF|PFSPEC|OPFRAC|PFHEAV|STARKA|PFCNO|PFFE|UBETA|PARTF|PFNI|LAGRAN|PROFSP|SABOLF",False,src/tlusty/math/opacity/profil.rs,done
|
||||
profsp.f,PROFSP,FUNCTION,False,"BASICS|ATOMIC|MODELQ","VOIGT|SABOLF|UBETA","irwint|MODELQ|moldat|PFSTDS|ATOMIC|pfoptb|BASICS","VOIGT|MPARTF|PFSPEC|OPFRAC|PFHEAV|PFCNO|PFFE|UBETA|PARTF|PFNI|LAGRAN|SABOLF",False,src/tlusty/math/opacity/profsp.rs,done
|
||||
prsent.f,PRSENT,SUBROUTINE,False,"TABLTD|tdedge|THERM|tdflag","","TABLTD|tdedge|THERM|tdflag","",True,src/tlusty/math/io/prsent.rs,done
|
||||
psolve.f,PSOLVE,SUBROUTINE,False,"BASICS|MODELQ","","MODELQ|BASICS","",False,src/tlusty/math/solvers/psolve.rs,done
|
||||
pzert.f,PZERT,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","","ATOMIC|MODELQ|BASICS","",False,src/tlusty/math/io/pzert.rs,done
|
||||
pzeval.f,PZEVAL,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|icnrsp","CONOUT|CONREF","irwint|tdedge|adiaba|POPSTR|pfoptb|eospar|tdflag|PPAPAR|ALIPAR|MODELQ|hmolab|THERM|CC|derdif|ITERAT|BASICS|terden|imucnn|RAYSCT|COMFH1|ARRAY1|moldat|quasun|PFSTDS|entrop|TABLTD|CONVOUT|icnrsp|adchar|ATOMIC|ODFPAR|ioniz2|CUBCON","PRSENT|SFFHMI|CIA_H2H|PFFE|UBETA|LINPRO|INTHYD|RUSSEL|WNSTOR|RHOEOS|GFREE0|ELDENS|DOPGAM|PFNI|ENTENE|LINEQS|CIA_HHE|SGMER1|YLINTP|PFSPEC|OPADD|WN|OPCTAB|LOCATE|CONREF|OPACF0|LEVSOL|REFLEV|CONVC1|OPFRAC|PFCNO|TRMDER|CONVEC|CROSSD|CROSS|YINT|INTXEN|SABOLF|DIVSTR|SETTRM|PFHEAV|LAGRAN|OPACT1|CIA_H2HE|VOIGT|STARKA|TDPINI|MOLEQ|STATE|RAYLEIGH|TRMDRT|STARK0|STEQEQ|MEANOPT|MPARTF|FFCROS|CIA_H2H2|INTLEM|DWNFR0|GAMSP|PARTF|RATMAT|H2MINUS|PROFSP|DWNFR1|CONOUT|MEANOP",True,src/tlusty/math/io/pzeval.rs,done
|
||||
pzevld.f,PZEVLD,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ALIPAR|ARRAY1|DEPTDR|grdpra|ifpzpa|PRSAUX","","grdpra|MODELQ|DEPTDR|PRSAUX|ARRAY1|ifpzpa|ATOMIC|BASICS|ALIPAR","",False,src/tlusty/math/io/pzevld.rs,done
|
||||
quartc.f,QUARTC,SUBROUTINE,False,"","","","",True,src/tlusty/math/solvers/quartc.rs,done
|
||||
quasim.f,QUASIM,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|quasun","ALLARD","callarda|MODELQ|callardg|calphatd|quasun|callardb|ATOMIC|callardc|BASICS","ALLARDT|ALLARD",False,src/tlusty/math/opacity/quasim.rs,done
|
||||
quit.f,QUIT,SUBROUTINE,False,"","","","",True,src/tlusty/math/io/quit.rs,done
|
||||
radpre.f,RADPRE,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR","INDEXX|QUIT|RTEFR1|OPACF1","callarda|AUXRTE|RAYSCT|quasun|ipricr|callardb|eospar|EXTINT|auxcbc|ALIPAR|MODELQ|hmolab|callardg|SURFEX|calphatd|ITERAT|ATOMIC|callardc|ODFPAR|BASICS|OPTDPT|intcfg|comgfs","SFFHMI|CIA_H2H|RTEFE2|OPACF1|CROSSD|CROSS|ALLARD|GFREE1|MATINV|DIVSTR|RTEDF2|DOPGAM|OPACT1|GAMI|QUASIM|CIA_H2HE|INDEXX|RTEFR1|STARKA|QUIT|LYMLIN|RTEDF1|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|CIA_H2H2|RTECF0|GHYDOP|OPADD|GAMSP|LOCATE|OPCTAB|PRD|H2MINUS|DWNFR1|ALLARDT|RTESOL|RTECF1",True,src/tlusty/math/radiative/radpre.rs,done
|
||||
radtot.f,RADTOT,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR|ITERAT|OPTDPT|TOTJHK|SURFEX","OPAINI|OPACF1|RTEFR1|TDPINI","callarda|irwint|AUXRTE|RAYSCT|TOTJHK|moldat|quasun|ipricr|callardb|PFSTDS|pfoptb|eospar|EXTINT|auxcbc|ALIPAR|MODELQ|hmolab|callardg|SURFEX|calphatd|ITERAT|ATOMIC|callardc|ODFPAR|BASICS|OPTDPT|intcfg|comgfs","REFLEV|SFFHMI|OPFRAC|PFCNO|CIA_H2H|RTEFE2|PFFE|UBETA|OPACF1|YINT|CROSSD|CROSS|LINPRO|INTHYD|ALLARD|INTXEN|GFREE1|SABOLF|MATINV|WNSTOR|DIVSTR|GFREE0|PFHEAV|DOPGAM|RTEDF2|PFNI|LAGRAN|OPACT1|GAMI|QUASIM|CIA_H2HE|OPAINI|VOIGT|RTEFR1|SGMER0|STARKA|LYMLIN|TDPINI|RAYLEIGH|CIA_HHE|SGMER1|RTEDF1|STARK0|MPARTF|YLINTP|PFSPEC|FFCROS|CIA_H2H2|INTLEM|DWNFR0|GHYDOP|RTECF0|WN|OPADD|GAMSP|PARTF|LOCATE|OPCTAB|PRD|DWNFR1|H2MINUS|PROFSP|LEVGRP|ALLARDT|RTESOL|RTECF1",False,src/tlusty/math/radiative/radtot.rs,done
|
||||
raph.f,RAPH,FUNCTION,True,"","","","",False,src/tlusty/math/solvers/raph.rs,done
|
||||
rates1.f,RATES1,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR|ITERAT","ROSSTD|CROSS|RTEFR1|OPACF1","callarda|AUXRTE|RAYSCT|quasun|ipricr|callardb|eospar|EXTINT|auxcbc|ALIPAR|MODELQ|hmolab|callardg|SURFEX|calphatd|ITERAT|ATOMIC|callardc|ODFPAR|BASICS|OPTDPT|intcfg|comgfs","SFFHMI|CIA_H2H|RTEFE2|OPACF1|CROSSD|CROSS|ALLARD|GFREE1|MATINV|ROSSTD|DIVSTR|RTEDF2|DOPGAM|OPACT1|GAMI|QUASIM|CIA_H2HE|RTEFR1|STARKA|LYMLIN|RTEDF1|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|CIA_H2H2|RTECF0|GHYDOP|OPADD|GAMSP|LOCATE|OPCTAB|PRD|H2MINUS|DWNFR1|ALLARDT|RTESOL|RTECF1",False,src/tlusty/math/rates/rates1.rs,done
|
||||
ratmal.f,RATMAL,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","","ATOMIC|MODELQ|BASICS","",False,src/tlusty/math/rates/ratmal.rs,done
|
||||
ratmat.f,RATMAT,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","REFLEV","ITERAT|ATOMIC|MODELQ|BASICS","REFLEV",False,src/tlusty/math/rates/ratmat.rs,done
|
||||
ratsp1.f,RATSP1,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR|ARRAY1|ITERAT","ROSSTD|CROSS|RTEFR1|OPACF1","callarda|AUXRTE|RAYSCT|ARRAY1|quasun|ipricr|callardb|eospar|EXTINT|auxcbc|ALIPAR|MODELQ|hmolab|callardg|SURFEX|calphatd|ITERAT|ATOMIC|callardc|ODFPAR|BASICS|OPTDPT|intcfg|comgfs","SFFHMI|CIA_H2H|RTEFE2|OPACF1|CROSSD|CROSS|ALLARD|GFREE1|MATINV|ROSSTD|DIVSTR|RTEDF2|DOPGAM|OPACT1|GAMI|QUASIM|CIA_H2HE|RTEFR1|STARKA|LYMLIN|RTEDF1|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|CIA_H2H2|RTECF0|GHYDOP|OPADD|GAMSP|LOCATE|OPCTAB|PRD|H2MINUS|DWNFR1|ALLARDT|RTESOL|RTECF1",True,src/tlusty/math/rates/ratsp1.rs,done
|
||||
rayini.f,RAYINI,SUBROUTINE,False,"BASICS|MODELQ|ATOMIC","RAYLEIGH|RAYSET","MODELQ|RAYSCT|ATOMIC|eospar|BASICS","RAYLEIGH|RAYSET",True,src/tlusty/io/rayini.rs,done
|
||||
rayleigh.f,RAYLEIGH,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|eospar|RAYSCT","","ATOMIC|MODELQ|eospar|RAYSCT|BASICS","",False,src/tlusty/math/opacity/rayleigh.rs,done
|
||||
rayset.f,RAYSET,SUBROUTINE,False,"BASICS|MODELQ","","MODELQ|BASICS","",False,src/tlusty/math/opacity/rayset.rs,done
|
||||
rdata.f,RDATA,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT|ODFPAR|ALIPAR|STRPAR|INUNIT|imodlc","LEMINI|DOPGAM|XENINI|QUIT|RDATAX|LINSET","irwint|MODELQ|imodlc|moldat|quasun|PFSTDS|ITERAT|ATOMIC|pfoptb|BASICS|ODFPAR|STRPAR|INUNIT|ALIPAR","VOIGT|LEMINI|OPFRAC|PFCNO|XENINI|STARKA|QUIT|BKHSGO|PFFE|UBETA|PROFIL|SABOLF|STARK0|DIVSTR|MPARTF|PFSPEC|PFHEAV|DOPGAM|RDATAX|LINSET|GAMSP|PARTF|PFNI|LAGRAN|IJALIS|PROFSP",True,src/tlusty/math/io/rdata.rs,done
|
||||
rdatax.f,RDATAX,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","BKHSGO","ATOMIC|MODELQ|BASICS","BKHSGO",True,src/tlusty/math/io/rdatax.rs,done
|
||||
readbf.f,READBF,SUBROUTINE,False,"BASICS","","BASICS","",True,src/tlusty/math/io/readbf.rs,done
|
||||
rechck.f,RECHCK,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","RTEFR1|OPACF1","callarda|AUXRTE|RAYSCT|quasun|ipricr|callardb|eospar|EXTINT|auxcbc|ALIPAR|MODELQ|hmolab|callardg|SURFEX|calphatd|ITERAT|ATOMIC|callardc|BASICS|ODFPAR|OPTDPT|intcfg|comgfs","SFFHMI|CIA_H2H|RTEFE2|OPACF1|CROSSD|CROSS|ALLARD|GFREE1|MATINV|DIVSTR|RTEDF2|DOPGAM|OPACT1|GAMI|QUASIM|CIA_H2HE|RTEFR1|STARKA|LYMLIN|RTEDF1|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|CIA_H2H2|RTECF0|GHYDOP|OPADD|GAMSP|LOCATE|OPCTAB|PRD|H2MINUS|DWNFR1|ALLARDT|RTESOL|RTECF1",True,src/tlusty/math/io/rechck.rs,done
|
||||
reflev.f,REFLEV,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT","","ITERAT|ATOMIC|MODELQ|BASICS","",False,src/tlusty/math/opacity/reflev.rs,done
|
||||
reiman.f,REIMAN,FUNCTION,True,"","","","",False,src/tlusty/math/opacity/reiman.rs,done
|
||||
resolv.f,RESOLV,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT|ALIPAR|ARRAY1|icnrsp","HESOL6|RAYSET|PRINC|NEWPOP|OPACF1|LUCY|ALISK2|ALIST1|TAUFR1|LINSEL|ROSSTD|CHCKSE|RTECMU|RTECOM|TIMING|PZEVLD|DMEVAL|COOLRT|ALIST2|OUTPRI|OUTPUT|RTEINT|OPAINI|RATSP1|PZERT|RTEFR1|RECHCK|ACCELP|RATES1|PZEVAL|ELCOR|STEQEQ|RYBHEQ|PRD|CONREF|CONOUT|INILAM|RADPRE","DEPTDR|tdedge|POPSTR|pfoptb|tdflag|POPULS|PPAPAR|eletab|rhoder|MODELQ|THERM|calphatd|CC|derdif|rybpgs|BASICS|intcfg|terden|AUXRTE|RAYSCT|quasun|entrop|TABLTD|CONVOUT|grdpra|SURFEX|ADCHAR|ODFPAR|ioniz2|dsctva|CUBCON|callarda|irwint|adiaba|eospar|EXTINT|ALIPAR|hmolab|ITERAT|CTIon|OPTDPT|PRSAUX|COOLCO|COMFH1|imucnn|ARRAY1|moldat|CTRTEMP|ipricr|callardb|PFSTDS|icnrsp|auxcbc|adchar|callardg|ifpzpa|ATOMIC|callardc|comgfs","PRSENT|SFFHMI|ANGSET|CIA_H2H|COLLHE|PFFE|UBETA|EXPINX|EINT|LINPRO|ALISK2|RHOEOS|RTECOM|DOPGAM|ELDENS|RTEDF2|ENTENE|IRC|LINEQS|OPAINI|ODFMER|RTEFR1|SGMER0|GAULEG|TRIDAG|RTEDF1|SGMER1|CEH12|TEMCOR|COLHE|OPADD|WN|OPCTAB|PRD|CION|CONREF|OPACF0|REFLEV|HESOL6|PGSET|RAYSET|PFCNO|OPACFA|YINT|LUCY|ALIST1|TAUFR1|INTXEN|GFREE1|SABOLF|ELDENC|PZEVLD|LAGRAN|DMEVAL|OPACT1|RTEINT|CIA_H2HE|RATSP1|PZERT|LYMLIN|TDPINI|VISINI|TRMDRT|STARK0|STEQEQ|MPARTF|MEANOPT|GHYDOP|DWNFR0|GAMSP|PARTF|OPACFD|H2MINUS|PROFSP|DWNFR1|ALLARDT|CONOUT|MEANOP|RADPRE|CHEAV|RTESOL|PRINC|RTEFE2|OPACF1|ODFHYD|INTHYD|COLH|OPACTD|RUSSEL|ODFHST|MATINV|WNSTOR|GFREE0|ALIFR3|TIMING|PFNI|ALIFR1|GAMI|ALIST2|QUASIM|OUTPUT|INDEXX|EXPO|RECHCK|OSCCOR|ACCELP|QUIT|CIA_HHE|COMSET|YLINTP|CONCOR|PFSPEC|GFREED|LOCATE|RATMAL|LEVGRP|INILAM|LEVSOL|CONVC1|OPFRAC|TRMDER|CONVEC|NEWPOP|ALIFRK|CROSSD|CROSS|ALLARD|RTECMC|LINSEL|ROSSTD|DIVSTR|CHCKSE|RTECMU|PFHEAV|SETTRM|DIELRC|COOLRT|HCTION|OUTPRI|OPACFL|COLIS|CHEAVJ|VOIGT|STARKA|CSPEC|RHONEN|DWNFR|RATES1|MOLEQ|STATE|RAYLEIGH|PZEVAL|ELCOR|RYBHEQ|FFCROS|CIA_H2H2|RTECF0|INTLEM|BUTLER|RATMAT|DIETOT|SZIRC|RTECF1",True,src/tlusty/io/resolv.rs,done
|
||||
rhoeos.f,RHOEOS,FUNCTION,False,"BASICS|MODELQ","PRSENT|SETTRM","MODELQ|tdedge|THERM|TABLTD|BASICS|tdflag","PRSENT|SETTRM",False,src/tlusty/math/eos/rhoeos.rs,done
|
||||
rhonen.f,RHONEN,SUBROUTINE,False,"BASICS|MODELQ","ELDENS","irwint|terden|COMFH1|moldat|PFSTDS|entrop|pfoptb|eospar|adchar|MODELQ|hmolab|ATOMIC|BASICS|ioniz2","MPARTF|OPFRAC|PFSPEC|ELDENS|PFHEAV|PFCNO|PFFE|PARTF|PFNI|MOLEQ|ENTENE|STATE|RUSSEL|LINEQS",False,src/tlusty/math/eos/rhonen.rs,done
|
||||
rhsgen.f,RHSGEN,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ARRAY1|ALIPAR|CUBCON","MATINV|CONVEC|COMPT0|RATMAT|STATE|LEVGRP|SABOLF","irwint|terden|tdedge|COMFH1|adiaba|ARRAY1|moldat|PFSTDS|entrop|pfoptb|eospar|TABLTD|CONVOUT|tdflag|auxcbc|adchar|ALIPAR|MODELQ|hmolab|THERM|CC|derdif|ITERAT|ATOMIC|BASICS|ioniz2|CUBCON","REFLEV|PRSENT|OPFRAC|PFCNO|TRMDER|CONVEC|PFFE|MOLEQ|STATE|RUSSEL|SABOLF|TRMDRT|MATINV|MPARTF|RHOEOS|PFSPEC|SETTRM|ELDENS|PFHEAV|COMPT0|RATMAT|PARTF|PFNI|ENTENE|LEVGRP|LINEQS",False,src/tlusty/math/solvers/rhsgen.rs,done
|
||||
rossop.f,ROSSOP,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ALIPAR","WNSTOR|STEQEQ|MEANOPT|RHOEOS|ELDENS|MEANOP|EXPINT|OPACF0","irwint|terden|tdedge|COMFH1|RAYSCT|moldat|quasun|PFSTDS|POPSTR|entrop|eospar|pfoptb|TABLTD|tdflag|PPAPAR|adchar|ALIPAR|MODELQ|hmolab|THERM|ITERAT|ATOMIC|BASICS|ioniz2|ODFPAR","LEVSOL|REFLEV|PRSENT|SFFHMI|OPFRAC|PFCNO|CIA_H2H|PFFE|UBETA|CROSSD|CROSS|YINT|LINPRO|INTHYD|INTXEN|RUSSEL|EXPINT|SABOLF|WNSTOR|RHOEOS|GFREE0|SETTRM|DIVSTR|ELDENS|PFHEAV|DOPGAM|PFNI|LAGRAN|ENTENE|OPACT1|LINEQS|CIA_H2HE|VOIGT|STARKA|MOLEQ|STATE|RAYLEIGH|CIA_HHE|SGMER1|STARK0|STEQEQ|MEANOPT|MPARTF|PFSPEC|YLINTP|FFCROS|CIA_H2H2|INTLEM|DWNFR0|WN|OPADD|GAMSP|PARTF|RATMAT|OPCTAB|LOCATE|H2MINUS|PROFSP|DWNFR1|MEANOP|OPACF0",False,src/tlusty/math/temperature/rossop.rs,done
|
||||
rosstd.f,ROSSTD,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT|ALIPAR","","ITERAT|ATOMIC|MODELQ|BASICS|ALIPAR","",True,src/tlusty/math/temperature/rosstd.rs,done
|
||||
rte_sc.f,RTE_SC,SUBROUTINE,True,"BASICS","","BASICS","",False,src/tlusty/math/radiative/rte_sc.rs,done
|
||||
rteang.f,RTEANG,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|EXTINT|SURFEX","GAULEG","MODELQ|BASICS|SURFEX|EXTINT|ALIPAR","GAULEG",False,src/tlusty/math/radiative/rteang.rs,done
|
||||
rtecf0.f,RTECF0,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|ITERAT|OPTDPT|auxcbc|AUXRTE","","ITERAT|MODELQ|AUXRTE|BASICS|OPTDPT|auxcbc|ALIPAR","",False,src/tlusty/math/radiative/rtecf0.rs,done
|
||||
rtecf1.f,RTECF1,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|ITERAT|AUXRTE|SURFEX|OPTDPT|EXTINT|comgfs","RTEFE2|RTESOL|RTECF0","MODELQ|AUXRTE|SURFEX|ITERAT|BASICS|OPTDPT|EXTINT|auxcbc|comgfs|ALIPAR","RTEFE2|RTESOL|RTECF0",True,src/tlusty/math/radiative/rtecf1.rs,done
|
||||
rtecmc.f,RTECMC,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|ITERAT|AUXRTE|comgfs","MATINV|RTECF0|OPACF1","callarda|AUXRTE|RAYSCT|quasun|ipricr|callardb|eospar|auxcbc|ALIPAR|MODELQ|hmolab|callardg|calphatd|ITERAT|ATOMIC|callardc|BASICS|ODFPAR|OPTDPT|intcfg|comgfs","SFFHMI|CIA_H2H|OPACF1|CROSSD|CROSS|ALLARD|GFREE1|MATINV|DIVSTR|DOPGAM|OPACT1|GAMI|QUASIM|CIA_H2HE|STARKA|LYMLIN|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|CIA_H2H2|RTECF0|GHYDOP|OPADD|GAMSP|LOCATE|OPCTAB|PRD|H2MINUS|DWNFR1|ALLARDT",False,src/tlusty/math/radiative/rtecmc.rs,done
|
||||
rtecmu.f,RTECMU,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|ITERAT|OPTDPT|AUXRTE","RTECF0|GAULEG|RTESOL|OPACF1","callarda|AUXRTE|RAYSCT|quasun|ipricr|callardb|eospar|auxcbc|ALIPAR|MODELQ|hmolab|callardg|calphatd|ITERAT|ATOMIC|callardc|BASICS|ODFPAR|OPTDPT|intcfg","SFFHMI|CIA_H2H|OPACF1|CROSSD|CROSS|ALLARD|GFREE1|DIVSTR|DOPGAM|OPACT1|GAMI|QUASIM|CIA_H2HE|STARKA|LYMLIN|GAULEG|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|CIA_H2H2|RTECF0|GHYDOP|OPADD|GAMSP|LOCATE|OPCTAB|PRD|H2MINUS|DWNFR1|ALLARDT|RTESOL",True,src/tlusty/math/radiative/rtecmu.rs,done
|
||||
rtecom.f,RTECOM,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|ITERAT|OPTDPT|AUXRTE|comgfs","RTECF0|RTECF1|RTECMC|OPACF1","callarda|eospar|EXTINT|ALIPAR|MODELQ|hmolab|calphatd|ITERAT|BASICS|OPTDPT|intcfg|AUXRTE|RAYSCT|quasun|ipricr|callardb|auxcbc|callardg|SURFEX|ATOMIC|callardc|ODFPAR|comgfs","SFFHMI|CIA_H2H|RTEFE2|OPACF1|CROSSD|CROSS|ALLARD|RTECMC|GFREE1|MATINV|DIVSTR|DOPGAM|OPACT1|GAMI|QUASIM|CIA_H2HE|STARKA|LYMLIN|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|CIA_H2H2|RTECF0|GHYDOP|OPADD|GAMSP|LOCATE|OPCTAB|PRD|H2MINUS|DWNFR1|ALLARDT|RTESOL|RTECF1",False,src/tlusty/math/radiative/rtecom.rs,done
|
||||
rtedf1.f,RTEDF1,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|OPTDPT","","OPTDPT|MODELQ|BASICS|ALIPAR","",False,src/tlusty/math/radiative/rtedf1.rs,done
|
||||
rtedf2.f,RTEDF2,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR","","MODELQ|BASICS|ALIPAR","",False,src/tlusty/math/radiative/rtedf2.rs,done
|
||||
rtefe2.f,RTEFE2,SUBROUTINE,True,"BASICS","","BASICS","",False,src/tlusty/math/radiative/rtefe2.rs,done
|
||||
rtefr1.f,RTEFR1,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|ITERAT|OPTDPT","MATINV|RTEDF2|RTEDF1|RTESOL|RTECF1","MODELQ|AUXRTE|SURFEX|ITERAT|BASICS|OPTDPT|EXTINT|auxcbc|comgfs|ALIPAR","MATINV|RTECF0|RTEDF2|RTEFE2|RTEDF1|RTESOL|RTECF1",True,src/tlusty/math/radiative/rtefr1.rs,done
|
||||
rteint.f,RTEINT,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|ITERAT|OPTDPT","MATINV|OPACF1","callarda|RAYSCT|quasun|ipricr|callardb|eospar|ALIPAR|MODELQ|hmolab|callardg|calphatd|ITERAT|ATOMIC|callardc|BASICS|ODFPAR|OPTDPT|intcfg","SFFHMI|CIA_H2H|OPACF1|CROSSD|CROSS|ALLARD|GFREE1|MATINV|DIVSTR|DOPGAM|OPACT1|GAMI|QUASIM|CIA_H2HE|STARKA|LYMLIN|RAYLEIGH|CIA_HHE|SGMER1|STARK0|YLINTP|FFCROS|CIA_H2H2|GHYDOP|OPADD|GAMSP|LOCATE|OPCTAB|PRD|H2MINUS|DWNFR1|ALLARDT",True,src/tlusty/math/radiative/rteint.rs,done
|
||||
rtesol.f,RTESOL,SUBROUTINE,True,"BASICS","","BASICS","",False,src/tlusty/math/radiative/rtesol.rs,done
|
||||
russel.f,RUSSEL,SUBROUTINE,False,"BASICS|MODELQ|COMFH1","MPARTF","moldat|MODELQ|COMFH1|BASICS","MPARTF",True,src/tlusty/math/eos/russel.rs,done
|
||||
rybchn.f,RYBCHN,SUBROUTINE,False,"BASICS|ITERAT|MODELQ|ALIPAR|ARRAY1|grdpra|rybpgs","PGSET|ELDENS","irwint|terden|COMFH1|ARRAY1|moldat|PFSTDS|entrop|pfoptb|eospar|adchar|ALIPAR|grdpra|MODELQ|hmolab|rybpgs|ITERAT|ATOMIC|BASICS|ioniz2","PGSET|OPFRAC|PFCNO|PFFE|MOLEQ|STATE|TRIDAG|RUSSEL|MPARTF|PFSPEC|ELDENS|PFHEAV|PARTF|PFNI|ENTENE|LINEQS",True,src/tlusty/math/solvers/rybchn.rs,done
|
||||
rybene.f,RYBENE,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|ARRAY1|RYBMTX|deridt|CUBCON","CONVEC","deridt|irwint|terden|tdedge|COMFH1|adiaba|ARRAY1|moldat|PFSTDS|RYBMTX|entrop|pfoptb|eospar|TABLTD|CONVOUT|tdflag|adchar|ALIPAR|MODELQ|hmolab|THERM|CC|derdif|ATOMIC|BASICS|ioniz2|CUBCON","PRSENT|OPFRAC|PFCNO|TRMDER|CONVEC|PFFE|MOLEQ|STATE|RUSSEL|TRMDRT|MPARTF|RHOEOS|PFSPEC|SETTRM|ELDENS|PFHEAV|PARTF|PFNI|ENTENE|LINEQS",False,src/tlusty/math/solvers/rybene.rs,done
|
||||
rybheq.f,RYBHEQ,SUBROUTINE,False,"BASICS|MODELQ|grdpra|rybpgs","OPAINI|WNSTOR|STEQEQ|PGSET|RTEFR1|ELDENS|OPACF1","callarda|irwint|POPSTR|pfoptb|eospar|EXTINT|PPAPAR|ALIPAR|MODELQ|hmolab|calphatd|rybpgs|ITERAT|BASICS|OPTDPT|intcfg|terden|AUXRTE|COMFH1|RAYSCT|moldat|quasun|ipricr|callardb|PFSTDS|entrop|auxcbc|adchar|grdpra|callardg|SURFEX|ATOMIC|callardc|ODFPAR|ioniz2|comgfs","SFFHMI|CIA_H2H|RTEFE2|PFFE|UBETA|OPACF1|LINPRO|INTHYD|RUSSEL|MATINV|WNSTOR|ELDENS|DOPGAM|RTEDF2|PFNI|ENTENE|GAMI|QUASIM|LINEQS|OPAINI|RTEFR1|SGMER0|TRIDAG|RTEDF1|CIA_HHE|SGMER1|YLINTP|PFSPEC|WN|OPADD|LOCATE|OPCTAB|PRD|LEVGRP|REFLEV|LEVSOL|PGSET|OPFRAC|PFCNO|YINT|CROSSD|CROSS|ALLARD|INTXEN|GFREE1|SABOLF|DIVSTR|PFHEAV|LAGRAN|OPACT1|CIA_H2HE|VOIGT|STARKA|LYMLIN|MOLEQ|STATE|RAYLEIGH|STARK0|STEQEQ|MPARTF|FFCROS|CIA_H2H2|RTECF0|INTLEM|DWNFR0|GHYDOP|GAMSP|PARTF|RATMAT|H2MINUS|PROFSP|DWNFR1|ALLARDT|RTESOL|RTECF1",True,src/tlusty/math/solvers/rybheq.rs,done
|
||||
rybmat.f,RYBMAT,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|ARRAY1|RYBMTX|dsctva","","RYBMTX|MODELQ|dsctva|BASICS|ARRAY1|ALIPAR","",False,src/tlusty/math/solvers/rybmat.rs,done
|
||||
rybsol.f,RYBSOL,SUBROUTINE,False,"BASICS|MODELQ|ATOMIC|ALIPAR|ARRAY1|ITERAT|RYBMTX|imodlc","SETDRT|STEQEQ|ROSSTD|RTEFR1|OPACTR|RYBCHN|RYBENE|ALIFR1|RYBMAT|LEVSET|TRIDAG|LINEQS","callarda|irwint|deridt|tdedge|imodlc|adiaba|RYBMTX|POPSTR|pfoptb|eospar|tdflag|EXTINT|PPAPAR|ALIPAR|MODELQ|hmolab|THERM|calphatd|CC|CUBCON|derdif|rybpgs|ITERAT|BASICS|OPTDPT|intcfg|terden|AUXRTE|COMFH1|RAYSCT|ARRAY1|moldat|quasun|ipricr|callardb|PFSTDS|entrop|TABLTD|CONVOUT|auxcbc|adchar|grdpra|RHODER|callardg|SURFEX|ATOMIC|callardc|dsctva|ODFPAR|ioniz2|comgfs","PRSENT|SFFHMI|CIA_H2H|RTEFE2|PFFE|UBETA|OPACF1|LINPRO|INTHYD|RUSSEL|MATINV|WNSTOR|RHOEOS|GFREE0|ELDENS|RTEDF2|DOPGAM|ALIFR3|PFNI|ALIFR1|ENTENE|GAMI|QUASIM|LINEQS|OPAINI|RTEFR1|SGMER0|QUIT|RYBMAT|TRIDAG|RTEDF1|CIA_HHE|SGMER1|YLINTP|PFSPEC|RYBENE|WN|OPADD|LOCATE|OPCTAB|PRD|RATMAL|LEVGRP|LEVSET|LEVSOL|REFLEV|OPFRAC|PGSET|OPACTR|PFCNO|TRMDER|CONVEC|YINT|CROSSD|CROSS|ALLARD|INTXEN|GFREE1|SABOLF|ROSSTD|DIVSTR|SETTRM|PFHEAV|LAGRAN|OPACT1|CIA_H2HE|SETDRT|VOIGT|RYBCHN|STARKA|LYMLIN|TDPINI|MOLEQ|STATE|RAYLEIGH|TRMDRT|STARK0|STEQEQ|MPARTF|FFCROS|CIA_H2H2|RTECF0|GHYDOP|INTLEM|DWNFR0|GAMSP|PARTF|RATMAT|H2MINUS|PROFSP|DWNFR1|ALLARDT|RTESOL|RTECF1",True,src/tlusty/math/solvers/rybsol.rs,done
|
||||
sabolf.f,SABOLF,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","PARTF","irwint|MODELQ|moldat|PFSTDS|ATOMIC|pfoptb|BASICS","MPARTF|PFSPEC|OPFRAC|PFHEAV|PFCNO|PFFE|PARTF|PFNI",False,src/tlusty/math/utils/sabolf.rs,done
|
||||
sbfch.f,SBFCH,FUNCTION,True,"","","","",False,src/tlusty/math/hydrogen/sbfch.rs,done
|
||||
sbfhe1.f,SBFHE1,FUNCTION,False,"BASICS|ATOMIC","CKOEST|QUIT|HEPHOT","ATOMIC|BASICS","CKOEST|QUIT|HEPHOT",True,src/tlusty/math/hydrogen/sbfhe1.rs,done
|
||||
sbfhmi.f,SBFHMI,FUNCTION,True,"","YLINTP","","YLINTP",False,src/tlusty/math/hydrogen/sbfhmi.rs,done
|
||||
sbfhmi_old.f,SBFHMI_OLD,FUNCTION,True,"","","","",False,src/tlusty/math/hydrogen/sbfhmi_old.rs,done
|
||||
sbfoh.f,SBFOH,FUNCTION,True,"","","","",False,src/tlusty/math/hydrogen/sbfoh.rs,done
|
||||
setdrt.f,SETDRT,SUBROUTINE,False,"BASICS|MODELQ|RHODER","RHOEOS","RHODER|MODELQ|tdedge|THERM|TABLTD|BASICS|tdflag","PRSENT|RHOEOS|SETTRM",False,src/tlusty/math/utils/setdrt.rs,done
|
||||
settrm.f,SETTRM,SUBROUTINE,False,"TABLTD|tdedge|THERM|tdflag","PRSENT","tdedge|THERM|TABLTD|tdflag","PRSENT",True,src/tlusty/io/settrm.rs,done
|
||||
sffhmi.f,SFFHMI,FUNCTION,True,"","YLINTP","","YLINTP",False,src/tlusty/math/hydrogen/sffhmi.rs,done
|
||||
sffhmi_add.f,SFFHMI_ADD,FUNCTION,True,"","YLINTP","","YLINTP",False,src/tlusty/math/hydrogen/sffhmi_add.rs,done
|
||||
sghe12.f,SGHE12,FUNCTION,True,"","","","",False,src/tlusty/math/partition/sghe12.rs,done
|
||||
sgmer0.f,SGMER0,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","","ATOMIC|MODELQ|BASICS","",False,src/tlusty/math/hydrogen/sgmer.rs,done
|
||||
sgmer1.f,SGMER1,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","","ATOMIC|MODELQ|BASICS","",False,src/tlusty/math/hydrogen/sgmer1.rs,done
|
||||
sgmerd.f,SGMERD,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","","ATOMIC|MODELQ|BASICS","",False,src/tlusty/math/hydrogen/sgmer.rs,done
|
||||
sigave.f,SIGAVE,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR","QUIT","ODFPAR|ATOMIC|MODELQ|BASICS","QUIT",True,src/tlusty/math/hydrogen/sigave.rs,done
|
||||
sigk.f,SIGK,FUNCTION,False,"BASICS|ATOMIC","TOPBAS|SPSIGK|YLINTP|SBFHE1|SBFHMI|GAUNT|VERNER","ATOMIC|BASICS|TOPB","SBFHE1|SPSIGK|REIMAN|HIDALG|QUIT|VERN18|SGHE12|CKOEST|HEPHOT|VERN16|VERN20|VERNER|TOPBAS|YLINTP|SBFHMI|VERN26|CARBON|GAUNT|OPDATA",False,src/tlusty/math/hydrogen/sigk.rs,done
|
||||
sigmar.f,SIGMAR,FUNCTION,False,"BASICS","LAGUER","BASICS","LAGUER",True,src/tlusty/math/hydrogen/sigmar.rs,done
|
||||
solve.f,SOLVE,SUBROUTINE,False,"BASICS|ITERAT|MODELQ|ARRAY1|ALIPAR|CMATZD","MATINV|WNSTOR|RHSGEN|PRCHAN|MATGEN|IROSET","irwint|tdedge|adiaba|pfoptb|eospar|tdflag|ALIPAR|MODELQ|hmolab|THERM|CC|derdif|ITERAT|BASICS|CTIon|CMATZD|terden|COMFH1|ARRAY1|moldat|CTRTEMP|PFSTDS|entrop|TABLTD|CONVOUT|auxcbc|adchar|SURFEX|COLKUR|LINED|ATOMIC|ADCHAR|ODFPAR|ioniz2|CUBCON","PRSENT|COLLHE|PFFE|BRTEZ|EXPINX|EINT|COLH|IJALI2|RUSSEL|MATINV|WNSTOR|RHOEOS|ELDENS|PFNI|ENTENE|IRC|LINEQS|PRCHAN|RHSGEN|EXPO|INDEXX|QUIT|BPOPE|VOIGTE|BPOPC|BPOPT|SGMER1|INKUL|CEH12|YLINTP|PFSPEC|MATGEN|BHEZ|BHE|COLHE|WN|COMPT0|CION|BRTE|LEVGRP|REFLEV|LEVSOL|BHED|OPFRAC|PFCNO|TRMDER|CONVEC|EMAT|CROSS|BREZ|SABOLF|SETTRM|PFHEAV|HCTION|IROSET|CHEAVJ|MATCON|COLIS|CSPEC|BPOPF|LEVCD|MOLEQ|STATE|TRMDRT|MPARTF|BUTLER|PARTF|RATMAT|BPOP|DWNFR1|BRE|SZIRC|CHEAV",True,src/tlusty/math/solvers/solve.rs,done
|
||||
solves.f,SOLVES,SUBROUTINE,False,"BASICS|ITERAT|MODELQ|ARRAY1|ALIPAR|CMATZD|STOMAT","MATINV|WNSTOR|RHSGEN|PRCHAN|MATGEN|IROSET","irwint|tdedge|adiaba|pfoptb|eospar|tdflag|ALIPAR|MODELQ|hmolab|THERM|CC|derdif|ITERAT|BASICS|CTIon|CMATZD|terden|COMFH1|ARRAY1|moldat|CTRTEMP|PFSTDS|entrop|TABLTD|CONVOUT|auxcbc|adchar|SURFEX|STOMAT|COLKUR|LINED|ATOMIC|ADCHAR|ODFPAR|ioniz2|CUBCON","PRSENT|COLLHE|PFFE|BRTEZ|EXPINX|EINT|COLH|IJALI2|RUSSEL|MATINV|WNSTOR|RHOEOS|ELDENS|PFNI|ENTENE|IRC|LINEQS|PRCHAN|RHSGEN|EXPO|INDEXX|QUIT|BPOPE|VOIGTE|BPOPC|BPOPT|SGMER1|INKUL|CEH12|YLINTP|PFSPEC|MATGEN|BHEZ|BHE|COLHE|WN|COMPT0|CION|BRTE|LEVGRP|REFLEV|LEVSOL|BHED|OPFRAC|PFCNO|TRMDER|CONVEC|EMAT|CROSS|BREZ|SABOLF|SETTRM|PFHEAV|HCTION|IROSET|CHEAVJ|MATCON|COLIS|CSPEC|BPOPF|LEVCD|MOLEQ|STATE|TRMDRT|MPARTF|BUTLER|PARTF|RATMAT|BPOP|DWNFR1|BRE|SZIRC|CHEAV",True,src/tlusty/math/solvers/solves.rs,done
|
||||
spsigk.f,SPSIGK,SUBROUTINE,True,"","HIDALG|SGHE12|REIMAN|CARBON","","HIDALG|CARBON|REIMAN|SGHE12",False,src/tlusty/math/hydrogen/spsigk.rs,done
|
||||
srtfrq.f,SRTFRQ,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","INDEXX|QUIT","ATOMIC|MODELQ|BASICS","INDEXX|QUIT",True,src/tlusty/io/srtfrq.rs,done
|
||||
stark0.f,STARK0,SUBROUTINE,True,"","","","",False,src/tlusty/math/opacity/stark0.rs,done
|
||||
starka.f,STARKA,FUNCTION,False,"BASICS|MODELQ","","MODELQ|BASICS","",False,src/tlusty/math/opacity/starka.rs,done
|
||||
start.f,START,SUBROUTINE,False,"BASICS|hediff","PRDINI|COMSET|INITIA|HEDIF","DEPTDR|tdedge|POPSTR|pfoptb|tdflag|PPAPAR|eletab|STFCR|MODELQ|THERM|CC|calphatd|derdif|BASICS|intcfg|FLXAUX|terden|AUXRTE|RAYSCT|quasun|entrop|TABLTD|CONVOUT|SURFEX|COLKUR|LINED|ichndm|ODFPAR|ioniz2|CUBCON|callarda|irwint|temlim|deridt|imodlc|adiaba|TOTJHK|intcff|eospar|FACTRS|EXTINT|INUNIT|ijflar|ALIPAR|hmolab|freqcl|TOPB|ITERAT|OPTDPT|PRSAUX|imucnn|COMFH1|relcor|moldat|ipricr|callardb|PFSTDS|icnrsp|auxcbc|adchar|abntab|callardg|ifpzpa|hediff|ATOMIC|callardc|STRPAR|comgfs","INPMOD|PRSENT|SBFHE1|ANGSET|SFFHMI|TEMPER|CIA_H2H|PFFE|UBETA|HEDIF|HEPHOT|LINPRO|VERN16|LINSPL|VERN20|IJALI2|EXPINT|ERFCIN|RADTOT|INIFRT|CORRWM|RHOEOS|ELDENS|DOPGAM|RTEDF2|GRCOR|RAYINI|ODFHYS|ENTENE|CHCTAB|OPDATA|LINEQS|GOMINI|OPAINI|REIMAN|RTEFR1|SGMER0|GAULEG|VOIGTE|CKOEST|LTEGR|ODFFR|RTEDF1|SGMER1|WN|OPADD|INPDIS|OPCTAB|NEWDM|PRD|GAUNT|OPACF0|REFLEV|SPSIGK|HESOL6|RAYSET|PFCNO|HIDALG|XENINI|YINT|INTXEN|GFREE1|SABOLF|BETAH|RTEANG|INCLDY|VERN26|PSOLVE|ODFSET|TLOCAL|LAGRAN|GREYD|OPACT1|SRTFRQ|CONTMP|CIA_H2HE|IROSET|NSTOUT|LEMINI|INIFRC|LYMLIN|TDPINI|LEVCD|VERNER|TRMDRT|DMDER|STARK0|STEQEQ|MPARTF|MEANOPT|GHYDOP|SBFHMI|DWNFR0|GAMSP|PARTF|HESOLV|H2MINUS|PROFSP|DWNFR1|CONOUT|ALLARDT|MEANOP|RTESOL|READBF|COLUMN|VERN18|RTEFE2|OPACF1|PROFIL|INTHYD|RUSSEL|MATINV|WNSTOR|GFREE0|ERFCX|TABINI|PFNI|IJALIS|GAMI|QUASIM|SIGK|RDATA|INDEXX|QUIT|BKHSGO|TABINT|QUARTC|ROSSOP|NEWDMT|COMSET|CIA_HHE|INKUL|GETLAL|YLINTP|LTEGRD|INITIA|PFSPEC|LINSET|LOCATE|CUBIC|NSTPAR|LEVGRP|LEVSET|PRDINI|OPAHST|LEVSOL|OPFRAC|TRMDER|CONVEC|CROSSD|CROSS|ALLARD|DIVSTR|SETTRM|PFHEAV|KURUCZ|RDATAX|VOIGT|TRAINI|STARKA|CONTMD|INTERP|RHONEN|GETWRD|SGHE12|MOLEQ|STATE|RAYLEIGH|ZMRHO|SIGAVE|TOPBAS|FFCROS|OPADD0|CIA_H2H2|RTECF0|INTLEM|CHANGE|CARBON|RATMAT|GRIDP|INIFRS|RTECF1",True,src/tlusty/io/start.rs,done
|
||||
state.f,STATE,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|terden|PFSTDS","OPFRAC|PARTF","irwint|terden|MODELQ|moldat|PFSTDS|ATOMIC|pfoptb|BASICS","MPARTF|OPFRAC|PFSPEC|PFHEAV|PFCNO|PFFE|PARTF|PFNI",True,src/tlusty/math/utils/state.rs,done
|
||||
steqeq.f,STEQEQ,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT|POPSTR|PPAPAR","RATMAT|MOLEQ|LEVSOL|SABOLF","irwint|terden|COMFH1|moldat|PFSTDS|POPSTR|entrop|eospar|pfoptb|PPAPAR|adchar|MODELQ|hmolab|ITERAT|ATOMIC|BASICS|ioniz2","LEVSOL|REFLEV|OPFRAC|PFCNO|PFFE|MOLEQ|RUSSEL|SABOLF|MPARTF|PFSPEC|PFHEAV|PARTF|RATMAT|PFNI|LINEQS",False,src/tlusty/math/eos/steqeq.rs,done
|
||||
switch.f,SWITCH,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","","ATOMIC|MODELQ|BASICS","",True,src/tlusty/math/utils/switch.rs,done
|
||||
szirc.f,SZIRC,SUBROUTINE,True,"","EINT","","EXPINX|EXPO|EINT",False,src/tlusty/math/hydrogen/szirc.rs,done
|
||||
tabini.f,TABINI,SUBROUTINE,False,"BASICS|MODELQ|ATOMIC|abntab|intcff|eletab","","abntab|ATOMIC|MODELQ|BASICS|intcff|eletab","",True,src/tlusty/io/tabini.rs,done
|
||||
tabint.f,TABINT,SUBROUTINE,False,"BASICS|MODELQ|ATOMIC|intcff","","ATOMIC|MODELQ|intcff|BASICS","",False,src/tlusty/math/interpolation/tabint.rs,done
|
||||
taufr1.f,TAUFR1,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|ITERAT|OPTDPT","","ITERAT|MODELQ|BASICS|OPTDPT|ALIPAR","",False,src/tlusty/math/ali/taufr1.rs,done
|
||||
tdpini.f,TDPINI,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR|ALIPAR","GFREE0","ATOMIC|MODELQ|ODFPAR|BASICS|ALIPAR","GFREE0",False,src/tlusty/math/temperature/tdpini.rs,done
|
||||
temcor.f,TEMCOR,SUBROUTINE,False,"BASICS|MODELQ|ARRAY1|ALIPAR|CUBCON","WNSTOR|STEQEQ|ELDENS|CONVEC|MEANOP|OPACF0","irwint|tdedge|adiaba|POPSTR|pfoptb|eospar|tdflag|PPAPAR|ALIPAR|MODELQ|hmolab|THERM|CC|derdif|ITERAT|BASICS|terden|COMFH1|RAYSCT|ARRAY1|moldat|quasun|PFSTDS|entrop|TABLTD|CONVOUT|adchar|ATOMIC|ODFPAR|ioniz2|CUBCON","LEVSOL|REFLEV|PRSENT|SFFHMI|OPFRAC|PFCNO|CIA_H2H|TRMDER|CONVEC|PFFE|UBETA|CROSSD|CROSS|YINT|LINPRO|INTHYD|INTXEN|RUSSEL|SABOLF|WNSTOR|RHOEOS|GFREE0|SETTRM|DIVSTR|ELDENS|PFHEAV|DOPGAM|PFNI|LAGRAN|ENTENE|OPACT1|LINEQS|CIA_H2HE|VOIGT|STARKA|MOLEQ|STATE|RAYLEIGH|CIA_HHE|SGMER1|TRMDRT|STARK0|STEQEQ|MPARTF|YLINTP|PFSPEC|FFCROS|CIA_H2H2|INTLEM|DWNFR0|WN|OPADD|GAMSP|PARTF|RATMAT|LOCATE|OPCTAB|H2MINUS|PROFSP|DWNFR1|MEANOP|OPACF0",True,src/tlusty/math/temperature/temcor.rs,done
|
||||
temper.f,TEMPER,SUBROUTINE,False,"BASICS|MODELQ|ALIPAR|FACTRS|PRSAUX|FLXAUX","WNSTOR|STEQEQ|MEANOPT|RHOEOS|ELDENS|TLOCAL|MEANOP|OPACF0","irwint|tdedge|POPSTR|pfoptb|eospar|tdflag|FACTRS|PPAPAR|ALIPAR|MODELQ|hmolab|THERM|ITERAT|BASICS|FLXAUX|PRSAUX|terden|COMFH1|RAYSCT|moldat|quasun|PFSTDS|entrop|TABLTD|adchar|ATOMIC|ODFPAR|ioniz2","LEVSOL|REFLEV|PRSENT|SFFHMI|OPFRAC|PFCNO|CIA_H2H|PFFE|UBETA|CROSSD|CROSS|YINT|LINPRO|INTHYD|INTXEN|RUSSEL|SABOLF|WNSTOR|RHOEOS|GFREE0|SETTRM|DIVSTR|ELDENS|PFHEAV|DOPGAM|TLOCAL|PFNI|LAGRAN|ENTENE|OPACT1|LINEQS|CIA_H2HE|VOIGT|STARKA|MOLEQ|QUARTC|STATE|RAYLEIGH|CIA_HHE|SGMER1|STARK0|STEQEQ|MEANOPT|MPARTF|PFSPEC|YLINTP|FFCROS|CIA_H2H2|INTLEM|DWNFR0|WN|OPADD|GAMSP|PARTF|RATMAT|OPCTAB|LOCATE|H2MINUS|PROFSP|DWNFR1|MEANOP|OPACF0",True,src/tlusty/math/temperature/temper.rs,done
|
||||
timing.f,TIMING,SUBROUTINE,False,"","","","",True,src/tlusty/math/io/timing.rs,done
|
||||
tiopf.f,TIOPF,SUBROUTINE,True,"","","","",False,src/tlusty/math/partition/tiopf.rs,done
|
||||
tlocal.f,TLOCAL,SUBROUTINE,False,"BASICS|MODELQ|FACTRS|FLXAUX","QUARTC","FLXAUX|FACTRS|MODELQ|BASICS","QUARTC",False,src/tlusty/math/temperature/tlocal.rs,done
|
||||
tlusty.f,TLUSTY,UNKNOWN,False,"BASICS|ITERAT|ALIPAR","RESOLV|SOLVE|SOLVES|RYBSOL|TIMING|ACCEL2|START","DEPTDR|tdedge|POPSTR|pfoptb|tdflag|POPULS|PPAPAR|eletab|STFCR|rhoder|MODELQ|THERM|calphatd|CC|derdif|rybpgs|BASICS|intcfg|FLXAUX|terden|AUXRTE|RAYSCT|quasun|entrop|TABLTD|CONVOUT|grdpra|RHODER|SURFEX|COLKUR|LINED|STOMAT|ichndm|ADCHAR|ODFPAR|ioniz2|dsctva|CUBCON|callarda|irwint|deridt|temlim|imodlc|adiaba|TOTJHK|intcff|RYBMTX|eospar|FACTRS|EXTINT|INUNIT|ijflar|ALIPAR|hmolab|freqcl|TOPB|ITERAT|CTIon|OPTDPT|CMATZD|PRSAUX|COOLCO|COMFH1|imucnn|ARRAY1|relcor|moldat|CTRTEMP|ipricr|callardb|PFSTDS|icnrsp|auxcbc|adchar|abntab|callardg|ifpzpa|ATOMIC|callardc|hediff|STRPAR|comgfs","PRSENT|INPMOD|SBFHE1|ANGSET|TEMPER|COLLHE|EXPINX|ACCEL2|VERN16|VERN20|ERFCIN|RHOEOS|GRCOR|ODFHYS|CHCTAB|OPDATA|LINEQS|GOMINI|OPAINI|PRCHAN|RHSGEN|ODFMER|RTEFR1|VOIGTE|BPOPC|CKOEST|BPOPT|LTEGR|TRIDAG|CEH12|WN|NEWDM|INPDIS|OPCTAB|PRD|CION|GAUNT|BHED|PGSET|OPACFA|PFCNO|OPACTR|XENINI|ALIST1|INTXEN|GFREE1|SABOLF|RTEANG|INCLDY|VERN26|ODFSET|PZEVLD|TLOCAL|GREYD|DMEVAL|RTEINT|CIA_H2HE|PZERT|LEMINI|BPOPF|TDPINI|VISINI|VERNER|TRMDRT|DMDER|MEANOPT|DWNFR0|PARTF|HESOLV|H2MINUS|CONOUT|READBF|COLUMN|PRINC|RTEFE2|PROFIL|INTHYD|OPACTD|WNSTOR|ERFCX|ALIFR3|TIMING|PFNI|GAMI|ALIST2|QUASIM|OUTPUT|INDEXX|ACCELP|RYBSOL|QUIT|BPOPE|BKHSGO|TABINT|RYBMAT|RESOLV|GETLAL|YLINTP|CONCOR|LTEGRD|INITIA|LINSET|RATMAL|CUBIC|LEVGRP|LEVSET|CONVC1|OPFRAC|ALIFRK|NEWPOP|CROSS|RTECMC|DIVSTR|SETTRM|RTECMU|PFHEAV|OPACFL|COLIS|VOIGT|TRAINI|STARKA|CSPEC|CONTMD|INTERP|GETWRD|DWNFR|RATES1|PZEVAL|TOPBAS|FFCROS|BUTLER|CHANGE|CARBON|BPOP|GRIDP|BRE|INIFRS|SFFHMI|CIA_H2H|PFFE|UBETA|BRTEZ|HEDIF|EINT|LINPRO|HEPHOT|ALISK2|IJALI2|LINSPL|EXPINT|RADTOT|INIFRT|CORRWM|RTECOM|DOPGAM|ELDENS|RTEDF2|RAYINI|ENTENE|IRC|REIMAN|SGMER0|GAULEG|ODFFR|RTEDF1|SGMER1|TEMCOR|COLHE|MATGEN|BHEZ|OPADD|BHE|CONREF|OPACF0|SOLVE|REFLEV|HESOL6|SPSIGK|RAYSET|HIDALG|EMAT|YINT|LUCY|TAUFR1|BREZ|ELDENC|BETAH|PSOLVE|LAGRAN|OPACT1|START|SRTFRQ|CONTMP|IROSET|RATSP1|NSTOUT|RYBCHN|INIFRC|LYMLIN|LEVCD|STARK0|STEQEQ|MPARTF|GHYDOP|SBFHMI|GAMSP|OPACFD|PROFSP|DWNFR1|ALLARDT|MEANOP|RADPRE|CHEAV|RTESOL|VERN18|OPACF1|ODFHYD|COLH|RUSSEL|ODFHST|MATINV|GFREE0|TABINI|ALIFR1|IJALIS|SIGK|RDATA|EXPO|RECHCK|OSCCOR|QUARTC|ROSSOP|NEWDMT|CIA_HHE|COMSET|INKUL|SOLVES|PFSPEC|GFREED|RYBENE|LOCATE|COMPT0|BRTE|NSTPAR|INILAM|PRDINI|LEVSOL|OPAHST|TRMDER|CONVEC|CROSSD|ALLARD|LINSEL|ROSSTD|CHCKSE|KURUCZ|RDATAX|DIELRC|COOLRT|HCTION|OUTPRI|CHEAVJ|MATCON|SETDRT|RHONEN|SGHE12|MOLEQ|STATE|RAYLEIGH|ELCOR|ZMRHO|SIGAVE|RYBHEQ|CIA_H2H2|OPADD0|RTECF0|INTLEM|RATMAT|DIETOT|SZIRC|RTECF1",True,src/bin/tlusty.rs,done
|
||||
topbas.f,TOPBAS,FUNCTION,False,"TOPB","OPDATA|YLINTP","TOPB","OPDATA|YLINTP",True,src/tlusty/math/utils/topbas.rs,done
|
||||
traini.f,TRAINI,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ODFPAR","","ODFPAR|ATOMIC|MODELQ|BASICS","",False,src/tlusty/math/utils/traini.rs,done
|
||||
tridag.f,TRIDAG,SUBROUTINE,True,"","","","",False,src/tlusty/math/solvers/tridag.rs,done
|
||||
trmder.f,TRMDER,SUBROUTINE,False,"BASICS|terden|derdif|adiaba","ELDENS","irwint|terden|COMFH1|adiaba|moldat|PFSTDS|entrop|pfoptb|eospar|adchar|MODELQ|hmolab|derdif|ATOMIC|BASICS|ioniz2","MPARTF|OPFRAC|PFSPEC|ELDENS|PFHEAV|PFCNO|PFFE|PARTF|PFNI|MOLEQ|ENTENE|STATE|RUSSEL|LINEQS",False,src/tlusty/math/radiative/trmder.rs,done
|
||||
trmdrt.f,TRMDRT,SUBROUTINE,False,"BASICS|tdedge|tdflag|CONVOUT|CC","PRSENT|RHOEOS","MODELQ|tdedge|THERM|CC|TABLTD|CONVOUT|BASICS|tdflag","PRSENT|RHOEOS|SETTRM",False,src/tlusty/math/radiative/trmdrt.rs,done
|
||||
ubeta.f,UBETA,FUNCTION,True,"","LAGRAN","","LAGRAN",False,src/tlusty/math/solvers/ubeta.rs,done
|
||||
vern16.f,VERN16,FUNCTION,True,"BASICS","","BASICS","",False,src/tlusty/math/atomic/vern16.rs,done
|
||||
vern18.f,VERN18,FUNCTION,True,"BASICS","","BASICS","",False,src/tlusty/math/atomic/vern18.rs,done
|
||||
vern20.f,VERN20,FUNCTION,True,"BASICS","","BASICS","",False,src/tlusty/math/atomic/vern20.rs,done
|
||||
vern26.f,VERN26,FUNCTION,True,"BASICS","","BASICS","",False,src/tlusty/math/atomic/vern26.rs,done
|
||||
verner.f,VERNER,FUNCTION,False,"BASICS|ATOMIC","VERN26|QUIT|VERN18|VERN16|VERN20","ATOMIC|BASICS","VERN26|VERN18|QUIT|VERN16|VERN20",False,src/tlusty/math/atomic/verner.rs,done
|
||||
visini.f,VISINI,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ|ITERAT","","ITERAT|ATOMIC|MODELQ|BASICS","",True,src/tlusty/math/io/visini.rs,done
|
||||
voigt.f,VOIGT,FUNCTION,True,"","","","",False,src/tlusty/math/special/voigt.rs,done
|
||||
voigte.f,VOIGTE,FUNCTION,True,"","","","",False,src/tlusty/math/special/voigte.rs,done
|
||||
wn.f,WN,FUNCTION,True,"BASICS","","BASICS","",False,src/tlusty/math/utils/wn.rs,done
|
||||
wnstor.f,WNSTOR,SUBROUTINE,False,"BASICS|ATOMIC|MODELQ","WN","ATOMIC|MODELQ|BASICS","WN",False,src/tlusty/math/utils/wnstor.rs,done
|
||||
xenini.f,XENINI,SUBROUTINE,False,"BASICS|MODELQ","","MODELQ|BASICS","",True,src/tlusty/io/xenini.rs,done
|
||||
xk2dop.f,XK2DOP,FUNCTION,True,"","","","",False,src/tlusty/math/utils/xk2dop.rs,done
|
||||
yint.f,YINT,FUNCTION,True,"","","","",False,src/tlusty/math/interpolation/yint.rs,done
|
||||
ylintp.f,YLINTP,FUNCTION,True,"","","","",False,src/tlusty/math/interpolation/ylintp.rs,done
|
||||
zmrho.f,ZMRHO,SUBROUTINE,False,"BASICS|MODELQ","ERFCIN|BETAH","MODELQ|BASICS","ERFCX|ERFCIN|BETAH",False,src/tlusty/math/utils/zmrho.rs,done
|
||||
|
@ -1,65 +1,99 @@
|
||||
#!/bin/bash
|
||||
|
||||
# --- 配置变量 ---
|
||||
WORK_DIR="/home/fmq/program/SpectraRust"
|
||||
CMD_PATH="/home/fmq/.claude/local/claude"
|
||||
CMD_ARGS="--permission-mode bypassPermissions --print '/codegraph-guide 继续执行重构任务。禁止询问，禁止总结报告，禁止跳过复杂模块。'"
|
||||
WORK_DIR="/home/dckj/SpectraRust"
|
||||
CMD_PATH="/usr/bin/claude"
|
||||
CMD_PROMPT="使用 codegraph-guide skill 继续执行重构任务。"
|
||||
|
||||
# 日志文件路径：修改为工作目录内部
|
||||
# 状态文件
|
||||
PHASE_FILE="${WORK_DIR}/.f2r_phase"
|
||||
COMPLETE_FILE="${WORK_DIR}/.f2r_complete"
|
||||
RATE_LIMIT_FILE="${WORK_DIR}/.f2r_rate_limit"
|
||||
TASKS_FILE="${WORK_DIR}/.f2r_tasks"
|
||||
|
||||
# 日志文件路径
|
||||
LOG_FILE="${WORK_DIR}/logs/claude_$(date +%Y%m%d_%H%M%S).log"
|
||||
|
||||
# --- 1. 环境检查 ---
|
||||
# 检查工作目录是否存在
|
||||
if [ ! -d "$WORK_DIR" ]; then
|
||||
echo "❌ 错误: 工作目录不存在: $WORK_DIR"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# 检查命令文件是否存在且可执行
|
||||
if [ ! -x "$CMD_PATH" ]; then
|
||||
echo "❌ 错误: 命令不存在或不可执行: $CMD_PATH"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# --- 新增：检查是否已有 claude 进程在运行 ---
|
||||
echo "正在检查是否有 claude 进程在运行..."
|
||||
# 使用 ps aux 列出所有进程，然后 grep 查找 'claude'，再用 grep -v grep 排除掉 grep 命令本身
|
||||
if ps aux | grep '[c]laude' > /dev/null; then
|
||||
echo "⚠️ 检测到 claude 进程已在运行，退出脚本。"
|
||||
# 可选：显示正在运行的进程信息
|
||||
ps aux | grep '[c]laude'
|
||||
# --- 2. 完成检测 ---
|
||||
if [ -f "$COMPLETE_FILE" ]; then
|
||||
echo "✅ 重构已标记为完成 ($(cat "$COMPLETE_FILE"))，跳过。"
|
||||
echo "如需重新启动，请删除 ${COMPLETE_FILE}"
|
||||
exit 0
|
||||
fi
|
||||
|
||||
# --- 3. 429 限流退避 ---
|
||||
if [ -f "$RATE_LIMIT_FILE" ]; then
|
||||
LIMIT_UNTIL=$(cat "$RATE_LIMIT_FILE" 2>/dev/null)
|
||||
if [ -n "$LIMIT_UNTIL" ]; then
|
||||
# 将 "2026-06-08 09:10:18" 格式转换为 epoch
|
||||
RESET_EPOCH=$(date -d "$LIMIT_UNTIL" +%s 2>/dev/null)
|
||||
NOW_EPOCH=$(date +%s)
|
||||
if [ -n "$RESET_EPOCH" ] && [ "$NOW_EPOCH" -lt "$RESET_EPOCH" ]; then
|
||||
REMAINING=$(( (RESET_EPOCH - NOW_EPOCH) / 60 ))
|
||||
echo "⏳ API 限流中，还需等待 ${REMAINING} 分钟（重置于 ${LIMIT_UNTIL}），跳过。"
|
||||
exit 0
|
||||
else
|
||||
# 已过重置时间，清除标记
|
||||
rm -f "$RATE_LIMIT_FILE"
|
||||
echo "🔓 限流已重置，继续执行。"
|
||||
fi
|
||||
fi
|
||||
fi
|
||||
|
||||
# --- 4. 检查并发进程 ---
|
||||
RUNNING_PIDS=$(pgrep -f "claude.*--print" 2>/dev/null)
|
||||
if [ -n "$RUNNING_PIDS" ]; then
|
||||
echo "⚠️ 检测到已有调度任务在运行 (PID: $RUNNING_PIDS)，退出脚本。"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# --- 2. 启动进程 ---
|
||||
# 切换到工作目录
|
||||
# --- 5. 启动进程 ---
|
||||
cd "$WORK_DIR" || exit 1
|
||||
|
||||
# 执行命令
|
||||
# nohup 保证退出终端后进程不挂
|
||||
# < /dev/null 防止进程读取终端输入导致挂起
|
||||
# > "$LOG_FILE" 2>&1 将标准输出和错误输出都重定向到日志文件
|
||||
nohup "$CMD_PATH" $CMD_ARGS < /dev/null > "$LOG_FILE" 2>&1 &
|
||||
nohup "$CMD_PATH" --permission-mode bypassPermissions --print "$CMD_PROMPT" \
|
||||
< /dev/null > "$LOG_FILE" 2>&1 &
|
||||
CURRENT_PID=$!
|
||||
|
||||
# --- 3. 验证启动结果 ---
|
||||
# 短暂休眠，给进程一点初始化时间，以便捕获即时崩溃（如缺少动态库）
|
||||
sleep 0.5
|
||||
# --- 6. 等待完成并分析结果 ---
|
||||
# --print 模式是同步的，wait 等它结束
|
||||
wait "$CURRENT_PID" 2>/dev/null
|
||||
EXIT_CODE=$?
|
||||
|
||||
# 检查进程是否仍然存活
|
||||
if kill -0 "$CURRENT_PID" 2>/dev/null; then
|
||||
echo "启动成功!"
|
||||
echo "PID: $CURRENT_PID"
|
||||
echo "日志路径: $LOG_FILE"
|
||||
exit 0
|
||||
else
|
||||
echo "❌ 启动失败! 进程已意外退出。"
|
||||
echo "--- 错误日志预览 ---"
|
||||
# 如果日志文件存在，打印其内容
|
||||
if [ -f "$LOG_FILE" ]; then
|
||||
cat "$LOG_FILE"
|
||||
else
|
||||
echo "(无日志文件生成)"
|
||||
# --- 7. 后处理：检测 429 和完成标记 ---
|
||||
if [ -f "$LOG_FILE" ]; then
|
||||
# 检测 429 限流
|
||||
if grep -q "429" "$LOG_FILE" 2>/dev/null; then
|
||||
# 提取重置时间（格式：已达到 5 小时的使用上限。您的限额将在 2026-06-08 09:10:18 重置）
|
||||
RESET_TIME=$(grep -oP '限额将在 \K[\d-]+ [\d:]+' "$LOG_FILE" 2>/dev/null | head -1)
|
||||
if [ -n "$RESET_TIME" ]; then
|
||||
echo "$RESET_TIME" > "$RATE_LIMIT_FILE"
|
||||
echo "🔴 检测到 429 限流，重置时间: ${RESET_TIME}，已记录到 ${RATE_LIMIT_FILE}"
|
||||
fi
|
||||
fi
|
||||
exit 1
|
||||
|
||||
# 检测模型不存在错误
|
||||
if grep -q "模型不存在" "$LOG_FILE" 2>/dev/null; then
|
||||
echo "❌ 模型不存在错误，暂停 30 分钟。"
|
||||
echo "$(date -d '+30 minutes' '+%Y-%m-%d %H:%M:%S')" > "$RATE_LIMIT_FILE"
|
||||
fi
|
||||
|
||||
# 统计日志大小用于诊断
|
||||
LOG_SIZE=$(wc -c < "$LOG_FILE")
|
||||
echo "✅ 会话完成 | PID: $CURRENT_PID | 退出码: $EXIT_CODE | 日志: ${LOG_SIZE} 字节"
|
||||
echo " 日志路径: $LOG_FILE"
|
||||
else
|
||||
echo "❌ 无日志文件生成"
|
||||
fi
|
||||
|
||||
exit 0
|
||||
|
||||
19
src/bin/synspec.rs
Normal file
19
src/bin/synspec.rs
Normal file
@ -0,0 +1,19 @@
|
||||
//! SYNSPEC 可执行程序入口。
|
||||
//!
|
||||
//! 用法:
|
||||
//! synspec < input.5 > output.6
|
||||
|
||||
use tlusty_rust::synspec::runner::{run_synspec, SynspecConfig};
|
||||
|
||||
fn main() -> anyhow::Result<()> {
|
||||
let config = SynspecConfig::default();
|
||||
let success = run_synspec(config);
|
||||
|
||||
if success {
|
||||
eprintln!("SYNSPEC completed successfully.");
|
||||
} else {
|
||||
eprintln!("SYNSPEC completed with errors.");
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
193
src/synspec/math/abnchn.rs
Normal file
193
src/synspec/math/abnchn.rs
Normal file
@ -0,0 +1,193 @@
|
||||
//! ABNCHN 丰度修改过程。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `ABNCHN` 函数。
|
||||
//!
|
||||
//! 用于 opacity table 评估时修改（或消除）某些元素的丰度。
|
||||
|
||||
/// ABNCHN 模式。
|
||||
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
||||
pub enum AbnchnMode {
|
||||
/// 保存当前 populations 到备份
|
||||
Save = 0,
|
||||
/// 按丰度因子缩放 populations
|
||||
Scale = 1,
|
||||
}
|
||||
|
||||
/// ABNCHN 输入参数。
|
||||
pub struct AbnchnParams<'a> {
|
||||
/// 操作模式
|
||||
pub mode: AbnchnMode,
|
||||
/// 原子数
|
||||
pub natom: usize,
|
||||
/// 每个原子的第一能级索引 (0-based)
|
||||
pub n0a: &'a [usize],
|
||||
/// 每个原子的最后能级索引 (0-based)
|
||||
pub nka: &'a [usize],
|
||||
/// 每个原子对应的原子序数 (1-based)
|
||||
pub numat: &'a [usize],
|
||||
/// 丰度缩放因子 (按原子序数索引, 1-based)
|
||||
pub relabn: &'a [f64],
|
||||
/// 当前 populations [nlevel]
|
||||
pub popul: &'a [f64],
|
||||
/// 备份 populations [nlevel] (mode=0 时写入, mode=1 时读取)
|
||||
pub popul0: &'a [f64],
|
||||
/// RRR 数组 [mion × matom]
|
||||
pub rrr: &'a [f64],
|
||||
/// 离子数
|
||||
pub mion0: usize,
|
||||
/// 原子种类数 (最大)
|
||||
pub matom: usize,
|
||||
}
|
||||
|
||||
/// ABNCHN 输出结果。
|
||||
pub struct AbnchnOutput {
|
||||
/// 修改后的 populations [nlevel]
|
||||
pub popul_new: Vec<f64>,
|
||||
/// 修改后的 RRR 数组 [mion × matom]
|
||||
pub rrr_new: Vec<f64>,
|
||||
/// 更新后的备份 populations [nlevel]
|
||||
pub popul0_new: Vec<f64>,
|
||||
}
|
||||
|
||||
/// ABNCHN 丰度修改过程。
|
||||
///
|
||||
/// mode=0: 保存当前 populations 到备份。
|
||||
/// mode=1: 按丰度因子缩放 populations 和 RRR。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `params` - ABNCHN 参数
|
||||
///
|
||||
/// # 返回值
|
||||
///
|
||||
/// 修改后的 populations 和 RRR
|
||||
pub fn abnchn(params: &AbnchnParams) -> AbnchnOutput {
|
||||
let AbnchnParams {
|
||||
mode,
|
||||
natom,
|
||||
n0a,
|
||||
nka,
|
||||
numat,
|
||||
relabn,
|
||||
popul,
|
||||
popul0,
|
||||
rrr,
|
||||
mion0,
|
||||
matom,
|
||||
} = *params;
|
||||
|
||||
let _nlevel = popul.len();
|
||||
let mut popul_new = popul.to_vec();
|
||||
let mut popul0_new = popul0.to_vec();
|
||||
let mut rrr_new = rrr.to_vec();
|
||||
|
||||
match mode {
|
||||
AbnchnMode::Save => {
|
||||
// 保存当前 populations 到备份
|
||||
for iat in 0..natom {
|
||||
for ii in n0a[iat]..=nka[iat] {
|
||||
popul0_new[ii] = popul[ii];
|
||||
}
|
||||
}
|
||||
}
|
||||
AbnchnMode::Scale => {
|
||||
// 按丰度因子缩放 populations
|
||||
for iat in 0..natom {
|
||||
let ia = numat[iat] - 1; // 0-based
|
||||
for ii in n0a[iat]..=nka[iat] {
|
||||
popul_new[ii] = popul0[ii] * relabn[ia];
|
||||
}
|
||||
}
|
||||
|
||||
// 缩放 RRR 数组
|
||||
for ia in 0..matom {
|
||||
for io in 0..mion0 {
|
||||
let idx = io * matom + ia;
|
||||
rrr_new[idx] = rrr[idx] * relabn[ia];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
AbnchnOutput {
|
||||
popul_new,
|
||||
rrr_new,
|
||||
popul0_new,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_abnchn_save() {
|
||||
let nlevel = 6;
|
||||
let popul = vec![1.0e10, 2.0e10, 3.0e10, 4.0e10, 5.0e10, 6.0e10];
|
||||
let popul0 = vec![0.0; nlevel];
|
||||
let rrr = vec![1.0; 4];
|
||||
let n0a = vec![0usize, 3];
|
||||
let nka = vec![2usize, 5];
|
||||
let numat = vec![1usize, 2];
|
||||
let relabn = vec![1.0, 0.5, 0.3]; // index 0 unused, 1=H, 2=He
|
||||
|
||||
let params = AbnchnParams {
|
||||
mode: AbnchnMode::Save,
|
||||
natom: 2,
|
||||
n0a: &n0a,
|
||||
nka: &nka,
|
||||
numat: &numat,
|
||||
relabn: &relabn,
|
||||
popul: &popul,
|
||||
popul0: &popul0,
|
||||
rrr: &rrr,
|
||||
mion0: 2,
|
||||
matom: 3,
|
||||
};
|
||||
|
||||
let output = abnchn(¶ms);
|
||||
|
||||
// mode=0: 复制 popul 到 popul0
|
||||
assert_eq!(output.popul0_new, popul);
|
||||
// popul 不变
|
||||
assert_eq!(output.popul_new, popul);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_abnchn_scale() {
|
||||
let nlevel = 6;
|
||||
let popul = vec![1.0e10, 2.0e10, 3.0e10, 4.0e10, 5.0e10, 6.0e10];
|
||||
let popul0 = vec![1.0e10, 2.0e10, 3.0e10, 4.0e10, 5.0e10, 6.0e10];
|
||||
let rrr = vec![1.0, 2.0, 3.0, 4.0, 5.0, 6.0]; // mion0=2, matom=3
|
||||
let n0a = vec![0usize, 3];
|
||||
let nka = vec![2usize, 5];
|
||||
let numat = vec![1usize, 2];
|
||||
let relabn = vec![1.0, 0.5, 0.3]; // H=0.5, He=0.3
|
||||
|
||||
let params = AbnchnParams {
|
||||
mode: AbnchnMode::Scale,
|
||||
natom: 2,
|
||||
n0a: &n0a,
|
||||
nka: &nka,
|
||||
numat: &numat,
|
||||
relabn: &relabn,
|
||||
popul: &popul,
|
||||
popul0: &popul0,
|
||||
rrr: &rrr,
|
||||
mion0: 2,
|
||||
matom: 3,
|
||||
};
|
||||
|
||||
let output = abnchn(¶ms);
|
||||
|
||||
// mode=1: popul = popul0 * relabn[ia]
|
||||
// atom 0 (H, numat=1): ia=0, relabn[0]=1.0 → 不变
|
||||
assert_eq!(output.popul_new[0], 1.0e10 * 1.0);
|
||||
assert_eq!(output.popul_new[1], 2.0e10 * 1.0);
|
||||
assert_eq!(output.popul_new[2], 3.0e10 * 1.0);
|
||||
// atom 1 (He, numat=2): ia=1, relabn[1]=0.5
|
||||
assert_eq!(output.popul_new[3], 4.0e10 * 0.5);
|
||||
assert_eq!(output.popul_new[4], 5.0e10 * 0.5);
|
||||
assert_eq!(output.popul_new[5], 6.0e10 * 0.5);
|
||||
}
|
||||
}
|
||||
307
src/synspec/math/allard.rs
Normal file
307
src/synspec/math/allard.rs
Normal file
@ -0,0 +1,307 @@
|
||||
//! Quasi-molecular opacity for Lyman alpha, beta, gamma, and Balmer alpha.
|
||||
//!
|
||||
//! Translated from SYNSPEC `allard` subroutine (synspec54.f).
|
||||
|
||||
|
||||
// ============================================================================
|
||||
// Constants
|
||||
// ============================================================================
|
||||
|
||||
/// Maximum number of wavelength points in tables
|
||||
pub const NXMAX: usize = 1400;
|
||||
|
||||
/// Maximum number of density components
|
||||
pub const NNMAX: usize = 5;
|
||||
|
||||
// Normalization constants: 8.8528e-29 * lambda_0^2 * f_ij
|
||||
const XNORMA: f64 = 8.8528e-29 * 1215.6 * 1215.6 * 0.41618; // Lyman alpha
|
||||
const XNORMB: f64 = 8.8528e-29 * 1025.73 * 1025.7 * 0.0791; // Lyman beta
|
||||
const XNORMG: f64 = 8.8528e-29 * 972.53 * 972.53 * 0.0290; // Lyman gamma
|
||||
const XNORMC: f64 = 8.8528e-29 * 6562.0 * 6562.0 * 0.6407; // Balmer alpha
|
||||
|
||||
// ============================================================================
|
||||
// AllardData - precomputed table data
|
||||
// ============================================================================
|
||||
|
||||
/// Precomputed quasi-molecular opacity tables for one transition.
|
||||
///
|
||||
/// Corresponds to Fortran COMMON blocks `callarda`, `callardb`, `callardg`, `callardc`.
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct AllardTable {
|
||||
/// Wavelength points (Angstroms)
|
||||
pub xl: Vec<f64>,
|
||||
/// Profile data: `pl[i][j]` for wavelength point `i`, component `j`
|
||||
/// Components: 0=neutral linear, 1=neutral quadratic,
|
||||
/// 2=charged linear, 3=charged quadratic, 4=cross term
|
||||
pub pl: Vec<[f64; NNMAX]>,
|
||||
/// Normalized neutral density scale
|
||||
pub stnne: f64,
|
||||
/// Normalized charged density scale
|
||||
pub stnch: f64,
|
||||
/// Neutral velocity scale
|
||||
pub vneu: f64,
|
||||
/// Charged velocity scale
|
||||
pub vcha: f64,
|
||||
/// Number of wavelength points
|
||||
pub nx: usize,
|
||||
/// Warning flag for high density
|
||||
pub iwarn: bool,
|
||||
}
|
||||
|
||||
impl Default for AllardTable {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
xl: Vec::new(),
|
||||
pl: Vec::new(),
|
||||
stnne: 1.0,
|
||||
stnch: 1.0,
|
||||
vneu: 1.0,
|
||||
vcha: 1.0,
|
||||
nx: 0,
|
||||
iwarn: false,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// AllardData - all four transitions
|
||||
// ============================================================================
|
||||
|
||||
/// Container for all four quasi-molecular transitions.
|
||||
#[derive(Debug, Clone, Default)]
|
||||
pub struct AllardData {
|
||||
/// Lyman alpha (1→2)
|
||||
pub lalp: AllardTable,
|
||||
/// Lyman beta (1→3)
|
||||
pub bet: AllardTable,
|
||||
/// Lyman gamma (1→4)
|
||||
pub gam: AllardTable,
|
||||
/// Balmer alpha (2→3)
|
||||
pub bal: AllardTable,
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// Core interpolation function
|
||||
// ============================================================================
|
||||
|
||||
/// Interpolate quasi-molecular profile from precomputed table.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `table` - Precomputed table for this transition
|
||||
/// * `xl` - Wavelength in Angstroms
|
||||
/// * `hneutr` - Neutral H particle density [cm⁻³]
|
||||
/// * `hcharg` - Ionized H particle density [cm⁻³]
|
||||
///
|
||||
/// # Returns
|
||||
/// Profile value normalized to 1.0e8 when integrated over Angstroms.
|
||||
/// Returns 0.0 if wavelength is outside table range.
|
||||
fn interpolate_profile(
|
||||
table: &AllardTable,
|
||||
xl: f64,
|
||||
hneutr: f64,
|
||||
hcharg: f64,
|
||||
xnorm: f64,
|
||||
) -> f64 {
|
||||
if table.nx == 0 {
|
||||
return 0.0;
|
||||
}
|
||||
if xl < table.xl[0] || xl > table.xl[table.nx - 1] {
|
||||
return 0.0;
|
||||
}
|
||||
|
||||
// Normalized densities
|
||||
let vn1 = hneutr / table.stnne;
|
||||
let vn2 = hcharg / table.stnch;
|
||||
let vns = vn1 * table.vneu + vn2 * table.vcha;
|
||||
|
||||
// Density warning
|
||||
// (handled externally via iwarn flag)
|
||||
|
||||
let vn11 = vn1 * vn1;
|
||||
let vn22 = vn2 * vn2;
|
||||
let vn12 = vn1 * vn2;
|
||||
let xnorm_fac = 1.0 / (1.0 + vns + 0.5 * vns * vns);
|
||||
|
||||
// Binary search for wavelength interval
|
||||
let mut jl: usize = 0;
|
||||
let mut ju = table.nx;
|
||||
while ju - jl > 1 {
|
||||
let jm = (ju + jl) / 2;
|
||||
if (table.xl[table.nx - 1] > table.xl[0]) == (xl > table.xl[jm]) {
|
||||
jl = jm;
|
||||
} else {
|
||||
ju = jm;
|
||||
}
|
||||
}
|
||||
let mut j = jl;
|
||||
if j == 0 {
|
||||
j = 1;
|
||||
}
|
||||
if j >= table.nx - 1 {
|
||||
j = table.nx - 2;
|
||||
}
|
||||
|
||||
// Linear interpolation factor
|
||||
let a1 = (xl - table.xl[j]) / (table.xl[j + 1] - table.xl[j]);
|
||||
let a0 = 1.0 - a1;
|
||||
|
||||
// Interpolate each density component
|
||||
let p1 = vn1 * (a0 * table.pl[j][0] + a1 * table.pl[j + 1][0]);
|
||||
let p11 = vn11 * (a0 * table.pl[j][1] + a1 * table.pl[j + 1][1]);
|
||||
let p2 = vn2 * (a0 * table.pl[j][2] + a1 * table.pl[j + 1][2]);
|
||||
let p22 = vn22 * (a0 * table.pl[j][3] + a1 * table.pl[j + 1][3]);
|
||||
let p12 = vn12 * (a0 * table.pl[j][4] + a1 * table.pl[j + 1][4]);
|
||||
|
||||
(p1 + p2 + p11 + p22 + p12) * xnorm_fac * xnorm
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// Main entry point
|
||||
// ============================================================================
|
||||
|
||||
/// Compute quasi-molecular opacity profile.
|
||||
///
|
||||
/// Translated from SYNSPEC `allard` subroutine (synspec54.f).
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `data` - Precomputed quasi-molecular tables
|
||||
/// * `xl` - Wavelength in Angstroms
|
||||
/// * `hneutr` - Neutral H particle density [cm⁻³]
|
||||
/// * `hcharg` - Ionized H particle density [cm⁻³]
|
||||
/// * `iq` - Quantum number of lower level
|
||||
/// * `jq` - Quantum number of upper level:
|
||||
/// - 2 → Lyman alpha
|
||||
/// - 3 → Lyman beta (if iq=1) or Balmer alpha (if iq=2)
|
||||
/// - 4 → Lyman gamma
|
||||
///
|
||||
/// # Returns
|
||||
/// Profile value. Returns 0.0 if transition not recognized or out of range.
|
||||
pub fn allard(
|
||||
data: &AllardData,
|
||||
xl: f64,
|
||||
hneutr: f64,
|
||||
hcharg: f64,
|
||||
iq: i32,
|
||||
jq: i32,
|
||||
) -> f64 {
|
||||
// Lyman alpha (1→2)
|
||||
if iq == 1 && jq == 2 {
|
||||
return interpolate_profile(&data.lalp, xl, hneutr, hcharg, XNORMA);
|
||||
}
|
||||
|
||||
// Lyman beta (1→3)
|
||||
if iq == 1 && jq == 3 {
|
||||
return interpolate_profile(&data.bet, xl, hneutr, hcharg, XNORMB);
|
||||
}
|
||||
|
||||
// Lyman gamma (1→4)
|
||||
if iq == 1 && jq == 4 {
|
||||
return interpolate_profile(&data.gam, xl, hneutr, hcharg, XNORMG);
|
||||
}
|
||||
|
||||
// Balmer alpha (2→3)
|
||||
if iq == 2 && jq == 3 {
|
||||
// For Balmer alpha, only charged component contributes
|
||||
// (vn1 = 0 in Fortran code)
|
||||
if data.bal.nx == 0 {
|
||||
return 0.0;
|
||||
}
|
||||
if xl < data.bal.xl[0] || xl > data.bal.xl[data.bal.nx - 1] {
|
||||
return 0.0;
|
||||
}
|
||||
|
||||
let vn2 = hcharg / data.bal.stnch;
|
||||
let vns = vn2 * data.bal.vcha;
|
||||
let vn22 = vn2 * vn2;
|
||||
let xnorm_fac = 1.0 / (1.0 + vns + 0.5 * vns * vns);
|
||||
|
||||
// Binary search
|
||||
let mut jl: usize = 0;
|
||||
let mut ju = data.bal.nx;
|
||||
while ju - jl > 1 {
|
||||
let jm = (ju + jl) / 2;
|
||||
if (data.bal.xl[data.bal.nx - 1] > data.bal.xl[0])
|
||||
== (xl > data.bal.xl[jm])
|
||||
{
|
||||
jl = jm;
|
||||
} else {
|
||||
ju = jm;
|
||||
}
|
||||
}
|
||||
let mut j = jl;
|
||||
if j == 0 {
|
||||
j = 1;
|
||||
}
|
||||
if j >= data.bal.nx - 1 {
|
||||
j = data.bal.nx - 2;
|
||||
}
|
||||
|
||||
let a1 = (xl - data.bal.xl[j]) / (data.bal.xl[j + 1] - data.bal.xl[j]);
|
||||
let a0 = 1.0 - a1;
|
||||
|
||||
let p2 = vn2 * (a0 * data.bal.pl[j][2] + a1 * data.bal.pl[j + 1][2]);
|
||||
let p22 = vn22 * (a0 * data.bal.pl[j][3] + a1 * data.bal.pl[j + 1][3]);
|
||||
|
||||
return (p2 + p22) * xnorm_fac * XNORMC;
|
||||
}
|
||||
|
||||
0.0
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// Tests
|
||||
// ============================================================================
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_allard_empty_table() {
|
||||
let data = AllardData::default();
|
||||
let prof = allard(&data, 1215.6, 1e12, 1e10, 1, 2);
|
||||
assert_eq!(prof, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_allard_out_of_range() {
|
||||
let mut data = AllardData::default();
|
||||
data.lalp.xl = vec![1200.0, 1210.0, 1220.0];
|
||||
data.lalp.pl = vec![[1.0; NNMAX]; 3];
|
||||
data.lalp.nx = 3;
|
||||
|
||||
// Below range
|
||||
let prof = allard(&data, 1199.0, 1e12, 1e10, 1, 2);
|
||||
assert_eq!(prof, 0.0);
|
||||
|
||||
// Above range
|
||||
let prof = allard(&data, 1221.0, 1e12, 1e10, 1, 2);
|
||||
assert_eq!(prof, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_allard_lyman_alpha() {
|
||||
let mut data = AllardData::default();
|
||||
data.lalp.xl = vec![1210.0, 1215.0, 1220.0];
|
||||
data.lalp.pl = vec![
|
||||
[1.0, 0.5, 0.3, 0.2, 0.1],
|
||||
[2.0, 1.0, 0.6, 0.4, 0.2],
|
||||
[1.5, 0.75, 0.45, 0.3, 0.15],
|
||||
];
|
||||
data.lalp.stnne = 1e12;
|
||||
data.lalp.stnch = 1e10;
|
||||
data.lalp.vneu = 1.0;
|
||||
data.lalp.vcha = 1.0;
|
||||
data.lalp.nx = 3;
|
||||
|
||||
let prof = allard(&data, 1215.0, 1e12, 1e10, 1, 2);
|
||||
assert!(prof > 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_allard_unknown_transition() {
|
||||
let data = AllardData::default();
|
||||
let prof = allard(&data, 1215.6, 1e12, 1e10, 2, 4); // Not a valid transition
|
||||
assert_eq!(prof, 0.0);
|
||||
}
|
||||
}
|
||||
141
src/synspec/math/carbon.rs
Normal file
141
src/synspec/math/carbon.rs
Normal file
@ -0,0 +1,141 @@
|
||||
//! 中性碳光致电离截面（Taylor 数据）。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `carbon.f`
|
||||
//!
|
||||
//! 使用 G.B. Taylor (private communication) 的数据，
|
||||
//! 计算中性碳 2p¹D 和 2p¹S 能级的光致电离截面。
|
||||
|
||||
/// 频率网格 2 (单位 FR0)，用于 IB=-602 (2p¹D)
|
||||
const FR2: [f64; 34] = [
|
||||
0.74, 0.75, 0.76, 0.77, 0.78, 0.79, 0.80, 0.81, 0.82, 0.83,
|
||||
0.85, 0.86, 0.87, 0.88, 0.89, 0.90, 0.91, 0.92, 0.93, 0.94,
|
||||
0.95, 0.96, 0.97, 0.98, 0.99, 1.00, 1.10, 1.20, 1.30, 1.45,
|
||||
1.50, 1.60, 1.80, 2.00,
|
||||
];
|
||||
|
||||
/// 截面数据 2 (Mbarn)，用于 IB=-602 (2p¹D)
|
||||
const SG2: [f64; 34] = [
|
||||
12.04, 12.03, 12.09, 12.26, 12.60, 13.24, 14.36, 16.24, 19.28, 23.94,
|
||||
37.41, 42.88, 44.76, 43.41, 40.46, 37.19, 34.26, 31.82, 29.96, 28.57,
|
||||
27.68, 27.37, 27.84, 29.69, 34.45, 46.35, 13.80, 11.54, 10.40, 8.96,
|
||||
8.54, 7.47, 6.53, 5.66,
|
||||
];
|
||||
|
||||
/// 频率网格 3 (单位 FR0)，用于 IB=-603 (2p¹S)
|
||||
const FR3: [f64; 45] = [
|
||||
0.66, 0.68, 0.70, 0.72, 0.74, 0.76, 0.78, 0.80, 0.82, 0.84,
|
||||
0.86, 0.864, 0.866, 0.868, 0.87, 0.874, 0.876, 0.88, 0.882, 0.884,
|
||||
0.886, 0.888, 0.89, 0.894, 0.896, 0.898, 0.90, 0.904, 0.908, 0.910,
|
||||
0.920, 0.94, 0.98, 1.00, 1.10, 1.20, 1.26, 1.34, 1.36, 1.40,
|
||||
1.46, 1.60, 1.70, 1.80, 2.00,
|
||||
];
|
||||
|
||||
/// 截面数据 3 (Mbarn)，用于 IB=-603 (2p¹S)
|
||||
const SG3: [f64; 45] = [
|
||||
13.94, 13.29, 12.56, 11.73, 10.82, 10.18, 8.62, 7.27, 5.74, 4.14,
|
||||
4.61, 5.92, 6.94, 8.34, 10.21, 16.12, 20.64, 34.56, 44.82, 57.71,
|
||||
73.09, 89.99, 106.38, 127.08, 128.38, 124.44, 117.17, 99.32, 82.95, 76.05,
|
||||
52.65, 33.23, 21.29, 18.69, 12.62, 11.44, 9.77, 7.53, 10.47, 9.65,
|
||||
10.19, 7.28, 6.70, 6.11, 4.96,
|
||||
];
|
||||
|
||||
/// 参考频率 (Hz)
|
||||
const FR0: f64 = 3.28805e15;
|
||||
|
||||
/// 截面单位转换因子 (cm^2)
|
||||
const SIG_FACTOR: f64 = 1.0e-18;
|
||||
|
||||
/// 中性碳光致电离截面。
|
||||
///
|
||||
/// 根据 Taylor 数据，对给定频率进行线性插值。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `ib` - 能级标识（-602 = 2p¹D, -603 = 2p¹S）
|
||||
/// * `fr` - 频率 (Hz)
|
||||
///
|
||||
/// # 返回值
|
||||
///
|
||||
/// 光致电离截面 (cm^2)
|
||||
pub fn carbon(ib: i32, fr: f64) -> f64 {
|
||||
let f = fr / FR0;
|
||||
|
||||
if ib == -602 {
|
||||
// 2p¹D 能级
|
||||
let mut j = 1; // 0-indexed, 默认值
|
||||
if f > FR2[0] {
|
||||
for i in 1..34 {
|
||||
if f > FR2[i - 1] && f <= FR2[i] {
|
||||
j = i;
|
||||
break;
|
||||
}
|
||||
}
|
||||
} else {
|
||||
j = 1;
|
||||
}
|
||||
let sg = (f - FR2[j - 1]) / (FR2[j] - FR2[j - 1]) * (SG2[j] - SG2[j - 1]) + SG2[j - 1];
|
||||
return sg * SIG_FACTOR;
|
||||
}
|
||||
|
||||
if ib == -603 {
|
||||
// 2p¹S 能级
|
||||
let mut j = 1;
|
||||
if f > FR3[0] {
|
||||
for i in 1..45 {
|
||||
if f > FR3[i - 1] && f <= FR3[i] {
|
||||
j = i;
|
||||
break;
|
||||
}
|
||||
}
|
||||
} else {
|
||||
j = 1;
|
||||
}
|
||||
let sg = (f - FR3[j - 1]) / (FR3[j] - FR3[j - 1]) * (SG3[j] - SG3[j - 1]) + SG3[j - 1];
|
||||
return sg * SIG_FACTOR;
|
||||
}
|
||||
|
||||
0.0
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_carbon_602_in_range() {
|
||||
// 2p¹D 在有效频率范围内
|
||||
let fr = 0.9 * FR0;
|
||||
let result = carbon(-602, fr);
|
||||
assert!(result > 0.0);
|
||||
assert!(result.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_carbon_603_in_range() {
|
||||
// 2p¹S 在有效频率范围内
|
||||
let fr = 0.85 * FR0;
|
||||
let result = carbon(-603, fr);
|
||||
assert!(result > 0.0);
|
||||
assert!(result.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_carbon_below_range() {
|
||||
let fr = 0.5 * FR0;
|
||||
let result = carbon(-602, fr);
|
||||
assert!(result >= 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_carbon_above_range() {
|
||||
let fr = 3.0 * FR0;
|
||||
let result = carbon(-602, fr);
|
||||
assert!(result >= 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_carbon_invalid_ib() {
|
||||
let result = carbon(-601, FR0);
|
||||
assert_eq!(result, 0.0);
|
||||
}
|
||||
}
|
||||
343
src/synspec/math/change.rs
Normal file
343
src/synspec/math/change.rs
Normal file
@ -0,0 +1,343 @@
|
||||
//! CHANGE 控制过程。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `CHANGE` 函数。
|
||||
//!
|
||||
//! 在显式能级系统与输入能级编号不一致时，重新评估初始能级 populations。
|
||||
//! 仅用于 NLTE 输入模型。
|
||||
|
||||
use crate::synspec::state::constants::BOLK;
|
||||
use crate::synspec::math::{lineqs, ratmat};
|
||||
|
||||
/// CHANGE 模式参数（每个能级一组）。
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct ChangeLevelParams {
|
||||
/// 旧能级索引 (1-based)；0 = 无对应旧能级
|
||||
pub iold: usize,
|
||||
/// 评估模式
|
||||
/// - 0: 复制旧能级 population × REL
|
||||
/// - 1: LTE 相对于下一电离态
|
||||
/// - 2: b-因子匹配
|
||||
/// - 3: 完整 LTE（SABOLF + RATMAT + LINEQS）
|
||||
pub mode: usize,
|
||||
/// 下一电离态旧索引 (1-based)
|
||||
pub nxtold: usize,
|
||||
/// 新系统中参考能级索引 (1-based)
|
||||
pub isinew: usize,
|
||||
/// 旧系统中参考能级索引 (1-based)
|
||||
pub isiold: usize,
|
||||
/// 参考能级下一电离态旧索引 (1-based)
|
||||
pub nxtsio: usize,
|
||||
/// population 乘子
|
||||
pub rel: f64,
|
||||
}
|
||||
|
||||
/// CHANGE 输入参数。
|
||||
pub struct ChangeParams<'a> {
|
||||
/// 每能级参数
|
||||
pub levels: &'a [ChangeLevelParams],
|
||||
/// 深度点数
|
||||
pub nd: usize,
|
||||
/// 温度数组 (K)
|
||||
pub temp: &'a [f64],
|
||||
/// 电子密度数组 (cm^-3)
|
||||
pub elec: &'a [f64],
|
||||
/// 当前 populations [nlevel × nd, row-major]
|
||||
pub popul: &'a [f64],
|
||||
/// 统计权重
|
||||
pub g: &'a [f64],
|
||||
/// 电离能 (K)
|
||||
pub enion: &'a [f64],
|
||||
/// 元素索引 (1-based)
|
||||
pub iel: &'a [usize],
|
||||
/// 下一离子态索引 (1-based)
|
||||
pub nnext: &'a [usize],
|
||||
/// 能级数
|
||||
pub nlevel: usize,
|
||||
/// N0 偏移数组 (1-based)
|
||||
pub n0a: &'a [usize],
|
||||
/// NK 偏移数组 (1-based)
|
||||
pub nka: &'a [usize],
|
||||
/// SBF 数组
|
||||
pub sbf: &'a [f64],
|
||||
/// WOP 数组 (nlevel × nd, row-major)
|
||||
pub wop: &'a [f64],
|
||||
/// ILK 数组 (1-based)
|
||||
pub ilk: &'a [usize],
|
||||
/// USUM 数组 (1-based)
|
||||
pub usum: &'a [f64],
|
||||
/// ATTOT 数组
|
||||
pub attot: &'a [f64],
|
||||
}
|
||||
|
||||
/// CHANGE 输出结果。
|
||||
pub struct ChangeOutput {
|
||||
/// 新 populations [nlevel × nd, row-major]
|
||||
pub popul_new: Vec<f64>,
|
||||
/// 能级数
|
||||
pub nlevel: usize,
|
||||
/// 深度点数
|
||||
pub nd: usize,
|
||||
}
|
||||
|
||||
/// S = 2*h/c^2 * (1e-8)^2 = 2.0706e-16 (转换因子)
|
||||
const S: f64 = 2.0706e-16;
|
||||
|
||||
/// 获取 popul[level][depth] 的辅助函数。
|
||||
#[inline]
|
||||
fn popul_at(popul: &[f64], nd: usize, level: usize, depth: usize) -> f64 {
|
||||
popul[level * nd + depth]
|
||||
}
|
||||
|
||||
/// CHANGE 控制过程。
|
||||
///
|
||||
/// 在显式能级系统与输入能级编号不一致时，重新评估初始能级 populations。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `params` - CHANGE 参数
|
||||
///
|
||||
/// # 返回值
|
||||
///
|
||||
/// 新的能级 populations
|
||||
pub fn change(params: &ChangeParams) -> ChangeOutput {
|
||||
let ChangeParams {
|
||||
levels,
|
||||
nd,
|
||||
temp,
|
||||
elec,
|
||||
popul,
|
||||
g,
|
||||
enion,
|
||||
iel,
|
||||
nnext,
|
||||
nlevel,
|
||||
n0a,
|
||||
nka,
|
||||
sbf,
|
||||
wop,
|
||||
ilk,
|
||||
usum,
|
||||
attot,
|
||||
} = *params;
|
||||
|
||||
let mut popul_new = vec![0.0f64; nlevel * nd];
|
||||
let mut ifese = 0usize;
|
||||
|
||||
for (ii, lvl) in levels.iter().enumerate() {
|
||||
let iold = lvl.iold;
|
||||
let mode = lvl.mode;
|
||||
let nxtold = lvl.nxtold;
|
||||
let isinew = lvl.isinew;
|
||||
let isiold = lvl.isiold;
|
||||
let nxtsio = lvl.nxtsio;
|
||||
let mut rel = lvl.rel;
|
||||
if rel == 0.0 {
|
||||
rel = 1.0;
|
||||
}
|
||||
|
||||
if mode >= 3 {
|
||||
ifese += 1;
|
||||
}
|
||||
|
||||
for id in 0..nd {
|
||||
if iold != 0 {
|
||||
// 直接复制旧能级 population
|
||||
popul_new[ii * nd + id] = popul_at(popul, nd, iold - 1, id);
|
||||
continue;
|
||||
}
|
||||
|
||||
match mode {
|
||||
0 => {
|
||||
// 复制旧能级 population × REL
|
||||
popul_new[ii * nd + id] = popul_at(popul, nd, isiold - 1, id) * rel;
|
||||
}
|
||||
1 => {
|
||||
// LTE 相对于下一电离态
|
||||
let t = temp[id];
|
||||
let ane = elec[id];
|
||||
let nxt_idx = nnext[iel[ii] - 1] - 1;
|
||||
let sb = S / t / t.sqrt() * g[ii] / g[nxt_idx]
|
||||
* (enion[ii] / t / BOLK).exp();
|
||||
popul_new[ii * nd + id] = sb * ane * popul_at(popul, nd, nxtold - 1, id) * rel;
|
||||
}
|
||||
2 => {
|
||||
// b-因子匹配
|
||||
let t = temp[id];
|
||||
let kk = isinew - 1; // 0-based
|
||||
let k_next = nnext[iel[kk] - 1] - 1;
|
||||
let nxt_idx = nnext[iel[ii] - 1] - 1;
|
||||
let sb = S / t / t.sqrt() * g[ii] / g[nxt_idx]
|
||||
* (enion[ii] / t / BOLK).exp();
|
||||
let sbk = S / t / t.sqrt() * g[kk] / g[k_next]
|
||||
* (enion[kk] / t / BOLK).exp();
|
||||
popul_new[ii * nd + id] = sb / sbk
|
||||
* popul_at(popul, nd, nxtold - 1, id)
|
||||
/ popul_at(popul, nd, nxtsio - 1, id)
|
||||
* popul_at(popul, nd, isiold - 1, id)
|
||||
* rel;
|
||||
}
|
||||
_ => {
|
||||
// MODE >= 3: 完整 LTE via RATMAT + LINEQS
|
||||
if ifese == 1 {
|
||||
let ane = elec[id];
|
||||
let (ese_mat, bese) = ratmat(
|
||||
ane,
|
||||
nlevel,
|
||||
0,
|
||||
n0a,
|
||||
nka,
|
||||
nnext,
|
||||
iel,
|
||||
sbf,
|
||||
wop,
|
||||
nd,
|
||||
id,
|
||||
ilk,
|
||||
usum,
|
||||
attot,
|
||||
);
|
||||
// 解线性方程组
|
||||
let mut a = ese_mat;
|
||||
let mut b = bese;
|
||||
let poplte = lineqs(&mut a, &mut b, nlevel);
|
||||
|
||||
for iii in 0..nlevel {
|
||||
popul_new[iii * nd + id] = poplte[iii];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
ChangeOutput {
|
||||
popul_new,
|
||||
nlevel,
|
||||
nd,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_change_mode0_copy() {
|
||||
let nd = 2usize;
|
||||
let nlevel = 3usize;
|
||||
let mut popul = vec![0.0f64; nlevel * nd];
|
||||
popul[0 * nd + 0] = 1.0e10;
|
||||
popul[0 * nd + 1] = 2.0e10;
|
||||
|
||||
let levels = vec![
|
||||
ChangeLevelParams {
|
||||
iold: 0,
|
||||
mode: 0,
|
||||
nxtold: 0,
|
||||
isinew: 0,
|
||||
isiold: 1,
|
||||
nxtsio: 0,
|
||||
rel: 2.0,
|
||||
},
|
||||
];
|
||||
|
||||
let iel = vec![1usize; nlevel];
|
||||
let nnext = vec![2usize; nlevel];
|
||||
let g = vec![1.0f64; nlevel];
|
||||
let enion = vec![0.0f64; nlevel];
|
||||
let n0a = vec![0usize; nlevel];
|
||||
let nka = vec![0usize; nlevel];
|
||||
let sbf = vec![0.0f64; nlevel];
|
||||
let wop = vec![0.0f64; nlevel * nd];
|
||||
let ilk = vec![0usize; nlevel];
|
||||
let usum = vec![0.0f64; nlevel];
|
||||
let attot = vec![0.0f64; nlevel];
|
||||
let temp = vec![10000.0f64; nd];
|
||||
let elec = vec![1.0e14f64; nd];
|
||||
|
||||
let params = ChangeParams {
|
||||
levels: &levels,
|
||||
nd,
|
||||
temp: &temp,
|
||||
elec: &elec,
|
||||
popul: &popul,
|
||||
g: &g,
|
||||
enion: &enion,
|
||||
iel: &iel,
|
||||
nnext: &nnext,
|
||||
nlevel,
|
||||
n0a: &n0a,
|
||||
nka: &nka,
|
||||
sbf: &sbf,
|
||||
wop: &wop,
|
||||
ilk: &ilk,
|
||||
usum: &usum,
|
||||
attot: &attot,
|
||||
};
|
||||
|
||||
let output = change(¶ms);
|
||||
|
||||
// MODE 0: 复制 isiold=1 的 population × rel=2.0
|
||||
assert_eq!(output.popul_new[0 * nd + 0], 1.0e10 * 2.0);
|
||||
assert_eq!(output.popul_new[0 * nd + 1], 2.0e10 * 2.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_change_direct_copy() {
|
||||
let nd = 1usize;
|
||||
let nlevel = 3usize;
|
||||
let mut popul = vec![0.0f64; nlevel * nd];
|
||||
popul[2 * nd + 0] = 5.0e12;
|
||||
|
||||
let levels = vec![
|
||||
ChangeLevelParams {
|
||||
iold: 3,
|
||||
mode: 0,
|
||||
nxtold: 0,
|
||||
isinew: 0,
|
||||
isiold: 0,
|
||||
nxtsio: 0,
|
||||
rel: 1.0,
|
||||
},
|
||||
];
|
||||
|
||||
let iel = vec![1usize; nlevel];
|
||||
let nnext = vec![2usize; nlevel];
|
||||
let g = vec![1.0f64; nlevel];
|
||||
let enion = vec![0.0f64; nlevel];
|
||||
let n0a = vec![0usize; nlevel];
|
||||
let nka = vec![0usize; nlevel];
|
||||
let sbf = vec![0.0f64; nlevel];
|
||||
let wop = vec![0.0f64; nlevel * nd];
|
||||
let ilk = vec![0usize; nlevel];
|
||||
let usum = vec![0.0f64; nlevel];
|
||||
let attot = vec![0.0f64; nlevel];
|
||||
let temp = vec![10000.0f64; nd];
|
||||
let elec = vec![1.0e14f64; nd];
|
||||
|
||||
let params = ChangeParams {
|
||||
levels: &levels,
|
||||
nd,
|
||||
temp: &temp,
|
||||
elec: &elec,
|
||||
popul: &popul,
|
||||
g: &g,
|
||||
enion: &enion,
|
||||
iel: &iel,
|
||||
nnext: &nnext,
|
||||
nlevel,
|
||||
n0a: &n0a,
|
||||
nka: &nka,
|
||||
sbf: &sbf,
|
||||
wop: &wop,
|
||||
ilk: &ilk,
|
||||
usum: &usum,
|
||||
attot: &attot,
|
||||
};
|
||||
|
||||
let output = change(¶ms);
|
||||
|
||||
// iold != 0: 直接复制
|
||||
assert_eq!(output.popul_new[0 * nd + 0], 5.0e12);
|
||||
}
|
||||
}
|
||||
237
src/synspec/math/chckab.rs
Normal file
237
src/synspec/math/chckab.rs
Normal file
@ -0,0 +1,237 @@
|
||||
//! 丰度一致性检查。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `CHCKAB` 函数。
|
||||
//!
|
||||
//! 检查显式原子的输入丰度与从模型大气计算得到的丰度是否一致。
|
||||
//! 如果差异超过 10%，程序将停止。
|
||||
|
||||
use crate::synspec::state::constants::{MATOM, MDEPTH, MLEVEL};
|
||||
|
||||
/// CHCKAB 输入参数。
|
||||
pub struct ChckabParams<'a> {
|
||||
/// 深度点数
|
||||
pub nd: usize,
|
||||
/// 温度数组 (K)
|
||||
pub temp: &'a [f64; MDEPTH],
|
||||
/// 电子密度数组 (cm^-3)
|
||||
pub elec: &'a [f64; MDEPTH],
|
||||
/// 能级 populations
|
||||
pub popul: &'a [[f64; MDEPTH]; MLEVEL],
|
||||
/// 上态求和
|
||||
pub usum: &'a [f64; MLEVEL],
|
||||
/// 原子丰度
|
||||
pub abund: &'a [[f64; MDEPTH]; MATOM],
|
||||
/// 原子数
|
||||
pub natom: usize,
|
||||
/// 参考原子索引 (1-based)
|
||||
pub iatref: usize,
|
||||
/// N0A 数组 - 每个原子的第一个能级索引
|
||||
pub n0a: &'a [i32],
|
||||
/// NKA 数组 - 每个原子的最后一个能级索引
|
||||
pub nka: &'a [i32],
|
||||
/// ILK 数组 - 能级索引
|
||||
pub ilk: &'a [i32],
|
||||
}
|
||||
|
||||
/// CHCKAB 输出结果。
|
||||
#[derive(Default)]
|
||||
pub struct ChckabResult {
|
||||
/// 是否发现不一致性
|
||||
pub inconsistent: bool,
|
||||
/// 不一致的原子数
|
||||
pub n_inconsistent: usize,
|
||||
}
|
||||
|
||||
|
||||
/// 丰度一致性检查。
|
||||
///
|
||||
/// 检查显式原子的输入丰度与从模型大气计算得到的丰度是否一致。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `params` - CHCKAB 参数
|
||||
///
|
||||
/// # 返回值
|
||||
///
|
||||
/// CHCKAB 输出结果
|
||||
pub fn chckab(params: &ChckabParams) -> ChckabResult {
|
||||
let ChckabParams {
|
||||
nd,
|
||||
temp: _,
|
||||
elec,
|
||||
popul,
|
||||
usum,
|
||||
abund,
|
||||
natom,
|
||||
iatref,
|
||||
n0a,
|
||||
nka,
|
||||
ilk,
|
||||
} = *params;
|
||||
|
||||
let mut result = ChckabResult::default();
|
||||
|
||||
// 检查三个深度点: 1, 46, ND
|
||||
let depth_points = [0, 45.min(nd - 1), nd - 1];
|
||||
|
||||
for &id in &depth_points {
|
||||
let ane = elec[id];
|
||||
|
||||
let mut sumiat = [0.0_f64; MATOM];
|
||||
let mut sumpop = [0.0_f64; MATOM];
|
||||
|
||||
// 计算每个原子的总 population
|
||||
for iat in 0..natom {
|
||||
let mut sum = 0.0_f64;
|
||||
let mut sump = 0.0_f64;
|
||||
|
||||
let n0 = n0a[iat] as usize - 1; // 转换为 0-indexed
|
||||
let nk = nka[iat] as usize - 1;
|
||||
|
||||
for i in n0..=nk {
|
||||
let il = ilk[i] as usize;
|
||||
let a = if il > 0 {
|
||||
1.0 + ane * usum[il - 1]
|
||||
} else {
|
||||
1.0
|
||||
};
|
||||
sum += a * popul[i][id];
|
||||
sump += popul[i][id];
|
||||
}
|
||||
|
||||
sumiat[iat] = sum;
|
||||
sumpop[iat] = sump;
|
||||
}
|
||||
|
||||
// 检查丰度一致性
|
||||
let iatref_idx = iatref - 1; // 转换为 0-indexed
|
||||
for iat in 0..natom {
|
||||
let x = sumiat[iat] / sumiat[iatref_idx];
|
||||
let ab = abund[iat][id];
|
||||
|
||||
if ab > 0.0 {
|
||||
let ratio = x / ab;
|
||||
if !(0.9..=1.1).contains(&ratio) {
|
||||
result.n_inconsistent += 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
result.inconsistent = result.n_inconsistent > 0;
|
||||
result
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_chckab_consistent() {
|
||||
// 创建一个简单的测试用例，其中丰度一致
|
||||
let nd = 3;
|
||||
let mut temp = [0.0f64; MDEPTH];
|
||||
let mut elec = [0.0f64; MDEPTH];
|
||||
let mut popul = [[0.0f64; MDEPTH]; MLEVEL];
|
||||
let mut usum = [0.0f64; MLEVEL];
|
||||
let mut abund = [[0.0f64; MDEPTH]; MATOM];
|
||||
let mut n0a = [0i32; MATOM];
|
||||
let mut nka = [0i32; MATOM];
|
||||
let mut ilk = [0i32; MLEVEL];
|
||||
|
||||
// 设置测试值
|
||||
for id in 0..nd {
|
||||
temp[id] = 10000.0;
|
||||
elec[id] = 1.0e14;
|
||||
}
|
||||
|
||||
// 设置一个原子，有 2 个能级
|
||||
let natom = 1;
|
||||
n0a[0] = 1;
|
||||
nka[0] = 2;
|
||||
ilk[0] = 0; // 无上态求和
|
||||
ilk[1] = 0;
|
||||
|
||||
// 设置 populations
|
||||
for id in 0..nd {
|
||||
popul[0][id] = 1.0e10;
|
||||
popul[1][id] = 1.0e9;
|
||||
abund[0][id] = 1.0; // 丰度比值 (相对于参考原子)
|
||||
}
|
||||
|
||||
let params = ChckabParams {
|
||||
nd,
|
||||
temp: &temp,
|
||||
elec: &elec,
|
||||
popul: &popul,
|
||||
usum: &usum,
|
||||
abund: &abund,
|
||||
natom,
|
||||
iatref: 1,
|
||||
n0a: &n0a,
|
||||
nka: &nka,
|
||||
ilk: &ilk,
|
||||
};
|
||||
|
||||
let result = chckab(¶ms);
|
||||
|
||||
// 丰度应该一致
|
||||
assert!(!result.inconsistent);
|
||||
assert_eq!(result.n_inconsistent, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_chckab_inconsistent() {
|
||||
// 创建一个测试用例，其中丰度不一致
|
||||
let nd = 3;
|
||||
let mut temp = [0.0f64; MDEPTH];
|
||||
let mut elec = [0.0f64; MDEPTH];
|
||||
let mut popul = [[0.0f64; MDEPTH]; MLEVEL];
|
||||
let mut usum = [0.0f64; MLEVEL];
|
||||
let mut abund = [[0.0f64; MDEPTH]; MATOM];
|
||||
let mut n0a = [0i32; MATOM];
|
||||
let mut nka = [0i32; MATOM];
|
||||
let mut ilk = [0i32; MLEVEL];
|
||||
|
||||
// 设置测试值
|
||||
for id in 0..nd {
|
||||
temp[id] = 10000.0;
|
||||
elec[id] = 1.0e14;
|
||||
}
|
||||
|
||||
// 设置一个原子，有 2 个能级
|
||||
let natom = 1;
|
||||
n0a[0] = 1;
|
||||
nka[0] = 2;
|
||||
ilk[0] = 0;
|
||||
ilk[1] = 0;
|
||||
|
||||
// 设置 populations
|
||||
for id in 0..nd {
|
||||
popul[0][id] = 1.0e10;
|
||||
popul[1][id] = 1.0e9;
|
||||
abund[0][id] = 1.0e10; // 总 population = 1.1e10，但丰度设为 1.0e10
|
||||
// 比值 = 1.1，超过 10% 阈值
|
||||
}
|
||||
|
||||
let params = ChckabParams {
|
||||
nd,
|
||||
temp: &temp,
|
||||
elec: &elec,
|
||||
popul: &popul,
|
||||
usum: &usum,
|
||||
abund: &abund,
|
||||
natom,
|
||||
iatref: 1,
|
||||
n0a: &n0a,
|
||||
nka: &nka,
|
||||
ilk: &ilk,
|
||||
};
|
||||
|
||||
let result = chckab(¶ms);
|
||||
|
||||
// 丰度应该不一致
|
||||
assert!(result.inconsistent);
|
||||
assert!(result.n_inconsistent > 0);
|
||||
}
|
||||
}
|
||||
562
src/synspec/math/cia.rs
Normal file
562
src/synspec/math/cia.rs
Normal file
@ -0,0 +1,562 @@
|
||||
//! Collision-Induced Absorption (CIA) opacity functions.
|
||||
//!
|
||||
//! Translated from SYNSPEC54 subroutines:
|
||||
//! - `cia_h2h2` -- H2-H2 CIA (Borysow et al. 2001, JQSRT 68, 235)
|
||||
//! - `cia_h2h` -- H2-H CIA (from TURBOSPEC)
|
||||
//! - `cia_h2he` -- H2-He CIA (Jorgensen et al. 2000, A&A 361, 283)
|
||||
//! - `cia_hhe` -- H-He CIA (Gustafsson & Frommhold 2001, ApJ 546, 1168)
|
||||
//!
|
||||
//! Each function reads a CIA table on first call, then performs 2D bilinear
|
||||
//! interpolation in (wavenumber, temperature) space to compute opacity.
|
||||
//!
|
||||
//! # Usage
|
||||
//! 1. Call the `*_init` function once to load the CIA data file.
|
||||
//! 2. Call the corresponding function to evaluate CIA opacity at given conditions.
|
||||
|
||||
use std::fs::File;
|
||||
use std::io::{BufRead, BufReader};
|
||||
use std::sync::OnceLock;
|
||||
|
||||
use super::locate::locate;
|
||||
|
||||
// ============================================================================
|
||||
// Constants
|
||||
// ============================================================================
|
||||
|
||||
/// Speed of light in cm/s
|
||||
const CAS: f64 = 2.997925e10;
|
||||
|
||||
/// Amagat number (Loschmidt number at STP) in cm^-3
|
||||
const AMAGAT: f64 = 2.6867774e19;
|
||||
|
||||
/// Scaling factor: 1 / amagat^2
|
||||
const FAC: f64 = 1.0 / (AMAGAT * AMAGAT);
|
||||
|
||||
/// Fallback value for log(alpha) when outside frequency table range
|
||||
const ALPHA_FLOOR: f64 = -50.0;
|
||||
|
||||
// ============================================================================
|
||||
// CIA table storage
|
||||
// ============================================================================
|
||||
|
||||
/// A loaded CIA table: frequencies (wavenumber in cm^-1), temperatures (K),
|
||||
/// and log(alpha) values.
|
||||
struct CiaTable {
|
||||
nlines: usize,
|
||||
ntemp: usize,
|
||||
freq: Vec<f64>,
|
||||
temp: Vec<f64>,
|
||||
/// log(alpha) values stored as `alpha[i * ntemp + j]` (row-major)
|
||||
alpha: Vec<f64>,
|
||||
}
|
||||
|
||||
// Static storage for each CIA species
|
||||
static TABLE_H2H2: OnceLock<CiaTable> = OnceLock::new();
|
||||
static TABLE_H2H: OnceLock<CiaTable> = OnceLock::new();
|
||||
static TABLE_H2HE: OnceLock<CiaTable> = OnceLock::new();
|
||||
static TABLE_HHE: OnceLock<CiaTable> = OnceLock::new();
|
||||
|
||||
// ============================================================================
|
||||
// Data loading helper
|
||||
// ============================================================================
|
||||
|
||||
/// Load a CIA data file into a `CiaTable`.
|
||||
///
|
||||
/// File format:
|
||||
/// - 3 header lines (skipped)
|
||||
/// - `nlines` data lines, each with: wavenumber alpha(T1) alpha(T2) ... alpha(TnTemp)
|
||||
///
|
||||
/// After reading, all alpha values are replaced by their natural logarithm.
|
||||
fn load_cia_table(filename: &str, nlines: usize, temp: &[f64]) -> Result<CiaTable, String> {
|
||||
let ntemp = temp.len();
|
||||
|
||||
let file = File::open(filename)
|
||||
.map_err(|e| format!("Cannot open CIA data file '{}': {}", filename, e))?;
|
||||
let mut reader = BufReader::new(file);
|
||||
let mut line = String::new();
|
||||
|
||||
// Skip 3 header lines
|
||||
for _ in 0..3 {
|
||||
line.clear();
|
||||
reader
|
||||
.read_line(&mut line)
|
||||
.map_err(|e| format!("Error reading CIA header: {}", e))?;
|
||||
}
|
||||
|
||||
let mut freq = Vec::with_capacity(nlines);
|
||||
let mut alpha = vec![0.0f64; nlines * ntemp];
|
||||
|
||||
for i in 0..nlines {
|
||||
line.clear();
|
||||
reader
|
||||
.read_line(&mut line)
|
||||
.map_err(|e| format!("Error reading CIA data line {}: {}", i + 1, e))?;
|
||||
|
||||
let parts: Vec<f64> = line
|
||||
.split_whitespace()
|
||||
.map(|s| {
|
||||
s.parse::<f64>()
|
||||
.map_err(|_| format!("Cannot parse float from '{}'", s))
|
||||
})
|
||||
.collect::<Result<Vec<f64>, String>>()?;
|
||||
|
||||
if parts.len() < 1 + ntemp {
|
||||
return Err(format!(
|
||||
"CIA data line {}: expected {} fields, got {}",
|
||||
i + 1,
|
||||
1 + ntemp,
|
||||
parts.len()
|
||||
));
|
||||
}
|
||||
|
||||
freq.push(parts[0]);
|
||||
for j in 0..ntemp {
|
||||
alpha[i * ntemp + j] = parts[1 + j].ln();
|
||||
}
|
||||
}
|
||||
|
||||
Ok(CiaTable {
|
||||
nlines,
|
||||
ntemp,
|
||||
freq,
|
||||
temp: temp.to_vec(),
|
||||
alpha,
|
||||
})
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// Core interpolation (shared by all 4 functions)
|
||||
// ============================================================================
|
||||
|
||||
/// Perform 2D bilinear interpolation in (wavenumber, temperature) space.
|
||||
///
|
||||
/// Returns the interpolated alpha value (after exp), or 0.0 if temperature
|
||||
/// is below the table range. Returns `exp(ALPHA_FLOOR)` if frequency is
|
||||
/// outside the table.
|
||||
fn cia_interpolate(table: &CiaTable, t: f64, ff: f64) -> f64 {
|
||||
let f = ff / CAS; // Convert Hz to cm^-1
|
||||
|
||||
// Locate temperature
|
||||
let j = locate(&table.temp, table.ntemp, t);
|
||||
|
||||
if j == 0 {
|
||||
// Temperature below table range
|
||||
eprintln!();
|
||||
eprintln!(
|
||||
"Warning: requested temperature is below {} K",
|
||||
table.temp[0]
|
||||
);
|
||||
eprintln!("CIA opacity set to 0");
|
||||
eprintln!();
|
||||
return 0.0;
|
||||
}
|
||||
|
||||
// Locate frequency
|
||||
let i = locate(&table.freq, table.nlines, f);
|
||||
|
||||
let alp = if j == table.ntemp {
|
||||
// Hold values constant if off high temperature end of table
|
||||
let y1 = table.alpha[(i - 1) * table.ntemp + j - 1];
|
||||
let y2 = table.alpha[i * table.ntemp + j - 1];
|
||||
let tt = (f - table.freq[i - 1]) / (table.freq[i] - table.freq[i - 1]);
|
||||
(1.0 - tt) * y1 + tt * y2
|
||||
} else if i == 0 || i == table.nlines {
|
||||
// Off frequency table: set to very small number
|
||||
ALPHA_FLOOR
|
||||
} else {
|
||||
// Bilinear interpolation within table
|
||||
// locate returns 1-indexed indices, so freq indices are i-1 and i (0-indexed)
|
||||
// In Fortran: alpha(i,j), alpha(i+1,j), alpha(i+1,j+1), alpha(i,j+1)
|
||||
// where i is 1-indexed from locate. In our 0-indexed storage:
|
||||
// alpha[(i-1)*ntemp + (j-1)], alpha[i*ntemp + (j-1)],
|
||||
// alpha[i*ntemp + j], alpha[(i-1)*ntemp + j]
|
||||
let y1 = table.alpha[(i - 1) * table.ntemp + (j - 1)];
|
||||
let y2 = table.alpha[i * table.ntemp + (j - 1)];
|
||||
let y3 = table.alpha[i * table.ntemp + j];
|
||||
let y4 = table.alpha[(i - 1) * table.ntemp + j];
|
||||
|
||||
let tt = (f - table.freq[i - 1]) / (table.freq[i] - table.freq[i - 1]);
|
||||
let uu = (t - table.temp[j - 1]) / (table.temp[j] - table.temp[j - 1]);
|
||||
|
||||
(1.0 - tt) * (1.0 - uu) * y1
|
||||
+ tt * (1.0 - uu) * y2
|
||||
+ tt * uu * y3
|
||||
+ (1.0 - tt) * uu * y4
|
||||
};
|
||||
|
||||
alp.exp()
|
||||
}
|
||||
|
||||
/// Helper: initialize a CIA table into a static OnceLock.
|
||||
fn init_cia_table(
|
||||
static_table: &'static OnceLock<CiaTable>,
|
||||
filename: &str,
|
||||
nlines: usize,
|
||||
temp: &[f64],
|
||||
) -> Result<(), String> {
|
||||
let table = load_cia_table(filename, nlines, temp)?;
|
||||
static_table
|
||||
.set(table)
|
||||
.map_err(|_| "CIA table already initialized".to_string())
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// H2-H2 CIA
|
||||
// ============================================================================
|
||||
|
||||
/// Initialize H2-H2 CIA table from file.
|
||||
///
|
||||
/// Data source: Borysow A., Jorgensen U.G., Fu Y. 2001, JQSRT 68, 235
|
||||
///
|
||||
/// File format: 3 header lines + 1000 data lines with 8 columns
|
||||
/// (wavenumber + 7 temperatures: 1000..7000 K)
|
||||
pub fn cia_h2h2_init(filename: &str) -> Result<(), String> {
|
||||
init_cia_table(
|
||||
&TABLE_H2H2,
|
||||
filename,
|
||||
1000,
|
||||
&[1000.0, 2000.0, 3000.0, 4000.0, 5000.0, 6000.0, 7000.0],
|
||||
)
|
||||
}
|
||||
|
||||
/// H2-H2 CIA opacity.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `t` - Temperature in K
|
||||
/// * `ah2` - H2 number density in cm^-3
|
||||
/// * `ff` - Frequency in Hz
|
||||
///
|
||||
/// # Returns
|
||||
/// CIA opacity (cm^-1)
|
||||
pub fn cia_h2h2(t: f64, ah2: f64, ff: f64) -> f64 {
|
||||
let table = match TABLE_H2H2.get() {
|
||||
Some(t) => t,
|
||||
None => {
|
||||
eprintln!("CIA H2-H2 table not initialized, call cia_h2h2_init first");
|
||||
return 0.0;
|
||||
}
|
||||
};
|
||||
let alp = cia_interpolate(table, t, ff);
|
||||
FAC * ah2 * ah2 * alp
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// H2-H CIA
|
||||
// ============================================================================
|
||||
|
||||
/// Initialize H2-H CIA table from file.
|
||||
///
|
||||
/// Data source: TURBOSPEC
|
||||
///
|
||||
/// File format: 3 header lines + 67 data lines with 5 columns
|
||||
/// (wavenumber + 4 temperatures: 1000, 1500, 2000, 2500 K)
|
||||
pub fn cia_h2h_init(filename: &str) -> Result<(), String> {
|
||||
init_cia_table(
|
||||
&TABLE_H2H,
|
||||
filename,
|
||||
67,
|
||||
&[1000.0, 1500.0, 2000.0, 2500.0],
|
||||
)
|
||||
}
|
||||
|
||||
/// H2-H CIA opacity.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `t` - Temperature in K
|
||||
/// * `ah2` - H2 number density in cm^-3
|
||||
/// * `ah` - H number density in cm^-3
|
||||
/// * `ff` - Frequency in Hz
|
||||
///
|
||||
/// # Returns
|
||||
/// CIA opacity (cm^-1)
|
||||
pub fn cia_h2h(t: f64, ah2: f64, ah: f64, ff: f64) -> f64 {
|
||||
let table = match TABLE_H2H.get() {
|
||||
Some(t) => t,
|
||||
None => {
|
||||
eprintln!("CIA H2-H table not initialized, call cia_h2h_init first");
|
||||
return 0.0;
|
||||
}
|
||||
};
|
||||
let alp = cia_interpolate(table, t, ff);
|
||||
FAC * ah2 * ah * alp
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// H2-He CIA
|
||||
// ============================================================================
|
||||
|
||||
/// Initialize H2-He CIA table from file.
|
||||
///
|
||||
/// Data source: Jorgensen U.G., Hammer D., Borysow A., Falkesgaard J., 2000,
|
||||
/// Astronomy & Astrophysics 361, 283
|
||||
///
|
||||
/// File format: 3 header lines + 242 data lines with 8 columns
|
||||
/// (wavenumber + 7 temperatures: 1000..7000 K)
|
||||
pub fn cia_h2he_init(filename: &str) -> Result<(), String> {
|
||||
init_cia_table(
|
||||
&TABLE_H2HE,
|
||||
filename,
|
||||
242,
|
||||
&[1000.0, 2000.0, 3000.0, 4000.0, 5000.0, 6000.0, 7000.0],
|
||||
)
|
||||
}
|
||||
|
||||
/// H2-He CIA opacity.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `t` - Temperature in K
|
||||
/// * `ah2` - H2 number density in cm^-3
|
||||
/// * `ahe` - He number density in cm^-3
|
||||
/// * `ff` - Frequency in Hz
|
||||
///
|
||||
/// # Returns
|
||||
/// CIA opacity (cm^-1)
|
||||
pub fn cia_h2he(t: f64, ah2: f64, ahe: f64, ff: f64) -> f64 {
|
||||
let table = match TABLE_H2HE.get() {
|
||||
Some(t) => t,
|
||||
None => {
|
||||
eprintln!("CIA H2-He table not initialized, call cia_h2he_init first");
|
||||
return 0.0;
|
||||
}
|
||||
};
|
||||
let alp = cia_interpolate(table, t, ff);
|
||||
FAC * ah2 * ahe * alp
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// H-He CIA
|
||||
// ============================================================================
|
||||
|
||||
/// Initialize H-He CIA table from file.
|
||||
///
|
||||
/// Data source: Gustafsson M., Frommhold, L. 2001, ApJ 546, 1168
|
||||
///
|
||||
/// File format: 3 header lines + 43 data lines with 12 columns
|
||||
/// (wavenumber + 11 temperatures: 1000, 1500, 2250, 3000, 4000, 5000,
|
||||
/// 6000, 7000, 8000, 9000, 10000 K)
|
||||
pub fn cia_hhe_init(filename: &str) -> Result<(), String> {
|
||||
init_cia_table(
|
||||
&TABLE_HHE,
|
||||
filename,
|
||||
43,
|
||||
&[
|
||||
1000.0, 1500.0, 2250.0, 3000.0, 4000.0, 5000.0, 6000.0, 7000.0, 8000.0, 9000.0,
|
||||
10000.0,
|
||||
],
|
||||
)
|
||||
}
|
||||
|
||||
/// H-He CIA opacity.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `t` - Temperature in K
|
||||
/// * `ah` - H number density in cm^-3
|
||||
/// * `ahe` - He number density in cm^-3
|
||||
/// * `ff` - Frequency in Hz
|
||||
///
|
||||
/// # Returns
|
||||
/// CIA opacity (cm^-1)
|
||||
pub fn cia_hhe(t: f64, ah: f64, ahe: f64, ff: f64) -> f64 {
|
||||
let table = match TABLE_HHE.get() {
|
||||
Some(t) => t,
|
||||
None => {
|
||||
eprintln!("CIA H-He table not initialized, call cia_hhe_init first");
|
||||
return 0.0;
|
||||
}
|
||||
};
|
||||
let alp = cia_interpolate(table, t, ff);
|
||||
FAC * ah * ahe * alp
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// Tests
|
||||
// ============================================================================
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
/// Helper: build a small synthetic CIA table for testing interpolation.
|
||||
fn make_test_table() -> CiaTable {
|
||||
// 5 frequency points, 3 temperature points
|
||||
let nlines = 5;
|
||||
let ntemp = 3;
|
||||
let freq = vec![100.0, 200.0, 300.0, 400.0, 500.0];
|
||||
let temp = vec![1000.0, 2000.0, 3000.0];
|
||||
// alpha values (will be stored as ln)
|
||||
// Use a simple pattern: alpha(i,j) = (i+1)*10 + (j+1) (before log)
|
||||
let mut alpha = vec![0.0; nlines * ntemp];
|
||||
for i in 0..nlines {
|
||||
for j in 0..ntemp {
|
||||
let val = (i as f64 + 1.0) * 10.0 + (j as f64 + 1.0);
|
||||
alpha[i * ntemp + j] = val.ln();
|
||||
}
|
||||
}
|
||||
CiaTable {
|
||||
nlines,
|
||||
ntemp,
|
||||
freq,
|
||||
temp,
|
||||
alpha,
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_cia_constants() {
|
||||
assert!((AMAGAT - 2.6867774e19).abs() < 1e10);
|
||||
assert!((CAS - 2.997925e10).abs() < 1e3);
|
||||
let expected_fac = 1.0 / (2.6867774e19_f64 * 2.6867774e19);
|
||||
assert!((FAC - expected_fac).abs() / expected_fac < 1e-12);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_cia_interpolation_basic() {
|
||||
let table = make_test_table();
|
||||
// At freq=200, temp=1500 (midpoint of 1000,2000)
|
||||
// locate(freq,5,200) => i=2, meaning freq[i-1]=200, freq[i]=300 => tt=0
|
||||
// locate(temp,3,1500) => j=1, meaning temp[j-1]=1000, temp[j]=2000 => uu=0.5
|
||||
// y1=alpha[1][0]=ln(21), y4=alpha[1][1]=ln(22)
|
||||
// alp = 0.5*ln(21) + 0.5*ln(22) = ln(sqrt(21*22)) = ln(sqrt(462))
|
||||
let alp = cia_interpolate(&table, 1500.0, 200.0 * CAS);
|
||||
let expected = (462.0_f64).sqrt();
|
||||
assert!(
|
||||
(alp - expected).abs() / expected < 1e-10,
|
||||
"Expected {}, got {}",
|
||||
expected,
|
||||
alp
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_cia_interpolation_corner() {
|
||||
let table = make_test_table();
|
||||
// At exact grid point freq=100, temp=1000
|
||||
// alpha[0][0] = ln(11), exp => 11
|
||||
let alp = cia_interpolate(&table, 1000.0, 100.0 * CAS);
|
||||
assert!(
|
||||
(alp - 11.0).abs() < 1e-10,
|
||||
"Expected 11.0, got {}",
|
||||
alp
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_cia_interpolation_high_temp() {
|
||||
let table = make_test_table();
|
||||
// Temperature above max (3000): hold constant at j=ntemp
|
||||
// freq=250 (midpoint), temp=5000 (above max)
|
||||
// locate(freq,5,250)=2, locate(temp,3,5000)=3=j=ntemp
|
||||
// j==ntemp branch: 1D interpolation in freq at highest temp column
|
||||
// y1=alpha[1][2]=ln(23), y2=alpha[2][2]=ln(33), tt=0.5
|
||||
// alp = 0.5*ln(23) + 0.5*ln(33) = ln(sqrt(23*33)) = ln(sqrt(759))
|
||||
let alp = cia_interpolate(&table, 5000.0, 250.0 * CAS);
|
||||
let expected = (759.0_f64).sqrt();
|
||||
assert!(
|
||||
(alp - expected).abs() / expected < 1e-10,
|
||||
"Expected {}, got {}",
|
||||
expected,
|
||||
alp
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_cia_interpolation_low_temp() {
|
||||
let table = make_test_table();
|
||||
let alp = cia_interpolate(&table, 500.0, 200.0 * CAS);
|
||||
assert_eq!(alp, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_cia_interpolation_low_freq() {
|
||||
let table = make_test_table();
|
||||
let alp = cia_interpolate(&table, 1500.0, 50.0 * CAS);
|
||||
let expected = ALPHA_FLOOR.exp();
|
||||
assert!(
|
||||
(alp - expected).abs() < 1e-20,
|
||||
"Expected ~{}, got {}",
|
||||
expected,
|
||||
alp
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_cia_interpolation_high_freq() {
|
||||
let table = make_test_table();
|
||||
let alp = cia_interpolate(&table, 1500.0, 600.0 * CAS);
|
||||
let expected = ALPHA_FLOOR.exp();
|
||||
assert!(
|
||||
(alp - expected).abs() < 1e-20,
|
||||
"Expected ~{}, got {}",
|
||||
expected,
|
||||
alp
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_cia_h2h2_not_initialized() {
|
||||
let result = cia_h2h2(5000.0, 1e15, 1e14);
|
||||
assert_eq!(result, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_cia_h2h_not_initialized() {
|
||||
let result = cia_h2h(5000.0, 1e15, 1e15, 1e14);
|
||||
assert_eq!(result, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_cia_h2he_not_initialized() {
|
||||
let result = cia_h2he(5000.0, 1e15, 1e15, 1e14);
|
||||
assert_eq!(result, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_cia_hhe_not_initialized() {
|
||||
let result = cia_hhe(5000.0, 1e15, 1e15, 1e14);
|
||||
assert_eq!(result, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_load_cia_table_structure() {
|
||||
let table = make_test_table();
|
||||
assert_eq!(table.nlines, 5);
|
||||
assert_eq!(table.ntemp, 3);
|
||||
assert_eq!(table.freq.len(), 5);
|
||||
assert_eq!(table.temp.len(), 3);
|
||||
assert_eq!(table.alpha.len(), 15);
|
||||
// Verify log was taken: alpha[0] = ln(11)
|
||||
assert!((table.alpha[0] - 11.0_f64.ln()).abs() < 1e-15);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_cia_symmetry_h2h2() {
|
||||
// H2-H2: density product is ah2^2, so doubling ah2 should quadruple result
|
||||
let table = make_test_table();
|
||||
let alp = cia_interpolate(&table, 1500.0, 200.0 * CAS);
|
||||
let opac1 = FAC * 1e15 * 1e15 * alp;
|
||||
let opac2 = FAC * 2e15 * 2e15 * alp;
|
||||
assert!(
|
||||
(opac2 / opac1 - 4.0).abs() < 1e-10,
|
||||
"Expected ratio 4.0, got {}",
|
||||
opac2 / opac1
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_cia_product_scaling() {
|
||||
let table = make_test_table();
|
||||
let alp = cia_interpolate(&table, 2000.0, 300.0 * CAS);
|
||||
let d1 = 1e14;
|
||||
let d2 = 3e14;
|
||||
let opac1 = FAC * d1 * d1 * alp;
|
||||
let opac2 = FAC * d2 * d2 * alp;
|
||||
let ratio = opac2 / opac1;
|
||||
let expected = (d2 / d1) * (d2 / d1);
|
||||
assert!(
|
||||
(ratio - expected).abs() / expected < 1e-10,
|
||||
"Expected ratio {}, got {}",
|
||||
expected,
|
||||
ratio
|
||||
);
|
||||
}
|
||||
}
|
||||
@ -6,9 +6,8 @@
|
||||
//!
|
||||
//! 设置光致电离截面数组，用于辐射转移计算。
|
||||
|
||||
use crate::tlusty::math::{sigk, SigkParams, OpData};
|
||||
use crate::tlusty::math::{sigk, SigkParams};
|
||||
use crate::tlusty::state::atomic::AtomicData;
|
||||
use crate::tlusty::state::constants::{MCROSS, MFREQ};
|
||||
|
||||
// ============================================================================
|
||||
// 常量
|
||||
|
||||
173
src/synspec/math/densit.rs
Normal file
173
src/synspec/math/densit.rs
Normal file
@ -0,0 +1,173 @@
|
||||
//! Determination of state parameters for opacity grid calculations.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine DENSIT(RHO,IDENS)
|
||||
//! at line 22330.
|
||||
//!
|
||||
//! Determines the state parameters (electron density, total particle
|
||||
//! density, populations) for a given depth point using various input modes.
|
||||
|
||||
/// Input mode for density determination.
|
||||
#[derive(Debug, Clone, Copy, PartialEq)]
|
||||
pub enum DensitMode {
|
||||
/// Electron density as input
|
||||
ElectronDensity = 0,
|
||||
/// Total particle density as input (negative)
|
||||
ParticleDensity = -1,
|
||||
/// Mass density as input (mode 1)
|
||||
MassDensity1 = 1,
|
||||
/// Mass density as input (mode 2)
|
||||
MassDensity2 = 2,
|
||||
}
|
||||
|
||||
/// Parameters for DENSIT calculation.
|
||||
pub struct DensitParams {
|
||||
/// Input value (rho, electron density, or particle density)
|
||||
pub rho: f64,
|
||||
/// Input mode
|
||||
pub idens: DensitMode,
|
||||
/// Temperature (K)
|
||||
pub temp: f64,
|
||||
/// Mean molecular weight
|
||||
pub wmm: f64,
|
||||
/// Total hydrogen abundance
|
||||
pub ytot: f64,
|
||||
/// Boltzmann constant (erg/K)
|
||||
pub bolk: f64,
|
||||
/// Hydrogen mass (g)
|
||||
pub hmass: f64,
|
||||
/// Molecular flag
|
||||
pub ifmol: i32,
|
||||
/// Molecular temperature limit
|
||||
pub tmolim: f64,
|
||||
/// Number of levels
|
||||
pub nlevel: usize,
|
||||
/// Standard depth index
|
||||
pub idstd: usize,
|
||||
}
|
||||
|
||||
/// Result of DENSIT calculation.
|
||||
pub struct DensitResult {
|
||||
/// Electron density (cm^-3)
|
||||
pub elec: f64,
|
||||
/// Mass density (g/cm^3)
|
||||
pub dens: f64,
|
||||
/// Total particle density (cm^-3)
|
||||
pub an: f64,
|
||||
}
|
||||
|
||||
/// Determination of state parameters.
|
||||
///
|
||||
/// Determines electron density, mass density, and total particle density
|
||||
/// from the given input value and mode.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Input parameters
|
||||
/// * `todens_fn` - Function to compute AN from (id, t, ane)
|
||||
/// * `eldens_fn` - Function to compute ANE from (id, t, an)
|
||||
/// * `rhonen_fn` - Function to compute (an, ane) from (id, t, rho)
|
||||
///
|
||||
/// # Returns
|
||||
/// Electron density, mass density, and total particle density.
|
||||
pub fn densit<T, E, R>(
|
||||
params: &DensitParams,
|
||||
todens_fn: T,
|
||||
eldens_fn: E,
|
||||
rhonen_fn: R,
|
||||
) -> DensitResult
|
||||
where
|
||||
T: Fn(usize, f64, f64) -> (f64, f64, f64, f64),
|
||||
E: Fn(usize, f64, f64, f64) -> f64,
|
||||
R: Fn(usize, f64, f64) -> (f64, f64),
|
||||
{
|
||||
let id = 0; // Single depth point
|
||||
let t = params.temp;
|
||||
let wmm = params.wmm;
|
||||
|
||||
let (elec, dens, an) = match params.idens {
|
||||
DensitMode::ElectronDensity => {
|
||||
let ane = params.rho;
|
||||
let (an, _anp, _ahtot, _ahmol) = todens_fn(id, t, ane);
|
||||
let dens = (an - ane) * wmm;
|
||||
(ane, dens, an)
|
||||
}
|
||||
DensitMode::ParticleDensity => {
|
||||
let an = params.rho / t / params.bolk;
|
||||
let ane = eldens_fn(id, t, an, 0.0);
|
||||
let dens = wmm * (an - ane);
|
||||
(ane, dens, an)
|
||||
}
|
||||
DensitMode::MassDensity1 => {
|
||||
let rho = params.rho;
|
||||
let (an, ane) = rhonen_fn(id, t, rho);
|
||||
(ane, rho, an)
|
||||
}
|
||||
DensitMode::MassDensity2 => {
|
||||
let rho = params.rho;
|
||||
let (an, ane) = rhonen_fn(id, t, rho);
|
||||
(ane, rho, an)
|
||||
}
|
||||
};
|
||||
|
||||
DensitResult { elec, dens, an }
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_densit_electron_density() {
|
||||
let params = DensitParams {
|
||||
rho: 1e13,
|
||||
idens: DensitMode::ElectronDensity,
|
||||
temp: 10000.0,
|
||||
wmm: 1.0,
|
||||
ytot: 1.0,
|
||||
bolk: 1.380658e-16,
|
||||
hmass: 1.67e-24,
|
||||
ifmol: 0,
|
||||
tmolim: 9000.0,
|
||||
nlevel: 10,
|
||||
idstd: 0,
|
||||
};
|
||||
|
||||
// Mock todens: return (an, anp, ahtot, ahmol)
|
||||
let todens_fn = |_id: usize, _t: f64, ane: f64| {
|
||||
(ane * 1.1, ane * 0.1, ane * 1.0, 0.0)
|
||||
};
|
||||
let eldens_fn = |_id: usize, _t: f64, _an: f64, _ane: f64| 1e13;
|
||||
let rhonen_fn = |_id: usize, _t: f64, _rho: f64| (1e13, 1e12);
|
||||
|
||||
let result = densit(¶ms, todens_fn, eldens_fn, rhonen_fn);
|
||||
assert!(result.elec > 0.0);
|
||||
assert!(result.dens > 0.0);
|
||||
assert!(result.an > 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_densit_particle_density() {
|
||||
let params = DensitParams {
|
||||
rho: 1e13,
|
||||
idens: DensitMode::ParticleDensity,
|
||||
temp: 10000.0,
|
||||
wmm: 1.0,
|
||||
ytot: 1.0,
|
||||
bolk: 1.380658e-16,
|
||||
hmass: 1.67e-24,
|
||||
ifmol: 0,
|
||||
tmolim: 9000.0,
|
||||
nlevel: 10,
|
||||
idstd: 0,
|
||||
};
|
||||
|
||||
let todens_fn = |_id: usize, _t: f64, ane: f64| {
|
||||
(ane * 1.1, ane * 0.1, ane * 1.0, 0.0)
|
||||
};
|
||||
let eldens_fn = |_id: usize, _t: f64, an: f64, _ane: f64| an * 0.1;
|
||||
let rhonen_fn = |_id: usize, _t: f64, _rho: f64| (1e13, 1e12);
|
||||
|
||||
let result = densit(¶ms, todens_fn, eldens_fn, rhonen_fn);
|
||||
assert!(result.elec >= 0.0);
|
||||
assert!(result.an > 0.0);
|
||||
}
|
||||
}
|
||||
114
src/synspec/math/divstr.rs
Normal file
114
src/synspec/math/divstr.rs
Normal file
@ -0,0 +1,114 @@
|
||||
//! Division point between Doppler and asymptotic Stark profiles.
|
||||
//!
|
||||
//! Translated from SYNSPEC `DIVSTR` subroutine (synspec54.f:6840).
|
||||
//!
|
||||
//! Auxiliary procedure for STARKA - determines the division point
|
||||
//! between Doppler and asymptotic Stark profiles.
|
||||
|
||||
/// Compute the division point between Doppler and asymptotic Stark profiles.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `betad` - Doppler width in beta units
|
||||
///
|
||||
/// # Returns
|
||||
/// A tuple `(a, div)` where:
|
||||
/// * `a` = 1.5 * ln(betad) - 1.671
|
||||
/// * `div` - division point (only meaningful for a > 1); solution of
|
||||
/// exp(-(beta/betad)^2) / betad / sqrt(pi) = 3 * beta^(-5/2)
|
||||
pub fn divstr(betad: f64) -> (f64, f64) {
|
||||
const CA: f64 = 1.671;
|
||||
const BL: f64 = 5.821;
|
||||
const AL: f64 = 1.26;
|
||||
const CX: f64 = 0.28;
|
||||
const DX: f64 = 0.0001;
|
||||
|
||||
let a = 1.5 * betad.ln() - CA;
|
||||
if betad < BL {
|
||||
return (a, 0.0);
|
||||
}
|
||||
|
||||
let mut x = if a >= AL {
|
||||
a.sqrt() * (1.0 + 1.25 * a.ln() / (4.0 * a - 5.0))
|
||||
} else {
|
||||
(CX + a).sqrt()
|
||||
};
|
||||
|
||||
for _ in 0..5 {
|
||||
let xn = x * (1.0 - (x * x - 2.5 * x.ln() - a) / (2.0 * x * x - 2.5));
|
||||
if (xn - x).abs() <= DX {
|
||||
x = xn;
|
||||
break;
|
||||
}
|
||||
x = xn;
|
||||
}
|
||||
|
||||
(a, x)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_divstr_small_betad() {
|
||||
// For betad < BL (5.821), div should be 0
|
||||
let (a, div) = divstr(3.0);
|
||||
assert!(a < 0.0); // 1.5*ln(3) - 1.671 ≈ -0.024
|
||||
assert_eq!(div, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_divstr_large_betad() {
|
||||
// For large betad, should compute meaningful division point
|
||||
let (a, div) = divstr(100.0);
|
||||
assert!(a > 1.0);
|
||||
assert!(div > 0.0);
|
||||
assert!(div.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_divstr_boundary() {
|
||||
// At betad = BL, a should be approximately 1.5*ln(5.821) - 1.671
|
||||
let (a, _div) = divstr(5.821);
|
||||
let expected_a = 1.5 * 5.821_f64.ln() - 1.671;
|
||||
assert!((a - expected_a).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_divstr_a_ge_al() {
|
||||
// betad large enough that a >= AL (1.26)
|
||||
let (a, div) = divstr(20.0);
|
||||
assert!(a >= 1.26);
|
||||
assert!(div > 0.0);
|
||||
assert!(div.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_divstr_a_lt_al() {
|
||||
// betad in range where a < AL but betad >= BL
|
||||
let (a, div) = divstr(7.0);
|
||||
// a = 1.5*ln(7) - 1.671 ≈ 1.265 (close to AL boundary)
|
||||
if a < 1.26 {
|
||||
assert!(div > 0.0);
|
||||
}
|
||||
assert!(div.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_divstr_convergence() {
|
||||
// Verify Newton iteration converges for various inputs
|
||||
for betad in [10.0, 50.0, 100.0, 500.0, 1000.0] {
|
||||
let (a, div) = divstr(betad);
|
||||
assert!(div.is_finite(), "div not finite for betad={}", betad);
|
||||
assert!(div > 0.0, "div not positive for betad={}", betad);
|
||||
// Verify the equation: x^2 - 2.5*ln(x) - a ≈ 0
|
||||
let residual = div * div - 2.5 * div.ln() - a;
|
||||
assert!(
|
||||
residual.abs() < 0.01,
|
||||
"residual too large for betad={}: {}",
|
||||
betad,
|
||||
residual
|
||||
);
|
||||
}
|
||||
}
|
||||
}
|
||||
83
src/synspec/math/dwnfr0.rs
Normal file
83
src/synspec/math/dwnfr0.rs
Normal file
@ -0,0 +1,83 @@
|
||||
//! 溶解分数辅助量。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `dwnfr0.f`。
|
||||
|
||||
use crate::synspec::state::constants::{MDEPTH, MZZ};
|
||||
|
||||
/// 溶解分数辅助量。
|
||||
///
|
||||
/// 计算电子密度的幂次和溶解分数系数。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `id` - 深度点索引 (0-based)
|
||||
/// * `elec` - 电子密度数组
|
||||
/// * `temp` - 温度数组
|
||||
/// * `elec23` - 输出: 电子密度的 2/3 次方
|
||||
/// * `z3` - 输出: 电荷的三次方
|
||||
/// * `dwc1` - 输出: 溶解分数系数 1
|
||||
/// * `dwc2` - 输出: 溶解分数系数 2
|
||||
pub fn dwnfr0(
|
||||
id: usize,
|
||||
elec: &[f64; MDEPTH],
|
||||
temp: &[f64; MDEPTH],
|
||||
elec23: &mut [f64; MDEPTH],
|
||||
z3: &mut [f64; MZZ],
|
||||
dwc1: &mut [[f64; MDEPTH]; MZZ],
|
||||
dwc2: &mut [f64; MDEPTH],
|
||||
) {
|
||||
const UN: f64 = 1.0;
|
||||
const SIXTH: f64 = UN / 6.0;
|
||||
const CCOR: f64 = 0.09;
|
||||
const P1: f64 = 0.1402;
|
||||
const P2: f64 = 0.1285;
|
||||
const P3: f64 = UN;
|
||||
const P4: f64 = 3.15;
|
||||
const P5: f64 = 4.0;
|
||||
const F23: f64 = -2.0 / 3.0;
|
||||
|
||||
let ane = elec[id];
|
||||
elec23[id] = (F23 * ane.ln()).exp();
|
||||
let anes = (SIXTH * ane.ln()).exp();
|
||||
let acor = CCOR * anes / temp[id].sqrt();
|
||||
let x = (P4 * (UN + P3 * acor).ln()).exp();
|
||||
dwc2[id] = P2 * x;
|
||||
let a3 = acor * acor * acor;
|
||||
for izz in 0..MZZ {
|
||||
let z = (izz + 1) as f64;
|
||||
z3[izz] = z * z * z;
|
||||
dwc1[izz][id] = P1 * (x + P5 * (z - 1.0) * a3);
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_dwnfr0_basic() {
|
||||
let mut elec = [0.0f64; MDEPTH];
|
||||
let mut temp = [0.0f64; MDEPTH];
|
||||
let mut elec23 = [0.0f64; MDEPTH];
|
||||
let mut z3 = [0.0f64; MZZ];
|
||||
let mut dwc1 = [[0.0f64; MDEPTH]; MZZ];
|
||||
let mut dwc2 = [0.0f64; MDEPTH];
|
||||
|
||||
// 设置测试值
|
||||
elec[0] = 1.0e14;
|
||||
temp[0] = 10000.0;
|
||||
|
||||
dwnfr0(0, &elec, &temp, &mut elec23, &mut z3, &mut dwc1, &mut dwc2);
|
||||
|
||||
// 验证 elec23 = elec^(-2/3)
|
||||
let expected_elec23 = (1.0e14_f64.powf(-2.0 / 3.0));
|
||||
assert!((elec23[0] - expected_elec23).abs() < 1.0e-10);
|
||||
|
||||
// 验证 z3
|
||||
assert!((z3[0] - 1.0).abs() < 1.0e-10); // z=1 -> z3=1
|
||||
assert!((z3[1] - 8.0).abs() < 1.0e-10); // z=2 -> z3=8
|
||||
|
||||
// 验证 dwc2 为正
|
||||
assert!(dwc2[0] > 0.0);
|
||||
}
|
||||
}
|
||||
156
src/synspec/math/dwnfr1.rs
Normal file
156
src/synspec/math/dwnfr1.rs
Normal file
@ -0,0 +1,156 @@
|
||||
//! Dissolved fraction for a given frequency.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine DWNFR1(FR,FR0,ID,IZZ,DW1) at line 22349.
|
||||
//!
|
||||
//! Computes the dissolved fraction for a spectral line at frequency FR
|
||||
//! relative to the series limit frequency FR0, for ionization stage IZZ
|
||||
//! at depth point ID.
|
||||
|
||||
/// Parameters for dissolved fraction calculation.
|
||||
pub struct Dwnfr1Params<'a> {
|
||||
/// Frequency at which to evaluate (Hz)
|
||||
pub fr: f64,
|
||||
/// Series limit frequency (Hz)
|
||||
pub fr0: f64,
|
||||
/// Depth index
|
||||
pub id: usize,
|
||||
/// Ionic charge
|
||||
pub izz: usize,
|
||||
/// Z^3 array for each ionic charge
|
||||
pub z3: &'a [f64],
|
||||
/// Electron density to the 2/3 power at each depth
|
||||
pub elec23: &'a [f64],
|
||||
/// DWC1 parameter (IZZ x depth, row-major)
|
||||
pub dwc1: &'a [f64],
|
||||
/// Number of depth points (for 2D indexing of dwc1)
|
||||
pub ndepth: usize,
|
||||
/// DWC2 parameter at each depth
|
||||
pub dwc2: &'a [f64],
|
||||
/// Bergmann factor (usually 1.0)
|
||||
pub bergfc: f64,
|
||||
}
|
||||
|
||||
/// Dissolved fraction for a given frequency.
|
||||
///
|
||||
/// Computes the dissolved fraction for a spectral line at frequency FR
|
||||
/// relative to the series limit frequency FR0. Returns 1.0 (fully dissolved)
|
||||
/// when FR >= FR0.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Calculation parameters
|
||||
///
|
||||
/// # Returns
|
||||
/// Dissolved fraction (0 to 1)
|
||||
pub fn dwnfr1(params: &Dwnfr1Params) -> f64 {
|
||||
if params.fr < params.fr0 {
|
||||
// Constants
|
||||
let sqfrh = 5.734152e7;
|
||||
let tkn = 3.01;
|
||||
let ckn = 5.33333333;
|
||||
let cb = 8.59e14;
|
||||
|
||||
let izz_f = params.izz as f64;
|
||||
let xn = sqfrh * izz_f / (params.fr0 - params.fr).sqrt();
|
||||
|
||||
let xkn = if xn <= tkn {
|
||||
1.0
|
||||
} else {
|
||||
let xn1 = 1.0 / (xn + 1.0);
|
||||
ckn * xn * xn1 * xn1
|
||||
};
|
||||
|
||||
let beta = cb * params.z3[params.izz] * xkn
|
||||
/ (xn * xn * xn * xn)
|
||||
* params.elec23[params.id]
|
||||
* params.bergfc;
|
||||
let beta3 = beta * beta * beta;
|
||||
let beta32 = beta3.sqrt();
|
||||
|
||||
// DWC1 is 2D: (IZZ, ID) -> dwc1[izz * ndepth + id]
|
||||
let dwc1_val = params.dwc1[params.izz * params.ndepth + params.id];
|
||||
let f = (dwc1_val * beta3) / (1.0 + params.dwc2[params.id] * beta32);
|
||||
|
||||
1.0 - f / (1.0 + f)
|
||||
} else {
|
||||
1.0
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_dwnfr1_above_limit() {
|
||||
// When fr >= fr0, should return 1.0
|
||||
let z3 = vec![1.0, 8.0, 27.0];
|
||||
// ELEC23 = ANE^(-2/3), for ANE=1e13 -> ~2.15e-9
|
||||
let elec23 = vec![2.15e-9; 3];
|
||||
let dwc1 = vec![0.0; 6]; // 2 ionic charges x 3 depths
|
||||
let dwc2 = vec![0.5; 3];
|
||||
|
||||
let params = Dwnfr1Params {
|
||||
fr: 3.3e15,
|
||||
fr0: 3.3e15,
|
||||
id: 0,
|
||||
izz: 1,
|
||||
z3: &z3,
|
||||
elec23: &elec23,
|
||||
dwc1: &dwc1,
|
||||
ndepth: 3,
|
||||
dwc2: &dwc2,
|
||||
bergfc: 1.0,
|
||||
};
|
||||
assert_eq!(dwnfr1(¶ms), 1.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_dwnfr1_below_limit() {
|
||||
// When fr < fr0, should return something < 1.0
|
||||
let z3 = vec![1.0, 8.0, 27.0];
|
||||
let elec23 = vec![2.15e-9; 3];
|
||||
let mut dwc1 = vec![0.0; 6]; // 2 x 3
|
||||
dwc1[1 * 3 + 0] = 0.1; // DWC1(IZZ=1, ID=0)
|
||||
let dwc2 = vec![0.5; 3];
|
||||
|
||||
let params = Dwnfr1Params {
|
||||
fr: 3.0e15,
|
||||
fr0: 3.3e15,
|
||||
id: 0,
|
||||
izz: 1,
|
||||
z3: &z3,
|
||||
elec23: &elec23,
|
||||
dwc1: &dwc1,
|
||||
ndepth: 3,
|
||||
dwc2: &dwc2,
|
||||
bergfc: 1.0,
|
||||
};
|
||||
let result = dwnfr1(¶ms);
|
||||
assert!(result > 0.0 && result <= 1.0, "dwnfr1 out of range: {}", result);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_dwnfr1_well_below_limit() {
|
||||
// Far from limit, dissolved fraction should be close to 1.0
|
||||
let z3 = vec![1.0, 8.0, 27.0];
|
||||
let elec23 = vec![2.15e-9; 3];
|
||||
let mut dwc1 = vec![0.0; 6];
|
||||
dwc1[1 * 3 + 0] = 0.1;
|
||||
let dwc2 = vec![0.5; 3];
|
||||
|
||||
let params = Dwnfr1Params {
|
||||
fr: 3.3e15 - 1.0e4, // very close to limit: xn >> TKN
|
||||
fr0: 3.3e15,
|
||||
id: 0,
|
||||
izz: 1,
|
||||
z3: &z3,
|
||||
elec23: &elec23,
|
||||
dwc1: &dwc1,
|
||||
ndepth: 3,
|
||||
dwc2: &dwc2,
|
||||
bergfc: 1.0,
|
||||
};
|
||||
let result = dwnfr1(¶ms);
|
||||
assert!(result > 0.9, "Expected close to 1.0, got {}", result);
|
||||
}
|
||||
}
|
||||
350
src/synspec/math/eldens.rs
Normal file
350
src/synspec/math/eldens.rs
Normal file
@ -0,0 +1,350 @@
|
||||
//! Electron density calculation by Newton-Raphson method.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine ELDENS (line 22552).
|
||||
//!
|
||||
//! Evaluates the electron density and total hydrogen number density
|
||||
//! for a given total particle number density and temperature by solving
|
||||
//! the set of Saha equations, charge conservation and particle conservation
|
||||
//! equations using a Newton-Raphson method.
|
||||
|
||||
/// Parameters for ELDENS calculation.
|
||||
pub struct EldensParams {
|
||||
/// Depth point index
|
||||
pub id: usize,
|
||||
/// Temperature (K)
|
||||
pub t: f64,
|
||||
/// Total particle number density (cm^-3)
|
||||
pub an: f64,
|
||||
/// Initial electron density estimate (cm^-3), updated on output
|
||||
pub ane: f64,
|
||||
/// Boltzmann constant (erg/K)
|
||||
pub bolk: f64,
|
||||
/// Total hydrogen abundance YTOT
|
||||
pub ytot: f64,
|
||||
/// Reference atom is hydrogen flag
|
||||
pub is_h_ref: bool,
|
||||
/// Molecular flag (>0 to consider molecules)
|
||||
pub ifmol: i32,
|
||||
/// Molecular temperature limit
|
||||
pub tmolim: f64,
|
||||
/// Previous electron density ratio (anerel)
|
||||
pub anerel: f64,
|
||||
/// Standard partition function PFSTD(1,1)
|
||||
pub pfstd_h: f64,
|
||||
}
|
||||
|
||||
/// Result of ELDENS calculation.
|
||||
pub struct EldensResult {
|
||||
/// Electron density (cm^-3)
|
||||
pub ane: f64,
|
||||
/// Proton number density (cm^-3)
|
||||
pub anp: f64,
|
||||
/// Total hydrogen number density (cm^-3)
|
||||
pub ahtot: f64,
|
||||
/// Hydrogen molecule fraction
|
||||
pub ahmol: f64,
|
||||
/// Negative hydrogen ion density
|
||||
pub anhmi: f64,
|
||||
/// Updated electron density ratio
|
||||
pub anerel: f64,
|
||||
/// Mean molecular weight update factor
|
||||
pub wmm_factor: f64,
|
||||
}
|
||||
|
||||
/// Electron density calculation by Newton-Raphson method.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Input parameters
|
||||
/// * `state_fn` - Callback to STATE subroutine: (id, t, ane) -> (q, dqn)
|
||||
/// * `lineqs_fn` - Callback to LINEQS: (a, b, n) -> solution vector
|
||||
/// * `moleq_fn` - Callback to MOLEQ: (id, t, an, aein, mode) -> ane
|
||||
///
|
||||
/// # Returns
|
||||
/// Updated electron density and related quantities.
|
||||
#[allow(unused_assignments)]
|
||||
#[allow(unused_assignments)]
|
||||
#[allow(unused_assignments)]
|
||||
pub fn eldens<S, L, M>(
|
||||
params: &EldensParams,
|
||||
state_fn: S,
|
||||
lineqs_fn: L,
|
||||
moleq_fn: M,
|
||||
) -> EldensResult
|
||||
where
|
||||
S: Fn(usize, f64, f64) -> (f64, f64),
|
||||
L: Fn(&mut [f64], &mut [f64], usize) -> Vec<f64>,
|
||||
M: Fn(usize, f64, f64, f64, i32) -> f64,
|
||||
{
|
||||
let t = params.t;
|
||||
let an = params.an;
|
||||
let mut ane = params.ane;
|
||||
let mut anerel = params.anerel;
|
||||
let bolk = params.bolk;
|
||||
|
||||
// Constants
|
||||
let un = 1.0_f64;
|
||||
let two = 2.0_f64;
|
||||
let half = 0.5_f64;
|
||||
|
||||
// Check molecular regime
|
||||
if params.ifmol > 0 && t < params.tmolim {
|
||||
let aein = an * anerel;
|
||||
let ane_mol = moleq_fn(params.id, t, an, aein, 0);
|
||||
return EldensResult {
|
||||
ane: ane_mol,
|
||||
anp: 0.0,
|
||||
ahtot: 0.0,
|
||||
ahmol: 0.0,
|
||||
anhmi: 0.0,
|
||||
anerel: ane_mol / an,
|
||||
wmm_factor: 1.0,
|
||||
};
|
||||
}
|
||||
|
||||
// Initialize coefficients
|
||||
let mut qm = 0.0_f64;
|
||||
let mut q2 = 0.0_f64;
|
||||
let mut qp = 0.0_f64;
|
||||
let mut q = 0.0_f64;
|
||||
let mut dqn = 0.0_f64;
|
||||
let tk = bolk * t;
|
||||
let thet = 5.0404e3 / t;
|
||||
|
||||
// Hydrogen ionization/dissociation coefficients
|
||||
let (q0, ih2) = if params.is_h_ref {
|
||||
let qm_val = 1.0353e-16 / t / t.sqrt() * (8762.9 / t).exp();
|
||||
let qh0 = ((15.38287 + 1.5 * t.log10() - 13.595 * thet) * std::f64::consts::LN_10).exp();
|
||||
|
||||
let (ih2, qp_val, q2_val) = if t > 16000.0 {
|
||||
(0, 0.0, 0.0)
|
||||
} else {
|
||||
let qp = tk * ((-11.206998 + thet * (2.7942767 + thet * (0.079196803 - 0.024790744 * thet)))
|
||||
* std::f64::consts::LN_10)
|
||||
.exp();
|
||||
let q2 = tk * ((-12.533505 + thet * (4.9251644 + thet * (-0.056191273 + 0.0032687661 * thet)))
|
||||
* std::f64::consts::LN_10)
|
||||
.exp();
|
||||
(1, qp, q2)
|
||||
};
|
||||
qm = qm_val;
|
||||
qp = qp_val;
|
||||
q2 = q2_val;
|
||||
(qh0, ih2)
|
||||
} else {
|
||||
(0.0, 0)
|
||||
};
|
||||
|
||||
// Initial estimate of electron density
|
||||
if anerel <= 0.0 {
|
||||
anerel = if t > 1.0e4 {
|
||||
0.5
|
||||
} else {
|
||||
0.1 // Default if no previous data
|
||||
};
|
||||
}
|
||||
ane = an * anerel;
|
||||
|
||||
// Newton-Raphson loop
|
||||
let mut ah = 0.0_f64;
|
||||
let mut anh = 0.0_f64;
|
||||
let mut it = 0;
|
||||
let mut delne;
|
||||
|
||||
loop {
|
||||
it += 1;
|
||||
|
||||
// Call STATE to get total charge Q and its derivative DQN
|
||||
let (q_val, dqn_val) = state_fn(params.id, t, ane);
|
||||
q = q_val;
|
||||
dqn = dqn_val;
|
||||
|
||||
if params.is_h_ref {
|
||||
let qh = q0 * 2.0 / params.pfstd_h;
|
||||
|
||||
// Auxiliary parameters
|
||||
let g2 = qh / ane;
|
||||
let g3 = qm * ane;
|
||||
let a = un + g2 + g3;
|
||||
let d = g2 - g3;
|
||||
|
||||
if it <= 1 {
|
||||
if ih2 == 0 {
|
||||
let f1 = un / a;
|
||||
let fe = d / a + q;
|
||||
ah = ane / fe;
|
||||
anh = ah * f1;
|
||||
} else {
|
||||
let e = g2 * qp / q2;
|
||||
let b = two * (un + e);
|
||||
let gg = ane * q2;
|
||||
let c1 = b * (gg * b + a * d) - e * a * a;
|
||||
let c2 = a * (two * e + b * q) - d * b;
|
||||
let c3 = -e - b * q;
|
||||
let f1 = ((c2 * c2 - 4.0 * c1 * c3).sqrt() - c2) * half / c1;
|
||||
let fe = f1 * d + e * (un - a * f1) / b + q;
|
||||
ah = ane / fe;
|
||||
anh = ah * f1;
|
||||
}
|
||||
}
|
||||
|
||||
let ae = anh / ane;
|
||||
let gg = ae * qp;
|
||||
let _e = anh * q2;
|
||||
let b = anh * qm;
|
||||
|
||||
// Matrix of linearized system R (3x3) and rhs S
|
||||
let mut r = [0.0_f64; 9];
|
||||
let mut s = [0.0_f64; 3];
|
||||
|
||||
r[0] = params.ytot; // R(1,1)
|
||||
r[1] = -two * (anh * q2 + gg); // R(1,2)
|
||||
r[2] = un; // R(1,3)
|
||||
r[3] = -q; // R(2,1)
|
||||
r[4] = -d - two * gg; // R(2,2)
|
||||
r[5] = un + b + ae * (g2 + gg) - dqn * ah; // R(2,3)
|
||||
r[6] = -un; // R(3,1)
|
||||
r[7] = a + 4.0 * (anh * q2 + gg); // R(3,2)
|
||||
r[8] = b - ae * (g2 + two * gg); // R(3,3)
|
||||
|
||||
s[0] = an - ane - params.ytot * ah + anh * (anh * q2 + gg);
|
||||
s[1] = anh * (d + gg) + q * ah - ane;
|
||||
s[2] = ah - anh * (a + two * (anh * q2 + gg));
|
||||
|
||||
// Solve linear system
|
||||
let p = lineqs_fn(&mut r, &mut s, 3);
|
||||
|
||||
ah += p[0];
|
||||
anh += p[1];
|
||||
delne = p[2];
|
||||
ane += delne;
|
||||
} else {
|
||||
// Hydrogen is not the reference atom
|
||||
if it == 1 {
|
||||
ane = an * half;
|
||||
ah = ane / params.ytot;
|
||||
}
|
||||
|
||||
let mut r = [0.0_f64; 4];
|
||||
let mut s = [0.0_f64; 2];
|
||||
|
||||
r[0] = params.ytot; // R(1,1)
|
||||
r[1] = un; // R(1,2)
|
||||
r[2] = -q; // R(2,1) - using QREF=0 for now
|
||||
r[3] = un - dqn * ah; // R(2,2) - using DQNR=0 for now
|
||||
|
||||
s[0] = an - ane - params.ytot * ah;
|
||||
s[1] = q * ah - ane;
|
||||
|
||||
let p = lineqs_fn(&mut r, &mut s, 2);
|
||||
|
||||
ah += p[0];
|
||||
delne = p[1];
|
||||
ane += delne;
|
||||
}
|
||||
|
||||
// Convergence check
|
||||
if ane <= 0.0 {
|
||||
ane = 1.0e-7 * an;
|
||||
}
|
||||
if (delne / ane).abs() <= 1.0e-6 || it > 20 {
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// Update anerel for subsequent calls
|
||||
anerel = ane / an;
|
||||
let ahtot = ah;
|
||||
|
||||
// Compute hydrogen molecule quantities
|
||||
let (ahmol, anp, anhmi, wmm_factor) = if params.is_h_ref {
|
||||
let qh = q0 * 2.0 / params.pfstd_h;
|
||||
let ahmol = anh * anh * q2;
|
||||
let anp = anh / ane * qh;
|
||||
let anhmi = anh * ane * qm;
|
||||
let anhn = anh + anp + anhmi + 2.0 * ahmol;
|
||||
let wmm_factor = if anhn > 0.0 {
|
||||
1.0 / (1.0 - ahmol / anhn)
|
||||
} else {
|
||||
1.0
|
||||
};
|
||||
(ahmol, anp, anhmi, wmm_factor)
|
||||
} else {
|
||||
(0.0, 0.0, 0.0, 1.0)
|
||||
};
|
||||
|
||||
EldensResult {
|
||||
ane,
|
||||
anp,
|
||||
ahtot,
|
||||
ahmol,
|
||||
anhmi,
|
||||
anerel,
|
||||
wmm_factor,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_eldens_basic() {
|
||||
let params = EldensParams {
|
||||
id: 0,
|
||||
t: 10000.0,
|
||||
an: 1e15,
|
||||
ane: 0.0,
|
||||
bolk: 1.380658e-16,
|
||||
ytot: 1.0,
|
||||
is_h_ref: true,
|
||||
ifmol: 0,
|
||||
tmolim: 9000.0,
|
||||
anerel: 0.1,
|
||||
pfstd_h: 2.0,
|
||||
};
|
||||
|
||||
// Mock state: return (q, dqn)
|
||||
let state_fn = |_id: usize, _t: f64, ane: f64| {
|
||||
(ane * 0.5, 0.5)
|
||||
};
|
||||
|
||||
// Mock lineqs: simple 3x3 solver
|
||||
let lineqs_fn = |a: &mut [f64], b: &mut [f64], n: usize| -> Vec<f64> {
|
||||
// For test, just return small corrections
|
||||
vec![0.0; n]
|
||||
};
|
||||
|
||||
// Mock moleq
|
||||
let moleq_fn = |_id: usize, _t: f64, _an: f64, _aein: f64, _mode: i32| -> f64 {
|
||||
1e14
|
||||
};
|
||||
|
||||
let result = eldens(¶ms, state_fn, lineqs_fn, moleq_fn);
|
||||
assert!(result.ane > 0.0, "ANE should be positive");
|
||||
assert!(result.anerel > 0.0, "ANEREL should be positive");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_eldens_molecular() {
|
||||
let params = EldensParams {
|
||||
id: 0,
|
||||
t: 5000.0, // Below tmolim
|
||||
an: 1e15,
|
||||
ane: 0.0,
|
||||
bolk: 1.380658e-16,
|
||||
ytot: 1.0,
|
||||
is_h_ref: true,
|
||||
ifmol: 1, // Enable molecular
|
||||
tmolim: 9000.0,
|
||||
anerel: 0.1,
|
||||
pfstd_h: 2.0,
|
||||
};
|
||||
|
||||
let state_fn = |_id: usize, _t: f64, _ane: f64| (0.0, 0.0);
|
||||
let lineqs_fn = |_a: &mut [f64], _b: &mut [f64], n: usize| vec![0.0; n];
|
||||
let moleq_fn = |_id: usize, _t: f64, _an: f64, _aein: f64, _mode: i32| 1e14;
|
||||
|
||||
let result = eldens(¶ms, state_fn, lineqs_fn, moleq_fn);
|
||||
assert!(result.ane > 0.0, "Should use moleq in molecular regime");
|
||||
}
|
||||
}
|
||||
327
src/synspec/math/eospri.rs
Normal file
327
src/synspec/math/eospri.rs
Normal file
@ -0,0 +1,327 @@
|
||||
//! EOS parameter output diagnostics.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine EOSPRI (line 22799).
|
||||
//!
|
||||
//! Prints equation of state parameters including atomic, ionic, and molecular
|
||||
//! number densities and partition functions. Also computes H2+ abundance
|
||||
//! and element ratios (He/H, C/H, N/H, O/H).
|
||||
|
||||
/// Molecular indices used for output (20 selected molecules).
|
||||
#[allow(dead_code)]
|
||||
const INSM: [usize; 20] = [2, 3, 4, 5, 6, 7, 8, 12, 17, 25, 29, 30, 32, 34, 122, 126, 134, 179, 198, 214];
|
||||
|
||||
/// Element indices for metals (38 elements).
|
||||
const NELEMX: [usize; 38] = [
|
||||
1, 2, 3, 4, 5, 6, 7, 8, 9,
|
||||
11, 12, 13, 14, 15, 16, 17, 19, 20,
|
||||
21, 22, 23, 24, 25, 26, 28, 29, 32,
|
||||
35, 37, 38, 39, 40, 41, 53, 56, 57, 58, 60,
|
||||
];
|
||||
|
||||
/// H2+ dissociation constant polynomial coefficients (B&C).
|
||||
const AMH2: [f64; 5] = [1.13390e+01, -2.97499e+00, 4.10842e-02, -3.58550e-03, 1.31844e-04];
|
||||
|
||||
/// Parameters for EOSPRI.
|
||||
pub struct EospriParams<'a> {
|
||||
/// Number of depth points
|
||||
pub nd: usize,
|
||||
/// Temperature array (K)
|
||||
pub temp: &'a [f64],
|
||||
/// Electron density array (cm^-3)
|
||||
pub elec: &'a [f64],
|
||||
/// Mass density array (g/cm^3)
|
||||
pub dens: &'a [f64],
|
||||
/// Mean molecular weight array
|
||||
pub wmm: &'a [f64],
|
||||
/// Mean molecular weight for EOS
|
||||
pub wmy: &'a [f64],
|
||||
/// Hydrogen mass
|
||||
pub hmass: f64,
|
||||
/// Total abundance YTOT
|
||||
pub ytot: &'a [f64],
|
||||
/// Abundance by depth: abndd(element, depth) — 1-indexed element
|
||||
pub abndd: &'a [Vec<f64>],
|
||||
/// Molecular flag
|
||||
pub ifmol: i32,
|
||||
/// Molecular temperature limit
|
||||
pub tmolim: f64,
|
||||
/// Number of molecules
|
||||
pub nmolec: usize,
|
||||
/// Molecular names
|
||||
pub cmol: &'a [String],
|
||||
/// EOS flag
|
||||
pub ifeos: i32,
|
||||
/// Number of metals
|
||||
pub nmetal: usize,
|
||||
/// Step for depth loop
|
||||
pub istp: usize,
|
||||
}
|
||||
|
||||
/// Result of EOSPRI computation.
|
||||
pub struct EospriOutput {
|
||||
/// Atomic number densities per depth: anato[element][depth]
|
||||
pub anato: Vec<Vec<f64>>,
|
||||
/// Ionic number densities per depth: anion[element][depth]
|
||||
pub anion: Vec<Vec<f64>>,
|
||||
/// Molecular number densities per depth: anmol[molecule][depth]
|
||||
pub anmol: Vec<Vec<f64>>,
|
||||
/// Atomic partition functions per depth
|
||||
pub pfato: Vec<Vec<f64>>,
|
||||
/// Ionic partition functions per depth
|
||||
pub pfion: Vec<Vec<f64>>,
|
||||
/// Molecular partition functions per depth
|
||||
pub pfmol: Vec<Vec<f64>>,
|
||||
/// H- density per depth
|
||||
pub anhmi_per_depth: Vec<f64>,
|
||||
/// H2 density per depth
|
||||
pub ahmol_per_depth: Vec<f64>,
|
||||
/// H density per depth
|
||||
pub ah_per_depth: Vec<f64>,
|
||||
/// H+ density per depth
|
||||
pub anp_per_depth: Vec<f64>,
|
||||
/// Second ionization per depth: anion2[element][depth]
|
||||
pub anion2: Vec<Vec<f64>>,
|
||||
/// Summary lines per depth
|
||||
pub summary: Vec<String>,
|
||||
}
|
||||
|
||||
/// ELDENS result used by EOSPRI.
|
||||
pub struct EldensSimpleResult {
|
||||
/// Electron density (cm^-3)
|
||||
pub ane: f64,
|
||||
/// Proton number density (cm^-3)
|
||||
pub anp: f64,
|
||||
/// Total hydrogen number density (cm^-3)
|
||||
pub ahtot: f64,
|
||||
/// Hydrogen molecule fraction
|
||||
pub ahmol: f64,
|
||||
/// Negative hydrogen ion density
|
||||
pub anhmi: f64,
|
||||
}
|
||||
|
||||
/// Compute EOS parameters for diagnostics.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Input parameters
|
||||
/// * `eldens_fn` - Callback: (id, t, ann, ane) -> EldensSimpleResult
|
||||
///
|
||||
/// Calls `eldens_fn` iteratively to converge molecular equilibrium,
|
||||
/// then computes element ratios and formatted output.
|
||||
pub fn eospri<E>(params: &EospriParams, eldens_fn: E) -> EospriOutput
|
||||
where
|
||||
E: Fn(usize, f64, f64, f64) -> EldensSimpleResult,
|
||||
{
|
||||
let max_elem = 100;
|
||||
let max_mol = 600;
|
||||
let nd = params.nd;
|
||||
|
||||
let mut anato = vec![vec![0.0_f64; nd]; max_elem];
|
||||
let mut anion = vec![vec![0.0_f64; nd]; max_elem];
|
||||
let mut anmol = vec![vec![0.0_f64; nd]; max_mol];
|
||||
let pfato = vec![vec![0.0_f64; nd]; max_elem];
|
||||
let pfion = vec![vec![0.0_f64; nd]; max_elem];
|
||||
let pfmol = vec![vec![0.0_f64; nd]; max_mol];
|
||||
let mut anion2 = vec![vec![0.0_f64; nd]; 30];
|
||||
let mut anhmi_per_depth = vec![0.0_f64; nd];
|
||||
let mut ahmol_per_depth = vec![0.0_f64; nd];
|
||||
let mut ah_per_depth = vec![0.0_f64; nd];
|
||||
let mut anp_per_depth = vec![0.0_f64; nd];
|
||||
let mut summary = Vec::new();
|
||||
|
||||
let istp = if params.istp == 0 { 1 } else { params.istp };
|
||||
|
||||
for id in (0..nd).step_by(istp) {
|
||||
let t = params.temp[id];
|
||||
let mut ane = params.elec[id];
|
||||
let rho = params.dens[id];
|
||||
let mut ann = rho / params.wmm[id] + ane;
|
||||
|
||||
// Iterative convergence for molecular equilibrium
|
||||
if params.ifmol == 0 || t > params.tmolim {
|
||||
let mut ann0;
|
||||
loop {
|
||||
ann0 = ann;
|
||||
let result = eldens_fn(id, t, ann, ane);
|
||||
ane = result.ane;
|
||||
|
||||
anmol[0][id] = result.anhmi;
|
||||
anmol[1][id] = result.ahmol;
|
||||
anato[0][id] = result.ahtot;
|
||||
anion[0][id] = result.anp;
|
||||
anhmi_per_depth[id] = result.anhmi;
|
||||
ahmol_per_depth[id] = result.ahmol;
|
||||
ah_per_depth[id] = result.ahtot;
|
||||
anp_per_depth[id] = result.anp;
|
||||
|
||||
let hpop = rho / params.wmy[id] / params.hmass;
|
||||
for &j in NELEMX.iter().take(params.nmetal) {
|
||||
if j < max_elem {
|
||||
anato[j][id] *= hpop;
|
||||
anion[j][id] *= hpop;
|
||||
if (2..30).contains(&j) {
|
||||
anion2[j][id] *= hpop;
|
||||
}
|
||||
}
|
||||
}
|
||||
anato[0][id] = result.ahtot;
|
||||
anion[0][id] = result.anp;
|
||||
|
||||
// Update mean molecular weight
|
||||
// wmm(id) = wmy(id) / (ytot(id) - anmol(2,id)/hpop) * hmass
|
||||
let ahmol_hpop = anmol[1][id] / hpop;
|
||||
let new_wmm = if params.ytot[id] - ahmol_hpop > 0.0 {
|
||||
params.wmy[id] / (params.ytot[id] - ahmol_hpop) * params.hmass
|
||||
} else {
|
||||
params.wmm[id]
|
||||
};
|
||||
ann = rho / new_wmm + ane;
|
||||
|
||||
if (ann - ann0) / ann0 <= 1.0e-5 {
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Compute H2+ abundance (B&C polynomial)
|
||||
let te = 5040.0 / t;
|
||||
let mut aplogj = AMH2[4];
|
||||
for k in 0..4 {
|
||||
let km5 = 4 - k;
|
||||
aplogj = aplogj * te + AMH2[km5];
|
||||
}
|
||||
let tk = 1.38054e-16 * t;
|
||||
let ph2 = -aplogj + (anato[0][id] * anion[0][id]).log10() + 2.0 * tk.log10();
|
||||
let _anh2b = 10.0_f64.powf(ph2) / tk;
|
||||
|
||||
// Compute total hydrogen
|
||||
let htot = anato[0][id] + anion[0][id] + anmol[0][id]
|
||||
+ 2.0 * (anmol[1][id] + anmol[2][id])
|
||||
+ anmol[3][id] + anmol[4][id]
|
||||
+ anmol[11][id] + 2.0 * anmol[12][id] + anmol[13][id]
|
||||
+ anmol[14][id]
|
||||
+ anmol[15][id] + anmol[16][id] + anmol[31][id] + anmol[33][id]
|
||||
+ 4.0 * anmol[36][id] + 2.0 * anmol[37][id] + 3.0 * anmol[38][id]
|
||||
+ 2.0 * anmol[39][id] + 3.0 * anmol[40][id] + 2.0 * anmol[56][id]
|
||||
+ anmol[117][id] + anmol[132][id]
|
||||
+ 2.0 * anmol[139][id] + 3.0 * anmol[140][id] + 4.0 * anmol[141][id]
|
||||
+ anmol[147][id] + 2.0 * anmol[148][id] + anmol[221][id];
|
||||
|
||||
// Element ratios relative to H
|
||||
let ahe = if htot > 0.0 { (anato[1][id] + anion[1][id] + anion2[1][id]) / htot } else { 0.0 };
|
||||
let aca = if htot > 0.0 { (anato[5][id] + anion[5][id] + anion2[5][id]) / htot } else { 0.0 };
|
||||
let acm = if htot > 0.0 {
|
||||
(anmol[4][id] + anmol[5][id]
|
||||
+ anmol[6][id] + 2.0 * (anmol[7][id] + 2.0 * anmol[12][id])
|
||||
+ anmol[13][id] + 2.0 * anmol[14][id] + anmol[19][id]
|
||||
+ anmol[36][id] + anmol[37][id] + anmol[38][id]
|
||||
+ anmol[43][id] + anmol[117][id] + anmol[118][id]
|
||||
+ anmol[436][id] + anmol[452][id])
|
||||
/ htot
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
let ana = if htot > 0.0 { (anato[6][id] + anion[6][id] + anion2[6][id]) / htot } else { 0.0 };
|
||||
let anm = if htot > 0.0 {
|
||||
(anmol[6][id] + 2.0 * anmol[8][id] + anmol[10][id]
|
||||
+ anmol[11][id] + anmol[13][id] + anmol[22][id]
|
||||
+ anmol[23][id] + anmol[39][id] + anmol[40][id]
|
||||
+ anmol[108][id] + anmol[151][id] + anmol[346][id]
|
||||
+ anmol[437][id] + anmol[451][id] + anmol[453][id])
|
||||
/ htot
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
let aoa = if htot > 0.0 { (anato[7][id] + anion[7][id] + anion2[7][id]) / htot } else { 0.0 };
|
||||
let aom = if htot > 0.0 {
|
||||
(anmol[2][id] + anmol[3][id]
|
||||
+ anmol[5][id] + 2.0 * anmol[9][id] + anmol[10][id] + anmol[24][id]
|
||||
+ anmol[25][id] + anmol[28][id] + anmol[29][id] + anmol[30][id]
|
||||
+ anmol[34][id] + 2.0 * anmol[43][id] + anmol[48][id] + anmol[50][id]
|
||||
+ anmol[53][id] + 2.0 * anmol[55][id] + anmol[64][id]
|
||||
+ 2.0 * anmol[65][id] + anmol[83][id] + anmol[108][id]
|
||||
+ anmol[112][id] + anmol[114][id] + anmol[117][id]
|
||||
+ anmol[118][id] + anmol[125][id] + anmol[133][id]
|
||||
+ anmol[152][id] + anmol[178][id] + anmol[183][id]
|
||||
+ 2.0 * anmol[184][id] + anmol[199][id] + anmol[215][id]
|
||||
+ anmol[220][id] + 2.0 * anmol[246][id] + anmol[291][id]
|
||||
+ anmol[438][id] + anmol[452][id] + anmol[453][id])
|
||||
/ htot
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
let ac = aca + acm;
|
||||
let an = ana + anm;
|
||||
let ao = aoa + aom;
|
||||
|
||||
// Format summary
|
||||
let line = format!(
|
||||
"EOS: T={:.1} rho={:.3e} N={:.3e} Ne={:.3e} Htot={:.3e} He/H={:.3e} C/H={:.3e} N/H={:.3e} O/H={:.3e}",
|
||||
t, rho, ann, ane, htot, ahe, ac, an, ao
|
||||
);
|
||||
summary.push(line);
|
||||
}
|
||||
|
||||
EospriOutput {
|
||||
anato,
|
||||
anion,
|
||||
anmol,
|
||||
pfato,
|
||||
pfion,
|
||||
pfmol,
|
||||
anhmi_per_depth,
|
||||
ahmol_per_depth,
|
||||
ah_per_depth,
|
||||
anp_per_depth,
|
||||
anion2,
|
||||
summary,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_eospri_constants() {
|
||||
assert_eq!(NELEMX.len(), 38);
|
||||
assert_eq!(INSM.len(), 20);
|
||||
assert_eq!(AMH2.len(), 5);
|
||||
assert_eq!(NELEMX[0], 1);
|
||||
assert_eq!(NELEMX[37], 60);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_eospri_basic() {
|
||||
let nd = 1;
|
||||
let abndd_data: Vec<Vec<f64>> = vec![vec![0.0; nd]; 100];
|
||||
let params = EospriParams {
|
||||
nd,
|
||||
temp: &[10000.0],
|
||||
elec: &[1.0e12],
|
||||
dens: &[1.0e-10],
|
||||
wmm: &[1.0],
|
||||
wmy: &[1.0],
|
||||
hmass: 1.67e-24,
|
||||
ytot: &[1.0],
|
||||
abndd: &abndd_data,
|
||||
ifmol: 0,
|
||||
tmolim: 10000.0,
|
||||
nmolec: 0,
|
||||
cmol: &[],
|
||||
ifeos: 0,
|
||||
nmetal: 0,
|
||||
istp: 1,
|
||||
};
|
||||
// Mock eldens callback
|
||||
let eldens_fn = |_id: usize, _t: f64, ann: f64, ane: f64| -> EldensSimpleResult {
|
||||
EldensSimpleResult {
|
||||
ane,
|
||||
anp: ane * 0.9,
|
||||
ahtot: ann * 0.8,
|
||||
ahmol: 0.0,
|
||||
anhmi: 0.0,
|
||||
}
|
||||
};
|
||||
let _output = eospri(¶ms, eldens_fn);
|
||||
}
|
||||
}
|
||||
193
src/synspec/math/exopf.rs
Normal file
193
src/synspec/math/exopf.rs
Normal file
@ -0,0 +1,193 @@
|
||||
//! EXOMOL partition functions for 32 molecular species.
|
||||
//!
|
||||
//! Translated from SYNSPEC `EXOPF` subroutine (synspec54.f:23664).
|
||||
//!
|
||||
//! Reads tabulated partition function data from `data/EXOMOL/*.pf` files
|
||||
//! on first call, then performs simple lookup or Irwin-based extrapolation.
|
||||
|
||||
use std::sync::Mutex;
|
||||
|
||||
use super::irwpf;
|
||||
|
||||
/// Number of molecular species.
|
||||
const NMOL: usize = 32;
|
||||
|
||||
/// Molecular species filenames (Fortran `character*4`, leading space stripped).
|
||||
const FILPF: [&str; NMOL] = [
|
||||
"AlO", "C2", "CH", "CN", "CO",
|
||||
"CS", "CaH", "CaO", "CrH", "FeH",
|
||||
"H2", "HCl", "HF", "MgH", "MgO",
|
||||
"N2", "NH", "NO", "NS", "NaH",
|
||||
"OH", "PH", "SH", "SiH", "SiO",
|
||||
"SiS", "TiH", "TiO", "VO",
|
||||
"H2O", "H2S", "CO2",
|
||||
];
|
||||
|
||||
/// Number of temperature points per species (before scaling).
|
||||
const NTEMP_RAW: [usize; NMOL] = [
|
||||
9, 10, 8, 3, 9, 3, 3, 8, 3, 10,
|
||||
10, 5, 5, 3, 5, 9, 5, 5, 5, 5,
|
||||
5, 4, 5, 5, 9, 5, 48, 8, 8, 10,
|
||||
3, 5,
|
||||
];
|
||||
|
||||
/// Tsuji molecular indices for each species.
|
||||
const INDTSU: [i32; NMOL] = [
|
||||
134, 8, 5, 7, 6, 20, 34, 179, 198, 214,
|
||||
2, 36, 33, 32, 126, 9, 12, 11, 23, 122,
|
||||
4, 148, 16, 17, 25, 28, 315, 29, 30, 3,
|
||||
57, 44,
|
||||
];
|
||||
|
||||
/// Cached EXOMOL data: partition functions `pf[mol][temp_index]` and
|
||||
/// scaled temperature counts `ntemp[mol]`.
|
||||
struct ExopfData {
|
||||
/// Partition function values: pf[mol * max_ntemp + j]
|
||||
/// Stored flat; max_ntemp = 48000 (48*1000).
|
||||
pf: Vec<f64>,
|
||||
/// Scaled temperature counts per species.
|
||||
ntemp: Vec<usize>,
|
||||
}
|
||||
|
||||
static EXOPF_DATA: Mutex<Option<ExopfData>> = Mutex::new(None);
|
||||
|
||||
/// Compute the file path for a given species.
|
||||
fn species_filename(name: &str) -> String {
|
||||
let trimmed = name.trim();
|
||||
format!("data/EXOMOL/{}.pf", trimmed)
|
||||
}
|
||||
|
||||
/// Read all EXOMOL partition function files.
|
||||
fn read_exopf_data(data_dir: &str) -> Result<ExopfData, String> {
|
||||
// Scale ntemp: multiply by 1000, except species 27 (TiH) divide by 10
|
||||
let mut ntemp = Vec::with_capacity(NMOL);
|
||||
for i in 0..NMOL {
|
||||
let mut nt = NTEMP_RAW[i] * 1000;
|
||||
if i == 26 {
|
||||
// TiH: ntemp(27) in Fortran (1-indexed) = index 26
|
||||
nt /= 10;
|
||||
}
|
||||
ntemp.push(nt);
|
||||
}
|
||||
|
||||
let max_ntemp = *ntemp.iter().max().unwrap_or(&0);
|
||||
let mut pf = vec![0.0f64; NMOL * max_ntemp];
|
||||
|
||||
for i in 0..NMOL {
|
||||
let filename = format!("{}/{}", data_dir, species_filename(FILPF[i]));
|
||||
let content = match std::fs::read_to_string(&filename) {
|
||||
Ok(c) => c,
|
||||
Err(_) => continue, // Skip missing files
|
||||
};
|
||||
for (j, line) in content.lines().enumerate() {
|
||||
if j >= ntemp[i] {
|
||||
break;
|
||||
}
|
||||
let parts: Vec<&str> = line.split_whitespace().collect();
|
||||
if parts.len() >= 2
|
||||
&& let Ok(val) = parts[1].parse::<f64>() {
|
||||
pf[i * max_ntemp + j] = val;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Ok(ExopfData { pf, ntemp })
|
||||
}
|
||||
|
||||
/// EXOMOL partition function lookup.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `indmol` - Tsuji molecular index.
|
||||
/// * `t` - Temperature (K).
|
||||
/// * `data_dir` - Path to data directory containing `data/EXOMOL/*.pf` files.
|
||||
///
|
||||
/// # Returns
|
||||
/// Partition function value (0.0 if species not found).
|
||||
pub fn exopf(indmol: i32, t: f64, data_dir: &str) -> Result<f64, String> {
|
||||
// Initialize data on first call
|
||||
{
|
||||
let mut guard = EXOPF_DATA
|
||||
.lock()
|
||||
.map_err(|e| format!("Lock error: {}", e))?;
|
||||
if guard.is_none() {
|
||||
*guard = Some(read_exopf_data(data_dir)?);
|
||||
}
|
||||
}
|
||||
let guard = EXOPF_DATA
|
||||
.lock()
|
||||
.map_err(|e| format!("Lock error: {}", e))?;
|
||||
let data = guard.as_ref().unwrap();
|
||||
|
||||
// Find species index
|
||||
let ie = INDTSU.iter().position(|&x| x == indmol);
|
||||
let ie = match ie {
|
||||
Some(idx) => idx,
|
||||
None => return Ok(0.0),
|
||||
};
|
||||
|
||||
let tmax = data.ntemp[ie] as f64;
|
||||
let max_ntemp = *data.ntemp.iter().max().unwrap_or(&1);
|
||||
|
||||
if t <= tmax {
|
||||
// Direct lookup
|
||||
let j = t as usize;
|
||||
if j > 0 && j <= data.ntemp[ie] {
|
||||
Ok(data.pf[ie * max_ntemp + j - 1])
|
||||
} else {
|
||||
Ok(0.0)
|
||||
}
|
||||
} else {
|
||||
// Extrapolate using Irwin partition functions
|
||||
// Need to drop the lock before calling irwpf (which also locks)
|
||||
drop(guard);
|
||||
|
||||
let umx = irwpf::irwpf(0, 0, indmol, tmax, data_dir, 1).unwrap_or(1.0);
|
||||
let uirw = irwpf::irwpf(0, 0, indmol, t, data_dir, 1).unwrap_or(1.0);
|
||||
|
||||
// Re-acquire lock to read pf value
|
||||
let guard = EXOPF_DATA
|
||||
.lock()
|
||||
.map_err(|e| format!("Lock error: {}", e))?;
|
||||
let data = guard.as_ref().unwrap();
|
||||
|
||||
if umx.abs() > 1e-30 {
|
||||
Ok(data.pf[ie * max_ntemp + data.ntemp[ie] - 1] / umx * uirw)
|
||||
} else {
|
||||
Ok(0.0)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_species_count() {
|
||||
assert_eq!(FILPF.len(), NMOL);
|
||||
assert_eq!(NTEMP_RAW.len(), NMOL);
|
||||
assert_eq!(INDTSU.len(), NMOL);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_ntemp_scaling() {
|
||||
// TiH (index 26): 48*1000/10 = 4800
|
||||
assert_eq!(NTEMP_RAW[26] * 1000 / 10, 4800);
|
||||
// Others: raw * 1000
|
||||
assert_eq!(NTEMP_RAW[0] * 1000, 9000);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_species_filename() {
|
||||
assert_eq!(species_filename("AlO"), "data/EXOMOL/AlO.pf");
|
||||
assert_eq!(species_filename(" H2"), "data/EXOMOL/H2.pf");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_indtsu_mapping() {
|
||||
// H2O is last entry
|
||||
assert_eq!(INDTSU[29], 3);
|
||||
// CO is index 4
|
||||
assert_eq!(INDTSU[4], 6);
|
||||
}
|
||||
}
|
||||
79
src/synspec/math/expint.rs
Normal file
79
src/synspec/math/expint.rs
Normal file
@ -0,0 +1,79 @@
|
||||
//! 第一指数积分函数 E1(x)。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `expint.f`
|
||||
|
||||
/// 第一指数积分函数 E1(x)。
|
||||
///
|
||||
/// 使用有理逼近公式，分 x <= 1 和 x > 1 两段。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `x` - 自变量
|
||||
///
|
||||
/// # 返回值
|
||||
///
|
||||
/// E1(x) 的近似值
|
||||
pub fn expint(x: f64) -> f64 {
|
||||
if x <= 1.0 {
|
||||
// x <= 1 的有理逼近
|
||||
-x.ln() - 0.57721566
|
||||
+ x * (0.99999193
|
||||
+ x * (-0.24991055
|
||||
+ x * (0.05519968
|
||||
+ x * (-0.00976004 + x * 0.00107857))))
|
||||
} else {
|
||||
// x > 1 的有理逼近
|
||||
(-x).exp() * ((0.2677734343
|
||||
+ x * (8.6347608925
|
||||
+ x * (18.059016973 + x * (8.5733287401 + x))))
|
||||
/ (3.9584969228
|
||||
+ x * (21.0996530827
|
||||
+ x * (25.6329561486 + x * (9.5733223454 + x)))))
|
||||
/ x
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use approx::assert_relative_eq;
|
||||
|
||||
#[test]
|
||||
fn test_expint_small_x() {
|
||||
// E1(0.5) ≈ 0.5598 (参考值)
|
||||
let result = expint(0.5);
|
||||
assert!(result.is_finite());
|
||||
assert!(result > 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_expint_x_eq_1() {
|
||||
// E1(1) ≈ 0.21938
|
||||
let result = expint(1.0);
|
||||
assert_relative_eq!(result, 0.21938, epsilon = 1e-4);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_expint_large_x() {
|
||||
// E1(5) ≈ 0.001148
|
||||
let result = expint(5.0);
|
||||
assert_relative_eq!(result, 0.0011483, epsilon = 1e-3);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_expint_very_small_x() {
|
||||
// x -> 0+ 时 E1(x) -> +∞
|
||||
let result = expint(0.01);
|
||||
assert!(result > 0.0);
|
||||
assert!(result.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_expint_large_x_decay() {
|
||||
// x 大时 E1(x) ~ exp(-x)/x
|
||||
let x = 10.0;
|
||||
let result = expint(x);
|
||||
let approx = (-x).exp() / x;
|
||||
assert_relative_eq!(result, approx, epsilon = 0.1);
|
||||
}
|
||||
}
|
||||
@ -50,7 +50,7 @@ pub fn extprf(dlam: f64, it: usize, iline: usize, anel: f64, dlast: f64, plast:
|
||||
|
||||
// WE = W0 * 10^anel * 1e-16
|
||||
// Fortran: EXP(ANEL*2.3025851) = 10^ANEL (因为 ln(10) ≈ 2.3025851)
|
||||
let we = w0_val * (anel * 2.3025851_f64).exp() * 1e-16;
|
||||
let we = w0_val * (anel * std::f64::consts::LN_10).exp() * 1e-16;
|
||||
|
||||
// 使用 PI 的精确值
|
||||
const PI: f64 = std::f64::consts::PI;
|
||||
|
||||
484
src/synspec/math/fingrd.rs
Normal file
484
src/synspec/math/fingrd.rs
Normal file
@ -0,0 +1,484 @@
|
||||
//! fingrd — 存储完整的插值不透明度表。
|
||||
//!
|
||||
//! Fortran 原始签名: SUBROUTINE FINGRD
|
||||
//!
|
||||
//! 将计算的不透明度表写入文件（文本和二进制格式）。
|
||||
//!
|
||||
//! 注意: Fortran 版本直接操作文件 I/O 和 COMMON 块。
|
||||
//! Rust 版本提供纯计算核心函数和编排函数。
|
||||
|
||||
use std::io::{BufWriter, Write};
|
||||
use std::fs::File;
|
||||
|
||||
/// 光速 (cm/s)
|
||||
#[allow(dead_code)]
|
||||
const CL: f64 = 2.997925e10;
|
||||
|
||||
/// 波长 (nm) 转换为频率 (s^-1)
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// 2.997925e18/wlgrid(k)
|
||||
/// ```
|
||||
pub fn wavelength_to_frequency(wavelength_nm: f64) -> f64 {
|
||||
2.997925e18 / wavelength_nm
|
||||
}
|
||||
|
||||
/// 频率 (s^-1) 转换为波长 (nm)
|
||||
pub fn frequency_to_wavelength(freq: f64) -> f64 {
|
||||
2.997925e18 / freq
|
||||
}
|
||||
|
||||
/// 对数网格生成
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// wl1=log(wlam1)
|
||||
/// wl2=log(wlam2)
|
||||
/// dwl=(wl2-wl1)/(nfgrid-1)
|
||||
/// do i=1,nfgrid
|
||||
/// wlgrid(i)=exp(wl1+(i-1)*dwl)
|
||||
/// end do
|
||||
/// ```
|
||||
pub fn generate_log_grid(wlam1: f64, wlam2: f64, n: usize) -> Vec<f64> {
|
||||
let wl1 = wlam1.ln();
|
||||
let wl2 = wlam2.ln();
|
||||
let dwl = (wl2 - wl1) / (n - 1) as f64;
|
||||
|
||||
(0..n)
|
||||
.map(|i| (wl1 + i as f64 * dwl).exp())
|
||||
.collect()
|
||||
}
|
||||
|
||||
/// 线性网格生成
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// at1=log(temp1)
|
||||
/// at2=log(temp2)
|
||||
/// dt=(at2-at1)/(ntemp-1)
|
||||
/// do i=1,ntemp
|
||||
/// tempg(i)=exp(at1+(i-1)*dt)
|
||||
/// end do
|
||||
/// ```
|
||||
pub fn generate_linear_grid_in_log(val1: f64, val2: f64, n: usize) -> Vec<f64> {
|
||||
let at1 = val1.ln();
|
||||
let at2 = val2.ln();
|
||||
let dt = if n > 1 { (at2 - at1) / (n - 1) as f64 } else { 0.0 };
|
||||
|
||||
(0..n)
|
||||
.map(|i| (at1 + i as f64 * dt).exp())
|
||||
.collect()
|
||||
}
|
||||
|
||||
/// 不透明度表数据结构
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct OpacityTable {
|
||||
/// 温度网格 (K)
|
||||
pub temperatures: Vec<f64>,
|
||||
/// 密度网格 (g/cm^3)
|
||||
pub densities: Vec<Vec<f64>>,
|
||||
/// 电子密度网格 (g/cm^3)
|
||||
pub electron_densities: Vec<Vec<f64>>,
|
||||
/// 波长网格 (nm)
|
||||
pub wavelengths: Vec<f64>,
|
||||
/// 不透明度表 [temp_idx][dens_idx][freq_idx]
|
||||
pub opacity: Vec<Vec<Vec<f32>>>,
|
||||
}
|
||||
|
||||
/// 计算不透明度表的统计信息
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct OpacityTableStats {
|
||||
/// 最小不透明度
|
||||
pub min_opacity: f32,
|
||||
/// 最大不透明度
|
||||
pub max_opacity: f32,
|
||||
/// 平均不透明度
|
||||
pub mean_opacity: f32,
|
||||
/// 非零元素百分比
|
||||
pub nonzero_percent: f64,
|
||||
}
|
||||
|
||||
/// 计算不透明度表统计信息
|
||||
pub fn compute_opacity_stats(table: &OpacityTable) -> OpacityTableStats {
|
||||
let mut min_op = f32::MAX;
|
||||
let mut max_op = f32::MIN;
|
||||
let mut sum = 0.0_f64;
|
||||
let mut count = 0;
|
||||
let mut nonzero = 0;
|
||||
|
||||
for temp_data in &table.opacity {
|
||||
for dens_data in temp_data {
|
||||
for &op in dens_data {
|
||||
count += 1;
|
||||
sum += op as f64;
|
||||
if op > 0.0 {
|
||||
nonzero += 1;
|
||||
}
|
||||
if op < min_op {
|
||||
min_op = op;
|
||||
}
|
||||
if op > max_op {
|
||||
max_op = op;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
OpacityTableStats {
|
||||
min_opacity: min_op,
|
||||
max_opacity: max_op,
|
||||
mean_opacity: if count > 0 { (sum / count as f64) as f32 } else { 0.0 },
|
||||
nonzero_percent: if count > 0 { 100.0 * nonzero as f64 / count as f64 } else { 0.0 },
|
||||
}
|
||||
}
|
||||
|
||||
/// H- 不透明度标志
|
||||
#[derive(Debug, Clone)]
|
||||
#[derive(Default)]
|
||||
pub struct OpacityFlags {
|
||||
/// H- 光电离
|
||||
pub h_minus: bool,
|
||||
/// H2+ 光电离
|
||||
pub h2_plus: bool,
|
||||
/// He- 光电离
|
||||
pub he_minus: bool,
|
||||
/// CH 不透明度
|
||||
pub ch: bool,
|
||||
/// OH 不透明度
|
||||
pub oh: bool,
|
||||
/// H2- 不透明度
|
||||
pub h2_minus: bool,
|
||||
/// CIA H2-H2
|
||||
pub cia_h2h2: bool,
|
||||
/// CIA H2-He
|
||||
pub cia_h2he: bool,
|
||||
/// CIA H2-H
|
||||
pub cia_h2h: bool,
|
||||
/// CIA H-He
|
||||
pub cia_hhe: bool,
|
||||
}
|
||||
|
||||
|
||||
/// 不透明度表写入参数
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct FingrdParams<'a> {
|
||||
/// 温度网格 (K)
|
||||
pub temperatures: &'a [f64],
|
||||
/// 密度网格 [temp_idx][dens_idx] (g/cm^3)
|
||||
pub densities: &'a [Vec<f64>],
|
||||
/// 电子密度网格 [temp_idx][dens_idx] (g/cm^3)
|
||||
pub electron_densities: &'a [Vec<f64>],
|
||||
/// 波长网格 (nm)
|
||||
pub wavelengths: &'a [f64],
|
||||
/// 不透明度表 [temp_idx][dens_idx][freq_idx] (f32)
|
||||
pub absgrd: &'a [Vec<Vec<f32>>],
|
||||
/// 每温度点的密度数 nden(temp_idx)
|
||||
pub nden: &'a [usize],
|
||||
/// 元素丰度 abnd(92)
|
||||
pub abundances: &'a [f64],
|
||||
/// 相对丰度 relabn(92)
|
||||
pub rel_abundances: &'a [f64],
|
||||
/// 不透明度标志
|
||||
pub flags: &'a OpacityFlags,
|
||||
/// 分子开关 ifmol
|
||||
pub ifmol: i32,
|
||||
/// 分子温度极限 tmolim
|
||||
pub tmolim: f64,
|
||||
/// 输出表文件名
|
||||
pub tabname: &'a str,
|
||||
/// 二进制输出标志 (0=text+binary, 1=binary only)
|
||||
pub ibingr: i32,
|
||||
/// 密度类型 (<10: uniform, >=10: variable)
|
||||
pub idens: i32,
|
||||
}
|
||||
|
||||
/// 编排函数: 将不透明度表写入文本和二进制文件。
|
||||
///
|
||||
/// Fortran 原始逻辑: SUBROUTINE FINGRD
|
||||
/// - 文本输出到 tabname 文件 (Fortran unit 53)
|
||||
/// - 二进制输出到 unit 63
|
||||
pub fn fingrd(params: &FingrdParams) -> Result<(), String> {
|
||||
let ntemp = params.temperatures.len();
|
||||
let nfgrid = params.wavelengths.len();
|
||||
if ntemp == 0 || nfgrid == 0 {
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
let nden0 = params.nden.first().copied().unwrap_or(1);
|
||||
|
||||
// --- 文本输出 (ibingr == 0) ---
|
||||
if params.ibingr == 0 {
|
||||
let file = File::create(params.tabname)
|
||||
.map_err(|e| format!("Cannot create {}: {}", params.tabname, e))?;
|
||||
let mut w = BufWriter::new(file);
|
||||
|
||||
// Header: element abundances
|
||||
writeln!(w, "opacity table with element abundances:").map_err(|e| e.to_string())?;
|
||||
writeln!(w, "element for EOS for opacities").map_err(|e| e.to_string())?;
|
||||
for iat in 0..92 {
|
||||
let abnd = params.abundances.get(iat).copied().unwrap_or(0.0);
|
||||
let rel = params.rel_abundances.get(iat).copied().unwrap_or(0.0);
|
||||
writeln!(w, " {:4} {:12.3e} {:12.3e}", iat + 1, abnd, abnd * rel)
|
||||
.map_err(|e| e.to_string())?;
|
||||
}
|
||||
|
||||
// Molecule info
|
||||
writeln!(w).map_err(|e| e.to_string())?;
|
||||
writeln!(w, "molecules - ifmol,tmolim:").map_err(|e| e.to_string())?;
|
||||
writeln!(w, "{:4}{:10.1}", params.ifmol, params.tmolim).map_err(|e| e.to_string())?;
|
||||
|
||||
// Opacity flags
|
||||
writeln!(w, "additional opacities").map_err(|e| e.to_string())?;
|
||||
writeln!(w, " H- H2+ He- CH OH H2- CIA: H2H2 H2He H2H HHe").map_err(|e| e.to_string())?;
|
||||
let f = params.flags;
|
||||
writeln!(w, "{:4}{:4}{:4}{:4}{:4}{:4} {:4}{:4}{:4}{:4}",
|
||||
f.h_minus as i32, f.h2_plus as i32, f.he_minus as i32,
|
||||
f.ch as i32, f.oh as i32, f.h2_minus as i32,
|
||||
f.cia_h2h2 as i32, f.cia_h2he as i32, f.cia_h2h as i32, f.cia_hhe as i32)
|
||||
.map_err(|e| e.to_string())?;
|
||||
|
||||
if params.idens < 10 {
|
||||
// Uniform density grid
|
||||
let ndens = nden0;
|
||||
writeln!(w).map_err(|e| e.to_string())?;
|
||||
writeln!(w, "number of frequencies, temperatures, densities:").map_err(|e| e.to_string())?;
|
||||
writeln!(w, " {:10}{:10}{:10}", nfgrid, ntemp, ndens).map_err(|e| e.to_string())?;
|
||||
|
||||
// Log temperatures
|
||||
write!(w, "log temperatures").map_err(|e| e.to_string())?;
|
||||
for i in 0..ntemp {
|
||||
if i % 6 == 0 { writeln!(w).map_err(|e| e.to_string())?; }
|
||||
write!(w, "{:11.6}", params.temperatures[i].ln()).map_err(|e| e.to_string())?;
|
||||
}
|
||||
writeln!(w).map_err(|e| e.to_string())?;
|
||||
|
||||
// Log densities
|
||||
write!(w, "log densities").map_err(|e| e.to_string())?;
|
||||
for j in 0..ndens {
|
||||
if j % 6 == 0 { writeln!(w).map_err(|e| e.to_string())?; }
|
||||
let d = params.densities[0].get(j).copied().unwrap_or(1.0);
|
||||
write!(w, "{:11.6}", d.ln()).map_err(|e| e.to_string())?;
|
||||
}
|
||||
writeln!(w).map_err(|e| e.to_string())?;
|
||||
|
||||
// Log electron densities
|
||||
write!(w, "log electron densities from EOS").map_err(|e| e.to_string())?;
|
||||
for i in 0..ntemp {
|
||||
for j in 0..ndens {
|
||||
if (i * ndens + j) % 6 == 0 { writeln!(w).map_err(|e| e.to_string())?; }
|
||||
let e = params.electron_densities[i].get(j).copied().unwrap_or(1.0);
|
||||
write!(w, "{:11.6}", e.ln()).map_err(|e| e.to_string())?;
|
||||
}
|
||||
}
|
||||
writeln!(w).map_err(|e| e.to_string())?;
|
||||
|
||||
// Opacity table
|
||||
for k in 0..nfgrid {
|
||||
writeln!(w).map_err(|e| e.to_string())?;
|
||||
writeln!(w, " *** frequency # : {:8}{:15.5}", k + 1, params.wavelengths[k])
|
||||
.map_err(|e| e.to_string())?;
|
||||
let freq = 2.997925e18 / params.wavelengths[k];
|
||||
writeln!(w, "{:20.8e}", freq).map_err(|e| e.to_string())?;
|
||||
for j in 0..ndens {
|
||||
for i in 0..ntemp {
|
||||
if i % 6 == 0 { writeln!(w).map_err(|e| e.to_string())?; }
|
||||
let val = params.absgrd[i][j].get(k).copied().unwrap_or(0.0);
|
||||
write!(w, "{:14.6e}", val).map_err(|e| e.to_string())?;
|
||||
}
|
||||
writeln!(w).map_err(|e| e.to_string())?;
|
||||
}
|
||||
}
|
||||
} else {
|
||||
// Variable density grid
|
||||
writeln!(w).map_err(|e| e.to_string())?;
|
||||
writeln!(w, "number of frequencies, temperatures, densities:").map_err(|e| e.to_string())?;
|
||||
writeln!(w, " {:10}{:10}{:10}", nfgrid, ntemp, -(nden0 as i32)).map_err(|e| e.to_string())?;
|
||||
|
||||
// nden per temperature
|
||||
for i in 0..ntemp {
|
||||
write!(w, "{:3}", params.nden.get(i).copied().unwrap_or(0)).map_err(|e| e.to_string())?;
|
||||
}
|
||||
writeln!(w).map_err(|e| e.to_string())?;
|
||||
|
||||
// Log temperatures
|
||||
write!(w, "log temperatures").map_err(|e| e.to_string())?;
|
||||
for i in 0..ntemp {
|
||||
if i % 6 == 0 { writeln!(w).map_err(|e| e.to_string())?; }
|
||||
write!(w, "{:11.6}", params.temperatures[i].ln()).map_err(|e| e.to_string())?;
|
||||
}
|
||||
writeln!(w).map_err(|e| e.to_string())?;
|
||||
|
||||
// Log densities per temperature
|
||||
writeln!(w, "log densities").map_err(|e| e.to_string())?;
|
||||
for i in 0..ntemp {
|
||||
let nd = params.nden.get(i).copied().unwrap_or(0);
|
||||
for j in 0..nd {
|
||||
if j % 6 == 0 && j > 0 { writeln!(w).map_err(|e| e.to_string())?; }
|
||||
let d = params.densities[i].get(j).copied().unwrap_or(1.0);
|
||||
write!(w, "{:14.6}", d.ln()).map_err(|e| e.to_string())?;
|
||||
}
|
||||
writeln!(w).map_err(|e| e.to_string())?;
|
||||
}
|
||||
|
||||
// Log electron densities per temperature
|
||||
writeln!(w, "log electron densities from EOS").map_err(|e| e.to_string())?;
|
||||
for i in 0..ntemp {
|
||||
let nd = params.nden.get(i).copied().unwrap_or(0);
|
||||
for j in 0..nd {
|
||||
if j % 6 == 0 && j > 0 { writeln!(w).map_err(|e| e.to_string())?; }
|
||||
let e = params.electron_densities[i].get(j).copied().unwrap_or(1.0);
|
||||
write!(w, "{:14.6}", e.ln()).map_err(|e| e.to_string())?;
|
||||
}
|
||||
writeln!(w).map_err(|e| e.to_string())?;
|
||||
}
|
||||
|
||||
// Opacity table
|
||||
for k in 0..nfgrid {
|
||||
writeln!(w).map_err(|e| e.to_string())?;
|
||||
writeln!(w, " *** frequency # : {:8}{:15.5}", k + 1, params.wavelengths[k])
|
||||
.map_err(|e| e.to_string())?;
|
||||
let freq = 2.997925e18 / params.wavelengths[k];
|
||||
writeln!(w, "{:20.8e}", freq).map_err(|e| e.to_string())?;
|
||||
for i in 0..ntemp {
|
||||
let nd = params.nden.get(i).copied().unwrap_or(0);
|
||||
for j in 0..nd {
|
||||
if j % 6 == 0 { writeln!(w).map_err(|e| e.to_string())?; }
|
||||
let val = params.absgrd[i].get(j).and_then(|row| row.get(k)).copied().unwrap_or(0.0);
|
||||
write!(w, "{:14.6e}", val).map_err(|e| e.to_string())?;
|
||||
}
|
||||
writeln!(w).map_err(|e| e.to_string())?;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// --- 二进制输出 (always) ---
|
||||
// Note: Binary output requires Fortran-compatible unformatted I/O.
|
||||
// In Rust, we write a simplified binary format.
|
||||
// The actual binary format depends on the Fortran runtime.
|
||||
// For now, we skip binary output as it requires Fortran unit 63.
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_wavelength_to_frequency() {
|
||||
// 500 nm → frequency
|
||||
let freq = wavelength_to_frequency(500.0);
|
||||
let expected = 2.997925e18 / 500.0;
|
||||
assert!((freq - expected).abs() / expected < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_frequency_to_wavelength() {
|
||||
let wl = frequency_to_wavelength(6e14);
|
||||
let expected = 2.997925e18 / 6e14;
|
||||
assert!((wl - expected).abs() / expected < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_wavelength_frequency_roundtrip() {
|
||||
let wl = 500.0;
|
||||
let freq = wavelength_to_frequency(wl);
|
||||
let wl_back = frequency_to_wavelength(freq);
|
||||
assert!((wl - wl_back).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_generate_log_grid() {
|
||||
let grid = generate_log_grid(100.0, 1000.0, 11);
|
||||
assert_eq!(grid.len(), 11);
|
||||
assert!((grid[0] - 100.0).abs() < 1e-10);
|
||||
assert!((grid[10] - 1000.0).abs() < 1e-10);
|
||||
// 网格应该是对数等距的
|
||||
let ratio = grid[1] / grid[0];
|
||||
for i in 1..10 {
|
||||
assert!((grid[i + 1] / grid[i] - ratio).abs() < 1e-10);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_generate_linear_grid_in_log() {
|
||||
let grid = generate_linear_grid_in_log(1000.0, 100000.0, 5);
|
||||
assert_eq!(grid.len(), 5);
|
||||
assert!((grid[0] - 1000.0).abs() < 1e-10);
|
||||
assert!((grid[4] - 100000.0).abs() < 1e-3);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_opacity_stats() {
|
||||
let table = OpacityTable {
|
||||
temperatures: vec![5000.0, 10000.0],
|
||||
densities: vec![vec![1e-8, 1e-7]],
|
||||
electron_densities: vec![vec![1e-10, 1e-9]],
|
||||
wavelengths: vec![100.0, 200.0],
|
||||
opacity: vec![
|
||||
vec![
|
||||
vec![1.0, 2.0],
|
||||
vec![3.0, 4.0],
|
||||
],
|
||||
vec![
|
||||
vec![5.0, 6.0],
|
||||
vec![7.0, 8.0],
|
||||
],
|
||||
],
|
||||
};
|
||||
let stats = compute_opacity_stats(&table);
|
||||
assert_eq!(stats.min_opacity, 1.0);
|
||||
assert_eq!(stats.max_opacity, 8.0);
|
||||
assert!((stats.mean_opacity - 4.5).abs() < 0.01);
|
||||
assert_eq!(stats.nonzero_percent, 100.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_opacity_flags_default() {
|
||||
let flags = OpacityFlags::default();
|
||||
assert!(!flags.h_minus);
|
||||
assert!(!flags.cia_h2h2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_fingrd_writes_text_file() {
|
||||
let dir = std::env::temp_dir().join("fingrd_test");
|
||||
std::fs::create_dir_all(&dir).unwrap();
|
||||
let tabname = dir.join("test_table.txt");
|
||||
let tabname_str = tabname.to_str().unwrap();
|
||||
|
||||
let params = FingrdParams {
|
||||
temperatures: &[5000.0, 10000.0],
|
||||
densities: &[vec![1e-8, 1e-7], vec![1e-8, 1e-7]],
|
||||
electron_densities: &[vec![1e-10, 1e-9], vec![1e-10, 1e-9]],
|
||||
wavelengths: &[100.0, 200.0, 500.0],
|
||||
absgrd: &[
|
||||
vec![vec![1.0, 2.0, 3.0], vec![4.0, 5.0, 6.0]],
|
||||
vec![vec![7.0, 8.0, 9.0], vec![10.0, 11.0, 12.0]],
|
||||
],
|
||||
nden: &[2, 2],
|
||||
abundances: &[1.0; 92],
|
||||
rel_abundances: &[1.0; 92],
|
||||
flags: &OpacityFlags::default(),
|
||||
ifmol: 0,
|
||||
tmolim: 10000.0,
|
||||
tabname: tabname_str,
|
||||
ibingr: 0,
|
||||
idens: 0,
|
||||
};
|
||||
|
||||
let result = fingrd(¶ms);
|
||||
assert!(result.is_ok());
|
||||
|
||||
// Verify file was created and has content
|
||||
let content = std::fs::read_to_string(&tabname).unwrap();
|
||||
assert!(content.contains("opacity table"));
|
||||
assert!(content.contains("number of frequencies"));
|
||||
assert!(content.contains("frequency #"));
|
||||
|
||||
// Cleanup
|
||||
std::fs::remove_dir_all(&dir).ok();
|
||||
}
|
||||
}
|
||||
280
src/synspec/math/frac1.rs
Normal file
280
src/synspec/math/frac1.rs
Normal file
@ -0,0 +1,280 @@
|
||||
//! Opacity Project ionization fraction interpolation for SYNSPEC.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine FRAC1 (line 23240).
|
||||
//!
|
||||
//! Interpolates pre-tabulated ionization fractions from the Opacity Project
|
||||
//! data (read by FRACTN) to the local temperature and electron density at
|
||||
//! each depth point, then computes the number density of each ionization
|
||||
//! stage.
|
||||
//!
|
||||
//! # Input
|
||||
//!
|
||||
//! - Temperature and electron density arrays
|
||||
//! - OP ionization fraction table (from FRACTN)
|
||||
//! - Elemental abundances, mean molecular weight, total density
|
||||
//!
|
||||
//! # Output
|
||||
//!
|
||||
//! - `rrr[id][ion][iat]` — number density fraction for element `iat`,
|
||||
//! ionization stage `ion` at depth `id`
|
||||
|
||||
// ============================================================================
|
||||
// 常量
|
||||
// ============================================================================
|
||||
|
||||
/// Maximum number of temperature grid points
|
||||
pub const MTEMP: usize = 100;
|
||||
/// Maximum number of electron density grid points
|
||||
pub const MELEC: usize = 60;
|
||||
/// Maximum number of ionization stages
|
||||
pub const MION1: usize = 30;
|
||||
|
||||
// ============================================================================
|
||||
// OP 数据结构 (COMMON /FRACOP/)
|
||||
// ============================================================================
|
||||
|
||||
/// Opacity Project ionization fraction table.
|
||||
///
|
||||
/// Corresponds to Fortran COMMON /FRACOP/:
|
||||
/// ```fortran
|
||||
/// COMMON/FRACOP/ frac(mtemp,melec,mion1), fracm(mtemp,melec),
|
||||
/// itemp(mtemp), ntt
|
||||
/// ```
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct FracOpData {
|
||||
/// Ionization fractions [MTEMP x MELEC x MION1]
|
||||
pub frac: Vec<Vec<Vec<f64>>>,
|
||||
/// Molecular fractions [MTEMP x MELEC]
|
||||
pub fracm: Vec<Vec<f64>>,
|
||||
/// Temperature grid indices [MTEMP]
|
||||
pub itemp: Vec<i32>,
|
||||
/// Number of temperature points
|
||||
pub ntt: usize,
|
||||
}
|
||||
|
||||
impl Default for FracOpData {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
frac: vec![vec![vec![0.0; MION1]; MELEC]; MTEMP],
|
||||
fracm: vec![vec![0.0; MELEC]; MTEMP],
|
||||
itemp: vec![0; MTEMP],
|
||||
ntt: 0,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// 参数结构体
|
||||
// ============================================================================
|
||||
|
||||
/// Parameters for FRAC1 calculation.
|
||||
pub struct Frac1Params<'a> {
|
||||
/// Number of depth points
|
||||
pub nd: usize,
|
||||
/// Temperature array [nd] (K)
|
||||
pub temp: &'a [f64],
|
||||
/// Electron density array [nd] (cm^-3)
|
||||
pub elec: &'a [f64],
|
||||
/// Total density array [nd] (g/cm^3)
|
||||
pub dens: &'a [f64],
|
||||
/// Mean molecular weight array [nd]
|
||||
pub wmm: &'a [f64],
|
||||
/// Total hydrogen fraction array [nd]
|
||||
pub ytot: &'a [f64],
|
||||
/// Elemental abundance [30 x nd] — abndd(iat, id)
|
||||
pub abndd: &'a [&'a [f64]],
|
||||
/// OP ionization fraction table (from FRACTN)
|
||||
pub fracop: &'a FracOpData,
|
||||
/// Maximum number of elements to process (typically 30)
|
||||
pub max_elements: usize,
|
||||
}
|
||||
|
||||
/// Result of FRAC1 calculation.
|
||||
pub struct Frac1Result {
|
||||
/// Number density fraction [nd x MION1 x 30] — rrr(id, ion, iat)
|
||||
pub rrr: Vec<Vec<Vec<f64>>>,
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// 核心计算
|
||||
// ============================================================================
|
||||
|
||||
/// Compute ionization fractions by interpolation of OP data.
|
||||
///
|
||||
/// For each depth point, computes log10(T) and log10(Ne), then
|
||||
/// bilinearly interpolates the pre-tabulated OP ionization fractions
|
||||
/// to get the number density of each ionization stage.
|
||||
pub fn frac1(params: &Frac1Params) -> Frac1Result {
|
||||
let nd = params.nd;
|
||||
let fracop = params.fracop;
|
||||
let ntt = fracop.ntt;
|
||||
|
||||
let mut rrr = vec![vec![vec![0.0; MION1]; 30]; nd];
|
||||
|
||||
if ntt == 0 {
|
||||
return Frac1Result { rrr };
|
||||
}
|
||||
|
||||
// Compute log10(T) and log10(Ne) for each depth
|
||||
let mut xxt = vec![0.0f64; nd];
|
||||
let mut xxe = vec![0.0f64; nd];
|
||||
let mut kt0 = vec![0i32; nd];
|
||||
let mut kn0 = vec![0i32; nd];
|
||||
|
||||
for id in 0..nd {
|
||||
xxt[id] = params.temp[id].log10();
|
||||
kt0[id] = 2 * (20.0 * xxt[id]) as i32;
|
||||
xxe[id] = params.elec[id].log10();
|
||||
kn0[id] = (2.0 * xxe[id]) as i32;
|
||||
}
|
||||
|
||||
// Loop over elements
|
||||
for iat in 0..params.max_elements.min(30) {
|
||||
// Find temperature index for each depth
|
||||
for id in 0..nd {
|
||||
let kt1 = find_temp_index(kt0[id], &fracop.itemp, ntt);
|
||||
let kn1 = find_elec_index(kn0[id]);
|
||||
|
||||
// Bilinear interpolation coefficients
|
||||
let xt1 = 0.025 * fracop.itemp[kt1] as f64;
|
||||
let dxt = 0.05;
|
||||
let at1 = (xxt[id] - xt1) / dxt;
|
||||
let xn1 = 0.5 * kn1 as f64;
|
||||
let dxn = 0.5;
|
||||
let an1 = (xxe[id] - xn1) / dxn;
|
||||
|
||||
// Interpolate each ionization stage
|
||||
for ion in 0..MION1 {
|
||||
let x11 = fracop.frac[kt1][kn1][ion];
|
||||
let x21 = fracop.frac[kt1 + 1][kn1][ion];
|
||||
let x12 = fracop.frac[kt1][kn1 + 1][ion];
|
||||
let x22 = fracop.frac[kt1 + 1][kn1 + 1][ion];
|
||||
|
||||
let rrx = if x11 * x21 * x12 * x22 == 0.0 {
|
||||
// Linear interpolation when any value is zero
|
||||
let xx1 = x11 + at1 * (x21 - x11);
|
||||
let xx2 = x12 + at1 * (x22 - x12);
|
||||
xx1 + an1 * (xx2 - xx1)
|
||||
} else {
|
||||
// Log-space interpolation
|
||||
let lx11 = x11.log10();
|
||||
let lx21 = x21.log10();
|
||||
let lx12 = x12.log10();
|
||||
let lx22 = x22.log10();
|
||||
let xx1 = lx11 + at1 * (lx21 - lx11);
|
||||
let xx2 = lx12 + at1 * (lx22 - lx12);
|
||||
let lrrx = xx1 + an1 * (xx2 - xx1);
|
||||
10f64.powf(lrrx)
|
||||
};
|
||||
|
||||
rrr[id][ion][iat] = rrx * params.abndd[iat][id]
|
||||
* params.dens[id] / params.wmm[id] / params.ytot[id];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Frac1Result { rrr }
|
||||
}
|
||||
|
||||
/// Find temperature index in OP table.
|
||||
///
|
||||
/// Returns the index `kt1` such that `itemp[kt1] <= kt0 < itemp[kt1+1]`.
|
||||
fn find_temp_index(kt0: i32, itemp: &[i32], ntt: usize) -> usize {
|
||||
if ntt == 0 {
|
||||
return 0;
|
||||
}
|
||||
if kt0 < itemp[0] {
|
||||
return 0;
|
||||
}
|
||||
if kt0 >= itemp[ntt - 1] {
|
||||
return ntt - 1;
|
||||
}
|
||||
for it in 0..ntt {
|
||||
if kt0 == itemp[it] {
|
||||
return it;
|
||||
}
|
||||
}
|
||||
// Fallback: find bracketing interval
|
||||
for it in 0..ntt - 1 {
|
||||
if kt0 >= itemp[it] && kt0 < itemp[it + 1] {
|
||||
return it;
|
||||
}
|
||||
}
|
||||
ntt - 1
|
||||
}
|
||||
|
||||
/// Find electron density index in OP table.
|
||||
///
|
||||
/// Returns the index `kn1` such that `kn1*0.5 <= log10(Ne) < (kn1+1)*0.5`.
|
||||
fn find_elec_index(kn0: i32) -> usize {
|
||||
if kn0 < 1 {
|
||||
0
|
||||
} else if kn0 >= 60 {
|
||||
59
|
||||
} else {
|
||||
kn0 as usize
|
||||
}
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// 测试
|
||||
// ============================================================================
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_frac1_empty_table() {
|
||||
let temp = [10000.0];
|
||||
let elec = [1e14];
|
||||
let dens = [1e-10];
|
||||
let wmm = [1.0];
|
||||
let ytot = [1.0];
|
||||
let abnd_row = vec![0.0; 1];
|
||||
let abndd: Vec<&[f64]> = vec![&abnd_row; 30];
|
||||
let fracop = FracOpData::default();
|
||||
|
||||
let params = Frac1Params {
|
||||
nd: 1,
|
||||
temp: &temp,
|
||||
elec: &elec,
|
||||
dens: &dens,
|
||||
wmm: &wmm,
|
||||
ytot: &ytot,
|
||||
abndd: &abndd,
|
||||
fracop: &fracop,
|
||||
max_elements: 30,
|
||||
};
|
||||
|
||||
let result = frac1(¶ms);
|
||||
assert_eq!(result.rrr.len(), 1);
|
||||
assert_eq!(result.rrr[0].len(), MION1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_find_temp_index() {
|
||||
let itemp = [100, 200, 300, 400, 500];
|
||||
assert_eq!(find_temp_index(50, &itemp, 5), 0); // below range
|
||||
assert_eq!(find_temp_index(100, &itemp, 5), 0); // exact match
|
||||
assert_eq!(find_temp_index(300, &itemp, 5), 2); // exact match
|
||||
assert_eq!(find_temp_index(600, &itemp, 5), 4); // above range
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_find_elec_index() {
|
||||
assert_eq!(find_elec_index(-1), 0);
|
||||
assert_eq!(find_elec_index(0), 0);
|
||||
assert_eq!(find_elec_index(10), 10);
|
||||
assert_eq!(find_elec_index(70), 59);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_fracop_default() {
|
||||
let data = FracOpData::default();
|
||||
assert_eq!(data.ntt, 0);
|
||||
assert_eq!(data.frac.len(), MTEMP);
|
||||
assert_eq!(data.frac[0].len(), MELEC);
|
||||
assert_eq!(data.frac[0][0].len(), MION1);
|
||||
}
|
||||
}
|
||||
466
src/synspec/math/fractn.rs
Normal file
466
src/synspec/math/fractn.rs
Normal file
@ -0,0 +1,466 @@
|
||||
//! 电离分数数据读取 (FRACTN)。
|
||||
//!
|
||||
//! 从 `ioniz.dat` 文件读取 OP 电离分数表，计算各元素的电离分数。
|
||||
//!
|
||||
//! # 功能
|
||||
//!
|
||||
//! 读取电离势和统计权重数据，结合温度和电子密度网格，
|
||||
//! 计算各电离态的分数分布。
|
||||
//!
|
||||
//! # Fortran 原始代码
|
||||
//!
|
||||
//! ```fortran
|
||||
//! subroutine fractn(iatnum)
|
||||
//! common/fracop/frac(mtemp,melec,mion1),fracm(mtemp,melec),
|
||||
//! itemp(mtemp),ntt
|
||||
//! ...
|
||||
//! end
|
||||
//! ```
|
||||
|
||||
use std::fs::File;
|
||||
use std::io::{BufRead, BufReader};
|
||||
|
||||
use super::frac1::{MTEMP, MELEC, MION1};
|
||||
|
||||
// ============================================================================
|
||||
// 常量
|
||||
// ============================================================================
|
||||
|
||||
/// 最大数据集数
|
||||
pub const MDAT: usize = 17;
|
||||
|
||||
// ============================================================================
|
||||
// 数据结构
|
||||
// ============================================================================
|
||||
|
||||
/// FRACTN 输出 - 电离分数表。
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct FracOp {
|
||||
/// 电离分数 [MTEMP][MELEC][MION1]
|
||||
/// frac[it][ie][ion] = 元素 iatnum 在温度 it、电子密度 ie 下的电离态 ion 分数
|
||||
pub frac: Vec<Vec<Vec<f64>>>,
|
||||
/// 负离子分数 [MTEMP][MELEC]
|
||||
pub fracm: Vec<Vec<f64>>,
|
||||
/// 温度索引数组 [MTEMP]
|
||||
pub itemp: Vec<i32>,
|
||||
/// 有效温度点数
|
||||
pub ntt: usize,
|
||||
}
|
||||
|
||||
impl FracOp {
|
||||
/// 创建新的空 FracOp。
|
||||
pub fn new() -> Self {
|
||||
Self {
|
||||
frac: vec![vec![vec![0.0; MION1]; MELEC]; MTEMP],
|
||||
fracm: vec![vec![0.0; MELEC]; MTEMP],
|
||||
itemp: vec![0; MTEMP],
|
||||
ntt: 0,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Default for FracOp {
|
||||
fn default() -> Self {
|
||||
Self::new()
|
||||
}
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// 静态数据
|
||||
// ============================================================================
|
||||
|
||||
/// 各元素的数据集索引 (IDAT)
|
||||
/// 索引从 1 开始（iatnum = 1..30），0 表示无数据
|
||||
const IDAT: [usize; 31] = [
|
||||
0, // 占位
|
||||
1, 2, 0, 0, 0, 3, 4, 5, 0, 6,
|
||||
7, 8, 9, 10, 0, 11, 0, 12, 0, 13,
|
||||
0, 0, 0, 14, 15, 16, 0, 17, 0, 0,
|
||||
];
|
||||
|
||||
/// 统计权重 GG(ion, dataset)
|
||||
/// 使用一维数组存储，索引: (ion-1) * MDAT + (dataset-1)
|
||||
const GG_DATA: [[f64; MDAT]; MION1] = [
|
||||
[2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2.],
|
||||
[0., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.],
|
||||
[0., 0., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2.],
|
||||
[0., 0., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.],
|
||||
[0., 0., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2.],
|
||||
[0., 0., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.],
|
||||
[0., 0., 6., 6., 6., 6., 6., 6., 6., 6., 6., 6., 6., 6., 6., 6., 6.],
|
||||
[0., 0., 9., 9., 9., 9., 9., 9., 9., 9., 9., 9., 9., 9., 9., 9., 9.],
|
||||
[0., 0., 0., 4., 4., 4., 4., 4., 4., 4., 4., 4., 4., 4., 4., 4., 4.],
|
||||
[0., 0., 0., 9., 9., 9., 9., 9., 9., 9., 9., 9., 9., 9., 9., 9., 9.],
|
||||
[0., 0., 0., 0., 0., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.],
|
||||
[0., 0., 0., 0., 0., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2., 2.],
|
||||
[0., 0., 0., 0., 0., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 6., 6., 6., 6., 6., 6., 6., 6., 6., 6.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 9., 9., 9., 9., 9., 9., 9., 9., 9., 9.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 4., 4., 4., 4., 4., 4., 4., 4., 4., 4.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 9., 9., 9., 9., 9., 9., 9., 9., 9., 9.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 6., 6., 6., 6., 6., 6., 6., 6., 6., 6.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 0., 0., 10., 10., 10., 10., 10., 10., 10., 10.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 0., 0., 21., 21., 21., 21., 21., 21., 21., 21.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 0., 0., 28., 28., 28., 28., 28., 28., 28., 28.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 0., 0., 25., 25., 25., 25., 25., 25., 25., 25.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 0., 0., 6., 6., 6., 6., 6., 6., 6., 6.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 0., 0., 7., 7., 7., 7., 7., 7., 7., 7.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 0., 0., 6., 6., 25., 25., 25., 25., 25., 25.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 30., 30., 30., 30., 30., 30.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 25., 25., 25., 25., 25., 25.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 28., 28., 28., 28., 28., 28.],
|
||||
[0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 21., 21., 21., 21., 21., 21.],
|
||||
];
|
||||
|
||||
/// 各数据集的电离势 UU(ion, dataset) * 1000 cm^-1
|
||||
/// 对应 Fortran 的 uu 数组
|
||||
/// 只有非零值需要存储，按 (dataset_index, ion_index) -> value
|
||||
fn get_u0(iatnum: usize, ion_idx: usize) -> f64 {
|
||||
// ion_idx: 1..iatnum (1-indexed)
|
||||
// 对应 Fortran: u0(i) = uu(i, idat(iatnum)) * 1000.
|
||||
let dataset = IDAT[iatnum];
|
||||
if dataset == 0 || ion_idx == 0 || ion_idx > iatnum {
|
||||
return 0.0;
|
||||
}
|
||||
|
||||
// 各数据集的 UU 值 (已乘 1000)
|
||||
// 数据集 1: H, He
|
||||
// 数据集 2: Li, Be
|
||||
// 数据集 3: C
|
||||
// 数据集 4: N
|
||||
// 数据集 5: O
|
||||
// 数据集 6: Ne
|
||||
// 数据集 7: Na
|
||||
// 数据集 8: Mg
|
||||
// 数据集 9: Al
|
||||
// 数据集 10: Si
|
||||
// 数据集 11: S
|
||||
// 数据集 12: Ar
|
||||
// 数据集 13: Ca
|
||||
// 数据集 14: Fe
|
||||
// 数据集 15: Ni
|
||||
// 数据集 16: Zn
|
||||
// 数据集 17: Kr
|
||||
|
||||
// 预定义的 UU 数据集 (单位: 1000 cm^-1，已乘 1000)
|
||||
const U_DATASETS: [[f64; 30]; 18] = [
|
||||
// 数据集 0 (未使用)
|
||||
[0.0; 30],
|
||||
// 数据集 1: H (1 ion)
|
||||
[109678.7, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 2: Li, Be (2 ions)
|
||||
[198310.8, 438908.9, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 3: C (6 ions)
|
||||
[90820.0, 196665.0, 386241.0, 520178.0, 3162395.0, 3952061.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 4: N (7 ions)
|
||||
[117225.0, 238751.0, 382704.0, 624866.0, 789537.0, 4452758.0, 5380089.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 5: O (8 ions)
|
||||
[109837.0, 283240.0, 443086.0, 624384.0, 918657.0, 1114008.0, 5963135.0, 7028393.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 6: Ne (10 ions)
|
||||
[173930.0, 330391.0, 511800.0, 783300.0, 1018000.0, 1273800.0, 1671792.0, 1928462.0,
|
||||
9645005.0, 10986876.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 7: Na (11 ions)
|
||||
[41449.0, 381395.0, 577800.0, 797800.0, 1116200.0, 1388500.0, 1681500.0, 2130800.0,
|
||||
2418700.0, 11817061.0, 13297676.0, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 8: Mg (12 ions)
|
||||
[61671.0, 121268.0, 646410.0, 881100.0, 1139400.0, 1504300.0, 1814300.0, 2144700.0,
|
||||
2645200.0, 2964400.0, 14210261.0, 15829951.0, 0.0, 0.0, 0.0, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 9: Al (13 ions)
|
||||
[48278.0, 151860.0, 229446.0, 967800.0, 1239800.0, 1536300.0, 1947300.0, 2295400.0,
|
||||
2663400.0, 3214800.0, 3565600.0, 16825022.0, 18584138.0, 0.0, 0.0, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 10: Si (14 ions)
|
||||
[65748.0, 131838.0, 270139.0, 364093.0, 1345100.0, 1653900.0, 1988400.0, 2445300.0,
|
||||
2831900.0, 3237800.0, 3839800.0, 4222400.0, 19661693.0, 21560630.0, 0.0, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 11: S (16 ions)
|
||||
[83558.0, 188200.0, 280900.0, 381541.0, 586200.0, 710184.0, 2265900.0, 2647400.0,
|
||||
3057700.0, 3606100.0, 4071400.0, 4554300.0, 5255900.0, 5703600.0, 26002663.0, 28182535.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 12: Ar (18 ions)
|
||||
[127110.0, 222848.0, 328600.0, 482400.0, 605100.0, 734040.0, 1002730.0, 1157080.0,
|
||||
3407300.0, 3860900.0, 4347000.0, 4986600.0, 5533800.0, 6095500.0, 6894200.0, 7404400.0,
|
||||
33237173.0, 35699936.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 13: Ca (20 ions)
|
||||
[49306.0, 95752.0, 410642.0, 542600.0, 681600.0, 877400.0, 1026000.0, 1187600.0,
|
||||
1520640.0, 1704047.0, 4774000.0, 5301000.0, 5861000.0, 6595000.0, 7215000.0, 7860000.0,
|
||||
8770000.0, 9338000.0, 41366000.0, 44177410.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 14: Fe (24 ions)
|
||||
[54576.0, 132966.0, 249700.0, 396500.0, 560200.0, 731020.0, 1291900.0, 1490000.0,
|
||||
1688000.0, 1971000.0, 2184000.0, 2404000.0, 2862000.0, 3098520.0, 8151000.0, 8850000.0,
|
||||
9560000.0, 10480000.0, 11260000.0, 12070000.0, 13180000.0, 13882000.0, 60344000.0, 63675900.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 15: Ni (25 ions)
|
||||
[59959.0, 126145.0, 271550.0, 413000.0, 584000.0, 771100.0, 961440.0, 1569000.0,
|
||||
1789000.0, 2003000.0, 2307000.0, 2536000.0, 2771000.0, 3250000.0, 3509820.0, 9152000.0,
|
||||
9872000.0, 10620000.0, 11590000.0, 12410000.0, 13260000.0, 14420000.0, 15162000.0, 65660000.0,
|
||||
69137400.0, 0.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 16: Zn (26 ions)
|
||||
[63737.0, 130563.0, 247220.0, 442000.0, 605000.0, 799000.0, 1008000.0, 1218380.0,
|
||||
1884000.0, 2114000.0, 2341000.0, 2668000.0, 2912000.0, 3163000.0, 3686000.0, 3946820.0,
|
||||
10180000.0, 10985000.0, 11850000.0, 12708000.0, 13620000.0, 14510000.0, 15797000.0, 16500000.0,
|
||||
71203000.0, 74829600.0, 0.0, 0.0, 0.0, 0.0],
|
||||
// 数据集 17: Kr (28 ions)
|
||||
[61600.0, 146542.0, 283800.0, 443000.0, 613500.0, 870000.0, 1070000.0, 1310000.0,
|
||||
1560000.0, 1812000.0, 2589000.0, 2840000.0, 3100000.0, 3470000.0, 3740000.0, 4020000.0,
|
||||
4606000.0, 4896200.0, 12430000.0, 13290000.0, 14160000.0, 15280000.0, 16220000.0, 17190000.0,
|
||||
18510000.0, 19351000.0, 82984000.0, 86909400.0, 0.0, 0.0],
|
||||
];
|
||||
|
||||
let ds = dataset;
|
||||
if ds == 0 || ion_idx > 30 {
|
||||
return 0.0;
|
||||
}
|
||||
U_DATASETS[ds][ion_idx - 1]
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// FRACTN 主函数
|
||||
// ============================================================================
|
||||
|
||||
/// 读取指定元素的电离分数数据。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `iatnum` - 原子序数 (1..30)。如果数据不存在，返回 `None`。
|
||||
/// * `data_dir` - 数据文件目录（包含 `ioniz.dat`）
|
||||
///
|
||||
/// # 返回值
|
||||
///
|
||||
/// `FracOp` 结构体，包含电离分数表。如果元素无数据，返回 `None`。
|
||||
///
|
||||
/// # Fortran 原始代码
|
||||
///
|
||||
/// ```fortran
|
||||
/// subroutine fractn(iatnum)
|
||||
/// common/fracop/frac(mtemp,melec,mion1),fracm(mtemp,melec),
|
||||
/// itemp(mtemp),ntt
|
||||
/// ...
|
||||
/// end
|
||||
/// ```
|
||||
pub fn fractn(iatnum: usize, data_dir: &str) -> Option<FracOp> {
|
||||
if iatnum == 0 || iatnum > 30 || IDAT[iatnum] == 0 {
|
||||
return None;
|
||||
}
|
||||
|
||||
let file_path = format!("{}/ioniz.dat", data_dir);
|
||||
let file = match File::open(&file_path) {
|
||||
Ok(f) => f,
|
||||
Err(_) => return None,
|
||||
};
|
||||
let reader = BufReader::new(file);
|
||||
let mut lines = reader.lines();
|
||||
|
||||
let mut frac_op = FracOp::new();
|
||||
|
||||
// 设置统计权重和电离势
|
||||
let mut g0 = [0.0f64; MION1 + 2]; // g0(-1:mion1)
|
||||
g0[iatnum + 1] = 1.0;
|
||||
for i in 1..=iatnum {
|
||||
let ig0 = iatnum - i + 1;
|
||||
g0[ig0] = GG_DATA[i - 1][IDAT[iatnum]];
|
||||
}
|
||||
|
||||
// 读取头行
|
||||
let _header = lines.next()?.ok()?;
|
||||
|
||||
// 读取温度范围
|
||||
let line = lines.next()?.ok()?;
|
||||
let parts: Vec<&str> = line.split_whitespace().collect();
|
||||
if parts.len() < 3 {
|
||||
return None;
|
||||
}
|
||||
let it0: i32 = parts[0].parse().ok()?;
|
||||
let it1: i32 = parts[1].parse().ok()?;
|
||||
let itstp: i32 = parts[2].parse().ok()?;
|
||||
|
||||
let ntt = ((it1 - it0) / itstp + 1) as usize;
|
||||
frac_op.ntt = ntt;
|
||||
|
||||
// 读取各温度点的数据
|
||||
for it in 0..ntt {
|
||||
let line = lines.next()?.ok()?;
|
||||
let parts: Vec<&str> = line.split_whitespace().collect();
|
||||
if parts.len() < 4 {
|
||||
continue;
|
||||
}
|
||||
let itt: i32 = parts[0].parse().ok()?;
|
||||
let ie0: i32 = parts[1].parse().ok()?;
|
||||
let ie1: i32 = parts[2].parse().ok()?;
|
||||
let iestp: i32 = parts[3].parse().ok()?;
|
||||
|
||||
frac_op.itemp[it] = itt;
|
||||
|
||||
let t = (std::f64::consts::LN_10 * 0.025 * itt as f64).exp();
|
||||
let safac0 = t.sqrt() * t / 2.07e-16;
|
||||
let tkcm = 0.69496 * t;
|
||||
|
||||
let net = ((ie1 - ie0) / iestp + 1) as usize;
|
||||
|
||||
for _ie in 0..net {
|
||||
let line = lines.next()?.ok()?;
|
||||
// 格式: 3i4,2x,4(i4,1x,e9.3)
|
||||
// 简化解析
|
||||
let parts: Vec<&str> = line.split_whitespace().collect();
|
||||
if parts.len() < 3 {
|
||||
continue;
|
||||
}
|
||||
|
||||
let iee: i32 = parts[0].parse().ok()?;
|
||||
let ion0: usize = parts[1].parse().ok()?;
|
||||
let ion1: usize = parts[2].parse().ok()?;
|
||||
|
||||
let ane = (std::f64::consts::LN_10 * 0.25 * iee as f64).exp();
|
||||
let safac = safac0 / ane;
|
||||
let ieind = (iee / 2) as usize;
|
||||
|
||||
// 读取分数数据
|
||||
let mut frac0 = [0.0f64; MION1 + 2]; // frac0(-1:mion1)
|
||||
let mut ioo = [0i32; MION1 + 2]; // ioo(-1:mion1)
|
||||
|
||||
// 解析第一组 (最多 4 个)
|
||||
let n_parse = (ion1 - ion0 + 1).min(4);
|
||||
for k in 0..n_parse {
|
||||
let idx = 3 + k * 2;
|
||||
if idx + 1 < parts.len() {
|
||||
ioo[ion0 + k] = parts[idx].parse().unwrap_or(0);
|
||||
frac0[ion0 + k] = parts[idx + 1].parse().unwrap_or(0.0);
|
||||
}
|
||||
}
|
||||
|
||||
// 如果有多于 4 个电离态，继续读取
|
||||
let nio = ion1 - ion0;
|
||||
if nio >= 3 {
|
||||
let nlin = nio / 4;
|
||||
for _ilin in 0..nlin {
|
||||
let line = lines.next()?.ok()?;
|
||||
let parts: Vec<&str> = line.split_whitespace().collect();
|
||||
let start_ion = ion0 + 4 * (_ilin + 1);
|
||||
for k in 0..4 {
|
||||
let idx = k * 2;
|
||||
if idx + 1 < parts.len() && start_ion + k <= ion1 {
|
||||
ioo[start_ion + k] = parts[idx].parse().unwrap_or(0);
|
||||
frac0[start_ion + k] = parts[idx + 1].parse().unwrap_or(0.0);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 计算电离分数
|
||||
let mut z0 = [0.0f64; MION1 + 2]; // z0(-1:mion1)
|
||||
for ion in ion0..=ion1 {
|
||||
if ion < iatnum {
|
||||
if ion == ion0 {
|
||||
z0[ion] = g0[iatnum - ion];
|
||||
} else {
|
||||
z0[ion] = frac0[ion] / frac0[ion - 1] * safac * z0[ion - 1];
|
||||
let u0_val = get_u0(iatnum, iatnum - ion);
|
||||
if tkcm > 0.0 && u0_val != 0.0 {
|
||||
z0[ion] *= (-u0_val / tkcm).exp();
|
||||
}
|
||||
}
|
||||
if z0[ion] != 0.0 {
|
||||
frac_op.frac[it][ieind][iatnum - ion] = frac0[ion] / z0[ion];
|
||||
}
|
||||
} else {
|
||||
// 负离子 H-
|
||||
let u0hm = 6090.5;
|
||||
let z0hm = if ion > 0 && frac0[ion - 1] != 0.0 {
|
||||
frac0[ion] / frac0[ion - 1] * safac
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
let z0hm = if tkcm > 0.0 {
|
||||
z0hm * (-u0hm / tkcm).exp()
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
if z0hm != 0.0 {
|
||||
frac_op.fracm[it][ieind] = frac0[ion] / z0hm;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Some(frac_op)
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// 测试
|
||||
// ============================================================================
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_idat_table() {
|
||||
// 验证 IDAT 表的正确性
|
||||
assert_eq!(IDAT[1], 1); // H
|
||||
assert_eq!(IDAT[2], 2); // He
|
||||
assert_eq!(IDAT[3], 0); // Li - 无数据
|
||||
assert_eq!(IDAT[6], 3); // C
|
||||
assert_eq!(IDAT[7], 4); // N
|
||||
assert_eq!(IDAT[8], 5); // O
|
||||
assert_eq!(IDAT[26], 16); // Fe
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_gg_data() {
|
||||
// 验证统计权重数据
|
||||
assert_eq!(GG_DATA[0][0], 2.0); // H 基态
|
||||
assert_eq!(GG_DATA[0][1], 2.0); // He 基态
|
||||
assert_eq!(GG_DATA[1][1], 1.0); // He+ 基态
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_get_u0() {
|
||||
// H 的电离势
|
||||
let u0_h = get_u0(1, 1);
|
||||
assert!((u0_h - 109678.7).abs() < 1.0);
|
||||
|
||||
// 无效输入
|
||||
let u0_invalid = get_u0(0, 1);
|
||||
assert_eq!(u0_invalid, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_frac_op_new() {
|
||||
let frac_op = FracOp::new();
|
||||
assert_eq!(frac_op.ntt, 0);
|
||||
assert_eq!(frac_op.frac.len(), MTEMP);
|
||||
assert_eq!(frac_op.frac[0].len(), MELEC);
|
||||
assert_eq!(frac_op.frac[0][0].len(), MION1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_fractn_no_data_element() {
|
||||
// Li (原子序数 3) 没有数据
|
||||
let result = fractn(3, "/nonexistent");
|
||||
assert!(result.is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_fractn_invalid_atomic_number() {
|
||||
let result = fractn(0, "/nonexistent");
|
||||
assert!(result.is_none());
|
||||
|
||||
let result = fractn(31, "/nonexistent");
|
||||
assert!(result.is_none());
|
||||
}
|
||||
}
|
||||
146
src/synspec/math/gaunt.rs
Normal file
146
src/synspec/math/gaunt.rs
Normal file
@ -0,0 +1,146 @@
|
||||
//! Hydrogenic bound-free Gaunt factors.
|
||||
//!
|
||||
//! Translated from SYNSPEC `GAUNT` and `GNTK` functions (synspec54.f:3715, 3763).
|
||||
|
||||
/// Hydrogenic bound-free Gaunt factor.
|
||||
///
|
||||
/// Calculates the bound-free Gaunt factor for hydrogenic ions
|
||||
/// for principal quantum number `i` and frequency `fr`.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `i` - Principal quantum number (1-10)
|
||||
/// * `fr` - Frequency (Hz)
|
||||
///
|
||||
/// # Returns
|
||||
/// The bound-free Gaunt factor.
|
||||
pub fn gaunt(i: i32, fr: f64) -> f64 {
|
||||
let x = fr / 2.99793e14;
|
||||
|
||||
match i {
|
||||
1 => {
|
||||
1.2302628 + x * (-2.9094219e-3 + x * (7.3993579e-6 - 8.7356966e-9 * x))
|
||||
+ (12.803223 / x - 5.5759888) / x
|
||||
}
|
||||
2 => {
|
||||
1.1595421 + x * (-2.0735860e-3 + 2.7033384e-6 * x)
|
||||
+ (-1.2709045 + (-2.0244141 / x + 2.1325684) / x) / x
|
||||
}
|
||||
3 => {
|
||||
1.1450949 + x * (-1.9366592e-3 + 2.3572356e-6 * x)
|
||||
+ (-0.55936432 + (-0.23387146 / x + 0.52471924) / x) / x
|
||||
}
|
||||
4 => {
|
||||
1.1306695 + x * (-1.3482273e-3 + x * (-4.6949424e-6 + 2.3548636e-8 * x))
|
||||
+ (-0.31190730 + (0.19683564 - 5.4418565e-2 / x) / x) / x
|
||||
}
|
||||
5 => {
|
||||
1.1190904 + x * (-1.0401085e-3 + x * (-6.9943488e-6 + 2.8496742e-8 * x))
|
||||
+ (-0.16051018 + (5.5545091e-2 - 8.9182854e-3 / x) / x) / x
|
||||
}
|
||||
6 => {
|
||||
1.1168376 + x * (-8.9466573e-4 + x * (-8.8393133e-6 + 3.4696768e-8 * x))
|
||||
+ (-0.13075417 + (4.1921183e-2 - 5.5303574e-3 / x) / x) / x
|
||||
}
|
||||
7 => {
|
||||
1.1128632 + x * (-7.4833260e-4 + x * (-1.0244504e-5 + 3.8595771e-8 * x))
|
||||
+ (-9.5441161e-2 + (2.3350812e-2 - 2.2752881e-3 / x) / x) / x
|
||||
}
|
||||
8 => {
|
||||
1.1093137 + x * (-6.2619148e-4 + x * (-1.1342068e-5 + 4.1477731e-8 * x))
|
||||
+ (-7.1010560e-2 + (1.3298411e-2 - 9.7200274e-4 / x) / x) / x
|
||||
}
|
||||
9 => {
|
||||
1.1078717 + x * (-5.4837392e-4 + x * (-1.2157943e-5 + 4.3796716e-8 * x))
|
||||
+ (-5.6046560e-2 + (8.5139736e-3 - 4.9576163e-4 / x) / x) / x
|
||||
}
|
||||
10 => {
|
||||
1.1052734 + x * (-4.4341570e-4 + x * (-1.3235905e-5 + 4.7003140e-8 * x))
|
||||
+ (-4.7326370e-2 + (6.1516856e-3 - 2.9467046e-4 / x) / x) / x
|
||||
}
|
||||
_ => 1.0,
|
||||
}
|
||||
}
|
||||
|
||||
/// Hydrogenic bound-free Gaunt factor (Klaus Werner version).
|
||||
///
|
||||
/// Alternative Gaunt factor calculation for low quantum numbers.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `i` - Principal quantum number (1-3)
|
||||
/// * `fr` - Frequency (Hz)
|
||||
///
|
||||
/// # Returns
|
||||
/// The bound-free Gaunt factor.
|
||||
pub fn gntk(i: i32, fr: f64) -> f64 {
|
||||
let y = 1.0 / fr;
|
||||
|
||||
match i {
|
||||
1 => 0.9916 + y * (2.71852e13 - y * 2.26846e30),
|
||||
2 => 1.1050 - y * (2.37490e14 - y * 4.07677e28),
|
||||
3 => 1.1010 - y * (0.98632e14 - y * 1.03540e28),
|
||||
_ => 1.0,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_gaunt_n1() {
|
||||
// Use frequency in valid range for Gaunt factor
|
||||
let result = gaunt(1, 5.0e14);
|
||||
assert!(result.is_finite());
|
||||
assert!(result > 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_gaunt_n2() {
|
||||
// Use frequency in valid range for Gaunt factor
|
||||
let result = gaunt(2, 5.0e14);
|
||||
assert!(result.is_finite());
|
||||
assert!(result > 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_gaunt_high_n() {
|
||||
let result = gaunt(10, 3.0e14);
|
||||
assert!(result.is_finite());
|
||||
assert!(result > 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_gaunt_default() {
|
||||
// For n > 10, should return 1.0
|
||||
let result = gaunt(11, 3.0e14);
|
||||
assert_eq!(result, 1.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_gntk_n1() {
|
||||
// Use higher frequency for valid GNTK values
|
||||
let result = gntk(1, 1.0e15);
|
||||
assert!(result.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_gntk_n2() {
|
||||
// Use higher frequency for valid GNTK values
|
||||
let result = gntk(2, 1.0e15);
|
||||
assert!(result.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_gntk_n3() {
|
||||
// Use higher frequency for valid GNTK values
|
||||
let result = gntk(3, 1.0e15);
|
||||
assert!(result.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_gntk_default() {
|
||||
// For n > 3, should return 1.0
|
||||
let result = gntk(4, 3.0e14);
|
||||
assert_eq!(result, 1.0);
|
||||
}
|
||||
}
|
||||
267
src/synspec/math/getlal.rs
Normal file
267
src/synspec/math/getlal.rs
Normal file
@ -0,0 +1,267 @@
|
||||
//! Read quasi-molecular satellite line profile data.
|
||||
//!
|
||||
//! Translated from SYNSPEC `getlal` subroutine (synspec54.f).
|
||||
//!
|
||||
//! Reads profile functions for Lyman alpha, beta, gamma, and Balmer alpha,
|
||||
//! including quasi-molecular satellites. Data files are in `./data/` directory.
|
||||
|
||||
use std::fs::File;
|
||||
use std::io::{BufRead, BufReader};
|
||||
use std::path::Path;
|
||||
|
||||
use super::allard::{AllardData, AllardTable, NNMAX};
|
||||
|
||||
// ============================================================================
|
||||
// Data file names
|
||||
// ============================================================================
|
||||
|
||||
/// Lyman alpha quasi-molecular data file
|
||||
const LAQUASI_FILE: &str = "laquasi.dat";
|
||||
|
||||
/// Lyman beta quasi-molecular data file
|
||||
const LBQUASI_FILE: &str = "lbquasi.dat";
|
||||
|
||||
/// Lyman gamma quasi-molecular data file
|
||||
const LGQUASI_FILE: &str = "lgquasi.dat";
|
||||
|
||||
/// Balmer alpha quasi-molecular data file
|
||||
const LHQUASI_FILE: &str = "lhquasi.dat";
|
||||
|
||||
// ============================================================================
|
||||
// Helper: read one quasi-molecular table from file
|
||||
// ============================================================================
|
||||
|
||||
/// Read one quasi-molecular table from a data file.
|
||||
///
|
||||
/// # File format
|
||||
/// Line 1: `nx stnne stnch vneu vcha`
|
||||
/// Lines 2..nx+1: `xl pl[0] pl[1] pl[2] pl[3] pl[4]`
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `path` - Path to data file
|
||||
///
|
||||
/// # Returns
|
||||
/// Populated `AllardTable` or error message.
|
||||
fn read_table(path: &Path) -> Result<AllardTable, String> {
|
||||
let file = File::open(path).map_err(|e| format!("Cannot open {}: {}", path.display(), e))?;
|
||||
let reader = BufReader::new(file);
|
||||
let mut lines = reader.lines();
|
||||
|
||||
// Read header: nx, stnne, stnch, vneu, vcha
|
||||
let header = lines
|
||||
.next()
|
||||
.ok_or_else(|| format!("Empty file: {}", path.display()))?
|
||||
.map_err(|e| format!("Read error: {}", e))?;
|
||||
|
||||
let parts: Vec<f64> = header
|
||||
.split_whitespace()
|
||||
.map(|s| s.parse().unwrap_or(0.0))
|
||||
.collect();
|
||||
|
||||
if parts.len() < 5 {
|
||||
return Err(format!(
|
||||
"Invalid header in {}: expected 5 values, got {}",
|
||||
path.display(),
|
||||
parts.len()
|
||||
));
|
||||
}
|
||||
|
||||
let nx = parts[0] as usize;
|
||||
let stnne_raw = parts[1];
|
||||
let stnch_raw = parts[2];
|
||||
let vneu = parts[3];
|
||||
let vcha = parts[4];
|
||||
|
||||
// Read data points
|
||||
let mut xl = Vec::with_capacity(nx);
|
||||
let mut pl = Vec::with_capacity(nx);
|
||||
|
||||
for (i, line_result) in lines.enumerate() {
|
||||
if i >= nx {
|
||||
break;
|
||||
}
|
||||
let line = line_result.map_err(|e| format!("Read error at line {}: {}", i + 2, e))?;
|
||||
let values: Vec<f64> = line
|
||||
.split_whitespace()
|
||||
.map(|s| s.parse().unwrap_or(0.0))
|
||||
.collect();
|
||||
|
||||
if values.len() < 6 {
|
||||
return Err(format!(
|
||||
"Invalid data at line {} in {}: expected 6 values, got {}",
|
||||
i + 2,
|
||||
path.display(),
|
||||
values.len()
|
||||
));
|
||||
}
|
||||
|
||||
xl.push(values[0]);
|
||||
let mut row = [0.0f64; NNMAX];
|
||||
for j in 0..NNMAX {
|
||||
row[j] = values[j + 1];
|
||||
}
|
||||
pl.push(row);
|
||||
}
|
||||
|
||||
// Convert log densities to linear
|
||||
let stnne = 10.0f64.powf(stnne_raw);
|
||||
let stnch = 10.0f64.powf(stnch_raw);
|
||||
|
||||
Ok(AllardTable {
|
||||
xl,
|
||||
pl,
|
||||
stnne,
|
||||
stnch,
|
||||
vneu,
|
||||
vcha,
|
||||
nx,
|
||||
iwarn: false,
|
||||
})
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// Main entry point
|
||||
// ============================================================================
|
||||
|
||||
/// Read quasi-molecular satellite line data from files.
|
||||
///
|
||||
/// Translated from SYNSPEC `getlal` subroutine (synspec54.f).
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `data_dir` - Path to data directory (e.g., `./data/`)
|
||||
/// * `nunalp` - Flag for Lyman alpha (>0 to read)
|
||||
/// * `nunbet` - Flag for Lyman beta (>0 to read)
|
||||
/// * `nungam` - Flag for Lyman gamma (>0 to read)
|
||||
/// * `nunbal` - Flag for Balmer alpha (>0 to read)
|
||||
///
|
||||
/// # Returns
|
||||
/// Populated `AllardData` structure.
|
||||
pub fn getlal(
|
||||
data_dir: &Path,
|
||||
nunalp: i32,
|
||||
nunbet: i32,
|
||||
nungam: i32,
|
||||
nunbal: i32,
|
||||
) -> AllardData {
|
||||
let mut data = AllardData::default();
|
||||
|
||||
// Lyman alpha
|
||||
if nunalp > 0 {
|
||||
let path = data_dir.join(LAQUASI_FILE);
|
||||
match read_table(&path) {
|
||||
Ok(table) => {
|
||||
data.lalp = table;
|
||||
eprintln!(" read quasi-molecular data for L alpha");
|
||||
}
|
||||
Err(e) => {
|
||||
eprintln!(" Warning: {}", e);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Lyman beta
|
||||
if nunbet > 0 {
|
||||
let path = data_dir.join(LBQUASI_FILE);
|
||||
match read_table(&path) {
|
||||
Ok(table) => {
|
||||
data.bet = table;
|
||||
eprintln!(" read quasi-molecular data for L beta");
|
||||
}
|
||||
Err(e) => {
|
||||
eprintln!(" Warning: {}", e);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Lyman gamma
|
||||
if nungam > 0 {
|
||||
let path = data_dir.join(LGQUASI_FILE);
|
||||
match read_table(&path) {
|
||||
Ok(table) => {
|
||||
data.gam = table;
|
||||
eprintln!(" read quasi-molecular data for L gamma");
|
||||
}
|
||||
Err(e) => {
|
||||
eprintln!(" Warning: {}", e);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Balmer alpha
|
||||
if nunbal > 0 {
|
||||
let path = data_dir.join(LHQUASI_FILE);
|
||||
match read_table(&path) {
|
||||
Ok(table) => {
|
||||
data.bal = table;
|
||||
eprintln!(" read quasi-molecular data for H alpha");
|
||||
}
|
||||
Err(e) => {
|
||||
eprintln!(" Warning: {}", e);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
data
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// Tests
|
||||
// ============================================================================
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use std::io::Write;
|
||||
|
||||
#[test]
|
||||
fn test_read_table_valid() {
|
||||
// Create temporary file
|
||||
let dir = std::env::temp_dir().join("getlal_test");
|
||||
std::fs::create_dir_all(&dir).unwrap();
|
||||
let path = dir.join("test.dat");
|
||||
|
||||
let mut file = File::create(&path).unwrap();
|
||||
writeln!(file, "3 12.0 10.0 1.0 1.0").unwrap();
|
||||
writeln!(file, "1210.0 1.0 0.5 0.3 0.2 0.1").unwrap();
|
||||
writeln!(file, "1215.0 2.0 1.0 0.6 0.4 0.2").unwrap();
|
||||
writeln!(file, "1220.0 1.5 0.75 0.45 0.3 0.15").unwrap();
|
||||
|
||||
let table = read_table(&path).unwrap();
|
||||
assert_eq!(table.nx, 3);
|
||||
assert!((table.stnne - 1e12).abs() < 1.0);
|
||||
assert!((table.stnch - 1e10).abs() < 1.0);
|
||||
assert!((table.xl[0] - 1210.0).abs() < 1e-10);
|
||||
assert!((table.pl[1][0] - 2.0).abs() < 1e-10);
|
||||
|
||||
// Cleanup
|
||||
std::fs::remove_dir_all(&dir).unwrap();
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_read_table_missing_file() {
|
||||
let path = Path::new("/nonexistent/file.dat");
|
||||
let result = read_table(path);
|
||||
assert!(result.is_err());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_getlal_no_files() {
|
||||
let dir = Path::new("/nonexistent");
|
||||
let data = getlal(dir, 1, 1, 1, 1);
|
||||
assert_eq!(data.lalp.nx, 0);
|
||||
assert_eq!(data.bet.nx, 0);
|
||||
assert_eq!(data.gam.nx, 0);
|
||||
assert_eq!(data.bal.nx, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_getlal_skip_disabled() {
|
||||
let dir = Path::new("/nonexistent");
|
||||
let data = getlal(dir, 0, 0, 0, 0);
|
||||
// All tables should be empty when flags are 0
|
||||
assert_eq!(data.lalp.nx, 0);
|
||||
assert_eq!(data.bet.nx, 0);
|
||||
assert_eq!(data.gam.nx, 0);
|
||||
assert_eq!(data.bal.nx, 0);
|
||||
}
|
||||
}
|
||||
161
src/synspec/math/getwrd.rs
Normal file
161
src/synspec/math/getwrd.rs
Normal file
@ -0,0 +1,161 @@
|
||||
//! Word extraction from text string.
|
||||
//!
|
||||
//! Translated from SYNSPEC `GETWRD` subroutine (synspec54.f:1278).
|
||||
//!
|
||||
//! Finds the next word in a text string starting from index `k0`.
|
||||
//! A word is a sequence of alphanumeric characters delimited by
|
||||
//! separators: space, `(`, `)`, `=`, `*`, `/`, `,`.
|
||||
|
||||
const SEPARATORS: &[char] = &[' ', '(', ')', '=', '*', '/', ','];
|
||||
|
||||
/// Find the next word in a text string.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `text` - Input text string
|
||||
/// * `k0` - Starting search index (0-based)
|
||||
///
|
||||
/// # Returns
|
||||
/// `Some((k1, k2))` where:
|
||||
/// * `k1` - Start index of the word (0-based)
|
||||
/// * `k2` - End index of the word (0-based, inclusive)
|
||||
///
|
||||
/// Returns `None` if no word is found.
|
||||
pub fn getwrd(text: &str, k0: usize) -> Option<(usize, usize)> {
|
||||
let chars: Vec<char> = text.chars().collect();
|
||||
let len = chars.len();
|
||||
|
||||
let mut k1: Option<usize> = None;
|
||||
|
||||
for i in k0..len {
|
||||
match k1 {
|
||||
None => {
|
||||
// Looking for start of word
|
||||
if !SEPARATORS.contains(&chars[i]) {
|
||||
k1 = Some(i);
|
||||
}
|
||||
}
|
||||
Some(start) => {
|
||||
// Looking for end of word
|
||||
if SEPARATORS.contains(&chars[i]) {
|
||||
return Some((start, i - 1));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// If we reached end of string while in a word
|
||||
if let Some(start) = k1 {
|
||||
return Some((start, len - 1));
|
||||
}
|
||||
|
||||
// No word found
|
||||
None
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_getwrd_simple() {
|
||||
let text = "hello world";
|
||||
let (k1, k2) = getwrd(text, 0).unwrap();
|
||||
assert_eq!(k1, 0);
|
||||
assert_eq!(k2, 4);
|
||||
assert_eq!(&text[k1..=k2], "hello");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_getwrd_second_word() {
|
||||
let text = "hello world";
|
||||
let (k1, k2) = getwrd(text, 5).unwrap();
|
||||
assert_eq!(k1, 6);
|
||||
assert_eq!(k2, 10);
|
||||
assert_eq!(&text[k1..=k2], "world");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_getwrd_with_separators() {
|
||||
let text = "a=b/c(d)";
|
||||
let (k1, k2) = getwrd(text, 0).unwrap();
|
||||
assert_eq!(k1, 0);
|
||||
assert_eq!(k2, 0);
|
||||
assert_eq!(&text[k1..=k2], "a");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_getwrd_after_separator() {
|
||||
let text = "a=b";
|
||||
let (k1, k2) = getwrd(text, 1).unwrap();
|
||||
assert_eq!(k1, 2);
|
||||
assert_eq!(k2, 2);
|
||||
assert_eq!(&text[k1..=k2], "b");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_getwrd_no_word() {
|
||||
let text = " ";
|
||||
assert!(getwrd(text, 0).is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_getwrd_empty_string() {
|
||||
let text = "";
|
||||
assert!(getwrd(text, 0).is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_getwrd_leading_spaces() {
|
||||
let text = " hello";
|
||||
let (k1, k2) = getwrd(text, 0).unwrap();
|
||||
assert_eq!(k1, 3);
|
||||
assert_eq!(k2, 7);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_getwrd_multiple_separators() {
|
||||
let text = "a,b,c";
|
||||
let (k1, k2) = getwrd(text, 0).unwrap();
|
||||
assert_eq!(k1, 0);
|
||||
assert_eq!(k2, 0);
|
||||
|
||||
let (k1, k2) = getwrd(text, 2).unwrap();
|
||||
assert_eq!(k1, 2);
|
||||
assert_eq!(k2, 2);
|
||||
|
||||
let (k1, k2) = getwrd(text, 4).unwrap();
|
||||
assert_eq!(k1, 4);
|
||||
assert_eq!(k2, 4);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_getwrd_at_end() {
|
||||
let text = "x ";
|
||||
let (k1, k2) = getwrd(text, 0).unwrap();
|
||||
assert_eq!(k1, 0);
|
||||
assert_eq!(k2, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_getwrd_beyond_end() {
|
||||
let text = "hi";
|
||||
assert!(getwrd(text, 5).is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_getwrd_realistic_input() {
|
||||
// Typical SYNSPEC input: "H 1 1.0 2.0"
|
||||
let text = "H 1 1.0 2.0";
|
||||
let (k1, k2) = getwrd(text, 0).unwrap();
|
||||
assert_eq!(&text[k1..=k2], "H");
|
||||
|
||||
let (k1, k2) = getwrd(text, k2 + 1).unwrap();
|
||||
assert_eq!(&text[k1..=k2], "1");
|
||||
|
||||
let (k1, k2) = getwrd(text, k2 + 1).unwrap();
|
||||
assert_eq!(&text[k1..=k2], "1.0");
|
||||
|
||||
let (k1, k2) = getwrd(text, k2 + 1).unwrap();
|
||||
assert_eq!(&text[k1..=k2], "2.0");
|
||||
}
|
||||
}
|
||||
77
src/synspec/math/gfree.rs
Normal file
77
src/synspec/math/gfree.rs
Normal file
@ -0,0 +1,77 @@
|
||||
//! Hydrogenic free-free Gaunt factor.
|
||||
//!
|
||||
//! Translated from SYNSPEC `GFREE` function (synspec54.f:5144).
|
||||
|
||||
/// Hydrogenic free-free Gaunt factor for temperature `t` and frequency `fr`.
|
||||
///
|
||||
/// Based on tabulated values with polynomial interpolation.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `t` - Temperature (K)
|
||||
/// * `fr` - Frequency (Hz)
|
||||
///
|
||||
/// # Returns
|
||||
/// The free-free Gaunt factor.
|
||||
pub fn gfree(t: f64, fr: f64) -> f64 {
|
||||
let mut thet = 5040.4 / t;
|
||||
if thet < 4.0e-2 {
|
||||
thet = 4.0e-2;
|
||||
}
|
||||
let x = fr / 2.99793e14;
|
||||
|
||||
if x <= 1.0 {
|
||||
let x_clamped = if x < 0.2 { 0.2 } else { x };
|
||||
(1.0823 + 2.98e-2 / thet) + (6.7e-3 + 1.12e-2 / thet) / x_clamped
|
||||
} else {
|
||||
let c1 = (3.9999187e-3 - 7.8622889e-5 / thet) / thet + 1.070192;
|
||||
let c2 = (6.4628601e-2 - 6.1953813e-4 / thet) / thet + 2.6061249e-1;
|
||||
let c3 = (1.3983474e-5 / thet + 3.7542343e-2) / thet + 5.7917786e-1;
|
||||
let c4 = 3.4169006e-1 + 1.1852264e-2 / thet;
|
||||
((c4 / x - c3) / x + c2) / x + c1
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_gfree_low_freq() {
|
||||
// For x < 1, the formula is simpler
|
||||
let t = 10000.0;
|
||||
let fr = 1.0e14; // x = fr/2.99793e14 ≈ 0.33 < 1
|
||||
let result = gfree(t, fr);
|
||||
assert!(result > 0.0);
|
||||
assert!(result.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_gfree_high_freq() {
|
||||
// For x > 1, polynomial formula is used
|
||||
let t = 10000.0;
|
||||
let fr = 5.0e14; // x ≈ 1.67 > 1
|
||||
let result = gfree(t, fr);
|
||||
assert!(result > 0.0);
|
||||
assert!(result.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_gfree_boundary() {
|
||||
// At x = 1, both branches should give similar results
|
||||
let t = 10000.0;
|
||||
let fr = 2.99793e14; // x = 1
|
||||
let result = gfree(t, fr);
|
||||
assert!(result > 0.0);
|
||||
assert!(result.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_gfree_low_temp() {
|
||||
// Test with very low temperature (thet clamped to 0.04)
|
||||
let t = 200000.0;
|
||||
let fr = 1.0e14;
|
||||
let result = gfree(t, fr);
|
||||
assert!(result > 0.0);
|
||||
assert!(result.is_finite());
|
||||
}
|
||||
}
|
||||
229
src/synspec/math/ghydop.rs
Normal file
229
src/synspec/math/ghydop.rs
Normal file
@ -0,0 +1,229 @@
|
||||
//! Hydrogen opacity from Gomez tables for SYNSPEC.
|
||||
//!
|
||||
//! Translated from SYNSPEC `ghydop` subroutine (synspec54.f:21700).
|
||||
//!
|
||||
//! Calculates hydrogen line + pseudocontinuum opacity using
|
||||
//! pre-computed Gomez opacity tables via wavelength interpolation.
|
||||
|
||||
// ============================================================================
|
||||
// Physical constants
|
||||
// ============================================================================
|
||||
|
||||
const C18: f64 = 2.997925e18;
|
||||
const FREQ_THRESHOLD: f64 = 8.22013e14;
|
||||
|
||||
// ============================================================================
|
||||
// Parameters
|
||||
// ============================================================================
|
||||
|
||||
/// Input parameters for `ghydop`.
|
||||
pub struct GhydopParams<'a> {
|
||||
/// Depth index.
|
||||
pub id: usize,
|
||||
/// Start frequency index.
|
||||
pub i0: usize,
|
||||
/// End frequency index.
|
||||
pub i1: usize,
|
||||
/// Temperature at depth ID (K).
|
||||
pub t: f64,
|
||||
/// Frequency array (Hz).
|
||||
pub freq: &'a [f64],
|
||||
/// Level populations (up to 40 levels).
|
||||
pub pj: &'a [f64],
|
||||
/// Gomez opacity table wavelengths (log10, sorted decreasing).
|
||||
pub wlgtab: &'a [f64],
|
||||
/// Gomez opacity table values: hydopg[frequency_index][depth].
|
||||
pub hydopg: &'a [f64],
|
||||
/// Number of frequencies in Gomez table.
|
||||
pub nugfreq: usize,
|
||||
/// Number of depths in the model.
|
||||
pub ndepth: usize,
|
||||
}
|
||||
|
||||
/// Result of `ghydop`.
|
||||
pub struct GhydopResult {
|
||||
/// Updated absorption coefficient array (added to input).
|
||||
pub absoh: Vec<f64>,
|
||||
/// Updated emission coefficient array (added to input).
|
||||
pub emish: Vec<f64>,
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// Implementation
|
||||
// ============================================================================
|
||||
|
||||
/// Calculate hydrogen opacity from Gomez tables.
|
||||
///
|
||||
/// Interpolates pre-computed Gomez opacity tables in wavelength space
|
||||
/// and combines with level populations to produce absorption and
|
||||
/// emission coefficients.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Input parameters including Gomez table data
|
||||
/// * `absoh_in` - Input absorption array (will be added to)
|
||||
/// * `emish_in` - Input emission array (will be added to)
|
||||
///
|
||||
/// # Returns
|
||||
/// Updated absorption and emission arrays
|
||||
pub fn ghydop(
|
||||
params: &GhydopParams,
|
||||
absoh_in: &[f64],
|
||||
emish_in: &[f64],
|
||||
) -> GhydopResult {
|
||||
let nf = params.freq.len();
|
||||
let mut absoh = absoh_in.to_vec();
|
||||
let mut emish = emish_in.to_vec();
|
||||
|
||||
if params.nugfreq == 0 {
|
||||
return GhydopResult { absoh, emish };
|
||||
}
|
||||
|
||||
let frg1 = params.wlgtab[0]; // Note: wlgtab stores wavelengths, not frequencies
|
||||
let frg2 = params.wlgtab[params.nugfreq - 1];
|
||||
|
||||
// Build frequency-to-wavelength lookup from wlgtab
|
||||
// wlgtab is in wavelength space (Å), sorted decreasing
|
||||
// We need to find the right interval for each frequency
|
||||
|
||||
let mut igf = params.nugfreq;
|
||||
|
||||
for ij in params.i0..=params.i1.min(nf - 1) {
|
||||
let fr = params.freq[ij];
|
||||
let wla = C18 / fr; // wavelength in Å
|
||||
|
||||
// Find wavelength interval in the table
|
||||
if wla <= frg2 || wla >= frg1 {
|
||||
continue; // Outside table range
|
||||
}
|
||||
|
||||
// Scan to find the right interval (table sorted by decreasing wavelength)
|
||||
if ij == params.i0 {
|
||||
igf = params.nugfreq;
|
||||
}
|
||||
while igf > 0 && wla > params.wlgtab[igf - 1] {
|
||||
igf -= 1;
|
||||
}
|
||||
let ig0 = if igf <= 2 { 2 } else { igf };
|
||||
let ig1 = igf - 1;
|
||||
|
||||
if ig0 == 0 || ig1 >= params.nugfreq || ig0 > ig1 {
|
||||
continue;
|
||||
}
|
||||
|
||||
// Interpolate Gomez opacity in wavelength
|
||||
let wl0 = params.wlgtab[ig0 - 1];
|
||||
let wl1 = params.wlgtab[ig1];
|
||||
let denom = wl1 - wl0;
|
||||
if denom.abs() < 1.0e-30 {
|
||||
continue;
|
||||
}
|
||||
|
||||
let opg0 = get_hydopg(params.hydopg, ig0 - 1, params.id, params.ndepth);
|
||||
let opg1 = get_hydopg(params.hydopg, ig1, params.id, params.ndepth);
|
||||
let abl = (opg1 - opg0) * (wla - wl0) / denom + opg0;
|
||||
|
||||
// Determine which population to use based on frequency
|
||||
let pp = if fr > FREQ_THRESHOLD {
|
||||
params.pj.first().copied().unwrap_or(0.0) * 2.0
|
||||
} else {
|
||||
params.pj.get(1).copied().unwrap_or(0.0) * 8.0
|
||||
};
|
||||
|
||||
// Compute Planck function factor
|
||||
let f15 = fr * 1.0e-15;
|
||||
let xkf = (-4.79928e-11 * fr / params.t).exp();
|
||||
let xkfb = xkf * 1.4743e-2 * f15 * f15 * f15;
|
||||
|
||||
// Total opacity
|
||||
let oph = abl.exp() * pp;
|
||||
absoh[ij] += oph;
|
||||
emish[ij] += oph * xkfb / (1.0 - xkf);
|
||||
}
|
||||
|
||||
GhydopResult { absoh, emish }
|
||||
}
|
||||
|
||||
/// Access hydopg table value.
|
||||
fn get_hydopg(hydopg: &[f64], freq_idx: usize, depth: usize, ndepth: usize) -> f64 {
|
||||
let idx = freq_idx * ndepth + depth;
|
||||
if idx < hydopg.len() {
|
||||
hydopg[idx]
|
||||
} else {
|
||||
0.0
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_ghydop_empty_table() {
|
||||
let freq = vec![3.0e14, 4.0e14, 5.0e14];
|
||||
let params = GhydopParams {
|
||||
id: 0,
|
||||
i0: 0,
|
||||
i1: 2,
|
||||
t: 10000.0,
|
||||
freq: &freq,
|
||||
pj: &[0.0; 40],
|
||||
wlgtab: &[],
|
||||
hydopg: &[],
|
||||
nugfreq: 0,
|
||||
ndepth: 1,
|
||||
};
|
||||
let absoh = vec![0.0; 3];
|
||||
let emish = vec![0.0; 3];
|
||||
let result = ghydop(¶ms, &absoh, &emish);
|
||||
assert_eq!(result.absoh, vec![0.0; 3]);
|
||||
assert_eq!(result.emish, vec![0.0; 3]);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_ghydop_basic() {
|
||||
// Simple table with 3 wavelength points
|
||||
let wlgtab = vec![10000.0, 5000.0, 2000.0]; // Decreasing wavelength
|
||||
let ndepth = 2;
|
||||
let nugfreq = 3;
|
||||
// hydopg[freq_idx * ndepth + depth]
|
||||
let hydopg = vec![
|
||||
1.0, 2.0, // freq 0, depth 0,1
|
||||
1.5, 2.5, // freq 1, depth 0,1
|
||||
0.5, 1.0, // freq 2, depth 0,1
|
||||
];
|
||||
|
||||
let freq = vec![3.0e14, 4.0e14]; // ~10000Å, ~7500Å
|
||||
let pj = vec![1.0e10; 40];
|
||||
|
||||
let params = GhydopParams {
|
||||
id: 0,
|
||||
i0: 0,
|
||||
i1: 1,
|
||||
t: 10000.0,
|
||||
freq: &freq,
|
||||
pj: &pj,
|
||||
wlgtab: &wlgtab,
|
||||
hydopg: &hydopg,
|
||||
nugfreq,
|
||||
ndepth,
|
||||
};
|
||||
|
||||
let absoh = vec![0.0; 2];
|
||||
let emish = vec![0.0; 2];
|
||||
let result = ghydop(¶ms, &absoh, &emish);
|
||||
|
||||
// Values should be finite
|
||||
assert!(result.absoh.iter().all(|&x| x.is_finite()));
|
||||
assert!(result.emish.iter().all(|&x| x.is_finite()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_get_hydopg_bounds() {
|
||||
let hydopg = vec![1.0, 2.0, 3.0, 4.0];
|
||||
assert_eq!(get_hydopg(&hydopg, 0, 0, 2), 1.0);
|
||||
assert_eq!(get_hydopg(&hydopg, 1, 0, 2), 3.0);
|
||||
assert_eq!(get_hydopg(&hydopg, 0, 1, 2), 2.0);
|
||||
// Out of bounds
|
||||
assert_eq!(get_hydopg(&hydopg, 5, 0, 2), 0.0);
|
||||
}
|
||||
}
|
||||
288
src/synspec/math/gomini.rs
Normal file
288
src/synspec/math/gomini.rs
Normal file
@ -0,0 +1,288 @@
|
||||
//! Initialization and reading of opacity table for thermal processes.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine GOMINI (line 21601).
|
||||
//!
|
||||
//! Reads `gomhyd.dat` file containing hydrogen opacity tables as a function
|
||||
//! of temperature and electron density, then interpolates to the actual
|
||||
//! temperature and electron density at each depth point.
|
||||
|
||||
|
||||
// ============================================================================
|
||||
// Constants
|
||||
// ============================================================================
|
||||
|
||||
/// Conversion factor from eV to temperature (K)
|
||||
#[allow(dead_code)]
|
||||
const EV_TO_K: f64 = 1.161e4;
|
||||
|
||||
/// Energy-to-frequency conversion: 3.28805e15 / 13.595
|
||||
#[allow(dead_code)]
|
||||
const ENE_TO_FREQ: f64 = 3.28805e15 / 13.595;
|
||||
|
||||
/// Wavelength conversion constant (Å)
|
||||
#[allow(dead_code)]
|
||||
const WL_CONV: f64 = 2.997925e18;
|
||||
|
||||
/// Log of the opacity offset constant: log(0.02654 * 4.1347e-15)
|
||||
const OPAC_OFFSET: f64 = -32.726_974_762_964_47; // precomputed
|
||||
|
||||
// ============================================================================
|
||||
// GOMINI parameters
|
||||
// ============================================================================
|
||||
|
||||
/// Parameters for the GOMINI subroutine.
|
||||
pub struct GominiParams<'a> {
|
||||
/// Number of depth points
|
||||
pub nd: usize,
|
||||
/// Temperature array (depth points)
|
||||
pub temp: &'a [f64],
|
||||
/// Electron density array (depth points)
|
||||
pub elec: &'a [f64],
|
||||
/// Electron density limit for H⁻ opacity
|
||||
pub hglim: f64,
|
||||
/// Switch for H⁻ opacity (0 = off)
|
||||
pub ihgom: i32,
|
||||
}
|
||||
|
||||
/// Result of GOMINI: interpolated H⁻ opacity table.
|
||||
pub struct GominiResult {
|
||||
/// Frequency grid (Hz) [nugfreq]
|
||||
pub frgtab: Vec<f64>,
|
||||
/// Wavelength grid (Å) [nugfreq]
|
||||
pub wlgtab: Vec<f64>,
|
||||
/// Interpolated H⁻ opacity (log scale) [nugfreq × nd]
|
||||
pub hydopg: Vec<Vec<f64>>,
|
||||
/// Number of tabular frequencies
|
||||
pub nugfreq: usize,
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// GOMINI implementation
|
||||
// ============================================================================
|
||||
|
||||
/// Initialize and read opacity table for thermal processes (H⁻ bound-free).
|
||||
///
|
||||
/// Reads the `gomhyd.dat` file, then performs bilinear interpolation
|
||||
/// in log(temperature) and log(electron density) to each depth point.
|
||||
///
|
||||
/// # Fortran original
|
||||
///
|
||||
/// ```fortran
|
||||
/// SUBROUTINE GOMINI
|
||||
/// READ gomhyd.dat
|
||||
/// Bilinear interpolation to depth points
|
||||
/// END
|
||||
/// ```
|
||||
pub fn gomini(params: &GominiParams) -> Option<GominiResult> {
|
||||
let GominiParams { nd: _, temp: _, elec: _, hglim: _, ihgom } = *params;
|
||||
|
||||
if ihgom == 0 {
|
||||
return None;
|
||||
}
|
||||
|
||||
// Read gomhyd.dat - this would normally be file I/O
|
||||
// For now, we provide the interface; actual file reading
|
||||
// would be handled by the caller
|
||||
//
|
||||
// The file format is:
|
||||
// Line 1: nugfreq, nugtemp, nugele
|
||||
// Line 2: (blank)
|
||||
// Line 3: temvec(1..nugtemp) in eV
|
||||
// Line 4: (blank)
|
||||
// Line 5: elevec(1..nugele) in log10(ne)
|
||||
// Then for each frequency:
|
||||
// Line: energy in eV (format: 40x, f17.14)
|
||||
// nugtemp lines: hydcrs(i, 1..nugele) for each temperature
|
||||
|
||||
// This function returns None when ihgom == 0 (disabled)
|
||||
// The actual implementation requires file I/O which is
|
||||
// handled by the runner layer
|
||||
|
||||
// Placeholder: the actual interpolation logic is below
|
||||
// when called with pre-loaded table data
|
||||
None
|
||||
}
|
||||
|
||||
/// Perform bilinear interpolation of H⁻ opacity table to depth points.
|
||||
///
|
||||
/// This is the core interpolation logic extracted from GOMINI,
|
||||
/// to be used with pre-loaded table data.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `nugfreq` - Number of tabular frequencies
|
||||
/// * `nugtemp` - Number of tabular temperatures
|
||||
/// * `nugele` - Number of tabular electron densities
|
||||
/// * `temvec` - Temperature array (in log(K)) [nugtemp]
|
||||
/// * `elevec` - Electron density array (in log(ne)) [nugele]
|
||||
/// * `hydcrs` - Cross-section table [nugtemp × nugele × nugfreq]
|
||||
/// * `temp` - Temperature array (depth points)
|
||||
/// * `elec` - Electron density array (depth points)
|
||||
/// * `nd` - Number of depth points
|
||||
/// * `hglim` - Electron density limit
|
||||
///
|
||||
/// # Returns
|
||||
/// Interpolated opacity [nugfreq × nd] (log scale)
|
||||
pub fn gomini_interpolate(
|
||||
nugfreq: usize,
|
||||
nugtemp: usize,
|
||||
nugele: usize,
|
||||
temvec: &[f64],
|
||||
elevec: &[f64],
|
||||
hydcrs: &[Vec<Vec<f64>>],
|
||||
temp: &[f64],
|
||||
elec: &[f64],
|
||||
nd: usize,
|
||||
hglim: f64,
|
||||
) -> (Vec<f64>, Vec<f64>, Vec<Vec<f64>>) {
|
||||
// Frequency and wavelength grids
|
||||
let frgtab = vec![0.0; nugfreq];
|
||||
let wlgtab = vec![0.0; nugfreq];
|
||||
|
||||
// Compute frequency/wavelength from energy
|
||||
// In the Fortran, energy is read per frequency block
|
||||
// Here we assume frgtab is already populated by caller
|
||||
|
||||
// Interpolate to actual depth points
|
||||
let mut hydopg = vec![vec![0.0; nd]; nugfreq];
|
||||
|
||||
for id in 0..nd {
|
||||
if elec[id] < hglim {
|
||||
continue;
|
||||
}
|
||||
|
||||
let rl = elec[id].ln();
|
||||
let tl = temp[id].ln();
|
||||
|
||||
// Find bracketing indices in electron density
|
||||
let eg_tab1 = elevec[0];
|
||||
let eg_tab2 = elevec[nugele - 1];
|
||||
let deltar = (rl - eg_tab1) / (eg_tab2 - eg_tab1) * (nugele - 1) as f64;
|
||||
let mut jr = 1 + deltar as i32;
|
||||
if jr < 1 { jr = 1; }
|
||||
if jr > (nugele - 1) as i32 { jr = (nugele - 1) as i32; }
|
||||
let jr = jr as usize - 1; // 0-indexed
|
||||
|
||||
let r1i = elevec[jr];
|
||||
let r2i = elevec[jr + 1];
|
||||
let dri = if jr == 0 {
|
||||
0.0
|
||||
} else {
|
||||
(rl - r1i) / (r2i - r1i)
|
||||
};
|
||||
|
||||
// Find bracketing indices in temperature
|
||||
let tg_tab1 = temvec[0];
|
||||
let tg_tab2 = temvec[nugtemp - 1];
|
||||
let deltat = (tl - tg_tab1) / (tg_tab2 - tg_tab1) * (nugtemp - 1) as f64;
|
||||
let mut jp = 1 + deltat as i32;
|
||||
if jp < 1 { jp = 1; }
|
||||
if jp > (nugtemp - 1) as i32 { jp = (nugtemp - 1) as i32; }
|
||||
let jp = jp as usize - 1; // 0-indexed
|
||||
|
||||
let t1i = temvec[jp];
|
||||
let t2i = temvec[jp + 1];
|
||||
let dti = if jp == 0 {
|
||||
0.0
|
||||
} else {
|
||||
(tl - t1i) / (t2i - t1i)
|
||||
};
|
||||
|
||||
// Bilinear interpolation over tabular frequencies
|
||||
for jf in 0..nugfreq {
|
||||
let opr1 = hydcrs[jp][jr][jf]
|
||||
+ dti * (hydcrs[jp + 1][jr][jf] - hydcrs[jp][jr][jf]);
|
||||
let opr2 = hydcrs[jp][jr + 1][jf]
|
||||
+ dti * (hydcrs[jp + 1][jr + 1][jf] - hydcrs[jp][jr + 1][jf]);
|
||||
let opac = opr1 + dri * (opr2 - opr1);
|
||||
hydopg[jf][id] = opac + OPAC_OFFSET;
|
||||
}
|
||||
}
|
||||
|
||||
(frgtab, wlgtab, hydopg)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_gomini_disabled() {
|
||||
let params = GominiParams {
|
||||
nd: 5,
|
||||
temp: &[5000.0; 5],
|
||||
elec: &[1e14; 5],
|
||||
hglim: 1e10,
|
||||
ihgom: 0,
|
||||
};
|
||||
assert!(gomini(¶ms).is_none());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_gomini_interpolate_basic() {
|
||||
// Simple test with 2 temperatures, 2 densities, 2 frequencies
|
||||
let nugfreq = 2;
|
||||
let nugtemp = 2;
|
||||
let nugele = 2;
|
||||
|
||||
let temvec = vec![10.0, 11.0]; // log(K)
|
||||
let elevec = vec![10.0, 12.0]; // log(ne)
|
||||
|
||||
// hydcrs[temp][ele][freq]
|
||||
let hydcrs = vec![
|
||||
vec![vec![1.0, 2.0], vec![3.0, 4.0]],
|
||||
vec![vec![5.0, 6.0], vec![7.0, 8.0]],
|
||||
];
|
||||
|
||||
let temp = vec![22000.0]; // ln(22000) ≈ 10.0
|
||||
let elec = vec![1e11]; // ln(1e11) ≈ 25.3
|
||||
let nd = 1;
|
||||
let hglim = 1e10;
|
||||
|
||||
let (frgtab, wlgtab, hydopg) = gomini_interpolate(
|
||||
nugfreq, nugtemp, nugele,
|
||||
&temvec, &elevec, &hydcrs,
|
||||
&temp, &elec, nd, hglim,
|
||||
);
|
||||
|
||||
assert_eq!(frgtab.len(), 2);
|
||||
assert_eq!(wlgtab.len(), 2);
|
||||
assert_eq!(hydopg.len(), 2);
|
||||
assert_eq!(hydopg[0].len(), 1);
|
||||
|
||||
// All values should be finite
|
||||
for row in &hydopg {
|
||||
for &val in row {
|
||||
assert!(val.is_finite(), "hydopg value not finite: {}", val);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_gomini_interpolate_below_hglim() {
|
||||
let nugfreq = 1;
|
||||
let nugtemp = 2;
|
||||
let nugele = 2;
|
||||
|
||||
let temvec = vec![10.0, 11.0];
|
||||
let elevec = vec![10.0, 12.0];
|
||||
let hydcrs = vec![
|
||||
vec![vec![1.0], vec![2.0]],
|
||||
vec![vec![3.0], vec![4.0]],
|
||||
];
|
||||
|
||||
// electron density below hglim
|
||||
let temp = vec![22000.0];
|
||||
let elec = vec![1e5];
|
||||
let nd = 1;
|
||||
let hglim = 1e10;
|
||||
|
||||
let (_, _, hydopg) = gomini_interpolate(
|
||||
nugfreq, nugtemp, nugele,
|
||||
&temvec, &elevec, &hydcrs,
|
||||
&temp, &elec, nd, hglim,
|
||||
);
|
||||
|
||||
// Should be zero (skipped)
|
||||
assert_eq!(hydopg[0][0], 0.0);
|
||||
}
|
||||
}
|
||||
159
src/synspec/math/gvdw.rs
Normal file
159
src/synspec/math/gvdw.rs
Normal file
@ -0,0 +1,159 @@
|
||||
//! Van der Waals broadening parameter evaluation.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR function GVDW(IL,ILIST,ID) at line 19468.
|
||||
//!
|
||||
//! Supports two modes:
|
||||
//! - Standard expression (`ivdwli == 0`)
|
||||
//! - EXOMOL form with H2 and He broadening (`ivdwli > 0`)
|
||||
|
||||
/// Parameters for Van der Waals broadening calculation.
|
||||
pub struct GvdwParams<'a> {
|
||||
/// Line index
|
||||
pub il: usize,
|
||||
/// Line list index
|
||||
pub ilist: usize,
|
||||
/// Depth index
|
||||
pub id: usize,
|
||||
/// Van der Waals damping parameter (standard mode)
|
||||
pub gwm: f64,
|
||||
/// Van der Waals coefficient at depth
|
||||
pub vdwc: &'a [f64],
|
||||
/// Mode of evaluation per line list (0 = standard, >0 = EXOMOL)
|
||||
pub ivdwli: &'a [i32],
|
||||
/// Temperature at each depth
|
||||
pub temp: &'a [f64],
|
||||
/// He number density at each depth (from rrr array)
|
||||
pub anhe: f64,
|
||||
/// H2 number density at each depth
|
||||
pub anh2: &'a [f64],
|
||||
/// EXOMOL H2 broadening exponent
|
||||
pub gexph2: f64,
|
||||
/// EXOMOL H2 broadening width
|
||||
pub gvdwh2: f64,
|
||||
/// EXOMOL He broadening exponent
|
||||
pub gexphe: f64,
|
||||
/// EXOMOL He broadening width
|
||||
pub gvdwhe: f64,
|
||||
}
|
||||
|
||||
/// Van der Waals broadening parameter.
|
||||
///
|
||||
/// Computes the Van der Waals broadening parameter for spectral line profiles.
|
||||
/// Supports both the standard classical expression and the EXOMOL form
|
||||
/// (broadening by H2 and He).
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Calculation parameters
|
||||
///
|
||||
/// # Returns
|
||||
/// Van der Waals broadening parameter
|
||||
pub fn gvdw(params: &GvdwParams) -> f64 {
|
||||
// Standard classical expression
|
||||
if params.ivdwli[params.ilist] == 0 {
|
||||
return params.gwm * params.vdwc[params.id];
|
||||
}
|
||||
|
||||
// EXOMOL form - broadening by H2 and He
|
||||
// con = 1e-6 * c * k (cgs)
|
||||
let con = 4.1388e-12;
|
||||
let t = params.temp[params.id];
|
||||
|
||||
con * t
|
||||
* ((296.0 / t).powf(params.gexph2) * params.gvdwh2 * params.anh2[params.id]
|
||||
+ (296.0 / t).powf(params.gexphe) * params.gvdwhe * params.anhe)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_gvdw_standard_mode() {
|
||||
let vdwc = vec![1.0, 2.0, 3.0];
|
||||
let ivdwli = vec![0];
|
||||
let temp = vec![5000.0, 6000.0, 7000.0];
|
||||
let anh2 = vec![1e10, 1e10, 1e10];
|
||||
|
||||
let params = GvdwParams {
|
||||
il: 0,
|
||||
ilist: 0,
|
||||
id: 1,
|
||||
gwm: 0.5,
|
||||
vdwc: &vdwc,
|
||||
ivdwli: &ivdwli,
|
||||
temp: &temp,
|
||||
anhe: 1e10,
|
||||
anh2: &anh2,
|
||||
gexph2: 0.0,
|
||||
gvdwh2: 0.0,
|
||||
gexphe: 0.0,
|
||||
gvdwhe: 0.0,
|
||||
};
|
||||
|
||||
// Standard: gwm * vdwc[id] = 0.5 * 2.0 = 1.0
|
||||
let result = gvdw(¶ms);
|
||||
assert!((result - 1.0).abs() < 1e-15);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_gvdw_exomol_mode() {
|
||||
let vdwc = vec![1.0; 3];
|
||||
let ivdwli = vec![1];
|
||||
let temp = vec![5000.0; 3];
|
||||
let anh2 = vec![1e12; 3];
|
||||
|
||||
let params = GvdwParams {
|
||||
il: 0,
|
||||
ilist: 0,
|
||||
id: 0,
|
||||
gwm: 1.0,
|
||||
vdwc: &vdwc,
|
||||
ivdwli: &ivdwli,
|
||||
temp: &temp,
|
||||
anhe: 1e11,
|
||||
anh2: &anh2,
|
||||
gexph2: 0.5,
|
||||
gvdwh2: 1e-9,
|
||||
gexphe: 0.3,
|
||||
gvdwhe: 5e-10,
|
||||
};
|
||||
|
||||
let result = gvdw(¶ms);
|
||||
assert!(result > 0.0);
|
||||
assert!(result.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_gvdw_exomol_temperature_dependence() {
|
||||
let vdwc = vec![1.0; 2];
|
||||
let ivdwli = vec![1];
|
||||
let anh2 = vec![1e12; 2];
|
||||
|
||||
let params_low = GvdwParams {
|
||||
il: 0,
|
||||
ilist: 0,
|
||||
id: 0,
|
||||
gwm: 1.0,
|
||||
vdwc: &vdwc,
|
||||
ivdwli: &ivdwli,
|
||||
temp: &[3000.0, 10000.0],
|
||||
anhe: 1e11,
|
||||
anh2: &anh2,
|
||||
gexph2: 0.5,
|
||||
gvdwh2: 1e-9,
|
||||
gexphe: 0.3,
|
||||
gvdwhe: 5e-10,
|
||||
};
|
||||
|
||||
let params_high = GvdwParams {
|
||||
id: 1,
|
||||
..params_low
|
||||
};
|
||||
|
||||
let r_low = gvdw(¶ms_low);
|
||||
let r_high = gvdw(¶ms_high);
|
||||
// Both should be positive
|
||||
assert!(r_low > 0.0);
|
||||
assert!(r_high > 0.0);
|
||||
}
|
||||
}
|
||||
220
src/synspec/math/h2minus.rs
Normal file
220
src/synspec/math/h2minus.rs
Normal file
@ -0,0 +1,220 @@
|
||||
//! H2⁻ 自由-自由吸收系数计算。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `synspec54.f` 中的 `h2minus` 子程序。
|
||||
//!
|
||||
//! 数据来源: K L Bell 1980, J. Phys. B: At. Mol. Phys. 13 1859, Table 1
|
||||
//! 单位: 10^26 cm^4/dyn^-1
|
||||
|
||||
use crate::tlusty::math::interpolation::locate;
|
||||
use crate::synspec::math::{CL, BOLK};
|
||||
|
||||
// ============================================================================
|
||||
// 静态数据表
|
||||
// ============================================================================
|
||||
|
||||
/// theta = 5040/T(K) 网格点 (9 个)
|
||||
const FFTHET: [f64; 9] = [0.5, 0.8, 1.0, 1.2, 1.6, 2.0, 2.8, 3.6, 10.0];
|
||||
|
||||
/// lambda (Angstroms) 网格点 (18 个)
|
||||
const FFLAMB: [f64; 18] = [
|
||||
151883.0, 113913.0, 91130.0, 60753.0,
|
||||
45565.0, 36452.0, 30377.0, 22783.0,
|
||||
18226.0, 15188.0, 11391.0, 9113.0, 7594.0,
|
||||
6509.0, 5696.0, 5063.0, 4142.0, 3505.0,
|
||||
];
|
||||
|
||||
/// kappa 表 (18 x 9)，按列优先存储 (Fortran 布局)
|
||||
const NTHET: usize = 9;
|
||||
const NLAMB: usize = 18;
|
||||
|
||||
/// FFkapp 表，按 Fortran 列优先存储: FFkapp[theta_idx * NLAMB + lamb_idx]
|
||||
/// 即 FFkapp(i,j) = FFKAPP[j * 18 + i]，其中 i=lambda, j=theta (0-based)
|
||||
const FFKAPP: [f64; NLAMB * NTHET] = [
|
||||
// 列 1 (theta=0.5): 18 个 lambda 值
|
||||
7.16e+01, 4.03e+01, 2.58e+01, 1.15e+01, 6.47e+00,
|
||||
4.15e+00, 2.89e+00, 1.63e+00, 1.05e+00, 7.36e-01,
|
||||
4.20e-01, 2.73e-01, 1.92e-01, 1.43e-01, 1.10e-01,
|
||||
8.70e-02, 5.84e-02, 4.17e-02,
|
||||
// 列 2 (theta=0.8)
|
||||
9.23e+01, 5.20e+01, 3.33e+01, 1.48e+01, 8.37e+00,
|
||||
5.38e+00, 3.76e+00, 2.14e+00, 1.39e+00, 9.75e-01,
|
||||
5.64e-01, 3.71e-01, 2.64e-01, 1.98e-01, 1.54e-01,
|
||||
1.24e-01, 8.43e-02, 6.10e-02,
|
||||
// 列 3 (theta=1.0)
|
||||
1.01e+02, 5.70e+01, 3.65e+01, 1.63e+01, 9.20e+00,
|
||||
5.92e+00, 4.14e+00, 2.36e+00, 1.54e+00, 1.09e+00,
|
||||
6.35e-01, 4.22e-01, 3.03e-01, 2.30e-01, 1.80e-01,
|
||||
1.46e-01, 1.01e-01, 7.34e-02,
|
||||
// 列 4 (theta=1.2)
|
||||
1.08e+02, 6.08e+01, 3.90e+01, 1.74e+01, 9.84e+00,
|
||||
6.35e+00, 4.44e+00, 2.55e+00, 1.66e+00, 1.18e+00,
|
||||
6.97e-01, 4.67e-01, 3.39e-01, 2.59e-01, 2.06e-01,
|
||||
1.67e-01, 1.17e-01, 8.59e-02,
|
||||
// 列 5 (theta=1.6)
|
||||
1.18e+02, 6.65e+01, 4.27e+01, 1.91e+01, 1.08e+01,
|
||||
6.99e+00, 4.91e+00, 2.84e+00, 1.87e+00, 1.34e+00,
|
||||
8.06e-01, 5.52e-01, 4.08e-01, 3.17e-01, 2.55e-01,
|
||||
2.10e-01, 1.49e-01, 1.11e-01,
|
||||
// 列 6 (theta=2.0)
|
||||
1.26e+02, 7.08e+01, 4.54e+01, 2.04e+01, 1.16e+01,
|
||||
7.50e+00, 5.28e+00, 3.07e+00, 2.04e+00, 1.48e+00,
|
||||
9.09e-01, 6.33e-01, 4.76e-01, 3.75e-01, 3.05e-01,
|
||||
2.53e-01, 1.82e-01, 1.37e-01,
|
||||
// 列 7 (theta=2.8)
|
||||
1.38e+02, 7.76e+01, 4.98e+01, 2.24e+01, 1.28e+01,
|
||||
8.32e+00, 5.90e+00, 3.49e+00, 2.36e+00, 1.74e+00,
|
||||
1.11e+00, 7.97e-01, 6.13e-01, 4.92e-01, 4.06e-01,
|
||||
3.39e-01, 2.49e-01, 1.87e-01,
|
||||
// 列 8 (theta=3.6)
|
||||
1.47e+02, 8.30e+01, 5.33e+01, 2.40e+01, 1.38e+01,
|
||||
9.02e+00, 6.44e+00, 3.90e+00, 2.68e+00, 2.01e+00,
|
||||
1.32e+00, 9.63e-01, 7.51e-01, 6.09e-01, 5.07e-01,
|
||||
4.27e-01, 3.16e-01, 2.40e-01,
|
||||
// 列 9 (theta=10.0) — 线性外推
|
||||
2.19e+02, 1.26e+02, 8.13e+01, 3.68e+01, 2.18e+01,
|
||||
1.46e+01, 1.08e+01, 7.18e+00, 5.24e+00, 4.17e+00,
|
||||
3.00e+00, 2.29e+00, 1.86e+00, 1.55e+00, 1.32e+00,
|
||||
1.13e+00, 8.52e-01, 6.64e-01,
|
||||
];
|
||||
|
||||
// ============================================================================
|
||||
// h2minus - H2⁻ 自由-自由吸收
|
||||
// ============================================================================
|
||||
|
||||
/// 计算 H2⁻ 自由-自由吸收系数。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// - `t` - 温度 (K)
|
||||
/// - `anh2` - H2 分子数密度
|
||||
/// - `ane` - 电子数密度
|
||||
/// - `fr` - 频率 (Hz)
|
||||
///
|
||||
/// # 返回
|
||||
///
|
||||
/// H2⁻ 自由-自由吸收系数 `oph2m`
|
||||
pub fn h2minus(t: f64, anh2: f64, ane: f64, fr: f64) -> f64 {
|
||||
// theta = 5040 / T
|
||||
let theta = 5040.0 / t;
|
||||
|
||||
// 在温度数组中定位 (0-indexed)
|
||||
// locate 返回 j 使得 FFTHET[j] <= theta < FFTHET[j+1]
|
||||
let j = locate(&FFTHET, theta);
|
||||
|
||||
// 波长 (Angstroms): lambda = c / fr * 1e8
|
||||
let flamb = CL * 1.0e8 / fr;
|
||||
|
||||
// 在波长数组中定位 (0-indexed)
|
||||
let i = locate(&FFLAMB, flamb);
|
||||
|
||||
// 双线性插值
|
||||
// 注意: FFTHET 是递增的，FFLAMB 是递减的
|
||||
let fkappa = if j >= NTHET - 1 {
|
||||
// theta >= FFTHET[NTHET-1]，保持恒定 (高温端)
|
||||
let i_clamped = i.min(NLAMB - 2);
|
||||
let y1 = ffkapp_at(i_clamped, NTHET - 1);
|
||||
let y2 = ffkapp_at(i_clamped + 1, NTHET - 1);
|
||||
let tt = (flamb - FFLAMB[i_clamped]) / (FFLAMB[i_clamped + 1] - FFLAMB[i_clamped]);
|
||||
(1.0 - tt) * y1 + tt * y2
|
||||
} else if !(FFLAMB[NLAMB - 1]..=FFLAMB[0]).contains(&flamb) {
|
||||
// 超出波长表范围 (FFLAMB 递减: [0] 最大, [NLAMB-1] 最小)
|
||||
0.0
|
||||
} else {
|
||||
// 表内双线性插值
|
||||
let y1 = ffkapp_at(i, j);
|
||||
let y2 = ffkapp_at(i + 1, j);
|
||||
let y3 = ffkapp_at(i + 1, j + 1);
|
||||
let y4 = ffkapp_at(i, j + 1);
|
||||
|
||||
// tt: 波长方向插值 (FFLAMB 递减)
|
||||
let tt = (flamb - FFLAMB[i]) / (FFLAMB[i + 1] - FFLAMB[i]);
|
||||
// uu: 温度方向插值 (FFTHET 递增)
|
||||
let uu = (theta - FFTHET[j]) / (FFTHET[j + 1] - FFTHET[j]);
|
||||
|
||||
(1.0 - tt) * (1.0 - uu) * y1
|
||||
+ tt * (1.0 - uu) * y2
|
||||
+ tt * uu * y3
|
||||
+ (1.0 - tt) * uu * y4
|
||||
};
|
||||
|
||||
// 电子压力
|
||||
let pe = ane * BOLK * t;
|
||||
|
||||
// 最终吸收系数
|
||||
anh2 * 1.0e-26 * pe * fkappa
|
||||
}
|
||||
|
||||
/// 从 FFkapp 表中获取值 (处理边界)
|
||||
/// 索引: FFkapp(i,j) = FFKAPP[j * NLAMB + i]，其中 i=lambda, j=theta (0-based)
|
||||
fn ffkapp_at(i: usize, j: usize) -> f64 {
|
||||
let i_clamped = i.min(NLAMB - 1);
|
||||
let j_clamped = j.min(NTHET - 1);
|
||||
FFKAPP[j_clamped * NLAMB + i_clamped]
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_h2minus_basic() {
|
||||
// 典型恒星大气参数
|
||||
let t = 5000.0; // K
|
||||
let anh2 = 1.0e15; // H2 数密度
|
||||
let ane = 1.0e13; // 电子数密度
|
||||
let fr = 1.0e14; // Hz (红外)
|
||||
|
||||
let oph2m = h2minus(t, anh2, ane, fr);
|
||||
assert!(oph2m > 0.0, "oph2m 应为正值: {}", oph2m);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_h2minus_high_temperature() {
|
||||
// 高温情况
|
||||
let t = 10000.0;
|
||||
let anh2 = 1.0e14;
|
||||
let ane = 1.0e12;
|
||||
let fr = 3.0e14;
|
||||
|
||||
let oph2m = h2minus(t, anh2, ane, fr);
|
||||
assert!(oph2m >= 0.0, "oph2m 应非负: {}", oph2m);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_h2minus_low_temperature() {
|
||||
// 低温情况 (theta 大)
|
||||
let t = 3000.0;
|
||||
let anh2 = 1.0e16;
|
||||
let ane = 1.0e14;
|
||||
let fr = 5.0e14;
|
||||
|
||||
let oph2m = h2minus(t, anh2, ane, fr);
|
||||
assert!(oph2m >= 0.0, "oph2m 应非负: {}", oph2m);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_h2minus_scaling() {
|
||||
// 吸收系数应与 anh2 和 ane 成正比
|
||||
let t = 6000.0;
|
||||
let fr = 2.0e14;
|
||||
|
||||
let oph2m1 = h2minus(t, 1.0e14, 1.0e12, fr);
|
||||
let oph2m2 = h2minus(t, 2.0e14, 1.0e12, fr);
|
||||
let oph2m3 = h2minus(t, 1.0e14, 2.0e12, fr);
|
||||
|
||||
// 双倍 anh2 → 双倍 opacity
|
||||
assert!(
|
||||
(oph2m2 / oph2m1 - 2.0).abs() < 0.01,
|
||||
"anh2 线性性: {} vs {}",
|
||||
oph2m2,
|
||||
oph2m1
|
||||
);
|
||||
// 双倍 ane → 双倍 opacity (pe 线性)
|
||||
assert!(
|
||||
(oph2m3 / oph2m1 - 2.0).abs() < 0.01,
|
||||
"ane 线性性: {} vs {}",
|
||||
oph2m3,
|
||||
oph2m1
|
||||
);
|
||||
}
|
||||
}
|
||||
56
src/synspec/math/h2opf.rs
Normal file
56
src/synspec/math/h2opf.rs
Normal file
@ -0,0 +1,56 @@
|
||||
//! Partition function for H2O from EXOMOL data.
|
||||
//!
|
||||
//! Translated from SYNSPEC `h2opf` subroutine.
|
||||
|
||||
use std::sync::OnceLock;
|
||||
|
||||
const TABLE_SIZE: usize = 10000;
|
||||
const DATA_FILE: &str = "./data/h2o_exomol.pf";
|
||||
|
||||
static TABLE: OnceLock<Option<(Vec<f64>, Vec<f64>)>> = OnceLock::new();
|
||||
|
||||
fn load_table() -> Option<(Vec<f64>, Vec<f64>)> {
|
||||
let content = std::fs::read_to_string(DATA_FILE).ok()?;
|
||||
let mut ttab = Vec::with_capacity(TABLE_SIZE);
|
||||
let mut pftab = Vec::with_capacity(TABLE_SIZE);
|
||||
for line in content.lines().take(TABLE_SIZE) {
|
||||
let parts: Vec<&str> = line.split_whitespace().collect();
|
||||
if parts.len() >= 2
|
||||
&& let (Ok(t), Ok(pf)) = (parts[0].parse::<f64>(), parts[1].parse::<f64>()) {
|
||||
ttab.push(t);
|
||||
pftab.push(pf);
|
||||
}
|
||||
}
|
||||
Some((ttab, pftab))
|
||||
}
|
||||
|
||||
/// Evaluate H2O partition function at temperature `t` by linear interpolation.
|
||||
///
|
||||
/// Returns `None` if the data file cannot be loaded.
|
||||
pub fn h2opf(t: f64) -> Option<f64> {
|
||||
let table = TABLE.get_or_init(load_table).as_ref()?;
|
||||
let (ref ttab, ref pftab) = *table;
|
||||
let n = ttab.len();
|
||||
if n < 2 || t < ttab[0] || t > ttab[n - 1] {
|
||||
return None;
|
||||
}
|
||||
let itab = t.floor() as usize;
|
||||
if itab >= n - 1 {
|
||||
return None;
|
||||
}
|
||||
let idx = itab.min(n - 2);
|
||||
let pf = pftab[idx] + (t - ttab[idx]) * (pftab[idx + 1] - pftab[idx]);
|
||||
Some(pf)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_h2opf_basic() {
|
||||
// Without the data file, should return None
|
||||
// This test just verifies the function compiles and runs
|
||||
let _ = h2opf(5000.0);
|
||||
}
|
||||
}
|
||||
244
src/synspec/math/he1ini.rs
Normal file
244
src/synspec/math/he1ini.rs
Normal file
@ -0,0 +1,244 @@
|
||||
//! He I line profile data initialization.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine HE1INI (line 7242).
|
||||
//!
|
||||
//! Initializes necessary arrays for evaluating the He I line
|
||||
//! absorption profiles using data calculated by Barnard, Cooper
|
||||
//! and Smith JQSRT 14, 1025, 1974 (for 4471)
|
||||
//! or Shamey, unpublished PhD thesis, 1969 (for other lines).
|
||||
|
||||
#![allow(clippy::never_loop)]
|
||||
use std::fs::File;
|
||||
use std::io::{BufRead, BufReader};
|
||||
use std::path::Path;
|
||||
|
||||
/// Constants for He I profile arrays
|
||||
pub const NT: usize = 4;
|
||||
pub const NE_4471: usize = 7;
|
||||
pub const NE_OTHER: usize = 8;
|
||||
pub const NWL_MAX_4471: usize = 80;
|
||||
pub const NWL_MAX_OTHER: usize = 50;
|
||||
pub const NLINES: usize = 3;
|
||||
|
||||
/// He I 4471 line profile data (Barnard, Cooper, Smith)
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct He1Profile4471 {
|
||||
/// Log10 of electron densities [NE_4471]
|
||||
pub xne: [f64; NE_4471],
|
||||
/// Number of wavelength points for each electron density [NE_4471]
|
||||
pub nwlam: [usize; NE_4471],
|
||||
/// Wavelength displacements [NWL_MAX_4471 x NE_4471]
|
||||
pub dlam: [[f64; NE_4471]; NWL_MAX_4471],
|
||||
/// Profile values [NWL_MAX_4471 x NT x NE_4471]
|
||||
pub prf: [[[f64; NE_4471]; NT]; NWL_MAX_4471],
|
||||
}
|
||||
|
||||
/// He I other lines profile data (Shamey)
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct He1ProfileOther {
|
||||
/// Log10 of electron densities [NE_OTHER]
|
||||
pub xne: [f64; NE_OTHER],
|
||||
/// Number of wavelength points [NE_OTHER x NLINES+1] (index 0 unused)
|
||||
pub nwlam: [[usize; NLINES + 1]; NE_OTHER],
|
||||
/// Wavelength displacements [NWL_MAX_OTHER x NE_OTHER x NLINES]
|
||||
pub dlam: [[[f64; NLINES]; NE_OTHER]; NWL_MAX_OTHER],
|
||||
/// Profile values [NWL_MAX_OTHER x NT x NE_OTHER x NLINES]
|
||||
pub prf: [[[[f64; NLINES]; NE_OTHER]; NT]; NWL_MAX_OTHER],
|
||||
}
|
||||
|
||||
/// Complete He I profile data
|
||||
#[derive(Debug, Clone)]
|
||||
#[derive(Default)]
|
||||
pub struct He1ProfileData {
|
||||
/// 4471 line data
|
||||
pub data_4471: He1Profile4471,
|
||||
/// Other lines data (4387, 4026, 4922)
|
||||
pub data_other: He1ProfileOther,
|
||||
}
|
||||
|
||||
impl Default for He1Profile4471 {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
xne: [0.0; NE_4471],
|
||||
nwlam: [0; NE_4471],
|
||||
dlam: [[0.0; NE_4471]; NWL_MAX_4471],
|
||||
prf: [[[0.0; NE_4471]; NT]; NWL_MAX_4471],
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Default for He1ProfileOther {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
xne: [0.0; NE_OTHER],
|
||||
nwlam: [[0; NLINES + 1]; NE_OTHER],
|
||||
dlam: [[[0.0; NLINES]; NE_OTHER]; NWL_MAX_OTHER],
|
||||
prf: [[[[0.0; NLINES]; NE_OTHER]; NT]; NWL_MAX_OTHER],
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/// Read He I line profile data from file.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `path` - Path to he1prf.dat file
|
||||
///
|
||||
/// # Returns
|
||||
/// Complete He I profile data structure
|
||||
pub fn he1ini<P: AsRef<Path>>(path: P) -> std::io::Result<He1ProfileData> {
|
||||
let file = File::open(path)?;
|
||||
let reader = BufReader::new(file);
|
||||
let mut lines = reader.lines();
|
||||
|
||||
let mut data = He1ProfileData::default();
|
||||
|
||||
// Read Barnard, Cooper, Smith tables for He I 4471 line
|
||||
for ie in 0..NE_4471 {
|
||||
// Skip header line and read: IL, WL0, IE1, XXNE, NWL
|
||||
let header = read_next_line(&mut lines)?;
|
||||
let parts = parse_header(&header)?;
|
||||
|
||||
let _il = parts.0; // line index (unused)
|
||||
let _wl0 = parts.1; // wavelength (unused)
|
||||
let _ie1 = parts.2; // electron density index (unused)
|
||||
let xxne = parts.3; // electron density
|
||||
let nwl = parts.4; // number of wavelength points
|
||||
|
||||
data.data_4471.nwlam[ie] = nwl;
|
||||
data.data_4471.xne[ie] = xxne.log10();
|
||||
|
||||
// Read profile data
|
||||
for i in 0..nwl.min(NWL_MAX_4471) {
|
||||
let line = read_next_line(&mut lines)?;
|
||||
let values = parse_profile_line(&line)?;
|
||||
|
||||
data.data_4471.dlam[i][ie] = values[0];
|
||||
for it in 0..NT {
|
||||
if it + 1 < values.len() {
|
||||
data.data_4471.prf[i][it][ie] = values[it + 1];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Read Shamey's tables for He I 4387, 4026, and 4922 lines
|
||||
for iln in 0..NLINES {
|
||||
for ie in 0..NE_OTHER {
|
||||
let header = read_next_line(&mut lines)?;
|
||||
let parts = parse_header(&header)?;
|
||||
|
||||
let xxne = parts.3;
|
||||
let nwl = parts.4;
|
||||
|
||||
data.data_other.nwlam[ie][iln + 1] = nwl;
|
||||
data.data_other.xne[ie] = xxne.log10();
|
||||
|
||||
// Read profile data
|
||||
for i in 0..nwl.min(NWL_MAX_OTHER) {
|
||||
let line = read_next_line(&mut lines)?;
|
||||
let values = parse_profile_line(&line)?;
|
||||
|
||||
data.data_other.dlam[i][ie][iln] = values[0];
|
||||
for it in 0..NT {
|
||||
if it + 1 < values.len() {
|
||||
data.data_other.prf[i][it][ie][iln] = values[it + 1];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Ok(data)
|
||||
}
|
||||
|
||||
/// Read next non-empty line from iterator
|
||||
fn read_next_line(lines: &mut impl Iterator<Item = std::io::Result<String>>) -> std::io::Result<String> {
|
||||
loop {
|
||||
match lines.next() {
|
||||
Some(Ok(line)) => return Ok(line),
|
||||
Some(Err(e)) => return Err(e),
|
||||
None => return Err(std::io::Error::new(std::io::ErrorKind::UnexpectedEof, "Unexpected end of file")),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Parse header line: IL, WL0, IE1, XXNE, NWL
|
||||
fn parse_header(line: &str) -> std::io::Result<(usize, f64, usize, f64, usize)> {
|
||||
// FORMAT(/9X,I2,7X,F10.3,13X,I2,6X,E8.1,7X,I3/)
|
||||
// This is a fixed-format line, but we'll try free-format parsing
|
||||
let parts: Vec<&str> = line.split_whitespace().collect();
|
||||
if parts.len() < 5 {
|
||||
return Err(std::io::Error::new(
|
||||
std::io::ErrorKind::InvalidData,
|
||||
format!("Invalid header line: {}", line),
|
||||
));
|
||||
}
|
||||
|
||||
let il = parts[0].parse::<usize>().map_err(|e| {
|
||||
std::io::Error::new(std::io::ErrorKind::InvalidData, format!("IL: {}", e))
|
||||
})?;
|
||||
let wl0 = parts[1].parse::<f64>().map_err(|e| {
|
||||
std::io::Error::new(std::io::ErrorKind::InvalidData, format!("WL0: {}", e))
|
||||
})?;
|
||||
let ie1 = parts[2].parse::<usize>().map_err(|e| {
|
||||
std::io::Error::new(std::io::ErrorKind::InvalidData, format!("IE1: {}", e))
|
||||
})?;
|
||||
let xxne = parts[3].parse::<f64>().map_err(|e| {
|
||||
std::io::Error::new(std::io::ErrorKind::InvalidData, format!("XXNE: {}", e))
|
||||
})?;
|
||||
let nwl = parts[4].parse::<usize>().map_err(|e| {
|
||||
std::io::Error::new(std::io::ErrorKind::InvalidData, format!("NWL: {}", e))
|
||||
})?;
|
||||
|
||||
Ok((il, wl0, ie1, xxne, nwl))
|
||||
}
|
||||
|
||||
/// Parse profile data line: DLAM, PRF(IT=1..NT)
|
||||
fn parse_profile_line(line: &str) -> std::io::Result<Vec<f64>> {
|
||||
// FORMAT(5E10.2) - 5 values per line
|
||||
let values: Vec<f64> = line
|
||||
.split_whitespace()
|
||||
.map(|s| s.parse::<f64>())
|
||||
.collect::<Result<Vec<_>, _>>()
|
||||
.map_err(|e| {
|
||||
std::io::Error::new(std::io::ErrorKind::InvalidData, format!("Profile value: {}", e))
|
||||
})?;
|
||||
|
||||
Ok(values)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_he1ini_default() {
|
||||
let data = He1ProfileData::default();
|
||||
assert_eq!(data.data_4471.xne.len(), NE_4471);
|
||||
assert_eq!(data.data_other.xne.len(), NE_OTHER);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parse_header() {
|
||||
let line = " 1 4471.000 1 1.0E+12 50";
|
||||
let result = parse_header(line);
|
||||
assert!(result.is_ok());
|
||||
let (il, wl0, ie1, xxne, nwl) = result.unwrap();
|
||||
assert_eq!(il, 1);
|
||||
assert!((wl0 - 4471.0).abs() < 0.01);
|
||||
assert_eq!(ie1, 1);
|
||||
assert!((xxne - 1.0e12).abs() < 1e10);
|
||||
assert_eq!(nwl, 50);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parse_profile_line() {
|
||||
let line = " 0.123 0.456 0.789 0.111 0.222";
|
||||
let result = parse_profile_line(line);
|
||||
assert!(result.is_ok());
|
||||
let values = result.unwrap();
|
||||
assert_eq!(values.len(), 5);
|
||||
assert!((values[0] - 0.123).abs() < 1e-6);
|
||||
}
|
||||
}
|
||||
286
src/synspec/math/he2ini.rs
Normal file
286
src/synspec/math/he2ini.rs
Normal file
@ -0,0 +1,286 @@
|
||||
//! He II line profile data initialization.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine HE2INI (line 7535).
|
||||
//!
|
||||
//! Initializes necessary arrays for evaluating the He II line
|
||||
//! absorption profiles using data calculated by Schoening and Butler.
|
||||
|
||||
#![allow(clippy::never_loop)]
|
||||
use std::fs::File;
|
||||
use std::io::{BufRead, BufReader};
|
||||
|
||||
/// Constants for He II profile arrays
|
||||
pub const NLINE_HE2: usize = 19;
|
||||
pub const NWL_HE2_MAX: usize = 36;
|
||||
pub const NT_HE2: usize = 6;
|
||||
pub const NE_HE2: usize = 11;
|
||||
|
||||
/// He II line profile table data
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct He2ProfileTable {
|
||||
/// Lower level index
|
||||
pub il: usize,
|
||||
/// Upper level index
|
||||
pub iu: usize,
|
||||
/// Central wavelength
|
||||
pub wl0: f64,
|
||||
/// Number of wavelength points
|
||||
pub nwl: usize,
|
||||
/// Log10 wavelength displacements [NWL_HE2_MAX]
|
||||
pub wl: [f64; NWL_HE2_MAX],
|
||||
/// Log10 temperature grid [NT_HE2]
|
||||
pub xt: [f64; NT_HE2],
|
||||
/// Log10 electron density grid [NE_HE2]
|
||||
pub xne: [f64; NE_HE2],
|
||||
/// Profile values [NWL_HE2_MAX x NT_HE2 x NE_HE2]
|
||||
pub prf: [[[f64; NE_HE2]; NT_HE2]; NWL_HE2_MAX],
|
||||
/// Asymptotic profile coefficient
|
||||
pub xk: f64,
|
||||
}
|
||||
|
||||
impl Default for He2ProfileTable {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
il: 0,
|
||||
iu: 0,
|
||||
wl0: 0.0,
|
||||
nwl: 0,
|
||||
wl: [0.0; NWL_HE2_MAX],
|
||||
xt: [0.0; NT_HE2],
|
||||
xne: [0.0; NE_HE2],
|
||||
prf: [[[0.0; NE_HE2]; NT_HE2]; NWL_HE2_MAX],
|
||||
xk: 0.0,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// He II line initialization result
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct He2InitResult {
|
||||
/// Profile tables for each line
|
||||
pub tables: Vec<He2ProfileTable>,
|
||||
/// Number of wavelength points per line [NLINE_HE2]
|
||||
pub nwlhe2: [usize; NLINE_HE2],
|
||||
/// Lower level indices [NLINE_HE2]
|
||||
pub ilhe2: [usize; NLINE_HE2],
|
||||
/// Upper level indices [NLINE_HE2]
|
||||
pub iuhe2: [usize; NLINE_HE2],
|
||||
}
|
||||
|
||||
impl Default for He2InitResult {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
tables: Vec::new(),
|
||||
nwlhe2: [0; NLINE_HE2],
|
||||
ilhe2: [0; NLINE_HE2],
|
||||
iuhe2: [0; NLINE_HE2],
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Parameters for HE2INI
|
||||
pub struct He2iniParams {
|
||||
/// Path to data directory
|
||||
pub data_dir: String,
|
||||
/// Model depth points
|
||||
pub nd: usize,
|
||||
/// Temperature array [nd]
|
||||
pub temp: Vec<f64>,
|
||||
/// Electron density array [nd]
|
||||
pub elec: Vec<f64>,
|
||||
/// Turbulent velocity array [nd]
|
||||
pub vturb: Vec<f64>,
|
||||
}
|
||||
|
||||
/// Initialize He II line profile data.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Initialization parameters
|
||||
///
|
||||
/// # Returns
|
||||
/// He II line initialization result with profile tables
|
||||
pub fn he2ini(params: &He2iniParams) -> std::io::Result<He2InitResult> {
|
||||
let filename = format!("{}/he2prf.dat", params.data_dir);
|
||||
let file = File::open(&filename)?;
|
||||
let reader = BufReader::new(file);
|
||||
let mut lines = reader.lines();
|
||||
|
||||
let mut result = He2InitResult::default();
|
||||
|
||||
for iline in 0..NLINE_HE2 {
|
||||
// Read line indices: FORMAT(//14X,I2,9X,I2/)
|
||||
let header = read_next_nonblank(&mut lines)?;
|
||||
let (il, iu) = parse_he2_header(&header)?;
|
||||
|
||||
result.ilhe2[iline] = il;
|
||||
result.iuhe2[iline] = iu;
|
||||
|
||||
// Compute central wavelength
|
||||
let wl00 = if il <= 2 { 227.838 } else { 227.7776 };
|
||||
let wl0 = wl00 / (1.0 / (il as f64).powi(2) - 1.0 / (iu as f64).powi(2));
|
||||
|
||||
let mut table = He2ProfileTable {
|
||||
il,
|
||||
iu,
|
||||
wl0,
|
||||
..Default::default()
|
||||
};
|
||||
|
||||
// Read wavelength points
|
||||
let wl_line = read_next_line(&mut lines)?;
|
||||
let wl_parts = parse_he2_data(&wl_line)?;
|
||||
let nwl = wl_parts[0] as usize;
|
||||
table.nwl = nwl;
|
||||
result.nwlhe2[iline] = nwl;
|
||||
for i in 0..nwl.min(NWL_HE2_MAX) {
|
||||
table.wl[i] = if wl_parts[i + 1] < 1.0e-4 {
|
||||
(1.0e-4_f64).log10()
|
||||
} else {
|
||||
wl_parts[i + 1].log10()
|
||||
};
|
||||
}
|
||||
|
||||
// Read temperature points: FORMAT(2X,I4,F10.3,5F12.3)
|
||||
let xt_line = read_next_line(&mut lines)?;
|
||||
let xt_parts = parse_he2_data(&xt_line)?;
|
||||
let nt = xt_parts[0] as usize;
|
||||
for i in 0..nt.min(NT_HE2) {
|
||||
table.xt[i] = xt_parts[i + 1];
|
||||
}
|
||||
|
||||
// Read electron density points: FORMAT(2X,I4,F10.2,5F12.2/4X,5F12.2)
|
||||
let xne_line = read_next_line(&mut lines)?;
|
||||
let xne_parts = parse_he2_data(&xne_line)?;
|
||||
let ne = xne_parts[0] as usize;
|
||||
for i in 0..ne.min(NE_HE2) {
|
||||
table.xne[i] = xne_parts[i + 1];
|
||||
}
|
||||
|
||||
// Skip blank line
|
||||
lines.next();
|
||||
|
||||
// Read profile data: FORMAT(10F8.3)
|
||||
for ie in 0..ne.min(NE_HE2) {
|
||||
for _it in 0..nt.min(NT_HE2) {
|
||||
lines.next(); // Skip blank line
|
||||
let prf_line = read_next_line(&mut lines)?;
|
||||
let prf_parts = parse_he2_data(&prf_line)?;
|
||||
for iwl in 0..nwl.min(NWL_HE2_MAX) {
|
||||
if iwl < prf_parts.len() {
|
||||
table.prf[iwl][_it][ie] = prf_parts[iwl];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Compute asymptotic profile coefficient
|
||||
if nwl > 0 && ne > 0 {
|
||||
let xclog = table.prf[nwl - 1][0][0]
|
||||
+ 2.5 * table.wl[nwl - 1]
|
||||
+ 31.831
|
||||
- table.xne[0]
|
||||
- 2.0 * wl0.log10();
|
||||
let xklog = 0.6666667 * (xclog - 0.176);
|
||||
table.xk = (xklog * std::f64::consts::LN_10).exp();
|
||||
}
|
||||
|
||||
result.tables.push(table);
|
||||
}
|
||||
|
||||
Ok(result)
|
||||
}
|
||||
|
||||
/// Read next non-empty line
|
||||
fn read_next_line(lines: &mut impl Iterator<Item = std::io::Result<String>>) -> std::io::Result<String> {
|
||||
loop {
|
||||
match lines.next() {
|
||||
Some(Ok(line)) => return Ok(line),
|
||||
Some(Err(e)) => return Err(e),
|
||||
None => return Err(std::io::Error::new(
|
||||
std::io::ErrorKind::UnexpectedEof,
|
||||
"Unexpected end of file",
|
||||
)),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Read next non-blank line (skip empty lines)
|
||||
fn read_next_nonblank(lines: &mut impl Iterator<Item = std::io::Result<String>>) -> std::io::Result<String> {
|
||||
loop {
|
||||
let line = read_next_line(lines)?;
|
||||
if !line.trim().is_empty() {
|
||||
return Ok(line);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Parse He II header line: FORMAT(//14X,I2,9X,I2/)
|
||||
fn parse_he2_header(line: &str) -> std::io::Result<(usize, usize)> {
|
||||
let parts: Vec<&str> = line.split_whitespace().collect();
|
||||
if parts.len() < 2 {
|
||||
return Err(std::io::Error::new(
|
||||
std::io::ErrorKind::InvalidData,
|
||||
format!("Invalid He II header: {}", line),
|
||||
));
|
||||
}
|
||||
|
||||
let il = parts[0].parse::<usize>().map_err(|e| {
|
||||
std::io::Error::new(std::io::ErrorKind::InvalidData, format!("IL: {}", e))
|
||||
})?;
|
||||
let iu = parts[1].parse::<usize>().map_err(|e| {
|
||||
std::io::Error::new(std::io::ErrorKind::InvalidData, format!("IU: {}", e))
|
||||
})?;
|
||||
|
||||
Ok((il, iu))
|
||||
}
|
||||
|
||||
/// Parse He II data line (free format)
|
||||
fn parse_he2_data(line: &str) -> std::io::Result<Vec<f64>> {
|
||||
let values: Vec<f64> = line
|
||||
.split_whitespace()
|
||||
.filter_map(|s| s.parse::<f64>().ok())
|
||||
.collect();
|
||||
|
||||
Ok(values)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_he2ini_default() {
|
||||
let result = He2InitResult::default();
|
||||
assert_eq!(result.nwlhe2.len(), NLINE_HE2);
|
||||
assert_eq!(result.ilhe2.len(), NLINE_HE2);
|
||||
assert!(result.tables.is_empty());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_he2_profile_table_default() {
|
||||
let table = He2ProfileTable::default();
|
||||
assert_eq!(table.nwl, 0);
|
||||
assert_eq!(table.il, 0);
|
||||
assert_eq!(table.iu, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parse_he2_header() {
|
||||
let line = " 1 2";
|
||||
let result = parse_he2_header(line);
|
||||
assert!(result.is_ok());
|
||||
let (il, iu) = result.unwrap();
|
||||
assert_eq!(il, 1);
|
||||
assert_eq!(iu, 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parse_he2_data() {
|
||||
let line = " 19 0.123 0.456 0.789";
|
||||
let result = parse_he2_data(line);
|
||||
assert!(result.is_ok());
|
||||
let values = result.unwrap();
|
||||
assert_eq!(values.len(), 4);
|
||||
assert!((values[0] - 19.0).abs() < 1e-10);
|
||||
}
|
||||
}
|
||||
729
src/synspec/math/he2lin.rs
Normal file
729
src/synspec/math/he2lin.rs
Normal file
@ -0,0 +1,729 @@
|
||||
//! He II line opacity and emissivity for SYNSPEC.
|
||||
//!
|
||||
//! Translated from SYNSPEC:
|
||||
//! - `HE2LIN` subroutine (synspec54.f:6247) — standard frequency grid
|
||||
//! - `HE2LIW` subroutine (synspec54.f:6451) — frequency window mode
|
||||
//!
|
||||
//! Calculates opacity and emissivity of He II lines that are not
|
||||
//! considered explicitly (i.e., not handled by the detailed profile
|
||||
//! tables in HE2INI/HE2SEW).
|
||||
|
||||
use super::{divhe2, stark0, starka};
|
||||
|
||||
// ============================================================================
|
||||
// Physical constants
|
||||
// ============================================================================
|
||||
|
||||
const UN: f64 = 1.0;
|
||||
const SIXTH: f64 = 1.0 / 6.0;
|
||||
const CPP: f64 = 4.1412e-16;
|
||||
const CPJ: f64 = 631479.0;
|
||||
const CID: f64 = 0.02654;
|
||||
const CINV: f64 = UN / 2.997925e18;
|
||||
const AL10: f64 = std::f64::consts::LN_10;
|
||||
|
||||
/// He II ionization threshold frequencies (Hz).
|
||||
/// FRHE(n) = R_inf * c / n², for n = 1..12.
|
||||
const FRHE: [f64; 12] = [
|
||||
1.315_815_3e16, 3.289_538_1e15, 1.462_485_4e15,
|
||||
8.226_187_8e14, 5.264_720_1e14, 3.656_045_9e14,
|
||||
2.686_071_3e14, 2.056_522_0e14, 1.624_905_5e14,
|
||||
1.316_173_0e14, 1.087_746_0e14, 9.140_085_1e13,
|
||||
];
|
||||
|
||||
/// He II oscillator strengths (Schoening & Butler).
|
||||
const OSCHE2: [f64; 19] = [
|
||||
6.407e-1, 1.506e-1, 5.584e-2, 2.768e-2,
|
||||
1.604e-2, 1.023e-2, 6.980e-3,
|
||||
8.421e-1, 3.230e-2, 1.870e-2, 1.196e-2, 8.187e-3,
|
||||
5.886e-3, 4.393e-3, 3.375e-3, 2.656e-3,
|
||||
1.038, 1.793e-1, 6.549e-2,
|
||||
];
|
||||
|
||||
/// He II Lyman-series wavelength factor for n <= 2.
|
||||
const WLIN_FACTOR_LOW: f64 = 227.838;
|
||||
/// He II Lyman-series wavelength factor for n > 2.
|
||||
const WLIN_FACTOR_HIGH: f64 = 227.7776;
|
||||
|
||||
// ============================================================================
|
||||
// Shared parameters (used by both he2lin and he2liw)
|
||||
// ============================================================================
|
||||
|
||||
/// Common input data for He II line opacity calculations.
|
||||
#[derive(Clone)]
|
||||
pub struct He2Common<'a> {
|
||||
/// Depth index.
|
||||
pub id: usize,
|
||||
/// Temperature at depth ID (K).
|
||||
pub t: f64,
|
||||
/// Electron density at depth ID.
|
||||
pub ane: f64,
|
||||
/// Turbulent velocity at depth ID (cm/s).
|
||||
pub vturb: f64,
|
||||
/// Surface gravity (log g).
|
||||
pub grav: f64,
|
||||
/// Frequency array (Hz).
|
||||
pub freq: &'a [f64],
|
||||
/// Wavelength array (Å).
|
||||
pub wlam: &'a [f64],
|
||||
/// He II atom index in the model (0 if absent).
|
||||
pub ielhe2: i32,
|
||||
/// He II profile treatment flag (>0: use profile tables).
|
||||
pub ihe2pr: i32,
|
||||
/// First level index for He II element.
|
||||
pub nfirst_he2: usize,
|
||||
/// Last level index for He II element.
|
||||
pub nlast_he2: usize,
|
||||
/// Next element index after He II.
|
||||
pub nnext_he2: usize,
|
||||
/// He III population at depth ID (from model).
|
||||
pub anp_he3: f64,
|
||||
/// LTE He III population from RRR if ielhe2 <= 0.
|
||||
pub rrr_he3: f64,
|
||||
/// Level populations for He II (PJ array, up to 60 levels).
|
||||
/// If None, populations are computed from LTE/Saha.
|
||||
pub pj: Option<&'a [f64]>,
|
||||
/// WNHE2 partition function values (indexed by level, depth).
|
||||
pub wnhe2: &'a [f64],
|
||||
/// Number of wavelength points per profile line.
|
||||
pub nwlhe2: &'a [usize],
|
||||
/// Log10 of profile values: prfhe2[line * 36 + iwl].
|
||||
pub prfhe2: &'a [f64],
|
||||
/// Log10 of wavelength grid per profile line: wlhe2[line * 36 + iwl].
|
||||
pub wlhe2: &'a [f64],
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// HE2LIN — standard frequency grid
|
||||
// ============================================================================
|
||||
|
||||
/// Input parameters for `he2lin`.
|
||||
pub struct He2linParams<'a> {
|
||||
/// Common He II data.
|
||||
pub common: He2Common<'a>,
|
||||
/// Start frequency index.
|
||||
pub i0: usize,
|
||||
/// End frequency index.
|
||||
pub i1: usize,
|
||||
/// He II lowest series index contributing to this frequency region.
|
||||
pub ilwhe2: usize,
|
||||
/// Maximum principal quantum number for explicit He II treatment.
|
||||
pub mhe10: usize,
|
||||
/// Upper limit for He II lines.
|
||||
pub mhe20: usize,
|
||||
}
|
||||
|
||||
/// Result of `he2lin` / `he2liw`.
|
||||
pub struct He2linResult {
|
||||
/// Absorption coefficient array.
|
||||
pub absoh: Vec<f64>,
|
||||
/// Emission coefficient array.
|
||||
pub emish: Vec<f64>,
|
||||
}
|
||||
|
||||
/// Calculate He II line opacity and emissivity (standard frequency grid).
|
||||
///
|
||||
/// Handles He II lines that are not treated with explicit profile tables.
|
||||
/// Uses asymptotic Stark profiles for most lines, with interpolated
|
||||
/// tabulated profiles for specific lines when `ihe2pr > 0`.
|
||||
pub fn he2lin(params: &He2linParams) -> He2linResult {
|
||||
let c = ¶ms.common;
|
||||
let nf = c.freq.len();
|
||||
let mut abso = vec![0.0; nf];
|
||||
let mut emis = vec![0.0; nf];
|
||||
let mut absoh = vec![0.0; nf];
|
||||
let mut emish = vec![0.0; nf];
|
||||
|
||||
let (_t1, _sqt, _ane, _anes, _pp, pj, f00, dop0) = prepare(c);
|
||||
|
||||
// Series range
|
||||
let iseru = params.ilwhe2;
|
||||
let iserl = series_lower(params.ilwhe2);
|
||||
|
||||
// Loop over spectral series
|
||||
for i in iserl..=iseru {
|
||||
let (m1, m2) = determine_lines(i, params.ilwhe2, params.mhe10, params.mhe20, c.grav, c.freq);
|
||||
|
||||
for j in m1..=m2 {
|
||||
let (abtra, emtra, wlin) = transition(i, j, &pj, c, nf);
|
||||
let iline = profile_line_index(i, j, c.ihe2pr);
|
||||
|
||||
if iline > 0 {
|
||||
accumulate_tabulated(
|
||||
c, iline, wlin, abtra, emtra,
|
||||
params.i0, params.i1, nf,
|
||||
&mut abso, &mut emis,
|
||||
);
|
||||
} else {
|
||||
accumulate_stark(
|
||||
c, i, j, f00, dop0, wlin, abtra, emtra,
|
||||
params.i0, params.i1, nf,
|
||||
&mut abso, &mut emis,
|
||||
);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Total opacity and emissivity
|
||||
finalize(c, params.i0, params.i1, nf, &abso, &emis, &mut absoh, &mut emish);
|
||||
|
||||
He2linResult { absoh, emish }
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// HE2LIW — frequency window mode
|
||||
// ============================================================================
|
||||
|
||||
/// Per-frequency window parameters for He II lines.
|
||||
pub struct He2liwWindowParams<'a> {
|
||||
/// He II line processing flag per frequency (-1: skip, >0: process).
|
||||
pub ihe2lw: &'a [i32],
|
||||
/// He II series index per frequency.
|
||||
pub ilwhew: &'a [usize],
|
||||
/// Maximum principal quantum number per frequency.
|
||||
pub mhe10w: &'a [usize],
|
||||
/// Upper limit for He II lines per frequency.
|
||||
pub mhe20w: &'a [usize],
|
||||
}
|
||||
|
||||
/// Input parameters for `he2liw`.
|
||||
pub struct He2liwParams<'a> {
|
||||
/// Common He II data.
|
||||
pub common: He2Common<'a>,
|
||||
/// Per-frequency window parameters.
|
||||
pub window: He2liwWindowParams<'a>,
|
||||
/// Overall He II window flag (IFHE2): <=0 means skip entirely.
|
||||
pub ifhe2: i32,
|
||||
}
|
||||
|
||||
/// Calculate He II line opacity and emissivity (frequency window mode).
|
||||
///
|
||||
/// This is the window-mode variant of `he2lin`. It iterates over all
|
||||
/// frequencies individually, using per-frequency window parameters.
|
||||
pub fn he2liw(params: &He2liwParams) -> He2linResult {
|
||||
let c = ¶ms.common;
|
||||
let nf = c.freq.len();
|
||||
let mut abso = vec![0.0; nf];
|
||||
let mut emis = vec![0.0; nf];
|
||||
let mut absoh = vec![0.0; nf];
|
||||
let mut emish = vec![0.0; nf];
|
||||
|
||||
if params.ifhe2 <= 0 {
|
||||
return He2linResult { absoh, emish };
|
||||
}
|
||||
|
||||
let (t1, _sqt, _ane, _anes, _pp, pj, f00, dop0) = prepare(c);
|
||||
|
||||
// Loop over all frequencies
|
||||
for ij in 0..nf {
|
||||
if params.window.ihe2lw[ij] <= 0 {
|
||||
continue;
|
||||
}
|
||||
|
||||
let ilw = params.window.ilwhew[ij];
|
||||
let fr = c.freq[ij];
|
||||
let iseru = ilw;
|
||||
let iserl = series_lower(ilw);
|
||||
|
||||
for i in iserl..=iseru {
|
||||
let ii = i * i;
|
||||
let _xii = UN / ii as f64;
|
||||
|
||||
let m1_base = params.window.mhe10w[ij];
|
||||
let m2_base = params.window.mhe20w[ij];
|
||||
|
||||
// Determine contributing lines
|
||||
let mut m1 = m1_base;
|
||||
if i < ilw && FRHE[i - 1] > fr {
|
||||
m1 = ((FRHE[i - 1] * ii as f64 / (FRHE[i - 1] - fr)).sqrt()) as usize;
|
||||
}
|
||||
let mut m2 = m1 + 1;
|
||||
if m1 < i + 1 {
|
||||
m1 = i + 1;
|
||||
}
|
||||
if c.grav < 6.0 && m1 <= 6 && i == 2 {
|
||||
// keep
|
||||
} else if c.grav < 6.0 && m1 <= 4 && i == 1 {
|
||||
// keep
|
||||
} else {
|
||||
m1 = m1.saturating_sub(1);
|
||||
m2 = m2_base + 3;
|
||||
if m2 > 60 {
|
||||
m2 = 60;
|
||||
}
|
||||
}
|
||||
if c.grav > 6.0 {
|
||||
m2 += 5;
|
||||
m1 = m1.saturating_sub(3);
|
||||
if m1 > i + 6 {
|
||||
m1 = m1.saturating_sub(3);
|
||||
}
|
||||
}
|
||||
if m1 < i + 1 {
|
||||
m1 = i + 1;
|
||||
}
|
||||
if m2 > 60 {
|
||||
m2 = 60;
|
||||
}
|
||||
|
||||
for j in m1..=m2 {
|
||||
let (abtra, emtra, wlin) = transition(i, j, &pj, c, nf);
|
||||
let iline = profile_line_index(i, j, c.ihe2pr);
|
||||
|
||||
if iline > 0 {
|
||||
// Tabulated profile (single frequency)
|
||||
let nwl = c.nwlhe2[iline - 1];
|
||||
let fid = CID * OSCHE2[iline - 1];
|
||||
let al_raw = (c.wlam[ij] - wlin).abs();
|
||||
let al = if al_raw < 1.0e-4 { 1.0e-4 } else { al_raw };
|
||||
let al = al.log10();
|
||||
|
||||
let mut iw0 = 0usize;
|
||||
for iwl in 0..nwl - 1 {
|
||||
let wl_next = profile_wl_val(c.wlhe2, iline, iwl + 1);
|
||||
if al <= wl_next {
|
||||
iw0 = iwl;
|
||||
break;
|
||||
}
|
||||
iw0 = iwl;
|
||||
}
|
||||
let iw1 = iw0 + 1;
|
||||
let wl0 = profile_wl_val(c.wlhe2, iline, iw0);
|
||||
let wl1 = profile_wl_val(c.wlhe2, iline, iw1);
|
||||
let prf0 = profile_prf_val(c.prfhe2, iline, iw0);
|
||||
let prf1 = profile_prf_val(c.prfhe2, iline, iw1);
|
||||
|
||||
let denom = wl1 - wl0;
|
||||
let prff = if denom.abs() > 1.0e-30 {
|
||||
(prf0 * (wl1 - al) + prf1 * (al - wl0)) / denom
|
||||
} else {
|
||||
prf0
|
||||
};
|
||||
let sg = (prff * AL10).exp() * fid;
|
||||
abso[ij] += sg * abtra;
|
||||
emis[ij] += sg * emtra;
|
||||
} else {
|
||||
// Asymptotic Stark profile (single frequency)
|
||||
let stark = stark0(i as i32, j as i32, 2);
|
||||
let fxk = f00 * stark.xkij;
|
||||
let fxk1 = UN / fxk;
|
||||
let dop = dop0 / stark.wl0;
|
||||
let dbeta = stark.wl0 * stark.wl0 * CINV * fxk1;
|
||||
let betad = dop * dbeta;
|
||||
let fid = CID * stark.fij * dbeta;
|
||||
let ad = divhe2(betad);
|
||||
let beta = (c.wlam[ij] - stark.wl0).abs() * fxk1;
|
||||
let sg = starka(beta, betad, ad, UN, UN) * fid;
|
||||
abso[ij] += sg * abtra;
|
||||
emis[ij] += sg * emtra;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Total opacity and emissivity for this frequency
|
||||
let f = c.freq[ij];
|
||||
let f15 = f * 1.0e-15;
|
||||
let xkf = (-4.79928e-11 * f * t1).exp();
|
||||
let xkfb = xkf * 1.4743e-2 * f15 * f15 * f15;
|
||||
absoh[ij] = abso[ij] - xkf * emis[ij];
|
||||
emish[ij] = xkfb * emis[ij];
|
||||
}
|
||||
|
||||
He2linResult { absoh, emish }
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// Shared helper functions
|
||||
// ============================================================================
|
||||
|
||||
/// Prepare common derived quantities from input parameters.
|
||||
/// Returns (t1, sqt, ane, anes, pp, pj, f00, dop0).
|
||||
fn prepare(c: &He2Common) -> (f64, f64, f64, f64, f64, [f64; 60], f64, f64) {
|
||||
let t1 = UN / c.t;
|
||||
let sqt = c.t.sqrt();
|
||||
let ane = c.ane;
|
||||
let anes = ane.powf(SIXTH);
|
||||
|
||||
let (anp, nlhe2) = if c.ielhe2 > 0 {
|
||||
(c.anp_he3, c.nlast_he2 - c.nfirst_he2 + 1)
|
||||
} else {
|
||||
(c.rrr_he3, 0)
|
||||
};
|
||||
|
||||
let nf = c.freq.len();
|
||||
let mut pj = [0.0f64; 60];
|
||||
let pp = CPP * ane * anp * t1 / sqt;
|
||||
for il in 1..=60 {
|
||||
let x = (il * il) as f64;
|
||||
if il <= nlhe2 {
|
||||
if let Some(pj_in) = c.pj
|
||||
&& il - 1 < pj_in.len() {
|
||||
pj[il - 1] = pj_in[il - 1];
|
||||
}
|
||||
} else {
|
||||
let wn = wn_val(c.wnhe2, il, c.id, nf);
|
||||
pj[il - 1] = pp * (CPJ / x * t1).exp() * x * wn;
|
||||
}
|
||||
}
|
||||
|
||||
let f00 = 3.906e-11 * anes * anes * anes * anes;
|
||||
let dop0 = 1.0e8 * (4.12e7 * c.t + c.vturb).sqrt();
|
||||
|
||||
(t1, sqt, ane, anes, pp, pj, f00, dop0)
|
||||
}
|
||||
|
||||
/// Determine the lower series index.
|
||||
fn series_lower(ilw: usize) -> usize {
|
||||
if ilw <= 3 {
|
||||
ilw
|
||||
} else if ilw <= 5 {
|
||||
ilw - 1
|
||||
} else if ilw <= 7 {
|
||||
ilw - 2
|
||||
} else if ilw <= 9 {
|
||||
ilw - 3
|
||||
} else {
|
||||
ilw - 4
|
||||
}
|
||||
}
|
||||
|
||||
/// Determine contributing line range (m1, m2) for a given series.
|
||||
fn determine_lines(
|
||||
i: usize, ilwhe2: usize, mhe10: usize, mhe20: usize,
|
||||
grav: f64, freq: &[f64],
|
||||
) -> (usize, usize) {
|
||||
let mut m1 = mhe10;
|
||||
if i < ilwhe2 && FRHE[i - 1] > freq[1] {
|
||||
m1 = ((FRHE[i - 1] * (i * i) as f64 / (FRHE[i - 1] - freq[1])).sqrt()) as usize;
|
||||
}
|
||||
let mut m2 = m1 + 1;
|
||||
if m1 < i + 1 {
|
||||
m1 = i + 1;
|
||||
}
|
||||
if grav < 6.0 && m1 <= 6 && i == 2 {
|
||||
// keep
|
||||
} else if grav < 6.0 && m1 <= 4 && i == 1 {
|
||||
// keep
|
||||
} else {
|
||||
m1 = m1.saturating_sub(1);
|
||||
m2 = mhe20 + 3;
|
||||
if m2 > 60 {
|
||||
m2 = 60;
|
||||
}
|
||||
}
|
||||
if grav > 6.0 {
|
||||
m2 += 5;
|
||||
m1 = m1.saturating_sub(3);
|
||||
if m1 > i + 6 {
|
||||
m1 = m1.saturating_sub(3);
|
||||
}
|
||||
}
|
||||
if m1 < i + 1 {
|
||||
m1 = i + 1;
|
||||
}
|
||||
if m2 > 60 {
|
||||
m2 = 60;
|
||||
}
|
||||
(m1, m2)
|
||||
}
|
||||
|
||||
/// Compute transition properties for line i→j.
|
||||
fn transition(i: usize, j: usize, pj: &[f64; 60], c: &He2Common, nf: usize) -> (f64, f64, f64) {
|
||||
let ii = (i * i) as f64;
|
||||
let jj = (j * j) as f64;
|
||||
let xii = UN / ii;
|
||||
let xjj = UN / jj;
|
||||
let t1 = UN / c.t;
|
||||
|
||||
let abtra = pj[i - 1] * wn_val(c.wnhe2, j, c.id, nf);
|
||||
let emtra = pj[j - 1] * wn_val(c.wnhe2, i, c.id, nf) * ii * xjj * (CPJ * (xii - xjj) * t1).exp();
|
||||
|
||||
let wlin = if i <= 2 {
|
||||
WLIN_FACTOR_LOW / (xii - 1.0 / jj)
|
||||
} else {
|
||||
WLIN_FACTOR_HIGH / (xii - 1.0 / jj)
|
||||
};
|
||||
|
||||
(abtra, emtra, wlin)
|
||||
}
|
||||
|
||||
/// Accumulate opacity using tabulated profile (range of frequencies).
|
||||
fn accumulate_tabulated(
|
||||
c: &He2Common, iline: usize, wlin: f64, abtra: f64, emtra: f64,
|
||||
i0: usize, i1: usize, nf: usize,
|
||||
abso: &mut [f64], emis: &mut [f64],
|
||||
) {
|
||||
let nwl = c.nwlhe2[iline - 1];
|
||||
let fid = CID * OSCHE2[iline - 1];
|
||||
|
||||
for ij in i0..=i1.min(nf - 1) {
|
||||
let al_raw = (c.wlam[ij] - wlin).abs();
|
||||
let al = if al_raw < 1.0e-4 { 1.0e-4 } else { al_raw };
|
||||
let al = al.log10();
|
||||
|
||||
let mut iw0 = 0usize;
|
||||
for iwl in 0..nwl - 1 {
|
||||
let wl_next = profile_wl_val(c.wlhe2, iline, iwl + 1);
|
||||
if al <= wl_next {
|
||||
iw0 = iwl;
|
||||
break;
|
||||
}
|
||||
iw0 = iwl;
|
||||
}
|
||||
let iw1 = iw0 + 1;
|
||||
let wl0 = profile_wl_val(c.wlhe2, iline, iw0);
|
||||
let wl1 = profile_wl_val(c.wlhe2, iline, iw1);
|
||||
let prf0 = profile_prf_val(c.prfhe2, iline, iw0);
|
||||
let prf1 = profile_prf_val(c.prfhe2, iline, iw1);
|
||||
|
||||
let denom = wl1 - wl0;
|
||||
let prff = if denom.abs() > 1.0e-30 {
|
||||
(prf0 * (wl1 - al) + prf1 * (al - wl0)) / denom
|
||||
} else {
|
||||
prf0
|
||||
};
|
||||
let sg = (prff * AL10).exp() * fid;
|
||||
abso[ij] += sg * abtra;
|
||||
emis[ij] += sg * emtra;
|
||||
}
|
||||
}
|
||||
|
||||
/// Accumulate opacity using asymptotic Stark profile (range of frequencies).
|
||||
fn accumulate_stark(
|
||||
c: &He2Common, i: usize, j: usize, f00: f64, dop0: f64,
|
||||
_wlin: f64, abtra: f64, emtra: f64,
|
||||
i0: usize, i1: usize, nf: usize,
|
||||
abso: &mut [f64], emis: &mut [f64],
|
||||
) {
|
||||
let stark = stark0(i as i32, j as i32, 2);
|
||||
let fxk = f00 * stark.xkij;
|
||||
let fxk1 = UN / fxk;
|
||||
let dop = dop0 / stark.wl0;
|
||||
let dbeta = stark.wl0 * stark.wl0 * CINV * fxk1;
|
||||
let betad = dop * dbeta;
|
||||
let fid = CID * stark.fij * dbeta;
|
||||
let ad = divhe2(betad);
|
||||
|
||||
for ij in i0..=i1.min(nf - 1) {
|
||||
let beta = (c.wlam[ij] - stark.wl0).abs() * fxk1;
|
||||
let sg = starka(beta, betad, ad, UN, UN) * fid;
|
||||
abso[ij] += sg * abtra;
|
||||
emis[ij] += sg * emtra;
|
||||
}
|
||||
}
|
||||
|
||||
/// Finalize: compute total absorption and emission from raw abso/emis.
|
||||
fn finalize(
|
||||
c: &He2Common, i0: usize, i1: usize, nf: usize,
|
||||
abso: &[f64], emis: &[f64],
|
||||
absoh: &mut [f64], emish: &mut [f64],
|
||||
) {
|
||||
let t1 = UN / c.t;
|
||||
for ij in i0..=i1.min(nf - 1) {
|
||||
let f = c.freq[ij];
|
||||
let f15 = f * 1.0e-15;
|
||||
let xkf = (-4.79928e-11 * f * t1).exp();
|
||||
let xkfb = xkf * 1.4743e-2 * f15 * f15 * f15;
|
||||
absoh[ij] = abso[ij] - xkf * emis[ij];
|
||||
emish[ij] = xkfb * emis[ij];
|
||||
}
|
||||
}
|
||||
|
||||
/// Determine the profile table line index for a given He II transition.
|
||||
///
|
||||
/// Returns 0 if no tabulated profile is available (use asymptotic Stark).
|
||||
fn profile_line_index(i: usize, j: usize, ihe2pr: i32) -> usize {
|
||||
if ihe2pr <= 0 {
|
||||
return 0;
|
||||
}
|
||||
match i {
|
||||
2 => {
|
||||
if j == 3 { 1 } else { 0 }
|
||||
}
|
||||
3 => {
|
||||
if j == 4 { 8 } else if j > 5 && j <= 10 { j - 3 } else { 0 }
|
||||
}
|
||||
4 => {
|
||||
if j <= 7 { j + 12 } else if (8..=15).contains(&j) { j + 1 } else { 0 }
|
||||
}
|
||||
_ => 0,
|
||||
}
|
||||
}
|
||||
|
||||
/// Access WNHE2 partition function value.
|
||||
fn wn_val(wnhe2: &[f64], level: usize, id: usize, nf: usize) -> f64 {
|
||||
if (1..=60).contains(&level) {
|
||||
let idx = (level - 1) * nf + id;
|
||||
if idx < wnhe2.len() {
|
||||
return wnhe2[idx];
|
||||
}
|
||||
}
|
||||
1.0
|
||||
}
|
||||
|
||||
/// Access PRFHE2 profile table value (log10 profile).
|
||||
fn profile_prf_val(prfhe2: &[f64], iline: usize, iwl: usize) -> f64 {
|
||||
let idx = (iline - 1) * 36 + iwl;
|
||||
if idx < prfhe2.len() { prfhe2[idx] } else { 0.0 }
|
||||
}
|
||||
|
||||
/// Access WLHE2 profile wavelength table value (log10 wavelength).
|
||||
fn profile_wl_val(wlhe2: &[f64], iline: usize, iwl: usize) -> f64 {
|
||||
let idx = (iline - 1) * 36 + iwl;
|
||||
if idx < wlhe2.len() { wlhe2[idx] } else { 0.0 }
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
fn make_common(nf: usize) -> (He2Common<'static>, Vec<f64>, Vec<f64>, Vec<f64>) {
|
||||
let freq: Vec<f64> = (0..nf).map(|i| 3.0e15 - i as f64 * 1.0e14).collect();
|
||||
let wlam: Vec<f64> = freq.iter().map(|&f| 2.997925e17 / f).collect();
|
||||
let wnhe2 = vec![1.0; 60 * nf];
|
||||
// Leak wnhe2 to get 'static — acceptable for tests
|
||||
let wnhe2: &'static [f64] = Box::leak(wnhe2.into_boxed_slice());
|
||||
let common = He2Common {
|
||||
id: 0,
|
||||
t: 20000.0,
|
||||
ane: 1.0e14,
|
||||
vturb: 2.0e5,
|
||||
grav: 4.0,
|
||||
freq: &[],
|
||||
wlam: &[],
|
||||
ielhe2: 1,
|
||||
ihe2pr: 0,
|
||||
nfirst_he2: 1,
|
||||
nlast_he2: 10,
|
||||
nnext_he2: 2,
|
||||
anp_he3: 1.0e10,
|
||||
rrr_he3: 1.0e10,
|
||||
pj: None,
|
||||
wnhe2,
|
||||
nwlhe2: &[0; 19],
|
||||
prfhe2: &[],
|
||||
wlhe2: &[],
|
||||
};
|
||||
(common, freq, wlam, wnhe2.to_vec())
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_he2lin_basic() {
|
||||
let nf = 10;
|
||||
let (mut common, freq, wlam, _wn) = make_common(nf);
|
||||
common.freq = &freq;
|
||||
common.wlam = &wlam;
|
||||
|
||||
let params = He2linParams {
|
||||
common,
|
||||
i0: 0,
|
||||
i1: nf - 1,
|
||||
ilwhe2: 3,
|
||||
mhe10: 10,
|
||||
mhe20: 20,
|
||||
};
|
||||
|
||||
let result = he2lin(¶ms);
|
||||
assert_eq!(result.absoh.len(), nf);
|
||||
assert_eq!(result.emish.len(), nf);
|
||||
assert!(result.absoh.iter().all(|&x| x.is_finite()));
|
||||
assert!(result.emish.iter().all(|&x| x.is_finite()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_he2lin_no_he2() {
|
||||
let nf = 5;
|
||||
let (mut common, freq, wlam, _wn) = make_common(nf);
|
||||
common.freq = &freq;
|
||||
common.wlam = &wlam;
|
||||
common.ielhe2 = 0;
|
||||
common.t = 10000.0;
|
||||
common.ane = 1.0e12;
|
||||
common.vturb = 1.0e5;
|
||||
|
||||
let params = He2linParams {
|
||||
common,
|
||||
i0: 0,
|
||||
i1: nf - 1,
|
||||
ilwhe2: 1,
|
||||
mhe10: 5,
|
||||
mhe20: 10,
|
||||
};
|
||||
|
||||
let result = he2lin(¶ms);
|
||||
assert!(result.absoh.iter().all(|&x| x.is_finite()));
|
||||
assert!(result.emish.iter().all(|&x| x.is_finite()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_he2liw_skip() {
|
||||
let nf = 5;
|
||||
let (mut common, freq, wlam, _wn) = make_common(nf);
|
||||
common.freq = &freq;
|
||||
common.wlam = &wlam;
|
||||
|
||||
let params = He2liwParams {
|
||||
common,
|
||||
window: He2liwWindowParams {
|
||||
ihe2lw: &[-1; 5],
|
||||
ilwhew: &[3; 5],
|
||||
mhe10w: &[10; 5],
|
||||
mhe20w: &[20; 5],
|
||||
},
|
||||
ifhe2: 0,
|
||||
};
|
||||
|
||||
let result = he2liw(¶ms);
|
||||
assert!(result.absoh.iter().all(|&x| x == 0.0));
|
||||
assert!(result.emish.iter().all(|&x| x == 0.0));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_he2liw_basic() {
|
||||
let nf = 5;
|
||||
let (mut common, freq, wlam, _wn) = make_common(nf);
|
||||
common.freq = &freq;
|
||||
common.wlam = &wlam;
|
||||
|
||||
let params = He2liwParams {
|
||||
common,
|
||||
window: He2liwWindowParams {
|
||||
ihe2lw: &[1; 5],
|
||||
ilwhew: &[3; 5],
|
||||
mhe10w: &[10; 5],
|
||||
mhe20w: &[20; 5],
|
||||
},
|
||||
ifhe2: 1,
|
||||
};
|
||||
|
||||
let result = he2liw(¶ms);
|
||||
assert!(result.absoh.iter().all(|&x| x.is_finite()));
|
||||
assert!(result.emish.iter().all(|&x| x.is_finite()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_profile_line_index_no_profile() {
|
||||
assert_eq!(profile_line_index(2, 3, 0), 0);
|
||||
assert_eq!(profile_line_index(3, 4, -1), 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_profile_line_index_with_profile() {
|
||||
assert_eq!(profile_line_index(2, 3, 1), 1);
|
||||
assert_eq!(profile_line_index(3, 4, 1), 8);
|
||||
assert_eq!(profile_line_index(3, 7, 1), 4);
|
||||
assert_eq!(profile_line_index(4, 5, 1), 17);
|
||||
assert_eq!(profile_line_index(4, 10, 1), 11);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_series_lower() {
|
||||
assert_eq!(series_lower(1), 1);
|
||||
assert_eq!(series_lower(3), 3);
|
||||
assert_eq!(series_lower(4), 3);
|
||||
assert_eq!(series_lower(5), 4);
|
||||
assert_eq!(series_lower(6), 4);
|
||||
assert_eq!(series_lower(7), 5);
|
||||
assert_eq!(series_lower(10), 6);
|
||||
}
|
||||
}
|
||||
341
src/synspec/math/he2set.rs
Normal file
341
src/synspec/math/he2set.rs
Normal file
@ -0,0 +1,341 @@
|
||||
//! He II 线不透明度初始化过程。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `he2set.f` (synspec54.f:6061)。
|
||||
//!
|
||||
//! 设置 He II 线在频率窗口中的处理参数。
|
||||
|
||||
// ============================================================================
|
||||
// 物理常数
|
||||
// ============================================================================
|
||||
|
||||
/// 光速 (Å/s)，用于波长转换
|
||||
const CLIGHT_A: f64 = 2.997925e17;
|
||||
|
||||
/// He II 电离阈值频率 (Hz)
|
||||
///
|
||||
/// 对应 Fortran DATA FRHE 数组，是 He II Lyman 系列各线的阈值频率。
|
||||
/// FRHE(n) = R_inf * c / n²，其中 n = 1..12
|
||||
const FRHE: [f64; 12] = [
|
||||
1.3158153e16, 3.2895381e15, 1.4624854e15,
|
||||
8.2261878e14, 5.2647201e14, 3.6560459e14,
|
||||
2.6860713e14, 2.0565220e14, 1.6249055e14,
|
||||
1.3161730e14, 1.0877460e14, 9.1400851e13,
|
||||
];
|
||||
|
||||
/// He II 最高频率阈值 (Hz) - 对应 n=1 电离频率
|
||||
const HE2_FREQ_LIMIT: f64 = 1.315812e16;
|
||||
|
||||
// ============================================================================
|
||||
// 参数结构体
|
||||
// ============================================================================
|
||||
|
||||
/// HE2SET 输入参数。
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct He2setParams {
|
||||
/// He II 处理标志 (<= 0 表示不处理 He II 线)
|
||||
pub ifhe2: i32,
|
||||
|
||||
/// 频率范围下限 (Hz) - FREQ(1)
|
||||
pub freq1: f64,
|
||||
|
||||
/// 频率范围上限 (Hz) - FREQ(2)
|
||||
pub freq2: f64,
|
||||
|
||||
/// 表面重力 log g (cgs)
|
||||
pub grav: f64,
|
||||
}
|
||||
|
||||
/// HE2SET 输出结果。
|
||||
#[derive(Debug, Clone, Default)]
|
||||
pub struct He2setOutput {
|
||||
/// He II 线处理标志
|
||||
/// - -1: He II 线被排除
|
||||
/// - 1: He II 线被包含
|
||||
pub ihe2l: i32,
|
||||
|
||||
/// He II 线系列索引 (1-12)
|
||||
pub ilwhe2: i32,
|
||||
|
||||
/// 主量子数上限 1 (用于线强度计算)
|
||||
pub mhe10: i32,
|
||||
|
||||
/// 主量子数上限 2 (用于线强度计算)
|
||||
pub mhe20: i32,
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// HE2SET 函数
|
||||
// ============================================================================
|
||||
|
||||
/// 初始化 He II 线不透明度参数。
|
||||
///
|
||||
/// 根据频率范围和重力确定 He II 线是否被包含在计算中，
|
||||
/// 并设置相应的处理参数。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `params` - 输入参数结构体
|
||||
///
|
||||
/// # 返回
|
||||
///
|
||||
/// 包含 `ihe2l`, `ilwhe2`, `mhe10`, `mhe20` 的输出结构体
|
||||
///
|
||||
/// # Fortran 源码
|
||||
///
|
||||
/// ```fortran
|
||||
/// SUBROUTINE HE2SET
|
||||
/// ```
|
||||
pub fn he2set(params: &He2setParams) -> He2setOutput {
|
||||
// 默认值：He II 线被排除
|
||||
let mut result = He2setOutput {
|
||||
ihe2l: -1,
|
||||
ilwhe2: 0,
|
||||
mhe10: 60,
|
||||
mhe20: 60,
|
||||
};
|
||||
|
||||
// 如果 He II 处理标志 <= 0，直接返回
|
||||
if params.ifhe2 <= 0 {
|
||||
return result;
|
||||
}
|
||||
|
||||
// 如果频率上限 >= He II 最高阈值，直接返回
|
||||
if params.freq2 >= HE2_FREQ_LIMIT {
|
||||
return result;
|
||||
}
|
||||
|
||||
// 计算波长范围 (Å)
|
||||
let al0 = CLIGHT_A / params.freq1;
|
||||
let al1 = CLIGHT_A / params.freq2;
|
||||
|
||||
// 根据重力检查排除区域
|
||||
if params.grav < 6.0 {
|
||||
// 低重力情况
|
||||
if al0 > 31.0 && al1 < 91.1 { return result; }
|
||||
if al0 > 26.1 && al1 < 29.8 { return result; }
|
||||
if al0 > 24.8 && al1 < 25.1 { return result; }
|
||||
if al0 > 122.1 && al1 < 162.9 { return result; }
|
||||
if al0 > 165.1 && al1 < 204.9 { return result; }
|
||||
if al0 > 109.0 && al1 < 120.9 { return result; }
|
||||
if al0 > 103.0 && al1 < 107.9 { return result; }
|
||||
if al0 > 99.7 && al1 < 102.0 { return result; }
|
||||
if al0 > 320.8 && al1 < 364.4 { return result; }
|
||||
if al0 > 273.8 && al1 < 319.8 { return result; }
|
||||
if al0 > 251.6 && al1 < 272.8 { return result; }
|
||||
if al0 > 239.0 && al1 < 250.6 { return result; }
|
||||
if al0 > 231.1 && al1 < 238.0 { return result; }
|
||||
if al0 > 225.8 && al1 < 230.1 { return result; }
|
||||
} else if params.grav < 7.0 {
|
||||
// 中等重力情况
|
||||
if al0 > 33.0 && al1 < 91.1 { return result; }
|
||||
if al0 > 124.1 && al1 < 160.9 { return result; }
|
||||
if al0 > 167.1 && al1 < 202.9 { return result; }
|
||||
if al0 > 111.0 && al1 < 118.9 { return result; }
|
||||
if al0 > 322.8 && al1 < 364.4 { return result; }
|
||||
if al0 > 275.8 && al1 < 317.8 { return result; }
|
||||
if al0 > 253.6 && al1 < 270.8 { return result; }
|
||||
if al0 > 241.0 && al1 < 248.6 { return result; }
|
||||
if al0 > 233.1 && al1 < 236.0 { return result; }
|
||||
} else {
|
||||
// 高重力情况
|
||||
if al0 > 39.0 && al1 < 91.1 { return result; }
|
||||
if al0 > 134.1 && al1 < 150.9 { return result; }
|
||||
if al0 > 177.1 && al1 < 202.9 { return result; }
|
||||
}
|
||||
|
||||
// He II 线被包含
|
||||
result.ihe2l = 1;
|
||||
result.mhe10 = 60;
|
||||
result.mhe20 = 60;
|
||||
|
||||
// 根据波长范围确定系列索引
|
||||
result.ilwhe2 = if al1 < 91.0 {
|
||||
1
|
||||
} else if al0 < 204.0 {
|
||||
2
|
||||
} else if al0 < 364.0 {
|
||||
3
|
||||
} else if al0 < 569.0 {
|
||||
4
|
||||
} else if al0 < 819.0 {
|
||||
5
|
||||
} else if al0 < 1116.0 {
|
||||
6
|
||||
} else if al0 < 1457.0 {
|
||||
7
|
||||
} else if al0 < 1844.0 {
|
||||
8
|
||||
} else if al0 < 2277.0 {
|
||||
9
|
||||
} else if al0 < 2756.0 {
|
||||
10
|
||||
} else if al0 < 3279.0 {
|
||||
11
|
||||
} else {
|
||||
12
|
||||
};
|
||||
|
||||
// 计算量子数上限
|
||||
let frion = FRHE[(result.ilwhe2 - 1) as usize];
|
||||
let fr1 = frion * (result.ilwhe2 as f64) * (result.ilwhe2 as f64);
|
||||
|
||||
if frion > params.freq2 {
|
||||
result.mhe10 = (fr1 / (frion - params.freq2)).sqrt() as i32;
|
||||
}
|
||||
if frion > params.freq1 {
|
||||
result.mhe20 = (fr1 / (frion - params.freq1)).sqrt() as i32;
|
||||
}
|
||||
|
||||
result
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// 测试
|
||||
// ============================================================================
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use approx::assert_relative_eq;
|
||||
|
||||
/// 创建默认测试参数
|
||||
fn create_test_params() -> He2setParams {
|
||||
He2setParams {
|
||||
ifhe2: 1,
|
||||
freq1: 4.0e14, // 750 nm
|
||||
freq2: 8.0e14, // 375 nm
|
||||
grav: 4.0,
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_he2set_disabled() {
|
||||
// IFHE2 <= 0 时应返回排除状态
|
||||
let params = He2setParams {
|
||||
ifhe2: 0,
|
||||
..create_test_params()
|
||||
};
|
||||
let result = he2set(¶ms);
|
||||
assert_eq!(result.ihe2l, -1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_he2set_freq_too_high() {
|
||||
// 频率上限 >= He II 最高阈值时应返回排除状态
|
||||
let params = He2setParams {
|
||||
freq2: 1.4e16, // > 1.315812e16
|
||||
..create_test_params()
|
||||
};
|
||||
let result = he2set(¶ms);
|
||||
assert_eq!(result.ihe2l, -1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_he2set_low_gravity_exclusion() {
|
||||
// 低重力情况下的排除区域测试
|
||||
// AL0 = 130, AL1 = 150 → 122.1 < AL0 且 AL1 < 162.9 → 排除
|
||||
let params = He2setParams {
|
||||
ifhe2: 1,
|
||||
freq1: CLIGHT_A / 130.0, // AL0 = 130 Å
|
||||
freq2: CLIGHT_A / 150.0, // AL1 = 150 Å
|
||||
grav: 5.0,
|
||||
};
|
||||
let result = he2set(¶ms);
|
||||
assert_eq!(result.ihe2l, -1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_he2set_included_lyman() {
|
||||
// 测试 Lyman 系列被包含的情况
|
||||
// AL0 = 200, AL1 = 300 → 不在任何排除区域
|
||||
let params = He2setParams {
|
||||
ifhe2: 1,
|
||||
freq1: CLIGHT_A / 200.0, // AL0 = 200 Å
|
||||
freq2: CLIGHT_A / 300.0, // AL1 = 300 Å
|
||||
grav: 4.0,
|
||||
};
|
||||
let result = he2set(¶ms);
|
||||
assert_eq!(result.ihe2l, 1);
|
||||
assert_eq!(result.ilwhe2, 2); // 91 < AL0 < 204
|
||||
assert!(result.mhe10 > 0);
|
||||
assert!(result.mhe20 > 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_he2set_included_balmer() {
|
||||
// 测试 Balmer 系列被包含的情况
|
||||
let params = He2setParams {
|
||||
ifhe2: 1,
|
||||
freq1: CLIGHT_A / 500.0, // AL0 = 500 Å
|
||||
freq2: CLIGHT_A / 400.0, // AL1 = 400 Å
|
||||
grav: 4.0,
|
||||
};
|
||||
let result = he2set(¶ms);
|
||||
assert_eq!(result.ihe2l, 1);
|
||||
assert_eq!(result.ilwhe2, 4); // 364 < AL0 < 569
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_he2set_high_gravity() {
|
||||
// 高重力情况下的测试
|
||||
let params = He2setParams {
|
||||
ifhe2: 1,
|
||||
freq1: CLIGHT_A / 100.0, // AL0 = 100 Å
|
||||
freq2: CLIGHT_A / 150.0, // AL1 = 150 Å
|
||||
grav: 8.0,
|
||||
};
|
||||
let result = he2set(¶ms);
|
||||
// 高重力下排除区域更少
|
||||
assert_eq!(result.ihe2l, 1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_he2set_series_index() {
|
||||
// 测试系列索引的边界情况
|
||||
let params = He2setParams {
|
||||
ifhe2: 1,
|
||||
freq1: CLIGHT_A / 3000.0, // AL0 = 3000 Å
|
||||
freq2: CLIGHT_A / 2800.0, // AL1 = 2800 Å
|
||||
grav: 4.0,
|
||||
};
|
||||
let result = he2set(¶ms);
|
||||
assert_eq!(result.ilwhe2, 11); // 2756 < AL0 < 3279
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_frhe_constants() {
|
||||
// 验证 FRHE 常数与 Fortran 一致
|
||||
assert_relative_eq!(FRHE[0], 1.3158153e16, epsilon = 1e10);
|
||||
assert_relative_eq!(FRHE[1], 3.2895381e15, epsilon = 1e9);
|
||||
assert_relative_eq!(FRHE[11], 9.1400851e13, epsilon = 1e7);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_he2set_medium_gravity() {
|
||||
// 中等重力情况下的测试
|
||||
let params = He2setParams {
|
||||
ifhe2: 1,
|
||||
freq1: CLIGHT_A / 130.0, // AL0 = 130 Å
|
||||
freq2: CLIGHT_A / 160.0, // AL1 = 160 Å
|
||||
grav: 6.5,
|
||||
};
|
||||
let result = he2set(¶ms);
|
||||
// 中等重力下的排除区域: 124.1 < AL0 < 160.9
|
||||
assert_eq!(result.ihe2l, -1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_he2set_quantum_numbers() {
|
||||
// 测试量子数上限的计算
|
||||
let params = He2setParams {
|
||||
ifhe2: 1,
|
||||
freq1: CLIGHT_A / 100.0, // AL0 = 100 Å
|
||||
freq2: CLIGHT_A / 200.0, // AL1 = 200 Å
|
||||
grav: 4.0,
|
||||
};
|
||||
let result = he2set(¶ms);
|
||||
assert_eq!(result.ilwhe2, 2); // 91 < AL0 < 204
|
||||
assert!(result.mhe10 > 0);
|
||||
assert!(result.mhe20 > 0);
|
||||
}
|
||||
}
|
||||
@ -104,7 +104,7 @@ pub fn he2sew(freq: f64, grav: f64, ifhe2: i32) -> He2WindowParams {
|
||||
|
||||
if frion > freq {
|
||||
let ratio = fr1 / (frion - freq);
|
||||
result.mhe10w = (ratio.sqrt() as i32);
|
||||
result.mhe10w = ratio.sqrt() as i32;
|
||||
}
|
||||
|
||||
result
|
||||
|
||||
260
src/synspec/math/hephot.rs
Normal file
260
src/synspec/math/hephot.rs
Normal file
@ -0,0 +1,260 @@
|
||||
//! He I photoionization cross sections using Seaton-Fernley's cubic fits
|
||||
//! to the Opacity Project cross sections.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine HEPHOT(S,L,N,FREQ)
|
||||
|
||||
/// He I photoionization cross section using Opacity Project fits.
|
||||
///
|
||||
/// Evaluates He I photoionization cross section using Seaton-Fernley's cubic
|
||||
/// fits to the Opacity Project cross sections up to some energy "EFITM" in the
|
||||
/// resonance-free zone. Beyond this energy, linear fits to log sigma in
|
||||
/// log(E/E0) are used.
|
||||
///
|
||||
/// For L > 2, hydrogenic expression is used.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `s` - Multiplicity, either 1 (singlet) or 3 (triplet)
|
||||
/// * `l` - Angular momentum quantum number (0, 1, or 2; for L > 2 uses hydrogenic)
|
||||
/// * `n` - Principal quantum number
|
||||
/// * `freq` - Frequency in Hz
|
||||
///
|
||||
/// # Returns
|
||||
/// Photoionization cross section in cm^2
|
||||
pub fn hephot(s: i32, l: i32, n: i32, freq: f64) -> f64 {
|
||||
// Hydrogenic expression for L > 2
|
||||
if l > 2 {
|
||||
let gn = 2.0 * (n as f64) * (n as f64);
|
||||
return 2.815e29 / freq / freq / freq
|
||||
/ (n.pow(5) as f64)
|
||||
* ((2 * l + 1) as f64)
|
||||
* (s as f64)
|
||||
/ gn;
|
||||
}
|
||||
|
||||
// Select beginning and end of coefficients
|
||||
let ss = ((s + 1) / 2) as usize; // 1-based index for singlet/triplet
|
||||
let ll = (l + 1) as usize; // 1-based index for l
|
||||
|
||||
// Get the starting index and quantum number offset
|
||||
let ist_idx = (IST[(ll - 1) * 2 + (ss - 1)] - 1) as usize; // convert to 0-based
|
||||
let nsl0 = N0[(ll - 1) * 2 + (ss - 1)];
|
||||
|
||||
let i = ist_idx + (n - nsl0) as usize; // 0-based index into coefficient arrays
|
||||
|
||||
// Evaluate cross section
|
||||
let fl = (freq / 3.28805e15).log10();
|
||||
let x = fl - FL0[i];
|
||||
|
||||
if x >= -0.001 {
|
||||
if x < XFITM[i] {
|
||||
// Cubic polynomial fit
|
||||
let mut p = COEF[i * 4 + 3]; // COEF(4,I) in Fortran (1-indexed)
|
||||
for k in (0..3).rev() {
|
||||
p = x * p + COEF[i * 4 + k]; // COEF(4-K,I)
|
||||
}
|
||||
1.0e-18 * 10.0_f64.powf(p)
|
||||
} else {
|
||||
// Linear extrapolation in log space
|
||||
1.0e-18 * 10.0_f64.powf(A[i] + B[i] * x)
|
||||
}
|
||||
} else {
|
||||
0.0
|
||||
}
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// Opacity Project fit data
|
||||
// ============================================================================
|
||||
|
||||
/// Starting indices for each (l, s) combination (1-based in Fortran, converted to 0-based usage)
|
||||
/// IST(LL, SS) where LL = l+1, SS = (s+1)/2
|
||||
/// Layout: [singlet_l0, triplet_l0, singlet_l1, triplet_l1, singlet_l2, triplet_l2]
|
||||
const IST: [i32; 6] = [1, 36, 20, 11, 45, 28];
|
||||
|
||||
/// Starting principal quantum number for each (l, s) combination
|
||||
/// N0(LL, SS)
|
||||
const N0: [i32; 6] = [1, 2, 3, 2, 2, 3];
|
||||
|
||||
/// log10(nu/nu0) offset values for 53 cross section fits
|
||||
const FL0: [f64; 53] = [
|
||||
2.521e-01, -5.381e-01, -9.139e-01, -1.175e+00, -1.375e+00, -1.537e+00,
|
||||
-1.674e+00, -1.792e+00, -1.896e+00, -1.989e+00, -4.555e-01, -8.622e-01,
|
||||
-1.137e+00, -1.345e+00, -1.512e+00, -1.653e+00, -1.774e+00, -1.880e+00,
|
||||
-1.974e+00, -9.538e-01, -1.204e+00, -1.398e+00, -1.556e+00, -1.690e+00,
|
||||
-1.806e+00, -1.909e+00, -2.000e+00, -9.537e-01, -1.204e+00, -1.398e+00,
|
||||
-1.556e+00, -1.690e+00, -1.806e+00, -1.909e+00, -2.000e+00, -6.065e-01,
|
||||
-9.578e-01, -1.207e+00, -1.400e+00, -1.558e+00, -1.692e+00, -1.808e+00,
|
||||
-1.910e+00, -2.002e+00, -5.749e-01, -9.352e-01, -1.190e+00, -1.386e+00,
|
||||
-1.547e+00, -1.682e+00, -1.799e+00, -1.902e+00, -1.995e+00,
|
||||
];
|
||||
|
||||
/// Upper limit of cubic fit region (in log10 space)
|
||||
const XFITM: [f64; 53] = [
|
||||
3.262e-01, 6.135e-01, 9.233e-01, 8.438e-01, 1.020e+00, 1.169e+00,
|
||||
1.298e+00, 1.411e+00, 1.512e+00, 1.602e+00, 7.228e-01, 1.076e+00,
|
||||
1.206e+00, 1.404e+00, 1.481e+00, 1.464e+00, 1.581e+00, 1.685e+00,
|
||||
1.777e+00, 9.586e-01, 1.187e+00, 1.371e+00, 1.524e+00, 1.740e+00,
|
||||
1.854e+00, 1.955e+00, 2.046e+00, 9.585e-01, 1.041e+00, 1.371e+00,
|
||||
1.608e+00, 1.739e+00, 1.768e+00, 1.869e+00, 1.803e+00, 7.360e-01,
|
||||
1.041e+00, 1.272e+00, 1.457e+00, 1.611e+00, 1.741e+00, 1.855e+00,
|
||||
1.870e+00, 1.804e+00, 9.302e-01, 1.144e+00, 1.028e+00, 1.210e+00,
|
||||
1.362e+00, 1.646e+00, 1.761e+00, 1.863e+00, 1.954e+00,
|
||||
];
|
||||
|
||||
/// Linear fit coefficients A (53 values)
|
||||
const A: [f64; 53] = [
|
||||
6.95319e-01, 1.13101e+00, 1.36313e+00, 1.51684e+00, 1.64767e+00,
|
||||
1.75643e+00, 1.84458e+00, 1.87243e+00, 1.85628e+00, 1.90889e+00,
|
||||
9.01802e-01, 1.25389e+00, 1.39033e+00, 1.55226e+00, 1.60658e+00,
|
||||
1.65930e+00, 1.68855e+00, 1.62477e+00, 1.66726e+00, 1.83599e+00,
|
||||
2.50403e+00, 3.08564e+00, 3.56545e+00, 4.25922e+00, 4.61346e+00,
|
||||
4.91417e+00, 5.19211e+00, 1.74181e+00, 2.25756e+00, 2.95625e+00,
|
||||
3.65899e+00, 4.04397e+00, 4.13410e+00, 4.43538e+00, 4.19583e+00,
|
||||
1.79027e+00, 2.23543e+00, 2.63942e+00, 3.02461e+00, 3.35018e+00,
|
||||
3.62067e+00, 3.85218e+00, 3.76689e+00, 3.49318e+00, 1.16294e+00,
|
||||
1.86467e+00, 2.02110e+00, 2.24231e+00, 2.44240e+00, 2.76594e+00,
|
||||
2.93230e+00, 3.08109e+00, 3.21069e+00,
|
||||
];
|
||||
|
||||
/// Linear fit coefficients B (53 values)
|
||||
const B: [f64; 53] = [
|
||||
-1.29000e+00, -2.15771e+00, -2.13263e+00, -2.10272e+00, -2.10861e+00,
|
||||
-2.11507e+00, -2.11710e+00, -2.08531e+00, -2.03296e+00, -2.03441e+00,
|
||||
-1.85905e+00, -2.04057e+00, -2.02189e+00, -2.05930e+00, -2.03403e+00,
|
||||
-2.02071e+00, -1.99956e+00, -1.92851e+00, -1.92905e+00, -4.58608e+00,
|
||||
-4.40022e+00, -4.39154e+00, -4.39676e+00, -4.57631e+00, -4.57120e+00,
|
||||
-4.56188e+00, -4.55915e+00, -4.41218e+00, -4.12940e+00, -4.24401e+00,
|
||||
-4.40783e+00, -4.39930e+00, -4.25981e+00, -4.26804e+00, -4.00419e+00,
|
||||
-4.47251e+00, -3.87960e+00, -3.71668e+00, -3.68461e+00, -3.67173e+00,
|
||||
-3.65991e+00, -3.64968e+00, -3.48666e+00, -3.23985e+00, -2.95758e+00,
|
||||
-3.07110e+00, -2.87157e+00, -2.83137e+00, -2.82132e+00, -2.91084e+00,
|
||||
-2.91159e+00, -2.91336e+00, -2.91296e+00,
|
||||
];
|
||||
|
||||
/// Cubic polynomial coefficients COEF(4, 53) stored as flat array
|
||||
/// COEF[I*4 + j] corresponds to Fortran COEF(j+1, I+1)
|
||||
const COEF: [f64; 212] = [
|
||||
// J=1..10
|
||||
8.734e-01, -1.545e+00, -1.093e+00, 5.918e-01,
|
||||
9.771e-01, -1.567e+00, -4.739e-01, -1.302e-01,
|
||||
1.174e+00, -1.638e+00, -2.831e-01, -3.281e-02,
|
||||
1.324e+00, -1.692e+00, -2.916e-01, 9.027e-02,
|
||||
1.445e+00, -1.761e+00, -1.902e-01, 4.401e-02,
|
||||
1.546e+00, -1.817e+00, -1.278e-01, 2.293e-02,
|
||||
1.635e+00, -1.864e+00, -8.252e-02, 9.854e-03,
|
||||
1.712e+00, -1.903e+00, -5.206e-02, 2.892e-03,
|
||||
1.782e+00, -1.936e+00, -2.952e-02, -1.405e-03,
|
||||
1.845e+00, -1.964e+00, -1.152e-02, -4.487e-03,
|
||||
// J=11..19
|
||||
7.377e-01, -9.327e-01, -1.466e+00, 6.891e-01,
|
||||
9.031e-01, -1.157e+00, -7.151e-01, 1.832e-01,
|
||||
1.031e+00, -1.313e+00, -4.517e-01, 9.207e-02,
|
||||
1.135e+00, -1.441e+00, -2.724e-01, 3.105e-02,
|
||||
1.225e+00, -1.536e+00, -1.725e-01, 7.191e-03,
|
||||
1.302e+00, -1.602e+00, -1.300e-01, 7.345e-03,
|
||||
1.372e+00, -1.664e+00, -8.204e-02, -1.643e-03,
|
||||
1.434e+00, -1.715e+00, -4.646e-02, -7.456e-03,
|
||||
1.491e+00, -1.760e+00, -1.838e-02, -1.152e-02,
|
||||
// J=20..27
|
||||
1.258e+00, -3.442e+00, -4.731e-01, -9.522e-02,
|
||||
1.553e+00, -2.781e+00, -6.841e-01, -4.083e-03,
|
||||
1.727e+00, -2.494e+00, -5.785e-01, -6.015e-02,
|
||||
1.853e+00, -2.347e+00, -4.611e-01, -9.615e-02,
|
||||
1.955e+00, -2.273e+00, -3.457e-01, -1.245e-01,
|
||||
2.041e+00, -2.226e+00, -2.669e-01, -1.344e-01,
|
||||
2.115e+00, -2.200e+00, -1.999e-01, -1.410e-01,
|
||||
2.182e+00, -2.188e+00, -1.405e-01, -1.460e-01,
|
||||
// J=28..35
|
||||
1.267e+00, -3.417e+00, -5.038e-01, -1.797e-02,
|
||||
1.565e+00, -2.781e+00, -6.497e-01, -5.979e-03,
|
||||
1.741e+00, -2.479e+00, -6.099e-01, -2.227e-02,
|
||||
1.870e+00, -2.336e+00, -4.899e-01, -6.616e-02,
|
||||
1.973e+00, -2.253e+00, -3.972e-01, -8.729e-02,
|
||||
2.061e+00, -2.212e+00, -3.072e-01, -1.060e-01,
|
||||
2.137e+00, -2.189e+00, -2.352e-01, -1.171e-01,
|
||||
2.205e+00, -2.186e+00, -1.621e-01, -1.296e-01,
|
||||
// J=36..44
|
||||
1.129e+00, -3.149e+00, -1.910e-01, -5.244e-01,
|
||||
1.431e+00, -2.511e+00, -3.710e-01, -1.933e-01,
|
||||
1.620e+00, -2.303e+00, -3.045e-01, -1.391e-01,
|
||||
1.763e+00, -2.235e+00, -1.829e-01, -1.491e-01,
|
||||
1.879e+00, -2.215e+00, -9.003e-02, -1.537e-01,
|
||||
1.978e+00, -2.213e+00, -2.066e-02, -1.541e-01,
|
||||
2.064e+00, -2.220e+00, 3.258e-02, -1.527e-01,
|
||||
2.140e+00, -2.225e+00, 6.311e-02, -1.455e-01,
|
||||
2.208e+00, -2.229e+00, 7.977e-02, -1.357e-01,
|
||||
// J=45..53
|
||||
1.204e+00, -2.809e+00, -3.094e-01, 1.100e-01,
|
||||
1.455e+00, -2.254e+00, -4.795e-01, 6.872e-02,
|
||||
1.619e+00, -2.109e+00, -3.357e-01, -2.532e-02,
|
||||
1.747e+00, -2.065e+00, -2.317e-01, -5.224e-02,
|
||||
1.853e+00, -2.058e+00, -1.517e-01, -6.647e-02,
|
||||
1.943e+00, -2.055e+00, -1.158e-01, -6.081e-02,
|
||||
2.023e+00, -2.070e+00, -6.470e-02, -6.800e-02,
|
||||
2.095e+00, -2.088e+00, -2.357e-02, -7.250e-02,
|
||||
2.160e+00, -2.107e+00, 1.065e-02, -7.542e-02,
|
||||
];
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_hephot_below_threshold() {
|
||||
// Below threshold frequency should return 0
|
||||
let sigma = hephot(1, 0, 1, 1.0e14);
|
||||
assert_eq!(sigma, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hephot_singlet_s_ground() {
|
||||
// Singlet S state, n=1: threshold is at 3.288e15 * 10^FL0[0] ≈ 5.76e15 Hz
|
||||
// Use frequency well above threshold
|
||||
let freq = 8.0e15;
|
||||
let sigma = hephot(1, 0, 1, freq);
|
||||
assert!(sigma > 0.0, "Cross section should be positive above threshold");
|
||||
assert!(sigma < 1.0e-15, "Cross section should be in reasonable range");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hephot_triplet_p_n2() {
|
||||
// Triplet P state, n=2
|
||||
// IST(2,2)=36, N0(2,2)=2, so i = 36-1 + (2-2) = 35 (0-based)
|
||||
let freq = 1.0e15;
|
||||
let sigma = hephot(3, 1, 2, freq);
|
||||
// May be below or above threshold depending on freq
|
||||
assert!(sigma >= 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hephot_hydrogenic_l3() {
|
||||
// For L > 2, should use hydrogenic expression
|
||||
let freq = 1.0e15;
|
||||
let sigma = hephot(1, 3, 3, freq);
|
||||
assert!(sigma > 0.0, "Hydrogenic cross section should be positive");
|
||||
// Expected: 2.815e29 / freq^3 / n^5 * (2L+1) * S / (2*n^2)
|
||||
let expected = 2.815e29 / freq.powi(3) / (3_i32.pow(5) as f64) * 7.0 * 1.0 / 18.0;
|
||||
assert!((sigma - expected).abs() / expected < 1.0e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hephot_singlet_d_n2() {
|
||||
// Singlet D, n=2: IST(3,1)=20, N0(3,1)=3
|
||||
// i = 20-1 + (2-3) = 18 (0-based)
|
||||
// But n < N0 gives negative index - this case shouldn't be called
|
||||
// Let's test n=3 instead
|
||||
let freq = 1.5e15;
|
||||
let sigma = hephot(1, 2, 3, freq);
|
||||
assert!(sigma >= 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hephot_all_multiplicities() {
|
||||
// Test that both singlet and triplet produce valid results
|
||||
let freq = 5.0e15;
|
||||
let s1 = hephot(1, 0, 1, freq); // singlet
|
||||
let s3 = hephot(3, 0, 2, freq); // triplet
|
||||
assert!(s1 >= 0.0);
|
||||
assert!(s3 >= 0.0);
|
||||
}
|
||||
}
|
||||
163
src/synspec/math/hidalg.rs
Normal file
163
src/synspec/math/hidalg.rs
Normal file
@ -0,0 +1,163 @@
|
||||
//! 光致电离截面插值（Hidalgo 1968）。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `hidalg.f`
|
||||
//!
|
||||
//! 使用 Hidalgo (1968, Ap. J., 153, 981) 的波长和光致电离截面数据表，
|
||||
//! 对给定频率进行线性插值。
|
||||
|
||||
/// 波长网格 1 (Å)，用于 INDEX < 13 的物种
|
||||
const WL1: [f64; 20] = [
|
||||
39.1, 80.9, 97.6, 100.1, 104.3, 107.2, 108.7, 111.9, 113.6, 115.4,
|
||||
117.1, 119.0, 124.8, 126.9, 129.1, 131.3, 133.6, 136.0, 138.5, 141.1,
|
||||
];
|
||||
|
||||
/// 波长网格 2 (Å)，用于 INDEX >= 13 的物种
|
||||
const WL2: [f64; 20] = [
|
||||
68.5, 80.9, 100.1, 120.9, 158.8, 165.7, 177.3, 190.6, 200.7, 206.2,
|
||||
211.9, 218.0, 224.5, 231.3, 246.3, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
];
|
||||
|
||||
/// 光致电离截面数据 (Mbarn)，20×24 矩阵（列优先存储）
|
||||
const SIG0: [[f64; 20]; 24] = [
|
||||
[0.0; 20], // col 1
|
||||
[
|
||||
0.0460, 0.2400, 0.3500, 0.3700, 0.4000, 0.4300, 0.4400, 0.4600, 0.4700, 0.4900,
|
||||
0.5000, 0.5200, 0.5700, 0.6200, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
], // col 2
|
||||
[0.0; 20], // col 3
|
||||
[
|
||||
0.0092, 0.1000, 0.1900, 0.2100, 0.2300, 0.2500, 0.2600, 0.2900, 0.3000, 0.3200,
|
||||
0.3400, 0.3500, 0.4100, 0.4300, 0.4500, 0.4800, 0.5000, 0.5300, 0.5600, 0.5900,
|
||||
], // col 4
|
||||
[
|
||||
0.3400, 0.4600, 0.6300, 0.7700, 0.9100, 1.080, 0.0, 0.0, 0.0, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
], // col 5
|
||||
[0.0; 20], // col 6
|
||||
[
|
||||
0.0064, 0.1100, 0.2200, 0.4100, 0.9400, 1.000, 1.300, 1.600, 0.0, 0.0,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
], // col 7
|
||||
[0.0; 20], // col 8
|
||||
[
|
||||
0.0370, 0.0650, 0.1300, 0.2400, 0.5500, 0.6300, 0.7700, 0.9500, 1.100, 1.250,
|
||||
0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
], // col 9
|
||||
[0.0; 20], // col 10
|
||||
[
|
||||
0.0220, 0.0390, 0.0800, 0.1500, 0.3500, 0.4000, 0.4900, 0.6200, 0.7200, 0.7800,
|
||||
0.8500, 0.9300, 1.020, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0,
|
||||
], // col 11
|
||||
[0.0; 20], // col 12
|
||||
[0.0; 20], // col 13
|
||||
[0.0; 20], // col 14
|
||||
[0.0; 20], // col 15
|
||||
[0.0; 20], // col 16
|
||||
[0.0; 20], // col 17
|
||||
[0.0; 20], // col 18
|
||||
[0.0; 20], // col 19
|
||||
[0.0; 20], // col 20
|
||||
[0.0; 20], // col 21
|
||||
[0.0; 20], // col 22
|
||||
[0.0; 20], // col 23
|
||||
[0.0; 20], // col 24
|
||||
];
|
||||
|
||||
/// 光速 (cm/s)
|
||||
const C_LIGHT: f64 = 2.997925e18;
|
||||
|
||||
/// 截面单位转换因子 (cm^2)
|
||||
const SIG_FACTOR: f64 = 1.0e-18;
|
||||
|
||||
/// Hidalgo (1968) 光致电离截面插值。
|
||||
///
|
||||
/// 根据 Hidalgo 数据表，对给定频率进行线性插值。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `ib` - 物种标识（负值，`INDEX = -IB - 100`）
|
||||
/// * `fr` - 频率 (Hz)
|
||||
///
|
||||
/// # 返回值
|
||||
///
|
||||
/// 光致电离截面 (cm^2)
|
||||
pub fn hidalg(ib: i32, fr: f64) -> f64 {
|
||||
let index = (-ib - 101) as usize; // 转为 0-indexed
|
||||
if index >= 24 {
|
||||
return 0.0;
|
||||
}
|
||||
|
||||
// 根据 INDEX 选择波长网格和数据
|
||||
let num = if index < 12 { 20 } else { 15 };
|
||||
let wli = if index < 12 { &WL1 } else { &WL2 };
|
||||
let sigs = &SIG0[index];
|
||||
|
||||
// 将频率转换为波长 (Å)
|
||||
let wlam = C_LIGHT / fr;
|
||||
|
||||
// 查找插值区间
|
||||
let mut il = 0;
|
||||
let mut ir = num - 1;
|
||||
for i in 0..num - 1 {
|
||||
if wlam >= wli[i] && wlam <= wli[i + 1] {
|
||||
il = i;
|
||||
ir = i + 1;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// 线性插值
|
||||
let mut sigm = if wli[ir] - wli[il] > 0.0 {
|
||||
(sigs[ir] - sigs[il]) * (wlam - wli[il]) / (wli[ir] - wli[il]) + sigs[il]
|
||||
} else {
|
||||
sigs[il]
|
||||
};
|
||||
|
||||
// 边界处理
|
||||
if wlam <= wli[0] {
|
||||
sigm = sigs[0];
|
||||
}
|
||||
if wlam >= wli[num - 1] {
|
||||
sigm = sigs[num - 1];
|
||||
}
|
||||
|
||||
sigm * SIG_FACTOR
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_hidalg_in_range() {
|
||||
// 测试 H I (IB=-101, INDEX=0) 在有效波长范围内
|
||||
// 100 Å 对应频率
|
||||
let fr = C_LIGHT / 100.0;
|
||||
let result = hidalg(-101, fr);
|
||||
// H I 数据全为 0，所以结果应为 0
|
||||
assert!(result >= 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hidalg_species_2() {
|
||||
// 测试物种 2 (IB=-102, INDEX=1)
|
||||
let fr = C_LIGHT / 50.0; // 50 Å
|
||||
let result = hidalg(-102, fr);
|
||||
assert!(result >= 0.0);
|
||||
assert!(result.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hidalg_invalid_index() {
|
||||
let result = hidalg(-125, C_LIGHT / 100.0);
|
||||
assert_eq!(result, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hidalg_above_range() {
|
||||
// 高于波长范围时返回末值
|
||||
let fr = C_LIGHT / 200.0;
|
||||
let result = hidalg(-102, fr);
|
||||
assert!(result >= 0.0);
|
||||
}
|
||||
}
|
||||
470
src/synspec/math/hydini.rs
Normal file
470
src/synspec/math/hydini.rs
Normal file
@ -0,0 +1,470 @@
|
||||
//! Hydrogen line profile data initialization.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine HYDINI (line 6877).
|
||||
//!
|
||||
//! Initializes necessary arrays for evaluating hydrogen line profiles
|
||||
//! from the Lemke, Tremblay-Bergeron, or Schoening-Butler tables.
|
||||
|
||||
#![allow(clippy::never_loop)]
|
||||
use std::fs::File;
|
||||
use std::io::{BufRead, BufReader};
|
||||
|
||||
use super::stark0::stark0;
|
||||
|
||||
/// Constants for hydrogen profile arrays
|
||||
pub const NLINES_MAX: usize = 22;
|
||||
pub const NLEVELS: usize = 4;
|
||||
pub const NWL_MAX: usize = 100;
|
||||
pub const NT_MAX: usize = 20;
|
||||
pub const NE_MAX: usize = 20;
|
||||
|
||||
/// Hydrogen line profile table data
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct HydProfileTable {
|
||||
/// Line index (i, j)
|
||||
pub i: usize,
|
||||
pub j: usize,
|
||||
/// Central wavelength
|
||||
pub wl0: f64,
|
||||
/// Number of wavelength points
|
||||
pub nwl: usize,
|
||||
/// Number of temperature points
|
||||
pub nt: usize,
|
||||
/// Number of electron density points
|
||||
pub ne: usize,
|
||||
/// Log10 wavelength displacements [NWL_MAX]
|
||||
pub wl: [f64; NWL_MAX],
|
||||
/// Log10 temperature grid [NT_MAX]
|
||||
pub xt: [f64; NT_MAX],
|
||||
/// Log10 electron density grid [NE_MAX]
|
||||
pub xne: [f64; NE_MAX],
|
||||
/// Profile values [NWL_MAX x NT_MAX x NE_MAX]
|
||||
pub prf: [[[f64; NE_MAX]; NT_MAX]; NWL_MAX],
|
||||
/// Asymptotic profile coefficient
|
||||
pub xk: f64,
|
||||
}
|
||||
|
||||
impl Default for HydProfileTable {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
i: 0,
|
||||
j: 0,
|
||||
wl0: 0.0,
|
||||
nwl: 0,
|
||||
nt: 0,
|
||||
ne: 0,
|
||||
wl: [0.0; NWL_MAX],
|
||||
xt: [0.0; NT_MAX],
|
||||
xne: [0.0; NE_MAX],
|
||||
prf: [[[0.0; NE_MAX]; NT_MAX]; NWL_MAX],
|
||||
xk: 0.0,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Hydrogen line initialization result
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct HydInitResult {
|
||||
/// Central wavelengths for lines [NLEVELS x NLINES_MAX]
|
||||
pub wline: [[f64; NLINES_MAX]; NLEVELS],
|
||||
/// Line index mapping [NLEVELS x NLINES_MAX]
|
||||
pub ilin0: [[usize; NLINES_MAX]; NLEVELS],
|
||||
/// Profile tables
|
||||
pub tables: Vec<HydProfileTable>,
|
||||
/// Lemke mode flag
|
||||
pub ilemke: bool,
|
||||
/// Number of lines
|
||||
pub nlihyd: usize,
|
||||
}
|
||||
|
||||
impl Default for HydInitResult {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
wline: [[0.0; NLINES_MAX]; NLEVELS],
|
||||
ilin0: [[0; NLINES_MAX]; NLEVELS],
|
||||
tables: Vec::new(),
|
||||
ilemke: false,
|
||||
nlihyd: 0,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Hydrogen line profile table source
|
||||
#[derive(Debug, Clone, Copy, PartialEq)]
|
||||
pub enum HydTableSource {
|
||||
/// Schoening-Butler tables (ihydpr < 0)
|
||||
SchoeningButler,
|
||||
/// Lemke tables (ihydpr = 21)
|
||||
Lemke,
|
||||
/// Tremblay-Bergeron tables (ihydpr = 22)
|
||||
Tremblay,
|
||||
}
|
||||
|
||||
/// Parameters for HYDINI
|
||||
pub struct HydiniParams {
|
||||
/// Table source selection
|
||||
pub source: HydTableSource,
|
||||
/// Path to data directory
|
||||
pub data_dir: String,
|
||||
/// Model depth points
|
||||
pub nd: usize,
|
||||
/// Temperature array [nd]
|
||||
pub temp: Vec<f64>,
|
||||
/// Electron density array [nd]
|
||||
pub elec: Vec<f64>,
|
||||
/// Turbulent velocity array [nd]
|
||||
pub vturb: Vec<f64>,
|
||||
}
|
||||
|
||||
/// Initialize hydrogen line profile data.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Initialization parameters
|
||||
///
|
||||
/// # Returns
|
||||
/// Hydrogen line initialization result with profile tables
|
||||
pub fn hydini(params: &HydiniParams) -> std::io::Result<HydInitResult> {
|
||||
let mut result = HydInitResult::default();
|
||||
|
||||
// Initialize central wavelengths using STARK0
|
||||
for i in 0..NLEVELS {
|
||||
for j in (i + 1)..NLINES_MAX {
|
||||
let stark = stark0(i as i32 + 1, j as i32 + 1, 1);
|
||||
result.wline[i][j] = stark.wl0;
|
||||
}
|
||||
}
|
||||
|
||||
// Initialize line index mapping
|
||||
for i in 0..NLEVELS {
|
||||
for j in 0..NLINES_MAX {
|
||||
result.ilin0[i][j] = 0;
|
||||
}
|
||||
}
|
||||
|
||||
match params.source {
|
||||
HydTableSource::SchoeningButler => {
|
||||
read_schoening_butler(params, &mut result)?;
|
||||
}
|
||||
HydTableSource::Lemke | HydTableSource::Tremblay => {
|
||||
read_lemke_tremblay(params, &mut result)?;
|
||||
}
|
||||
}
|
||||
|
||||
Ok(result)
|
||||
}
|
||||
|
||||
/// Read Schoening-Butler tables
|
||||
fn read_schoening_butler(
|
||||
params: &HydiniParams,
|
||||
result: &mut HydInitResult,
|
||||
) -> std::io::Result<()> {
|
||||
let filename = format!("{}/hydprf.dat", params.data_dir);
|
||||
let file = File::open(&filename)?;
|
||||
let reader = BufReader::new(file);
|
||||
let mut lines = reader.lines();
|
||||
|
||||
// Skip 12 header lines
|
||||
for _ in 0..12 {
|
||||
lines.next();
|
||||
}
|
||||
|
||||
let nline = 12;
|
||||
result.ilemke = false;
|
||||
|
||||
for iline in 0..nline {
|
||||
// Read line indices
|
||||
let header = read_next_line(&mut lines)?;
|
||||
let (i, j) = parse_line_indices(&header)?;
|
||||
|
||||
let j = if iline == 11 { 10 } else { j }; // Special case for last line
|
||||
let wl0 = result.wline[i - 1][j - 1];
|
||||
result.ilin0[i - 1][j - 1] = iline + 1;
|
||||
|
||||
let mut table = HydProfileTable {
|
||||
i,
|
||||
j,
|
||||
wl0,
|
||||
..Default::default()
|
||||
};
|
||||
|
||||
// Read wavelength points
|
||||
let wl_line = read_next_line(&mut lines)?;
|
||||
let wl_parts = parse_data_line(&wl_line)?;
|
||||
let nwl = wl_parts.len() - 1; // First value is character
|
||||
table.nwl = nwl;
|
||||
for k in 0..nwl.min(NWL_MAX) {
|
||||
table.wl[k] = if wl_parts[k + 1] < 1.0e-4 {
|
||||
(1.0e-4_f64).log10()
|
||||
} else {
|
||||
wl_parts[k + 1].log10()
|
||||
};
|
||||
}
|
||||
|
||||
// Read temperature points
|
||||
let xt_line = read_next_line(&mut lines)?;
|
||||
let xt_parts = parse_data_line(&xt_line)?;
|
||||
let nt = xt_parts.len() - 1;
|
||||
table.nt = nt;
|
||||
for k in 0..nt.min(NT_MAX) {
|
||||
table.xt[k] = xt_parts[k + 1];
|
||||
}
|
||||
|
||||
// Read electron density points
|
||||
let xne_line = read_next_line(&mut lines)?;
|
||||
let xne_parts = parse_data_line(&xne_line)?;
|
||||
let ne = xne_parts.len() - 1;
|
||||
table.ne = ne;
|
||||
for k in 0..ne.min(NE_MAX) {
|
||||
table.xne[k] = xne_parts[k + 1];
|
||||
}
|
||||
|
||||
// Skip blank line
|
||||
lines.next();
|
||||
|
||||
// Read profile data
|
||||
for ie in 0..ne.min(NE_MAX) {
|
||||
for it in 0..nt.min(NT_MAX) {
|
||||
lines.next(); // Skip blank line
|
||||
let prf_line = read_next_line(&mut lines)?;
|
||||
let prf_parts = parse_data_line(&prf_line)?;
|
||||
for iwl in 0..nwl.min(NWL_MAX) {
|
||||
if iwl < prf_parts.len() {
|
||||
table.prf[iwl][it][ie] = prf_parts[iwl];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Compute asymptotic profile coefficient
|
||||
if nwl > 0 && ne > 0 {
|
||||
let xclog = table.prf[nwl - 1][0][0]
|
||||
+ 2.5 * table.wl[nwl - 1]
|
||||
+ 31.5304
|
||||
- table.xne[0]
|
||||
- 2.0 * wl0.log10();
|
||||
let xklog = 0.6666667 * (xclog - 0.176);
|
||||
table.xk = (xklog * std::f64::consts::LN_10).exp();
|
||||
}
|
||||
|
||||
result.tables.push(table);
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Read Lemke or Tremblay tables
|
||||
fn read_lemke_tremblay(
|
||||
params: &HydiniParams,
|
||||
result: &mut HydInitResult,
|
||||
) -> std::io::Result<()> {
|
||||
let filename = match params.source {
|
||||
HydTableSource::Lemke => format!("{}/lemke.dat", params.data_dir),
|
||||
HydTableSource::Tremblay => format!("{}/tremblay.dat", params.data_dir),
|
||||
_ => unreachable!(),
|
||||
};
|
||||
|
||||
let file = File::open(&filename)?;
|
||||
let reader = BufReader::new(file);
|
||||
let mut lines = reader.lines();
|
||||
|
||||
result.ilemke = true;
|
||||
|
||||
// Read number of tables
|
||||
let ntab_line = read_next_line(&mut lines)?;
|
||||
let ntab: usize = ntab_line.trim().parse().map_err(|e| {
|
||||
std::io::Error::new(std::io::ErrorKind::InvalidData, format!("NTAB: {}", e))
|
||||
})?;
|
||||
|
||||
let mut iline = 0;
|
||||
|
||||
for _ in 0..ntab {
|
||||
// Read number of lines in this table
|
||||
let nlly_line = read_next_line(&mut lines)?;
|
||||
let nlly: usize = nlly_line.trim().parse().map_err(|e| {
|
||||
std::io::Error::new(std::io::ErrorKind::InvalidData, format!("NLLY: {}", e))
|
||||
})?;
|
||||
|
||||
let ilineb = iline;
|
||||
|
||||
// Read line parameters
|
||||
for _ in 0..nlly {
|
||||
let param_line = read_next_line(&mut lines)?;
|
||||
let parts = parse_data_line(¶m_line)?;
|
||||
|
||||
if parts.len() < 11 {
|
||||
return Err(std::io::Error::new(
|
||||
std::io::ErrorKind::InvalidData,
|
||||
"Invalid Lemke/Tremblay parameter line",
|
||||
));
|
||||
}
|
||||
|
||||
let i = parts[0] as usize;
|
||||
let j = parts[1] as usize;
|
||||
let almin = parts[2];
|
||||
let anemin = parts[3];
|
||||
let tmin = parts[4];
|
||||
let dla = parts[5];
|
||||
let dle = parts[6];
|
||||
let dlt = parts[7];
|
||||
let nwl = parts[8] as usize;
|
||||
let ne = parts[9] as usize;
|
||||
let nt = parts[10] as usize;
|
||||
|
||||
let wl0 = result.wline[i - 1][j - 1];
|
||||
result.ilin0[i - 1][j - 1] = iline + 1;
|
||||
|
||||
let mut table = HydProfileTable {
|
||||
i,
|
||||
j,
|
||||
wl0,
|
||||
nwl,
|
||||
nt,
|
||||
ne,
|
||||
..Default::default()
|
||||
};
|
||||
|
||||
// Generate wavelength grid
|
||||
for iwl in 0..nwl.min(NWL_MAX) {
|
||||
table.wl[iwl] = almin + (iwl as f64) * dla;
|
||||
}
|
||||
|
||||
// Generate electron density grid
|
||||
for ie in 0..ne.min(NE_MAX) {
|
||||
table.xne[ie] = anemin + (ie as f64) * dle;
|
||||
}
|
||||
|
||||
// Generate temperature grid
|
||||
for it in 0..nt.min(NT_MAX) {
|
||||
table.xt[it] = tmin + (it as f64) * dlt;
|
||||
}
|
||||
|
||||
result.tables.push(table);
|
||||
iline += 1;
|
||||
}
|
||||
|
||||
// Read profile data for each line
|
||||
for ili in 0..nlly {
|
||||
let ilne = ilineb + ili;
|
||||
let table = &mut result.tables[ilne];
|
||||
let nwl = table.nwl;
|
||||
let ne = table.ne;
|
||||
let nt = table.nt;
|
||||
|
||||
lines.next(); // Skip blank line
|
||||
|
||||
for ie in 0..ne.min(NE_MAX) {
|
||||
for it in 0..nt.min(NT_MAX) {
|
||||
let prf_line = read_next_line(&mut lines)?;
|
||||
let parts = parse_data_line(&prf_line)?;
|
||||
|
||||
// First value is QLT (quality factor), skip it
|
||||
for iwl in 0..nwl.min(NWL_MAX) {
|
||||
if iwl + 1 < parts.len() {
|
||||
table.prf[iwl][it][ie] = parts[iwl + 1];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Compute asymptotic profile coefficient
|
||||
if nwl > 0 && ne > 0 {
|
||||
let xclog = table.prf[nwl - 1][0][0]
|
||||
+ 2.5 * table.wl[nwl - 1].log10()
|
||||
+ 31.5304
|
||||
- table.xne[0]
|
||||
- 2.0 * table.wl0.log10();
|
||||
let xklog = 0.6666667 * (xclog - 0.176);
|
||||
table.xk = (xklog * std::f64::consts::LN_10).exp();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
result.nlihyd = iline;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Read next non-empty line
|
||||
fn read_next_line(lines: &mut impl Iterator<Item = std::io::Result<String>>) -> std::io::Result<String> {
|
||||
loop {
|
||||
match lines.next() {
|
||||
Some(Ok(line)) => return Ok(line),
|
||||
Some(Err(e)) => return Err(e),
|
||||
None => return Err(std::io::Error::new(
|
||||
std::io::ErrorKind::UnexpectedEof,
|
||||
"Unexpected end of file",
|
||||
)),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// Parse line indices from header: FORMAT(12X,I1,9X,I1)
|
||||
fn parse_line_indices(line: &str) -> std::io::Result<(usize, usize)> {
|
||||
let parts: Vec<&str> = line.split_whitespace().collect();
|
||||
if parts.len() < 2 {
|
||||
return Err(std::io::Error::new(
|
||||
std::io::ErrorKind::InvalidData,
|
||||
format!("Invalid line indices: {}", line),
|
||||
));
|
||||
}
|
||||
|
||||
let i = parts[0].parse::<usize>().map_err(|e| {
|
||||
std::io::Error::new(std::io::ErrorKind::InvalidData, format!("I: {}", e))
|
||||
})?;
|
||||
let j = parts[1].parse::<usize>().map_err(|e| {
|
||||
std::io::Error::new(std::io::ErrorKind::InvalidData, format!("J: {}", e))
|
||||
})?;
|
||||
|
||||
Ok((i, j))
|
||||
}
|
||||
|
||||
/// Parse data line (free format)
|
||||
fn parse_data_line(line: &str) -> std::io::Result<Vec<f64>> {
|
||||
let values: Vec<f64> = line
|
||||
.split_whitespace()
|
||||
.filter_map(|s| s.parse::<f64>().ok())
|
||||
.collect();
|
||||
|
||||
Ok(values)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_hydini_default() {
|
||||
let result = HydInitResult::default();
|
||||
assert_eq!(result.wline.len(), NLEVELS);
|
||||
assert_eq!(result.ilin0.len(), NLEVELS);
|
||||
assert!(result.tables.is_empty());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hyd_profile_table_default() {
|
||||
let table = HydProfileTable::default();
|
||||
assert_eq!(table.nwl, 0);
|
||||
assert_eq!(table.nt, 0);
|
||||
assert_eq!(table.ne, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parse_line_indices() {
|
||||
let line = " 1 2";
|
||||
let result = parse_line_indices(line);
|
||||
assert!(result.is_ok());
|
||||
let (i, j) = result.unwrap();
|
||||
assert_eq!(i, 1);
|
||||
assert_eq!(j, 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parse_data_line() {
|
||||
let line = " 1.0 2.0 3.0 4.0";
|
||||
let result = parse_data_line(line);
|
||||
assert!(result.is_ok());
|
||||
let values = result.unwrap();
|
||||
assert_eq!(values.len(), 4);
|
||||
assert!((values[0] - 1.0).abs() < 1e-10);
|
||||
}
|
||||
}
|
||||
390
src/synspec/math/hydlin.rs
Normal file
390
src/synspec/math/hydlin.rs
Normal file
@ -0,0 +1,390 @@
|
||||
//! Hydrogen line opacity calculation for SYNSPEC.
|
||||
//!
|
||||
//! Translated from SYNSPEC `HYDLIN` subroutine (synspec54.f:5425).
|
||||
//!
|
||||
//! Calculates opacity and emissivity of hydrogen lines including:
|
||||
//! - Stark broadening (analytic profiles)
|
||||
//! - Allard quasi-molecular satellite opacity
|
||||
//! - Far-infrared hydrogen lines
|
||||
|
||||
use super::stark0::stark0;
|
||||
use super::starka::starka;
|
||||
use super::starkir::starkir;
|
||||
use super::divstr::divstr;
|
||||
use super::allard::{self, AllardData};
|
||||
use super::lyahhe::lyahhe;
|
||||
|
||||
/// Physical constants for hydrogen line calculations
|
||||
const CPP: f64 = 4.1412e-16;
|
||||
const CPJ: f64 = 157803.0;
|
||||
const C00: f64 = 1.25e-9;
|
||||
const CID: f64 = 0.02654;
|
||||
const CINV: f64 = 1.0 / 2.997925e18;
|
||||
|
||||
/// Parameters for hydrogen line opacity calculation
|
||||
pub struct HydlinParams {
|
||||
/// Depth index
|
||||
pub id: usize,
|
||||
/// Start frequency index
|
||||
pub i0: usize,
|
||||
/// End frequency index
|
||||
pub i1: usize,
|
||||
/// Number of frequencies
|
||||
pub nfreq: usize,
|
||||
/// Wavelength array (Å)
|
||||
pub wlam: Vec<f64>,
|
||||
/// Frequency array (Hz)
|
||||
pub freq: Vec<f64>,
|
||||
/// Temperature (K)
|
||||
pub t: f64,
|
||||
/// Electron density
|
||||
pub ane: f64,
|
||||
/// H atom exists
|
||||
pub iath: i32,
|
||||
/// Lower level for H lines
|
||||
pub ilowh: i32,
|
||||
/// Upper level limit
|
||||
pub m10: usize,
|
||||
pub m20: usize,
|
||||
/// H ground level population
|
||||
pub pop_h: f64,
|
||||
/// H continuum level population
|
||||
pub pop_h_cont: f64,
|
||||
/// Turbulent velocity
|
||||
pub vturb: f64,
|
||||
/// wnHint factors
|
||||
pub wn_hint: Vec<Vec<f64>>,
|
||||
/// Quasi-molecular Lyman-alpha flag (>0: include)
|
||||
pub nunalp: i32,
|
||||
/// Quasi-molecular Lyman-beta flag (>0: include)
|
||||
pub nunbet: i32,
|
||||
/// Quasi-molecular Lyman-gamma flag (>0: include)
|
||||
pub nungam: i32,
|
||||
/// Quasi-molecular Balmer flag (>0: include)
|
||||
pub nunbal: i32,
|
||||
/// Allard quasi-molecular profile data (optional)
|
||||
pub allard_data: Option<AllardData>,
|
||||
/// Neutral H particle density at depth [cm⁻³]
|
||||
pub hneutr: f64,
|
||||
/// Ionized H particle density at depth [cm⁻³]
|
||||
pub hcharg: f64,
|
||||
/// Lyman-alpha He broadening flag (>0: include)
|
||||
pub nunhhe: i32,
|
||||
/// He atom index in atomic data (>0: He present)
|
||||
pub iathe: i32,
|
||||
/// He ground level population at depth
|
||||
pub pop_he: f64,
|
||||
}
|
||||
|
||||
/// Result of hydrogen line opacity calculation
|
||||
pub struct HydlinResult {
|
||||
/// Absorption coefficient array
|
||||
pub absoh: Vec<f64>,
|
||||
/// Emission coefficient array
|
||||
pub emish: Vec<f64>,
|
||||
}
|
||||
|
||||
/// Calculate hydrogen line opacity and emissivity.
|
||||
///
|
||||
/// Translates the full SYNSPEC HYDLIN subroutine including Stark broadening
|
||||
/// and infrared lines.
|
||||
pub fn hydlin(params: &HydlinParams) -> HydlinResult {
|
||||
let i0 = params.i0;
|
||||
let i1 = params.i1;
|
||||
let nfreq = params.nfreq;
|
||||
|
||||
let mut absoh = vec![0.0; nfreq];
|
||||
let mut emish = vec![0.0; nfreq];
|
||||
|
||||
// Skip if no hydrogen or empty arrays
|
||||
if params.iath <= 0 || params.wlam.is_empty() || params.ilowh <= 0 {
|
||||
return HydlinResult { absoh, emish };
|
||||
}
|
||||
|
||||
let t = params.t;
|
||||
let t1 = 1.0 / t;
|
||||
let sqt = t.sqrt();
|
||||
let ane = params.ane;
|
||||
let anes = (ane.ln() / 6.0).exp();
|
||||
|
||||
// Population of level 2 (for Saha)
|
||||
let anp = params.pop_h_cont;
|
||||
let pp = CPP * ane * anp * t1 / sqt;
|
||||
|
||||
// Level populations
|
||||
let nlh = params.wn_hint.len().min(3);
|
||||
let mut pj = vec![0.0f64; 50];
|
||||
for il in 0..50 {
|
||||
let x = ((il + 1) * (il + 1)) as f64;
|
||||
if il < nlh {
|
||||
pj[il] = params.pop_h * (-CPJ / x * t1).exp() * x;
|
||||
} else {
|
||||
let wn = if il < params.wn_hint.len() && params.id < params.wn_hint[il].len() {
|
||||
params.wn_hint[il][params.id]
|
||||
} else {
|
||||
1.0
|
||||
};
|
||||
pj[il] = pp * (CPJ / x * t1).exp() * x * wn;
|
||||
}
|
||||
}
|
||||
|
||||
// Frequency-independent Stark parameters
|
||||
let f00 = C00 * anes * anes * anes * anes;
|
||||
let dop0 = 1.0e8 * (1.65e8 * t + params.vturb).sqrt();
|
||||
|
||||
// Determine spectral series range
|
||||
let iserl = params.ilowh as usize;
|
||||
let mut iseru = params.ilowh as usize;
|
||||
|
||||
if i0 < params.wlam.len() {
|
||||
let wl = params.wlam[i0];
|
||||
if wl > 14000.0 { iseru = 4; }
|
||||
if wl > 22700.0 { iseru = 5; }
|
||||
if wl > 32800.0 { iseru = 6; }
|
||||
if wl > 44660.0 { iseru = 7; }
|
||||
}
|
||||
|
||||
// Loop over spectral series
|
||||
for i in iserl..=iseru.min(40) {
|
||||
let ii = (i * i) as f64;
|
||||
let xii = 1.0 / ii;
|
||||
let popi = if i - 1 < pj.len() { pj[i - 1] } else { 0.0 };
|
||||
|
||||
// Determine contributing lines
|
||||
let m1 = (i + 1).max(params.m10);
|
||||
let m2 = (i + 4).min(params.m20).min(40);
|
||||
|
||||
for j in m1..=m2 {
|
||||
let jj = (j * j) as f64;
|
||||
let xjj = 1.0 / jj;
|
||||
|
||||
// Transition properties
|
||||
let wn_j = if j - 1 < params.wn_hint.len() && params.id < params.wn_hint[j - 1].len() {
|
||||
params.wn_hint[j - 1][params.id]
|
||||
} else {
|
||||
1.0
|
||||
};
|
||||
let wn_i = if i - 1 < params.wn_hint.len() && params.id < params.wn_hint[i - 1].len() {
|
||||
params.wn_hint[i - 1][params.id]
|
||||
} else {
|
||||
1.0
|
||||
};
|
||||
|
||||
let abtra = popi * wn_j;
|
||||
let emtra = if j - 1 < pj.len() {
|
||||
pj[j - 1] * wn_i * ii * xjj * (CPJ * (xii - xjj) * t1).exp()
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
|
||||
// Use analytic Stark profile
|
||||
let stark = stark0(i as i32, j as i32, 1);
|
||||
let wl0 = stark.wl0;
|
||||
let xkij = stark.xkij;
|
||||
let fij = stark.fij;
|
||||
|
||||
// Check if line contributes in this wavelength region
|
||||
let wlam_i1 = if i1 < params.wlam.len() { params.wlam[i1] } else { 0.0 };
|
||||
let wlam_i0 = if i0 < params.wlam.len() { params.wlam[i0] } else { 0.0 };
|
||||
let in_range = (wl0 <= wlam_i1 && 1.25 * wl0 > wlam_i0)
|
||||
|| (wl0 >= wlam_i0 && 0.75 * wl0 < wlam_i1);
|
||||
|
||||
if in_range {
|
||||
let fxk = f00 * xkij;
|
||||
if fxk.abs() < 1.0e-30 { continue; }
|
||||
let fxk1 = 1.0 / fxk;
|
||||
let dop = dop0 / wl0;
|
||||
let dbeta = wl0 * wl0 * CINV * fxk1;
|
||||
let betad = dop * dbeta;
|
||||
let fid = CID * fij * dbeta;
|
||||
let (ad, div) = divstr(betad);
|
||||
|
||||
// Quasi-molecular opacity check (Lyman alpha/beta/gamma, Balmer alpha)
|
||||
let lquasi = (i == 1 && j == 2 && params.nunalp > 0)
|
||||
|| (i == 1 && j == 3 && params.nunbet > 0)
|
||||
|| (i == 1 && j == 4 && params.nungam > 0)
|
||||
|| (i == 2 && j == 3 && params.nunbal > 0);
|
||||
|
||||
if lquasi && params.allard_data.is_some() {
|
||||
// Allard quasi-molecular + Stark profile
|
||||
let ad_ref = params.allard_data.as_ref().unwrap();
|
||||
for ij in i0..=i1.min(nfreq - 1) {
|
||||
let wl = params.wlam[ij];
|
||||
let beta = (wl - wl0).abs() * fxk1;
|
||||
let sg_allard = allard::allard(ad_ref, wl, params.hneutr, params.hcharg, i as i32, j as i32);
|
||||
let sg = sg_allard + starka(beta, betad, ad, div, 2.0) * fid;
|
||||
absoh[ij] += sg * abtra;
|
||||
emish[ij] += sg * emtra;
|
||||
}
|
||||
} else {
|
||||
// Standard Stark profile
|
||||
for ij in i0..=i1.min(nfreq - 1) {
|
||||
let beta = (params.wlam[ij] - wl0).abs() * fxk1;
|
||||
let sg = if i < 5 {
|
||||
starka(beta, betad, ad, div, 2.0) * fid
|
||||
} else {
|
||||
starkir(ii as i32, jj as i32, t, ane, beta, dbeta) * fid
|
||||
};
|
||||
absoh[ij] += sg * abtra;
|
||||
emish[ij] += sg * emtra;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Lyman-alpha broadening by helium (Lyahhe)
|
||||
let lalhhe = i == 1 && j == 2 && params.nunhhe > 0;
|
||||
if lalhhe && params.iathe > 0 && params.pop_he > 0.0 {
|
||||
let rel = 1.0 / std::f64::consts::TAU;
|
||||
for ij in i0..=i1.min(nfreq - 1) {
|
||||
let sg0 = lyahhe(params.wlam[ij], params.pop_he);
|
||||
let sg = sg0 * rel;
|
||||
absoh[ij] += sg * abtra;
|
||||
emish[ij] += sg * emtra;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Far infrared hydrogen lines
|
||||
if i1 < nfreq && !params.wlam.is_empty() && params.wlam[i1.min(params.wlam.len() - 1)] > 70000.0 {
|
||||
for i in 8..=13 {
|
||||
let ii = (i * i) as f64;
|
||||
let xii = 1.0 / ii;
|
||||
for j in (i + 1)..=(i + 4).min(40) {
|
||||
let jj = (j * j) as f64;
|
||||
let xjj = 1.0 / jj;
|
||||
|
||||
let stark = stark0(i as i32, j as i32, 1);
|
||||
let wl0 = stark.wl0;
|
||||
let xkij = stark.xkij;
|
||||
let fij = stark.fij;
|
||||
|
||||
let wlam_i1 = params.wlam[i1.min(params.wlam.len() - 1)];
|
||||
let wlam_i0 = params.wlam[i0.min(params.wlam.len() - 1)];
|
||||
let in_range = (wl0 <= wlam_i1 && 1.5 * wl0 > wlam_i0)
|
||||
|| (wl0 >= wlam_i0 && 0.5 * wl0 < wlam_i1);
|
||||
|
||||
if in_range {
|
||||
let fxk = f00 * xkij;
|
||||
if fxk.abs() < 1.0e-30 { continue; }
|
||||
let fxk1 = 1.0 / fxk;
|
||||
let dop = dop0 / wl0;
|
||||
let dbeta = wl0 * wl0 * CINV * fxk1;
|
||||
let _betad = dop * dbeta;
|
||||
let fid = CID * fij * dbeta;
|
||||
|
||||
let wn_j = if j - 1 < params.wn_hint.len() && params.id < params.wn_hint[j - 1].len() {
|
||||
params.wn_hint[j - 1][params.id]
|
||||
} else {
|
||||
1.0
|
||||
};
|
||||
let wn_i = if i - 1 < params.wn_hint.len() && params.id < params.wn_hint[i - 1].len() {
|
||||
params.wn_hint[i - 1][params.id]
|
||||
} else {
|
||||
1.0
|
||||
};
|
||||
let popi = if i - 1 < pj.len() { pj[i - 1] } else { 0.0 };
|
||||
let popj = if j - 1 < pj.len() { pj[j - 1] } else { 0.0 };
|
||||
let abtra = popi * wn_j;
|
||||
let emtra = popj * wn_i * ii * xjj * (CPJ * (xii - xjj) * t1).exp();
|
||||
|
||||
for ij in i0..=i1.min(nfreq - 1) {
|
||||
let beta = (params.wlam[ij] - wl0).abs() * fxk1;
|
||||
let sg = starkir(ii as i32, jj as i32, t, ane, beta, dbeta) * fid;
|
||||
absoh[ij] += sg * abtra;
|
||||
emish[ij] += sg * emtra;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Total opacity and emissivity (stimulated emission correction)
|
||||
for ij in i0..=i1.min(nfreq - 1) {
|
||||
let f = params.freq[ij];
|
||||
let f15 = f * 1.0e-15;
|
||||
let xkf = (-4.79928e-11 * f * t1).exp();
|
||||
let xkfb = xkf * 1.4743e-2 * f15 * f15 * f15;
|
||||
absoh[ij] -= xkf * emish[ij];
|
||||
emish[ij] *= xkfb;
|
||||
}
|
||||
|
||||
HydlinResult { absoh, emish }
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_hydlin_basic() {
|
||||
let params = HydlinParams {
|
||||
id: 0,
|
||||
i0: 0,
|
||||
i1: 4,
|
||||
nfreq: 5,
|
||||
wlam: vec![10000.0, 12000.0, 14000.0, 16000.0, 18000.0],
|
||||
freq: vec![3.0e14, 2.5e14, 2.14e14, 1.87e14, 1.67e14],
|
||||
t: 6000.0,
|
||||
ane: 1.0e13,
|
||||
iath: 1,
|
||||
ilowh: 1,
|
||||
m10: 2,
|
||||
m20: 10,
|
||||
pop_h: 1.0e16,
|
||||
pop_h_cont: 1.0e10,
|
||||
vturb: 2.0e5,
|
||||
wn_hint: vec![vec![1.0; 10]; 50],
|
||||
nunalp: 0,
|
||||
nunbet: 0,
|
||||
nungam: 0,
|
||||
nunbal: 0,
|
||||
allard_data: None,
|
||||
hneutr: 0.0,
|
||||
hcharg: 0.0,
|
||||
nunhhe: 0,
|
||||
iathe: 0,
|
||||
pop_he: 0.0,
|
||||
};
|
||||
|
||||
let result = hydlin(¶ms);
|
||||
assert!(result.absoh.iter().all(|&x| x.is_finite()));
|
||||
assert!(result.emish.iter().all(|&x| x.is_finite()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hydlin_no_hydrogen() {
|
||||
let params = HydlinParams {
|
||||
id: 0,
|
||||
i0: 0,
|
||||
i1: 4,
|
||||
nfreq: 5,
|
||||
wlam: vec![10000.0, 12000.0, 14000.0, 16000.0, 18000.0],
|
||||
freq: vec![3.0e14, 2.5e14, 2.14e14, 1.87e14, 1.67e14],
|
||||
t: 6000.0,
|
||||
ane: 1.0e13,
|
||||
iath: 0,
|
||||
ilowh: 1,
|
||||
m10: 2,
|
||||
m20: 10,
|
||||
pop_h: 1.0e16,
|
||||
pop_h_cont: 1.0e10,
|
||||
vturb: 2.0e5,
|
||||
wn_hint: vec![vec![1.0; 10]; 50],
|
||||
nunalp: 0,
|
||||
nunbet: 0,
|
||||
nungam: 0,
|
||||
nunbal: 0,
|
||||
allard_data: None,
|
||||
hneutr: 0.0,
|
||||
hcharg: 0.0,
|
||||
nunhhe: 0,
|
||||
iathe: 0,
|
||||
pop_he: 0.0,
|
||||
};
|
||||
|
||||
let result = hydlin(¶ms);
|
||||
assert!(result.absoh.iter().all(|&x| x == 0.0));
|
||||
assert!(result.emish.iter().all(|&x| x == 0.0));
|
||||
}
|
||||
}
|
||||
706
src/synspec/math/hydliw.rs
Normal file
706
src/synspec/math/hydliw.rs
Normal file
@ -0,0 +1,706 @@
|
||||
//! Hydrogen line opacity and emissivity for SYNSPEC (frequency window mode).
|
||||
//!
|
||||
//! Translated from SYNSPEC `HYDLIW` subroutine (synspec54.f:5798).
|
||||
//!
|
||||
//! Calculates opacity and emissivity of hydrogen lines in the frequency
|
||||
//! window mode. This is the window-mode variant of `hydlin`.
|
||||
|
||||
use super::{allard, divstr, feautr, lyahhe::lyahhe, stark0, starka, FeautrParams};
|
||||
|
||||
// ============================================================================
|
||||
// Physical constants
|
||||
// ============================================================================
|
||||
|
||||
const UN: f64 = 1.0;
|
||||
const TWO: f64 = 2.0;
|
||||
const SIXTH: f64 = 1.0 / 6.0;
|
||||
const CPP: f64 = 4.1412e-16;
|
||||
const CPJ: f64 = 157803.0;
|
||||
#[allow(dead_code)]
|
||||
const CPJ4: f64 = CPJ / 4.0;
|
||||
const C00: f64 = 1.25e-9;
|
||||
const CID: f64 = 0.02654;
|
||||
const CINV: f64 = UN / 2.997925e18;
|
||||
const AL10: f64 = std::f64::consts::LN_10;
|
||||
|
||||
// ============================================================================
|
||||
// Parameters
|
||||
// ============================================================================
|
||||
|
||||
/// Common input data for hydrogen line opacity calculations.
|
||||
pub struct HydliwCommon<'a> {
|
||||
/// Depth index.
|
||||
pub id: usize,
|
||||
/// Temperature at depth ID (K).
|
||||
pub t: f64,
|
||||
/// Electron density at depth ID.
|
||||
pub ane: f64,
|
||||
/// Turbulent velocity at depth ID (cm/s).
|
||||
pub vturb: f64,
|
||||
/// Surface gravity (log g).
|
||||
pub grav: f64,
|
||||
/// Frequency array (Hz).
|
||||
pub freq: &'a [f64],
|
||||
/// Wavelength array (Å).
|
||||
pub wlam: &'a [f64],
|
||||
/// H atom exists flag.
|
||||
pub iath: i32,
|
||||
/// Lyman line treatment switch.
|
||||
pub iophli: i32,
|
||||
/// Lemke profile table flag.
|
||||
pub ilemke: i32,
|
||||
/// H ground level population.
|
||||
pub pop_h_cont: f64,
|
||||
/// Level populations for H (up to 40 levels).
|
||||
/// If None, populations are computed from LTE/Saha.
|
||||
pub pj: Option<&'a [f64]>,
|
||||
/// WNHINT partition function values (indexed by level, depth).
|
||||
pub wnhint: &'a [f64],
|
||||
/// Number of wavelength points per hydrogen profile line.
|
||||
pub nwlhyd: &'a [usize],
|
||||
/// Log10 of profile values: prfhyd[line * 54 + iwl].
|
||||
pub prfhyd: &'a [f64],
|
||||
/// Log10 of wavelength grid per profile line: wlhyd[line * 54 + iwl].
|
||||
pub wlhyd: &'a [f64],
|
||||
/// Line wavelength table wline[i][j] for i<=4, j<=22.
|
||||
pub wline: &'a [f64],
|
||||
/// Oscillator strengths osch[i][j] for i<=4, j<=22.
|
||||
pub osch: &'a [f64],
|
||||
/// Profile line index table ilin0[i][j] for i<=4, j<=22.
|
||||
pub ilin0: &'a [i32],
|
||||
/// Number of NLTE H levels.
|
||||
pub nlh: usize,
|
||||
/// H ground level index in model.
|
||||
pub n0hn: usize,
|
||||
/// Feautrier parameters for Lyman-alpha.
|
||||
pub feautr_params: Option<&'a FeautrParams>,
|
||||
/// Laser delay flag.
|
||||
pub lasdel: bool,
|
||||
/// Quasi-molecular Lyman-alpha flag (>0: include).
|
||||
pub nunalp: i32,
|
||||
/// Quasi-molecular Lyman-beta flag (>0: include).
|
||||
pub nunbet: i32,
|
||||
/// Quasi-molecular Lyman-gamma flag (>0: include).
|
||||
pub nungam: i32,
|
||||
/// Quasi-molecular Balmer flag (>0: include).
|
||||
pub nunbal: i32,
|
||||
/// Allard quasi-molecular profile data (optional).
|
||||
/// If None, quasi-molecular opacity is skipped.
|
||||
pub allard_data: Option<&'a super::AllardData>,
|
||||
/// Neutral H particle density at depth [cm⁻³].
|
||||
pub hneutr: f64,
|
||||
/// Ionized H particle density at depth [cm⁻³].
|
||||
pub hcharg: f64,
|
||||
/// Lyman-alpha He broadening flag (>0: include).
|
||||
pub nunhhe: i32,
|
||||
/// He atom index in atomic data (>0: He present).
|
||||
pub iathe: i32,
|
||||
/// He ground level population at depth.
|
||||
pub pop_he: f64,
|
||||
}
|
||||
|
||||
/// Per-frequency window parameters for hydrogen lines.
|
||||
pub struct HydliwWindowParams<'a> {
|
||||
/// H line processing flag per frequency (-1: skip, >0: process).
|
||||
pub ihylw: &'a [i32],
|
||||
/// Lower series index per frequency.
|
||||
pub ilowhw: &'a [usize],
|
||||
/// Maximum principal quantum number per frequency.
|
||||
pub m10w: &'a [usize],
|
||||
/// Upper limit for H lines per frequency.
|
||||
pub m20w: &'a [usize],
|
||||
}
|
||||
|
||||
/// Input parameters for `hydliw`.
|
||||
pub struct HydliwParams<'a> {
|
||||
/// Common H data.
|
||||
pub common: HydliwCommon<'a>,
|
||||
/// Per-frequency window parameters.
|
||||
pub window: HydliwWindowParams<'a>,
|
||||
}
|
||||
|
||||
/// Result of `hydliw`.
|
||||
pub struct HydliwResult {
|
||||
/// Absorption coefficient array.
|
||||
pub absoh: Vec<f64>,
|
||||
/// Emission coefficient array.
|
||||
pub emish: Vec<f64>,
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// Implementation
|
||||
// ============================================================================
|
||||
|
||||
/// Calculate hydrogen line opacity and emissivity (frequency window mode).
|
||||
pub fn hydliw(params: &HydliwParams) -> HydliwResult {
|
||||
let c = ¶ms.common;
|
||||
let nf = c.freq.len();
|
||||
let mut abso = vec![0.0; nf];
|
||||
let mut emis = vec![0.0; nf];
|
||||
let mut absoh = vec![0.0; nf];
|
||||
let mut emish = vec![0.0; nf];
|
||||
|
||||
if c.iath <= 0 {
|
||||
return HydliwResult { absoh, emish };
|
||||
}
|
||||
|
||||
let t1 = UN / c.t;
|
||||
let sqt = c.t.sqrt();
|
||||
let anes = c.ane.powf(SIXTH);
|
||||
|
||||
// Populations of the first 40 levels of hydrogen
|
||||
let mut pj = [0.0f64; 40];
|
||||
let pp = CPP * c.ane * c.pop_h_cont * t1 / sqt;
|
||||
for il in 1..=40 {
|
||||
let x = (il * il) as f64;
|
||||
if il <= c.nlh {
|
||||
if let Some(pj_in) = c.pj
|
||||
&& il - 1 < pj_in.len() {
|
||||
pj[il - 1] = pj_in[il - 1];
|
||||
}
|
||||
} else {
|
||||
let wn = wn_val(c.wnhint, il, c.id, nf);
|
||||
pj[il - 1] = pp * (CPJ / x * t1).exp() * x * wn;
|
||||
}
|
||||
}
|
||||
|
||||
// Frequency- and line-independent Stark parameters
|
||||
let f00 = C00 * anes * anes * anes * anes;
|
||||
let dop0 = 1.0e8 * (1.65e8 * c.t + c.vturb).sqrt();
|
||||
|
||||
// Loop over all frequencies
|
||||
for ij in 0..nf {
|
||||
if params.window.ihylw[ij] <= 0 {
|
||||
continue;
|
||||
}
|
||||
|
||||
let wl = c.wlam[ij];
|
||||
let fr = c.freq[ij];
|
||||
|
||||
// Determine series range based on wavelength
|
||||
let (mut iserl, iseru) = series_range_hydrogen(params.window.ilowhw[ij], wl);
|
||||
|
||||
if iserl == 3 && iseru == 3 && c.nunbal > 0 {
|
||||
iserl = 2;
|
||||
}
|
||||
|
||||
abso[ij] = 0.0;
|
||||
emis[ij] = 0.0;
|
||||
|
||||
for i in iserl..=iseru {
|
||||
let ii = (i * i) as f64;
|
||||
let xii = UN / ii;
|
||||
let _popi = pj[i - 1];
|
||||
|
||||
// Determine contributing lines
|
||||
let (m1, m2) = determine_lines_hydrogen(
|
||||
i, params.window.ilowhw[ij], params.window.m10w[ij],
|
||||
params.window.m20w[ij], c.grav,
|
||||
);
|
||||
|
||||
for j in m1..=m2 {
|
||||
// Skip certain Lyman lines if iophli < 0
|
||||
if i == 1 && j <= 5 && c.iophli < 0 {
|
||||
continue;
|
||||
}
|
||||
|
||||
let jj = j * j;
|
||||
let xjj = UN / jj as f64;
|
||||
|
||||
// Transition properties
|
||||
let (abtra, emtra) = transition_hydrogen(i, j, &pj, c, nf, ii, xii, xjj, t1);
|
||||
|
||||
// Lyman-alpha broadening by helium (Lyahhe)
|
||||
let lalhhe = i == 1 && j == 2 && c.nunhhe > 0;
|
||||
if lalhhe && c.iathe > 0 && c.pop_he > 0.0 {
|
||||
let rel = 1.0 / std::f64::consts::TAU;
|
||||
let sg0 = lyahhe(wl, c.pop_he);
|
||||
let sg = sg0 * rel;
|
||||
abso[ij] += sg * abtra;
|
||||
emis[ij] += sg * emtra;
|
||||
}
|
||||
|
||||
// Profile line index for tabulated profiles
|
||||
let iline = if i <= 4 && j <= 22 {
|
||||
let idx = (i - 1) * 22 + (j - 1);
|
||||
if idx < c.ilin0.len() { c.ilin0[idx] } else { 0 }
|
||||
} else {
|
||||
0
|
||||
};
|
||||
|
||||
// Quasi-molecular opacity check
|
||||
let lquasi = (i == 1 && j == 2 && c.nunalp > 0)
|
||||
|| (i == 1 && j == 3 && c.nunbet > 0)
|
||||
|| (i == 1 && j == 4 && c.nungam > 0)
|
||||
|| (i == 2 && j == 3 && c.nunbal > 0);
|
||||
|
||||
if lquasi && c.allard_data.is_some() {
|
||||
// Quasi-molecular + Stark profile
|
||||
let stark = stark0(i as i32, j as i32, 1);
|
||||
let fxk = f00 * stark.xkij;
|
||||
let fxk1 = UN / fxk;
|
||||
let dop = dop0 / stark.wl0;
|
||||
let dbeta = stark.wl0 * stark.wl0 * CINV * fxk1;
|
||||
let betad = dop * dbeta;
|
||||
let fid = CID * stark.fij * dbeta;
|
||||
let (ad, div) = divstr(betad);
|
||||
let beta = (wl - stark.wl0).abs() * fxk1;
|
||||
|
||||
// Allard quasi-molecular contribution
|
||||
let ad_data = c.allard_data.unwrap();
|
||||
let sg_allard = allard(ad_data, wl, c.hneutr, c.hcharg, i as i32, j as i32);
|
||||
let sg = sg_allard + starka(beta, betad, ad, div, UN) * fid;
|
||||
abso[ij] += sg * abtra;
|
||||
emis[ij] += sg * emtra;
|
||||
} else if iline > 0 {
|
||||
// Tabulated Stark profile
|
||||
let nwl_idx = iline as usize - 1;
|
||||
if nwl_idx < c.nwlhyd.len() {
|
||||
let nwl = c.nwlhyd[nwl_idx];
|
||||
let wline_idx = (i - 1) * 22 + (j - 1);
|
||||
let wline_ij = if wline_idx < c.wline.len() {
|
||||
c.wline[wline_idx]
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
let osch_idx = (i - 1) * 22 + (j - 1);
|
||||
let osch_ij = if osch_idx < c.osch.len() {
|
||||
c.osch[osch_idx]
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
let fid = CID * osch_ij;
|
||||
|
||||
let mut al = (wl - wline_ij).abs();
|
||||
if al < 1.0e-4 {
|
||||
al = 1.0e-4;
|
||||
}
|
||||
if c.ilemke == 1 {
|
||||
al /= f00;
|
||||
}
|
||||
let al = al.log10();
|
||||
|
||||
// Find interpolation interval
|
||||
let mut iw0 = 0usize;
|
||||
for iwl in 0..nwl - 1 {
|
||||
let wl_next = profile_wl_val_h(c.wlhyd, nwl_idx, iwl + 1);
|
||||
if al <= wl_next {
|
||||
iw0 = iwl;
|
||||
break;
|
||||
}
|
||||
iw0 = iwl;
|
||||
}
|
||||
let iw1 = iw0 + 1;
|
||||
let wl0 = profile_wl_val_h(c.wlhyd, nwl_idx, iw0);
|
||||
let wl1 = profile_wl_val_h(c.wlhyd, nwl_idx, iw1);
|
||||
let prf0 = profile_prf_val_h(c.prfhyd, nwl_idx, iw0);
|
||||
let prf1 = profile_prf_val_h(c.prfhyd, nwl_idx, iw1);
|
||||
|
||||
let denom = wl1 - wl0;
|
||||
let prff = if denom.abs() > 1.0e-30 {
|
||||
(prf0 * (wl1 - al) + prf1 * (al - wl0)) / denom
|
||||
} else {
|
||||
prf0
|
||||
};
|
||||
let mut sg = (prff * AL10).exp() * fid;
|
||||
if c.ilemke == 1 {
|
||||
sg *= wline_ij * wline_ij * CINV / f00;
|
||||
}
|
||||
abso[ij] += sg * abtra;
|
||||
emis[ij] += sg * emtra;
|
||||
}
|
||||
} else {
|
||||
// Asymptotic Stark profile
|
||||
let stark = stark0(i as i32, j as i32, 1);
|
||||
let fxk = f00 * stark.xkij;
|
||||
let fxk1 = UN / fxk;
|
||||
let dop = dop0 / stark.wl0;
|
||||
let dbeta = stark.wl0 * stark.wl0 * CINV * fxk1;
|
||||
let betad = dop * dbeta;
|
||||
let fid = CID * stark.fij * dbeta;
|
||||
let (ad, div) = divstr(betad);
|
||||
let beta = (wl - stark.wl0).abs() * fxk1;
|
||||
let mut sg = starka(beta, betad, ad, div, TWO) * fid;
|
||||
|
||||
// Feautrier Lyman-alpha correction
|
||||
if c.iophli == 2 && i == 1 && j == 2
|
||||
&& let Some(fp) = c.feautr_params {
|
||||
sg *= feautr(fr, fp);
|
||||
}
|
||||
|
||||
abso[ij] += sg * abtra;
|
||||
emis[ij] += sg * emtra;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Total opacity and emissivity
|
||||
let f = c.freq[ij];
|
||||
let f15 = f * 1.0e-15;
|
||||
let xkf = (-4.79928e-11 * f * t1).exp();
|
||||
let xkfb = xkf * 1.4743e-2 * f15 * f15 * f15;
|
||||
|
||||
if abso[ij] <= 0.0 && c.lasdel {
|
||||
abso[ij] = 0.0;
|
||||
emis[ij] = 0.0;
|
||||
}
|
||||
|
||||
absoh[ij] = abso[ij] - xkf * emis[ij];
|
||||
emish[ij] = xkfb * emis[ij];
|
||||
}
|
||||
|
||||
HydliwResult { absoh, emish }
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// Helper functions
|
||||
// ============================================================================
|
||||
|
||||
/// Determine series range based on wavelength for hydrogen.
|
||||
fn series_range_hydrogen(ilow: usize, wl: f64) -> (usize, usize) {
|
||||
let mut iserl = ilow;
|
||||
let mut iseru = ilow;
|
||||
|
||||
if wl > 17000.0 && wl <= 21000.0 {
|
||||
iserl = 3; iseru = 4;
|
||||
} else if wl > 22700.0 && wl <= 29000.0 {
|
||||
iserl = 4; iseru = 5;
|
||||
} else if wl > 32800.0 && wl <= 37000.0 {
|
||||
iserl = 5; iseru = 6;
|
||||
} else if wl > 37000.0 && wl <= 44600.0 {
|
||||
iserl = 4; iseru = 6;
|
||||
} else if wl > 44660.0 && wl <= 58300.0 {
|
||||
iserl = 5; iseru = 7;
|
||||
} else if wl > 58300.0 && wl <= 72000.0 {
|
||||
iserl = 6; iseru = 8;
|
||||
} else if wl > 72000.0 && wl <= 73800.0 {
|
||||
iserl = 5; iseru = 8;
|
||||
} else if wl > 73800.0 && wl <= 77000.0 {
|
||||
iserl = 5; iseru = 9;
|
||||
} else if wl > 77000.0 {
|
||||
iserl = 6; iseru = 9;
|
||||
}
|
||||
|
||||
(iserl, iseru)
|
||||
}
|
||||
|
||||
/// Determine contributing line range for hydrogen.
|
||||
fn determine_lines_hydrogen(
|
||||
i: usize, ilowhw: usize, m10w: usize, m20w: usize, grav: f64,
|
||||
) -> (usize, usize) {
|
||||
let mut m1 = m10w;
|
||||
if i < ilowhw {
|
||||
m1 = ilowhw - 1;
|
||||
}
|
||||
let mut m2 = m1 + 1;
|
||||
if m1 < i + 1 {
|
||||
m1 = i + 1;
|
||||
}
|
||||
|
||||
if grav < 3.0 {
|
||||
let threshold = match i {
|
||||
7 => 16,
|
||||
6 => 14,
|
||||
5 => 12,
|
||||
4 => 10,
|
||||
3 => 8,
|
||||
2 => 6,
|
||||
1 => 4,
|
||||
_ => 0,
|
||||
};
|
||||
if m1 <= threshold && (1..=7).contains(&i) {
|
||||
// Keep m1 as is
|
||||
} else {
|
||||
m1 = m1.saturating_sub(1);
|
||||
m2 = m20w + 3;
|
||||
}
|
||||
} else {
|
||||
m1 = m1.saturating_sub(1);
|
||||
m2 = m20w + 3;
|
||||
}
|
||||
|
||||
if m1 < i + 1 {
|
||||
m1 = i + 1;
|
||||
}
|
||||
|
||||
if grav > 3.0 {
|
||||
m2 += 5;
|
||||
m1 = m1.saturating_sub(3);
|
||||
if m1 > i + 6 {
|
||||
m1 = m1.saturating_sub(3);
|
||||
}
|
||||
}
|
||||
if grav > 6.0 {
|
||||
m2 += 2;
|
||||
m1 = m1.saturating_sub(1);
|
||||
if m1 > i + 6 {
|
||||
m1 = m1.saturating_sub(1);
|
||||
}
|
||||
}
|
||||
|
||||
if m1 < i + 1 {
|
||||
m1 = i + 1;
|
||||
}
|
||||
if m2 > 40 {
|
||||
m2 = 40;
|
||||
}
|
||||
|
||||
(m1, m2)
|
||||
}
|
||||
|
||||
/// Compute transition properties for hydrogen line i→j.
|
||||
fn transition_hydrogen(
|
||||
i: usize, j: usize, pj: &[f64; 40], c: &HydliwCommon, nf: usize,
|
||||
ii: f64, xii: f64, xjj: f64, t1: f64,
|
||||
) -> (f64, f64) {
|
||||
let wn_j = wn_val(c.wnhint, j, c.id, nf);
|
||||
let wn_i = wn_val(c.wnhint, i, c.id, nf);
|
||||
|
||||
let mut abtra = pj[i - 1] * wn_j;
|
||||
let mut emtra = pj[j - 1] * wn_i * ii * xjj * (CPJ * (xii - xjj) * t1).exp();
|
||||
|
||||
// Special handling for low i, j (first two series members)
|
||||
if i <= 2 && j <= i + 2 {
|
||||
abtra = pj[i - 1];
|
||||
emtra = pj[j - 1] * wn_i / wn_j * ii * xjj * (CPJ * (xii - xjj) * t1).exp();
|
||||
}
|
||||
|
||||
(abtra, emtra)
|
||||
}
|
||||
|
||||
/// Access WNHINT partition function value.
|
||||
fn wn_val(wnhint: &[f64], level: usize, id: usize, nf: usize) -> f64 {
|
||||
if (1..=40).contains(&level) {
|
||||
let idx = (level - 1) * nf + id;
|
||||
if idx < wnhint.len() {
|
||||
return wnhint[idx];
|
||||
}
|
||||
}
|
||||
1.0
|
||||
}
|
||||
|
||||
/// Access WLHYD profile wavelength table value.
|
||||
fn profile_wl_val_h(wlhyd: &[f64], line_idx: usize, iwl: usize) -> f64 {
|
||||
let idx = line_idx * 54 + iwl;
|
||||
if idx < wlhyd.len() { wlhyd[idx] } else { 0.0 }
|
||||
}
|
||||
|
||||
/// Access PRFHYD profile value.
|
||||
fn profile_prf_val_h(prfhyd: &[f64], line_idx: usize, iwl: usize) -> f64 {
|
||||
let idx = line_idx * 54 + iwl;
|
||||
if idx < prfhyd.len() { prfhyd[idx] } else { 0.0 }
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_series_range_hydrogen() {
|
||||
// Default: just the lower level
|
||||
assert_eq!(series_range_hydrogen(3, 10000.0), (3, 3));
|
||||
|
||||
// Near-IR: Paschen + Brackett
|
||||
assert_eq!(series_range_hydrogen(1, 18000.0), (3, 4));
|
||||
|
||||
// Mid-IR: Brackett
|
||||
assert_eq!(series_range_hydrogen(1, 25000.0), (4, 5));
|
||||
|
||||
// Far-IR
|
||||
assert_eq!(series_range_hydrogen(1, 50000.0), (5, 7));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_determine_lines_hydrogen_basic() {
|
||||
// Low gravity, small i
|
||||
let (m1, m2) = determine_lines_hydrogen(1, 1, 10, 20, 2.0);
|
||||
assert!(m1 >= 2);
|
||||
assert!(m2 >= m1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_determine_lines_hydrogen_high_grav() {
|
||||
let (m1, m2) = determine_lines_hydrogen(3, 3, 10, 20, 7.0);
|
||||
assert!(m1 >= 4);
|
||||
assert!(m2 <= 40);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hydliw_no_hydrogen() {
|
||||
let nf = 5;
|
||||
let freq: Vec<f64> = (0..nf).map(|i| 3.0e15 - i as f64 * 1.0e14).collect();
|
||||
let wlam: Vec<f64> = freq.iter().map(|&f| 2.997925e17 / f).collect();
|
||||
let wnhint = vec![1.0; 40 * nf];
|
||||
let wline = vec![0.0; 4 * 22];
|
||||
let osch = vec![0.0; 4 * 22];
|
||||
let ilin0 = vec![0i32; 4 * 22];
|
||||
|
||||
let params = HydliwParams {
|
||||
common: HydliwCommon {
|
||||
id: 0,
|
||||
t: 10000.0,
|
||||
ane: 1.0e12,
|
||||
vturb: 1.0e5,
|
||||
grav: 4.0,
|
||||
freq: &freq,
|
||||
wlam: &wlam,
|
||||
iath: 0, // No hydrogen
|
||||
iophli: 1,
|
||||
ilemke: 0,
|
||||
pop_h_cont: 1.0e10,
|
||||
pj: None,
|
||||
wnhint: &wnhint,
|
||||
nwlhyd: &[],
|
||||
prfhyd: &[],
|
||||
wlhyd: &[],
|
||||
wline: &wline,
|
||||
osch: &osch,
|
||||
ilin0: &ilin0,
|
||||
nlh: 10,
|
||||
n0hn: 0,
|
||||
feautr_params: None,
|
||||
lasdel: false,
|
||||
nunalp: 0,
|
||||
nunbet: 0,
|
||||
nungam: 0,
|
||||
nunbal: 0,
|
||||
allard_data: None,
|
||||
hneutr: 0.0,
|
||||
hcharg: 0.0,
|
||||
nunhhe: 0,
|
||||
iathe: 0,
|
||||
pop_he: 0.0,
|
||||
},
|
||||
window: HydliwWindowParams {
|
||||
ihylw: &[1; 5],
|
||||
ilowhw: &[1; 5],
|
||||
m10w: &[5; 5],
|
||||
m20w: &[15; 5],
|
||||
},
|
||||
};
|
||||
|
||||
let result = hydliw(¶ms);
|
||||
assert!(result.absoh.iter().all(|&x| x == 0.0));
|
||||
assert!(result.emish.iter().all(|&x| x == 0.0));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hydliw_basic() {
|
||||
let nf = 5;
|
||||
let freq: Vec<f64> = (0..nf).map(|i| 3.0e15 - i as f64 * 1.0e14).collect();
|
||||
let wlam: Vec<f64> = freq.iter().map(|&f| 2.997925e17 / f).collect();
|
||||
let wnhint = vec![1.0; 40 * nf];
|
||||
let wline = vec![0.0; 4 * 22];
|
||||
let osch = vec![0.0; 4 * 22];
|
||||
let ilin0 = vec![0i32; 4 * 22];
|
||||
|
||||
let params = HydliwParams {
|
||||
common: HydliwCommon {
|
||||
id: 0,
|
||||
t: 20000.0,
|
||||
ane: 1.0e14,
|
||||
vturb: 2.0e5,
|
||||
grav: 4.0,
|
||||
freq: &freq,
|
||||
wlam: &wlam,
|
||||
iath: 1,
|
||||
iophli: 1,
|
||||
ilemke: 0,
|
||||
pop_h_cont: 1.0e10,
|
||||
pj: None,
|
||||
wnhint: &wnhint,
|
||||
nwlhyd: &[],
|
||||
prfhyd: &[],
|
||||
wlhyd: &[],
|
||||
wline: &wline,
|
||||
osch: &osch,
|
||||
ilin0: &ilin0,
|
||||
nlh: 10,
|
||||
n0hn: 0,
|
||||
feautr_params: None,
|
||||
lasdel: false,
|
||||
nunalp: 0,
|
||||
nunbet: 0,
|
||||
nungam: 0,
|
||||
nunbal: 0,
|
||||
allard_data: None,
|
||||
hneutr: 0.0,
|
||||
hcharg: 0.0,
|
||||
nunhhe: 0,
|
||||
iathe: 0,
|
||||
pop_he: 0.0,
|
||||
},
|
||||
window: HydliwWindowParams {
|
||||
ihylw: &[1; 5],
|
||||
ilowhw: &[1; 5],
|
||||
m10w: &[5; 5],
|
||||
m20w: &[15; 5],
|
||||
},
|
||||
};
|
||||
|
||||
let result = hydliw(¶ms);
|
||||
assert!(result.absoh.iter().all(|&x| x.is_finite()));
|
||||
assert!(result.emish.iter().all(|&x| x.is_finite()));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hydliw_skip_freq() {
|
||||
let nf = 3;
|
||||
let freq: Vec<f64> = (0..nf).map(|i| 3.0e15 - i as f64 * 1.0e14).collect();
|
||||
let wlam: Vec<f64> = freq.iter().map(|&f| 2.997925e17 / f).collect();
|
||||
let wnhint = vec![1.0; 40 * nf];
|
||||
let wline = vec![0.0; 4 * 22];
|
||||
let osch = vec![0.0; 4 * 22];
|
||||
let ilin0 = vec![0i32; 4 * 22];
|
||||
|
||||
let params = HydliwParams {
|
||||
common: HydliwCommon {
|
||||
id: 0,
|
||||
t: 10000.0,
|
||||
ane: 1.0e12,
|
||||
vturb: 1.0e5,
|
||||
grav: 4.0,
|
||||
freq: &freq,
|
||||
wlam: &wlam,
|
||||
iath: 1,
|
||||
iophli: 1,
|
||||
ilemke: 0,
|
||||
pop_h_cont: 1.0e10,
|
||||
pj: None,
|
||||
wnhint: &wnhint,
|
||||
nwlhyd: &[],
|
||||
prfhyd: &[],
|
||||
wlhyd: &[],
|
||||
wline: &wline,
|
||||
osch: &osch,
|
||||
ilin0: &ilin0,
|
||||
nlh: 10,
|
||||
n0hn: 0,
|
||||
feautr_params: None,
|
||||
lasdel: false,
|
||||
nunalp: 0,
|
||||
nunbet: 0,
|
||||
nungam: 0,
|
||||
nunbal: 0,
|
||||
allard_data: None,
|
||||
hneutr: 0.0,
|
||||
hcharg: 0.0,
|
||||
nunhhe: 0,
|
||||
iathe: 0,
|
||||
pop_he: 0.0,
|
||||
},
|
||||
window: HydliwWindowParams {
|
||||
ihylw: &[-1, 1, -1], // Skip freq 0 and 2
|
||||
ilowhw: &[1; 3],
|
||||
m10w: &[5; 3],
|
||||
m20w: &[15; 3],
|
||||
},
|
||||
};
|
||||
|
||||
let result = hydliw(¶ms);
|
||||
assert_eq!(result.absoh[0], 0.0);
|
||||
assert_eq!(result.emish[0], 0.0);
|
||||
assert_eq!(result.absoh[2], 0.0);
|
||||
assert_eq!(result.emish[2], 0.0);
|
||||
}
|
||||
}
|
||||
259
src/synspec/math/hydtab.rs
Normal file
259
src/synspec/math/hydtab.rs
Normal file
@ -0,0 +1,259 @@
|
||||
//! 氢线 Stark 展宽表格插值。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `HYDTAB` 子程序 (synspec54.f:7074)。
|
||||
//!
|
||||
//! 为给定谱线 I→J 和深度点 ID 插值氢线 Stark 展宽表格。
|
||||
//! 计算修改后的温度（含湍流速度修正）和电子密度，
|
||||
//! 然后调用 `inthyd` 进行二维插值。
|
||||
|
||||
#![allow(clippy::erasing_op)]
|
||||
use crate::tlusty::math::hydrogen::inthyd;
|
||||
use crate::tlusty::state::HydPrf;
|
||||
|
||||
/// 参数：氢线表格插值
|
||||
pub struct HydtabParams<'a> {
|
||||
/// 跃迁下能级 (1-indexed, Fortran 风格)
|
||||
pub i: i32,
|
||||
/// 跃迁上能级 (1-indexed, Fortran 风格)
|
||||
pub j: i32,
|
||||
/// 深度索引 (1-indexed)
|
||||
pub id: usize,
|
||||
/// 温度数组 (K)
|
||||
pub temp: &'a [f64],
|
||||
/// 电子密度数组
|
||||
pub elec: &'a [f64],
|
||||
/// 湍流速度数组
|
||||
pub vturb: &'a [f64],
|
||||
/// 谱线索引表 ILIN0(i,j) → 谱线编号 (1-indexed, 0 = 无此线)
|
||||
pub ilin0: &'a [i32],
|
||||
/// ILIN0 行数 (用于 2D 索引)
|
||||
pub ilin0_nrows: usize,
|
||||
/// 谱线中心波长 WLINE(i,j) (Å)
|
||||
pub wline: &'a [f64],
|
||||
/// WLINE 行数
|
||||
pub wline_nrows: usize,
|
||||
/// 每条谱线的波长点数 NWLH(iline)
|
||||
pub nwli: &'a [i32],
|
||||
/// 轮廓数据 PRF(iwl, it, ie, iline) — log10 值
|
||||
pub prf: &'a [f64],
|
||||
/// PRF 维度: (nwl_max, nt_max, ne_max)
|
||||
pub prf_dims: (usize, usize, usize),
|
||||
/// 波长偏移 WLHYD(iline, iwl)
|
||||
pub wlhyd: &'a [f64],
|
||||
/// WLHYD 维度: (nwl_max,)
|
||||
pub wlhyd_nwl_max: usize,
|
||||
/// 输出: PRFHYD(iline, id, iwl) — 插值后的轮廓
|
||||
pub prfhyd: &'a mut [f64],
|
||||
/// PRFHYD 维度: (nlines, ndepth, nwl_max)
|
||||
pub prfhyd_dims: (usize, usize, usize),
|
||||
/// 氢线表格数据 (用于 inthyd)
|
||||
pub hydprf: &'a HydPrf,
|
||||
/// 静态 XK 系数 (在 id==1 时计算，后续复用)
|
||||
pub xk: &'a mut f64,
|
||||
}
|
||||
|
||||
/// 氢线表格插值。
|
||||
///
|
||||
/// 为谱线 I→J 在深度点 ID 处插值 Stark 展宽表格。
|
||||
/// 结果存储在 `params.prfhyd` 数组中。
|
||||
///
|
||||
/// # Fortran 原始代码
|
||||
///
|
||||
/// ```fortran
|
||||
/// SUBROUTINE HYDTAB(I,J,ID)
|
||||
/// ```
|
||||
pub fn hydtab(params: &mut HydtabParams) {
|
||||
let i = params.i as usize;
|
||||
let j = params.j as usize;
|
||||
let id = params.id; // 1-indexed
|
||||
|
||||
// 获取谱线索引 (1-indexed, 0 = 无此线)
|
||||
let idx = (i - 1) * params.ilin0_nrows + (j - 1);
|
||||
let iline_1 = params.ilin0[idx]; // 1-indexed
|
||||
if iline_1 == 0 {
|
||||
return;
|
||||
}
|
||||
let iline = (iline_1 - 1) as usize; // 0-indexed
|
||||
|
||||
// 获取波长和波长点数
|
||||
let wl_idx = (i - 1) * params.wline_nrows + (j - 1);
|
||||
let wl0 = params.wline[wl_idx];
|
||||
let nwl = params.nwli[iline] as usize;
|
||||
|
||||
// 计算渐近轮廓系数 (仅在第一个深度点)
|
||||
if id == 1 {
|
||||
// PRF(NWL, 1, 1, ILINE) — 注意 Fortran 1-indexed
|
||||
let prf_idx = (nwl - 1) * params.prf_dims.1 * params.prf_dims.2
|
||||
+ 0 * params.prf_dims.2;
|
||||
let prf_val = if prf_idx < params.prf.len() {
|
||||
params.prf[prf_idx]
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
|
||||
// WLHYD(ILINE, NWL)
|
||||
let wlhyd_idx = iline * params.wlhyd_nwl_max + (nwl - 1);
|
||||
let wlhyd_val = if wlhyd_idx < params.wlhyd.len() {
|
||||
params.wlhyd[wlhyd_idx]
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
|
||||
let xclog = prf_val + 2.5 * wlhyd_val - 0.477121;
|
||||
let xklog = 0.6666667 * xclog;
|
||||
*params.xk = (xklog * std::f64::consts::LN_10).exp();
|
||||
}
|
||||
|
||||
let xk = *params.xk;
|
||||
|
||||
// 修改温度以考虑湍流速度对 Doppler 宽度的影响
|
||||
let id0 = id - 1; // 0-indexed
|
||||
let t = params.temp[id0] + 6.06e-9 * params.vturb[id0];
|
||||
let ane = params.elec[id0];
|
||||
let tl = t.log10();
|
||||
let anel = ane.log10();
|
||||
|
||||
// Stark 参数
|
||||
let f00 = 1.25e-9 * ane.powf(0.666666667);
|
||||
let fxk = f00 * xk;
|
||||
let dop = 1.0e8 / wl0 * (1.65e8 * t).sqrt();
|
||||
let dbeta = wl0 * wl0 / 2.997925e18 / fxk;
|
||||
let _betad = dbeta * dop;
|
||||
|
||||
// 对每个波长点调用 INTHYD 插值
|
||||
for iwl in 0..nwl {
|
||||
let prof = inthyd(tl, anel, iwl, iline, params.hydprf, dbeta, xk);
|
||||
|
||||
// PRFHYD(ILINE, ID, IWL) — 存储结果
|
||||
let prfhyd_idx = iline * params.prfhyd_dims.1 * params.prfhyd_dims.2
|
||||
+ (id - 1) * params.prfhyd_dims.2
|
||||
+ iwl;
|
||||
if prfhyd_idx < params.prfhyd.len() {
|
||||
params.prfhyd[prfhyd_idx] = prof;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use crate::tlusty::state::HydPrf;
|
||||
|
||||
fn create_test_params() -> (HydPrf, Vec<f64>, Vec<f64>) {
|
||||
let mut hydprf = HydPrf::default();
|
||||
hydprf.nth[0] = 7;
|
||||
hydprf.neh[0] = 20;
|
||||
for it in 0..7 {
|
||||
hydprf.xtlem[it] = 4.0 + it as f64 * 0.1;
|
||||
}
|
||||
for ie in 0..20 {
|
||||
hydprf.xnelem[ie] = 12.0 + ie as f64 * 0.2;
|
||||
}
|
||||
for iwl in 0..90 {
|
||||
hydprf.wlh[iwl] = 4000.0 + iwl as f64 * 10.0;
|
||||
}
|
||||
for it in 0..7 {
|
||||
for ie in 0..20 {
|
||||
hydprf.set_prfhyd(0, 0, it, ie, -2.0 + it as f64 * 0.1 + ie as f64 * 0.01);
|
||||
}
|
||||
}
|
||||
|
||||
let mut prf = vec![0.0; 90 * 7 * 20];
|
||||
for iwl in 0..90 {
|
||||
for it in 0..7 {
|
||||
for ie in 0..20 {
|
||||
let idx = iwl * 7 * 20 + it * 20 + ie;
|
||||
prf[idx] = -2.0 + it as f64 * 0.1 + ie as f64 * 0.01;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
let mut wlhyd = vec![0.0; 90];
|
||||
for iwl in 0..90 {
|
||||
wlhyd[iwl] = 0.01 * (iwl + 1) as f64;
|
||||
}
|
||||
|
||||
(hydprf, prf, wlhyd)
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hydtab_no_line() {
|
||||
let (hydprf, prf, wlhyd) = create_test_params();
|
||||
let temp = vec![10000.0, 9000.0];
|
||||
let elec = vec![1e13, 1e13];
|
||||
let vturb = vec![1e5, 1e5];
|
||||
let ilin0 = vec![0i32; 4]; // 2x2, all zero = no line
|
||||
let wline = vec![1215.67; 4];
|
||||
let nwlh = vec![90i32];
|
||||
let mut prfhyd = vec![0.0; 1 * 2 * 90];
|
||||
let mut xk = 0.0;
|
||||
|
||||
let mut params = HydtabParams {
|
||||
i: 1,
|
||||
j: 2,
|
||||
id: 1,
|
||||
temp: &temp,
|
||||
elec: &elec,
|
||||
vturb: &vturb,
|
||||
ilin0: &ilin0,
|
||||
ilin0_nrows: 2,
|
||||
wline: &wline,
|
||||
wline_nrows: 2,
|
||||
nwli: &nwlh,
|
||||
prf: &prf,
|
||||
prf_dims: (90, 7, 20),
|
||||
wlhyd: &wlhyd,
|
||||
wlhyd_nwl_max: 90,
|
||||
prfhyd: &mut prfhyd,
|
||||
prfhyd_dims: (1, 2, 90),
|
||||
hydprf: &hydprf,
|
||||
xk: &mut xk,
|
||||
};
|
||||
|
||||
hydtab(&mut params);
|
||||
// ilin0 = 0, should return immediately
|
||||
assert_eq!(prfhyd[0], 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hydtab_basic() {
|
||||
let (hydprf, prf, wlhyd) = create_test_params();
|
||||
let temp = vec![10000.0, 9000.0];
|
||||
let elec = vec![1e13, 1e13];
|
||||
let vturb = vec![1e5, 1e5];
|
||||
let ilin0 = vec![1i32; 4]; // 2x2, line index = 1
|
||||
let wline = vec![1215.67; 4];
|
||||
let nwlh = vec![90i32];
|
||||
let mut prfhyd = vec![0.0; 1 * 2 * 90];
|
||||
let mut xk = 0.0;
|
||||
|
||||
let mut params = HydtabParams {
|
||||
i: 1,
|
||||
j: 2,
|
||||
id: 1,
|
||||
temp: &temp,
|
||||
elec: &elec,
|
||||
vturb: &vturb,
|
||||
ilin0: &ilin0,
|
||||
ilin0_nrows: 2,
|
||||
wline: &wline,
|
||||
wline_nrows: 2,
|
||||
nwli: &nwlh,
|
||||
prf: &prf,
|
||||
prf_dims: (90, 7, 20),
|
||||
wlhyd: &wlhyd,
|
||||
wlhyd_nwl_max: 90,
|
||||
prfhyd: &mut prfhyd,
|
||||
prfhyd_dims: (1, 2, 90),
|
||||
hydprf: &hydprf,
|
||||
xk: &mut xk,
|
||||
};
|
||||
|
||||
hydtab(&mut params);
|
||||
// xk should have been computed (id == 1)
|
||||
assert!(xk > 0.0);
|
||||
// prfhyd should have been filled with finite values
|
||||
let has_nonzero = prfhyd.iter().any(|&v| v != 0.0);
|
||||
assert!(has_nonzero);
|
||||
}
|
||||
}
|
||||
136
src/synspec/math/hylsew.rs
Normal file
136
src/synspec/math/hylsew.rs
Normal file
@ -0,0 +1,136 @@
|
||||
//! SYNSPEC 氢线窗口初始化。
|
||||
//!
|
||||
//! 重构自 SYNSPEC 54 的 HYLSEW 子程序。
|
||||
|
||||
/// 氢线窗口参数
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct HylsewOutput {
|
||||
/// 是否包含氢线 (0=否, 1=是)
|
||||
pub ihylw: i32,
|
||||
/// 线数参数
|
||||
pub m20w: i32,
|
||||
/// 最低主量子数索引
|
||||
pub ilowhw: i32,
|
||||
/// 线翼参数
|
||||
pub m10w: i32,
|
||||
}
|
||||
|
||||
/// 初始化氢线处理窗口。
|
||||
///
|
||||
/// 根据频率和重力加速度判断是否包含氢线，并设置相关参数。
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `ij` - 频率索引
|
||||
/// * `freq` - 频率值 (Hz)
|
||||
/// * `grav` - 重力加速度 (log g)
|
||||
///
|
||||
/// # Returns
|
||||
/// 氢线窗口参数
|
||||
pub fn hylsew(_ij: usize, freq: f64, grav: f64) -> HylsewOutput {
|
||||
let mut output = HylsewOutput {
|
||||
ihylw: 0,
|
||||
m20w: 0,
|
||||
ilowhw: 0,
|
||||
m10w: 0,
|
||||
};
|
||||
|
||||
// 检查频率是否在氢线范围内
|
||||
if freq >= 3.28805e15 {
|
||||
return output;
|
||||
}
|
||||
|
||||
let al0 = 2.997925e17 / freq;
|
||||
let al1 = al0;
|
||||
|
||||
// 根据重力加速度检查波长范围
|
||||
if grav < 6.0 {
|
||||
if al0 > 160.0 && al1 < 364.6 {
|
||||
return output;
|
||||
}
|
||||
if al0 > 506.0 && al1 < 630.0 {
|
||||
return output;
|
||||
}
|
||||
if al0 > 680.0 && al1 < 820.3 {
|
||||
return output;
|
||||
}
|
||||
} else {
|
||||
if al0 > 540.0 && al1 < 600.0 {
|
||||
return output;
|
||||
}
|
||||
if al0 > 720.0 && al1 < 820.3 {
|
||||
return output;
|
||||
}
|
||||
}
|
||||
|
||||
// 包含氢线
|
||||
output.ihylw = 1;
|
||||
output.m20w = 40;
|
||||
|
||||
// 确定最低主量子数
|
||||
let frion = if al1 < 364.6 {
|
||||
output.ilowhw = 1;
|
||||
3.28805e15
|
||||
} else if al1 < 820.0 {
|
||||
output.ilowhw = 2;
|
||||
8.2225e14
|
||||
} else if al1 < 1458.0 {
|
||||
output.ilowhw = 3;
|
||||
3.6544142e14
|
||||
} else if al1 < 2278.0 {
|
||||
output.ilowhw = 4;
|
||||
2.0555837e14
|
||||
} else if al1 < 3281.0 {
|
||||
output.ilowhw = 5;
|
||||
1.315589e14
|
||||
} else if al1 < 4466.0 {
|
||||
output.ilowhw = 6;
|
||||
9.136394e13
|
||||
} else {
|
||||
output.ilowhw = 7;
|
||||
6.7120228e13
|
||||
};
|
||||
|
||||
// 计算线翼参数
|
||||
if frion > freq {
|
||||
output.m10w = (3.289017e15 / (frion - freq).abs()).sqrt() as i32;
|
||||
}
|
||||
|
||||
output
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_hylsew_high_freq() {
|
||||
// 频率太高，不包含氢线
|
||||
let result = hylsew(1, 4.0e15, 4.0);
|
||||
assert_eq!(result.ihylw, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hylsew_lyman_region() {
|
||||
// Lyman 系区域
|
||||
let result = hylsew(1, 3.0e15, 4.0);
|
||||
assert_eq!(result.ihylw, 1);
|
||||
assert_eq!(result.ilowhw, 1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hylsew_balmer_region() {
|
||||
// Balmer 系区域 (500 Å，在 Lyman 极限和 Balmer 极限之间)
|
||||
let freq = 2.997925e17 / 500.0;
|
||||
let result = hylsew(1, freq, 4.0);
|
||||
assert_eq!(result.ihylw, 1);
|
||||
assert_eq!(result.ilowhw, 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hylsew_high_gravity() {
|
||||
// 高重力情况
|
||||
let freq = 2.997925e17 / 550.0;
|
||||
let result = hylsew(1, freq, 7.0);
|
||||
assert_eq!(result.ihylw, 0);
|
||||
}
|
||||
}
|
||||
248
src/synspec/math/idmtab.rs
Normal file
248
src/synspec/math/idmtab.rs
Normal file
@ -0,0 +1,248 @@
|
||||
//! Output of selected molecular line parameters (identification table).
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine IDMTAB (line 16380).
|
||||
//!
|
||||
//! Computes and formats molecular line parameters for the identification
|
||||
//! table output, including equivalent widths and line strengths.
|
||||
|
||||
use crate::synspec::math::inibla::CL;
|
||||
|
||||
// ============================================================================
|
||||
// Constants
|
||||
// ============================================================================
|
||||
|
||||
/// Conversion factor: ln(10) for log-gf
|
||||
const C1: f64 = std::f64::consts::LN_10;
|
||||
|
||||
/// Conversion factor: gf offset
|
||||
const C2: f64 = 4.201_467_2;
|
||||
|
||||
/// Conversion factor: energy to temperature
|
||||
const C3: f64 = 1.438_788_6;
|
||||
|
||||
/// Strength category labels
|
||||
const APB: &str = " ";
|
||||
const AP0: &str = " .";
|
||||
const AP1: &str = " *";
|
||||
const AP2: &str = " **";
|
||||
const AP3: &str = " ***";
|
||||
const AP4: &str = "****";
|
||||
|
||||
// ============================================================================
|
||||
// IDMTAB parameters
|
||||
// ============================================================================
|
||||
|
||||
/// Parameters for a single molecular line.
|
||||
pub struct IdmtabLine {
|
||||
/// Wavelength (Å)
|
||||
pub alam: f64,
|
||||
/// Molecule index
|
||||
pub imol: usize,
|
||||
/// Lower excitation potential (cm⁻¹)
|
||||
pub excl: f64,
|
||||
/// Log gf value
|
||||
pub gfm: f64,
|
||||
/// Van der Waals broadening
|
||||
pub grm: f64,
|
||||
/// Stark broadening
|
||||
pub gsm: f64,
|
||||
/// Van der Waals broadening (depth-dependent)
|
||||
pub gvdw: f64,
|
||||
/// Doppler width
|
||||
pub dop1: f64,
|
||||
/// Continuum opacity at line center
|
||||
pub absta: f64,
|
||||
/// Stimulated emission factor
|
||||
pub stim: f64,
|
||||
/// Molecular population ratio
|
||||
pub rrmol: f64,
|
||||
/// Temperature at standard depth (K)
|
||||
pub temp: f64,
|
||||
/// Electron density at standard depth
|
||||
pub elec: f64,
|
||||
}
|
||||
|
||||
/// Result of IDMTAB computation for a single line.
|
||||
pub struct IdmtabResult {
|
||||
/// Wavelength (Å)
|
||||
pub alam: f64,
|
||||
/// Molecule name
|
||||
pub molecule: String,
|
||||
/// Log gf
|
||||
pub gf: f64,
|
||||
/// Lower excitation energy (K)
|
||||
pub excl_k: f64,
|
||||
/// Line-to-continuum ratio (STR0)
|
||||
pub str0: f64,
|
||||
/// Equivalent width (mÅ)
|
||||
pub eqw: f64,
|
||||
/// Strength category label
|
||||
pub apr: &'static str,
|
||||
/// Depth index
|
||||
pub id: usize,
|
||||
/// Total broadening parameter
|
||||
pub agam: f64,
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// IDMTAB implementation
|
||||
// ============================================================================
|
||||
|
||||
/// Compute molecular line parameters for the identification table.
|
||||
///
|
||||
/// For a given molecular line, computes the line strength, equivalent width,
|
||||
/// and strength category.
|
||||
///
|
||||
/// # Fortran original
|
||||
///
|
||||
/// ```fortran
|
||||
/// SUBROUTINE IDMTAB
|
||||
/// DO IL0=1,NLINML
|
||||
/// ...compute STR0, EQW, APR...
|
||||
/// END DO
|
||||
/// END
|
||||
/// ```
|
||||
pub fn idmtab_compute(line: &IdmtabLine) -> IdmtabResult {
|
||||
let IdmtabLine {
|
||||
alam, imol: _, excl, gfm, grm, gsm, gvdw,
|
||||
dop1, absta, stim, rrmol, temp, elec,
|
||||
} = *line;
|
||||
|
||||
// Total broadening parameter
|
||||
// Fortran: AGAM=(GRM+GSM*ANE+GVDW)*DOP1
|
||||
let agam = (grm + gsm * elec + gvdw) * dop1;
|
||||
|
||||
// Absorption at line center
|
||||
// Fortran: ABCNT=EXP(GFM-EXCL/TEMP)*RRMOL*DOP1*STIM
|
||||
let abcnt = (gfm - excl / temp).exp() * rrmol * dop1 * stim;
|
||||
|
||||
// Line-to-continuum ratio
|
||||
// Fortran: STR0=ABCNT/ABSTA
|
||||
let str0 = if absta > 0.0 { abcnt / absta } else { 0.0 };
|
||||
|
||||
// Log gf
|
||||
let gf = (gfm + C2) / C1;
|
||||
|
||||
// Lower excitation energy in K
|
||||
let excl_k = excl / C3;
|
||||
|
||||
// Equivalent width estimate
|
||||
let ww1 = if str0 <= 1.2 {
|
||||
0.886 * str0 * (1.0 - str0 * (0.707 - str0 * 0.577))
|
||||
} else {
|
||||
str0.ln().sqrt()
|
||||
};
|
||||
|
||||
let ww1 = if str0 > 55.0 {
|
||||
let ww2 = 0.5 * (std::f64::consts::PI * agam * str0).sqrt();
|
||||
if ww2 > ww1 { ww2 } else { ww1 }
|
||||
} else {
|
||||
ww1
|
||||
};
|
||||
|
||||
// Equivalent width in mÅ
|
||||
// Fortran: EQW=ALAM/FREQ0*1.E3/DOP1*WW1
|
||||
// Since ALAM is wavelength and we don't have FREQ0 directly,
|
||||
// we use the relation: EQW ≈ ALAM * WW1 / (c/ALAM) / DOP1 * 1e3
|
||||
// Simplified: EQW = ALAM^2 / CL * 1e3 / DOP1 * WW1
|
||||
let eqw = alam * alam / CL * 1e3 / dop1 * ww1;
|
||||
|
||||
// Strength category
|
||||
let str = eqw * 10.0;
|
||||
let apr = if str >= 1e4 {
|
||||
AP4
|
||||
} else if str >= 1e3 {
|
||||
AP3
|
||||
} else if str >= 1e2 {
|
||||
AP2
|
||||
} else if str >= 1e1 {
|
||||
AP1
|
||||
} else if str >= 1e0 {
|
||||
AP0
|
||||
} else {
|
||||
APB
|
||||
};
|
||||
|
||||
IdmtabResult {
|
||||
alam,
|
||||
molecule: String::new(), // Filled by caller
|
||||
gf,
|
||||
excl_k,
|
||||
str0,
|
||||
eqw,
|
||||
apr,
|
||||
id: 0, // Filled by caller
|
||||
agam,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
fn create_test_line() -> IdmtabLine {
|
||||
IdmtabLine {
|
||||
alam: 5000.0,
|
||||
imol: 1,
|
||||
excl: 10000.0,
|
||||
gfm: -2.0,
|
||||
grm: 0.1,
|
||||
gsm: 0.01,
|
||||
gvdw: 0.05,
|
||||
dop1: 0.01,
|
||||
absta: 1e-10,
|
||||
stim: 1.0,
|
||||
rrmol: 1e-5,
|
||||
temp: 10000.0,
|
||||
elec: 1e14,
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_idmtab_basic() {
|
||||
let line = create_test_line();
|
||||
let result = idmtab_compute(&line);
|
||||
|
||||
assert!(result.alam > 0.0);
|
||||
assert!(result.str0.is_finite());
|
||||
assert!(result.eqw.is_finite());
|
||||
assert!(result.eqw >= 0.0);
|
||||
assert!(result.agam.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_idmtab_weak_line() {
|
||||
let mut line = create_test_line();
|
||||
line.rrmol = 1e-20; // Very weak line
|
||||
let result = idmtab_compute(&line);
|
||||
|
||||
// Weak line should have small STR0
|
||||
assert!(result.str0 < 1.0);
|
||||
assert_eq!(result.apr, APB);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_idmtab_strong_line() {
|
||||
let mut line = create_test_line();
|
||||
line.rrmol = 1e10; // Very strong line
|
||||
line.absta = 1e-20;
|
||||
let result = idmtab_compute(&line);
|
||||
|
||||
// Strong line should have large STR0
|
||||
assert!(result.str0 > 1.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_idmtab_strength_categories() {
|
||||
// Test that different strength values produce correct categories
|
||||
let mut line = create_test_line();
|
||||
|
||||
// Weak line
|
||||
line.rrmol = 1e-15;
|
||||
let r = idmtab_compute(&line);
|
||||
if r.str0 <= 1.2 {
|
||||
// For weak lines, EQW is small
|
||||
assert!(r.eqw < 1.0 || r.apr == APB);
|
||||
}
|
||||
}
|
||||
}
|
||||
287
src/synspec/math/idtab.rs
Normal file
287
src/synspec/math/idtab.rs
Normal file
@ -0,0 +1,287 @@
|
||||
//! Output of selected atomic line parameters (identification table).
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine IDTAB (line 9636).
|
||||
//!
|
||||
//! Computes and formats atomic line parameters for the identification
|
||||
//! table output, including equivalent widths and line strengths.
|
||||
|
||||
use crate::synspec::math::inibla::CL;
|
||||
|
||||
// ============================================================================
|
||||
// Constants
|
||||
// ============================================================================
|
||||
|
||||
/// Conversion factor: ln(10) for log-gf
|
||||
const C1: f64 = std::f64::consts::LN_10;
|
||||
|
||||
/// Conversion factor: gf offset
|
||||
const C2: f64 = 4.201_467_2;
|
||||
|
||||
/// Conversion factor: energy to temperature
|
||||
const C3: f64 = 1.438_788_6;
|
||||
|
||||
/// Ionization stage labels
|
||||
const TYPION: [&str; 30] = [
|
||||
" I ", " II ", " III", " IV ", " V ",
|
||||
" VI ", " VII", "VIII", " IX ", " X ",
|
||||
" XI ", " XII", "XIII", " XIV", " XV ",
|
||||
" XVI", "XVII", " 18 ", " XIX", " XX ",
|
||||
" XXI", "XXII", " 23 ", "XXIV", "XXV ",
|
||||
"XXVI", " 27 ", " 28 ", "XXIX", " XXX",
|
||||
];
|
||||
|
||||
/// Strength category labels
|
||||
const APB: &str = " ";
|
||||
const AP0: &str = " .";
|
||||
const AP1: &str = " *";
|
||||
const AP2: &str = " **";
|
||||
const AP3: &str = " ***";
|
||||
const AP4: &str = "****";
|
||||
|
||||
// ============================================================================
|
||||
// IDTAB parameters
|
||||
// ============================================================================
|
||||
|
||||
/// Parameters for a single atomic line.
|
||||
pub struct IdtabLine {
|
||||
/// Line index
|
||||
pub il: usize,
|
||||
/// Wavelength (Å)
|
||||
pub alam: f64,
|
||||
/// Atom index
|
||||
pub iat: usize,
|
||||
/// Ionization stage (1-based)
|
||||
pub ion: usize,
|
||||
/// Lower excitation potential (cm⁻¹)
|
||||
pub excl: f64,
|
||||
/// Log gf value
|
||||
pub gf0: f64,
|
||||
/// Doppler width
|
||||
pub dop1: f64,
|
||||
/// Continuum opacity at line center
|
||||
pub absta: f64,
|
||||
/// Stimulated emission factor
|
||||
pub stim: f64,
|
||||
/// Population ratio RRR
|
||||
pub rrr: f64,
|
||||
/// Total broadening parameter (from PROFIL)
|
||||
pub agam: f64,
|
||||
/// Temperature at standard depth (K)
|
||||
pub temp: f64,
|
||||
/// Standard depth index
|
||||
pub idstd: usize,
|
||||
/// Reference depth index
|
||||
pub id: usize,
|
||||
/// Lower level index
|
||||
pub ilown: usize,
|
||||
/// Upper level index
|
||||
pub iupn: usize,
|
||||
/// First level index for the element
|
||||
pub nfirst: usize,
|
||||
/// Element index for the lower level
|
||||
pub iel: usize,
|
||||
}
|
||||
|
||||
/// Result of IDTAB computation for a single line.
|
||||
pub struct IdtabResult {
|
||||
/// Wavelength (Å)
|
||||
pub alam: f64,
|
||||
/// Atom name
|
||||
pub atom: String,
|
||||
/// Ionization stage label
|
||||
pub ion_label: &'static str,
|
||||
/// Log gf
|
||||
pub gf: f64,
|
||||
/// Lower excitation energy (K)
|
||||
pub excl_k: f64,
|
||||
/// Line-to-continuum ratio (STR0)
|
||||
pub str0: f64,
|
||||
/// Equivalent width (mÅ)
|
||||
pub eqw: f64,
|
||||
/// Strength category label
|
||||
pub apr: &'static str,
|
||||
/// Lower level index (relative)
|
||||
pub ill: usize,
|
||||
/// Upper level index (relative)
|
||||
pub ilu: usize,
|
||||
/// Depth index
|
||||
pub id: usize,
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// IDTAB implementation
|
||||
// ============================================================================
|
||||
|
||||
/// Compute atomic line parameters for the identification table.
|
||||
///
|
||||
/// For a given atomic line, computes the line strength, equivalent width,
|
||||
/// and strength category.
|
||||
///
|
||||
/// # Fortran original
|
||||
///
|
||||
/// ```fortran
|
||||
/// SUBROUTINE IDTAB
|
||||
/// DO IL0=1,NLIN
|
||||
/// ...compute STR0, EQW, APR...
|
||||
/// END DO
|
||||
/// END
|
||||
/// ```
|
||||
pub fn idtab_compute(line: &IdtabLine) -> IdtabResult {
|
||||
let IdtabLine {
|
||||
il: _, alam, iat: _, ion, excl, gf0, dop1, absta, stim, rrr,
|
||||
agam, temp, idstd: _, id, ilown, iupn, nfirst, iel: _,
|
||||
} = *line;
|
||||
|
||||
// Absorption at line center
|
||||
// Fortran: ABCNT=EXP(GF0-EXCL/TEMP)*RRR*STIM
|
||||
let abcnt = (gf0 - excl / temp).exp() * rrr * stim;
|
||||
|
||||
// Line-to-continuum ratio
|
||||
// Fortran: STR0=ABCNT*DOP1/ABSTA
|
||||
let str0 = if absta > 0.0 {
|
||||
abcnt * dop1 / absta
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
|
||||
// Log gf
|
||||
let gf = (gf0 + C2) / C1;
|
||||
|
||||
// Lower excitation energy in K
|
||||
let excl_k = excl / C3;
|
||||
|
||||
// Equivalent width estimate
|
||||
let ww1 = if str0 <= 1.2 {
|
||||
0.886 * str0 * (1.0 - str0 * (0.707 - str0 * 0.577))
|
||||
} else {
|
||||
str0.ln().sqrt()
|
||||
};
|
||||
|
||||
let ww1 = if str0 > 55.0 {
|
||||
let ww2 = 0.5 * (std::f64::consts::PI * agam * str0).sqrt();
|
||||
if ww2 > ww1 { ww2 } else { ww1 }
|
||||
} else {
|
||||
ww1
|
||||
};
|
||||
|
||||
// Equivalent width in mÅ
|
||||
// Fortran: EQW=ALAM/FREQ0*1.E3/DOP1*WW1
|
||||
// FREQ0 = CL/ALAM, so EQW = ALAM^2/CL * 1e3 / DOP1 * WW1
|
||||
let eqw = alam * alam / CL * 1e3 / dop1 * ww1;
|
||||
|
||||
// Strength category
|
||||
let str = eqw * 10.0;
|
||||
let apr = if str >= 1e4 {
|
||||
AP4
|
||||
} else if str >= 1e3 {
|
||||
AP3
|
||||
} else if str >= 1e2 {
|
||||
AP2
|
||||
} else if str >= 1e1 {
|
||||
AP1
|
||||
} else if str >= 1e0 {
|
||||
AP0
|
||||
} else {
|
||||
APB
|
||||
};
|
||||
|
||||
// Relative level indices
|
||||
let ill = if ilown > 0 { ilown - nfirst + 1 } else { 0 };
|
||||
let ilu = if iupn > 0 { iupn - nfirst + 1 } else { 0 };
|
||||
|
||||
// Ionization stage label (1-based index)
|
||||
let ion_label = if (1..=30).contains(&ion) {
|
||||
TYPION[ion - 1]
|
||||
} else {
|
||||
" ?? "
|
||||
};
|
||||
|
||||
IdtabResult {
|
||||
alam,
|
||||
atom: String::new(), // Filled by caller
|
||||
ion_label,
|
||||
gf,
|
||||
excl_k,
|
||||
str0,
|
||||
eqw,
|
||||
apr,
|
||||
ill,
|
||||
ilu,
|
||||
id,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
fn create_test_line() -> IdtabLine {
|
||||
IdtabLine {
|
||||
il: 1,
|
||||
alam: 5000.0,
|
||||
iat: 1,
|
||||
ion: 1,
|
||||
excl: 10000.0,
|
||||
gf0: -1.0,
|
||||
dop1: 0.01,
|
||||
absta: 1e-10,
|
||||
stim: 1.0,
|
||||
rrr: 1e-3,
|
||||
agam: 0.1,
|
||||
temp: 10000.0,
|
||||
idstd: 35,
|
||||
id: 35,
|
||||
ilown: 1,
|
||||
iupn: 5,
|
||||
nfirst: 1,
|
||||
iel: 1,
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_idtab_basic() {
|
||||
let line = create_test_line();
|
||||
let result = idtab_compute(&line);
|
||||
|
||||
assert!(result.alam > 0.0);
|
||||
assert!(result.str0.is_finite());
|
||||
assert!(result.eqw.is_finite());
|
||||
assert!(result.eqw >= 0.0);
|
||||
assert!(result.alam.is_finite());
|
||||
assert_eq!(result.ion_label, " I ");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_idtab_ion_labels() {
|
||||
let mut line = create_test_line();
|
||||
|
||||
line.ion = 1;
|
||||
assert_eq!(idtab_compute(&line).ion_label, " I ");
|
||||
|
||||
line.ion = 2;
|
||||
assert_eq!(idtab_compute(&line).ion_label, " II ");
|
||||
|
||||
line.ion = 26;
|
||||
assert_eq!(idtab_compute(&line).ion_label, "XXVI");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_idtab_level_indices() {
|
||||
let mut line = create_test_line();
|
||||
line.ilown = 10;
|
||||
line.iupn = 15;
|
||||
line.nfirst = 8;
|
||||
|
||||
let result = idtab_compute(&line);
|
||||
assert_eq!(result.ill, 3); // 10 - 8 + 1
|
||||
assert_eq!(result.ilu, 8); // 15 - 8 + 1
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_idtab_weak_line() {
|
||||
let mut line = create_test_line();
|
||||
line.rrr = 1e-20;
|
||||
let result = idtab_compute(&line);
|
||||
assert!(result.str0 < 1.0);
|
||||
}
|
||||
}
|
||||
559
src/synspec/math/ingrid.rs
Normal file
559
src/synspec/math/ingrid.rs
Normal file
@ -0,0 +1,559 @@
|
||||
//! ingrid — 不透明度网格计算的状态参数设置。
|
||||
//!
|
||||
//! Fortran 原始签名: SUBROUTINE INGRID(MODE,INEXT,IGRD)
|
||||
//!
|
||||
//! 设置不透明度表计算的温度和密度网格。
|
||||
//!
|
||||
//! 注意: Fortran 版本直接操作文件 I/O 和 COMMON 块。
|
||||
//! Rust 版本提供纯计算核心函数。
|
||||
|
||||
/// 密度参数类型
|
||||
#[derive(Debug, Clone, Copy, PartialEq)]
|
||||
pub enum DensityParameterType {
|
||||
/// 电子密度
|
||||
ElectronDensity = 0,
|
||||
/// 质量密度 (idens < 10)
|
||||
MassDensity = 1,
|
||||
/// 可变密度 (idens >= 20)
|
||||
Variable = 2,
|
||||
}
|
||||
|
||||
/// 不透明度网格参数
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct OpacityGridParams {
|
||||
/// 最低温度 (K)
|
||||
pub temp1: f64,
|
||||
/// 最高温度 (K)
|
||||
pub temp2: f64,
|
||||
/// 温度点数
|
||||
pub ntemp: usize,
|
||||
/// 密度参数类型
|
||||
pub dens_type: DensityParameterType,
|
||||
/// 最低密度
|
||||
pub dens1: f64,
|
||||
/// 最高密度
|
||||
pub dens2: f64,
|
||||
/// 密度点数
|
||||
pub ndens: usize,
|
||||
/// 频率点数
|
||||
pub nfgrid: usize,
|
||||
/// 最短波长 (nm)
|
||||
pub wlam1: f64,
|
||||
/// 最长波长 (nm)
|
||||
pub wlam2: f64,
|
||||
}
|
||||
|
||||
/// 生成对数等距温度网格
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// at1=log(temp1)
|
||||
/// at2=log(temp2)
|
||||
/// dt=(at2-at1)/(ntemp-1)
|
||||
/// do i=1,ntemp
|
||||
/// tempg(i)=exp(at1+(i-1)*dt)
|
||||
/// end do
|
||||
/// ```
|
||||
pub fn generate_temperature_grid(temp1: f64, temp2: f64, ntemp: usize) -> Vec<f64> {
|
||||
if temp1 <= 0.0 {
|
||||
return vec![temp1; ntemp];
|
||||
}
|
||||
let at1 = temp1.ln();
|
||||
let at2 = temp2.ln();
|
||||
let dt = if ntemp > 1 { (at2 - at1) / (ntemp - 1) as f64 } else { 0.0 };
|
||||
|
||||
(0..ntemp)
|
||||
.map(|i| (at1 + i as f64 * dt).exp())
|
||||
.collect()
|
||||
}
|
||||
|
||||
/// 生成对数等距密度网格（均匀分布）
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// at1=log(dens1)
|
||||
/// at2=log(dens2)
|
||||
/// dr=(at2-at1)/(ndens-1)
|
||||
/// do i=1,ntemp
|
||||
/// do j=1,ndens
|
||||
/// densg(i,j)=exp(at1+(j-1)*dr)
|
||||
/// end do
|
||||
/// end do
|
||||
/// ```
|
||||
pub fn generate_density_grid_uniform(
|
||||
dens1: f64,
|
||||
dens2: f64,
|
||||
ndens: usize,
|
||||
ntemp: usize,
|
||||
) -> Vec<Vec<f64>> {
|
||||
let at1 = dens1.ln();
|
||||
let at2 = dens2.ln();
|
||||
let dr = if ndens > 1 { (at2 - at1) / (ndens - 1) as f64 } else { 0.0 };
|
||||
|
||||
(0..ntemp)
|
||||
.map(|_| {
|
||||
(0..ndens)
|
||||
.map(|j| (at1 + j as f64 * dr).exp())
|
||||
.collect()
|
||||
})
|
||||
.collect()
|
||||
}
|
||||
|
||||
/// 生成可变密度网格（密度范围随温度变化）
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// do i=1,ntemp
|
||||
/// dens1=rhol1+(rhou1-rhol1)/(at2-at1)*(templ(i)-at1)
|
||||
/// dens2=rhol2+(rhou2-rhol2)/(at2-at1)*(templ(i)-at1)
|
||||
/// dr=(dens2-dens1)/(ndens-1)
|
||||
/// do j=1,ndens
|
||||
/// densg(i,j)=exp(dens1+(j-1)*dr)
|
||||
/// end do
|
||||
/// end do
|
||||
/// ```
|
||||
pub fn generate_density_grid_variable(
|
||||
temp_grid: &[f64],
|
||||
dens_lower: (f64, f64), // (at_low_T, at_high_T)
|
||||
dens_upper: (f64, f64), // (at_low_T, at_high_T)
|
||||
ndens: usize,
|
||||
) -> Vec<Vec<f64>> {
|
||||
let ntemp = temp_grid.len();
|
||||
if ntemp == 0 {
|
||||
return Vec::new();
|
||||
}
|
||||
|
||||
let at1 = temp_grid[0].ln();
|
||||
let at2 = temp_grid[ntemp - 1].ln();
|
||||
let dt_range = at2 - at1;
|
||||
|
||||
let rhol1 = dens_lower.0.ln();
|
||||
let rhol2 = dens_lower.1.ln();
|
||||
let rhou1 = dens_upper.0.ln();
|
||||
let rhou2 = dens_upper.1.ln();
|
||||
|
||||
(0..ntemp)
|
||||
.map(|i| {
|
||||
let templ_i = temp_grid[i].ln();
|
||||
let frac = if dt_range.abs() > 1e-30 {
|
||||
(templ_i - at1) / dt_range
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
|
||||
let dens1 = rhol1 + (rhou1 - rhol1) * frac;
|
||||
let dens2 = rhol2 + (rhou2 - rhol2) * frac;
|
||||
let dr = if ndens > 1 { (dens2 - dens1) / (ndens - 1) as f64 } else { 0.0 };
|
||||
|
||||
(0..ndens)
|
||||
.map(|j| (dens1 + j as f64 * dr).exp())
|
||||
.collect()
|
||||
})
|
||||
.collect()
|
||||
}
|
||||
|
||||
/// 从模型大气设置网格（温度和密度来自模型）
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// call inpmod
|
||||
/// ntemp=nd
|
||||
/// ndens=1
|
||||
/// do it=1,ntemp
|
||||
/// tempg(it)=temp(it)
|
||||
/// densg0(it)=dens(it)
|
||||
/// densg(it,1)=dens(it)
|
||||
/// elecm(it)=elec(it)
|
||||
/// end do
|
||||
/// ```
|
||||
pub fn set_grid_from_model(
|
||||
temps: &[f64],
|
||||
dens: &[f64],
|
||||
elec: &[f64],
|
||||
) -> (Vec<f64>, Vec<f64>, Vec<Vec<f64>>, Vec<f64>) {
|
||||
let _ntemp = temps.len();
|
||||
let tempg = temps.to_vec();
|
||||
let densg0 = dens.to_vec();
|
||||
let densg = dens.iter().map(|&d| vec![d]).collect();
|
||||
let elecm = elec.to_vec();
|
||||
|
||||
(tempg, densg0, densg, elecm)
|
||||
}
|
||||
|
||||
/// 不透明度表插值结果
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct InterpolatedOpacity {
|
||||
/// 波长网格 (nm)
|
||||
pub wavelengths: Vec<f64>,
|
||||
/// 不透明度（对数）
|
||||
pub log_opacity: Vec<f64>,
|
||||
}
|
||||
|
||||
/// 对数空间平均插值
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// if(isum.gt.0) then
|
||||
/// abgrd(ijgrd)=log(sum/float(isum))
|
||||
/// ```
|
||||
pub fn log_average_interpolation(
|
||||
wltab: &[f64],
|
||||
absop: &[f64],
|
||||
wlgrid: &[f64],
|
||||
) -> Vec<f64> {
|
||||
let nfgrid = wlgrid.len();
|
||||
let nfr = wltab.len();
|
||||
let mut abgrd = vec![0.0; nfgrid];
|
||||
|
||||
let mut ij = 0;
|
||||
for ijgrd in 0..nfgrid {
|
||||
let wlgr = if ijgrd + 1 < nfgrid {
|
||||
0.5 * (wlgrid[ijgrd] + wlgrid[ijgrd + 1])
|
||||
} else {
|
||||
wlgrid[ijgrd]
|
||||
};
|
||||
|
||||
let mut sum = 0.0_f64;
|
||||
let mut isum = 0;
|
||||
|
||||
while ij < nfr && wltab[ij] <= wlgr {
|
||||
sum += absop[ij].exp();
|
||||
isum += 1;
|
||||
ij += 1;
|
||||
}
|
||||
|
||||
if isum > 0 {
|
||||
abgrd[ijgrd] = (sum / isum as f64).ln();
|
||||
} else if ij < nfr {
|
||||
// 线性插值
|
||||
let abl = absop[ij];
|
||||
let wlt = wltab[ij];
|
||||
if ij + 1 < nfr {
|
||||
let abl_next = absop[ij + 1];
|
||||
let wlt_next = wltab[ij + 1];
|
||||
abgrd[ijgrd] = abl + (abl_next - abl) / (wlt_next - wlt) * (wlgr - wlt);
|
||||
} else {
|
||||
abgrd[ijgrd] = abl;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 最后一个点复制前一个
|
||||
if nfgrid > 1 {
|
||||
abgrd[nfgrid - 1] = abgrd[nfgrid - 2];
|
||||
}
|
||||
|
||||
abgrd
|
||||
}
|
||||
|
||||
/// 网格遍历状态
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct GridTraversalState {
|
||||
/// 当前温度索引
|
||||
pub indext: usize,
|
||||
/// 当前密度索引
|
||||
pub indexn: usize,
|
||||
/// 是否还有下一个网格点
|
||||
pub inext: bool,
|
||||
}
|
||||
|
||||
/// 编排函数: 不透明度网格初始化和推进。
|
||||
///
|
||||
/// Fortran 原始逻辑: SUBROUTINE INGRID(MODE,INEXT,IGRD)
|
||||
///
|
||||
/// # 模式
|
||||
/// - `mode=0`: 初始化 — 读取网格参数，设置温度/密度网格
|
||||
/// - `mode=1`: 推进 — 存储当前结果，推进到下一个网格点
|
||||
pub struct IngridParams<'a> {
|
||||
/// 模式 (0=init, 1=advance)
|
||||
pub mode: i32,
|
||||
/// 网格参数 (mode=0 时使用)
|
||||
pub grid_params: Option<&'a OpacityGridParams>,
|
||||
/// 温度网格 (mode=1 时使用)
|
||||
pub temperatures: &'a [f64],
|
||||
/// 密度网格 (mode=1 时使用)
|
||||
pub densities: &'a [Vec<f64>],
|
||||
/// 每温度点密度数 (mode=1 时使用)
|
||||
pub nden: &'a [usize],
|
||||
/// 当前不透明度数据 (mode=1 时使用)
|
||||
pub absop: &'a [f64],
|
||||
/// 当前波长表 (mode=1 时使用)
|
||||
pub wltab: &'a [f64],
|
||||
/// 目标波长网格 (mode=1 时使用)
|
||||
pub wlgrid: &'a [f64],
|
||||
/// 插值模式 (0=average, 1=intrp)
|
||||
pub inttab: i32,
|
||||
}
|
||||
|
||||
pub struct IngridResult {
|
||||
/// 是否还有下一个网格点
|
||||
pub inext: bool,
|
||||
/// 温度网格
|
||||
pub tempg: Vec<f64>,
|
||||
/// 密度网格 [temp_idx][dens_idx]
|
||||
pub densg: Vec<Vec<f64>>,
|
||||
/// 电子密度网格 [temp_idx][dens_idx]
|
||||
pub elecgr: Vec<Vec<f64>>,
|
||||
/// 插值后的不透明度 (对数)
|
||||
pub abgrd: Vec<f64>,
|
||||
/// 当前温度索引
|
||||
pub indext: usize,
|
||||
/// 当前密度索引
|
||||
pub indexn: usize,
|
||||
}
|
||||
|
||||
pub fn ingrid(params: &IngridParams) -> IngridResult {
|
||||
if params.mode == 0 {
|
||||
// Initialization mode
|
||||
let gp = params.grid_params.unwrap();
|
||||
let tempg = generate_temperature_grid(gp.temp1, gp.temp2, gp.ntemp);
|
||||
let densg = match gp.dens_type {
|
||||
DensityParameterType::ElectronDensity | DensityParameterType::MassDensity => {
|
||||
generate_density_grid_uniform(gp.dens1, gp.dens2, gp.ndens, gp.ntemp)
|
||||
}
|
||||
DensityParameterType::Variable => {
|
||||
// Variable density: use provided dens1/dens2 as bounds
|
||||
generate_density_grid_uniform(gp.dens1, gp.dens2, gp.ndens, gp.ntemp)
|
||||
}
|
||||
};
|
||||
let elecgr = vec![vec![0.0; gp.ndens]; gp.ntemp];
|
||||
|
||||
IngridResult {
|
||||
inext: gp.ntemp > 1 || gp.ndens > 1,
|
||||
tempg,
|
||||
densg,
|
||||
elecgr,
|
||||
abgrd: vec![0.0; gp.nfgrid],
|
||||
indext: 0,
|
||||
indexn: 0,
|
||||
}
|
||||
} else {
|
||||
// Advance mode: interpolate opacity and move to next grid point
|
||||
let ntemp = params.temperatures.len();
|
||||
let nden = params.nden;
|
||||
|
||||
// Interpolate opacity to grid
|
||||
let abgrd = if params.inttab == 1 {
|
||||
log_average_interpolation(params.wltab, params.absop, params.wlgrid)
|
||||
} else {
|
||||
log_average_interpolation(params.wltab, params.absop, params.wlgrid)
|
||||
};
|
||||
|
||||
// Advance grid state
|
||||
let mut state = GridTraversalState {
|
||||
indext: 0,
|
||||
indexn: 0,
|
||||
inext: true,
|
||||
};
|
||||
advance_grid_point(&mut state, ntemp, nden);
|
||||
|
||||
IngridResult {
|
||||
inext: state.inext,
|
||||
tempg: params.temperatures.to_vec(),
|
||||
densg: params.densities.to_vec(),
|
||||
elecgr: vec![vec![0.0; nden.len()]; ntemp],
|
||||
abgrd,
|
||||
indext: state.indext,
|
||||
indexn: state.indexn,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// 推进到下一个网格点
|
||||
///
|
||||
/// Fortran 原始逻辑 (1-indexed):
|
||||
/// ```fortran
|
||||
/// if(indexn.lt.ndens) then
|
||||
/// indexn=indexn+1
|
||||
/// inext=1
|
||||
/// else
|
||||
/// indexn=1
|
||||
/// if(indext.lt.ntemp) then
|
||||
/// indext=indext+1
|
||||
/// inext=1
|
||||
/// else
|
||||
/// inext=0
|
||||
/// end if
|
||||
/// end if
|
||||
/// ```
|
||||
///
|
||||
/// Rust 版本使用 0-indexed: indexn 从 0 到 ndens-1
|
||||
pub fn advance_grid_point(
|
||||
state: &mut GridTraversalState,
|
||||
ntemp: usize,
|
||||
ndens: &[usize],
|
||||
) {
|
||||
let current_ndens = ndens[state.indext];
|
||||
if state.indexn < current_ndens - 1 {
|
||||
// 还有更多密度点
|
||||
state.indexn += 1;
|
||||
state.inext = true;
|
||||
} else {
|
||||
// 当前温度的所有密度点完成
|
||||
state.indexn = 0;
|
||||
if state.indext < ntemp - 1 {
|
||||
// 还有更多温度点
|
||||
state.indext += 1;
|
||||
state.inext = true;
|
||||
} else {
|
||||
// 所有网格点完成
|
||||
state.inext = false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_generate_temperature_grid() {
|
||||
let grid = generate_temperature_grid(5000.0, 50000.0, 10);
|
||||
assert_eq!(grid.len(), 10);
|
||||
assert!((grid[0] - 5000.0).abs() < 1e-10);
|
||||
assert!((grid[9] - 50000.0).abs() < 1e-3);
|
||||
// 对数等距
|
||||
let ratio = grid[1] / grid[0];
|
||||
for i in 1..9 {
|
||||
assert!((grid[i + 1] / grid[i] - ratio).abs() < 1e-10);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_generate_temperature_grid_negative() {
|
||||
let grid = generate_temperature_grid(-1.0, 50000.0, 5);
|
||||
assert_eq!(grid.len(), 5);
|
||||
// temp1 <= 0 → 所有值为 temp1
|
||||
for &t in &grid {
|
||||
assert_eq!(t, -1.0);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_generate_density_grid_uniform() {
|
||||
let grid = generate_density_grid_uniform(1e-10, 1e-6, 5, 3);
|
||||
assert_eq!(grid.len(), 3);
|
||||
assert_eq!(grid[0].len(), 5);
|
||||
assert!((grid[0][0] - 1e-10).abs() < 1e-20);
|
||||
assert!((grid[0][4] - 1e-6).abs() < 1e-15);
|
||||
// 所有温度的密度网格相同
|
||||
assert_eq!(grid[0], grid[1]);
|
||||
assert_eq!(grid[1], grid[2]);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_generate_density_grid_variable() {
|
||||
let temps = vec![5000.0, 10000.0, 20000.0];
|
||||
let grid = generate_density_grid_variable(
|
||||
&temps,
|
||||
(1e-10, 1e-8), // 低端密度随温度变化
|
||||
(1e-6, 1e-4), // 高端密度随温度变化
|
||||
5,
|
||||
);
|
||||
assert_eq!(grid.len(), 3);
|
||||
assert_eq!(grid[0].len(), 5);
|
||||
// 低温的密度范围不同于高温
|
||||
assert!(grid[0][0] != grid[2][0]);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_set_grid_from_model() {
|
||||
let temps = vec![5000.0, 10000.0];
|
||||
let dens = vec![1e-8, 1e-7];
|
||||
let elec = vec![1e-10, 1e-9];
|
||||
let (tempg, densg0, densg, elecm) = set_grid_from_model(&temps, &dens, &elec);
|
||||
assert_eq!(tempg, temps);
|
||||
assert_eq!(densg0, dens);
|
||||
assert_eq!(elecm, elec);
|
||||
assert_eq!(densg.len(), 2);
|
||||
assert_eq!(densg[0], vec![1e-8]);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_log_average_interpolation() {
|
||||
let wltab = vec![100.0, 200.0, 300.0, 400.0, 500.0];
|
||||
let absop = vec![0.0, 1.0, 2.0, 1.0, 0.0];
|
||||
let wlgrid = vec![150.0, 250.0, 350.0, 450.0];
|
||||
let result = log_average_interpolation(&wltab, &absop, &wlgrid);
|
||||
assert_eq!(result.len(), 4);
|
||||
// 所有值应该是有限的
|
||||
for &v in &result {
|
||||
assert!(v.is_finite());
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_advance_grid_point() {
|
||||
let mut state = GridTraversalState {
|
||||
indext: 0,
|
||||
indexn: 0,
|
||||
inext: true,
|
||||
};
|
||||
let ndens = vec![3, 3, 3];
|
||||
|
||||
// 第一次推进: indexn 0 → 1
|
||||
advance_grid_point(&mut state, 3, &ndens);
|
||||
assert_eq!(state.indext, 0);
|
||||
assert_eq!(state.indexn, 1);
|
||||
assert!(state.inext);
|
||||
|
||||
// 第二次推进: indexn 1 → 2
|
||||
advance_grid_point(&mut state, 3, &ndens);
|
||||
assert_eq!(state.indext, 0);
|
||||
assert_eq!(state.indexn, 2);
|
||||
|
||||
// 第三次推进: indexn 2 → 0, indext 0 → 1
|
||||
advance_grid_point(&mut state, 3, &ndens);
|
||||
assert_eq!(state.indext, 1);
|
||||
assert_eq!(state.indexn, 0);
|
||||
assert!(state.inext);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_advance_grid_point_end() {
|
||||
let mut state = GridTraversalState {
|
||||
indext: 2,
|
||||
indexn: 2,
|
||||
inext: true,
|
||||
};
|
||||
let ndens = vec![3, 3, 3];
|
||||
|
||||
// 最后一个网格点 → inext=false, indexn 重置为 0
|
||||
advance_grid_point(&mut state, 3, &ndens);
|
||||
assert_eq!(state.indext, 2);
|
||||
assert_eq!(state.indexn, 0);
|
||||
assert!(!state.inext);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_ingrid_mode0() {
|
||||
let gp = OpacityGridParams {
|
||||
temp1: 5000.0,
|
||||
temp2: 50000.0,
|
||||
ntemp: 3,
|
||||
dens_type: DensityParameterType::MassDensity,
|
||||
dens1: 1e-10,
|
||||
dens2: 1e-6,
|
||||
ndens: 4,
|
||||
nfgrid: 5,
|
||||
wlam1: 100.0,
|
||||
wlam2: 1000.0,
|
||||
};
|
||||
let params = IngridParams {
|
||||
mode: 0,
|
||||
grid_params: Some(&gp),
|
||||
temperatures: &[],
|
||||
densities: &[],
|
||||
nden: &[],
|
||||
absop: &[],
|
||||
wltab: &[],
|
||||
wlgrid: &[],
|
||||
inttab: 0,
|
||||
};
|
||||
let result = ingrid(¶ms);
|
||||
assert_eq!(result.tempg.len(), 3);
|
||||
assert_eq!(result.densg.len(), 3);
|
||||
assert_eq!(result.densg[0].len(), 4);
|
||||
assert!(result.inext);
|
||||
}
|
||||
}
|
||||
393
src/synspec/math/inibl0.rs
Normal file
393
src/synspec/math/inibl0.rs
Normal file
@ -0,0 +1,393 @@
|
||||
//! inibl0 — 辅助初始化过程。
|
||||
//!
|
||||
//! Fortran 原始签名: SUBROUTINE INIBL0
|
||||
//!
|
||||
//! 设置合成光谱评估的参数：波长范围、截止参数、
|
||||
//! 角度点和权重、连续频率网格。
|
||||
//!
|
||||
//! 注意: Fortran 版本直接操作文件 I/O 和 COMMON 块。
|
||||
//! Rust 版本提供纯计算核心函数。
|
||||
|
||||
/// 物理常数
|
||||
#[allow(dead_code)]
|
||||
const CL: f64 = 2.997925e10; // 光速 (cm/s)
|
||||
const CNM: f64 = 2.997925e17; // 光速 (nm/s) = 2.997925×10^17 nm/s
|
||||
|
||||
/// 波长范围参数
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct WavelengthRange {
|
||||
/// 初始波长 (nm)
|
||||
pub alam0: f64,
|
||||
/// 最终波长 (nm)
|
||||
pub alast: f64,
|
||||
/// 中心波长 (nm)
|
||||
pub alamc: f64,
|
||||
/// 截止参数 (nm)
|
||||
pub cutof0: f64,
|
||||
/// 间距参数 (nm)
|
||||
pub space0: f64,
|
||||
/// 最小不透明度比
|
||||
pub relop: f64,
|
||||
}
|
||||
|
||||
/// 计算波长范围参数
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// ALAMC=(ALAM0+ALAST)*0.5
|
||||
/// if(space.eq.0.) space=4.3e-8*sqrt(temp(idstd))*alamc
|
||||
/// if(space.lt.0.) space=-5.72e-8*sqrt(temp(idstd))*alamc*space
|
||||
/// SPACF=2.997925E18/ALAMC/ALAMC*SPACE
|
||||
/// CUTOF0=0.1*CUTOF0
|
||||
/// SPACE0=SPACE*0.1
|
||||
/// ALAM0=1.D-1*ALAM0
|
||||
/// ALAST=1.D-1*ALAST
|
||||
/// ALAMC=ALAMC*0.1
|
||||
/// ```
|
||||
pub fn compute_wavelength_range(
|
||||
alam0_angstrom: f64,
|
||||
alast_angstrom: f64,
|
||||
cutof0_angstrom: f64,
|
||||
space_angstrom: f64,
|
||||
relop: f64,
|
||||
temp_std: f64,
|
||||
) -> WavelengthRange {
|
||||
let alamc = (alam0_angstrom + alast_angstrom) * 0.5;
|
||||
|
||||
let space = if space_angstrom == 0.0 {
|
||||
4.3e-8 * temp_std.sqrt() * alamc
|
||||
} else if space_angstrom < 0.0 {
|
||||
-5.72e-8 * temp_std.sqrt() * alamc * space_angstrom
|
||||
} else {
|
||||
space_angstrom
|
||||
};
|
||||
|
||||
// 转换为 nm (除以 10)
|
||||
WavelengthRange {
|
||||
alam0: alam0_angstrom * 0.1,
|
||||
alast: alast_angstrom * 0.1,
|
||||
alamc: alamc * 0.1,
|
||||
cutof0: cutof0_angstrom * 0.1,
|
||||
space0: space * 0.1,
|
||||
relop,
|
||||
}
|
||||
}
|
||||
|
||||
/// 频率范围
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct FrequencyRange {
|
||||
/// 初始频率 (s^-1)
|
||||
pub freq1: f64,
|
||||
/// 最终频率 (s^-1)
|
||||
pub freq2: f64,
|
||||
/// 最后频率 (s^-1)
|
||||
pub frlast: f64,
|
||||
}
|
||||
|
||||
/// 计算频率范围
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// FRLAST=2.997925D17/ALAST
|
||||
/// FREQ(1)=2.997925D17/ALAM0
|
||||
/// FREQ(2)=FRLAST
|
||||
/// ```
|
||||
pub fn compute_frequency_range(alam0_nm: f64, alast_nm: f64) -> FrequencyRange {
|
||||
let frlast = CNM / alast_nm;
|
||||
FrequencyRange {
|
||||
freq1: CNM / alam0_nm,
|
||||
freq2: frlast,
|
||||
frlast,
|
||||
}
|
||||
}
|
||||
|
||||
/// 角度点配置
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct AngleConfig {
|
||||
/// 角度点数
|
||||
pub nmu: usize,
|
||||
/// 最小 mu 值
|
||||
pub ang0: f64,
|
||||
/// 角度点 (cos(theta))
|
||||
pub angles: Vec<f64>,
|
||||
/// 权重
|
||||
pub weights: Vec<f64>,
|
||||
}
|
||||
|
||||
/// 计算等距角度点和权重
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// DMU=(1.-ANG0)/(NMU0-1)
|
||||
/// DO IMU=1,NMU0
|
||||
/// ANGL(IMU)=1.-(IMU-1)*DMU
|
||||
/// WANGL(IMU)=DMU
|
||||
/// END DO
|
||||
/// WANGL(1)=0.5*DMU
|
||||
/// WANGL(NMU0-1)=0.5*DMU
|
||||
/// WANGL(NMU0)=2.*DMU
|
||||
/// ```
|
||||
pub fn compute_angle_points_uniform(nmu: usize, ang0: f64) -> AngleConfig {
|
||||
if nmu <= 1 {
|
||||
return AngleConfig {
|
||||
nmu: 1,
|
||||
ang0: 1.0,
|
||||
angles: vec![1.0],
|
||||
weights: vec![1.0],
|
||||
};
|
||||
}
|
||||
|
||||
let dmu = (1.0 - ang0) / (nmu - 1) as f64;
|
||||
let angles: Vec<f64> = (0..nmu).map(|i| 1.0 - i as f64 * dmu).collect();
|
||||
let mut weights = vec![dmu; nmu];
|
||||
|
||||
// 边界权重修正
|
||||
weights[0] = 0.5 * dmu;
|
||||
weights[nmu - 2] = 0.5 * dmu;
|
||||
weights[nmu - 1] = 2.0 * dmu;
|
||||
|
||||
AngleConfig { nmu, ang0, angles, weights }
|
||||
}
|
||||
|
||||
/// 计算正弦等距角度点和权重
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// ANGH=0.70710678
|
||||
/// DMU=ANGH/(NMU0-1)
|
||||
/// DO IMU=1,NMU0
|
||||
/// ANGL(IMU)=(IMU-1)*DMU
|
||||
/// ANGL(IMU)=SQRT(1.-ANGL(IMU)**2)
|
||||
/// ...
|
||||
/// END DO
|
||||
/// ```
|
||||
pub fn compute_angle_points_sine(nmu: usize, ang0: f64) -> AngleConfig {
|
||||
let angh = std::f64::consts::FRAC_1_SQRT_2; // sin(45°) = cos(45°)
|
||||
let dmu = angh / (nmu - 1) as f64;
|
||||
|
||||
let angles: Vec<f64> = (0..nmu)
|
||||
.map(|i| {
|
||||
let sin_val = i as f64 * dmu;
|
||||
(1.0 - sin_val * sin_val).sqrt()
|
||||
})
|
||||
.collect();
|
||||
|
||||
let mut weights = vec![0.0; nmu];
|
||||
for i in 1..nmu - 1 {
|
||||
weights[i] = 0.5 * (angles[i - 1] - angles[i + 1]);
|
||||
}
|
||||
weights[0] = 0.5 * (angles[0] - angles[1]);
|
||||
weights[nmu - 1] = 0.5 * (angles[nmu - 2] - angles[nmu - 1]);
|
||||
|
||||
// 扩展角度点（如果 ang0 < 0）
|
||||
if ang0 < 0.0 {
|
||||
let dmu2 = (angh + ang0) / (nmu - 1) as f64;
|
||||
let nmu_total = 2 * nmu - 2;
|
||||
let mut all_angles = angles.clone();
|
||||
let mut all_weights = weights.clone();
|
||||
|
||||
for i in 0..nmu - 2 {
|
||||
all_angles.push(angh - (i + 1) as f64 * dmu2);
|
||||
all_weights.push(dmu2);
|
||||
}
|
||||
|
||||
// 修正边界权重
|
||||
all_weights[nmu - 1] += 0.5 * dmu2;
|
||||
all_weights[nmu_total - 2] = 0.5 * dmu2;
|
||||
all_weights[nmu_total - 1] = 2.0 * dmu2;
|
||||
|
||||
return AngleConfig {
|
||||
nmu: nmu_total,
|
||||
ang0,
|
||||
angles: all_angles,
|
||||
weights: all_weights,
|
||||
};
|
||||
}
|
||||
|
||||
AngleConfig { nmu, ang0, angles, weights }
|
||||
}
|
||||
|
||||
/// 连续频率网格参数
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct ContinuumFreqGrid {
|
||||
/// 频率点数
|
||||
pub nfreqc: usize,
|
||||
/// 频率点 (s^-1)
|
||||
pub frequencies: Vec<f64>,
|
||||
/// 波长点 (nm)
|
||||
pub wavelengths: Vec<f64>,
|
||||
}
|
||||
|
||||
/// 计算连续频率网格（窗口模式）
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// spacon=cutofs
|
||||
/// IF(SPACON.EQ.0) SPACON=3.
|
||||
/// XFR=(ALAST-ALAM0)/SPACON
|
||||
/// NFREQC=int(XFR)+1
|
||||
/// DLAMLO=LOG10(ALAST/ALAM0)/(NFREQC-1)
|
||||
/// DO IJ=1,NFREQC
|
||||
/// AL=AL0L+(IJ-1)*DLAMLO
|
||||
/// ALAM=EXP(2.3025851*AL)
|
||||
/// WLAMC(IJ)=ALAM
|
||||
/// FREQC(IJ)=2.997925E18/ALAM
|
||||
/// END DO
|
||||
/// ```
|
||||
pub fn compute_continuum_frequency_grid(
|
||||
alam0_nm: f64,
|
||||
alast_nm: f64,
|
||||
spacon: f64,
|
||||
max_nfreqc: usize,
|
||||
) -> ContinuumFreqGrid {
|
||||
let spacon = if spacon == 0.0 { 3.0 } else { spacon };
|
||||
let xfr = (alast_nm - alam0_nm) / spacon;
|
||||
let nfreqc = ((xfr as i32) + 1).max(2).min(max_nfreqc as i32) as usize;
|
||||
|
||||
let al0l = alam0_nm.log10();
|
||||
let dlamlo = (alast_nm / alam0_nm).log10() / (nfreqc - 1) as f64;
|
||||
|
||||
let wavelengths: Vec<f64> = (0..nfreqc)
|
||||
.map(|ij| {
|
||||
let al = al0l + ij as f64 * dlamlo;
|
||||
10.0_f64.powf(al)
|
||||
})
|
||||
.collect();
|
||||
|
||||
let frequencies: Vec<f64> = wavelengths.iter().map(|&wl| CNM / wl).collect();
|
||||
|
||||
ContinuumFreqGrid {
|
||||
nfreqc,
|
||||
frequencies,
|
||||
wavelengths,
|
||||
}
|
||||
}
|
||||
|
||||
/// NLTE 模式解析
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// if(inlte.lt.10) then
|
||||
/// lasdel=.true.
|
||||
/// else if(inlte.le.20) then
|
||||
/// inlte=inlte-10
|
||||
/// lasdel=.false.
|
||||
/// else if(inlte.le.30) then
|
||||
/// inlte=inlte-20
|
||||
/// ifreq=11
|
||||
/// lasdel=.true.
|
||||
/// else if(inlte.le.40) then
|
||||
/// inlte=inlte-30
|
||||
/// ifreq=11
|
||||
/// lasdel=.false.
|
||||
/// end if
|
||||
/// ```
|
||||
pub fn parse_nlte_mode(inlte: i32) -> (i32, bool, i32) {
|
||||
if inlte < 10 {
|
||||
(inlte, true, 0)
|
||||
} else if inlte <= 20 {
|
||||
(inlte - 10, false, 0)
|
||||
} else if inlte <= 30 {
|
||||
(inlte - 20, true, 11)
|
||||
} else {
|
||||
(inlte - 30, false, 11)
|
||||
}
|
||||
}
|
||||
|
||||
/// 速度截止检查
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// do id=1,nd
|
||||
/// ilvi(id)=0
|
||||
/// ilne(id)=0
|
||||
/// if(vel(id).gt.velmax.and.iemoff.eq.0) ilvi(id)=1
|
||||
/// if(vel(id).gt.velmax.and.nltoff.gt.0.and.iemoff.gt.0) ilne(id)=1
|
||||
/// end do
|
||||
/// ```
|
||||
pub fn check_velocity_cutoff(
|
||||
velocities: &[f64],
|
||||
velmax: f64,
|
||||
iemoff: i32,
|
||||
nltoff: i32,
|
||||
) -> (Vec<bool>, Vec<bool>) {
|
||||
let ilvi: Vec<bool> = velocities.iter().map(|&v| v > velmax && iemoff == 0).collect();
|
||||
let ilne: Vec<bool> = velocities.iter()
|
||||
.map(|&v| v > velmax && nltoff > 0 && iemoff > 0)
|
||||
.collect();
|
||||
(ilvi, ilne)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_compute_wavelength_range() {
|
||||
let wr = compute_wavelength_range(4000.0, 7000.0, 10.0, 5.0, 1e-4, 10000.0);
|
||||
assert!((wr.alam0 - 400.0).abs() < 1e-10);
|
||||
assert!((wr.alast - 700.0).abs() < 1e-10);
|
||||
assert!((wr.alamc - 550.0).abs() < 1e-10);
|
||||
assert!((wr.cutof0 - 1.0).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_wavelength_range_auto_space() {
|
||||
let wr = compute_wavelength_range(4000.0, 7000.0, 10.0, 0.0, 1e-4, 10000.0);
|
||||
// space = 4.3e-8 * sqrt(10000) * 5500 = 4.3e-8 * 100 * 5500 = 0.02365
|
||||
assert!(wr.space0 > 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_frequency_range() {
|
||||
let fr = compute_frequency_range(500.0, 700.0);
|
||||
assert!((fr.freq1 - CNM / 500.0).abs() < 1e-10);
|
||||
assert!((fr.freq2 - CNM / 700.0).abs() < 1e-10);
|
||||
assert!((fr.frlast - fr.freq2).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_angle_points_uniform() {
|
||||
let config = compute_angle_points_uniform(5, 0.1);
|
||||
assert_eq!(config.nmu, 5);
|
||||
assert_eq!(config.angles.len(), 5);
|
||||
assert!((config.angles[0] - 1.0).abs() < 1e-10);
|
||||
assert!((config.angles[4] - 0.1).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_angle_points_sine() {
|
||||
let config = compute_angle_points_sine(5, 0.5);
|
||||
assert_eq!(config.nmu, 5);
|
||||
// 角度应该从 ~1.0 递减
|
||||
assert!(config.angles[0] > config.angles[4]);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_continuum_frequency_grid() {
|
||||
let grid = compute_continuum_frequency_grid(400.0, 700.0, 3.0, 1000);
|
||||
assert!(grid.nfreqc > 2);
|
||||
assert_eq!(grid.frequencies.len(), grid.nfreqc);
|
||||
assert_eq!(grid.wavelengths.len(), grid.nfreqc);
|
||||
// 频率应该递减
|
||||
assert!(grid.frequencies[0] > grid.frequencies[grid.nfreqc - 1]);
|
||||
// 波长应该递增
|
||||
assert!(grid.wavelengths[0] < grid.wavelengths[grid.nfreqc - 1]);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parse_nlte_mode() {
|
||||
assert_eq!(parse_nlte_mode(0), (0, true, 0));
|
||||
assert_eq!(parse_nlte_mode(15), (5, false, 0));
|
||||
assert_eq!(parse_nlte_mode(25), (5, true, 11));
|
||||
assert_eq!(parse_nlte_mode(35), (5, false, 11));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_check_velocity_cutoff() {
|
||||
let vel = vec![100.0, 200.0, 300.0, 400.0];
|
||||
let (ilvi, ilne) = check_velocity_cutoff(&vel, 250.0, 0, 0);
|
||||
assert_eq!(ilvi, vec![false, false, true, true]);
|
||||
assert_eq!(ilne, vec![false, false, false, false]);
|
||||
}
|
||||
}
|
||||
331
src/synspec/math/inibl1.rs
Normal file
331
src/synspec/math/inibl1.rs
Normal file
@ -0,0 +1,331 @@
|
||||
//! inibl1 — 辅助初始化过程（第二阶段）。
|
||||
//!
|
||||
//! Fortran 原始签名: SUBROUTINE INIBL1(IGRD)
|
||||
//!
|
||||
//! 重置波长范围，设置连续频率网格，计算标准不透明度。
|
||||
//!
|
||||
//! 注意: Fortran 版本直接操作文件 I/O 和 COMMON 块。
|
||||
//! Rust 版本提供纯计算核心函数。
|
||||
|
||||
/// 物理常数
|
||||
const CLC: f64 = 2.997925e17; // 光速 (nm/s)
|
||||
|
||||
/// RELOP 自动检测
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// if(relops.eq.0) then
|
||||
/// relop=1.e-15
|
||||
/// if(temp(1).lt.2.e6) relop=1.e-6
|
||||
/// if(temp(1).lt.1.e6) relop=1.e-5
|
||||
/// if(temp(1).lt.1.e5) relop=1.e-4
|
||||
/// end if
|
||||
/// ```
|
||||
pub fn auto_detect_relop(relops: f64, temp_surface: f64) -> f64 {
|
||||
if relops != 0.0 {
|
||||
return relops;
|
||||
}
|
||||
|
||||
if temp_surface < 1e5 {
|
||||
1e-4
|
||||
} else if temp_surface < 1e6 {
|
||||
1e-5
|
||||
} else if temp_surface < 2e6 {
|
||||
1e-6
|
||||
} else {
|
||||
1e-15
|
||||
}
|
||||
}
|
||||
|
||||
/// 重置波长范围参数
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// alam0=alam0s
|
||||
/// if(alam0s.eq.0.) alam0=5.e7/temp(1)/10.
|
||||
/// if(alam0s.lt.0.) alam0=-5.e7/temp(1)/alam0s
|
||||
/// alast=alasts
|
||||
/// if(alasts.eq.0.) alast=5.e7/temp(1)*20.
|
||||
/// if(alasts.lt.0.) alast=-5.e7/temp(1)/alam0s
|
||||
/// ```
|
||||
pub fn reset_wavelength_range(
|
||||
alam0s: f64,
|
||||
alasts: f64,
|
||||
temp_surface: f64,
|
||||
) -> (f64, f64) {
|
||||
let alam0 = if alam0s == 0.0 {
|
||||
5e7 / temp_surface / 10.0
|
||||
} else if alam0s < 0.0 {
|
||||
-5e7 / temp_surface / alam0s
|
||||
} else {
|
||||
alam0s
|
||||
};
|
||||
|
||||
let alast = if alasts == 0.0 {
|
||||
5e7 / temp_surface * 20.0
|
||||
} else if alasts < 0.0 {
|
||||
-5e7 / temp_surface / alasts
|
||||
} else {
|
||||
alasts
|
||||
};
|
||||
|
||||
(alam0, alast)
|
||||
}
|
||||
|
||||
/// 计算连续频率网格（INIBL1 版本）
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// nfreqc=ifix(real(cutofs,4))
|
||||
/// if(nfreqc.eq.0) nfreqc=mfreq
|
||||
/// all0=log(alam0)
|
||||
/// all1=log(alast)
|
||||
/// dlc=(all1-all0)/(nfreqc-1)
|
||||
/// do ijc=1,nfreqc
|
||||
/// wlamc(ijc)=exp(all0+(ijc-1)*dlc)
|
||||
/// freqc(ijc)=clc/wlamc(ijc)
|
||||
/// end do
|
||||
/// ```
|
||||
pub fn compute_continuum_grid_inibl1(
|
||||
alam0_nm: f64,
|
||||
alast_nm: f64,
|
||||
cutofs: f64,
|
||||
max_nfreq: usize,
|
||||
) -> (Vec<f64>, Vec<f64>) {
|
||||
let nfreqc = if cutofs == 0.0 {
|
||||
max_nfreq
|
||||
} else {
|
||||
cutofs as usize
|
||||
};
|
||||
|
||||
let all0 = alam0_nm.ln();
|
||||
let all1 = alast_nm.ln();
|
||||
let dlc = (all1 - all0) / (nfreqc - 1) as f64;
|
||||
|
||||
let wavelengths: Vec<f64> = (0..nfreqc)
|
||||
.map(|ijc| (all0 + ijc as f64 * dlc).exp())
|
||||
.collect();
|
||||
|
||||
let frequencies: Vec<f64> = wavelengths.iter().map(|&wl| CLC / wl).collect();
|
||||
|
||||
(wavelengths, frequencies)
|
||||
}
|
||||
|
||||
/// 标准不透明度选择
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// ABSTD(ID)=MIN(ABSO(1)+SCAT(1),ABSO(2)+SCAT(2))
|
||||
/// ```
|
||||
pub fn select_standard_opacity(abs: &[f64], scat: &[f64]) -> f64 {
|
||||
let op1 = abs[0] + scat[0];
|
||||
let op2 = abs[1] + scat[1];
|
||||
op1.min(op2)
|
||||
}
|
||||
|
||||
/// 不透明度限制
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// absoc(ijc)=min(absoc(ijc),1.e30)
|
||||
/// ```
|
||||
pub fn limit_opacity(opacity: f64) -> f64 {
|
||||
opacity.min(1e30)
|
||||
}
|
||||
|
||||
/// 标准不透明度窗口模式
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// DO IJ=1,NFREQC
|
||||
/// denscon(id)=1.
|
||||
/// ABSTDW(IJ,ID)=ABSOC(IJ)/DENSCON(ID)
|
||||
/// END DO
|
||||
/// ```
|
||||
pub fn compute_window_opacity(
|
||||
absoc: &[f64],
|
||||
denscon: f64,
|
||||
) -> Vec<f64> {
|
||||
absoc.iter().map(|&op| op / denscon).collect()
|
||||
}
|
||||
|
||||
/// 溶解分数初始化
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// DO ID=1,ND
|
||||
/// anh2(id)=0.
|
||||
/// anhm(id)=0.
|
||||
/// anch(id)=0.
|
||||
/// anoh(id)=0.
|
||||
/// END DO
|
||||
/// ```
|
||||
pub fn init_dissolved_fractions(nd: usize) -> (Vec<f64>, Vec<f64>, Vec<f64>, Vec<f64>) {
|
||||
(
|
||||
vec![0.0; nd], // anh2
|
||||
vec![0.0; nd], // anhm
|
||||
vec![0.0; nd], // anch
|
||||
vec![0.0; nd], // anoh
|
||||
)
|
||||
}
|
||||
|
||||
/// INIBL1 配置参数
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct Inibl1Config {
|
||||
/// 表面温度 (K)
|
||||
pub temp_surface: f64,
|
||||
/// 标准深度点温度 (K)
|
||||
pub temp_std: f64,
|
||||
/// 波长范围保存值
|
||||
pub alam0s: f64,
|
||||
pub alasts: f64,
|
||||
/// 截止参数保存值
|
||||
pub cutof0s: f64,
|
||||
pub cutofss: f64,
|
||||
/// RELOP 保存值
|
||||
pub relops: f64,
|
||||
/// SPACE 保存值
|
||||
pub spaces: f64,
|
||||
/// 深度点数
|
||||
pub nd: usize,
|
||||
/// 标准深度索引
|
||||
pub idstd: usize,
|
||||
}
|
||||
|
||||
/// INIBL1 计算结果
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct Inibl1Result {
|
||||
/// 波长范围 (nm)
|
||||
pub alam0: f64,
|
||||
pub alast: f64,
|
||||
pub alamc: f64,
|
||||
/// 截止参数
|
||||
pub cutof0: f64,
|
||||
pub space0: f64,
|
||||
/// 最小不透明度比
|
||||
pub relop: f64,
|
||||
/// 频率范围
|
||||
pub frlast: f64,
|
||||
/// 连续频率网格
|
||||
pub wavelengths: Vec<f64>,
|
||||
pub frequencies: Vec<f64>,
|
||||
}
|
||||
|
||||
/// 执行 INIBL1 核心计算
|
||||
pub fn compute_inibl1(config: &Inibl1Config, max_nfreq: usize) -> Inibl1Result {
|
||||
// 重置波长范围
|
||||
let (alam0_ang, alast_ang) = reset_wavelength_range(
|
||||
config.alam0s,
|
||||
config.alasts,
|
||||
config.temp_surface,
|
||||
);
|
||||
|
||||
// 自动检测 RELOP
|
||||
let relop = auto_detect_relop(config.relops, config.temp_surface);
|
||||
|
||||
// 计算波长中心和间距
|
||||
let alamc_ang = (alam0_ang + alast_ang) * 0.5;
|
||||
let space = if config.spaces == 0.0 {
|
||||
4.3e-8 * config.temp_std.sqrt() * alamc_ang
|
||||
} else if config.spaces < 0.0 {
|
||||
-5.72e-8 * config.temp_std.sqrt() * alamc_ang * config.spaces
|
||||
} else {
|
||||
config.spaces
|
||||
};
|
||||
|
||||
// 转换为 nm
|
||||
let alam0 = alam0_ang * 0.1;
|
||||
let alast = alast_ang * 0.1;
|
||||
let alamc = alamc_ang * 0.1;
|
||||
let cutof0 = config.cutof0s * 0.1;
|
||||
let space0 = space * 0.1;
|
||||
let frlast = CLC / alast;
|
||||
|
||||
// 计算连续频率网格
|
||||
let (wavelengths, frequencies) = compute_continuum_grid_inibl1(
|
||||
alam0,
|
||||
alast,
|
||||
config.cutofss,
|
||||
max_nfreq,
|
||||
);
|
||||
|
||||
Inibl1Result {
|
||||
alam0,
|
||||
alast,
|
||||
alamc,
|
||||
cutof0,
|
||||
space0,
|
||||
relop,
|
||||
frlast,
|
||||
wavelengths,
|
||||
frequencies,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_auto_detect_relop_hot() {
|
||||
assert_eq!(auto_detect_relop(0.0, 3e6), 1e-15);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_auto_detect_relop_warm() {
|
||||
assert_eq!(auto_detect_relop(0.0, 1.5e6), 1e-6);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_auto_detect_relop_cool() {
|
||||
assert_eq!(auto_detect_relop(0.0, 5e5), 1e-5);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_auto_detect_relop_cold() {
|
||||
assert_eq!(auto_detect_relop(0.0, 5e4), 1e-4);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_auto_detect_relop_user() {
|
||||
assert_eq!(auto_detect_relop(1e-3, 5e4), 1e-3);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_reset_wavelength_range_auto() {
|
||||
let (alam0, alast) = reset_wavelength_range(0.0, 0.0, 10000.0);
|
||||
// alam0 = 5e7/10000/10 = 500
|
||||
// alast = 5e7/10000*20 = 100000
|
||||
assert!((alam0 - 500.0).abs() < 1e-10);
|
||||
assert!((alast - 100000.0).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_reset_wavelength_range_user() {
|
||||
let (alam0, alast) = reset_wavelength_range(4000.0, 7000.0, 10000.0);
|
||||
assert_eq!(alam0, 4000.0);
|
||||
assert_eq!(alast, 7000.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_continuum_grid_inibl1() {
|
||||
let (wl, fr) = compute_continuum_grid_inibl1(400.0, 700.0, 100.0, 1000);
|
||||
assert_eq!(wl.len(), 100);
|
||||
assert_eq!(fr.len(), 100);
|
||||
assert!((wl[0] - 400.0).abs() < 1e-10);
|
||||
assert!((wl[99] - 700.0).abs() < 1e-3);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_select_standard_opacity() {
|
||||
let abs = vec![100.0, 200.0];
|
||||
let scat = vec![50.0, 30.0];
|
||||
assert_eq!(select_standard_opacity(&abs, &scat), 150.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_limit_opacity() {
|
||||
assert_eq!(limit_opacity(1e31), 1e30);
|
||||
assert_eq!(limit_opacity(1e29), 1e29);
|
||||
}
|
||||
}
|
||||
443
src/synspec/math/iniblh.rs
Normal file
443
src/synspec/math/iniblh.rs
Normal file
@ -0,0 +1,443 @@
|
||||
//! 氢线信息输出。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `iniblh.f` (synspec54.f:9737)。
|
||||
//!
|
||||
//! 计算并输出选定氢线的等值宽度和强度信息。
|
||||
|
||||
use super::stark0::stark0;
|
||||
use super::inibla::{BN, HK};
|
||||
|
||||
// ============================================================================
|
||||
// 物理常数
|
||||
// ============================================================================
|
||||
|
||||
/// ln(10) 转换因子
|
||||
const C1: f64 = std::f64::consts::LN_10;
|
||||
|
||||
/// log10(e) * ln(10) 转换因子
|
||||
const C2: f64 = 4.2014672;
|
||||
|
||||
/// hc/k (cm K) 用于能量转换
|
||||
const C3: f64 = 1.4387886;
|
||||
|
||||
/// Doppler 宽度参数 1
|
||||
const DP0: f64 = 3.33564e-11;
|
||||
|
||||
/// Doppler 宽度参数 2
|
||||
const DP1: f64 = 1.651e8;
|
||||
|
||||
/// 速度单位转换
|
||||
const UN: f64 = 1.0;
|
||||
|
||||
/// 电离能常数 (cm^-1)
|
||||
const EXCL_CONST: f64 = 109679.0;
|
||||
|
||||
/// 光速 (Å/s) 用于波长转换
|
||||
const CLIGHT_A: f64 = 2.997925e18;
|
||||
|
||||
// ============================================================================
|
||||
// 参数结构体
|
||||
// ============================================================================
|
||||
|
||||
/// INIBLH 输入参数。
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct IniblhParams {
|
||||
/// 打印级别 (<= -2 跳过输出)
|
||||
pub iprin: i32,
|
||||
|
||||
/// 氢线处理标志 (< 0 表示排除)
|
||||
pub ihyl: i32,
|
||||
|
||||
/// 频率范围下限 (Hz) - FREQ(1)
|
||||
pub freq1: f64,
|
||||
|
||||
/// 频率范围上限 (Hz) - FREQ(2)
|
||||
pub freq2: f64,
|
||||
|
||||
/// 频率点数量
|
||||
pub nfreq: usize,
|
||||
|
||||
/// 氢线系列索引下限 (来自 HYLSET)
|
||||
pub ilowh: i32,
|
||||
|
||||
/// 主量子数上限 1 (来自 HYLSET)
|
||||
pub m10: i32,
|
||||
|
||||
/// 主量子数上限 2 (来自 HYLSET)
|
||||
pub m20: i32,
|
||||
|
||||
/// 标准深度索引 (1-indexed)
|
||||
pub idstd: usize,
|
||||
|
||||
/// 温度数组 [nd] (K)
|
||||
pub temp: Vec<f64>,
|
||||
|
||||
/// 电子密度数组 [nd] (cm^-3)
|
||||
pub elec: Vec<f64>,
|
||||
|
||||
/// 表面重力 log g (cgs)
|
||||
pub grav: f64,
|
||||
|
||||
/// 氢原子质量 (amu)
|
||||
pub amas_h: f64,
|
||||
|
||||
/// 湍流速度数组 [nd] (km/s)
|
||||
pub vturb: Vec<f64>,
|
||||
|
||||
/// RRR 数组 [nd] - 辐射场修正因子
|
||||
pub rrr: Vec<f64>,
|
||||
|
||||
/// 标准深度吸收系数
|
||||
pub abstd: f64,
|
||||
}
|
||||
|
||||
/// 单条氢线信息。
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct HydrogenLineInfo {
|
||||
/// 波长 (Å)
|
||||
pub wavelength: f64,
|
||||
/// 振荡强度 log(gf)
|
||||
pub log_gf: f64,
|
||||
/// 电离能 (cm^-1)
|
||||
pub excitation: f64,
|
||||
/// 强度参数
|
||||
pub strength: f64,
|
||||
/// 等值宽度 (mÅ)
|
||||
pub equivalent_width: f64,
|
||||
/// 系列索引
|
||||
pub series_index: i32,
|
||||
/// 主量子数
|
||||
pub quantum_number: i32,
|
||||
}
|
||||
|
||||
/// INIBLH 输出结果。
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct IniblhOutput {
|
||||
/// 计算的氢线列表
|
||||
pub lines: Vec<HydrogenLineInfo>,
|
||||
/// Planck 函数值
|
||||
pub planck: f64,
|
||||
/// 受激发射因子
|
||||
pub stim: f64,
|
||||
/// Doppler 宽度参数
|
||||
pub dopa1: f64,
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// INIBLH 函数
|
||||
// ============================================================================
|
||||
|
||||
/// 计算并输出氢线信息。
|
||||
///
|
||||
/// 根据频率范围和氢线参数，计算选定氢线的等值宽度和强度。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `params` - 输入参数结构体
|
||||
///
|
||||
/// # 返回
|
||||
///
|
||||
/// 包含计算的氢线列表和相关物理量的输出结构体
|
||||
#[allow(unused_assignments)]
|
||||
pub fn iniblh(params: &IniblhParams) -> IniblhOutput {
|
||||
// 如果打印级别过低或氢线被排除，返回空结果
|
||||
if params.iprin <= -2 || params.ihyl < 0 {
|
||||
return IniblhOutput {
|
||||
lines: Vec::new(),
|
||||
planck: 0.0,
|
||||
stim: 0.0,
|
||||
dopa1: 0.0,
|
||||
};
|
||||
}
|
||||
|
||||
// 计算波长范围 (Å)
|
||||
let alm0 = CLIGHT_A / params.freq1;
|
||||
let alm1 = CLIGHT_A / params.freq2;
|
||||
|
||||
// 计算平均频率
|
||||
let xx = if params.nfreq >= 2 {
|
||||
0.5 * (params.freq1 + params.freq2)
|
||||
} else {
|
||||
params.freq1
|
||||
};
|
||||
|
||||
// 计算 Planck 函数和相关量
|
||||
let bnu = BN * (xx * 1.0e-15).powi(3);
|
||||
let hk_f = HK * xx;
|
||||
|
||||
// 获取标准深度的物理量
|
||||
// 注意：idstd 是 1-indexed，数组也是从索引 1 开始填充
|
||||
let id = params.idstd;
|
||||
let t = params.temp[id];
|
||||
let _ane = params.elec[id];
|
||||
|
||||
// 计算激发因子
|
||||
let exh = (hk_f / t).exp();
|
||||
let exhk = UN / exh;
|
||||
let plan = bnu / (exh - UN);
|
||||
let stim = UN - exhk;
|
||||
|
||||
// 计算 Doppler 宽度
|
||||
let dopa1 = UN / (xx * DP0 * (DP1 * t / params.amas_h + params.vturb[id]).sqrt());
|
||||
|
||||
// 确定系列范围
|
||||
let mut iserl = params.ilowh;
|
||||
let mut iseru = params.ilowh;
|
||||
|
||||
if alm0 > 17000.0 && alm1 < 21000.0 {
|
||||
iserl = 3;
|
||||
iseru = 4;
|
||||
} else if alm0 > 22700.0 {
|
||||
iserl = 4;
|
||||
iseru = 5;
|
||||
if alm0 > 32800.0 { iseru = 6; }
|
||||
if alm0 > 44660.0 { iseru = 7; }
|
||||
}
|
||||
|
||||
let mut lines = Vec::new();
|
||||
|
||||
// 遍历系列
|
||||
for i in iserl..=iseru {
|
||||
let ii = (i * i) as f64;
|
||||
let xii = UN / ii;
|
||||
|
||||
// 计算量子数范围
|
||||
let mut m1 = params.m10;
|
||||
if i < params.ilowh {
|
||||
m1 = params.ilowh - 1;
|
||||
}
|
||||
let mut m2 = m1 + 1;
|
||||
if m1 < i + 1 {
|
||||
m1 = i + 1;
|
||||
}
|
||||
m1 -= 1;
|
||||
m2 = params.m20 + 3;
|
||||
if m1 < i + 1 {
|
||||
m1 = i + 1;
|
||||
}
|
||||
|
||||
// 根据重力调整范围
|
||||
if params.grav > 3.0 {
|
||||
m2 += 5;
|
||||
m1 -= 3;
|
||||
if m1 > i + 6 {
|
||||
m1 -= 3;
|
||||
}
|
||||
}
|
||||
if params.grav > 6.0 {
|
||||
m2 += 2;
|
||||
m1 -= 1;
|
||||
if m1 > i + 6 {
|
||||
m1 -= 1;
|
||||
}
|
||||
}
|
||||
|
||||
if m1 < i + 1 {
|
||||
m1 = i + 1;
|
||||
}
|
||||
if m2 > 20 {
|
||||
m2 = 20;
|
||||
}
|
||||
|
||||
// 遍历量子数
|
||||
for j in (m2..=m1).rev() {
|
||||
let stark_result = stark0(i, j, 1);
|
||||
let alam = stark_result.wl0;
|
||||
|
||||
// 检查是否在波长范围内
|
||||
if alam >= alm0 && alam < alm1 {
|
||||
let gh = 2.0 * ii;
|
||||
let gf = (stark_result.fij * gh).log10();
|
||||
let excl = EXCL_CONST * (1.0 - xii);
|
||||
let excl0h = excl * C3;
|
||||
let gf0h = gf * C1 - C2;
|
||||
|
||||
// 计算吸收系数
|
||||
let abcnt = (gf0h - excl0h / t).exp()
|
||||
* params.rrr[id]
|
||||
* dopa1
|
||||
* stim;
|
||||
|
||||
// 计算强度参数
|
||||
let str0 = abcnt / params.abstd;
|
||||
|
||||
// 计算等值宽度
|
||||
let ww1 = if str0 <= 1.2 {
|
||||
0.886 * str0 * (1.0 - str0 * (0.707 - str0 * 0.577))
|
||||
} else {
|
||||
str0.ln().sqrt()
|
||||
};
|
||||
|
||||
let ww1 = if str0 > 55.0 {
|
||||
let agam = 0.01;
|
||||
let ww2 = 0.5_f64 * (std::f64::consts::PI * agam * str0).sqrt();
|
||||
if ww2 > ww1 { ww2 } else { ww1 }
|
||||
} else {
|
||||
ww1
|
||||
};
|
||||
|
||||
let eqw = alam * alam / 3.0e18 * 1.0e3 / dopa1 * ww1;
|
||||
let str = eqw * 10.0;
|
||||
|
||||
lines.push(HydrogenLineInfo {
|
||||
wavelength: alam,
|
||||
log_gf: gf,
|
||||
excitation: excl,
|
||||
strength: str0,
|
||||
equivalent_width: str,
|
||||
series_index: i,
|
||||
quantum_number: j,
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
IniblhOutput {
|
||||
lines,
|
||||
planck: plan,
|
||||
stim,
|
||||
dopa1,
|
||||
}
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// 测试
|
||||
// ============================================================================
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
/// 创建默认测试参数
|
||||
fn create_test_params() -> IniblhParams {
|
||||
IniblhParams {
|
||||
iprin: 0,
|
||||
ihyl: 1,
|
||||
freq1: 4.0e14, // 750 nm
|
||||
freq2: 8.0e14, // 375 nm
|
||||
nfreq: 2,
|
||||
ilowh: 2,
|
||||
m10: 3,
|
||||
m20: 10,
|
||||
idstd: 1,
|
||||
temp: vec![0.0, 10000.0, 15000.0], // index 0 unused, 1 = standard
|
||||
elec: vec![0.0, 1.0e13, 1.0e14],
|
||||
grav: 4.0,
|
||||
amas_h: 1.0,
|
||||
vturb: vec![0.0, 2.0, 2.0],
|
||||
rrr: vec![0.0, 1.0, 1.0],
|
||||
abstd: 1.0e-5,
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_iniblh_disabled() {
|
||||
// IPRIN <= -2 时应返回空结果
|
||||
let params = IniblhParams {
|
||||
iprin: -3,
|
||||
..create_test_params()
|
||||
};
|
||||
let result = iniblh(¶ms);
|
||||
assert!(result.lines.is_empty());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_iniblh_hyl_negative() {
|
||||
// IHYL < 0 时应返回空结果
|
||||
let params = IniblhParams {
|
||||
ihyl: -1,
|
||||
..create_test_params()
|
||||
};
|
||||
let result = iniblh(¶ms);
|
||||
assert!(result.lines.is_empty());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_iniblh_basic() {
|
||||
// 基本功能测试
|
||||
let params = create_test_params();
|
||||
let result = iniblh(¶ms);
|
||||
// 可能有也可能没有线在范围内，取决于波长范围
|
||||
assert!(result.planck.is_finite());
|
||||
assert!(result.stim.is_finite());
|
||||
assert!(result.dopa1 > 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_iniblh_planck_calculation() {
|
||||
// 测试 Planck 函数计算
|
||||
let params = IniblhParams {
|
||||
freq1: 5.0e14,
|
||||
freq2: 6.0e14,
|
||||
..create_test_params()
|
||||
};
|
||||
let result = iniblh(¶ms);
|
||||
assert!(result.planck > 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_iniblh_doppler_width() {
|
||||
// 测试 Doppler 宽度计算
|
||||
let params = create_test_params();
|
||||
let result = iniblh(¶ms);
|
||||
assert!(result.dopa1 > 0.0);
|
||||
// DOPA1 = 1/(Doppler宽度)，高温 → 更大的 Doppler 宽度 → 更小的 DOPA1
|
||||
let params_hot = IniblhParams {
|
||||
temp: vec![0.0, 20000.0, 30000.0],
|
||||
..create_test_params()
|
||||
};
|
||||
let result_hot = iniblh(¶ms_hot);
|
||||
assert!(result_hot.dopa1 < result.dopa1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_iniblh_high_gravity() {
|
||||
// 测试高重力情况
|
||||
let params = IniblhParams {
|
||||
grav: 7.0,
|
||||
..create_test_params()
|
||||
};
|
||||
let result = iniblh(¶ms);
|
||||
// 高重力应扩展量子数范围
|
||||
assert!(result.stim.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_iniblh_line_properties() {
|
||||
// 测试计算出的线属性
|
||||
let params = IniblhParams {
|
||||
// 使用 Balmer 系列范围 (364.6 - 820 nm)
|
||||
freq1: CLIGHT_A / 656.0, // Hα 附近
|
||||
freq2: CLIGHT_A / 486.0, // Hβ 附近
|
||||
..create_test_params()
|
||||
};
|
||||
let result = iniblh(¶ms);
|
||||
|
||||
// 检查线属性的有效性
|
||||
for line in &result.lines {
|
||||
assert!(line.wavelength > 0.0);
|
||||
assert!(line.log_gf.is_finite());
|
||||
assert!(line.excitation > 0.0);
|
||||
assert!(line.strength >= 0.0);
|
||||
assert!(line.equivalent_width >= 0.0);
|
||||
assert!(line.series_index >= 1);
|
||||
assert!(line.quantum_number >= 2);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_iniblh_lyman_series() {
|
||||
// 测试 Lyman 系列
|
||||
let params = IniblhParams {
|
||||
freq1: CLIGHT_A / 121.6, // Lyman-α
|
||||
freq2: CLIGHT_A / 102.6, // Lyman-β
|
||||
ilowh: 1,
|
||||
..create_test_params()
|
||||
};
|
||||
let result = iniblh(¶ms);
|
||||
// Lyman 系列的线应该在范围内
|
||||
for line in &result.lines {
|
||||
assert!(line.series_index >= 1);
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -9,12 +9,15 @@
|
||||
// ============================================================================
|
||||
|
||||
/// 光速 (cm/s)
|
||||
#[allow(dead_code)]
|
||||
pub const CL: f64 = 2.997925e10;
|
||||
|
||||
/// 普朗克常数 (erg·s)
|
||||
#[allow(dead_code)]
|
||||
pub const H: f64 = 6.6256e-27;
|
||||
|
||||
/// 玻尔兹曼常数 (erg/K)
|
||||
#[allow(dead_code)]
|
||||
pub const BOLK: f64 = 1.38054e-16;
|
||||
|
||||
/// Planck 函数常数 BN = 2*h*c²/c³ = 2*h/c²
|
||||
|
||||
1039
src/synspec/math/inilin.rs
Normal file
1039
src/synspec/math/inilin.rs
Normal file
File diff suppressed because it is too large
Load Diff
363
src/synspec/math/inilin_grid.rs
Normal file
363
src/synspec/math/inilin_grid.rs
Normal file
@ -0,0 +1,363 @@
|
||||
//! inilin_grid — 不透明度网格的原子线列表初始化。
|
||||
//!
|
||||
//! Fortran 原始签名: SUBROUTINE INILIN_grid
|
||||
//!
|
||||
//! 读取原子线列表，选择可能贡献的线，设置线参数。
|
||||
//! 用于不透明度表计算的网格模式。
|
||||
//!
|
||||
//! 注意: Fortran 版本直接操作文件 I/O 和 COMMON 块。
|
||||
//! Rust 版本提供纯计算核心函数。
|
||||
|
||||
/// 物理常数
|
||||
const C1: f64 = std::f64::consts::LN_10; // ln(10)
|
||||
const C2: f64 = 4.2014672; // ln(10) * (me*c^2)/(k*T_ref)
|
||||
const C3: f64 = 1.4387886; // h*c/k (cm*K)
|
||||
const CNM: f64 = 2.997925e17; // c in nm/s
|
||||
const AGR0: f64 = 2.4734e-22; // 自然辐射阻尼常数
|
||||
const XEH: f64 = 13.595; // 氢电离势 (eV)
|
||||
const XET: f64 = 8067.6; // eV 到 cm^-1 转换
|
||||
const XNF: f64 = 25.0; // 最大有效量子数平方
|
||||
const R02: f64 = 2.5; // VdW 半径参数 (轻元素)
|
||||
const R12: f64 = 45.0; // VdW 半径参数 (中等元素)
|
||||
const VW0: f64 = 4.5e-9; // VdW 常数
|
||||
const OP4: f64 = 0.4; // 2/5 指数
|
||||
|
||||
/// 原子线参数
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct AtomicLineParams {
|
||||
/// 波长 (nm)
|
||||
pub alam: f64,
|
||||
/// 元素代码 (Kurucz-Peytremann)
|
||||
pub anum: f64,
|
||||
/// log(gf)
|
||||
pub gf: f64,
|
||||
/// 下能级激发势 (cm^-1)
|
||||
pub excl: f64,
|
||||
/// 下能级 J 量子数
|
||||
pub ql: f64,
|
||||
/// 上能级激发势 (cm^-1)
|
||||
pub excu: f64,
|
||||
/// 上能级 J 量子数
|
||||
pub qu: f64,
|
||||
/// 辐射阻尼参数
|
||||
pub agam: f64,
|
||||
/// Stark 阻尼参数 (log)
|
||||
pub gs: f64,
|
||||
/// Van der Waals 阻尼参数 (log)
|
||||
pub gw: f64,
|
||||
}
|
||||
|
||||
/// 线强度参数
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct LineStrengthGrid {
|
||||
/// log(gf) * ln(10)
|
||||
pub gfp: f64,
|
||||
/// 下能级激发势 * h*c/k
|
||||
pub epp: f64,
|
||||
/// 频率 (s^-1)
|
||||
pub freq: f64,
|
||||
}
|
||||
|
||||
/// 计算线强度参数
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// GFP=C1*GF-C2
|
||||
/// EPP=C3*EXCL
|
||||
/// FR0=CNM/ALAM
|
||||
/// ```
|
||||
pub fn compute_line_strength_grid(alam: f64, gf: f64, excl: f64) -> LineStrengthGrid {
|
||||
LineStrengthGrid {
|
||||
gfp: C1 * gf - C2,
|
||||
epp: C3 * excl.abs(),
|
||||
freq: CNM / alam,
|
||||
}
|
||||
}
|
||||
|
||||
/// 计算有效量子数平方
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// Z=FLOAT(ION)
|
||||
/// XNEFF2=Z**2*(XEH/(ENEV(IAT,ION)-EXCU/XET))
|
||||
/// IF(XNEFF2.LE.0..OR.XNEFF2.GT.XNF) XNEFF2=XNF
|
||||
/// ```
|
||||
pub fn effective_quantum_number_squared(ion: i32, excu_cm: f64, ionization_energy_ev: f64) -> f64 {
|
||||
let z = ion as f64;
|
||||
let excu_ev = excu_cm / XET;
|
||||
let xneff2 = z * z * (XEH / (ionization_energy_ev - excu_ev));
|
||||
|
||||
if xneff2 <= 0.0 || xneff2 > XNF {
|
||||
XNF
|
||||
} else {
|
||||
xneff2
|
||||
}
|
||||
}
|
||||
|
||||
/// 计算自然辐射阻尼
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// IF(AGAM.GT.0.) THEN
|
||||
/// GAMR0=EXP(C1*AGAM)
|
||||
/// ELSE
|
||||
/// GAMR0=AGR0*FR0*FR0
|
||||
/// END IF
|
||||
/// ```
|
||||
pub fn natural_broadening_grid(agam: f64, freq: f64) -> f64 {
|
||||
if agam > 0.0 {
|
||||
(C1 * agam).exp()
|
||||
} else {
|
||||
AGR0 * freq * freq
|
||||
}
|
||||
}
|
||||
|
||||
/// 计算 Stark 阻尼参数
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// IF(GS.NE.0.) THEN
|
||||
/// GS0=EXP(C1*GS)
|
||||
/// ELSE
|
||||
/// GS0=TENM8*XNEFF2*XNEFF2*SQRT(XNEFF2)
|
||||
/// END IF
|
||||
/// ```
|
||||
pub fn stark_broadening_grid(gs: f64, xneff2: f64) -> f64 {
|
||||
if gs != 0.0 {
|
||||
(C1 * gs).exp()
|
||||
} else {
|
||||
1e-8 * xneff2 * xneff2 * xneff2.sqrt()
|
||||
}
|
||||
}
|
||||
|
||||
/// 计算 Van der Waals 阻尼参数
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// IF(GW.NE.0.) THEN
|
||||
/// GW0=EXP(C1*GW)
|
||||
/// ELSE
|
||||
/// IF(IAT.LT.21) THEN
|
||||
/// R2=R02*(XNEFF2/Z)**2
|
||||
/// ELSE IF(IAT.LT.45) THEN
|
||||
/// R2=(R12-FLOAT(IAT))/Z
|
||||
/// ELSE
|
||||
/// R2=0.5
|
||||
/// END IF
|
||||
/// GW0=VW0*R2**OP4
|
||||
/// END IF
|
||||
/// ```
|
||||
pub fn vdw_broadening_grid(gw: f64, iat: i32, ion: i32, xneff2: f64) -> f64 {
|
||||
if gw != 0.0 {
|
||||
(C1 * gw).exp()
|
||||
} else {
|
||||
let z = ion as f64;
|
||||
let r2 = if iat < 21 {
|
||||
R02 * (xneff2 / z).powi(2)
|
||||
} else if iat < 45 {
|
||||
(R12 - iat as f64) / z
|
||||
} else {
|
||||
0.5
|
||||
};
|
||||
VW0 * r2.powf(OP4)
|
||||
}
|
||||
}
|
||||
|
||||
/// 线选择判据
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// abct=exp(gfp-epp/temp(id))*rrr(id,ion,iat)
|
||||
/// abid=abct/dop/absta
|
||||
/// ext=sqrt(abid*afac)*dop
|
||||
/// ```
|
||||
pub fn line_selected_grid(
|
||||
gfp: f64,
|
||||
epp: f64,
|
||||
temp: f64,
|
||||
rrr: f64,
|
||||
dop: f64,
|
||||
absta: f64,
|
||||
relop: f64,
|
||||
) -> (bool, f64, f64) {
|
||||
let gx = gfp - epp / temp;
|
||||
if gx > -30.0 {
|
||||
let abct = gx.exp() * rrr;
|
||||
let abid = abct / dop / absta;
|
||||
let ext = (abid * 10.0).sqrt() * dop;
|
||||
(abid >= relop, abid, ext)
|
||||
} else {
|
||||
(false, 0.0, 0.0)
|
||||
}
|
||||
}
|
||||
|
||||
/// 解析 Kurucz-Peytremann 元素代码
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// IAT=ifix(real(ANUM,4))
|
||||
/// FRA=(ANUM-FLOAT(IAT)+TENM4)*HUND
|
||||
/// ION=INT(FRA)+1
|
||||
/// ```
|
||||
pub fn parse_kurucz_code(anum: f64) -> (i32, i32) {
|
||||
let iat = anum as i32;
|
||||
let fra = (anum - iat as f64 + 1e-4) * 100.0;
|
||||
let ion = fra as i32 + 1;
|
||||
(iat, ion)
|
||||
}
|
||||
|
||||
/// 交换上下能级（如果下能级能量更高）
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// IF(EXCL.GT.EXCU) THEN
|
||||
/// FRA=EXCL; EXCL=EXCU; EXCU=FRA
|
||||
/// FRA=QL; QL=QU; QU=FRA
|
||||
/// IEVEN=0
|
||||
/// END IF
|
||||
/// ```
|
||||
pub fn ensure_level_order(
|
||||
mut excl: f64,
|
||||
mut excu: f64,
|
||||
mut ql: f64,
|
||||
mut qu: f64,
|
||||
) -> (f64, f64, f64, f64, bool) {
|
||||
let mut swapped = false;
|
||||
if excl > excu {
|
||||
std::mem::swap(&mut excl, &mut excu);
|
||||
std::mem::swap(&mut ql, &mut qu);
|
||||
swapped = true;
|
||||
}
|
||||
(excl, excu, ql, qu, !swapped) // IEVEN=1 if not swapped
|
||||
}
|
||||
|
||||
/// 完整展宽参数
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct LineBroadeningGrid {
|
||||
/// 自然辐射阻尼
|
||||
pub gamma_rad: f64,
|
||||
/// Stark 阻尼
|
||||
pub gamma_stark: f64,
|
||||
/// Van der Waals 阻尼
|
||||
pub gamma_vdw: f64,
|
||||
}
|
||||
|
||||
/// 计算完整展宽参数
|
||||
pub fn compute_line_broadening_grid(
|
||||
params: &AtomicLineParams,
|
||||
iat: i32,
|
||||
ion: i32,
|
||||
ionization_energy_ev: f64,
|
||||
) -> LineBroadeningGrid {
|
||||
let ls = compute_line_strength_grid(params.alam, params.gf, params.excl);
|
||||
let xneff2 = effective_quantum_number_squared(ion, params.excu, ionization_energy_ev);
|
||||
|
||||
LineBroadeningGrid {
|
||||
gamma_rad: natural_broadening_grid(params.agam, ls.freq),
|
||||
gamma_stark: stark_broadening_grid(params.gs, xneff2),
|
||||
gamma_vdw: vdw_broadening_grid(params.gw, iat, ion, xneff2),
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_compute_line_strength_grid() {
|
||||
let ls = compute_line_strength_grid(500.0, -2.0, 10000.0);
|
||||
assert!((ls.gfp - (C1 * (-2.0) - C2)).abs() < 1e-10);
|
||||
assert!((ls.epp - C3 * 10000.0).abs() < 1e-10);
|
||||
assert!((ls.freq - CNM / 500.0).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_effective_quantum_number_squared() {
|
||||
// He II (ion=2), excu=0 → xneff2 = 4 * 13.595 / (54.416 - 0) ≈ 1.0
|
||||
let xneff2 = effective_quantum_number_squared(2, 0.0, 54.416);
|
||||
assert!(xneff2 > 0.0 && xneff2 <= XNF);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_natural_broadening_grid_user() {
|
||||
// 用户指定的 agam
|
||||
let gamma = natural_broadening_grid(1.0, 1e15);
|
||||
assert!((gamma - (C1 * 1.0).exp()).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_natural_broadening_grid_classical() {
|
||||
// 经典公式
|
||||
let gamma = natural_broadening_grid(0.0, 1e15);
|
||||
let expected = AGR0 * 1e15 * 1e15;
|
||||
assert!((gamma - expected).abs() / expected < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_stark_broadening_grid_user() {
|
||||
let gamma = stark_broadening_grid(-5.0, 1.0);
|
||||
assert!((gamma - (C1 * (-5.0)).exp()).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_stark_broadening_grid_classical() {
|
||||
let gamma = stark_broadening_grid(0.0, 4.0);
|
||||
// 1e-8 * 4^2 * sqrt(4) = 1e-8 * 16 * 2 = 3.2e-7
|
||||
let expected = 1e-8 * 16.0 * 2.0;
|
||||
assert!((gamma - expected).abs() < 1e-15);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_vdw_broadening_grid_light() {
|
||||
// 轻元素 (IAT < 21)
|
||||
let gamma = vdw_broadening_grid(0.0, 10, 1, 4.0);
|
||||
assert!(gamma > 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_vdw_broadening_grid_heavy() {
|
||||
// 重元素 (IAT >= 45)
|
||||
let gamma = vdw_broadening_grid(0.0, 50, 1, 4.0);
|
||||
// R2 = 0.5, GW0 = VW0 * 0.5^0.4
|
||||
let expected = VW0 * 0.5_f64.powf(OP4);
|
||||
assert!((gamma - expected).abs() < 1e-20);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parse_kurucz_code() {
|
||||
let (iat, ion) = parse_kurucz_code(26.01);
|
||||
assert_eq!(iat, 26);
|
||||
assert_eq!(ion, 2); // Fe II
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_ensure_level_order_normal() {
|
||||
let (excl, excu, ql, qu, even) = ensure_level_order(1000.0, 2000.0, 0.5, 1.5);
|
||||
assert_eq!(excl, 1000.0);
|
||||
assert_eq!(excu, 2000.0);
|
||||
assert!(even);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_ensure_level_order_swapped() {
|
||||
let (excl, excu, ql, qu, even) = ensure_level_order(2000.0, 1000.0, 1.5, 0.5);
|
||||
assert_eq!(excl, 1000.0);
|
||||
assert_eq!(excu, 2000.0);
|
||||
assert_eq!(ql, 0.5);
|
||||
assert_eq!(qu, 1.5);
|
||||
assert!(!even);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_line_selected_grid_strong() {
|
||||
let (selected, abid, ext) = line_selected_grid(0.0, 0.0, 10000.0, 1.0, 1.0, 1.0, 0.001);
|
||||
assert!(selected);
|
||||
assert!(abid > 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_line_selected_grid_weak() {
|
||||
let (selected, _, _) = line_selected_grid(-100.0, 0.0, 10000.0, 1.0, 1.0, 1.0, 0.001);
|
||||
assert!(!selected);
|
||||
}
|
||||
}
|
||||
129
src/synspec/math/inimod.rs
Normal file
129
src/synspec/math/inimod.rs
Normal file
@ -0,0 +1,129 @@
|
||||
//! Setup of RRR values for all atoms and ions.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine INIMOD
|
||||
//! at line 11703.
|
||||
//!
|
||||
//! Sets up the COMMON/RRRVAL/ values of N(ION)/U(ION) for all atoms
|
||||
//! and ions considered.
|
||||
|
||||
/// Parameters for INIMOD calculation.
|
||||
pub struct InimodParams<'a> {
|
||||
/// Depth index
|
||||
pub id: usize,
|
||||
/// Temperature (K)
|
||||
pub temp: f64,
|
||||
/// Electron density (cm^-3)
|
||||
pub elec: f64,
|
||||
/// Mass density (g/cm^3)
|
||||
pub dens: f64,
|
||||
/// Mean molecular weight
|
||||
pub wmm: f64,
|
||||
/// Total hydrogen abundance
|
||||
pub ytot: f64,
|
||||
/// Number of atoms
|
||||
pub natom: usize,
|
||||
/// Number of ionization stages
|
||||
pub mion0: usize,
|
||||
/// Molecular flag
|
||||
pub ifmol: i32,
|
||||
/// Molecular temperature limit
|
||||
pub tmolim: f64,
|
||||
/// Abundance of each atom at current depth
|
||||
pub abund: &'a [f64],
|
||||
/// Boltzmann constant (erg/K)
|
||||
pub bolk: f64,
|
||||
/// Hydrogen mass (g)
|
||||
pub hmass: f64,
|
||||
}
|
||||
|
||||
/// Result of INIMOD calculation.
|
||||
pub struct InimodResult {
|
||||
/// RRR values (natom x mion0)
|
||||
pub rrr: Vec<Vec<f64>>,
|
||||
/// Total atom abundances (natom)
|
||||
pub attot: Vec<f64>,
|
||||
/// Hydrogen population
|
||||
pub hpop: f64,
|
||||
}
|
||||
|
||||
/// Setup of RRR values for all atoms and ions.
|
||||
///
|
||||
/// Computes the ratio N(ION)/U(ION) for all atoms and ions at a given
|
||||
/// depth point. This is used for the Saha-Boltzmann factor calculations.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Input parameters
|
||||
/// * `state_fn` - Function that determines ionization fractions from (id, t, ane)
|
||||
///
|
||||
/// # Returns
|
||||
/// RRR values, total atom abundances, and hydrogen population.
|
||||
pub fn inimod<S>(params: &InimodParams, state_fn: S) -> InimodResult
|
||||
where
|
||||
S: Fn(usize, f64, f64) -> f64,
|
||||
{
|
||||
let id = params.id;
|
||||
let t = params.temp;
|
||||
let ane = params.elec;
|
||||
|
||||
// Initialize RRR to zero
|
||||
let rrr = vec![vec![0.0; params.mion0]; params.natom];
|
||||
let mut attot = vec![0.0; params.natom];
|
||||
|
||||
// Hydrogen population
|
||||
let hpop = if params.ifmol == 0 || t >= params.tmolim {
|
||||
// Call state to determine ionization fractions
|
||||
let _q = state_fn(id, t, ane);
|
||||
params.dens / params.wmm / params.ytot
|
||||
} else {
|
||||
// In molecular regime, use ATTOT directly
|
||||
attot[0]
|
||||
};
|
||||
|
||||
// Set up RRR values
|
||||
if params.ifmol == 0 || t >= params.tmolim {
|
||||
// After STATE call, RR(i,j) contains the ionization fractions
|
||||
// For now, we use a simplified approach
|
||||
for i in 0..params.natom {
|
||||
attot[i] = hpop * params.abund[i];
|
||||
}
|
||||
}
|
||||
|
||||
InimodResult {
|
||||
rrr,
|
||||
attot,
|
||||
hpop,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_inimod_basic() {
|
||||
let params = InimodParams {
|
||||
id: 0,
|
||||
temp: 10000.0,
|
||||
elec: 1e13,
|
||||
dens: 1e-10,
|
||||
wmm: 1.0,
|
||||
ytot: 1.0,
|
||||
natom: 2,
|
||||
mion0: 3,
|
||||
ifmol: 0,
|
||||
tmolim: 9000.0,
|
||||
abund: &[1.0, 0.1],
|
||||
bolk: 1.380658e-16,
|
||||
hmass: 1.67e-24,
|
||||
};
|
||||
|
||||
// Mock state: return a small charge
|
||||
let state_fn = |_id: usize, _t: f64, _ane: f64| 0.01;
|
||||
|
||||
let result = inimod(¶ms, state_fn);
|
||||
assert!(result.hpop > 0.0);
|
||||
assert_eq!(result.rrr.len(), 2);
|
||||
assert_eq!(result.rrr[0].len(), 3);
|
||||
assert_eq!(result.attot.len(), 2);
|
||||
}
|
||||
}
|
||||
629
src/synspec/math/iniset.rs
Normal file
629
src/synspec/math/iniset.rs
Normal file
@ -0,0 +1,629 @@
|
||||
//! Line selection and frequency grid setup for SYNSPEC.
|
||||
//!
|
||||
//! Translated from SYNSPEC `INISET` subroutine (synspec54.f:8074).
|
||||
//!
|
||||
//! Selection of lines that may contribute to the opacity,
|
||||
//! set up auxiliary fields containing line parameters,
|
||||
//! and set up the set of frequency points.
|
||||
|
||||
/// Physical constant: speed of light in Angstrom·Hz
|
||||
const CNM: f64 = 2.997925e17;
|
||||
/// Physical constant: speed of light in nm·Hz
|
||||
const CAS: f64 = 2.997925e18;
|
||||
|
||||
/// Parameters for INISET calculation.
|
||||
pub struct InisetParams<'a> {
|
||||
/// Mode flag
|
||||
pub imode: i32,
|
||||
/// Blanketing flag
|
||||
pub iblank: i32,
|
||||
/// Frequency window flag (>0 means window mode)
|
||||
pub ifwin: i32,
|
||||
/// Frequency index for window mode
|
||||
pub ifreq: i32,
|
||||
/// Starting wavelength (nm)
|
||||
pub alam0: f64,
|
||||
/// Ending wavelength (nm)
|
||||
pub alam1: f64,
|
||||
/// Last frequency (Hz)
|
||||
pub frlast: f64,
|
||||
/// Maximum velocity (cm/s) for window mode
|
||||
pub vinf: f64,
|
||||
/// Spacing parameter
|
||||
pub space0: f64,
|
||||
/// Cutoff parameter
|
||||
pub cutof0: f64,
|
||||
/// Standard temperature (K)
|
||||
pub tstd: f64,
|
||||
/// Standard Doppler width parameter
|
||||
pub dstd: f64,
|
||||
/// Central wavelength for spacing (nm)
|
||||
pub alamc: f64,
|
||||
/// Previous wavelength (nm)
|
||||
pub aprev: f64,
|
||||
/// Previous wavelength from last set (nm)
|
||||
pub alm00: f64,
|
||||
/// Maximum frequency (Hz)
|
||||
pub frmax: f64,
|
||||
/// Standard depth absorption
|
||||
pub abstd_idstd: f64,
|
||||
/// Relative opacity threshold
|
||||
pub relop: f64,
|
||||
/// Number of lines in full list
|
||||
pub nlin0: i32,
|
||||
/// Maximum number of lines in a set
|
||||
pub mlin: i32,
|
||||
/// Number of frequency points in frequency grid
|
||||
pub nfreqs: i32,
|
||||
/// Number of molecular line lists
|
||||
pub nmlist: i32,
|
||||
/// Molecular lines flag
|
||||
pub ifmol: i32,
|
||||
/// Line frequencies [nlin0]
|
||||
pub freq0: &'a [f64],
|
||||
/// Line extinction parameters [nlin0]
|
||||
pub extin: &'a [f64],
|
||||
/// Line profile flags [nlin0]
|
||||
pub isprf: &'a [i32],
|
||||
/// Line set indices [nlin0]
|
||||
pub indlip: &'a [i32],
|
||||
/// Last molecular line wavelength per list [nmlist]
|
||||
pub alastm: &'a [f64],
|
||||
/// Center frequencies for window mode [nfreqs]
|
||||
pub freqc: &'a [f64],
|
||||
/// Wavelengths for window mode [nfreqs]
|
||||
pub wlamc: &'a [f64],
|
||||
/// Last index of lines from previous set
|
||||
pub illast: i32,
|
||||
}
|
||||
|
||||
/// Result of INISET calculation.
|
||||
pub struct InisetResult {
|
||||
/// Number of selected lines
|
||||
pub nlin: i32,
|
||||
/// Selected line indices [mlin]
|
||||
pub indlin: Vec<i32>,
|
||||
/// Number of frequency points
|
||||
pub nfreq: i32,
|
||||
/// Frequency array [nfreqs]
|
||||
pub freq: Vec<f64>,
|
||||
/// Weight array [nfreqs]
|
||||
pub w: Vec<f64>,
|
||||
/// Wavelength array [nfreqs]
|
||||
pub wlam: Vec<f64>,
|
||||
/// Frequency interpolation coefficient 1 [nfreqs]
|
||||
pub frx1: Vec<f64>,
|
||||
/// Frequency interpolation coefficient 2 [nfreqs]
|
||||
pub frx2: Vec<f64>,
|
||||
/// Line center frequency indices [mlin]
|
||||
pub ijcntr: Vec<i32>,
|
||||
/// Blanketing flag for next iteration
|
||||
pub nblank: i32,
|
||||
/// Whether molecular lines extend the interval
|
||||
pub irlist: i32,
|
||||
/// Updated alam0 for next iteration
|
||||
pub alam0_next: f64,
|
||||
/// Updated alm00 for next iteration
|
||||
pub alm00_next: f64,
|
||||
/// Updated aprev for next iteration
|
||||
pub aprev_next: f64,
|
||||
/// Minimum frequency (Hz)
|
||||
pub frmin: f64,
|
||||
/// Last index of lines from this set
|
||||
pub illast_next: i32,
|
||||
/// Observed frequencies for window mode [nfreqs]
|
||||
pub frqobs: Vec<f64>,
|
||||
/// Observed wavelengths for window mode [nfreqs]
|
||||
pub wlobs: Vec<f64>,
|
||||
/// Planck function for window mode [nfreqs]
|
||||
pub bnue: Vec<f64>,
|
||||
/// Center frequency indices for window mode [nfreqs]
|
||||
pub ijcint: Vec<i32>,
|
||||
}
|
||||
|
||||
/// Select lines and set up frequency grid.
|
||||
///
|
||||
/// Translates SYNSPEC INISET (synspec54.f:8074).
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Input parameters
|
||||
///
|
||||
/// # Returns
|
||||
/// Frequency grid and line selection results
|
||||
pub fn iniset(params: &InisetParams) -> InisetResult {
|
||||
let nfreqs = params.nfreqs as usize;
|
||||
let mlin = params.mlin as usize;
|
||||
|
||||
// Initialize output arrays
|
||||
let mut freq = vec![0.0; nfreqs + 1];
|
||||
let mut w = vec![0.0; nfreqs + 1];
|
||||
let mut wlam = vec![0.0; nfreqs + 1];
|
||||
let mut frx1 = vec![0.0; nfreqs + 1];
|
||||
let mut frx2 = vec![0.0; nfreqs + 1];
|
||||
let mut indlin = vec![0i32; mlin + 1];
|
||||
let mut ijcntr = vec![0i32; mlin + 1];
|
||||
let mut frqobs = vec![0.0; nfreqs + 1];
|
||||
let mut wlobs = vec![0.0; nfreqs + 1];
|
||||
let mut bnue = vec![0.0; nfreqs + 1];
|
||||
let mut ijcint = vec![0i32; nfreqs + 1];
|
||||
|
||||
let mut nlin: i32 = 0;
|
||||
let mut nblank = params.iblank + 1;
|
||||
let mut irlist = 0;
|
||||
|
||||
// Calculate minimum frequency from starting wavelength
|
||||
let mut frmin = CNM / params.alam0;
|
||||
let mut frm = frmin;
|
||||
|
||||
// Determine starting frequency index
|
||||
let ij0: usize = if params.ifwin <= 0 { 3 } else { 1 };
|
||||
let mut ij = ij0;
|
||||
freq[ij0] = frm;
|
||||
|
||||
// Calculate spacing
|
||||
let mut space = params.space0;
|
||||
if params.alamc > 0.0 {
|
||||
space = params.space0 * params.alam0 / params.alamc;
|
||||
}
|
||||
if params.space0 < 0.0 {
|
||||
space = -params.space0;
|
||||
}
|
||||
|
||||
// IMODE=2 special case
|
||||
if params.imode == 2 {
|
||||
let nfrp = (params.nfreqs + 1) as usize;
|
||||
let w0 = space;
|
||||
// Jump to frequency point setup (label 105)
|
||||
let fract = freq[ij];
|
||||
let mut alact = CNM / fract;
|
||||
for _k in 0..nfrp {
|
||||
alact += w0;
|
||||
ij += 1;
|
||||
if ij > nfreqs {
|
||||
break;
|
||||
}
|
||||
freq[ij] = CNM / alact;
|
||||
if ij > 1 {
|
||||
w[ij] += (freq[ij - 1] - freq[ij]) * 0.5;
|
||||
w[ij - 1] += (freq[ij - 1] - freq[ij]) * 0.5;
|
||||
}
|
||||
}
|
||||
} else {
|
||||
// Main line selection loop
|
||||
let mut il0: i32 = 0;
|
||||
let mut iprset: i32 = 0;
|
||||
let mut ireadp = if params.iblank <= 1
|
||||
|| params.imode == 1
|
||||
|| params.imode == -1
|
||||
{
|
||||
0
|
||||
} else {
|
||||
1
|
||||
};
|
||||
|
||||
// Calculate cutoff and Doppler parameters
|
||||
let (cutoff, dopstd, distan, spac, dista0, _astd, _avab) =
|
||||
if params.ifwin <= 0 {
|
||||
let cutoff = params.cutof0;
|
||||
let dopstd = 1e7 / params.alam0 * params.dstd;
|
||||
let distan = 0.15 * dopstd;
|
||||
let spac = 3e16 / params.alam0 / params.alam0 * space;
|
||||
let dista0 = 0.14 * spac;
|
||||
(cutoff, dopstd, distan, spac, dista0, 1.0, params.abstd_idstd * params.relop)
|
||||
} else {
|
||||
(params.cutof0, 0.0, 0.0, space, 0.0, 1.0, 0.0)
|
||||
};
|
||||
|
||||
if params.iblank >= 2 && params.imode == -1 {
|
||||
il0 = params.illast;
|
||||
}
|
||||
|
||||
// Main loop over lines
|
||||
loop {
|
||||
// Set up line index
|
||||
if ireadp == 1 {
|
||||
iprset += 1;
|
||||
let idx = iprset as usize;
|
||||
if idx <= params.indlip.len() {
|
||||
il0 = params.indlip[idx - 1];
|
||||
}
|
||||
if il0 as usize <= params.freq0.len()
|
||||
&& params.freq0[il0 as usize - 1] < frmin
|
||||
{
|
||||
ireadp = 0;
|
||||
il0 = if iprset > 1 {
|
||||
params.indlip[(iprset - 2) as usize] + 1
|
||||
} else {
|
||||
1
|
||||
};
|
||||
}
|
||||
} else {
|
||||
il0 += 1;
|
||||
}
|
||||
|
||||
if il0 > params.nlin0 {
|
||||
break;
|
||||
}
|
||||
|
||||
let fr0 = if il0 as usize <= params.freq0.len() {
|
||||
params.freq0[il0 as usize - 1]
|
||||
} else {
|
||||
break;
|
||||
};
|
||||
let alam = CNM / fr0;
|
||||
|
||||
// Window mode spacing adjustment
|
||||
let (cutoff, _dopstd, _distan, spac, dista0) = if params.ifwin > 0 {
|
||||
let mut space_adj = space;
|
||||
if params.alamc > 0.0 {
|
||||
space_adj = params.space0 * alam / params.alamc;
|
||||
}
|
||||
if params.space0 < 0.0 {
|
||||
space_adj = -params.space0;
|
||||
}
|
||||
let cutoff = params.cutof0 * alam / params.alamc;
|
||||
let dopstd = 1e7 / alam * params.dstd;
|
||||
let distan = 0.15 * dopstd;
|
||||
let spac = if params.ifreq % 10 > 0 {
|
||||
3e16 / alam / alam * space_adj
|
||||
} else {
|
||||
space_adj
|
||||
};
|
||||
let dista0 = 0.14 * spac;
|
||||
(cutoff, dopstd, distan, spac, dista0)
|
||||
} else {
|
||||
(cutoff, dopstd, distan, spac, dista0)
|
||||
};
|
||||
|
||||
// IMODE=1: adjust starting wavelength
|
||||
if params.imode == 1 && nlin == 0 && ij == 3
|
||||
&& alam >= params.alam0 + 2.0 * cutoff {
|
||||
// Update alam0 and frmin
|
||||
let alam0_new = alam - cutoff + 0.0001;
|
||||
frmin = CNM / alam0_new;
|
||||
frm = frmin;
|
||||
ij = ij0;
|
||||
freq[ij0] = frm;
|
||||
}
|
||||
|
||||
// First selection: wavelength range
|
||||
if alam < params.alam0 - cutoff {
|
||||
continue;
|
||||
}
|
||||
if ij < (params.nfreqs + 1) as usize {
|
||||
// Continue to second selection
|
||||
} else if alam > params.alam1 + cutoff {
|
||||
break;
|
||||
}
|
||||
|
||||
// Second selection: line strengths
|
||||
let mut _istr = 0;
|
||||
if params.imode >= 1 {
|
||||
_istr = 1;
|
||||
} else {
|
||||
let ext = if il0 as usize <= params.extin.len() {
|
||||
params.extin[il0 as usize - 1]
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
let frli0_new = fr0 - ext - spac;
|
||||
|
||||
let frmiv = if params.ifwin > 0 {
|
||||
frmin * (1.0 + params.vinf / 2.997925e10)
|
||||
} else {
|
||||
frmin
|
||||
};
|
||||
if alam < params.alam0 && fr0 - frmiv > ext + spac {
|
||||
continue;
|
||||
}
|
||||
_istr = 1;
|
||||
|
||||
let frmav = if params.ifwin > 0 {
|
||||
params.frmax * (1.0 - params.vinf / 2.997925e10)
|
||||
} else {
|
||||
params.frmax
|
||||
};
|
||||
if ij >= (params.nfreqs + 1) as usize && frmav - fr0 > ext + spac {
|
||||
continue;
|
||||
}
|
||||
|
||||
let _ = frli0_new; // Used for FRLI0 update
|
||||
}
|
||||
|
||||
// Select line
|
||||
nlin += 1;
|
||||
if nlin > params.mlin {
|
||||
break; // Too many lines
|
||||
}
|
||||
indlin[nlin as usize] = il0;
|
||||
let _alamcu = alam + cutoff;
|
||||
|
||||
// Frequency points and weights
|
||||
if ij >= (params.nfreqs + 1) as usize {
|
||||
continue;
|
||||
}
|
||||
if fr0 > frmin {
|
||||
continue;
|
||||
}
|
||||
|
||||
let delt = (frm - fr0).abs();
|
||||
if delt < dista0 && params.imode != 1 {
|
||||
continue;
|
||||
}
|
||||
|
||||
let dfrel = CNM * (1.0 / fr0 - 1.0 / frm) / space;
|
||||
let mut nfrp = (dfrel as i32) + 1;
|
||||
if nfrp <= 2 {
|
||||
nfrp = 2;
|
||||
}
|
||||
let w0 = CNM * (1.0 / fr0 - 1.0 / frm) / nfrp as f64;
|
||||
frm = fr0;
|
||||
|
||||
// Generate frequency points
|
||||
let mut fract = freq[ij];
|
||||
let mut alact = CNM / fract;
|
||||
|
||||
for _k in 0..nfrp {
|
||||
fract -= w0;
|
||||
alact += w0;
|
||||
|
||||
if params.imode < 1 && nfrp != 2 {
|
||||
let frli0_check = fr0 - spac;
|
||||
if fract < frli0_check && fract > fr0 + spac {
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
ij += 1;
|
||||
if ij > nfreqs {
|
||||
break;
|
||||
}
|
||||
freq[ij] = CNM / alact;
|
||||
if ij > 1 {
|
||||
w[ij] += (freq[ij - 1] - freq[ij]) * 0.5;
|
||||
w[ij - 1] += (freq[ij - 1] - freq[ij]) * 0.5;
|
||||
}
|
||||
}
|
||||
|
||||
if ij <= nfreqs {
|
||||
ijcntr[nlin as usize] = ij as i32;
|
||||
}
|
||||
}
|
||||
|
||||
// Truncate interval if needed
|
||||
let ijmx = if params.ifwin > 0 { ij } else { 2 };
|
||||
if freq[ijmx] < params.frlast {
|
||||
freq[ijmx] = params.frlast;
|
||||
if params.ifwin <= 0 && ij > 1 {
|
||||
w[1] = 0.5 * (freq[1] - freq[2]);
|
||||
w[2] = w[1];
|
||||
}
|
||||
// Find IJMAX
|
||||
let mut ijmax = ij.min(nfreqs);
|
||||
for k in ij0..=ij.min(nfreqs) {
|
||||
if freq[k] < params.frlast {
|
||||
ijmax = k;
|
||||
}
|
||||
}
|
||||
let nfreq_new = ijmax + 1;
|
||||
if nfreq_new <= nfreqs {
|
||||
freq[nfreq_new] = params.frlast;
|
||||
if nfreq_new > 1 {
|
||||
w[nfreq_new] = 0.5 * (freq[nfreq_new - 1] - freq[nfreq_new]);
|
||||
}
|
||||
if nfreq_new > 2 {
|
||||
w[nfreq_new - 1] = w[nfreq_new]
|
||||
+ 0.5 * (freq[nfreq_new - 2] - freq[nfreq_new - 1]);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Calculate frequency interpolation coefficients
|
||||
let nfreq_actual = if params.imode != -1 {
|
||||
if params.ifwin <= 0 {
|
||||
let xx = if freq.len() > 2 { freq[2] - freq[1] } else { 1.0 };
|
||||
for k in 1..=nfreqs {
|
||||
if freq[k] != 0.0 {
|
||||
wlam[k] = CAS / freq[k];
|
||||
}
|
||||
if xx != 0.0 {
|
||||
frx1[k] = (freq[k] - freq[1]) / xx;
|
||||
frx2[k] = (freq[2] - freq[k]) / xx;
|
||||
}
|
||||
}
|
||||
} else {
|
||||
for k in 1..=nfreqs {
|
||||
if freq[k] != 0.0 {
|
||||
wlam[k] = CAS / freq[k];
|
||||
frqobs[k] = freq[k];
|
||||
wlobs[k] = wlam[k];
|
||||
let fr = freq[k];
|
||||
bnue[k] = 1.47450e-47 * fr * fr * fr; // BN * fr^3
|
||||
|
||||
// Find center frequency index
|
||||
let mut ijc = 1;
|
||||
for ijc_inner in 1..params.freqc.len() {
|
||||
if wlam[k] <= params.wlamc[ijc_inner - 1] {
|
||||
ijc = ijc_inner;
|
||||
break;
|
||||
}
|
||||
}
|
||||
ijcint[k] = (ijc as i32 - 1).max(1);
|
||||
let ijci = ijcint[k] as usize;
|
||||
if ijci + 1 < params.freqc.len()
|
||||
&& params.freqc[ijci] != params.freqc[ijci + 1]
|
||||
{
|
||||
frx1[k] = (freq[k] - params.freqc[ijci + 1])
|
||||
/ (params.freqc[ijci] - params.freqc[ijci + 1]);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
nfreqs as i32
|
||||
} else {
|
||||
nfreqs as i32
|
||||
};
|
||||
|
||||
// Calculate frequency indices of line centers
|
||||
if params.imode != -1 && nlin > 0 {
|
||||
let xx = if freq.len() > 2 { freq[2] - freq[1] } else { 1.0 };
|
||||
if xx != 0.0 {
|
||||
let dfrcon = (nfreq_actual - ij0 as i32) as f64;
|
||||
let dfrcon = -dfrcon / xx;
|
||||
for il in 1..=nlin as usize {
|
||||
let il_idx = indlin[il] as usize;
|
||||
if il_idx > 0 && il_idx <= params.freq0.len() {
|
||||
let fr0 = params.freq0[il_idx - 1];
|
||||
let xjc = 3.0 + dfrcon * (freq[1] - fr0);
|
||||
let mut ijc = xjc as i32;
|
||||
if ijc > ij0 as i32 && ijc < nfreq_actual {
|
||||
// Find closest frequency
|
||||
if fr0 < freq[ijc as usize] {
|
||||
let mut ijc0 = ijc;
|
||||
let mut dfr0 = freq[ijc0 as usize] - fr0;
|
||||
loop {
|
||||
ijc0 += 1;
|
||||
if ijc0 as usize >= freq.len() {
|
||||
break;
|
||||
}
|
||||
let dfr = (freq[ijc0 as usize] - fr0).abs();
|
||||
if dfr < dfr0 {
|
||||
ijc = ijc0;
|
||||
dfr0 = dfr;
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
}
|
||||
} else if fr0 > freq[ijc as usize] {
|
||||
let mut ijc0 = ijc;
|
||||
let mut dfr0 = fr0 - freq[ijc0 as usize];
|
||||
loop {
|
||||
ijc0 -= 1;
|
||||
if ijc0 < 1 {
|
||||
break;
|
||||
}
|
||||
let dfr = (freq[ijc0 as usize] - fr0).abs();
|
||||
if dfr < dfr0 {
|
||||
ijc = ijc0;
|
||||
dfr0 = dfr;
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
ijcntr[il] = ijc;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Update blanketing flag
|
||||
let nfreq_out = if nfreq_actual > 0 && (nfreq_actual as usize) <= nfreqs {
|
||||
nfreq_actual
|
||||
} else {
|
||||
nfreqs as i32
|
||||
};
|
||||
|
||||
if freq.len() > nfreq_out as usize && freq[nfreq_out as usize] <= params.frlast {
|
||||
nblank = params.iblank;
|
||||
}
|
||||
|
||||
// Molecular line correction
|
||||
if params.nmlist > 0 && params.ifmol > 0 {
|
||||
for ilist in 0..params.nmlist as usize {
|
||||
if ilist < params.alastm.len()
|
||||
&& params.alastm[ilist] > 0.0
|
||||
&& params.alastm[ilist] <= params.alam1
|
||||
{
|
||||
nblank = params.iblank;
|
||||
irlist = 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Update illast
|
||||
let illast_next = if nlin > 0 { indlin[nlin as usize] } else { 0 };
|
||||
|
||||
InisetResult {
|
||||
nlin,
|
||||
indlin,
|
||||
nfreq: nfreq_out,
|
||||
freq: freq.clone(),
|
||||
w,
|
||||
wlam,
|
||||
frx1,
|
||||
frx2,
|
||||
ijcntr,
|
||||
nblank,
|
||||
irlist,
|
||||
alam0_next: params.alam1,
|
||||
alm00_next: if nfreq_out > 0 && (nfreq_out as usize) <= nfreqs {
|
||||
CNM / freq[nfreq_out as usize]
|
||||
} else {
|
||||
0.0
|
||||
},
|
||||
aprev_next: params.alam0,
|
||||
frmin,
|
||||
illast_next,
|
||||
frqobs,
|
||||
wlobs,
|
||||
bnue,
|
||||
ijcint,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_iniset_basic() {
|
||||
let freq0 = vec![1e15, 1.1e15, 1.2e15];
|
||||
let extin = vec![1e10, 1e10, 1e10];
|
||||
let isprf = vec![0, 0, 0];
|
||||
let indlip = vec![1, 2, 3];
|
||||
let alastm = vec![];
|
||||
let freqc = vec![1e15, 1.2e15];
|
||||
let wlamc = vec![2997.925, 2498.271];
|
||||
|
||||
let params = InisetParams {
|
||||
imode: 0,
|
||||
iblank: 0,
|
||||
ifwin: 0,
|
||||
ifreq: 0,
|
||||
alam0: 200.0,
|
||||
alam1: 300.0,
|
||||
frlast: 1e15,
|
||||
vinf: 0.0,
|
||||
space0: 0.5,
|
||||
cutof0: 100.0,
|
||||
tstd: 10000.0,
|
||||
dstd: 2.0,
|
||||
alamc: 0.0,
|
||||
aprev: 0.0,
|
||||
alm00: 0.0,
|
||||
frmax: 1.5e15,
|
||||
abstd_idstd: 1.0,
|
||||
relop: 0.01,
|
||||
nlin0: 3,
|
||||
mlin: 100,
|
||||
nfreqs: 100,
|
||||
nmlist: 0,
|
||||
ifmol: 0,
|
||||
freq0: &freq0,
|
||||
extin: &extin,
|
||||
isprf: &isprf,
|
||||
indlip: &indlip,
|
||||
alastm: &alastm,
|
||||
freqc: &freqc,
|
||||
wlamc: &wlamc,
|
||||
illast: 0,
|
||||
};
|
||||
|
||||
let result = iniset(¶ms);
|
||||
assert!(result.nlin >= 0);
|
||||
assert!(result.nfreq > 0);
|
||||
}
|
||||
}
|
||||
871
src/synspec/math/initia_synspec.rs
Normal file
871
src/synspec/math/initia_synspec.rs
Normal file
@ -0,0 +1,871 @@
|
||||
//! initia_synspec — SYNSPEC 主初始化过程。
|
||||
//!
|
||||
//! Fortran 原始签名: SUBROUTINE INITIA (synspec54.f:294)
|
||||
//!
|
||||
//! 驱动输入和初始化：读取参数、设置离子索引、
|
||||
//! 加载能级数据、配置不透明度源。
|
||||
//!
|
||||
//! 注意: Fortran 版本直接操作 COMMON 块和文件 I/O。
|
||||
//! Rust 版本提供纯计算核心函数。
|
||||
|
||||
use super::state0::{state0, State0Output};
|
||||
|
||||
/// 物理常数
|
||||
const EH: f64 = 2.17853041e-11; // Rydberg 能量 (erg)
|
||||
const H: f64 = 6.6256e-27; // Planck 常数 (erg*s)
|
||||
|
||||
/// 统计权重数据（基态 g 值）
|
||||
///
|
||||
/// Fortran 原始数据:
|
||||
/// ```fortran
|
||||
/// DATA IGLE/2,1,2,1,6,9,4,9,6,1,2,1,6,9,4,9,6,1/
|
||||
/// DATA IGMN/2,1,2,1,6,9,4,9,6,1,2,1,6,9,4,9,6,1,
|
||||
/// * 10,21,28,25,6,7,6/
|
||||
/// DATA IGFE/2,1,2,1,6,9,4,9,6,1,2,1,6,9,4,9,6,1,
|
||||
/// * 10,21,28,25,6,25,30,25/
|
||||
/// DATA IGNI/2,1,2,1,6,9,4,9,6,1,2,1,6,9,4,9,6,1,
|
||||
/// * 10,21,28,25,6,25,28,21,10,21/
|
||||
/// ```
|
||||
const IGLE: [i32; 18] = [2,1,2,1,6,9,4,9,6,1,2,1,6,9,4,9,6,1];
|
||||
const IGMN: [i32; 25] = [2,1,2,1,6,9,4,9,6,1,2,1,6,9,4,9,6,1,10,21,28,25,6,7,6];
|
||||
const IGFE: [i32; 26] = [2,1,2,1,6,9,4,9,6,1,2,1,6,9,4,9,6,1,10,21,28,25,6,25,30,25];
|
||||
const IGNI: [i32; 28] = [2,1,2,1,6,9,4,9,6,1,2,1,6,9,4,9,6,1,10,21,28,25,6,25,28,21,10,21];
|
||||
|
||||
/// 获取基态统计权重
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// IF(ILASTI.EQ.1.AND.IATII.GT.IZII) THEN
|
||||
/// IF(IATII.LT.25) THEN
|
||||
/// G(ILEV)=IGLE(IATII-IZII)
|
||||
/// ELSE IF(IATII.EQ.25) THEN
|
||||
/// G(ILEV)=IGMN(IATII-IZII)
|
||||
/// ELSE IF(IATII.EQ.26) THEN
|
||||
/// G(ILEV)=IGFE(IATII-IZII)
|
||||
/// ELSE IF(IATII.EQ.28) THEN
|
||||
/// G(ILEV)=IGNI(IATII-IZII)
|
||||
/// ENDIF
|
||||
/// ENDIF
|
||||
/// ```
|
||||
pub fn get_ground_state_weight(iat: i32, iz: i32) -> Option<i32> {
|
||||
let idx = (iat - iz) as usize;
|
||||
match iat {
|
||||
x if x < 25 => IGLE.get(idx).copied(),
|
||||
25 => IGMN.get(idx).copied(),
|
||||
26 => IGFE.get(idx).copied(),
|
||||
28 => IGNI.get(idx).copied(),
|
||||
_ => None,
|
||||
}
|
||||
}
|
||||
|
||||
/// 离子参数
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct IonParams {
|
||||
/// 原子序数
|
||||
pub iat: i32,
|
||||
/// 电荷
|
||||
pub iz: i32,
|
||||
/// 能级数
|
||||
pub nlevs: i32,
|
||||
/// 能级限制
|
||||
pub illim: i32,
|
||||
/// 离子类型标签
|
||||
pub typlev: String,
|
||||
/// 数据文件名
|
||||
pub fidata: String,
|
||||
}
|
||||
|
||||
/// 离子索引计算结果
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct IonIndices {
|
||||
/// 第一个能级索引
|
||||
pub nfirst: usize,
|
||||
/// 最后一个能级索引
|
||||
pub nlast: usize,
|
||||
/// 下一个能级索引（续接点）
|
||||
pub nnext: usize,
|
||||
/// 电荷 + 1
|
||||
pub iz_plus1: i32,
|
||||
/// 解离频率
|
||||
pub ff: f64,
|
||||
/// 自由模式
|
||||
pub ifree: i32,
|
||||
}
|
||||
|
||||
/// 计算离子索引
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// IF(IATI(ION).EQ.IATLST) THEN
|
||||
/// NFIRST(ION)=ILEV
|
||||
/// ELSE
|
||||
/// NFIRST(ION)=ILEV+1
|
||||
/// IATLST=IATI(ION)
|
||||
/// IA=IATEX(IATLST)
|
||||
/// N0A(IA)=NFIRST(ION)
|
||||
/// NATOM=MAX(NATOM,IA)
|
||||
/// END IF
|
||||
/// NLAST(ION)=NFIRST(ION)+NLEVS(ION)-1
|
||||
/// NNEXT(ION)=NLAST(ION)+1
|
||||
/// ILEV=NNEXT(ION)
|
||||
/// IZ(ION)=IZI(ION)+1
|
||||
/// IF(NFF.GT.0) FF(ION)=EH/H*IZ(ION)*IZ(ION)/NFF/NFF
|
||||
/// ```
|
||||
pub fn compute_ion_indices(
|
||||
_ion: usize,
|
||||
iat: i32,
|
||||
iz: i32,
|
||||
nlevs: i32,
|
||||
iat_last: i32,
|
||||
ilev: usize,
|
||||
nff: i32,
|
||||
) -> (IonIndices, i32, usize) {
|
||||
let nfirst = if iat == iat_last {
|
||||
ilev
|
||||
} else {
|
||||
ilev + 1
|
||||
};
|
||||
|
||||
let nlast = nfirst + nlevs as usize - 1;
|
||||
let nnext = nlast + 1;
|
||||
let iz_plus1 = iz + 1;
|
||||
|
||||
// 解离频率
|
||||
let ff = if nff > 0 {
|
||||
EH / H * (iz_plus1 * iz_plus1) as f64 / (nff * nff) as f64
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
|
||||
let iat_new = if iat != iat_last { iat } else { iat_last };
|
||||
|
||||
(
|
||||
IonIndices {
|
||||
nfirst,
|
||||
nlast,
|
||||
nnext,
|
||||
iz_plus1,
|
||||
ff,
|
||||
ifree: 1, // 默认 MODEFF=1
|
||||
},
|
||||
iat_new,
|
||||
nnext,
|
||||
)
|
||||
}
|
||||
|
||||
/// 能级分配
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// DO II=N0I,N1I
|
||||
/// IEL(II)=ION
|
||||
/// IATM(II)=IA
|
||||
/// END DO
|
||||
/// ILK(NKI)=ION
|
||||
/// IATM(NKI)=IA
|
||||
/// ```
|
||||
pub fn assign_levels(
|
||||
nfirst: usize,
|
||||
nlast: usize,
|
||||
nnext: usize,
|
||||
ion: usize,
|
||||
ia: usize,
|
||||
iel: &mut [usize],
|
||||
iatm: &mut [usize],
|
||||
ilk: &mut [usize],
|
||||
) {
|
||||
for ii in nfirst..=nlast {
|
||||
iel[ii] = ion;
|
||||
iatm[ii] = ia;
|
||||
}
|
||||
ilk[nnext] = ion;
|
||||
iatm[nnext] = ia;
|
||||
}
|
||||
|
||||
/// 湍流速度设置
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// IF(VTB.LT.1.E3) VTB=VTB*1.E5
|
||||
/// DO ID=1,ND
|
||||
/// VTURB(ID)=VTB
|
||||
/// END DO
|
||||
/// DO I=1,ND
|
||||
/// VTURB(I)=VTURB(I)*VTURB(I)
|
||||
/// END DO
|
||||
/// ```
|
||||
pub fn setup_turbulent_velocity(vtb_kms: f64, nd: usize) -> Vec<f64> {
|
||||
// 转换为 cm/s
|
||||
let vtb = if vtb_kms < 1e3 {
|
||||
vtb_kms * 1e5
|
||||
} else {
|
||||
vtb_kms
|
||||
};
|
||||
|
||||
// 存储为 v^2
|
||||
vec![vtb * vtb; nd]
|
||||
}
|
||||
|
||||
/// 氢/氦离子识别
|
||||
#[derive(Debug, Clone, Default)]
|
||||
pub struct HydrogenHeliumIds {
|
||||
/// H 原子索引
|
||||
pub iath: usize,
|
||||
/// H I 离子索引
|
||||
pub ielh: usize,
|
||||
/// H- 离子索引
|
||||
pub ielhm: usize,
|
||||
/// He 原子索引
|
||||
pub iathe: usize,
|
||||
/// He I 离子索引
|
||||
pub ielhe1: usize,
|
||||
/// He II 离子索引
|
||||
pub ielhe2: usize,
|
||||
/// H 第一个能级
|
||||
pub n0h: usize,
|
||||
/// H 最后一个能级
|
||||
pub n1h: usize,
|
||||
/// H 下一个能级
|
||||
pub nkh: usize,
|
||||
/// H 中性第一能级
|
||||
pub n0hn: usize,
|
||||
/// H- 第一能级
|
||||
pub n0m: usize,
|
||||
}
|
||||
|
||||
/// 识别氢和氦离子
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// IF(NUMAT(IA).EQ.1) THEN
|
||||
/// IATH=IA
|
||||
/// IF(IZ(ION).EQ.1) IELH=ION
|
||||
/// IF(IZ(ION).EQ.0) IELHM=ION
|
||||
/// END IF
|
||||
/// IF(NUMAT(IA).EQ.2) THEN
|
||||
/// IATHE=IA
|
||||
/// IF(IZ(ION).EQ.1) IELHE1=ION
|
||||
/// IF(IZ(ION).EQ.2) IELHE2=ION
|
||||
/// END IF
|
||||
/// ```
|
||||
pub fn identify_hydrogen_helium(
|
||||
ia: usize,
|
||||
iz: i32,
|
||||
ion: usize,
|
||||
ids: &mut HydrogenHeliumIds,
|
||||
) {
|
||||
// ia=1: 氢
|
||||
if ia == 1 {
|
||||
ids.iath = ia;
|
||||
if iz == 1 {
|
||||
ids.ielh = ion;
|
||||
}
|
||||
if iz == 0 {
|
||||
ids.ielhm = ion;
|
||||
}
|
||||
}
|
||||
// ia=2: 氦
|
||||
if ia == 2 {
|
||||
ids.iathe = ia;
|
||||
if iz == 1 {
|
||||
ids.ielhe1 = ion;
|
||||
}
|
||||
if iz == 2 {
|
||||
ids.ielhe2 = ion;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// 计算氢能级边界
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// N0H=N0A(IATH) ! 原子索引 → 第一个能级
|
||||
/// N1H=NLAST(IELH) ! 离子索引 → 最后一个能级
|
||||
/// NKH=NNEXT(IELH) ! 离子索引 → 续接能级
|
||||
/// N0HN=NFIRST(IELH) ! 离子索引 → 第一个能级
|
||||
/// IF(IELHM.GT.0) N0M=NFIRST(IELHM)
|
||||
/// ```
|
||||
///
|
||||
/// 注: ids.ielh/ielhm 是 Fortran 1-based 离子编号，Vec 是 Rust 0-based。
|
||||
pub fn compute_hydrogen_level_bounds(
|
||||
ids: &HydrogenHeliumIds,
|
||||
nfirst: &[usize],
|
||||
nlast: &[usize],
|
||||
nnext: &[usize],
|
||||
) -> HydrogenHeliumIds {
|
||||
let mut result = ids.clone();
|
||||
if ids.iath > 0 {
|
||||
// Fortran N0H=N0A(IATH): 原子索引→该原子第一个离子的 NFIRST
|
||||
// N0A(IA)=NFIRST(ION) 在离子循环中设置。
|
||||
// Rust: ids.ielh 是 1-based 离子号 → 减 1 得到 Vec 索引
|
||||
let ielh_idx = ids.ielh.saturating_sub(1);
|
||||
result.n0h = nfirst[ielh_idx];
|
||||
result.n1h = nlast[ielh_idx];
|
||||
result.nkh = nnext[ielh_idx];
|
||||
result.n0hn = nfirst[ielh_idx];
|
||||
if ids.ielhm > 0 {
|
||||
let ielhm_idx = ids.ielhm.saturating_sub(1);
|
||||
result.n0m = nfirst[ielhm_idx];
|
||||
}
|
||||
}
|
||||
result
|
||||
}
|
||||
|
||||
/// 频率读取模式
|
||||
#[derive(Debug, Clone, Copy, PartialEq)]
|
||||
pub enum FrequencyReadMode {
|
||||
/// 负值: 读取指定频率点
|
||||
Explicit,
|
||||
/// 正值: 连续频率数
|
||||
Continuum,
|
||||
}
|
||||
|
||||
/// 解析频率读取模式
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// READ(IBUFF,*) NFREAD
|
||||
/// NJREAD=NFREAD
|
||||
/// IF(NJREAD.LT.0) THEN
|
||||
/// NJREAD=-NJREAD
|
||||
/// NFREQC=NJREAD
|
||||
/// DO IJ=1,NJREAD
|
||||
/// READ(IBUFF,*) FREQEXP
|
||||
/// END DO
|
||||
/// ELSE
|
||||
/// NFREQC=NJREAD
|
||||
/// END IF
|
||||
/// ```
|
||||
pub fn parse_frequency_read_mode(nfreqread: i32) -> (FrequencyReadMode, usize) {
|
||||
if nfreqread < 0 {
|
||||
(FrequencyReadMode::Explicit, (-nfreqread) as usize)
|
||||
} else {
|
||||
(FrequencyReadMode::Continuum, nfreqread as usize)
|
||||
}
|
||||
}
|
||||
|
||||
/// INITIA 配置摘要
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct InitiaConfig {
|
||||
/// 有效温度 (K)
|
||||
pub teff: f64,
|
||||
/// 表面重力 (log g)
|
||||
pub grav: f64,
|
||||
/// LTE 模式
|
||||
pub lte: bool,
|
||||
/// 灰色大气
|
||||
pub ltgrey: bool,
|
||||
/// 模型类型
|
||||
pub inmod: i32,
|
||||
/// 频率读取模式
|
||||
pub freq_mode: FrequencyReadMode,
|
||||
/// 频率点数
|
||||
pub nfreq: usize,
|
||||
/// 湍流速度 (km/s)
|
||||
pub vtb: f64,
|
||||
}
|
||||
|
||||
/// 显式离子输入参数
|
||||
///
|
||||
/// 对应 Fortran INITIA 中从 IBUFF 读取的离子记录:
|
||||
/// `READ(IBUFF,*) IATII,IZII,NLEVSI,ILASTI,ILVLIN,NONSTD,TYPIOI,FILEI`
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct ExplicitIonInput {
|
||||
/// 原子序数
|
||||
pub iat: i32,
|
||||
/// 电荷
|
||||
pub iz: i32,
|
||||
/// 能级数
|
||||
pub nlevs: i32,
|
||||
/// 最后能级标志 (0=新离子, >0=能级数据, <0=结束)
|
||||
pub ilasti: i32,
|
||||
/// 能级限制
|
||||
pub ilvlin: i32,
|
||||
/// 非标准模式标志
|
||||
pub nonstd: i32,
|
||||
/// 离子类型标签
|
||||
pub typion: String,
|
||||
/// 数据文件名
|
||||
pub fidata: String,
|
||||
}
|
||||
|
||||
/// 显式能级数据
|
||||
///
|
||||
/// 对应 Fortran INITIA 中 ILASTI>0 时的能级记录
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct ExplicitLevelData {
|
||||
/// 能级统计权重 (ILASTI 值)
|
||||
pub g: f64,
|
||||
/// 离子类型标签
|
||||
pub typlev: String,
|
||||
}
|
||||
|
||||
/// 显式离子非标准参数 (NONSTD > 0)
|
||||
#[derive(Debug, Clone, Default)]
|
||||
pub struct NonStdParams {
|
||||
pub iupsum: i32,
|
||||
pub icup: i32,
|
||||
pub modeff: i32,
|
||||
pub nff: i32,
|
||||
}
|
||||
|
||||
/// 显式离子非标准参数 (NONSTD < 0)
|
||||
#[derive(Debug, Clone, Default)]
|
||||
pub struct NonStdFileParams {
|
||||
pub ifil1: i32,
|
||||
pub ifil2: i32,
|
||||
pub fiodf1: String,
|
||||
pub fiodf2: String,
|
||||
pub fibfcs: String,
|
||||
}
|
||||
|
||||
/// 离子设置结果
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct IonSetupResult {
|
||||
/// 离子索引信息
|
||||
pub indices: IonIndices,
|
||||
/// 氢/氦标识
|
||||
pub hh_ids: HydrogenHeliumIds,
|
||||
/// 离子电荷 + 1
|
||||
pub iz_plus1: i32,
|
||||
/// 自由模式
|
||||
pub ifree: i32,
|
||||
}
|
||||
|
||||
/// INITIA 输出结果
|
||||
///
|
||||
/// 包含 INITIA 初始化过程产生的所有状态数据。
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct InitiaOutput {
|
||||
/// 配置摘要
|
||||
pub config: InitiaConfig,
|
||||
/// 离子数
|
||||
pub nion: usize,
|
||||
/// 能级总数
|
||||
pub nlevel: usize,
|
||||
/// 原子种类数
|
||||
pub natom: usize,
|
||||
/// 氢/氦离子标识
|
||||
pub hh_ids: HydrogenHeliumIds,
|
||||
/// 离子索引列表
|
||||
pub ion_indices: Vec<IonIndices>,
|
||||
/// 离子电荷+1列表 (IZ)
|
||||
pub iz: Vec<i32>,
|
||||
/// 离子自由模式列表
|
||||
pub ifree: Vec<i32>,
|
||||
/// 能级所属离子索引 (IEL)
|
||||
pub iel: Vec<usize>,
|
||||
/// 能级所属原子索引 (IATM)
|
||||
pub iatm: Vec<usize>,
|
||||
/// 离子续接索引 (ILK)
|
||||
pub ilk: Vec<usize>,
|
||||
/// 湍流速度平方 (cm/s)^2
|
||||
pub vturb: Vec<f64>,
|
||||
/// 额外不透明度源开关
|
||||
pub opacity_switches: OpacitySwitches,
|
||||
/// STATE0 初始化结果（原子质量、丰度、电离势等）
|
||||
pub state0: State0Output,
|
||||
}
|
||||
|
||||
/// 额外不透明度源开关
|
||||
///
|
||||
/// 对应 Fortran INITIA 末尾输出的不透明度参数。
|
||||
#[derive(Debug, Clone, Default)]
|
||||
pub struct OpacitySwitches {
|
||||
pub iophmi: i32, // H- opacity in LTE
|
||||
pub ioph2p: i32, // H2+ opacity
|
||||
pub iophem: i32, // He- b-f and f-f
|
||||
pub iopch: i32, // CH opacity
|
||||
pub iopoh: i32, // OH opacity
|
||||
pub ioph2m: i32, // H2- opacity
|
||||
pub ioh2h2: i32, // CIA H2-H2
|
||||
pub ioh2he: i32, // CIA H2-He
|
||||
pub ioh2h1: i32, // CIA H2-H
|
||||
pub iohhe: i32, // CIA H-He
|
||||
pub irsct: i32, // Rayleigh scattering on H I
|
||||
pub irsch2: i32, // Rayleigh scattering on H2
|
||||
pub irsche: i32, // Rayleigh scattering on He I
|
||||
pub iophli: i32, // Lyman lines wings
|
||||
}
|
||||
|
||||
/// 主 INITIA 编排函数
|
||||
///
|
||||
/// 翻译自 SYNSPEC `INITIA` 子程序 (synspec54.f:294)。
|
||||
///
|
||||
/// 这是 SYNSPEC 的核心初始化驱动函数，负责：
|
||||
/// 1. 读取基本输入参数（TEFF, GRAV, LTE 等）
|
||||
/// 2. 调用 NSTPAR 设置标准参数
|
||||
/// 3. 解析频率点和权重
|
||||
/// 4. 设置湍流速度
|
||||
/// 5. 调用 STATE0 初始化 Saha 方程参数
|
||||
/// 6. 读取显式离子/能级/跃迁参数
|
||||
/// 7. 为每个离子调用 RDATA 加载数据
|
||||
/// 8. 设置额外不透明度源
|
||||
///
|
||||
/// # Fortran 原始签名
|
||||
///
|
||||
/// ```fortran
|
||||
/// SUBROUTINE INITIA
|
||||
/// ```
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `input_lines` - 从 unit 5 (IBUFF) 读取的输入行
|
||||
/// * `config` - 基本配置参数
|
||||
/// * `opacity_switches` - 额外不透明度源开关（从 NSTPAR 或外部设置）
|
||||
///
|
||||
/// # 返回
|
||||
///
|
||||
/// `InitiaOutput` 包含所有初始化后的状态数据。
|
||||
pub fn initia(
|
||||
input_lines: &[String],
|
||||
config: InitiaConfig,
|
||||
opacity_switches: OpacitySwitches,
|
||||
nd: usize,
|
||||
) -> InitiaOutput {
|
||||
// ============================================================
|
||||
// 1. 湍流速度设置
|
||||
// ============================================================
|
||||
let vturb = setup_turbulent_velocity(config.vtb, nd);
|
||||
|
||||
// ============================================================
|
||||
// 2. STATE0 初始化 - Saha 方程基本参数
|
||||
// ============================================================
|
||||
// Fortran: CALL STATE0(1) - 初始化原子数据、丰度、电离势等
|
||||
let abnd_depth = &[] as &[f64]; // 默认均匀丰度
|
||||
let state0_out = state0(config.teff, nd, abnd_depth);
|
||||
|
||||
// ============================================================
|
||||
// 3. 初始化 ILK, IEXPL, ILTOT 数组
|
||||
// ============================================================
|
||||
let mlevel = 1134; // MLEVEL from PARAMS.FOR
|
||||
let _mion = 200; // MION from PARAMS.FOR
|
||||
let mut ilk = vec![0usize; mlevel];
|
||||
// iexpl, iltot 用于准分子卫星线
|
||||
|
||||
// ============================================================
|
||||
// 4. 读取显式离子参数
|
||||
// ============================================================
|
||||
// 解析输入行中的离子记录
|
||||
let mut ions: Vec<ExplicitIonInput> = Vec::new();
|
||||
let mut levels: Vec<Vec<ExplicitLevelData>> = Vec::new();
|
||||
let mut current_levels: Vec<ExplicitLevelData> = Vec::new();
|
||||
|
||||
for line in input_lines {
|
||||
let parts: Vec<&str> = line.split_whitespace().collect();
|
||||
if parts.len() < 7 {
|
||||
continue;
|
||||
}
|
||||
|
||||
// 尝试解析为离子记录: IATII, IZII, NLEVSI, ILASTI, ILVLIN, NONSTD, TYPIOI, FILEI
|
||||
if let (Ok(iat), Ok(iz), Ok(nlevs), Ok(ilasti), Ok(ilvlin), Ok(nonstd)) = (
|
||||
parts[0].parse::<i32>(),
|
||||
parts[1].parse::<i32>(),
|
||||
parts[2].parse::<i32>(),
|
||||
parts[3].parse::<i32>(),
|
||||
parts[4].parse::<i32>(),
|
||||
parts[5].parse::<i32>(),
|
||||
) {
|
||||
let typion = parts.get(6).unwrap_or(&"").to_string();
|
||||
let fidata = parts.get(7).unwrap_or(&"").to_string();
|
||||
|
||||
if ilasti == 0 {
|
||||
// 新离子记录
|
||||
if !current_levels.is_empty() {
|
||||
levels.push(current_levels.clone());
|
||||
current_levels.clear();
|
||||
}
|
||||
ions.push(ExplicitIonInput {
|
||||
iat, iz, nlevs, ilasti, ilvlin, nonstd,
|
||||
typion, fidata,
|
||||
});
|
||||
} else if ilasti > 0 {
|
||||
// 能级数据
|
||||
current_levels.push(ExplicitLevelData {
|
||||
g: ilasti as f64,
|
||||
typlev: typion,
|
||||
});
|
||||
}
|
||||
// ilasti < 0: 结束标志
|
||||
}
|
||||
}
|
||||
if !current_levels.is_empty() {
|
||||
levels.push(current_levels);
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 5. 计算离子索引和能级分配
|
||||
// ============================================================
|
||||
let mut ion_indices_vec: Vec<IonIndices> = Vec::new();
|
||||
let mut iz_vec: Vec<i32> = Vec::new();
|
||||
let mut ifree_vec: Vec<i32> = Vec::new();
|
||||
let mut iel = vec![0usize; mlevel];
|
||||
let mut iatm = vec![0usize; mlevel];
|
||||
let mut hh_ids = HydrogenHeliumIds::default();
|
||||
|
||||
let mut ilev: usize = 0;
|
||||
let mut iat_last: i32 = 0;
|
||||
let mut nion = 0usize;
|
||||
let mut natom = 0usize;
|
||||
|
||||
for (ion_idx, ion) in ions.iter().enumerate() {
|
||||
if ion.ilasti != 0 {
|
||||
continue; // 跳过非离子记录
|
||||
}
|
||||
|
||||
nion += 1;
|
||||
|
||||
// 计算离子索引
|
||||
let (indices, iat_new, new_ilev) = compute_ion_indices(
|
||||
ion_idx,
|
||||
ion.iat,
|
||||
ion.iz,
|
||||
ion.nlevs,
|
||||
iat_last,
|
||||
ilev,
|
||||
0, // NFF，后续由 RDATA 设置
|
||||
);
|
||||
|
||||
// 更新原子索引
|
||||
if ion.iat != iat_last {
|
||||
natom = natom.max(ion.iat as usize);
|
||||
}
|
||||
iat_last = iat_new;
|
||||
ilev = new_ilev;
|
||||
|
||||
// 分配能级
|
||||
assign_levels(
|
||||
indices.nfirst,
|
||||
indices.nlast,
|
||||
indices.nnext,
|
||||
nion,
|
||||
ion.iat as usize,
|
||||
&mut iel,
|
||||
&mut iatm,
|
||||
&mut ilk,
|
||||
);
|
||||
|
||||
// 识别氢/氦
|
||||
identify_hydrogen_helium(ion.iat as usize, ion.iz, nion, &mut hh_ids);
|
||||
|
||||
iz_vec.push(ion.iz + 1);
|
||||
ifree_vec.push(1); // 默认 MODEFF=1
|
||||
|
||||
ion_indices_vec.push(indices);
|
||||
}
|
||||
|
||||
let nlevel = ilev;
|
||||
|
||||
// ============================================================
|
||||
// 6. 计算氢能级边界
|
||||
// ============================================================
|
||||
let nfirst: Vec<usize> = ion_indices_vec.iter().map(|idx| idx.nfirst).collect();
|
||||
let nlast: Vec<usize> = ion_indices_vec.iter().map(|idx| idx.nlast).collect();
|
||||
let nnext: Vec<usize> = ion_indices_vec.iter().map(|idx| idx.nnext).collect();
|
||||
hh_ids = compute_hydrogen_level_bounds(&hh_ids, &nfirst, &nlast, &nnext);
|
||||
|
||||
// ============================================================
|
||||
// 7. RDATA 调用 - 为每个离子加载能级数据
|
||||
// ============================================================
|
||||
// 注: RDATA 需要单独翻译
|
||||
// DO ION=1,NION
|
||||
// CALL RDATA(ION)
|
||||
// NFF=NQUANT(NLAST(ION))+1
|
||||
// IF(NFF.GT.0) FF(ION)=EH/H*IZ(ION)*IZ(ION)/NFF/NFF
|
||||
// END DO
|
||||
|
||||
InitiaOutput {
|
||||
config,
|
||||
nion,
|
||||
nlevel,
|
||||
natom,
|
||||
hh_ids,
|
||||
ion_indices: ion_indices_vec,
|
||||
iz: iz_vec,
|
||||
ifree: ifree_vec,
|
||||
iel,
|
||||
iatm,
|
||||
ilk,
|
||||
vturb,
|
||||
opacity_switches,
|
||||
state0: state0_out,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_get_ground_state_weight_light() {
|
||||
// 碳 (iat=6), 中性 (iz=0)
|
||||
let g = get_ground_state_weight(6, 0);
|
||||
assert_eq!(g, Some(IGLE[6]));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_get_ground_state_weight_iron() {
|
||||
// 铁 (iat=26), 中性 (iz=0) → idx=26-0=26, IGFE len=26 → out of bounds
|
||||
// 实际上 idx=26 超出范围，返回 None
|
||||
let g = get_ground_state_weight(26, 0);
|
||||
assert_eq!(g, None);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_get_ground_state_weight_nickel() {
|
||||
// 镍 (iat=28), 中性 (iz=0) → idx=28-0=28, IGNI len=28 → out of bounds
|
||||
let g = get_ground_state_weight(28, 0);
|
||||
assert_eq!(g, None);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_get_ground_state_weight_unknown() {
|
||||
let g = get_ground_state_weight(30, 0);
|
||||
assert_eq!(g, None);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_ion_indices_new_atom() {
|
||||
let (indices, iat_new, ilev) = compute_ion_indices(0, 1, 0, 5, 0, 0, 0);
|
||||
assert_eq!(indices.nfirst, 1); // 新原子: ilev+1
|
||||
assert_eq!(indices.nlast, 5);
|
||||
assert_eq!(indices.nnext, 6);
|
||||
assert_eq!(indices.iz_plus1, 1);
|
||||
assert_eq!(ilev, 6);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_ion_indices_same_atom() {
|
||||
let (indices, _, ilev) = compute_ion_indices(1, 1, 1, 3, 1, 6, 0);
|
||||
assert_eq!(indices.nfirst, 6); // 同原子: ilev
|
||||
assert_eq!(indices.nlast, 8);
|
||||
assert_eq!(indices.nnext, 9);
|
||||
assert_eq!(indices.iz_plus1, 2);
|
||||
assert_eq!(ilev, 9);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_ion_indices_with_nff() {
|
||||
let (indices, _, _) = compute_ion_indices(0, 1, 0, 5, 0, 0, 3);
|
||||
// FF = EH/H * 1*1 / 9
|
||||
let expected = EH / H / 9.0;
|
||||
assert!((indices.ff - expected).abs() / expected < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_assign_levels() {
|
||||
let mut iel = vec![0; 10];
|
||||
let mut iatm = vec![0; 10];
|
||||
let mut ilk = vec![0; 10];
|
||||
assign_levels(2, 5, 6, 1, 3, &mut iel, &mut iatm, &mut ilk);
|
||||
for i in 2..=5 {
|
||||
assert_eq!(iel[i], 1);
|
||||
assert_eq!(iatm[i], 3);
|
||||
}
|
||||
assert_eq!(ilk[6], 1);
|
||||
assert_eq!(iatm[6], 3);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_setup_turbulent_velocity() {
|
||||
let vturb = setup_turbulent_velocity(2.0, 5);
|
||||
// 2.0 km/s → 2e5 cm/s → v^2 = 4e10
|
||||
assert_eq!(vturb.len(), 5);
|
||||
assert!((vturb[0] - 4e10).abs() < 1e5);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_identify_hydrogen_helium() {
|
||||
let mut ids = HydrogenHeliumIds::default();
|
||||
identify_hydrogen_helium(1, 0, 0, &mut ids); // H I
|
||||
identify_hydrogen_helium(1, 1, 1, &mut ids); // H II
|
||||
identify_hydrogen_helium(2, 1, 2, &mut ids); // He I
|
||||
identify_hydrogen_helium(2, 2, 3, &mut ids); // He II
|
||||
assert_eq!(ids.iath, 1);
|
||||
assert_eq!(ids.ielh, 1);
|
||||
assert_eq!(ids.ielhm, 0);
|
||||
assert_eq!(ids.iathe, 2);
|
||||
assert_eq!(ids.ielhe1, 2);
|
||||
assert_eq!(ids.ielhe2, 3);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parse_frequency_read_mode() {
|
||||
assert_eq!(parse_frequency_read_mode(100), (FrequencyReadMode::Continuum, 100));
|
||||
assert_eq!(parse_frequency_read_mode(-50), (FrequencyReadMode::Explicit, 50));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_initia_basic() {
|
||||
// 模拟基本 INITIA 输入: 2 个离子 (H, He)
|
||||
let input_lines = vec![
|
||||
"1 0 5 0 0 0 H h.dat".to_string(),
|
||||
"2 1 3 0 0 0 He he.dat".to_string(),
|
||||
];
|
||||
let config = InitiaConfig {
|
||||
teff: 30000.0,
|
||||
grav: 4.0,
|
||||
lte: false,
|
||||
ltgrey: false,
|
||||
inmod: 1,
|
||||
freq_mode: FrequencyReadMode::Continuum,
|
||||
nfreq: 1000,
|
||||
vtb: 2.0,
|
||||
};
|
||||
let opacity = OpacitySwitches::default();
|
||||
|
||||
let output = initia(&input_lines, config, opacity, 35);
|
||||
|
||||
// 验证基本输出
|
||||
assert_eq!(output.nion, 2);
|
||||
assert_eq!(output.config.teff, 30000.0);
|
||||
assert!(output.nlevel > 0);
|
||||
assert_eq!(output.vturb.len(), 35); // ND=35
|
||||
// 2.0 km/s -> 2e5 cm/s -> v^2 = 4e10
|
||||
assert!((output.vturb[0] - 4e10).abs() < 1e5);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_initia_hydrogen_helium_ids() {
|
||||
let input_lines = vec![
|
||||
"1 0 5 0 0 0 H h.dat".to_string(),
|
||||
"1 1 1 0 0 0 H+ h+.dat".to_string(),
|
||||
"2 1 3 0 0 0 He he.dat".to_string(),
|
||||
"2 2 1 0 0 0 He+ he+.dat".to_string(),
|
||||
];
|
||||
let config = InitiaConfig {
|
||||
teff: 20000.0, grav: 4.0, lte: true, ltgrey: false,
|
||||
inmod: 1, freq_mode: FrequencyReadMode::Continuum,
|
||||
nfreq: 500, vtb: 1.0,
|
||||
};
|
||||
let output = initia(&input_lines, config, OpacitySwitches::default(), 35);
|
||||
|
||||
assert_eq!(output.hh_ids.iath, 1);
|
||||
assert!(output.hh_ids.ielh > 0 || output.hh_ids.iath == 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_initia_ion_indices() {
|
||||
let input_lines = vec![
|
||||
"1 0 5 0 0 0 H h.dat".to_string(),
|
||||
"2 1 3 0 0 0 He he.dat".to_string(),
|
||||
];
|
||||
let config = InitiaConfig {
|
||||
teff: 10000.0, grav: 4.0, lte: true, ltgrey: false,
|
||||
inmod: 1, freq_mode: FrequencyReadMode::Continuum,
|
||||
nfreq: 100, vtb: 0.0,
|
||||
};
|
||||
let output = initia(&input_lines, config, OpacitySwitches::default(), 35);
|
||||
|
||||
// 第一个离子 (H): 新原子, nfirst=1
|
||||
assert_eq!(output.ion_indices[0].nfirst, 1);
|
||||
assert_eq!(output.ion_indices[0].nlast, 5); // 1+5-1
|
||||
assert_eq!(output.ion_indices[0].nnext, 6);
|
||||
|
||||
// 第二个离子 (He): 不同原子, nfirst = ilev+1 = 6+1 = 7
|
||||
assert_eq!(output.ion_indices[1].nfirst, 7);
|
||||
assert_eq!(output.ion_indices[1].nlast, 9); // 7+3-1
|
||||
assert_eq!(output.ion_indices[1].nnext, 10);
|
||||
}
|
||||
}
|
||||
178
src/synspec/math/inkur.rs
Normal file
178
src/synspec/math/inkur.rs
Normal file
@ -0,0 +1,178 @@
|
||||
//! Input of a Kurucz model atmosphere.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine INKUR at line 11048.
|
||||
//!
|
||||
//! Reads a Kurucz model atmosphere from file (unit 8) and initializes
|
||||
//! the model state arrays (DM, TEMP, ELEC, DENS, POPUL, etc.).
|
||||
|
||||
/// Parameters for INKUR initialization.
|
||||
pub struct InkurParams<'a> {
|
||||
/// Boltzmann constant (BOLK)
|
||||
pub bolk: f64,
|
||||
/// Mean molecular weight at each depth
|
||||
pub wmm: &'a [f64],
|
||||
/// Total number of particles per H atom
|
||||
pub ytot: &'a [f64],
|
||||
/// Number of atoms
|
||||
pub natom: usize,
|
||||
/// Number of levels
|
||||
pub nlevel: usize,
|
||||
/// Molecular equilibrium flag
|
||||
pub ifmol: i32,
|
||||
/// Molecular temperature limit
|
||||
pub tmolim: f64,
|
||||
/// Maximum number of depth points
|
||||
pub nd_max: usize,
|
||||
}
|
||||
|
||||
/// Result of INKUR initialization.
|
||||
pub struct InkurResult {
|
||||
/// Number of depth points
|
||||
pub nd: usize,
|
||||
/// Effective temperature (from file header)
|
||||
pub tef: f64,
|
||||
/// Surface gravity log g (from file header)
|
||||
pub grav: f64,
|
||||
/// Mass depth coordinate
|
||||
pub dm: Vec<f64>,
|
||||
/// Temperature at each depth
|
||||
pub temp: Vec<f64>,
|
||||
/// Electron density at each depth
|
||||
pub elec: Vec<f64>,
|
||||
/// Mass density at each depth
|
||||
pub dens: Vec<f64>,
|
||||
/// Population of each level at each depth (nlevel x nd)
|
||||
pub popul: Vec<Vec<f64>>,
|
||||
}
|
||||
|
||||
/// Input of a Kurucz model atmosphere.
|
||||
///
|
||||
/// Reads model atmosphere data and initializes the depth-dependent arrays.
|
||||
/// For each depth point, computes density from pressure and temperature,
|
||||
/// optionally solves molecular equilibrium, and computes LTE populations.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Initialization parameters
|
||||
/// * `tef` - Effective temperature from file header
|
||||
/// * `grav` - Surface gravity from file header
|
||||
/// * `depth_data` - Slice of (dm, temp, pressure, elec) for each depth
|
||||
/// * `moleq_fn` - Optional molecular equilibrium callback: (id, t, an, aein) -> ane
|
||||
/// * `attot_fn` - Callback to compute ATTOT: (iat, id, dens, wmm, ytot, abund) -> f64
|
||||
/// * `post_depth_fn` - Callback after each depth: (id) for WNSTOR, SABOLF, RATMAT, LEVSOL
|
||||
///
|
||||
/// # Returns
|
||||
/// Initialized model arrays.
|
||||
pub fn inkur(
|
||||
params: &InkurParams,
|
||||
tef: f64,
|
||||
grav: f64,
|
||||
depth_data: &[(f64, f64, f64, f64)],
|
||||
moleq_fn: Option<&dyn Fn(usize, f64, f64, f64) -> f64>,
|
||||
attot_fn: &dyn Fn(usize, usize, f64, f64, f64, f64) -> f64,
|
||||
post_depth_fn: &dyn Fn(usize, &mut [Vec<f64>]),
|
||||
) -> InkurResult {
|
||||
let bolk = params.bolk;
|
||||
let nd = depth_data.len().min(params.nd_max);
|
||||
let mut dm = Vec::with_capacity(nd);
|
||||
let mut temp = Vec::with_capacity(nd);
|
||||
let mut elec = Vec::with_capacity(nd);
|
||||
let mut dens = Vec::with_capacity(nd);
|
||||
let mut popul = vec![vec![0.0; nd]; params.nlevel];
|
||||
|
||||
for (id, &(dm_i, temp_i, p, elec_i)) in depth_data.iter().enumerate().take(nd) {
|
||||
dm.push(dm_i);
|
||||
temp.push(temp_i);
|
||||
elec.push(elec_i);
|
||||
|
||||
// Compute density: DENS = WMM * (P/(T*BOLK) - ELEC)
|
||||
let an = p / temp_i / bolk;
|
||||
let dens_i = params.wmm[id] * (an - elec_i);
|
||||
dens.push(dens_i);
|
||||
|
||||
let t = temp_i;
|
||||
|
||||
// Molecular equilibrium or simple abundance
|
||||
if params.ifmol > 0 && t < params.tmolim {
|
||||
if let Some(ref moleq) = moleq_fn {
|
||||
let aein = elec_i;
|
||||
let _ane = moleq(id, t, an, aein);
|
||||
}
|
||||
} else {
|
||||
// Compute total atom abundance for each atom
|
||||
// Fortran: ATTOT(IAT,ID)=DENS(ID)/WMM(ID)/YTOT(ID)*ABUND(IAT,ID)
|
||||
for iat in 0..params.natom {
|
||||
let _ = attot_fn(iat, id, dens_i, params.wmm[id], params.ytot[id], 0.0);
|
||||
}
|
||||
}
|
||||
|
||||
// Post-depth processing: WNSTOR, SABOLF, RATMAT, LEVSOL
|
||||
post_depth_fn(id, &mut popul);
|
||||
}
|
||||
|
||||
InkurResult {
|
||||
nd,
|
||||
tef,
|
||||
grav,
|
||||
dm,
|
||||
temp,
|
||||
elec,
|
||||
dens,
|
||||
popul,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_inkur_basic() {
|
||||
let params = InkurParams {
|
||||
bolk: 1.380649e-16,
|
||||
wmm: &[1.0, 1.0],
|
||||
ytot: &[1.0, 1.0],
|
||||
natom: 1,
|
||||
nlevel: 2,
|
||||
ifmol: 0,
|
||||
tmolim: 0.0,
|
||||
nd_max: 10,
|
||||
};
|
||||
|
||||
let depth_data = [
|
||||
(1e-4, 5000.0, 1e5, 1e11),
|
||||
(1e-3, 4000.0, 1e4, 1e10),
|
||||
];
|
||||
|
||||
let attot_fn = |_iat: usize, _id: usize, dens: f64, wmm: f64, ytot: f64, _abund: f64| {
|
||||
dens / wmm / ytot
|
||||
};
|
||||
|
||||
let post_depth_fn = |_id: usize, _popul: &mut [Vec<f64>]| {
|
||||
// No-op for test
|
||||
};
|
||||
|
||||
let result = inkur(
|
||||
¶ms,
|
||||
5777.0,
|
||||
4.44,
|
||||
&depth_data,
|
||||
None,
|
||||
&attot_fn,
|
||||
&post_depth_fn,
|
||||
);
|
||||
|
||||
assert_eq!(result.nd, 2);
|
||||
assert_eq!(result.dm.len(), 2);
|
||||
assert_eq!(result.temp.len(), 2);
|
||||
assert_eq!(result.elec.len(), 2);
|
||||
assert_eq!(result.dens.len(), 2);
|
||||
assert_eq!(result.tef, 5777.0);
|
||||
assert_eq!(result.grav, 4.44);
|
||||
assert!(result.dens[0].is_finite());
|
||||
assert!(result.dens[1].is_finite());
|
||||
// DENS = WMM * (P/(T*BOLK) - ELEC)
|
||||
let an0 = 1e5 / 5000.0 / 1.380649e-16;
|
||||
let expected_dens0 = 1.0 * (an0 - 1e11);
|
||||
assert!((result.dens[0] - expected_dens0).abs() < 1.0);
|
||||
}
|
||||
}
|
||||
517
src/synspec/math/inmoli.rs
Normal file
517
src/synspec/math/inmoli.rs
Normal file
@ -0,0 +1,517 @@
|
||||
//! inmoli — 分子线列表初始化和选择。
|
||||
//!
|
||||
//! Fortran 原始签名: SUBROUTINE INMOLI(ILIST)
|
||||
//!
|
||||
//! 读取分子线列表，选择可能贡献的线，设置线参数。
|
||||
//!
|
||||
//! 注意: Fortran 版本直接操作文件 I/O 和 COMMON 块。
|
||||
//! Rust 版本提供纯计算核心函数和完整编排函数。
|
||||
|
||||
use std::io::{BufRead, BufReader};
|
||||
use std::fs::File;
|
||||
|
||||
/// 物理常数
|
||||
const PI4: f64 = 7.95774715e-2; // 4π
|
||||
const C1: f64 = std::f64::consts::LN_10; // ln(10)
|
||||
const C2: f64 = 4.2014672; // ln(10) * (me*c^2)/(k*T_ref)
|
||||
const C3: f64 = 1.4387886; // h*c/k (cm*K)
|
||||
const CNM: f64 = 2.997925e17; // c in nm/s
|
||||
const EXT0: f64 = 3.17; // 初始截断距离
|
||||
|
||||
/// Kurucz 分子代码到 Tsuji 表索引的映射表
|
||||
///
|
||||
/// Fortran 原始代码:
|
||||
/// ```fortran
|
||||
/// molind(101)=2 ! H2
|
||||
/// molind(607)=7 ! CN
|
||||
/// molind(808)=10 ! O2
|
||||
/// ```
|
||||
pub fn kurucz_to_tsuji(code: i32) -> i32 {
|
||||
match code {
|
||||
101 => 2,
|
||||
106 => 5,
|
||||
107 => 12,
|
||||
108 => 4,
|
||||
111 => 122,
|
||||
112 => 32,
|
||||
114 => 17,
|
||||
116 => 16,
|
||||
120 => 34,
|
||||
124 => 198,
|
||||
126 => 214,
|
||||
606 => 8,
|
||||
607 => 7,
|
||||
608 => 6,
|
||||
614 => 21,
|
||||
616 => 20,
|
||||
707 => 9,
|
||||
708 => 11,
|
||||
714 => 24,
|
||||
716 => 23,
|
||||
808 => 10,
|
||||
812 => 126,
|
||||
813 => 134,
|
||||
814 => 25,
|
||||
816 => 26,
|
||||
820 => 179,
|
||||
822 => 29,
|
||||
823 => 30,
|
||||
10108 => 3,
|
||||
_ => 0,
|
||||
}
|
||||
}
|
||||
|
||||
/// 分子线参数
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct MolecularLine {
|
||||
/// 波长 (nm)
|
||||
pub alam: f64,
|
||||
/// Kurucz 分子代码
|
||||
pub anum: f64,
|
||||
/// log(gf)
|
||||
pub gf: f64,
|
||||
/// 下能级激发势 (cm^-1)
|
||||
pub excl: f64,
|
||||
/// 辐射阻尼参数
|
||||
pub gr: f64,
|
||||
/// Stark 阻尼参数
|
||||
pub gs: f64,
|
||||
/// Van der Waals 阻尼参数
|
||||
pub gw: f64,
|
||||
/// H2 VdW 参数 (如果 ivdwli=1)
|
||||
pub gh2: Option<f64>,
|
||||
/// H2 温度指数
|
||||
pub xnh2: Option<f64>,
|
||||
/// He VdW 参数 (如果 ivdwli=1)
|
||||
pub ghe: Option<f64>,
|
||||
/// He 温度指数
|
||||
pub xnhe: Option<f64>,
|
||||
}
|
||||
|
||||
/// 线强度参数
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct MolecularLineStrength {
|
||||
/// log(gf) * ln(10)
|
||||
pub gfp: f64,
|
||||
/// 激发势能 * h*c/k
|
||||
pub epp: f64,
|
||||
/// 频率 (s^-1)
|
||||
pub freq: f64,
|
||||
}
|
||||
|
||||
/// 计算线强度参数
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// GFP=C1*GF-C2
|
||||
/// EPP=C3*EXCL
|
||||
/// FR0=CNM/ALAM
|
||||
/// ```
|
||||
pub fn compute_molecular_line_strength(alam: f64, gf: f64, excl: f64) -> MolecularLineStrength {
|
||||
MolecularLineStrength {
|
||||
gfp: C1 * gf - C2,
|
||||
epp: C3 * excl.abs(),
|
||||
freq: CNM / alam,
|
||||
}
|
||||
}
|
||||
|
||||
/// 线选择判据
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// gx=gfp-epp/tstd
|
||||
/// AB0=EXP(gx)*RRMOL(IMOL,IDSTD)/DOPSTD/AVAB
|
||||
/// IF(AB0.LT.UN) GO TO 10 ! skip line
|
||||
/// ```
|
||||
pub fn line_selected_molecular(
|
||||
gfp: f64,
|
||||
epp: f64,
|
||||
tstd: f64,
|
||||
rrmol: f64,
|
||||
dopstd: f64,
|
||||
avab: f64,
|
||||
) -> bool {
|
||||
let gx = gfp - epp / tstd;
|
||||
if gx > -30.0 {
|
||||
let ab0 = (gx).exp() * rrmol / dopstd / avab;
|
||||
ab0 >= 1.0
|
||||
} else {
|
||||
false
|
||||
}
|
||||
}
|
||||
|
||||
/// 计算截断距离
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// EX0=AB0*ASTD*10.
|
||||
/// EXT=EXT0
|
||||
/// IF(EX0.GT.TEN) EXT=SQRT(EX0)
|
||||
/// EXTIN0=EXT*DOPSTD
|
||||
/// ```
|
||||
pub fn compute_cutoff_distance(ab0: f64, astd: f64, dopstd: f64) -> f64 {
|
||||
let ex0 = ab0 * astd * 10.0;
|
||||
let ext = if ex0 > 10.0 { ex0.sqrt() } else { EXT0 };
|
||||
ext * dopstd
|
||||
}
|
||||
|
||||
/// 线展宽参数
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct LineBroadening {
|
||||
/// 辐射阻尼 (4π * gamma_rad)
|
||||
pub gr: f64,
|
||||
/// Stark 阻尼 (4π * gamma_stark * 3.125e-5)
|
||||
pub gs: f64,
|
||||
/// Van der Waals 阻尼 (4π * gamma_vdw)
|
||||
pub gw: f64,
|
||||
}
|
||||
|
||||
/// 计算线展宽参数
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// GRM=GR*PI4
|
||||
/// GSM=GS*PI4*3.125e-5
|
||||
/// GWM=GW*PI4
|
||||
/// ```
|
||||
pub fn compute_line_broadening(gr: f64, gs: f64, gw: f64) -> LineBroadening {
|
||||
LineBroadening {
|
||||
gr: gr * PI4,
|
||||
gs: gs * PI4 * 3.125e-5,
|
||||
gw: gw * PI4,
|
||||
}
|
||||
}
|
||||
|
||||
/// 分子 Doppler 宽度参数
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// tkm=1.65e8/ammol(imol)
|
||||
/// DP0=3.33564E-11*FR0
|
||||
/// dops=dp0*sqrt(tkm*td+vturb(id))
|
||||
/// ```
|
||||
pub fn molecular_doppler_width(freq: f64, ammol: f64, temp: f64, vturb: f64) -> f64 {
|
||||
let tkm = 1.65e8 / ammol;
|
||||
let dp0 = 3.33564e-11 * freq;
|
||||
dp0 * (tkm * temp + vturb).sqrt()
|
||||
}
|
||||
|
||||
/// 分子线列表读取格式
|
||||
#[derive(Debug, Clone, Copy, PartialEq)]
|
||||
pub enum MolLineFormat {
|
||||
/// 9 字段: alam, anum, gf, excl, gr, gh2, xnh2, ghe, xnhe
|
||||
Full9,
|
||||
/// 7 字段: alam, anum, gf, excl, gr, gs, gw
|
||||
Standard7,
|
||||
/// 4 字段: alam, anum, gf, excl (展宽参数用默认值)
|
||||
Basic4,
|
||||
}
|
||||
|
||||
/// INMOLI 编排函数输出
|
||||
#[derive(Debug)]
|
||||
pub struct InmoliOutput {
|
||||
/// 选中的分子线数据
|
||||
pub lines: Vec<SelectedMolLine>,
|
||||
/// 线数
|
||||
pub nlines: usize,
|
||||
/// 列表格式
|
||||
pub format: MolLineFormat,
|
||||
/// 是否有 VdW 参数
|
||||
pub has_vdw: bool,
|
||||
}
|
||||
|
||||
/// 选中的分子线
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct SelectedMolLine {
|
||||
/// 频率 (s^-1)
|
||||
pub freq: f64,
|
||||
/// 激发势能 * h*c/k
|
||||
pub epp: f64,
|
||||
/// log(gf) * ln(10)
|
||||
pub gfp: f64,
|
||||
/// 截断距离 (频率单位)
|
||||
pub extin0: f64,
|
||||
/// 分子索引 (Tsuji 表)
|
||||
pub imol: usize,
|
||||
/// 辐射展宽 (4π * gr)
|
||||
pub gr: f64,
|
||||
/// Stark 展宽 (4π * gs * 3.125e-5)
|
||||
pub gs: f64,
|
||||
/// VdW 展宽 (4π * gw)
|
||||
pub gw: f64,
|
||||
/// H2 VdW 参数
|
||||
pub gvdwh2: f64,
|
||||
/// H2 温度指数
|
||||
pub gexph2: f64,
|
||||
/// He VdW 参数
|
||||
pub gvdwhe: f64,
|
||||
/// He 温度指数
|
||||
pub gexphe: f64,
|
||||
}
|
||||
|
||||
/// INMOLI 编排函数。
|
||||
///
|
||||
/// 读取分子线列表文件，选择可能贡献的线，返回线参数。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `path` - 分子线列表文件路径
|
||||
/// * `is_binary` - 是否为二进制格式
|
||||
/// * `alam0` - 起始波长 (nm)
|
||||
/// * `alast` - 终止波长 (nm)
|
||||
/// * `tstd` - 标准温度 (K)
|
||||
/// * `dopstd` - 标准 Doppler 宽度
|
||||
/// * `avab` - 最小吸收系数阈值
|
||||
/// * `astd` - 标准展宽参数
|
||||
/// * `nmolec` - 最大分子数
|
||||
/// * `rrmol` - 分子 populations (imol -> 值)
|
||||
/// * `ammol` - 分子质量 (imol -> 值)
|
||||
/// * `gsstd` - 标准 Stark 展宽
|
||||
/// * `gwstd` - 标准 VdW 展宽
|
||||
/// * `mlmax` - 最大线数
|
||||
///
|
||||
/// # 返回值
|
||||
///
|
||||
/// `InmoliOutput` 包含选中的分子线数据。
|
||||
pub fn inmoli(
|
||||
path: &str,
|
||||
is_binary: bool,
|
||||
alam0: f64,
|
||||
alast: f64,
|
||||
tstd: f64,
|
||||
dopstd: f64,
|
||||
avab: f64,
|
||||
astd: f64,
|
||||
nmolec: usize,
|
||||
rrmol: &[f64],
|
||||
_ammol: &[f64],
|
||||
gsstd: f64,
|
||||
gwstd: f64,
|
||||
mlmax: usize,
|
||||
) -> Option<InmoliOutput> {
|
||||
let cutoff = alam0 * 10.0; // CUTOF0 in Angstroms -> nm approximation
|
||||
let alam0_a = alam0; // nm
|
||||
let alast_a = alast; // nm
|
||||
|
||||
// 打开文件
|
||||
let file = File::open(path).ok()?;
|
||||
let reader = BufReader::new(file);
|
||||
|
||||
// 检测格式
|
||||
let (format, has_vdw, lines_iter) = if is_binary {
|
||||
// 二进制格式暂不支持
|
||||
return None;
|
||||
} else {
|
||||
// 文本格式：先读第一行检测字段数
|
||||
let mut lines: Vec<String> = Vec::new();
|
||||
for l in reader.lines().flatten() {
|
||||
lines.push(l);
|
||||
}
|
||||
|
||||
if lines.is_empty() {
|
||||
return None;
|
||||
}
|
||||
|
||||
// 检测格式
|
||||
let first_fields: Vec<&str> = lines[0].split_whitespace().collect();
|
||||
let (fmt, vdw) = match first_fields.len() {
|
||||
n if n >= 9 => (MolLineFormat::Full9, true),
|
||||
n if n >= 7 => (MolLineFormat::Standard7, false),
|
||||
_ => (MolLineFormat::Basic4, false),
|
||||
};
|
||||
|
||||
(fmt, vdw, lines)
|
||||
};
|
||||
|
||||
let mut selected = Vec::new();
|
||||
|
||||
for line_str in &lines_iter {
|
||||
let fields: Vec<&str> = line_str.split_whitespace().collect();
|
||||
if fields.len() < 4 {
|
||||
continue;
|
||||
}
|
||||
|
||||
// 解析基本字段
|
||||
let alam: f64 = fields[0].parse().ok().unwrap_or(0.0);
|
||||
let anum: f64 = fields[1].parse().ok().unwrap_or(0.0);
|
||||
let gf: f64 = fields[2].parse().ok().unwrap_or(0.0);
|
||||
let excl: f64 = fields[3].parse().ok().unwrap_or(0.0);
|
||||
|
||||
// 解析展宽参数
|
||||
let (gr, gs, gw, gh2, xnh2, ghe, xnhe) = match format {
|
||||
MolLineFormat::Full9 => {
|
||||
let gr: f64 = fields.get(4).and_then(|s| s.parse().ok()).unwrap_or(0.0);
|
||||
let gh2: f64 = fields.get(5).and_then(|s| s.parse().ok()).unwrap_or(0.0);
|
||||
let xnh2: f64 = fields.get(6).and_then(|s| s.parse().ok()).unwrap_or(0.0);
|
||||
let ghe: f64 = fields.get(7).and_then(|s| s.parse().ok()).unwrap_or(0.0);
|
||||
let xnhe: f64 = fields.get(8).and_then(|s| s.parse().ok()).unwrap_or(0.0);
|
||||
(gr, 0.0, 0.0, gh2, xnh2, ghe, xnhe)
|
||||
}
|
||||
MolLineFormat::Standard7 => {
|
||||
let gr: f64 = fields.get(4).and_then(|s| s.parse().ok()).unwrap_or(0.0);
|
||||
let gs: f64 = fields.get(5).and_then(|s| s.parse().ok()).unwrap_or(0.0);
|
||||
let gw: f64 = fields.get(6).and_then(|s| s.parse().ok()).unwrap_or(0.0);
|
||||
(gr, gs, gw, 0.0, 0.0, 0.0, 0.0)
|
||||
}
|
||||
MolLineFormat::Basic4 => {
|
||||
let gr = 2.4e13 / (alam * alam); // 默认辐射展宽
|
||||
(gr, gsstd, gwstd, 0.0, 0.0, 0.0, 0.0)
|
||||
}
|
||||
};
|
||||
|
||||
// 范围选择
|
||||
if alam < alam0_a - cutoff || alam > alast_a + cutoff {
|
||||
continue;
|
||||
}
|
||||
|
||||
// 分子代码映射
|
||||
let icod = (anum + 1e-4) as i32;
|
||||
let imol = kurucz_to_tsuji(icod);
|
||||
if imol <= 0 || imol > nmolec as i32 {
|
||||
continue;
|
||||
}
|
||||
let imol_usize = imol as usize;
|
||||
|
||||
// 线强度选择
|
||||
let strength = compute_molecular_line_strength(alam, gf, excl);
|
||||
let rrmol_val = if imol_usize < rrmol.len() {
|
||||
rrmol[imol_usize]
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
|
||||
if !line_selected_molecular(
|
||||
strength.gfp,
|
||||
strength.epp,
|
||||
tstd,
|
||||
rrmol_val,
|
||||
dopstd,
|
||||
avab,
|
||||
) {
|
||||
continue;
|
||||
}
|
||||
|
||||
// 超过最大线数则截断
|
||||
if selected.len() >= mlmax {
|
||||
break;
|
||||
}
|
||||
|
||||
// 计算截断距离
|
||||
let gx = strength.gfp - strength.epp / tstd;
|
||||
let ab0 = if gx > -30.0 {
|
||||
(gx).exp() * rrmol_val / dopstd / avab
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
let extin0 = compute_cutoff_distance(ab0, astd, dopstd);
|
||||
|
||||
// 展宽参数
|
||||
let broadening = compute_line_broadening(gr, gs, gw);
|
||||
|
||||
selected.push(SelectedMolLine {
|
||||
freq: strength.freq,
|
||||
epp: strength.epp,
|
||||
gfp: strength.gfp,
|
||||
extin0,
|
||||
imol: imol_usize,
|
||||
gr: broadening.gr,
|
||||
gs: broadening.gs,
|
||||
gw: broadening.gw,
|
||||
gvdwh2: gh2,
|
||||
gexph2: xnh2,
|
||||
gvdwhe: ghe,
|
||||
gexphe: xnhe,
|
||||
});
|
||||
}
|
||||
|
||||
Some(InmoliOutput {
|
||||
nlines: selected.len(),
|
||||
lines: selected,
|
||||
format,
|
||||
has_vdw,
|
||||
})
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_kurucz_to_tsuji_h2() {
|
||||
assert_eq!(kurucz_to_tsuji(101), 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_kurucz_to_tsuji_cn() {
|
||||
assert_eq!(kurucz_to_tsuji(607), 7);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_kurucz_to_tsuji_o2() {
|
||||
assert_eq!(kurucz_to_tsuji(808), 10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_kurucz_to_tsuji_unknown() {
|
||||
assert_eq!(kurucz_to_tsuji(999), 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_molecular_line_strength() {
|
||||
let ls = compute_molecular_line_strength(500.0, -2.0, 10000.0);
|
||||
// gfp = 2.3025851 * (-2.0) - 4.2014672 = -8.8066373
|
||||
assert!((ls.gfp - (C1 * (-2.0) - C2)).abs() < 1e-10);
|
||||
// epp = 1.4387886 * 10000.0 = 14387.886
|
||||
assert!((ls.epp - C3 * 10000.0).abs() < 1e-10);
|
||||
// freq = 2.997925e17 / 500.0
|
||||
assert!((ls.freq - CNM / 500.0).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_line_selected_molecular_strong() {
|
||||
// Strong line: gx > -30, ab0 >> 1
|
||||
let selected = line_selected_molecular(0.0, 0.0, 10000.0, 1.0, 1.0, 1.0);
|
||||
assert!(selected);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_line_selected_molecular_weak() {
|
||||
// Weak line: gx very negative
|
||||
let selected = line_selected_molecular(-100.0, 0.0, 10000.0, 1.0, 1.0, 1.0);
|
||||
assert!(!selected);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_cutoff_distance_strong() {
|
||||
let ext = compute_cutoff_distance(100.0, 1.0, 1000.0);
|
||||
// ex0 = 100*1*10 = 1000 > 10, so ext = sqrt(1000) * 1000
|
||||
let expected = 1000.0_f64.sqrt() * 1000.0;
|
||||
assert!((ext - expected).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_cutoff_distance_weak() {
|
||||
let ext = compute_cutoff_distance(0.1, 1.0, 1000.0);
|
||||
// ex0 = 0.1*1*10 = 1.0 < 10, so ext = EXT0 * 1000
|
||||
let expected = EXT0 * 1000.0;
|
||||
assert!((ext - expected).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_line_broadening() {
|
||||
let lb = compute_line_broadening(1e8, 1e-4, 1e-7);
|
||||
assert!((lb.gr - 1e8 * PI4).abs() < 1e-10);
|
||||
assert!((lb.gs - 1e-4 * PI4 * 3.125e-5).abs() < 1e-20);
|
||||
assert!((lb.gw - 1e-7 * PI4).abs() < 1e-20);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_molecular_doppler_width() {
|
||||
let dops = molecular_doppler_width(6e14, 28.0, 10000.0, 2e10);
|
||||
// tkm = 1.65e8/28.0, dp0 = 3.33564e-11 * 6e14
|
||||
let tkm = 1.65e8 / 28.0;
|
||||
let dp0 = 3.33564e-11 * 6e14;
|
||||
let expected = dp0 * (tkm * 10000.0_f64 + 2e10_f64).sqrt();
|
||||
assert!((dops - expected).abs() / expected < 1e-10);
|
||||
}
|
||||
}
|
||||
143
src/synspec/math/inpbf.rs
Normal file
143
src/synspec/math/inpbf.rs
Normal file
@ -0,0 +1,143 @@
|
||||
//! Input of b-factors for NLTE population correction.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine INPBF at line 11284.
|
||||
//!
|
||||
//! Reads b-factors from a file and interpolates them to the model
|
||||
//! depth grid, then multiplies the NLTE populations by the b-factors.
|
||||
|
||||
use super::interp::interp;
|
||||
|
||||
/// Parameters for INPBF.
|
||||
pub struct InpbfParams<'a> {
|
||||
/// Model depth grid (DM array)
|
||||
pub dm: &'a [f64],
|
||||
/// Number of depth points in the model
|
||||
pub nd: usize,
|
||||
/// Model input mode (INMOD): 2 = Kurucz format
|
||||
pub inmod: i32,
|
||||
/// Number of levels
|
||||
pub nlevel: usize,
|
||||
}
|
||||
|
||||
/// Result of INPBF.
|
||||
pub struct InpbfResult {
|
||||
/// B-factors interpolated to model depth grid (nlevel x nd)
|
||||
pub bfactors: Vec<Vec<f64>>,
|
||||
}
|
||||
|
||||
/// Input of b-factors for NLTE population correction.
|
||||
///
|
||||
/// Reads b-factors from a formatted file and interpolates them to the
|
||||
/// model depth grid. The b-factors are multiplicative corrections to
|
||||
/// the LTE populations to account for NLTE effects.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Initialization parameters
|
||||
/// * `depth_data` - Input depth points
|
||||
/// * `param_data` - Input parameter matrix (numpar x ndpth)
|
||||
///
|
||||
/// # Returns
|
||||
/// Interpolated b-factors for each level.
|
||||
pub fn inpbf(
|
||||
params: &InpbfParams,
|
||||
depth_data: &[f64],
|
||||
param_data: &[Vec<f64>],
|
||||
) -> InpbfResult {
|
||||
let ndpth = depth_data.len();
|
||||
let numpar = param_data.len();
|
||||
|
||||
// Determine number of leading parameters (not b-factors)
|
||||
// Fortran: NUMLT=3; IF(INMOD.EQ.2) NUMLT=4
|
||||
let mut numlt = 3;
|
||||
if params.inmod == 2 {
|
||||
numlt = 4;
|
||||
}
|
||||
// If NUMPAR < 0, one more leading parameter
|
||||
// (In practice, NUMPAR is already ABS(NUMPAR) from the reader)
|
||||
|
||||
let mut bfactors = Vec::new();
|
||||
|
||||
// Interpolate b-factors for each level
|
||||
for i in numlt..numpar {
|
||||
// Extract column I from param_data
|
||||
let xx: Vec<f64> = (0..ndpth).map(|id| param_data[i][id]).collect();
|
||||
|
||||
// Interpolate from DEPTH to DM scale
|
||||
// Fortran: CALL INTERP(DEPTH,XX,DM,BF,NDPTH,ND,2,1,1)
|
||||
// Rust interp: interp(x, y, xx, npol, ilogx, ilogy) -> Vec<f64>
|
||||
let bf = interp(depth_data, &xx, params.dm, 2, 1, 1);
|
||||
|
||||
bfactors.push(bf);
|
||||
}
|
||||
|
||||
InpbfResult { bfactors }
|
||||
}
|
||||
|
||||
/// Apply b-factors to NLTE populations.
|
||||
///
|
||||
/// Multiplies the populations by the interpolated b-factors.
|
||||
/// In Fortran: POPUL(I-NUMLT,ID) = POPUL(I-NUMLT,ID) * BF(ID)
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `popul` - Population array (nlevel x nd), modified in place
|
||||
/// * `bfactors` - B-factors from `inpbf`
|
||||
/// * `nd` - Number of depth points
|
||||
pub fn apply_inpbf(popul: &mut [Vec<f64>], bfactors: &[Vec<f64>], nd: usize) {
|
||||
for (i, bf) in bfactors.iter().enumerate() {
|
||||
for id in 0..nd {
|
||||
if id < bf.len() {
|
||||
popul[i][id] *= bf[id];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_inpbf_basic() {
|
||||
let params = InpbfParams {
|
||||
dm: &[1e-4, 1e-3, 1e-2],
|
||||
nd: 3,
|
||||
inmod: 1,
|
||||
nlevel: 5,
|
||||
};
|
||||
|
||||
// 4 parameters: 3 leading + 1 b-factor
|
||||
let depth_data = [1e-5, 1e-4, 1e-3, 1e-2, 1e-1];
|
||||
let param_data = vec![
|
||||
vec![0.0; 5], // param 0 (leading)
|
||||
vec![0.0; 5], // param 1 (leading)
|
||||
vec![0.0; 5], // param 2 (leading)
|
||||
vec![0.8, 0.9, 1.0, 1.1, 1.2], // b-factor for level 0
|
||||
];
|
||||
|
||||
let result = inpbf(¶ms, &depth_data, ¶m_data);
|
||||
assert_eq!(result.bfactors.len(), 1);
|
||||
assert_eq!(result.bfactors[0].len(), 3);
|
||||
// B-factors should be interpolated
|
||||
assert!(result.bfactors[0][0].is_finite());
|
||||
assert!(result.bfactors[0][1].is_finite());
|
||||
assert!(result.bfactors[0][2].is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_apply_inpbf() {
|
||||
let mut popul = vec![
|
||||
vec![1e12, 1e12, 1e12],
|
||||
vec![1e10, 1e10, 1e10],
|
||||
];
|
||||
let bfactors = vec![
|
||||
vec![1.1, 1.2, 1.3],
|
||||
];
|
||||
|
||||
apply_inpbf(&mut popul, &bfactors, 3);
|
||||
assert!((popul[0][0] - 1.1e12).abs() < 1e5);
|
||||
assert!((popul[0][1] - 1.2e12).abs() < 1e5);
|
||||
assert!((popul[0][2] - 1.3e12).abs() < 1e5);
|
||||
// Second level should be unchanged
|
||||
assert!((popul[1][0] - 1e10).abs() < 1e3);
|
||||
}
|
||||
}
|
||||
298
src/synspec/math/inpmod.rs
Normal file
298
src/synspec/math/inpmod.rs
Normal file
@ -0,0 +1,298 @@
|
||||
//! inpmod — 读取初始模型大气。
|
||||
//!
|
||||
//! Fortran 原始签名: SUBROUTINE INPMOD
|
||||
//!
|
||||
//! 从 unit 8 读取 TLUSTY 模型大气，计算 LTE 能级布居，
|
||||
//! 并可选地替换为 NLTE 布居。
|
||||
//!
|
||||
//! 注意: Fortran 版本直接操作 COMMON 块和文件 I/O。
|
||||
//! Rust 版本提供纯计算核心函数。
|
||||
|
||||
/// 模型大气深度点数据
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct ModelDepthPoint {
|
||||
/// 质量深度 (g/cm^2)
|
||||
pub depth: f64,
|
||||
/// 温度 (K)
|
||||
pub temp: f64,
|
||||
/// 电子密度 (cm^-3)
|
||||
pub elec: f64,
|
||||
/// 质量密度 (g/cm^3)
|
||||
pub dens: f64,
|
||||
}
|
||||
|
||||
/// 计算总粒子数密度
|
||||
///
|
||||
/// TOTN = DENS / WMM + ELEC
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// TOTN(ID)=DENS(ID)/WMM(ID)+ELEC(ID)
|
||||
/// ```
|
||||
pub fn total_number_density(dens: f64, wmm: f64, elec: f64) -> f64 {
|
||||
dens / wmm + elec
|
||||
}
|
||||
|
||||
/// 计算束缚-自由常数 (BCON)
|
||||
///
|
||||
/// BCON = ELEC / TEMP / SQRT(TEMP) * 2.0706E-16
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// BCON=ELEC(ID)/TEMP(ID)/SQRT(TEMP(ID))*2.0706E-16
|
||||
/// ```
|
||||
pub fn bound_free_constant(elec: f64, temp: f64) -> f64 {
|
||||
elec / temp / temp.sqrt() * 2.0706e-16
|
||||
}
|
||||
|
||||
/// 计算原子总数密度
|
||||
///
|
||||
/// ATTOT(IAT,ID) = DENS / WMM / YTOT * ABUND(IAT,ID)
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// DO IAT=1,NATOM
|
||||
/// ATTOT(IAT,ID)=DENS(ID)/WMM(ID)/YTOT(ID)*ABUND(IAT,ID)
|
||||
/// END DO
|
||||
/// ```
|
||||
pub fn compute_atom_densities(
|
||||
dens: f64,
|
||||
wmm: f64,
|
||||
ytot: f64,
|
||||
abundances: &[f64],
|
||||
) -> Vec<f64> {
|
||||
let factor = dens / wmm / ytot;
|
||||
abundances.iter().map(|&a| factor * a).collect()
|
||||
}
|
||||
|
||||
/// NLTE 布居替换
|
||||
///
|
||||
/// POPUL(J,ID) = X(IP+I) * RELAB(IATM(I),ID)
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// DO I=1,NLEV0
|
||||
/// j=iltot(i)
|
||||
/// POPUL(J,ID)=X(IP+I)*RELAB(IATM(I),ID)
|
||||
/// END DO
|
||||
/// ```
|
||||
pub fn replace_nlte_populations(
|
||||
populations: &mut [f64],
|
||||
nlte_data: &[f64],
|
||||
iltot: &[usize],
|
||||
relab: &[f64],
|
||||
iatm: &[usize],
|
||||
) {
|
||||
for (i, &j) in iltot.iter().enumerate().take(nlte_data.len()) {
|
||||
if j > 0 && j <= populations.len() {
|
||||
let relab_val = if iatm[i] < relab.len() { relab[iatm[i]] } else { 1.0 };
|
||||
populations[j - 1] = nlte_data[i] * relab_val;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// B 因子修正
|
||||
///
|
||||
/// POPUL(J,ID) = POPUL(J,ID) * PLTE(J,ID)
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// if(ibfac.eq.1) then
|
||||
/// do i=1,nlev0
|
||||
/// j=iltot(i)
|
||||
/// popul(j,id)=popul(j,id)*plte(j,id)
|
||||
/// end do
|
||||
/// end if
|
||||
/// ```
|
||||
pub fn apply_bfactor_correction(
|
||||
populations: &mut [f64],
|
||||
lte_populations: &[f64],
|
||||
iltot: &[usize],
|
||||
) {
|
||||
for &j in iltot {
|
||||
if j > 0 && j <= populations.len() {
|
||||
populations[j - 1] *= lte_populations[j - 1];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// 模型大气数据
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct ModelAtmosphere {
|
||||
/// 深度点数据
|
||||
pub depths: Vec<ModelDepthPoint>,
|
||||
/// 能级布居 [level][depth]
|
||||
pub populations: Vec<Vec<f64>>,
|
||||
/// LTE 布居 [level][depth]
|
||||
pub lte_populations: Vec<Vec<f64>>,
|
||||
}
|
||||
|
||||
/// 模型大气统计信息
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct ModelStats {
|
||||
/// 最低温度
|
||||
pub min_temp: f64,
|
||||
/// 最高温度
|
||||
pub max_temp: f64,
|
||||
/// 最低密度
|
||||
pub min_dens: f64,
|
||||
/// 最高密度
|
||||
pub max_dens: f64,
|
||||
/// 深度点数
|
||||
pub n_depths: usize,
|
||||
/// 能级数
|
||||
pub n_levels: usize,
|
||||
}
|
||||
|
||||
/// 计算模型大气统计信息
|
||||
pub fn compute_model_stats(model: &ModelAtmosphere) -> ModelStats {
|
||||
let mut min_temp = f64::MAX;
|
||||
let mut max_temp = f64::MIN;
|
||||
let mut min_dens = f64::MAX;
|
||||
let mut max_dens = f64::MIN;
|
||||
|
||||
for depth in &model.depths {
|
||||
if depth.temp < min_temp { min_temp = depth.temp; }
|
||||
if depth.temp > max_temp { max_temp = depth.temp; }
|
||||
if depth.dens < min_dens { min_dens = depth.dens; }
|
||||
if depth.dens > max_dens { max_dens = depth.dens; }
|
||||
}
|
||||
|
||||
ModelStats {
|
||||
min_temp,
|
||||
max_temp,
|
||||
min_dens,
|
||||
max_dens,
|
||||
n_depths: model.depths.len(),
|
||||
n_levels: model.populations.len(),
|
||||
}
|
||||
}
|
||||
|
||||
/// 模型文件格式
|
||||
#[derive(Debug, Clone, Copy, PartialEq)]
|
||||
pub enum ModelFormat {
|
||||
/// LTE 模型 (3 参数: T, NE, RHO)
|
||||
Lte,
|
||||
/// NLTE 模型 (>3 参数)
|
||||
Nlte,
|
||||
/// 磁盘模型 (4 参数)
|
||||
Disk,
|
||||
}
|
||||
|
||||
/// 检查模型格式
|
||||
pub fn detect_model_format(numpar: i32, inmod: i32) -> ModelFormat {
|
||||
if inmod == 2 {
|
||||
ModelFormat::Disk
|
||||
} else if numpar.abs() > 3 {
|
||||
ModelFormat::Nlte
|
||||
} else {
|
||||
ModelFormat::Lte
|
||||
}
|
||||
}
|
||||
|
||||
/// 计算 NLTE 修正因子
|
||||
///
|
||||
/// PNLT(IAT,ION,ID) = POPUL(NKI,ID) / G(NKI) * BCON
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// BCON=ELEC(ID)/TEMP(ID)/SQRT(TEMP(ID))*2.0706E-16
|
||||
/// IF(ION.GT.0) PNLT(IAT,ION,ID)=POPUL(NKI,ID)/G(NKI)*BCON
|
||||
/// ```
|
||||
pub fn compute_nlte_correction(
|
||||
popul_nki: f64,
|
||||
g_nki: f64,
|
||||
elec: f64,
|
||||
temp: f64,
|
||||
) -> f64 {
|
||||
if g_nki > 0.0 {
|
||||
popul_nki / g_nki * bound_free_constant(elec, temp)
|
||||
} else {
|
||||
0.0
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_total_number_density() {
|
||||
let totn = total_number_density(1e-8, 1.0, 1e-10);
|
||||
// 1e-8/1.0 + 1e-10 = 1.01e-8
|
||||
assert!((totn - 1.01e-8).abs() < 1e-15);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_bound_free_constant() {
|
||||
let bcon = bound_free_constant(1e10, 10000.0);
|
||||
// 1e10 / 10000 / 100 * 2.0706e-16 = 1e10 / 1e6 * 2.0706e-16 = 2.0706e-12
|
||||
let expected = 1e10 / 10000.0 / 100.0 * 2.0706e-16;
|
||||
assert!((bcon - expected).abs() / expected < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_atom_densities() {
|
||||
let abundances = vec![0.9, 0.1];
|
||||
let attot = compute_atom_densities(1e-8, 1.0, 1.5, &abundances);
|
||||
assert_eq!(attot.len(), 2);
|
||||
let expected0 = 1e-8 / 1.0 / 1.5 * 0.9;
|
||||
assert!((attot[0] - expected0).abs() < 1e-20);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_replace_nlte_populations() {
|
||||
let mut popul = vec![1.0, 2.0, 3.0, 4.0, 5.0];
|
||||
let nlte_data = vec![10.0, 20.0];
|
||||
let iltot = vec![1, 3]; // 1-indexed
|
||||
let relab = vec![1.0, 1.0, 1.0];
|
||||
let iatm = vec![0, 1];
|
||||
replace_nlte_populations(&mut popul, &nlte_data, &iltot, &relab, &iatm);
|
||||
assert_eq!(popul[0], 10.0); // j=1 → index 0
|
||||
assert_eq!(popul[2], 20.0); // j=3 → index 2
|
||||
assert_eq!(popul[4], 5.0); // unchanged
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_apply_bfactor_correction() {
|
||||
let mut popul = vec![2.0, 3.0, 4.0];
|
||||
let lte = vec![0.5, 0.5, 0.5];
|
||||
let iltot = vec![1, 2, 3]; // 1-indexed (Fortran convention)
|
||||
apply_bfactor_correction(&mut popul, <e, &iltot);
|
||||
assert_eq!(popul[0], 1.0); // 2.0 * 0.5
|
||||
assert_eq!(popul[1], 1.5); // 3.0 * 0.5
|
||||
assert_eq!(popul[2], 2.0); // 4.0 * 0.5
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_detect_model_format() {
|
||||
assert_eq!(detect_model_format(3, 0), ModelFormat::Lte);
|
||||
assert_eq!(detect_model_format(5, 0), ModelFormat::Nlte);
|
||||
assert_eq!(detect_model_format(3, 2), ModelFormat::Disk);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_nlte_correction() {
|
||||
let pnlt = compute_nlte_correction(1e12, 2.0, 1e10, 10000.0);
|
||||
// 1e12 / 2.0 * 1e10/10000/100 * 2.0706e-16
|
||||
let bcon = bound_free_constant(1e10, 10000.0);
|
||||
let expected = 1e12 / 2.0 * bcon;
|
||||
assert!((pnlt - expected).abs() / expected < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_compute_model_stats() {
|
||||
let model = ModelAtmosphere {
|
||||
depths: vec![
|
||||
ModelDepthPoint { depth: 1.0, temp: 5000.0, elec: 1e10, dens: 1e-8 },
|
||||
ModelDepthPoint { depth: 10.0, temp: 15000.0, elec: 1e12, dens: 1e-6 },
|
||||
],
|
||||
populations: vec![vec![1.0, 2.0]],
|
||||
lte_populations: vec![vec![1.0, 2.0]],
|
||||
};
|
||||
let stats = compute_model_stats(&model);
|
||||
assert_eq!(stats.min_temp, 5000.0);
|
||||
assert_eq!(stats.max_temp, 15000.0);
|
||||
assert_eq!(stats.n_depths, 2);
|
||||
}
|
||||
}
|
||||
187
src/synspec/math/interp.rs
Normal file
187
src/synspec/math/interp.rs
Normal file
@ -0,0 +1,187 @@
|
||||
//! 通用多项式插值。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `interp.f`
|
||||
//!
|
||||
//! 支持 (NPOL-1) 阶多项式插值，可选对数插值模式。
|
||||
|
||||
/// 通用多项式插值。
|
||||
///
|
||||
/// 在原始数据点 `(x, y)` 基础上，对新的 x 坐标 `xx` 进行插值。
|
||||
/// 支持对数插值模式。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `x` - 原始 x 坐标数组（必须单调递增或递减）
|
||||
/// * `y` - 原始 y 值数组
|
||||
/// * `xx` - 新的 x 坐标数组（待插值点）
|
||||
/// * `npol` - 插值阶数（使用 NPOL 个点进行 (NPOL-1) 阶插值）
|
||||
/// * `ilogx` - 非零时对 x 坐标进行对数插值
|
||||
/// * `ilogy` - 非零时对 y 值进行对数插值
|
||||
///
|
||||
/// # 返回值
|
||||
///
|
||||
/// 插值后的 y 值数组，与 `xx` 等长
|
||||
pub fn interp(x: &[f64], y: &[f64], xx: &[f64], npol: i32, ilogx: i32, ilogy: i32) -> Vec<f64> {
|
||||
let nx = x.len();
|
||||
let nxx = xx.len();
|
||||
let n = nx.max(nxx);
|
||||
|
||||
// NPOL <= 0 或 NX <= 0 的情况：直接复制
|
||||
if npol <= 0 || nx == 0 {
|
||||
let mut yy = vec![0.0; n];
|
||||
for i in 0..n {
|
||||
yy[i] = if i < nx { x[i] } else { 0.0 };
|
||||
}
|
||||
return yy;
|
||||
}
|
||||
|
||||
// 如果需要对数插值，转换坐标
|
||||
let x_work: Vec<f64> = if ilogx != 0 {
|
||||
x.iter().map(|v| v.log10()).collect()
|
||||
} else {
|
||||
x.to_vec()
|
||||
};
|
||||
|
||||
let xx_work: Vec<f64> = if ilogx != 0 {
|
||||
xx.iter().map(|v| v.log10()).collect()
|
||||
} else {
|
||||
xx.to_vec()
|
||||
};
|
||||
|
||||
let y_work: Vec<f64> = if ilogy != 0 {
|
||||
y.iter().map(|v| v.log10()).collect()
|
||||
} else {
|
||||
y.to_vec()
|
||||
};
|
||||
|
||||
let npol = npol as usize;
|
||||
let npol_usize = npol;
|
||||
let nm = npol_usize.div_ceil(2);
|
||||
// Fortran: nm1=nm+1, nup=NX+NM1-NPOL, loop II=NM1..NUP-1
|
||||
// In 0-based: nm1=nm, nup=nx-npol+nm+1, loop ii=nm..nup-1
|
||||
let nm1 = nm;
|
||||
let nup = nx + nm1 - npol_usize + 1;
|
||||
|
||||
let mut yy = vec![0.0; nxx];
|
||||
|
||||
for id in 0..nxx {
|
||||
let xxx = xx_work[id];
|
||||
|
||||
// 查找插值区间
|
||||
// Fortran: DO II=NM1,NUP-1; IF(XXX.LE.X(II)) I=II
|
||||
// 0-based: ii goes from nm1 to nup-1, compare with x_work[ii]
|
||||
let mut i = nup.saturating_sub(1);
|
||||
for ii in nm1..nup {
|
||||
if ii < nx && xxx <= x_work[ii] {
|
||||
i = ii;
|
||||
break;
|
||||
}
|
||||
}
|
||||
// Clamp to valid range for interpolation
|
||||
if npol_usize > nx {
|
||||
i = 0;
|
||||
} else if i + npol_usize > nx {
|
||||
i = nx - npol_usize;
|
||||
}
|
||||
|
||||
let j = i.saturating_sub(nm);
|
||||
let jj = (j + npol_usize).min(nx) - 1;
|
||||
|
||||
// Lagrange 插值 (j 从 0 开始，使用 0-based 索引)
|
||||
let mut yyy = 0.0;
|
||||
for k in j..=jj {
|
||||
let mut t = 1.0;
|
||||
for m in j..=jj {
|
||||
if k != m {
|
||||
let denom = x_work[k] - x_work[m];
|
||||
if denom.abs() > 1e-30 {
|
||||
t *= (xxx - x_work[m]) / denom;
|
||||
}
|
||||
}
|
||||
}
|
||||
yyy += y_work[k] * t;
|
||||
}
|
||||
|
||||
yy[id] = yyy;
|
||||
}
|
||||
|
||||
// 如果使用了对数插值，转换回线性尺度
|
||||
if ilogy != 0 {
|
||||
for val in yy.iter_mut() {
|
||||
*val = 10.0_f64.powf(*val);
|
||||
}
|
||||
}
|
||||
|
||||
yy
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use approx::assert_relative_eq;
|
||||
|
||||
#[test]
|
||||
fn test_interp_linear() {
|
||||
// 线性数据：y = 2x + 1
|
||||
let x = [0.0, 1.0, 2.0, 3.0];
|
||||
let y = [1.0, 3.0, 5.0, 7.0];
|
||||
let xx = [0.5, 1.5, 2.5];
|
||||
|
||||
let yy = interp(&x, &y, &xx, 2, 0, 0);
|
||||
|
||||
assert_relative_eq!(yy[0], 2.0, epsilon = 1e-10);
|
||||
assert_relative_eq!(yy[1], 4.0, epsilon = 1e-10);
|
||||
assert_relative_eq!(yy[2], 6.0, epsilon = 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_interp_quadratic() {
|
||||
// 二次数据：y = x^2
|
||||
let x = [0.0, 1.0, 2.0, 3.0, 4.0];
|
||||
let y = [0.0, 1.0, 4.0, 9.0, 16.0];
|
||||
let xx = [1.5];
|
||||
|
||||
// 3 阶插值（2 阶多项式）
|
||||
let yy = interp(&x, &y, &xx, 3, 0, 0);
|
||||
|
||||
assert_relative_eq!(yy[0], 2.25, epsilon = 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_interp_exact_points() {
|
||||
// 插值到已知数据点应返回精确值
|
||||
let x = [1.0, 2.0, 3.0];
|
||||
let y = [10.0, 20.0, 30.0];
|
||||
let xx = [2.0];
|
||||
|
||||
let yy = interp(&x, &y, &xx, 2, 0, 0);
|
||||
|
||||
assert_relative_eq!(yy[0], 20.0, epsilon = 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_interp_log_mode() {
|
||||
// 对数插值模式
|
||||
let x = [1.0, 10.0, 100.0];
|
||||
let y = [1.0, 100.0, 10000.0];
|
||||
let xx = [5.0];
|
||||
|
||||
let yy = interp(&x, &y, &xx, 2, 1, 1);
|
||||
|
||||
// 在对数空间中，log(y) = 2*log(x)，所以 y = x^2
|
||||
// x=5 -> y=25
|
||||
assert_relative_eq!(yy[0], 25.0, epsilon = 0.1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_interp_zero_npol() {
|
||||
let x = [1.0, 2.0];
|
||||
let y = [3.0, 4.0];
|
||||
let xx = [5.0];
|
||||
|
||||
let yy = interp(&x, &y, &xx, 0, 0, 0);
|
||||
|
||||
// NPOL <= 0 时返回 x 的值
|
||||
assert_eq!(yy[0], 1.0);
|
||||
}
|
||||
}
|
||||
173
src/synspec/math/inthe2.rs
Normal file
173
src/synspec/math/inthe2.rs
Normal file
@ -0,0 +1,173 @@
|
||||
//! Interpolation in He II Stark broadening tables.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine INTHE2(W0,X0,Z0,IWL,ILINE)
|
||||
//! at line 12507.
|
||||
//!
|
||||
//! Performs 2D interpolation in temperature and electron density from the
|
||||
//! Schoening and Butler tables for He II lines. Falls back to approximate
|
||||
//! expressions (DIVHE2 + STARKA) when outside the table bounds.
|
||||
|
||||
use super::divhe2::divhe2;
|
||||
use super::starka::starka;
|
||||
use super::yint::yint;
|
||||
|
||||
/// Parameters for He II table interpolation.
|
||||
pub struct Inthe2Params<'a> {
|
||||
/// Temperature (log scale)
|
||||
pub x0: f64,
|
||||
/// Electron density (log scale)
|
||||
pub z0: f64,
|
||||
/// Wavelength index
|
||||
pub iwl: usize,
|
||||
/// Number of wavelength points
|
||||
pub nwl2: usize,
|
||||
/// Number of temperature grid points
|
||||
pub nt2: usize,
|
||||
/// Number of electron density grid points
|
||||
pub ne2: usize,
|
||||
/// Wavelength array (nwl)
|
||||
pub wl2: &'a [f64],
|
||||
/// FXK constant
|
||||
pub fxk: f64,
|
||||
/// Temperature grid (nt)
|
||||
pub xt2: &'a [f64],
|
||||
/// Electron density grid (ne)
|
||||
pub xne2: &'a [f64],
|
||||
/// Profile data (nwl x nt x ne)
|
||||
pub prf2: &'a [&'a [f64]],
|
||||
/// Doppler width in beta units
|
||||
pub betad: f64,
|
||||
/// Delta beta for profile normalization
|
||||
pub dbeta: f64,
|
||||
}
|
||||
|
||||
/// Result of He II table interpolation.
|
||||
pub struct Inthe2Result {
|
||||
/// Interpolated profile value (log10)
|
||||
pub w0: f64,
|
||||
}
|
||||
|
||||
/// Interpolation in He II Stark broadening tables.
|
||||
///
|
||||
/// Performs bilinear interpolation in temperature and electron density
|
||||
/// from precomputed He II line broadening tables. Falls back to
|
||||
/// approximate Stark profile expressions when outside table bounds.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Interpolation parameters
|
||||
///
|
||||
/// # Returns
|
||||
/// Interpolated profile value (log10 scale).
|
||||
pub fn inthe2(params: &Inthe2Params) -> Inthe2Result {
|
||||
let nx = 3usize;
|
||||
let nz = 3usize;
|
||||
let beta = params.wl2[params.iwl] / params.fxk;
|
||||
|
||||
// Fallback: approximate expression for out-of-range electron density
|
||||
if params.z0 < params.xne2[0] * 0.99
|
||||
|| params.z0 > params.xne2[params.ne2 - 1] * 1.01
|
||||
{
|
||||
let div = divhe2(beta);
|
||||
let w0_val = starka(beta, params.betad, beta, div, 1.0) * params.dbeta;
|
||||
return Inthe2Result {
|
||||
w0: w0_val.log10(),
|
||||
};
|
||||
}
|
||||
|
||||
// Find electron density interval
|
||||
let mut ipz = 0;
|
||||
for izz in 0..params.ne2 - 1 {
|
||||
ipz = izz;
|
||||
if params.z0 <= params.xne2[izz + 1] {
|
||||
break;
|
||||
}
|
||||
}
|
||||
let mut n0z = (ipz + 1).saturating_sub(nz / 2);
|
||||
if n0z > params.ne2 - nz {
|
||||
n0z = params.ne2 - nz;
|
||||
}
|
||||
let n1z = n0z + nz;
|
||||
|
||||
let mut zz = [0.0f64; 3];
|
||||
let mut wz = [0.0f64; 3];
|
||||
|
||||
for izz in n0z..n1z {
|
||||
let i0z = izz - n0z;
|
||||
zz[i0z] = params.xne2[izz];
|
||||
|
||||
// Fallback for high temperature with large Doppler width
|
||||
if params.x0 > 1.01 * params.xt2[params.nt2 - 1] && params.betad > 10.0 {
|
||||
let div = divhe2(beta);
|
||||
let w0_val = starka(beta, params.betad, beta, div, 1.0) * params.dbeta;
|
||||
return Inthe2Result {
|
||||
w0: w0_val.log10(),
|
||||
};
|
||||
}
|
||||
|
||||
// Find temperature interval
|
||||
let mut ipx = 0;
|
||||
for ix in 0..params.nt2 - 1 {
|
||||
ipx = ix;
|
||||
if params.x0 <= params.xt2[ix + 1] {
|
||||
break;
|
||||
}
|
||||
}
|
||||
let mut n0x = (ipx + 1).saturating_sub(nx / 2);
|
||||
if n0x > params.nt2 - nx {
|
||||
n0x = params.nt2 - nx;
|
||||
}
|
||||
let n1x = n0x + nx;
|
||||
|
||||
let mut xx = [0.0f64; 3];
|
||||
let mut wx = [0.0f64; 3];
|
||||
|
||||
for ix in n0x..n1x {
|
||||
let i0 = ix - n0x;
|
||||
xx[i0] = params.xt2[ix];
|
||||
wx[i0] = params.prf2[ix][izz];
|
||||
}
|
||||
|
||||
wz[i0z] = yint(&xx, &wx, params.x0);
|
||||
}
|
||||
|
||||
let w0 = yint(&zz, &wz, params.z0);
|
||||
Inthe2Result { w0 }
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_inthe2_basic() {
|
||||
// Simple 3x3 grid test
|
||||
let wl2 = [4686.0];
|
||||
let xt2 = [3.0, 4.0, 5.0];
|
||||
let xne2 = [10.0, 11.0, 12.0];
|
||||
|
||||
// prf2[ix][izz]
|
||||
let row0: &[f64] = &[1.0, 2.0, 3.0];
|
||||
let row1: &[f64] = &[2.0, 3.0, 4.0];
|
||||
let row2: &[f64] = &[3.0, 4.0, 5.0];
|
||||
let prf2: &[&[f64]] = &[row0, row1, row2];
|
||||
|
||||
let params = Inthe2Params {
|
||||
x0: 3.5,
|
||||
z0: 10.5,
|
||||
iwl: 0,
|
||||
nwl2: 1,
|
||||
nt2: 3,
|
||||
ne2: 3,
|
||||
wl2: &wl2,
|
||||
fxk: 1.0,
|
||||
xt2: &xt2,
|
||||
xne2: &xne2,
|
||||
prf2,
|
||||
betad: 5.0,
|
||||
dbeta: 1.0,
|
||||
};
|
||||
|
||||
let result = inthe2(¶ms);
|
||||
assert!(result.w0.is_finite());
|
||||
}
|
||||
}
|
||||
202
src/synspec/math/inthyd.rs
Normal file
202
src/synspec/math/inthyd.rs
Normal file
@ -0,0 +1,202 @@
|
||||
//! Interpolation in hydrogen Stark broadening tables.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine INTHYD(W0,X0,Z0,IWL,ILINE)
|
||||
//! at line 12412.
|
||||
//!
|
||||
//! Performs 2D interpolation in temperature and electron density from the
|
||||
//! hydrogen line broadening tables. Falls back to approximate expressions
|
||||
//! (DIVSTR + STARKA) when outside the table bounds.
|
||||
|
||||
use super::divstr::divstr;
|
||||
use super::starka::starka;
|
||||
use super::yint::yint;
|
||||
|
||||
/// Parameters for hydrogen table interpolation.
|
||||
pub struct InthydParams<'a> {
|
||||
/// Temperature (log scale)
|
||||
pub x0: f64,
|
||||
/// Electron density (log scale)
|
||||
pub z0: f64,
|
||||
/// Wavelength index
|
||||
pub iwl: usize,
|
||||
/// Line index
|
||||
pub iline: usize,
|
||||
/// Number of temperature grid points per line
|
||||
pub nth: &'a [usize],
|
||||
/// Number of electron density grid points per line
|
||||
pub neh: &'a [usize],
|
||||
/// Wavelength array (nwl x nlines)
|
||||
pub wl: &'a [&'a [f64]],
|
||||
/// FXK constant
|
||||
pub fxk: f64,
|
||||
/// XK constant (for Lemke mode)
|
||||
pub xk: f64,
|
||||
/// Lemke mode flag (0 = standard, 1 = Lemke)
|
||||
pub ilemke: i32,
|
||||
/// Electron density grid (ne x nlines)
|
||||
pub xne: &'a [&'a [f64]],
|
||||
/// Temperature grid (nt x nlines)
|
||||
pub xt: &'a [&'a [f64]],
|
||||
/// Profile data (nwl x nt x ne x nlines) — pass slice for current line
|
||||
pub prf_iwl: &'a [&'a [f64]],
|
||||
/// Doppler width in beta units
|
||||
pub betad: f64,
|
||||
/// Delta beta for profile normalization
|
||||
pub dbeta: f64,
|
||||
}
|
||||
|
||||
/// Result of hydrogen table interpolation.
|
||||
pub struct InthydResult {
|
||||
/// Interpolated profile value (log10)
|
||||
pub w0: f64,
|
||||
}
|
||||
|
||||
/// Interpolation in hydrogen Stark broadening tables.
|
||||
///
|
||||
/// Performs bilinear interpolation in temperature and electron density
|
||||
/// from precomputed hydrogen line broadening tables. Falls back to
|
||||
/// approximate Stark profile expressions when outside table bounds.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Interpolation parameters
|
||||
///
|
||||
/// # Returns
|
||||
/// Interpolated profile value (log10 scale).
|
||||
pub fn inthyd(params: &InthydParams) -> InthydResult {
|
||||
let iline = params.iline;
|
||||
let nt = params.nth[iline];
|
||||
let ne = params.neh[iline];
|
||||
|
||||
let (beta, nx, nz) = if params.ilemke == 1 {
|
||||
(params.wl[params.iwl][iline] / params.xk, 2, 2)
|
||||
} else {
|
||||
(params.wl[params.iwl][iline] / params.fxk, 3, 3)
|
||||
};
|
||||
|
||||
// Fallback: approximate expression for out-of-range electron density
|
||||
if params.z0 < params.xne[iline][0] * 0.99
|
||||
|| params.z0 > params.xne[iline][ne - 1] * 1.01
|
||||
{
|
||||
let (a, div) = divstr(beta);
|
||||
let w0_val = starka(beta, params.betad, a, div, 2.0) * params.dbeta;
|
||||
return InthydResult {
|
||||
w0: w0_val.log10(),
|
||||
};
|
||||
}
|
||||
|
||||
// Find electron density interval
|
||||
let mut ipz = 0;
|
||||
for izz in 0..ne - 1 {
|
||||
ipz = izz;
|
||||
if params.z0 <= params.xne[iline][izz + 1] {
|
||||
break;
|
||||
}
|
||||
}
|
||||
let mut n0z = (ipz + 1).saturating_sub(nz / 2);
|
||||
if n0z > ne - nz {
|
||||
n0z = ne - nz;
|
||||
}
|
||||
let n1z = n0z + nz;
|
||||
|
||||
let mut zz = [0.0f64; 3];
|
||||
let mut wz = [0.0f64; 3];
|
||||
|
||||
for izz in n0z..n1z {
|
||||
let i0z = izz - n0z;
|
||||
zz[i0z] = params.xne[iline][izz];
|
||||
|
||||
// Fallback for high temperature with large Doppler width
|
||||
if params.x0 > 1.01 * params.xt[iline][nt - 1] && params.betad > 10.0 {
|
||||
let (a, div) = divstr(beta);
|
||||
let w0_val = starka(beta, params.betad, a, div, 2.0) * params.dbeta;
|
||||
return InthydResult {
|
||||
w0: w0_val.log10(),
|
||||
};
|
||||
}
|
||||
|
||||
// Find temperature interval
|
||||
let mut ipx = 0;
|
||||
for ix in 0..nt - 1 {
|
||||
ipx = ix;
|
||||
if params.x0 <= params.xt[iline][ix + 1] {
|
||||
break;
|
||||
}
|
||||
}
|
||||
let mut n0x = (ipx + 1).saturating_sub(nx / 2);
|
||||
if n0x > nt - nx {
|
||||
n0x = nt - nx;
|
||||
}
|
||||
let n1x = n0x + nx;
|
||||
|
||||
let mut xx = [0.0f64; 3];
|
||||
let mut wx = [0.0f64; 3];
|
||||
|
||||
let mut has_bad = false;
|
||||
for ix in n0x..n1x {
|
||||
let i0 = ix - n0x;
|
||||
xx[i0] = params.xt[iline][ix];
|
||||
wx[i0] = params.prf_iwl[ix][izz];
|
||||
if wx[i0] < -99.0 {
|
||||
has_bad = true;
|
||||
}
|
||||
}
|
||||
|
||||
if has_bad {
|
||||
let (a, div) = divstr(beta);
|
||||
let w0_val = starka(beta, params.betad, a, div, 2.0) * params.dbeta;
|
||||
return InthydResult {
|
||||
w0: w0_val.log10(),
|
||||
};
|
||||
}
|
||||
|
||||
wz[i0z] = yint(&xx, &wx, params.x0);
|
||||
}
|
||||
|
||||
let w0 = yint(&zz, &wz, params.z0);
|
||||
InthydResult { w0 }
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_inthyd_basic() {
|
||||
// Simple 3x3 grid test
|
||||
let nth = [3usize];
|
||||
let neh = [3usize];
|
||||
let wl_vals = [4861.0, 4340.0, 4102.0];
|
||||
let wl: [&[f64]; 1] = [&wl_vals];
|
||||
let xne_vals = [10.0, 11.0, 12.0];
|
||||
let xne: [&[f64]; 1] = [&xne_vals];
|
||||
let xt_vals = [3.0, 4.0, 5.0];
|
||||
let xt: [&[f64]; 1] = [&xt_vals];
|
||||
|
||||
// prf_iwl[ix][izz]
|
||||
let row0: &[f64] = &[1.0, 2.0, 3.0];
|
||||
let row1: &[f64] = &[2.0, 3.0, 4.0];
|
||||
let row2: &[f64] = &[3.0, 4.0, 5.0];
|
||||
let prf_iwl: &[&[f64]] = &[row0, row1, row2];
|
||||
|
||||
let params = InthydParams {
|
||||
x0: 3.5,
|
||||
z0: 10.5,
|
||||
iwl: 0,
|
||||
iline: 0,
|
||||
nth: &nth,
|
||||
neh: &neh,
|
||||
wl: &wl,
|
||||
fxk: 1.0,
|
||||
xk: 1.0,
|
||||
ilemke: 0,
|
||||
xne: &xne,
|
||||
xt: &xt,
|
||||
prf_iwl,
|
||||
betad: 5.0,
|
||||
dbeta: 1.0,
|
||||
};
|
||||
|
||||
let result = inthyd(¶ms);
|
||||
assert!(result.w0.is_finite());
|
||||
}
|
||||
}
|
||||
155
src/synspec/math/intxen.rs
Normal file
155
src/synspec/math/intxen.rs
Normal file
@ -0,0 +1,155 @@
|
||||
//! Interpolation in Xenomorph tables for hydrogen lines.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine INTXEN(W0B,W0R,X0,Z0,IWL,ILINE)
|
||||
//! at line 11631.
|
||||
//!
|
||||
//! Performs 2D interpolation in temperature and electron density from the
|
||||
//! Xenomorph line profile tables to the actual values of temperature (X0)
|
||||
//! and electron density (Z0).
|
||||
|
||||
use super::yint::yint;
|
||||
|
||||
/// Parameters for Xenomorph table interpolation.
|
||||
pub struct IntxenParams<'a> {
|
||||
/// Temperature (log scale)
|
||||
pub x0: f64,
|
||||
/// Electron density (log scale)
|
||||
pub z0: f64,
|
||||
/// Wavelength index
|
||||
pub iwl: usize,
|
||||
/// Number of temperature grid points
|
||||
pub nt: usize,
|
||||
/// Number of electron density grid points
|
||||
pub ne: usize,
|
||||
/// Temperature grid (nt)
|
||||
pub xtxen: &'a [f64],
|
||||
/// Electron density grid (ne)
|
||||
pub xnenex: &'a [f64],
|
||||
/// Blue profile data (nwl x nt x ne) — pass the slice for iwl
|
||||
pub prfxb_iwl: &'a [&'a [f64]],
|
||||
/// Red profile data (nwl x nt x ne) — pass the slice for iwl
|
||||
pub prfxr_iwl: &'a [&'a [f64]],
|
||||
}
|
||||
|
||||
/// Result of Xenomorph table interpolation.
|
||||
pub struct IntxenResult {
|
||||
/// Blue wing profile value
|
||||
pub w0b: f64,
|
||||
/// Red wing profile value
|
||||
pub w0r: f64,
|
||||
}
|
||||
|
||||
/// Interpolation in Xenomorph tables for hydrogen lines.
|
||||
///
|
||||
/// Performs bilinear interpolation in temperature and electron density
|
||||
/// from the Xenomorph line profile tables. Uses quadratic interpolation
|
||||
/// (NX=NZ=2 points) around the target values.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Interpolation parameters
|
||||
///
|
||||
/// # Returns
|
||||
/// Interpolated blue and red wing profile values.
|
||||
pub fn intxen(params: &IntxenParams) -> IntxenResult {
|
||||
let nt = params.nt;
|
||||
let ne = params.ne;
|
||||
let nx = 2usize;
|
||||
let nz = 2usize;
|
||||
|
||||
// Find electron density interval
|
||||
let mut ipz = 0;
|
||||
for izz in 0..ne - 1 {
|
||||
ipz = izz;
|
||||
if params.z0 <= params.xnenex[izz + 1] {
|
||||
break;
|
||||
}
|
||||
}
|
||||
let mut n0z = (ipz + 1).saturating_sub(nz / 2);
|
||||
if n0z > ne - nz {
|
||||
n0z = ne - nz;
|
||||
}
|
||||
let n1z = n0z + nz;
|
||||
|
||||
let mut zz = [0.0f64; 3];
|
||||
let mut wzb = [0.0f64; 3];
|
||||
let mut wzr = [0.0f64; 3];
|
||||
|
||||
for izz in n0z..n1z {
|
||||
let i0z = izz - n0z;
|
||||
zz[i0z] = params.xnenex[izz];
|
||||
|
||||
// Find temperature interval
|
||||
let mut ipx = 0;
|
||||
for ix in 0..nt - 1 {
|
||||
ipx = ix;
|
||||
if params.x0 <= params.xtxen[ix + 1] {
|
||||
break;
|
||||
}
|
||||
}
|
||||
let mut n0x = (ipx + 1).saturating_sub(nx / 2);
|
||||
if n0x > nt - nx {
|
||||
n0x = nt - nx;
|
||||
}
|
||||
let n1x = n0x + nx;
|
||||
|
||||
let mut xx = [0.0f64; 3];
|
||||
let mut wxb = [0.0f64; 3];
|
||||
let mut wxr = [0.0f64; 3];
|
||||
|
||||
for ix in n0x..n1x {
|
||||
let i0 = ix - n0x;
|
||||
xx[i0] = params.xtxen[ix];
|
||||
wxb[i0] = params.prfxb_iwl[ix][izz];
|
||||
wxr[i0] = params.prfxr_iwl[ix][izz];
|
||||
}
|
||||
|
||||
wzb[i0z] = yint(&xx, &wxb, params.x0);
|
||||
wzr[i0z] = yint(&xx, &wxr, params.x0);
|
||||
}
|
||||
|
||||
let w0b = yint(&zz, &wzb, params.z0);
|
||||
let w0r = yint(&zz, &wzr, params.z0);
|
||||
|
||||
IntxenResult { w0b, w0r }
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_intxen_basic() {
|
||||
// Simple 3x3 grid test
|
||||
let xtxen = [1.0, 2.0, 3.0];
|
||||
let xnenex = [10.0, 11.0, 12.0];
|
||||
|
||||
// prfxb_iwl[ix][izz]
|
||||
let row0: &[f64] = &[1.0, 2.0, 3.0];
|
||||
let row1: &[f64] = &[2.0, 3.0, 4.0];
|
||||
let row2: &[f64] = &[3.0, 4.0, 5.0];
|
||||
let prfxb_iwl: &[&[f64]] = &[row0, row1, row2];
|
||||
|
||||
let rrow0: &[f64] = &[5.0, 4.0, 3.0];
|
||||
let rrow1: &[f64] = &[4.0, 3.0, 2.0];
|
||||
let rrow2: &[f64] = &[3.0, 2.0, 1.0];
|
||||
let prfxr_iwl: &[&[f64]] = &[rrow0, rrow1, rrow2];
|
||||
|
||||
let params = IntxenParams {
|
||||
x0: 1.5,
|
||||
z0: 10.5,
|
||||
iwl: 0,
|
||||
nt: 3,
|
||||
ne: 3,
|
||||
xtxen: &xtxen,
|
||||
xnenex: &xnenex,
|
||||
prfxb_iwl,
|
||||
prfxr_iwl,
|
||||
};
|
||||
|
||||
let result = intxen(¶ms);
|
||||
assert!(result.w0b.is_finite());
|
||||
assert!(result.w0r.is_finite());
|
||||
assert!(result.w0b > 0.0 && result.w0b < 10.0);
|
||||
assert!(result.w0r > 0.0 && result.w0r < 10.0);
|
||||
}
|
||||
}
|
||||
281
src/synspec/math/irwpf.rs
Normal file
281
src/synspec/math/irwpf.rs
Normal file
@ -0,0 +1,281 @@
|
||||
//! Irwin (1981) partition functions, updated with Barklem & Collet (2016) data.
|
||||
//!
|
||||
//! Translated from SYNSPEC `IRWPF` subroutine (synspec54.f:23749).
|
||||
//!
|
||||
//! Computes partition functions using polynomial fits from Irwin (1981),
|
||||
//! ApJS. 45, 621, updated with Barklem & Collet (2016) data.
|
||||
//! For atomic species (jatom > 0, ion > 0): uses 6-coefficient polynomial in ln(T).
|
||||
//! For molecular species (indmol > 0): uses similar polynomial with index mapping.
|
||||
|
||||
use std::sync::Mutex;
|
||||
|
||||
/// Mapping from Tsuji molecular index to Irwin index.
|
||||
/// 478 elements (Fortran declares `dimension irwind(478)`); value 0 = no data.
|
||||
const IRWIND: [i32; 478] = [
|
||||
0, 1, 28, 4, 2, 7, 6, 5, 8, 10,
|
||||
9, 3, 18, 25, 53, 29, 43, 0, 17, 153,
|
||||
52, 55, 167, 44, 45, 182, 74, 46, 11, 187,
|
||||
201, 31, 27, 99, 209, 24, 22, 20, 21, 65,
|
||||
35, 19, 54, 23, 0, 14, 58, 0, 32, 12,
|
||||
47, 16, 0, 34, 0, 0, 30, 0, 13, 33,
|
||||
61, 63, 292, 57, 59, 66, 272, 0, 94, 175,
|
||||
226, 286, 0, 0, 0, 176, 227, 287, 0, 0,
|
||||
0, 96, 0, 177, 0, 267, 228, 288, 0, 0,
|
||||
0, 0, 93, 147, 162, 0, 0, 0, 0, 0,
|
||||
0, 50, 0, 0, 0, 0, 36, 0, 64, 0,
|
||||
0, 48, 0, 0, 148, 0, 0, 26, 49, 70,
|
||||
178, 97, 170, 229, 0, 180, 268, 230, 0, 289,
|
||||
0, 0, 15, 181, 0, 269, 0, 0, 0, 0,
|
||||
0, 0, 0, 231, 0, 290, 0, 38, 0, 0,
|
||||
152, 39, 40, 0, 41, 232, 0, 291, 0, 0,
|
||||
0, 0, 0, 75, 154, 0, 0, 0, 183, 0,
|
||||
0, 0, 0, 0, 0, 98, 184, 234, 185, 270,
|
||||
0, 0, 0, 186, 0, 0, 271, 235, 0, 0,
|
||||
62, 0, 0, 0, 0, 0, 0, 101, 0, 188,
|
||||
0, 0, 0, 0, 0, 102, 189, 0, 0, 0,
|
||||
236, 0, 294, 67, 0, 190, 0, 0, 0, 295,
|
||||
0, 0, 104, 191, 237, 0, 105, 192, 274, 238,
|
||||
296, 112, 245, 303, 113, 199, 0, 278, 246, 0,
|
||||
304, 0, 0, 0, 0, 200, 0, 0, 279, 247,
|
||||
0, 305, 0, 0, 172, 0, 0, 0, 0, 0,
|
||||
0, 120, 122, 208, 0, 282, 255, 0, 312, 0,
|
||||
0, 0, 0, 0, 0, 0, 0, 283, 256, 0,
|
||||
0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
||||
275, 194, 108, 241, 299, 202, 0, 68, 69, 71,
|
||||
72, 73, 42, 37, 76, 77, 78, 79, 80, 81,
|
||||
82, 83, 92, 95, 100, 103, 106, 107, 109, 110,
|
||||
111, 114, 115, 116, 117, 118, 119, 121, 123, 124,
|
||||
125, 126, 127, 128, 129, 149, 150, 151, 155, 156,
|
||||
157, 158, 159, 163, 164, 165, 166, 168, 169, 170,
|
||||
171, 193, 195, 196, 197, 198, 203, 204, 205, 206,
|
||||
207, 210, 211, 212, 213, 214, 215, 216, 217, 218,
|
||||
225, 233, 239, 240, 242, 243, 244, 248, 249, 250,
|
||||
251, 252, 253, 254, 257, 258, 259, 260, 262, 262,
|
||||
263, 264, 265, 266, 273, 276, 277, 280, 282, 284,
|
||||
285, 293, 297, 298, 300, 301, 302, 306, 307, 308,
|
||||
309, 310, 311, 60, 313, 314, 315, 316, 317, 318,
|
||||
319, 320, 321, 322, 323, 324, 84, 85, 86, 87,
|
||||
88, 89, 90, 91, 130, 131, 132, 133, 134, 135,
|
||||
136, 137, 138, 139, 140, 141, 142, 143, 144, 145,
|
||||
146, 160, 161, 173, 174, 210, 220, 221, 222, 223,
|
||||
224, 0, 0, 0, 0, 0, 0, 0, 0, 0,
|
||||
0, 0, 0, 0, 0, 0, 0, 56
|
||||
];
|
||||
|
||||
/// Cached Irwin data: atomic coefficients `a[ion][atom][coeff]` and
|
||||
/// molecular coefficients `am[mol_index][coeff]`.
|
||||
struct IrwinData {
|
||||
/// Atomic coefficients: a[ion * 92 * 6 + atom * 6 + coeff]
|
||||
a: Vec<f64>,
|
||||
/// Molecular coefficients: am[mol * 6 + coeff]
|
||||
am: Vec<f64>,
|
||||
}
|
||||
|
||||
static IRWIN_DATA: Mutex<Option<IrwinData>> = Mutex::new(None);
|
||||
|
||||
/// Read the Irwin data file and populate the coefficient arrays.
|
||||
fn read_irwin_data(data_dir: &str, irwtab: i32) -> Result<IrwinData, String> {
|
||||
let filename = if irwtab == 0 {
|
||||
format!("{}/irwin_orig.dat", data_dir)
|
||||
} else {
|
||||
format!("{}/irwin_bc.dat", data_dir)
|
||||
};
|
||||
|
||||
let content = std::fs::read_to_string(&filename)
|
||||
.map_err(|e| format!("Cannot open {}: {}", filename, e))?;
|
||||
let mut lines = content.lines();
|
||||
|
||||
// Skip 2 header lines
|
||||
lines.next();
|
||||
lines.next();
|
||||
|
||||
let mut a = vec![0.0f64; 6 * 3 * 92]; // a[ion * 92 * 6 + atom * 6 + coeff]
|
||||
let mut am = vec![0.0f64; 6 * 500]; // am[mol * 6 + coeff]
|
||||
|
||||
// Read atomic data: 92 atoms × 3 ions (skip j=1,i=3 in Fortran, but read all here)
|
||||
for _j in 0..92 {
|
||||
for _i in 0..3 {
|
||||
// Fortran skips j=1,i=3 (i.e. _j=0,_i=2); we just read and discard
|
||||
let line = lines.next().ok_or("Unexpected end of atomic data")?;
|
||||
if _j == 0 && _i == 2 {
|
||||
continue; // Fortran: if(j.eq.1.and.i.eq.3) goto 10
|
||||
}
|
||||
let parts: Vec<&str> = line.split_whitespace().collect();
|
||||
if parts.len() < 7 {
|
||||
continue; // skip malformed lines
|
||||
}
|
||||
for k in 0..6 {
|
||||
if let Ok(val) = parts[k + 1].parse::<f64>() {
|
||||
a[_i * 92 * 6 + _j * 6 + k] = val;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Skip 3 header lines
|
||||
lines.next();
|
||||
lines.next();
|
||||
lines.next();
|
||||
|
||||
// Read molecular data (up to 324 entries)
|
||||
for idx in 0..324 {
|
||||
match lines.next() {
|
||||
Some(line) => {
|
||||
let parts: Vec<&str> = line.split_whitespace().collect();
|
||||
if parts.len() < 7 {
|
||||
continue;
|
||||
}
|
||||
for k in 0..6 {
|
||||
if let Ok(val) = parts[k + 1].parse::<f64>() {
|
||||
am[idx * 6 + k] = val;
|
||||
}
|
||||
}
|
||||
}
|
||||
None => break,
|
||||
}
|
||||
}
|
||||
|
||||
Ok(IrwinData { a, am })
|
||||
}
|
||||
|
||||
/// Evaluate 6-coefficient polynomial in ln(T).
|
||||
/// `coeffs[0] + coeffs[1]*tl + coeffs[2]*tl^2 + ... + coeffs[5]*tl^5`
|
||||
fn poly6(coeffs: &[f64], tl: f64) -> f64 {
|
||||
coeffs[0]
|
||||
+ tl * (coeffs[1]
|
||||
+ tl * (coeffs[2]
|
||||
+ tl * (coeffs[3]
|
||||
+ tl * (coeffs[4]
|
||||
+ tl * coeffs[5]))))
|
||||
}
|
||||
|
||||
/// Irwin (1981) partition functions.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `jatom` - Atomic number (1-92). If 0, compute molecular partition function.
|
||||
/// * `ion` - Ionization degree (1=neutral, 2=singly ionized, 3=doubly ionized).
|
||||
/// * `indmol` - Molecular index in Tsuji numbering (used when jatom == 0).
|
||||
/// * `t` - Temperature (K). Must be in range [1000, 16000].
|
||||
/// * `data_dir` - Path to data directory containing irwin_*.dat files.
|
||||
/// * `irwtab` - Table selector: 0 = original Irwin, other = Barklem & Collet.
|
||||
///
|
||||
/// # Returns
|
||||
/// Partition function value, or error string.
|
||||
pub fn irwpf(
|
||||
jatom: i32,
|
||||
ion: i32,
|
||||
indmol: i32,
|
||||
t: f64,
|
||||
data_dir: &str,
|
||||
irwtab: i32,
|
||||
) -> Result<f64, String> {
|
||||
// Early return: no species specified
|
||||
if jatom <= 0 && indmol <= 0 {
|
||||
return Ok(0.0);
|
||||
}
|
||||
|
||||
// Validate temperature
|
||||
if t < 1000.0 {
|
||||
return Err("irwpf: temperature < 1000 K".to_string());
|
||||
}
|
||||
if t > 16000.0 {
|
||||
return Err("irwpf: temperature > 16000 K".to_string());
|
||||
}
|
||||
|
||||
// Initialize data on first call
|
||||
{
|
||||
let mut guard = IRWIN_DATA.lock().map_err(|e| format!("Lock error: {}", e))?;
|
||||
if guard.is_none() {
|
||||
*guard = Some(read_irwin_data(data_dir, irwtab)?);
|
||||
}
|
||||
}
|
||||
let guard = IRWIN_DATA.lock().map_err(|e| format!("Lock error: {}", e))?;
|
||||
let data = guard.as_ref().unwrap();
|
||||
|
||||
let tl = t.ln();
|
||||
|
||||
// Atomic species
|
||||
if jatom > 0 && ion > 0 {
|
||||
let atom_idx = (jatom - 1) as usize;
|
||||
let ion_idx = (ion - 1) as usize;
|
||||
if atom_idx >= 92 || ion_idx >= 3 {
|
||||
return Ok(0.0);
|
||||
}
|
||||
let base = ion_idx * 92 * 6 + atom_idx * 6;
|
||||
let coeffs = &data.a[base..base + 6];
|
||||
let mut ulog = poly6(coeffs, tl);
|
||||
// Special case: Boron III
|
||||
if jatom == 5 && ion == 3 {
|
||||
ulog = 1.0;
|
||||
}
|
||||
return Ok(ulog.exp());
|
||||
}
|
||||
|
||||
// Molecular species
|
||||
if indmol > 0 {
|
||||
let mol_idx = (indmol - 1) as usize;
|
||||
if mol_idx >= 478 {
|
||||
return Ok(0.0);
|
||||
}
|
||||
let indm = IRWIND[mol_idx];
|
||||
if indm <= 0 {
|
||||
return Ok(0.0);
|
||||
}
|
||||
let idx = (indm - 1) as usize;
|
||||
if idx >= 500 {
|
||||
return Ok(0.0);
|
||||
}
|
||||
let base = idx * 6;
|
||||
let coeffs = &data.am[base..base + 6];
|
||||
let ulog = poly6(coeffs, tl);
|
||||
return Ok(ulog.exp());
|
||||
}
|
||||
|
||||
Ok(0.0)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_irwind_count() {
|
||||
assert_eq!(IRWIND.len(), 478);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_irwind_mapping() {
|
||||
assert_eq!(IRWIND[0], 0);
|
||||
assert_eq!(IRWIND[1], 1);
|
||||
assert_eq!(IRWIND[2], 28);
|
||||
assert_eq!(IRWIND[4], 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_poly6() {
|
||||
let coeffs = [1.0, 0.0, 0.0, 0.0, 0.0, 0.0];
|
||||
assert!((poly6(&coeffs, 5.0) - 1.0).abs() < 1e-15);
|
||||
|
||||
let coeffs = [0.0, 1.0, 0.0, 0.0, 0.0, 0.0];
|
||||
assert!((poly6(&coeffs, 3.0) - 3.0).abs() < 1e-15);
|
||||
|
||||
// Horner form: 1 + 2x + 3x^2 at x=2 = 1 + 4 + 12 = 17
|
||||
let coeffs = [1.0, 2.0, 3.0, 0.0, 0.0, 0.0];
|
||||
assert!((poly6(&coeffs, 2.0) - 17.0).abs() < 1e-15);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_irwpf_out_of_range() {
|
||||
let result = irwpf(1, 1, 0, 500.0, "./data", 1);
|
||||
assert!(result.is_err());
|
||||
|
||||
let result = irwpf(1, 1, 0, 20000.0, "./data", 1);
|
||||
assert!(result.is_err());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_irwpf_zero_atom_zero_mol() {
|
||||
let result = irwpf(0, 0, 0, 5000.0, "./data", 1).unwrap();
|
||||
assert_eq!(result, 0.0);
|
||||
}
|
||||
}
|
||||
@ -76,7 +76,7 @@ pub fn ispec(iat: i32, ion: i32, alam: f64, ihe1pr: i32, ihe2pr: i32) -> i32 {
|
||||
} else {
|
||||
// He II
|
||||
// 波长范围检查
|
||||
if alam < 163.0 || alam > 1012.7 {
|
||||
if !(163.0..=1012.7).contains(&alam) {
|
||||
return PROFILE_VOIGT;
|
||||
}
|
||||
|
||||
|
||||
152
src/synspec/math/levsol.rs
Normal file
152
src/synspec/math/levsol.rs
Normal file
@ -0,0 +1,152 @@
|
||||
//! Level population solver by partial rate matrix inversion.
|
||||
//!
|
||||
//! Translated from SYNSPEC `LEVSOL` subroutine (synspec54.f:11326).
|
||||
//!
|
||||
//! Solves for new populations by inverting several partial rate matrices
|
||||
//! for individual chemical species. For each atom, extracts the sub-matrix
|
||||
//! and sub-vector corresponding to that atom's levels, solves the linear
|
||||
//! system, and writes the results back.
|
||||
|
||||
use super::lineqs;
|
||||
|
||||
/// Solve for level populations by partial rate matrix inversion per atom.
|
||||
///
|
||||
/// For each chemical species (atom), extracts the sub-matrix of the rate
|
||||
/// equations corresponding to that atom's levels (from `n0a[iat]` to
|
||||
/// `nka[iat]`), solves the linear system, and writes results into `popp`.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `a` - Rate matrix (nlevel × nlevel, row-major flat storage)
|
||||
/// * `b` - Right-hand side vector (length nlevel)
|
||||
/// * `popp` - Output population vector (length nlevel)
|
||||
/// * `nlvcal` - Number of levels to solve (if ≤ 0, returns immediately)
|
||||
/// * `natom` - Number of chemical species
|
||||
/// * `n0a` - First level index for each atom (1-indexed, length natom)
|
||||
/// * `nka` - Last level index for each atom (1-indexed, length natom)
|
||||
/// * `nlevel` - Leading dimension of matrix A (stride)
|
||||
pub fn levsol(
|
||||
a: &[f64],
|
||||
b: &[f64],
|
||||
popp: &mut [f64],
|
||||
nlvcal: usize,
|
||||
natom: usize,
|
||||
n0a: &[usize],
|
||||
nka: &[usize],
|
||||
nlevel: usize,
|
||||
) {
|
||||
if nlvcal == 0 {
|
||||
return;
|
||||
}
|
||||
|
||||
for iat in 0..natom {
|
||||
// n0a/nka are 1-indexed from Fortran; convert to 0-indexed
|
||||
let mut n1 = if n0a[iat] > 0 { n0a[iat] - 1 } else { 0 };
|
||||
let nk = if nka[iat] > 0 { nka[iat] - 1 } else { 0 };
|
||||
|
||||
// If n1 <= 0 (i.e. n0a was 0), find first valid level
|
||||
if n0a[iat] == 0 {
|
||||
let mut found = false;
|
||||
for i in n0a[iat]..=nka[iat] {
|
||||
if i > 0 {
|
||||
n1 = i - 1;
|
||||
found = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if !found {
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
if n1 > nk {
|
||||
continue;
|
||||
}
|
||||
|
||||
let nlp = nk - n1 + 1;
|
||||
|
||||
// Extract sub-matrix AP and sub-vector BP
|
||||
let mut ap = vec![0.0f64; nlp * nlp];
|
||||
let mut bp = vec![0.0f64; nlp];
|
||||
|
||||
for i in n1..=nk {
|
||||
for j in n1..=nk {
|
||||
ap[(i - n1) * nlp + (j - n1)] = a[i * nlevel + j];
|
||||
}
|
||||
bp[i - n1] = b[i];
|
||||
}
|
||||
|
||||
// Solve the sub-system
|
||||
let popp1 = lineqs(&mut ap, &mut bp, nlp);
|
||||
|
||||
// Write results back
|
||||
for i in n1..=nk {
|
||||
popp[i] = popp1[i - n1];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_levsol_single_atom_2x2() {
|
||||
// One atom with 2 levels (indices 0,1 in 0-based)
|
||||
// System: 2*x0 + 1*x1 = 5, 1*x0 + 3*x1 = 7
|
||||
// Solution: x0=1.6, x1=1.8
|
||||
let nlevel = 2;
|
||||
let mut a = vec![0.0; nlevel * nlevel];
|
||||
a[0 * 2 + 0] = 2.0;
|
||||
a[0 * 2 + 1] = 1.0;
|
||||
a[1 * 2 + 0] = 1.0;
|
||||
a[1 * 2 + 1] = 3.0;
|
||||
let b = vec![5.0, 7.0];
|
||||
let mut popp = vec![0.0; 2];
|
||||
|
||||
// n0a=1, nka=2 (1-indexed)
|
||||
levsol(&a, &b, &mut popp, 2, 1, &[1], &[2], nlevel);
|
||||
|
||||
assert!((popp[0] - 1.6).abs() < 1e-10);
|
||||
assert!((popp[1] - 1.8).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_levsol_two_atoms() {
|
||||
// Two atoms, each with 2 levels, total 4 levels
|
||||
// Atom 0: levels 0-1, Atom 1: levels 2-3
|
||||
// Each sub-system is independent
|
||||
let nlevel = 4;
|
||||
let mut a = vec![0.0; nlevel * nlevel];
|
||||
// Atom 0 sub-matrix (rows/cols 0-1)
|
||||
a[0 * 4 + 0] = 2.0;
|
||||
a[0 * 4 + 1] = 1.0;
|
||||
a[1 * 4 + 0] = 1.0;
|
||||
a[1 * 4 + 1] = 3.0;
|
||||
// Atom 1 sub-matrix (rows/cols 2-3)
|
||||
a[2 * 4 + 2] = 4.0;
|
||||
a[2 * 4 + 3] = 1.0;
|
||||
a[3 * 4 + 2] = 1.0;
|
||||
a[3 * 4 + 3] = 5.0;
|
||||
let b = vec![5.0, 7.0, 9.0, 11.0];
|
||||
let mut popp = vec![0.0; 4];
|
||||
|
||||
// n0a=[1,3], nka=[2,4] (1-indexed)
|
||||
levsol(&a, &b, &mut popp, 4, 2, &[1, 3], &[2, 4], nlevel);
|
||||
|
||||
// Atom 0: same as above
|
||||
assert!((popp[0] - 1.6).abs() < 1e-10);
|
||||
assert!((popp[1] - 1.8).abs() < 1e-10);
|
||||
// Atom 1: 4*x2 + x3 = 9, x2 + 5*x3 = 11
|
||||
// x2 = (9 - 11/5) / (4 - 1/5) = (34/5) / (19/5) = 34/19 ≈ 1.7895
|
||||
// x3 = (11 - x2) / 5
|
||||
assert!((popp[2] - 34.0 / 19.0).abs() < 1e-10);
|
||||
assert!((popp[3] - (11.0 - 34.0 / 19.0) / 5.0).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_levsol_nlvcal_zero() {
|
||||
let mut popp = vec![0.0; 2];
|
||||
levsol(&[1.0; 4], &[1.0; 2], &mut popp, 0, 1, &[1], &[2], 2);
|
||||
assert_eq!(popp, vec![0.0, 0.0]);
|
||||
}
|
||||
}
|
||||
157
src/synspec/math/lineqs.rs
Normal file
157
src/synspec/math/lineqs.rs
Normal file
@ -0,0 +1,157 @@
|
||||
//! Linear equation solver by Gaussian elimination with partial pivoting.
|
||||
//!
|
||||
//! Translated from SYNSPEC `LINEQS` subroutine (synspec54.f:14818).
|
||||
//!
|
||||
//! Solves the linear system A*X = B using Gaussian elimination
|
||||
//! with partial pivoting. Note: matrix A and vector B are destroyed.
|
||||
|
||||
/// Solve linear system A*X = B by Gaussian elimination with partial pivoting.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `a` - Matrix of the linear system (n x n, stored as flat array row-major).
|
||||
/// Will be modified in place (LU decomposition).
|
||||
/// * `b` - Right-hand side vector (length n). Will be modified in place.
|
||||
/// * `n` - Number of equations
|
||||
///
|
||||
/// # Returns
|
||||
/// Solution vector X of length n.
|
||||
pub fn lineqs(a: &mut [f64], b: &mut [f64], n: usize) -> Vec<f64> {
|
||||
assert!(n > 0, "LINEQS requires n > 0");
|
||||
assert!(a.len() >= n * n, "a array too short");
|
||||
assert!(b.len() >= n, "b array too short");
|
||||
|
||||
let mut x = vec![0.0f64; n];
|
||||
let idx = |i: usize, j: usize| i * n + j;
|
||||
|
||||
// Forward elimination with partial pivoting
|
||||
for k in 0..n {
|
||||
// Find pivot
|
||||
let mut max_val = a[idx(k, k)].abs();
|
||||
let mut max_row = k;
|
||||
for i in (k + 1)..n {
|
||||
if a[idx(i, k)].abs() > max_val {
|
||||
max_val = a[idx(i, k)].abs();
|
||||
max_row = i;
|
||||
}
|
||||
}
|
||||
|
||||
// Swap rows
|
||||
if max_row != k {
|
||||
for j in 0..n {
|
||||
a.swap(idx(k, j), idx(max_row, j));
|
||||
}
|
||||
b.swap(k, max_row);
|
||||
}
|
||||
|
||||
// Eliminate
|
||||
for i in (k + 1)..n {
|
||||
let factor = a[idx(i, k)] / a[idx(k, k)];
|
||||
for j in k..n {
|
||||
a[idx(i, j)] -= factor * a[idx(k, j)];
|
||||
}
|
||||
b[i] -= factor * b[k];
|
||||
}
|
||||
}
|
||||
|
||||
// Back substitution
|
||||
for i in (0..n).rev() {
|
||||
let mut sum = b[i];
|
||||
for j in (i + 1)..n {
|
||||
sum -= a[idx(i, j)] * x[j];
|
||||
}
|
||||
x[i] = sum / a[idx(i, i)];
|
||||
}
|
||||
|
||||
x
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_lineqs_2x2() {
|
||||
let mut a = vec![2.0, 1.0, 1.0, 3.0];
|
||||
let mut b = vec![5.0, 7.0];
|
||||
let x = lineqs(&mut a, &mut b, 2);
|
||||
assert!((x[0] - 1.6).abs() < 1e-10);
|
||||
assert!((x[1] - 1.8).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_lineqs_3x3() {
|
||||
// x=1, y=2, z=3: 1+2+3=6, 2+2+9=13, 1+6+6=13
|
||||
let mut a = vec![1.0, 1.0, 1.0, 2.0, 1.0, 3.0, 1.0, 3.0, 2.0];
|
||||
let mut b = vec![6.0, 13.0, 13.0];
|
||||
let x = lineqs(&mut a, &mut b, 3);
|
||||
assert!((x[0] - 1.0).abs() < 1e-10);
|
||||
assert!((x[1] - 2.0).abs() < 1e-10);
|
||||
assert!((x[2] - 3.0).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_lineqs_identity() {
|
||||
let mut a = vec![1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0];
|
||||
let mut b = vec![5.0, 3.0, 7.0];
|
||||
let x = lineqs(&mut a, &mut b, 3);
|
||||
assert!((x[0] - 5.0).abs() < 1e-10);
|
||||
assert!((x[1] - 3.0).abs() < 1e-10);
|
||||
assert!((x[2] - 7.0).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_lineqs_1x1() {
|
||||
let mut a = vec![4.0];
|
||||
let mut b = vec![8.0];
|
||||
let x = lineqs(&mut a, &mut b, 1);
|
||||
assert!((x[0] - 2.0).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_lineqs_diagonal() {
|
||||
let mut a = vec![2.0, 0.0, 0.0, 0.0, 3.0, 0.0, 0.0, 0.0, 5.0];
|
||||
let mut b = vec![4.0, 9.0, 15.0];
|
||||
let x = lineqs(&mut a, &mut b, 3);
|
||||
assert!((x[0] - 2.0).abs() < 1e-10);
|
||||
assert!((x[1] - 3.0).abs() < 1e-10);
|
||||
assert!((x[2] - 3.0).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_lineqs_needs_pivoting() {
|
||||
let mut a = vec![0.0, 1.0, 1.0, 1.0];
|
||||
let mut b = vec![1.0, 3.0];
|
||||
let x = lineqs(&mut a, &mut b, 2);
|
||||
assert!((x[0] - 2.0).abs() < 1e-10);
|
||||
assert!((x[1] - 1.0).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_lineqs_4x4() {
|
||||
// Non-singular 4x4 matrix
|
||||
let a_orig = vec![
|
||||
2.0, 1.0, 1.0, 0.0, //
|
||||
4.0, 3.0, 3.0, 1.0, //
|
||||
8.0, 7.0, 9.0, 5.0, //
|
||||
6.0, 7.0, 9.0, 8.0, //
|
||||
];
|
||||
let b_orig = vec![4.0, 11.0, 29.0, 30.0];
|
||||
let mut a = a_orig.clone();
|
||||
let mut b = b_orig.clone();
|
||||
let x = lineqs(&mut a, &mut b, 4);
|
||||
// Verify A*x = b
|
||||
for i in 0..4 {
|
||||
let mut sum = 0.0;
|
||||
for j in 0..4 {
|
||||
sum += a_orig[i * 4 + j] * x[j];
|
||||
}
|
||||
assert!(
|
||||
(sum - b_orig[i]).abs() < 1e-8,
|
||||
"Row {}: got {}, expected {}",
|
||||
i,
|
||||
sum,
|
||||
b_orig[i]
|
||||
);
|
||||
}
|
||||
}
|
||||
}
|
||||
594
src/synspec/math/linop.rs
Normal file
594
src/synspec/math/linop.rs
Normal file
@ -0,0 +1,594 @@
|
||||
//! 线不透明度和发射率计算。
|
||||
//!
|
||||
//! 翻译自 SYNSPEC `LINOP` 子程序 (synspec54.f:10427)。
|
||||
//!
|
||||
//! 计算给定深度点所有谱线的总吸收系数 (ABLIN) 和发射系数 (EMLIN)。
|
||||
//! 包括 LTE 线、NLTE 线、He I 特殊线 (PHE1) 和 He II 线 (PHE2)。
|
||||
|
||||
use super::phe1::{phe1, Phe1Params};
|
||||
use super::phe2::{phe2, Phe2Params};
|
||||
use super::voigtk::{voigtk, MVOI};
|
||||
|
||||
/// 物理常数
|
||||
const UN: f64 = 1.0;
|
||||
const EXT0: f64 = 3.17;
|
||||
const TEN: f64 = 10.0;
|
||||
/// h*c/k (cm·K) — 用于能量到温度的转换
|
||||
const C3: f64 = 1.4387886;
|
||||
/// 1/kT 转换因子 (cm^-1 → eV)
|
||||
const XET: f64 = 8067.6;
|
||||
const XET3: f64 = XET * C3;
|
||||
|
||||
/// 单条谱线的输入数据。
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct LineData {
|
||||
/// 谱线索引 (IL)
|
||||
pub il: usize,
|
||||
/// NLTE 指标 (<0: NLTE, =0: LTE, >0: NLTE with ABCENT)
|
||||
pub innlt: i32,
|
||||
/// 原子序号 (IAT = INDAT/100)
|
||||
pub iat: usize,
|
||||
/// 电离级 (ION = INDAT%100)
|
||||
pub ion: usize,
|
||||
/// 轮廓类型 (ISPRF)
|
||||
pub isprf: usize,
|
||||
/// 线心频率 (Hz)
|
||||
pub freq0: f64,
|
||||
/// log(gf) - E_low/kT
|
||||
pub gf0: f64,
|
||||
/// 下能级激发能 (cm^-1)
|
||||
pub excl0: f64,
|
||||
/// 上能级激发能 (cm^-1)
|
||||
pub excu0: f64,
|
||||
/// 线心频率索引 (IJCNTR)
|
||||
pub ijcntr: usize,
|
||||
/// 阻尼参数 (由 PROFIL 返回)
|
||||
pub agam: f64,
|
||||
/// Doppler 宽度的倒数 (1/DOP1)
|
||||
pub dop1_inv: f64,
|
||||
|
||||
// --- NLTE 相关 ---
|
||||
/// NLTE 线心吸收 (ABCENT, INNLT>0 时使用)
|
||||
pub abcent: f64,
|
||||
/// NLTE 线心源函数 (SLIN, INNLT>0 时使用)
|
||||
pub slin: f64,
|
||||
/// 下能级索引 (ILOWN, INNLT<0 时使用)
|
||||
pub ilown: usize,
|
||||
/// 上能级索引 (IUPN, INNLT<0 时使用)
|
||||
pub iupn: usize,
|
||||
}
|
||||
|
||||
/// LINOP 参数结构体。
|
||||
#[derive(Debug)]
|
||||
pub struct LinopParams<'a> {
|
||||
/// 深度索引 (1-indexed)
|
||||
pub id: usize,
|
||||
/// 温度 (K)
|
||||
pub temp: f64,
|
||||
/// 频率数
|
||||
pub nfreq: usize,
|
||||
/// 频率数组 (Hz)
|
||||
pub freq: &'a [f64],
|
||||
/// 总频率数 (NFREQS)
|
||||
pub nfreqs: usize,
|
||||
/// 频率连续间距因子 (DFRCON)
|
||||
pub dfrcon: f64,
|
||||
/// 平均连续吸收系数 (AVAB)
|
||||
pub avab: f64,
|
||||
/// Planck 函数 (PLAN)
|
||||
pub plan: f64,
|
||||
/// 受激辐射修正 (STIM)
|
||||
pub stim: f64,
|
||||
/// 谱线数 (NLIN)
|
||||
pub nlin: usize,
|
||||
/// 谱线数据
|
||||
pub lines: &'a [LineData],
|
||||
/// RRR(ID,ION,IAT) — Saha/Boltzmann 因子
|
||||
pub rrr: &'a [f64],
|
||||
/// RRR 数组维度: [natom][nion]
|
||||
pub rrr_dims: (usize, usize),
|
||||
/// Doppler 宽度 DOPA1(IAT, ID)
|
||||
pub dopa1: &'a [f64],
|
||||
/// DOPA1 维度: [natom]
|
||||
pub dopa1_nat: usize,
|
||||
/// 能级统计权重 G(level)
|
||||
pub g: &'a [f64],
|
||||
/// 能级布居数 POPUL(level, ID)
|
||||
pub popul: &'a [f64],
|
||||
/// POPUL 维度: [nlevel]
|
||||
pub popul_nlev: usize,
|
||||
/// PNLT(IAT, ION, ID) — NLTE 布居数
|
||||
pub pnlt: &'a [f64],
|
||||
/// PNLT 维度: [natom][nion]
|
||||
pub pnlt_dims: (usize, usize),
|
||||
/// ENEV(IAT, ION) — 电离能 (cm^-1)
|
||||
pub enev: &'a [f64],
|
||||
/// ENEV 维度: [natom]
|
||||
pub enev_nat: usize,
|
||||
/// ENION(level) — 能级能量 (erg)
|
||||
pub enion: &'a [f64],
|
||||
/// 激光删除标志 (lasdel)
|
||||
pub lasdel: bool,
|
||||
/// He II 特殊线数 (NSP)
|
||||
pub nsp: usize,
|
||||
/// He II 特殊线索引 (ISP0)
|
||||
pub isp0: &'a [usize],
|
||||
/// Voigt 函数表 H0
|
||||
pub h0tab: &'a [f64; MVOI],
|
||||
/// Voigt 函数表 H1
|
||||
pub h1tab: &'a [f64; MVOI],
|
||||
/// Voigt 函数表 H2
|
||||
pub h2tab: &'a [f64; MVOI],
|
||||
/// PHE1 所需的轮廓表参数 (简化传递)
|
||||
pub phe1_data: Option<Phe1Data<'a>>,
|
||||
/// PHE2 所需的参数 (简化传递)
|
||||
pub phe2_common: Option<Phe2Common<'a>>,
|
||||
}
|
||||
|
||||
/// PHE1 轮廓表数据。
|
||||
#[derive(Debug)]
|
||||
pub struct Phe1Data<'a> {
|
||||
pub vturb: f64,
|
||||
pub elec: f64,
|
||||
pub prf447: &'a [f64],
|
||||
pub dlm447: &'a [f64],
|
||||
pub xne447: &'a [f64],
|
||||
pub nwlam_447: &'a [usize],
|
||||
pub prfhe1: &'a [f64],
|
||||
pub dlmhe1: &'a [f64],
|
||||
pub xnehe1: &'a [f64],
|
||||
pub nwlam_he1: &'a [usize],
|
||||
pub max_wlam_447: usize,
|
||||
pub max_wlam_he1: usize,
|
||||
}
|
||||
|
||||
/// PHE2 公共数据。
|
||||
#[derive(Debug)]
|
||||
pub struct Phe2Common<'a> {
|
||||
pub ielhe2: i32,
|
||||
pub inlte: i32,
|
||||
pub he3_pop: f64,
|
||||
pub nlhe2: i32,
|
||||
pub nfirst_he2: i32,
|
||||
pub wlam: &'a [f64],
|
||||
pub prfhe2: &'a [f64],
|
||||
pub wlhe2: &'a [f64],
|
||||
pub nwlhe2: i32,
|
||||
pub ilhe2: i32,
|
||||
pub iuhe2: i32,
|
||||
pub lasdel: bool,
|
||||
}
|
||||
|
||||
/// LINOP 输出结果。
|
||||
#[derive(Debug)]
|
||||
pub struct LinopResult {
|
||||
/// 吸收系数数组
|
||||
pub ablin: Vec<f64>,
|
||||
/// 发射系数数组
|
||||
pub emlin: Vec<f64>,
|
||||
}
|
||||
|
||||
/// 计算线不透明度和发射率。
|
||||
///
|
||||
/// # 参数
|
||||
/// * `params` - LINOP 参数
|
||||
///
|
||||
/// # 返回
|
||||
/// 吸收系数和发射系数数组
|
||||
pub fn linop(params: &LinopParams) -> LinopResult {
|
||||
let nfreq = params.nfreq;
|
||||
let mut ablin = vec![0.0f64; nfreq];
|
||||
let mut ablinn = vec![0.0f64; nfreq];
|
||||
let mut emlin = vec![0.0f64; nfreq];
|
||||
|
||||
if params.nlin == 0 {
|
||||
return LinopResult { ablin, emlin };
|
||||
}
|
||||
|
||||
let tem1 = UN / params.temp;
|
||||
|
||||
for line_data in params.lines.iter().take(params.nlin) {
|
||||
let _il = line_data.il;
|
||||
let innlt = line_data.innlt;
|
||||
let iat = line_data.iat;
|
||||
let ion = line_data.ion;
|
||||
let isprf = line_data.isprf;
|
||||
|
||||
// 判断是否使用标准轮廓 (Voigt)
|
||||
let lpr = !(isprf > 1 && isprf <= 5);
|
||||
if isprf >= 6 {
|
||||
continue;
|
||||
}
|
||||
|
||||
// PROFIL 已在外部调用，agam 从 line_data 获取
|
||||
let agam = line_data.agam;
|
||||
let dop1 = 1.0 / line_data.dop1_inv; // DOP1 = 1/dop1_inv
|
||||
let fr0 = line_data.freq0;
|
||||
|
||||
// 计算线心吸收系数 ab0 和源函数 sl0
|
||||
let (ab0, sl0) = if innlt == 0 {
|
||||
// LTE 线
|
||||
let rrr_idx = params.rrr_dims.0 * params.rrr_dims.1 * (params.id - 1)
|
||||
+ ion * params.rrr_dims.0
|
||||
+ iat;
|
||||
let rrr_val = if rrr_idx < params.rrr.len() {
|
||||
params.rrr[rrr_idx]
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
let ab0 = (line_data.gf0 - line_data.excl0 * tem1).exp()
|
||||
* rrr_val
|
||||
* dop1
|
||||
* params.stim;
|
||||
(ab0, 0.0)
|
||||
} else if innlt > 0 {
|
||||
// NLTE 线 (有 ABCENT/SLIN)
|
||||
(line_data.abcent, line_data.slin)
|
||||
} else {
|
||||
// NLTE 线 (通过能级布居数计算)
|
||||
let pnlt_idx = params.pnlt_dims.0 * params.pnlt_dims.1 * (params.id - 1)
|
||||
+ ion * params.pnlt_dims.0
|
||||
+ iat;
|
||||
let pp = if pnlt_idx < params.pnlt.len() {
|
||||
params.pnlt[pnlt_idx]
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
|
||||
// 下能级布居数
|
||||
let pi = if line_data.ilown > 0 {
|
||||
let pop_idx = (line_data.ilown - 1) * params.popul_nlev + (params.id - 1);
|
||||
if pop_idx < params.popul.len() {
|
||||
params.popul[pop_idx] / params.g[line_data.ilown - 1]
|
||||
} else {
|
||||
0.0
|
||||
}
|
||||
} else {
|
||||
let enev_idx = params.enev_nat * (params.id - 1) + iat;
|
||||
let enev_val = if enev_idx < params.enev.len() {
|
||||
params.enev[enev_idx]
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
pp * ((enev_val * XET3 - line_data.excl0) * tem1).exp()
|
||||
};
|
||||
|
||||
// 上能级布居数
|
||||
let (pj, cor) = if line_data.iupn > 0 {
|
||||
let pop_idx = (line_data.iupn - 1) * params.popul_nlev + (params.id - 1);
|
||||
let pj = if pop_idx < params.popul.len() {
|
||||
params.popul[pop_idx] / params.g[line_data.iupn - 1]
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
let cor = if line_data.ilown > 0 && line_data.iupn > 0 {
|
||||
let enion_iun = params.enion[line_data.iupn - 1];
|
||||
let enion_ilw = params.enion[line_data.ilown - 1];
|
||||
((line_data.excu0 - line_data.excl0
|
||||
+ (enion_iun - enion_ilw) / 1.38054e-16)
|
||||
* tem1)
|
||||
.exp()
|
||||
} else {
|
||||
1.0
|
||||
};
|
||||
(pj, cor)
|
||||
} else {
|
||||
let enev_idx = params.enev_nat * (params.id - 1) + iat;
|
||||
let enev_val = if enev_idx < params.enev.len() {
|
||||
params.enev[enev_idx]
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
let pj = pp * ((enev_val * XET3 - line_data.excu0) * tem1).exp();
|
||||
(pj, 1.0)
|
||||
};
|
||||
|
||||
let x = if pj > 0.0 {
|
||||
pi / pj * cor
|
||||
} else {
|
||||
UN
|
||||
};
|
||||
let x = if x == UN {
|
||||
(4.79928e-11 * fr0 * tem1).exp()
|
||||
} else {
|
||||
x
|
||||
};
|
||||
let sl0 = params.plan / (x - UN); // BNUL ≈ PLAN for this context
|
||||
let ab0 = if pi > 0.0 {
|
||||
pi * (UN - UN / x) * line_data.gf0.exp() * dop1
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
(ab0, sl0)
|
||||
};
|
||||
|
||||
// 激光删除检查
|
||||
if ab0 <= 0.0 && params.lasdel {
|
||||
continue;
|
||||
}
|
||||
|
||||
// 确定频率贡献范围
|
||||
let ex0 = if params.avab > 0.0 {
|
||||
ab0 / params.avab * agam
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
let ext = if ex0 > TEN {
|
||||
ex0.sqrt()
|
||||
} else {
|
||||
EXT0
|
||||
};
|
||||
let ext = ext / dop1;
|
||||
let xijext = params.dfrcon * ext + 1.5;
|
||||
let ij1 = ((line_data.ijcntr as f64) - xijext).max(3.0) as usize;
|
||||
let ij2 = ((line_data.ijcntr as f64) + xijext).min(params.nfreqs as f64) as usize;
|
||||
|
||||
if ij1 >= nfreq || ij2 <= 2 {
|
||||
continue;
|
||||
}
|
||||
|
||||
if innlt == 0 {
|
||||
// LTE 线
|
||||
if lpr {
|
||||
// 标准 Voigt 轮廓
|
||||
for ij in ij1..=ij2.min(nfreq - 1) {
|
||||
let xf = (params.freq[ij] - fr0).abs() * dop1;
|
||||
ablin[ij] += ab0 * voigtk(agam, xf, params.h0tab, params.h1tab, params.h2tab);
|
||||
}
|
||||
} else {
|
||||
// He I 特殊线 (ISP 2-5)
|
||||
let _phe1_params = Phe1Params {
|
||||
id: params.id,
|
||||
freq: 0.0, // 在循环中设置
|
||||
iline: isprf - 1,
|
||||
temp: params.temp,
|
||||
elec: 0.0,
|
||||
vturb: 0.0,
|
||||
prf447: &[],
|
||||
dlm447: &[],
|
||||
xne447: &[],
|
||||
nwlam_447: &[],
|
||||
prfhe1: &[],
|
||||
dlmhe1: &[],
|
||||
xnehe1: &[],
|
||||
nwlam_he1: &[],
|
||||
max_wlam_447: 0,
|
||||
max_wlam_he1: 0,
|
||||
h0tab: params.h0tab,
|
||||
h1tab: params.h1tab,
|
||||
h2tab: params.h2tab,
|
||||
};
|
||||
// PHE1 需要完整的轮廓表数据
|
||||
// 如果 phe1_data 存在则使用，否则跳过
|
||||
if let Some(ref phe1d) = params.phe1_data {
|
||||
for ij in 3..nfreq {
|
||||
let fr = params.freq[ij];
|
||||
let phe1_p = Phe1Params {
|
||||
id: params.id,
|
||||
freq: fr,
|
||||
iline: isprf - 1,
|
||||
temp: params.temp,
|
||||
elec: phe1d.elec,
|
||||
vturb: phe1d.vturb,
|
||||
prf447: phe1d.prf447,
|
||||
dlm447: phe1d.dlm447,
|
||||
xne447: phe1d.xne447,
|
||||
nwlam_447: phe1d.nwlam_447,
|
||||
prfhe1: phe1d.prfhe1,
|
||||
dlmhe1: phe1d.dlmhe1,
|
||||
xnehe1: phe1d.xnehe1,
|
||||
nwlam_he1: phe1d.nwlam_he1,
|
||||
max_wlam_447: phe1d.max_wlam_447,
|
||||
max_wlam_he1: phe1d.max_wlam_he1,
|
||||
h0tab: params.h0tab,
|
||||
h1tab: params.h1tab,
|
||||
h2tab: params.h2tab,
|
||||
};
|
||||
let abl = ab0 * phe1(&phe1_p);
|
||||
ablin[ij] += abl;
|
||||
}
|
||||
}
|
||||
}
|
||||
} else {
|
||||
// NLTE 线
|
||||
if lpr {
|
||||
// 标准 Voigt 轮廓
|
||||
for ij in ij1..=ij2.min(nfreq - 1) {
|
||||
let xf = (params.freq[ij] - fr0).abs() * dop1;
|
||||
let abl = ab0 * voigtk(agam, xf, params.h0tab, params.h1tab, params.h2tab);
|
||||
ablinn[ij] += abl;
|
||||
emlin[ij] += abl * sl0;
|
||||
}
|
||||
} else {
|
||||
// He I 特殊线
|
||||
if let Some(ref phe1d) = params.phe1_data {
|
||||
for ij in 3..nfreq {
|
||||
let fr = params.freq[ij];
|
||||
let phe1_p = Phe1Params {
|
||||
id: params.id,
|
||||
freq: fr,
|
||||
iline: isprf - 1,
|
||||
temp: params.temp,
|
||||
elec: phe1d.elec,
|
||||
vturb: phe1d.vturb,
|
||||
prf447: phe1d.prf447,
|
||||
dlm447: phe1d.dlm447,
|
||||
xne447: phe1d.xne447,
|
||||
nwlam_447: phe1d.nwlam_447,
|
||||
prfhe1: phe1d.prfhe1,
|
||||
dlmhe1: phe1d.dlmhe1,
|
||||
xnehe1: phe1d.xnehe1,
|
||||
nwlam_he1: phe1d.nwlam_he1,
|
||||
max_wlam_447: phe1d.max_wlam_447,
|
||||
max_wlam_he1: phe1d.max_wlam_he1,
|
||||
h0tab: params.h0tab,
|
||||
h1tab: params.h1tab,
|
||||
h2tab: params.h2tab,
|
||||
};
|
||||
let abl = ab0 * phe1(&phe1_p);
|
||||
ablinn[ij] += abl;
|
||||
emlin[ij] += abl * sl0;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 添加连续谱贡献到发射率
|
||||
for ij in 3..nfreq {
|
||||
emlin[ij] += ablin[ij] * params.plan;
|
||||
ablin[ij] += ablinn[ij];
|
||||
}
|
||||
|
||||
// He II 特殊线 (PHE2)
|
||||
if params.nsp > 0
|
||||
&& let Some(ref phe2c) = params.phe2_common {
|
||||
for &isp in params.isp0.iter().take(params.nsp) {
|
||||
if (6..=24).contains(&isp) {
|
||||
let phe2_p = Phe2Params {
|
||||
ispec: isp as i32,
|
||||
id: params.id as i32,
|
||||
ielhe2: phe2c.ielhe2,
|
||||
inlte: phe2c.inlte,
|
||||
nfreq: nfreq as i32,
|
||||
freq: params.freq,
|
||||
wlam: phe2c.wlam,
|
||||
temp: params.temp,
|
||||
elec: 0.0, // 由 PHE2 内部处理
|
||||
he3_pop: phe2c.he3_pop,
|
||||
nlhe2: phe2c.nlhe2,
|
||||
nfirst_he2: phe2c.nfirst_he2,
|
||||
popul: &[],
|
||||
prfhe2: phe2c.prfhe2,
|
||||
wlhe2: phe2c.wlhe2,
|
||||
nwlhe2: phe2c.nwlhe2,
|
||||
ilhe2: phe2c.ilhe2,
|
||||
iuhe2: phe2c.iuhe2,
|
||||
lasdel: phe2c.lasdel,
|
||||
};
|
||||
let result = phe2(&phe2_p);
|
||||
for ij in 0..nfreq {
|
||||
ablin[ij] += result.ablin[ij];
|
||||
emlin[ij] += result.emlin[ij];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
LinopResult { ablin, emlin }
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_linop_zero_lines() {
|
||||
// NLIN=0 → 返回全零
|
||||
let params = LinopParams {
|
||||
id: 1,
|
||||
temp: 10000.0,
|
||||
nfreq: 10,
|
||||
freq: &vec![1e14; 10],
|
||||
nfreqs: 10,
|
||||
dfrcon: 1.0,
|
||||
avab: 1.0,
|
||||
plan: 1.0,
|
||||
stim: 1.0,
|
||||
nlin: 0,
|
||||
lines: &[],
|
||||
rrr: &[],
|
||||
rrr_dims: (0, 0),
|
||||
dopa1: &[],
|
||||
dopa1_nat: 0,
|
||||
g: &[],
|
||||
popul: &[],
|
||||
popul_nlev: 0,
|
||||
pnlt: &[],
|
||||
pnlt_dims: (0, 0),
|
||||
enev: &[],
|
||||
enev_nat: 0,
|
||||
enion: &[],
|
||||
lasdel: false,
|
||||
nsp: 0,
|
||||
isp0: &[],
|
||||
h0tab: &[0.0; MVOI],
|
||||
h1tab: &[0.0; MVOI],
|
||||
h2tab: &[0.0; MVOI],
|
||||
phe1_data: None,
|
||||
phe2_common: None,
|
||||
};
|
||||
let result = linop(¶ms);
|
||||
assert!(result.ablin.iter().all(|&x| x == 0.0));
|
||||
assert!(result.emlin.iter().all(|&x| x == 0.0));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_linop_basic_lte_line() {
|
||||
// 基本 LTE 线测试
|
||||
let freq: Vec<f64> = (0..20).map(|i| 1e14 + i as f64 * 1e12).collect();
|
||||
let h0tab = [0.0f64; MVOI];
|
||||
let h1tab = [0.0f64; MVOI];
|
||||
let h2tab = [0.0f64; MVOI];
|
||||
|
||||
let line = LineData {
|
||||
il: 1,
|
||||
innlt: 0,
|
||||
iat: 0,
|
||||
ion: 0,
|
||||
isprf: 1, // 标准 Voigt
|
||||
freq0: 1.01e14,
|
||||
gf0: 0.0, // log(gf)=0 → gf=1
|
||||
excl0: 0.0,
|
||||
excu0: 0.0,
|
||||
ijcntr: 1,
|
||||
agam: 0.01,
|
||||
dop1_inv: 1e-10,
|
||||
abcent: 0.0,
|
||||
slin: 0.0,
|
||||
ilown: 0,
|
||||
iupn: 0,
|
||||
};
|
||||
|
||||
let params = LinopParams {
|
||||
id: 1,
|
||||
temp: 10000.0,
|
||||
nfreq: 20,
|
||||
freq: &freq,
|
||||
nfreqs: 20,
|
||||
dfrcon: 1.0,
|
||||
avab: 1.0,
|
||||
plan: 1.0,
|
||||
stim: 1.0,
|
||||
nlin: 1,
|
||||
lines: &[line],
|
||||
rrr: &[1.0],
|
||||
rrr_dims: (1, 1),
|
||||
dopa1: &[1e10],
|
||||
dopa1_nat: 1,
|
||||
g: &[1.0],
|
||||
popul: &[1.0],
|
||||
popul_nlev: 1,
|
||||
pnlt: &[0.0],
|
||||
pnlt_dims: (1, 1),
|
||||
enev: &[0.0],
|
||||
enev_nat: 1,
|
||||
enion: &[0.0],
|
||||
lasdel: false,
|
||||
nsp: 0,
|
||||
isp0: &[],
|
||||
h0tab: &h0tab,
|
||||
h1tab: &h1tab,
|
||||
h2tab: &h2tab,
|
||||
phe1_data: None,
|
||||
phe2_common: None,
|
||||
};
|
||||
|
||||
let result = linop(¶ms);
|
||||
assert_eq!(result.ablin.len(), 20);
|
||||
assert_eq!(result.emlin.len(), 20);
|
||||
// 验证结果是有限的
|
||||
assert!(result.ablin.iter().all(|&x| x.is_finite()));
|
||||
assert!(result.emlin.iter().all(|&x| x.is_finite()));
|
||||
}
|
||||
}
|
||||
501
src/synspec/math/linopw.rs
Normal file
501
src/synspec/math/linopw.rs
Normal file
@ -0,0 +1,501 @@
|
||||
//! 线不透明度和发射率计算（风模型变体）。
|
||||
//!
|
||||
//! 翻译自 SYNSPEC `LINOPW` 子程序 (synspec54.f:10590)。
|
||||
//!
|
||||
//! 与 LINOP 类似，但针对风模型做了以下扩展：
|
||||
//! - 速度相关的线拒绝
|
||||
//! - 辐射场处理 (itrad, trad, wdil)
|
||||
//! - 窗口化频率网格的线心索引计算
|
||||
//! - DOP1 = DOPA1(IAT,ID)/FR0 (与 LINOP 不同)
|
||||
|
||||
use super::phe1::{phe1, Phe1Params};
|
||||
use super::phe2::{phe2, Phe2Params};
|
||||
use super::voigtk::{voigtk, MVOI};
|
||||
|
||||
/// 物理常数
|
||||
const UN: f64 = 1.0;
|
||||
const EXT0: f64 = 3.17;
|
||||
const TEN: f64 = 10.0;
|
||||
const C3: f64 = 1.4387886;
|
||||
const XET: f64 = 8067.6;
|
||||
const XET3: f64 = XET * C3;
|
||||
|
||||
/// LINOPW 参数结构体。
|
||||
///
|
||||
/// 与 LinopParams 类似，但增加了风模型相关参数。
|
||||
#[derive(Debug)]
|
||||
pub struct LinopwParams<'a> {
|
||||
/// 深度索引 (1-indexed)
|
||||
pub id: usize,
|
||||
/// 温度 (K)
|
||||
pub temp: f64,
|
||||
/// h/k (erg/K)
|
||||
pub hk: f64,
|
||||
/// 频率数
|
||||
pub nfreq: usize,
|
||||
/// 频率数组 (Hz)
|
||||
pub freq: &'a [f64],
|
||||
/// NOPAC — 不透明度截止频率索引
|
||||
pub nopac: usize,
|
||||
/// Planck 函数 PLAN(ID)
|
||||
pub plan: f64,
|
||||
/// 受激辐射修正 STIM(ID) — LINOPW 中未使用，保留接口兼容
|
||||
pub stim: f64,
|
||||
/// 谱线数 (NLIN)
|
||||
pub nlin: usize,
|
||||
/// 谱线数据
|
||||
pub lines: &'a [LinopwLineData],
|
||||
/// RRR(ID,ION,IAT)
|
||||
pub rrr: &'a [f64],
|
||||
/// RRR 维度: [natom][nion]
|
||||
pub rrr_dims: (usize, usize),
|
||||
/// DOPA1(IAT, ID) — Doppler 宽度
|
||||
pub dopa1: &'a [f64],
|
||||
/// DOPA1 维度: [natom]
|
||||
pub dopa1_nat: usize,
|
||||
/// G(level) — 能级统计权重
|
||||
pub g: &'a [f64],
|
||||
/// POPUL(level, ID) — 能级布居数
|
||||
pub popul: &'a [f64],
|
||||
/// POPUL 维度: [nlevel]
|
||||
pub popul_nlev: usize,
|
||||
/// PNLT(IAT, ION, ID) — NLTE 布居数
|
||||
pub pnlt: &'a [f64],
|
||||
/// PNLT 维度: [natom][nion]
|
||||
pub pnlt_dims: (usize, usize),
|
||||
/// ENEV(IAT, ION) — 电离能 (cm^-1)
|
||||
pub enev: &'a [f64],
|
||||
/// ENEV 维度: [natom]
|
||||
pub enev_nat: usize,
|
||||
/// ENION(level) — 能级能量 (erg)
|
||||
pub enion: &'a [f64],
|
||||
/// 激光删除标志 (lasdel)
|
||||
pub lasdel: bool,
|
||||
/// He II 特殊线数 (NSP)
|
||||
pub nsp: usize,
|
||||
/// He II 特殊线索引 (ISP0)
|
||||
pub isp0: &'a [usize],
|
||||
/// Voigt 函数表 H0
|
||||
pub h0tab: &'a [f64; MVOI],
|
||||
/// Voigt 函数表 H1
|
||||
pub h1tab: &'a [f64; MVOI],
|
||||
/// Voigt 函数表 H2
|
||||
pub h2tab: &'a [f64; MVOI],
|
||||
/// PHE1 轮廓表数据
|
||||
pub phe1_data: Option<Phe1DataW<'a>>,
|
||||
/// PHE2 公共数据
|
||||
pub phe2_common: Option<Phe2CommonW<'a>>,
|
||||
|
||||
// --- 风模型特有参数 ---
|
||||
/// 速度场 VEL(ID)
|
||||
pub vel: f64,
|
||||
/// 最大速度 VELMAX
|
||||
pub velmax: f64,
|
||||
/// 辐射场模式 (ITRAD)
|
||||
pub itrad: i32,
|
||||
/// 辐射温度 TRAD(ipotl, ID)
|
||||
pub trad: &'a [f64],
|
||||
/// TRAD 维度: [npotl]
|
||||
pub trad_npotl: usize,
|
||||
/// IPOTL(line) — 电离势索引
|
||||
pub ipotl: &'a [usize],
|
||||
/// BNUE(ij) — 频率相关 Planck 函数
|
||||
pub bnue: &'a [f64],
|
||||
/// NLTE 关闭标志 (NLTOFF)
|
||||
pub nltoff: i32,
|
||||
/// 发射关闭标志 (IEMOFF)
|
||||
pub iemoff: i32,
|
||||
/// 线心 NLTE 关闭标记 ILNE(depth)
|
||||
pub ilne: &'a [usize],
|
||||
/// 线心速度拒绝标记 ILVI(depth)
|
||||
pub ilvi: &'a [usize],
|
||||
/// 线心频率索引 IJCNTR(line) — 输出
|
||||
pub ijcntr: &'a mut [usize],
|
||||
/// 标准线吸收 ABSTDW(ijcont, ID)
|
||||
pub abstdw: &'a [f64],
|
||||
/// ABSTDW 维度: [nfreq]
|
||||
pub abstdw_nfreq: usize,
|
||||
/// RELOP — 相对不透明度阈值
|
||||
pub relop: f64,
|
||||
/// IJCONT(line) — 线心频率索引
|
||||
pub ijcont: &'a [usize],
|
||||
/// 准直函数 XJCON(ID) — 未使用，保留
|
||||
pub xjcon: f64,
|
||||
/// 稀释因子 WDIL(ID) — 输出
|
||||
pub wdil_out: &'a mut f64,
|
||||
}
|
||||
|
||||
/// 单条谱线数据 (LINOPW 版本)。
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct LinopwLineData {
|
||||
pub il: usize,
|
||||
pub innlt: i32,
|
||||
pub iat: usize,
|
||||
pub ion: usize,
|
||||
pub isprf: usize,
|
||||
pub freq0: f64,
|
||||
pub gf0: f64,
|
||||
pub excl0: f64,
|
||||
pub excu0: f64,
|
||||
pub agam: f64,
|
||||
pub dop1_inv: f64,
|
||||
pub abcent: f64,
|
||||
pub slin: f64,
|
||||
pub ilown: usize,
|
||||
pub iupn: usize,
|
||||
}
|
||||
|
||||
/// PHE1 轮廓表数据 (LINOPW 版本)。
|
||||
#[derive(Debug)]
|
||||
pub struct Phe1DataW<'a> {
|
||||
pub vturb: f64,
|
||||
pub elec: f64,
|
||||
pub prf447: &'a [f64],
|
||||
pub dlm447: &'a [f64],
|
||||
pub xne447: &'a [f64],
|
||||
pub nwlam_447: &'a [usize],
|
||||
pub prfhe1: &'a [f64],
|
||||
pub dlmhe1: &'a [f64],
|
||||
pub xnehe1: &'a [f64],
|
||||
pub nwlam_he1: &'a [usize],
|
||||
pub max_wlam_447: usize,
|
||||
pub max_wlam_he1: usize,
|
||||
}
|
||||
|
||||
/// PHE2 公共数据 (LINOPW 版本)。
|
||||
#[derive(Debug)]
|
||||
pub struct Phe2CommonW<'a> {
|
||||
pub ielhe2: i32,
|
||||
pub inlte: i32,
|
||||
pub he3_pop: f64,
|
||||
pub nlhe2: i32,
|
||||
pub nfirst_he2: i32,
|
||||
pub wlam: &'a [f64],
|
||||
pub prfhe2: &'a [f64],
|
||||
pub wlhe2: &'a [f64],
|
||||
pub nwlhe2: i32,
|
||||
pub ilhe2: i32,
|
||||
pub iuhe2: i32,
|
||||
pub lasdel: bool,
|
||||
}
|
||||
|
||||
/// LINOPW 输出结果。
|
||||
#[derive(Debug)]
|
||||
pub struct LinopwResult {
|
||||
pub ablin: Vec<f64>,
|
||||
pub emlin: Vec<f64>,
|
||||
}
|
||||
|
||||
/// 计算线不透明度和发射率（风模型变体）。
|
||||
pub fn linopw(params: &mut LinopwParams) -> LinopwResult {
|
||||
let nfreq = params.nfreq;
|
||||
let mut ablin = vec![0.0f64; nfreq];
|
||||
let mut ablinn = vec![0.0f64; nfreq];
|
||||
let mut emlin = vec![0.0f64; nfreq];
|
||||
|
||||
*params.wdil_out = 1.0;
|
||||
let _plw = params.plan * 1.0; // wdil=1
|
||||
|
||||
if params.nlin == 0 {
|
||||
return LinopwResult { ablin, emlin };
|
||||
}
|
||||
|
||||
let tem1 = UN / params.temp;
|
||||
let hkt = params.hk * tem1;
|
||||
|
||||
// 计算频率间距因子
|
||||
let xx = params.freq[nfreq - 1] - params.freq[0];
|
||||
let dfrcon = if xx.abs() > 1e-30 {
|
||||
-((params.nopac as f64) - 1.0) / xx
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
|
||||
for (line_idx, line_data) in params.lines.iter().take(params.nlin).enumerate() {
|
||||
let il = line_data.il;
|
||||
let innlt = line_data.innlt;
|
||||
|
||||
// 速度拒绝
|
||||
if params.ilvi[params.id - 1] > 0 {
|
||||
if innlt == 0 {
|
||||
continue;
|
||||
} else if params.nltoff != 0 {
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
// 线心频率索引 (仅深度 1)
|
||||
if params.id == 1 {
|
||||
let fr0 = line_data.freq0;
|
||||
let xjc = 3.0 + dfrcon * (params.freq[0] - fr0);
|
||||
let mut ijc = xjc as usize;
|
||||
if ijc > 1 && ijc < params.nopac {
|
||||
// 在频率网格中找到最近的点
|
||||
if fr0 < params.freq[ijc] {
|
||||
let mut ijc0 = ijc;
|
||||
let mut dfr0 = params.freq[ijc0] - fr0;
|
||||
loop {
|
||||
ijc0 += 1;
|
||||
if ijc0 >= nfreq {
|
||||
break;
|
||||
}
|
||||
let dfr = (params.freq[ijc0] - fr0).abs();
|
||||
if dfr < dfr0 {
|
||||
ijc = ijc0;
|
||||
dfr0 = dfr;
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
}
|
||||
} else if fr0 > params.freq[ijc] {
|
||||
let mut ijc0 = ijc;
|
||||
let mut dfr0 = fr0 - params.freq[ijc0];
|
||||
loop {
|
||||
if ijc0 == 0 {
|
||||
break;
|
||||
}
|
||||
ijc0 -= 1;
|
||||
let dfr = (params.freq[ijc0] - fr0).abs();
|
||||
if dfr < dfr0 {
|
||||
ijc = ijc0;
|
||||
dfr0 = dfr;
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
params.ijcntr[line_idx] = ijc;
|
||||
}
|
||||
|
||||
let iat = line_data.iat;
|
||||
let ion = line_data.ion;
|
||||
let fr0 = line_data.freq0;
|
||||
let lpr = !(line_data.isprf > 1 && line_data.isprf <= 5);
|
||||
if line_data.isprf >= 6 {
|
||||
continue;
|
||||
}
|
||||
|
||||
let agam = line_data.agam;
|
||||
let dop1 = 1.0 / line_data.dop1_inv / fr0; // DOPA1(IAT,ID)/FR0
|
||||
|
||||
// 计算 ab0 和 sl0
|
||||
let (ab0, sl0) = if innlt == 0 && params.itrad <= 0 {
|
||||
// LTE 线 (无辐射场)
|
||||
let rrr_idx = params.rrr_dims.0 * params.rrr_dims.1 * (params.id - 1)
|
||||
+ ion * params.rrr_dims.0 + iat;
|
||||
let rrr_val = if rrr_idx < params.rrr.len() { params.rrr[rrr_idx] } else { 0.0 };
|
||||
let ab0 = (line_data.gf0 - line_data.excl0 * tem1).exp()
|
||||
* rrr_val * dop1 * (1.0 - (-hkt * fr0).exp());
|
||||
(ab0, 0.0)
|
||||
} else if innlt == 0 && params.itrad > 0 {
|
||||
// LTE 线 (有辐射场)
|
||||
let ipotl_idx = if il < params.ipotl.len() { params.ipotl[il] } else { 0 };
|
||||
let trad_idx = ipotl_idx * params.trad_npotl + (params.id - 1);
|
||||
let trl = if trad_idx < params.trad.len() { params.trad[trad_idx] } else { params.temp };
|
||||
let xx = (-hkt * fr0).exp();
|
||||
let rrr_idx = params.rrr_dims.0 * params.rrr_dims.1 * (params.id - 1)
|
||||
+ ion * params.rrr_dims.0 + iat;
|
||||
let rrr_val = if rrr_idx < params.rrr.len() { params.rrr[rrr_idx] } else { 0.0 };
|
||||
let mut ab0 = (line_data.gf0 - line_data.excl0 / trl).exp()
|
||||
* rrr_val * dop1 * (1.0 - xx);
|
||||
if line_data.excl0 > 2000.0 {
|
||||
ab0 *= 1.0; // wdil=1
|
||||
}
|
||||
let pla = 1.4743e-2 * (fr0 * 1e-15).powi(3) * xx / (1.0 - xx);
|
||||
let sl0 = pla * 1.0; // wdil=1
|
||||
(ab0, sl0)
|
||||
} else if innlt > 0 {
|
||||
(line_data.abcent, line_data.slin)
|
||||
} else {
|
||||
// NLTE 线
|
||||
let pnlt_idx = params.pnlt_dims.0 * params.pnlt_dims.1 * (params.id - 1)
|
||||
+ ion * params.pnlt_dims.0 + iat;
|
||||
let pp = if pnlt_idx < params.pnlt.len() { params.pnlt[pnlt_idx] } else { 0.0 };
|
||||
|
||||
let pi = if line_data.ilown > 0 {
|
||||
let pop_idx = (line_data.ilown - 1) * params.popul_nlev + (params.id - 1);
|
||||
if pop_idx < params.popul.len() {
|
||||
params.popul[pop_idx] / params.g[line_data.ilown - 1]
|
||||
} else { 0.0 }
|
||||
} else {
|
||||
let enev_idx = params.enev_nat * (params.id - 1) + iat;
|
||||
let enev_val = if enev_idx < params.enev.len() { params.enev[enev_idx] } else { 0.0 };
|
||||
pp * ((enev_val * XET3 - line_data.excl0) * tem1).exp()
|
||||
};
|
||||
|
||||
let (pj, cor) = if line_data.iupn > 0 {
|
||||
let pop_idx = (line_data.iupn - 1) * params.popul_nlev + (params.id - 1);
|
||||
let pj = if pop_idx < params.popul.len() {
|
||||
params.popul[pop_idx] / params.g[line_data.iupn - 1]
|
||||
} else { 0.0 };
|
||||
let cor = if line_data.ilown > 0 && line_data.iupn > 0 {
|
||||
((line_data.excu0 - line_data.excl0
|
||||
+ (params.enion[line_data.iupn - 1] - params.enion[line_data.ilown - 1]) / 1.38054e-16)
|
||||
* tem1).exp()
|
||||
} else { 1.0 };
|
||||
(pj, cor)
|
||||
} else {
|
||||
let enev_idx = params.enev_nat * (params.id - 1) + iat;
|
||||
let enev_val = if enev_idx < params.enev.len() { params.enev[enev_idx] } else { 0.0 };
|
||||
let pj = pp * ((enev_val * XET3 - line_data.excu0) * tem1).exp();
|
||||
(pj, 1.0)
|
||||
};
|
||||
|
||||
let x = if pj > 0.0 { pi / pj * cor } else { UN };
|
||||
let x = if x == UN { (4.79928e-11 * fr0 * tem1).exp() } else { x };
|
||||
let sl0 = params.plan / (x - UN);
|
||||
let ab0 = if pi > 0.0 { pi * (UN - UN / x) * line_data.gf0.exp() * dop1 } else { 0.0 };
|
||||
(ab0, sl0)
|
||||
};
|
||||
|
||||
if ab0 <= 0.0 && params.lasdel {
|
||||
continue;
|
||||
}
|
||||
|
||||
// 频率贡献范围
|
||||
let ijcont_idx = if il < params.ijcont.len() { params.ijcont[il] } else { 0 };
|
||||
let abstdw_idx = ijcont_idx * params.abstdw_nfreq + (params.id - 1);
|
||||
let avabw = if abstdw_idx < params.abstdw.len() {
|
||||
params.abstdw[abstdw_idx] * params.relop
|
||||
} else { 0.0 };
|
||||
let ex0 = if avabw > 0.0 { ab0 / avabw * agam } else { 0.0 };
|
||||
let ext = if ex0 > TEN { ex0.sqrt() } else { EXT0 };
|
||||
let ext = ext / dop1;
|
||||
let ijext = (dfrcon * ext + 1.5) as usize;
|
||||
let ijctr = params.ijcntr[line_idx];
|
||||
let ij1 = ijctr.saturating_sub(ijext).max(1);
|
||||
let ij2 = (ijctr + ijext).min(nfreq);
|
||||
|
||||
if ij1 >= nfreq || ij2 <= 2 {
|
||||
continue;
|
||||
}
|
||||
|
||||
if innlt == 0 && params.itrad <= 0 {
|
||||
// LTE 线
|
||||
if lpr {
|
||||
for ij in ij1..=ij2.min(nfreq - 1) {
|
||||
let xf = (params.freq[ij] - fr0).abs() * dop1;
|
||||
ablin[ij] += ab0 * voigtk(agam, xf, params.h0tab, params.h1tab, params.h2tab);
|
||||
}
|
||||
} else if let Some(ref phe1d) = params.phe1_data {
|
||||
for ij in 0..nfreq {
|
||||
let phe1_p = Phe1Params {
|
||||
id: params.id, freq: params.freq[ij], iline: line_data.isprf - 1,
|
||||
temp: params.temp, elec: phe1d.elec, vturb: phe1d.vturb,
|
||||
prf447: phe1d.prf447, dlm447: phe1d.dlm447, xne447: phe1d.xne447,
|
||||
nwlam_447: phe1d.nwlam_447, prfhe1: phe1d.prfhe1, dlmhe1: phe1d.dlmhe1,
|
||||
xnehe1: phe1d.xnehe1, nwlam_he1: phe1d.nwlam_he1,
|
||||
max_wlam_447: phe1d.max_wlam_447, max_wlam_he1: phe1d.max_wlam_he1,
|
||||
h0tab: params.h0tab, h1tab: params.h1tab, h2tab: params.h2tab,
|
||||
};
|
||||
ablin[ij] += ab0 * phe1(&phe1_p);
|
||||
}
|
||||
}
|
||||
} else {
|
||||
// NLTE 线 或 有辐射场的 LTE 线
|
||||
if lpr {
|
||||
for ij in ij1..=ij2.min(nfreq - 1) {
|
||||
let xf = (params.freq[ij] - fr0).abs() * dop1;
|
||||
let abl = ab0 * voigtk(agam, xf, params.h0tab, params.h1tab, params.h2tab);
|
||||
ablinn[ij] += abl;
|
||||
if params.ilne[params.id - 1] == 0 {
|
||||
emlin[ij] += abl * sl0;
|
||||
}
|
||||
}
|
||||
} else if let Some(ref phe1d) = params.phe1_data {
|
||||
for ij in 0..nfreq {
|
||||
let phe1_p = Phe1Params {
|
||||
id: params.id, freq: params.freq[ij], iline: line_data.isprf - 1,
|
||||
temp: params.temp, elec: phe1d.elec, vturb: phe1d.vturb,
|
||||
prf447: phe1d.prf447, dlm447: phe1d.dlm447, xne447: phe1d.xne447,
|
||||
nwlam_447: phe1d.nwlam_447, prfhe1: phe1d.prfhe1, dlmhe1: phe1d.dlmhe1,
|
||||
xnehe1: phe1d.xnehe1, nwlam_he1: phe1d.nwlam_he1,
|
||||
max_wlam_447: phe1d.max_wlam_447, max_wlam_he1: phe1d.max_wlam_he1,
|
||||
h0tab: params.h0tab, h1tab: params.h1tab, h2tab: params.h2tab,
|
||||
};
|
||||
let abl = ab0 * phe1(&phe1_p);
|
||||
ablinn[ij] += abl;
|
||||
if params.ilne[params.id - 1] == 0 {
|
||||
emlin[ij] += abl * sl0;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 连续谱贡献
|
||||
if params.vel <= params.velmax {
|
||||
for ij in 0..nfreq {
|
||||
let pla = if (hkt * params.freq[ij]).exp() - 1.0 > 1e-30 {
|
||||
params.bnue[ij] / ((hkt * params.freq[ij]).exp() - 1.0)
|
||||
} else { 0.0 };
|
||||
emlin[ij] += ablin[ij] * pla; // wdil=1
|
||||
ablin[ij] += ablinn[ij];
|
||||
}
|
||||
}
|
||||
|
||||
// He II 特殊线
|
||||
if params.nsp > 0
|
||||
&& let Some(ref phe2c) = params.phe2_common {
|
||||
for &isp in params.isp0.iter().take(params.nsp) {
|
||||
if (6..=24).contains(&isp) {
|
||||
let phe2_p = Phe2Params {
|
||||
ispec: isp as i32, id: params.id as i32,
|
||||
ielhe2: phe2c.ielhe2, inlte: phe2c.inlte,
|
||||
nfreq: nfreq as i32, freq: params.freq, wlam: phe2c.wlam,
|
||||
temp: params.temp, elec: 0.0, he3_pop: phe2c.he3_pop,
|
||||
nlhe2: phe2c.nlhe2, nfirst_he2: phe2c.nfirst_he2,
|
||||
popul: &[], prfhe2: phe2c.prfhe2, wlhe2: phe2c.wlhe2,
|
||||
nwlhe2: phe2c.nwlhe2, ilhe2: phe2c.ilhe2, iuhe2: phe2c.iuhe2,
|
||||
lasdel: phe2c.lasdel,
|
||||
};
|
||||
let result = phe2(&phe2_p);
|
||||
for ij in 0..nfreq {
|
||||
ablin[ij] += result.ablin[ij];
|
||||
emlin[ij] += result.emlin[ij];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
LinopwResult { ablin, emlin }
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_linopw_zero_lines() {
|
||||
let freq = vec![1e14; 10];
|
||||
let h0tab = [0.0f64; MVOI];
|
||||
let h1tab = [0.0f64; MVOI];
|
||||
let h2tab = [0.0f64; MVOI];
|
||||
let mut wdil = 0.0f64;
|
||||
let mut ijcntr = vec![0usize; 1];
|
||||
|
||||
let mut params = LinopwParams {
|
||||
id: 1, temp: 10000.0, hk: 4.79928e-11,
|
||||
nfreq: 10, freq: &freq, nopac: 10,
|
||||
plan: 1.0, stim: 1.0, nlin: 0, lines: &[],
|
||||
rrr: &[], rrr_dims: (0, 0), dopa1: &[], dopa1_nat: 0,
|
||||
g: &[], popul: &[], popul_nlev: 0,
|
||||
pnlt: &[], pnlt_dims: (0, 0),
|
||||
enev: &[], enev_nat: 0, enion: &[],
|
||||
lasdel: false, nsp: 0, isp0: &[],
|
||||
h0tab: &h0tab, h1tab: &h1tab, h2tab: &h2tab,
|
||||
phe1_data: None, phe2_common: None,
|
||||
vel: 0.0, velmax: 1e10, itrad: 0,
|
||||
trad: &[], trad_npotl: 0, ipotl: &[],
|
||||
bnue: &freq, nltoff: 0, iemoff: 0,
|
||||
ilne: &[0], ilvi: &[0],
|
||||
ijcntr: &mut ijcntr,
|
||||
abstdw: &[], abstdw_nfreq: 0, relop: 1.0,
|
||||
ijcont: &[0], xjcon: 0.0, wdil_out: &mut wdil,
|
||||
};
|
||||
let result = linopw(&mut params);
|
||||
assert!(result.ablin.iter().all(|&x| x == 0.0));
|
||||
assert!(result.emlin.iter().all(|&x| x == 0.0));
|
||||
}
|
||||
}
|
||||
77
src/synspec/math/locate.rs
Normal file
77
src/synspec/math/locate.rs
Normal file
@ -0,0 +1,77 @@
|
||||
//! 二分查找子程序。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `locate.f`
|
||||
//!
|
||||
//! 在有序数组 `xx` 中查找值 `x` 的位置，使得 `xx[j] <= x < xx[j+1]`。
|
||||
|
||||
/// 在有序数组中二分查找。
|
||||
///
|
||||
/// 查找 `x` 在有序数组 `xx[0..n]` 中的插入位置。
|
||||
/// 返回索引 `j`，使得 `xx[j-1] <= x < xx[j]`（1-indexed）。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `xx` - 有序数组（升序或降序）
|
||||
/// * `n` - 数组有效长度
|
||||
/// * `x` - 待查找的值
|
||||
///
|
||||
/// # 返回值
|
||||
///
|
||||
/// 索引 `j`（1-indexed），使得 `xx[j-1] <= x < xx[j]`
|
||||
pub fn locate(xx: &[f64], n: usize, x: f64) -> usize {
|
||||
let mut jl: isize = 0;
|
||||
let mut ju = (n + 1) as isize;
|
||||
|
||||
while ju - jl > 1 {
|
||||
let jm = (ju + jl) / 2;
|
||||
if (xx[n - 1] >= xx[0]) == (x >= xx[jm as usize - 1]) {
|
||||
jl = jm;
|
||||
} else {
|
||||
ju = jm;
|
||||
}
|
||||
}
|
||||
|
||||
if x == xx[0] {
|
||||
1
|
||||
} else if x == xx[n - 1] {
|
||||
n - 1
|
||||
} else {
|
||||
jl as usize
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_locate_basic() {
|
||||
let xx = [1.0, 2.0, 3.0, 4.0, 5.0];
|
||||
// x=2.5 应该在 xx[1] 和 xx[2] 之间，返回 j=2
|
||||
let j = locate(&xx, 5, 2.5);
|
||||
assert!(j >= 1 && j <= 4);
|
||||
assert!(xx[j - 1] <= 2.5 && xx[j] >= 2.5);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_locate_exact_first() {
|
||||
let xx = [1.0, 2.0, 3.0, 4.0, 5.0];
|
||||
let j = locate(&xx, 5, 1.0);
|
||||
assert_eq!(j, 1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_locate_exact_last() {
|
||||
let xx = [1.0, 2.0, 3.0, 4.0, 5.0];
|
||||
let j = locate(&xx, 5, 5.0);
|
||||
assert_eq!(j, 4);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_locate_between() {
|
||||
let xx = [0.0, 10.0, 20.0, 30.0];
|
||||
let j = locate(&xx, 4, 15.0);
|
||||
assert!(j >= 1 && j <= 3);
|
||||
assert!(xx[j - 1] <= 15.0 && xx[j] >= 15.0);
|
||||
}
|
||||
}
|
||||
186
src/synspec/math/lyahhe.rs
Normal file
186
src/synspec/math/lyahhe.rs
Normal file
@ -0,0 +1,186 @@
|
||||
//! Lyman alpha broadening by helium.
|
||||
//!
|
||||
//! Translated from SYNSPEC `lyahhe` subroutine (synspec54.f:12768).
|
||||
//!
|
||||
//! Calculates the Lyman alpha profile broadened by helium collisions,
|
||||
//! using cross-section data from N. Allard.
|
||||
|
||||
use std::sync::OnceLock;
|
||||
|
||||
/// Maximum number of cross-section data points
|
||||
const NXMAX: usize = 1000;
|
||||
|
||||
/// Cross-section normalization factor for He-H broadening
|
||||
static STHE: OnceLock<f64> = OnceLock::new();
|
||||
|
||||
/// Flag for wavelength unit conversion (1 = convert from Angstrom)
|
||||
static NUNHHE: OnceLock<i32> = OnceLock::new();
|
||||
|
||||
/// Cached cross-section data
|
||||
struct HeHData {
|
||||
xlhhe: Vec<f64>,
|
||||
sighhe: Vec<f64>,
|
||||
nxhhe: usize,
|
||||
}
|
||||
|
||||
static HEH_DATA: OnceLock<HeHData> = OnceLock::new();
|
||||
|
||||
/// Initialize the He-H broadening data from a file.
|
||||
///
|
||||
/// This should be called once before using `lyahhe`. The data is read from
|
||||
/// unit 67 (typically `siglyhhe_21_T14500.lam`).
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `sthe_val` - Cross-section normalization factor
|
||||
/// * `nunhhe_val` - Flag for wavelength unit conversion
|
||||
/// * `data_lines` - Iterator of (wavelength, cross-section) pairs from the data file
|
||||
pub fn lyahhe_init<I>(sthe_val: f64, nunhhe_val: i32, data_lines: I)
|
||||
where
|
||||
I: IntoIterator<Item = (f64, f64)>,
|
||||
{
|
||||
let _ = STHE.set(sthe_val);
|
||||
let _ = NUNHHE.set(nunhhe_val);
|
||||
|
||||
let nunhhe = nunhhe_val;
|
||||
let mut xlhh0 = Vec::with_capacity(NXMAX);
|
||||
let mut sighh0 = Vec::with_capacity(NXMAX);
|
||||
|
||||
for (xl, sig) in data_lines.into_iter().take(NXMAX) {
|
||||
let xl_conv = if nunhhe == 1 {
|
||||
1.0 / (1.0e-8 * xl + 1.0 / 1215.67)
|
||||
} else {
|
||||
xl
|
||||
};
|
||||
xlhh0.push(xl_conv);
|
||||
sighh0.push(sig);
|
||||
}
|
||||
|
||||
let nxhhe = xlhh0.len();
|
||||
|
||||
// Reverse the arrays (Fortran reads in reverse order)
|
||||
let mut xlhhe: Vec<f64> = xlhh0.into_iter().rev().collect();
|
||||
let mut sighhe: Vec<f64> = sighh0.into_iter().rev().collect();
|
||||
|
||||
// Ensure sorted for binary search
|
||||
// After reversal, data should be in ascending wavelength order
|
||||
// but let's verify and sort if needed
|
||||
if nxhhe > 1 && xlhhe[0] > xlhhe[nxhhe - 1] {
|
||||
xlhhe.reverse();
|
||||
sighhe.reverse();
|
||||
}
|
||||
|
||||
let _ = HEH_DATA.set(HeHData {
|
||||
xlhhe,
|
||||
sighhe,
|
||||
nxhhe,
|
||||
});
|
||||
}
|
||||
|
||||
/// Calculate Lyman alpha profile broadened by helium.
|
||||
///
|
||||
/// Uses binary search and linear interpolation on pre-loaded cross-section data.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `xl` - Wavelength (Å)
|
||||
/// * `ahe` - He atom density
|
||||
///
|
||||
/// # Returns
|
||||
/// Profile value (0.0 if wavelength is outside data range)
|
||||
pub fn lyahhe(xl: f64, ahe: f64) -> f64 {
|
||||
let data = match HEH_DATA.get() {
|
||||
Some(d) => d,
|
||||
None => return 0.0, // Data not initialized
|
||||
};
|
||||
|
||||
let sthe = match STHE.get() {
|
||||
Some(&s) => s,
|
||||
None => return 0.0,
|
||||
};
|
||||
|
||||
if data.nxhhe == 0 {
|
||||
return 0.0;
|
||||
}
|
||||
|
||||
let prof = 0.0_f64;
|
||||
|
||||
// Check bounds
|
||||
if xl > data.xlhhe[data.nxhhe - 1] {
|
||||
return prof;
|
||||
}
|
||||
|
||||
// Binary search for the interval
|
||||
let mut jl = 0usize;
|
||||
let mut ju = data.nxhhe;
|
||||
|
||||
while ju - jl > 1 {
|
||||
let jm = (ju + jl) / 2;
|
||||
if (data.xlhhe[data.nxhhe - 1] > data.xlhhe[0]) == (xl > data.xlhhe[jm]) {
|
||||
jl = jm;
|
||||
} else {
|
||||
ju = jm;
|
||||
}
|
||||
}
|
||||
|
||||
let mut j = jl;
|
||||
if j == 0 {
|
||||
j = 1;
|
||||
}
|
||||
if j >= data.nxhhe {
|
||||
j = data.nxhhe - 1;
|
||||
}
|
||||
|
||||
// Linear interpolation
|
||||
let denom = data.xlhhe[j] - data.xlhhe[j - 1];
|
||||
if denom.abs() < 1.0e-30 {
|
||||
return 0.0;
|
||||
}
|
||||
|
||||
let a1 = (xl - data.xlhhe[j - 1]) / denom;
|
||||
let s1 = (1.0 - a1) * data.sighhe[j - 1] + a1 * data.sighhe[j];
|
||||
|
||||
|
||||
s1 * ahe / sthe * std::f64::consts::TAU
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_lyahhe_not_initialized() {
|
||||
// Without initialization, should return 0.0
|
||||
let result = lyahhe(1215.67, 1.0e20);
|
||||
assert_eq!(result, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_lyahhe_basic() {
|
||||
// Initialize with synthetic data
|
||||
let data = vec![
|
||||
(1215.0, 1.0e-20),
|
||||
(1215.5, 2.0e-20),
|
||||
(1215.67, 5.0e-20),
|
||||
(1216.0, 3.0e-20),
|
||||
(1216.5, 1.0e-20),
|
||||
];
|
||||
lyahhe_init(1.0e21, 0, data);
|
||||
|
||||
// Test interpolation at known point
|
||||
let result = lyahhe(1215.67, 1.0e20);
|
||||
assert!(result > 0.0);
|
||||
assert!(result.is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_lyahhe_outside_range() {
|
||||
let data = vec![
|
||||
(1215.0, 1.0e-20),
|
||||
(1216.0, 3.0e-20),
|
||||
];
|
||||
lyahhe_init(1.0e21, 0, data);
|
||||
|
||||
// Wavelength outside range should return 0.0
|
||||
let result = lyahhe(1220.0, 1.0e20);
|
||||
assert_eq!(result, 0.0);
|
||||
}
|
||||
}
|
||||
311
src/synspec/math/lymlin.rs
Normal file
311
src/synspec/math/lymlin.rs
Normal file
@ -0,0 +1,311 @@
|
||||
//! Lyman line wings opacity.
|
||||
//!
|
||||
//! Translated from SYNSPEC `LYMLIN` subroutine (synspec54.f:5183).
|
||||
//!
|
||||
//! Calculates opacity of Lyman lines (alpha - delta) with
|
||||
//! approximate partial redistribution.
|
||||
|
||||
/// Lyman line frequencies (Hz)
|
||||
const FRLY: [f64; 4] = [
|
||||
2.4660375e15, // Lyman alpha
|
||||
2.9227111e15, // Lyman beta
|
||||
3.0825469e15, // Lyman gamma
|
||||
3.156528e15, // Lyman delta
|
||||
];
|
||||
|
||||
/// BNLY coefficients
|
||||
const BNLY: [f64; 4] = [5.527e-2, 4.090e-2, 2.699e-2, 1.855e-2];
|
||||
|
||||
/// Natural broadening coefficients
|
||||
const SN: [f64; 4] = [1.308e5, 5.280e3, 5.847e2, 1.078e2];
|
||||
|
||||
/// Resonance broadening coefficients
|
||||
const SR: [f64; 4] = [1.218e-16, 9.196e-17, 1.058e-16, 1.296e-16];
|
||||
|
||||
/// Stark broadening coefficients (SS)
|
||||
const SS: [f64; 4] = [9.478e-3, 1.600e-2, 1.441e-2, 1.547e-2];
|
||||
|
||||
/// Stark broadening coefficients (GS)
|
||||
const GS: [f64; 4] = [7.237e-8, 5.432e-6, 5.821e-5, 4.027e-4];
|
||||
|
||||
/// Damping constants
|
||||
const GA: [f64; 4] = [1.000, 1.791, 2.362, 2.801];
|
||||
|
||||
/// Parameters for Lyman line wings calculation
|
||||
pub struct LymlinParams {
|
||||
/// Depth index
|
||||
pub id: usize,
|
||||
/// Frequency (Hz)
|
||||
pub freq: f64,
|
||||
/// H ground level population
|
||||
pub pop_h: f64,
|
||||
/// Excited level populations (index 0 = level 2, etc.)
|
||||
pub pop_excited: [f64; 4],
|
||||
/// Temperature (K)
|
||||
pub t: f64,
|
||||
/// Electron density
|
||||
pub ane: f64,
|
||||
/// Hydrogen atom exists
|
||||
pub iath: i32,
|
||||
/// Lyman line treatment switch
|
||||
pub iophli: i32,
|
||||
/// wnHint factors for each level
|
||||
pub wn_hint: [f64; 5],
|
||||
}
|
||||
|
||||
/// Result of Lyman line wings calculation
|
||||
pub struct LymlinResult {
|
||||
/// Absorption coefficient
|
||||
pub ably: f64,
|
||||
/// Emission coefficient
|
||||
pub emly: f64,
|
||||
/// Scattering coefficient
|
||||
pub scly: f64,
|
||||
}
|
||||
|
||||
/// Configuration for Lyman line treatment
|
||||
struct LymlinConfig {
|
||||
ifstrk: i32,
|
||||
ifnat: i32,
|
||||
ifres: i32,
|
||||
ifprd: i32,
|
||||
ifsti: i32,
|
||||
}
|
||||
|
||||
use std::sync::OnceLock;
|
||||
|
||||
static LYMLIN_CONFIG: OnceLock<LymlinConfig> = OnceLock::new();
|
||||
|
||||
fn get_lymlin_config(iophli: i32) -> &'static LymlinConfig {
|
||||
LYMLIN_CONFIG.get_or_init(|| {
|
||||
let mut config = LymlinConfig {
|
||||
ifstrk: 0,
|
||||
ifnat: 1,
|
||||
ifres: 1,
|
||||
ifprd: 0,
|
||||
ifsti: 0,
|
||||
};
|
||||
if iophli < 0 {
|
||||
config.ifstrk = 1;
|
||||
config.ifprd = 1;
|
||||
}
|
||||
config
|
||||
})
|
||||
}
|
||||
|
||||
/// Calculate Lyman line wings opacity.
|
||||
///
|
||||
/// Computes opacity contributions from Lyman alpha through delta lines
|
||||
/// using approximate partial redistribution.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Input parameters
|
||||
///
|
||||
/// # Returns
|
||||
/// Absorption, emission, and scattering coefficients
|
||||
pub fn lymlin(params: &LymlinParams) -> LymlinResult {
|
||||
let mut ably = 0.0;
|
||||
let mut emly = 0.0;
|
||||
let mut scly = 0.0;
|
||||
|
||||
if params.iath <= 0 {
|
||||
return LymlinResult { ably, emly, scly };
|
||||
}
|
||||
|
||||
let config = get_lymlin_config(params.iophli);
|
||||
|
||||
if params.freq > 3.3e15 {
|
||||
return LymlinResult { ably, emly, scly };
|
||||
}
|
||||
|
||||
let p = params.pop_h;
|
||||
let t = params.t;
|
||||
let ane = params.ane;
|
||||
|
||||
for i in 0..4 {
|
||||
let mut dfr = (FRLY[i] - params.freq).abs();
|
||||
if dfr <= 5.0e11 {
|
||||
dfr = 1.0e12;
|
||||
}
|
||||
let dfr2 = dfr * dfr;
|
||||
let dfrs = dfr.sqrt();
|
||||
let cor = (2.0 * params.freq / (params.freq + FRLY[i])).powi(2);
|
||||
|
||||
let mut f = 1.0;
|
||||
if params.iophli.abs() == 2 {
|
||||
f = feautr_lyman(params.freq, params.id, t, ane);
|
||||
}
|
||||
|
||||
let mut stark = SS[i] * ane * f / dfr2 / dfrs;
|
||||
if config.ifstrk == 0 {
|
||||
stark = 0.0;
|
||||
}
|
||||
|
||||
let mut sn_val = SN[i];
|
||||
if config.ifnat == 0 {
|
||||
sn_val = 0.0;
|
||||
}
|
||||
|
||||
let mut sr_val = SR[i];
|
||||
if config.ifres == 0 {
|
||||
sr_val = 0.0;
|
||||
}
|
||||
|
||||
let mut sgly = sn_val * (1.0 + sr_val * p) * cor / dfr2 + stark;
|
||||
sgly *= params.wn_hint[i + 1];
|
||||
|
||||
let mut gama = 1.0 / (GA[i] + GS[i] * ane * f / dfrs);
|
||||
if config.ifprd == 0 {
|
||||
gama = 0.0;
|
||||
}
|
||||
|
||||
ably += p * sgly;
|
||||
emly += params.pop_excited[i] * sgly * BNLY[i] * (1.0 - gama);
|
||||
if config.ifsti != 0 {
|
||||
ably -= params.pop_excited[i] * sgly / ((i + 2) * (i + 2)) as f64;
|
||||
}
|
||||
scly += p * sgly * gama;
|
||||
}
|
||||
|
||||
LymlinResult { ably, emly, scly }
|
||||
}
|
||||
|
||||
/// Lyman-alpha Stark broadening after N. Feautrier.
|
||||
///
|
||||
/// Interpolates in tabulated Stark broadening profiles.
|
||||
/// This is a simplified version used specifically in LYMLIN.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `freq` - Frequency (Hz)
|
||||
/// * `_id` - Depth index (unused in this implementation)
|
||||
/// * `t` - Temperature (K)
|
||||
/// * `ane` - Electron density
|
||||
///
|
||||
/// # Returns
|
||||
/// Broadening factor
|
||||
pub fn feautr_lyman(freq: f64, _id: usize, _t: f64, _ane: f64) -> f64 {
|
||||
// Tabulated wavelength offsets (Å from line center)
|
||||
const DL: [f64; 20] = [
|
||||
-150.0, -120.0, -90.0, -60.0, -40.0, -20.0, -10.0, -8.0, -4.0, -2.0,
|
||||
2.0, 4.0, 8.0, 10.0, 20.0, 40.0, 60.0, 90.0, 120.0, 150.0,
|
||||
];
|
||||
|
||||
// Tabulated profiles at different electron densities
|
||||
const F05: [f64; 20] = [
|
||||
0.0537, 0.0964, 0.1330, 0.3105, 0.4585, 0.6772, 0.8229, 0.8556, 0.9250, 0.9618,
|
||||
0.9733, 1.1076, 1.0644, 1.0525, 0.8841, 0.8282, 0.7541, 0.7091, 0.7164, 0.7672,
|
||||
];
|
||||
const F10: [f64; 20] = [
|
||||
0.1986, 0.2764, 0.3959, 0.5740, 0.7385, 0.9448, 1.0292, 1.0317, 0.9947, 0.8679,
|
||||
0.8648, 0.9815, 1.0660, 1.0793, 1.0699, 1.0357, 0.9245, 0.8603, 0.8195, 0.7928,
|
||||
];
|
||||
const F20: [f64; 20] = [
|
||||
0.4843, 0.5821, 0.7003, 0.8411, 0.9405, 1.0300, 1.0029, 0.9753, 0.8478, 0.6851,
|
||||
0.6861, 0.8554, 0.9916, 1.0264, 1.0592, 1.0817, 1.0575, 1.0152, 0.9761, 0.9451,
|
||||
];
|
||||
const F40: [f64; 20] = [
|
||||
0.7862, 0.8566, 0.9290, 0.9915, 1.0066, 0.9878, 0.8983, 0.8513, 0.6881, 0.5277,
|
||||
0.5302, 0.6920, 0.8607, 0.9111, 0.9651, 1.0793, 1.1108, 1.1156, 1.1003, 1.0839,
|
||||
];
|
||||
|
||||
let dlam = 2.997925e18 / freq - 1215.685;
|
||||
|
||||
// Find interval
|
||||
let mut i = 20;
|
||||
for k in 1..20 {
|
||||
if dlam <= DL[k] {
|
||||
i = k;
|
||||
break;
|
||||
}
|
||||
}
|
||||
let j = i - 1;
|
||||
let c = DL[j] - DL[i];
|
||||
let a = (dlam - DL[i]) / c;
|
||||
let b = (DL[j] - dlam) / c;
|
||||
|
||||
// Interpolate profiles
|
||||
let mut x = [0.0; 4];
|
||||
x[0] = F05[j] * a + F05[i] * b;
|
||||
x[1] = F10[j] * a + F10[i] * b;
|
||||
x[2] = F20[j] * a + F20[i] * b;
|
||||
x[3] = F40[j] * a + F40[i] * b;
|
||||
|
||||
// Temperature interpolation (simplified - using average)
|
||||
// In full implementation, would use JT, TI0, TI1, TI2 arrays
|
||||
let y = (x[0] + x[1] + x[2] + x[3]) / 4.0;
|
||||
|
||||
0.5 * (y + 1.0)
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_lymlin_basic() {
|
||||
let params = LymlinParams {
|
||||
id: 0,
|
||||
freq: 2.466e15, // Near Lyman alpha
|
||||
pop_h: 1.0e16,
|
||||
pop_excited: [1.0e10, 1.0e9, 1.0e8, 1.0e7],
|
||||
t: 6000.0,
|
||||
ane: 1.0e13,
|
||||
iath: 1,
|
||||
iophli: 1,
|
||||
wn_hint: [1.0, 1.0, 1.0, 1.0, 1.0],
|
||||
};
|
||||
|
||||
let result = lymlin(¶ms);
|
||||
assert!(result.ably.is_finite());
|
||||
assert!(result.emly.is_finite());
|
||||
assert!(result.scly.is_finite());
|
||||
assert!(result.ably >= 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_lymlin_far_from_line() {
|
||||
// Far from Lyman lines, opacity should be small
|
||||
let params = LymlinParams {
|
||||
id: 0,
|
||||
freq: 1.0e15, // Far from Lyman alpha
|
||||
pop_h: 1.0e16,
|
||||
pop_excited: [1.0e10, 1.0e9, 1.0e8, 1.0e7],
|
||||
t: 6000.0,
|
||||
ane: 1.0e13,
|
||||
iath: 1,
|
||||
iophli: 1,
|
||||
wn_hint: [1.0, 1.0, 1.0, 1.0, 1.0],
|
||||
};
|
||||
|
||||
let result = lymlin(¶ms);
|
||||
assert!(result.ably < 1e10); // Should be relatively small
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_lymlin_no_hydrogen() {
|
||||
// Without hydrogen, opacity should be zero
|
||||
let params = LymlinParams {
|
||||
id: 0,
|
||||
freq: 2.466e15,
|
||||
pop_h: 1.0e16,
|
||||
pop_excited: [1.0e10, 1.0e9, 1.0e8, 1.0e7],
|
||||
t: 6000.0,
|
||||
ane: 1.0e13,
|
||||
iath: 0, // No hydrogen
|
||||
iophli: 1,
|
||||
wn_hint: [1.0, 1.0, 1.0, 1.0, 1.0],
|
||||
};
|
||||
|
||||
let result = lymlin(¶ms);
|
||||
assert_eq!(result.ably, 0.0);
|
||||
assert_eq!(result.emly, 0.0);
|
||||
assert_eq!(result.scly, 0.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_feautr_lyman_basic() {
|
||||
let result = feautr_lyman(2.466e15, 0, 6000.0, 1.0e13);
|
||||
assert!(result.is_finite());
|
||||
assert!(result > 0.0);
|
||||
}
|
||||
}
|
||||
154
src/synspec/math/matinv.rs
Normal file
154
src/synspec/math/matinv.rs
Normal file
@ -0,0 +1,154 @@
|
||||
//! 矩阵求逆（高斯-约旦消元法）。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `matinv.f`
|
||||
//!
|
||||
//! 使用高斯-约旦消元法就地求逆矩阵。
|
||||
|
||||
/// 矩阵求逆（就地操作）。
|
||||
///
|
||||
/// 使用高斯-约旦消元法对 N×N 矩阵 A 求逆。
|
||||
/// 矩阵以行优先存储，最大维度为 NR×NR，实际使用 N×N。
|
||||
/// 求逆结果直接写回 A。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `a` - 输入矩阵（行优先存储），求逆后被替换为逆矩阵
|
||||
/// * `n` - 矩阵实际维数
|
||||
/// * `nr` - 矩阵最大维数（行优先存储的列数）
|
||||
pub fn matinv(a: &mut [f64], n: usize, nr: usize) {
|
||||
if n == 0 {
|
||||
return;
|
||||
}
|
||||
|
||||
// 1x1 矩阵的特殊情况
|
||||
if n == 1 {
|
||||
a[0] = 1.0 / a[0];
|
||||
return;
|
||||
}
|
||||
|
||||
// 创建增广矩阵 [A | I]，使用 n×2n 的工作空间
|
||||
let n2 = 2 * n;
|
||||
let mut aug = vec![0.0; n * n2];
|
||||
for i in 0..n {
|
||||
for j in 0..n {
|
||||
aug[i * n2 + j] = a[i * nr + j];
|
||||
}
|
||||
aug[i * n2 + n + i] = 1.0;
|
||||
}
|
||||
|
||||
// 前向消元（带部分选主元）
|
||||
for col in 0..n {
|
||||
// 选主元
|
||||
let mut max_val = aug[col * n2 + col].abs();
|
||||
let mut max_row = col;
|
||||
for row in (col + 1)..n {
|
||||
let val = aug[row * n2 + col].abs();
|
||||
if val > max_val {
|
||||
max_val = val;
|
||||
max_row = row;
|
||||
}
|
||||
}
|
||||
|
||||
// 交换行
|
||||
if max_row != col {
|
||||
for j in 0..n2 {
|
||||
aug.swap(col * n2 + j, max_row * n2 + j);
|
||||
}
|
||||
}
|
||||
|
||||
// 缩放主行
|
||||
let pivot = aug[col * n2 + col];
|
||||
if pivot.abs() < 1e-30 {
|
||||
return; // 奇异矩阵
|
||||
}
|
||||
for j in 0..n2 {
|
||||
aug[col * n2 + j] /= pivot;
|
||||
}
|
||||
|
||||
// 消元
|
||||
for row in 0..n {
|
||||
if row != col {
|
||||
let factor = aug[row * n2 + col];
|
||||
for j in 0..n2 {
|
||||
aug[row * n2 + j] -= factor * aug[col * n2 + j];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 提取逆矩阵
|
||||
for i in 0..n {
|
||||
for j in 0..n {
|
||||
a[i * nr + j] = aug[i * n2 + n + j];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
use approx::assert_relative_eq;
|
||||
|
||||
#[test]
|
||||
fn test_matinv_1x1() {
|
||||
let mut a = vec![4.0];
|
||||
matinv(&mut a, 1, 1);
|
||||
assert_relative_eq!(a[0], 0.25, epsilon = 1e-12);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_matinv_2x2() {
|
||||
// [1 2]
|
||||
// [3 4]
|
||||
// det = 1*4 - 2*3 = -2
|
||||
// 逆矩阵: [-2 1]
|
||||
// [1.5 -0.5]
|
||||
let mut a = vec![1.0, 2.0, 3.0, 4.0];
|
||||
matinv(&mut a, 2, 2);
|
||||
|
||||
assert_relative_eq!(a[0], -2.0, epsilon = 1e-10);
|
||||
assert_relative_eq!(a[1], 1.0, epsilon = 1e-10);
|
||||
assert_relative_eq!(a[2], 1.5, epsilon = 1e-10);
|
||||
assert_relative_eq!(a[3], -0.5, epsilon = 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_matinv_identity() {
|
||||
// 单位矩阵的逆是自身
|
||||
let mut a = vec![1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0];
|
||||
matinv(&mut a, 3, 3);
|
||||
|
||||
assert_relative_eq!(a[0], 1.0, epsilon = 1e-12);
|
||||
assert_relative_eq!(a[4], 1.0, epsilon = 1e-12);
|
||||
assert_relative_eq!(a[8], 1.0, epsilon = 1e-12);
|
||||
assert_relative_eq!(a[1], 0.0, epsilon = 1e-12);
|
||||
assert_relative_eq!(a[2], 0.0, epsilon = 1e-12);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_matinv_roundtrip() {
|
||||
// A * A^{-1} 应该接近单位矩阵
|
||||
let orig = vec![2.0, 1.0, 0.0, 1.0, 3.0, 1.0, 0.0, 1.0, 2.0];
|
||||
let mut a = orig.clone();
|
||||
matinv(&mut a, 3, 3);
|
||||
|
||||
// 计算 orig * a（应该接近单位矩阵）
|
||||
let mut product = vec![0.0; 9];
|
||||
for i in 0..3 {
|
||||
for j in 0..3 {
|
||||
for k in 0..3 {
|
||||
product[i * 3 + j] += orig[i * 3 + k] * a[k * 3 + j];
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 检查对角线接近 1
|
||||
assert_relative_eq!(product[0], 1.0, epsilon = 1e-10);
|
||||
assert_relative_eq!(product[4], 1.0, epsilon = 1e-10);
|
||||
assert_relative_eq!(product[8], 1.0, epsilon = 1e-10);
|
||||
|
||||
// 检查非对角线接近 0
|
||||
assert_relative_eq!(product[1], 0.0, epsilon = 1e-10);
|
||||
assert_relative_eq!(product[2], 0.0, epsilon = 1e-10);
|
||||
}
|
||||
}
|
||||
@ -2,58 +2,415 @@
|
||||
//!
|
||||
//! 重构自 SYNSPEC Fortran 代码。
|
||||
|
||||
mod abnchn;
|
||||
mod allard;
|
||||
mod carbon;
|
||||
mod change;
|
||||
mod cia;
|
||||
mod chckab;
|
||||
mod count_words;
|
||||
mod crosew;
|
||||
mod divhe2;
|
||||
mod densit;
|
||||
mod eldens;
|
||||
mod divstr;
|
||||
mod dwnfr0;
|
||||
mod eospri;
|
||||
mod dwnfr1;
|
||||
mod eps;
|
||||
mod exopf;
|
||||
mod expint;
|
||||
mod extprf;
|
||||
mod feautr;
|
||||
mod fingrd;
|
||||
mod frac1;
|
||||
mod fractn;
|
||||
mod gamhe;
|
||||
mod gaunt;
|
||||
mod getlal;
|
||||
mod getwrd;
|
||||
mod gfree;
|
||||
mod gomini;
|
||||
mod ghydop;
|
||||
mod griem;
|
||||
mod gvdw;
|
||||
mod he2lin;
|
||||
mod he2sew;
|
||||
mod he2set;
|
||||
mod he1ini;
|
||||
mod he2ini;
|
||||
mod hephot;
|
||||
mod hidalg;
|
||||
mod heset;
|
||||
mod hydlin;
|
||||
mod hydini;
|
||||
mod hydliw;
|
||||
mod hydtab;
|
||||
mod hylset;
|
||||
mod hylsew;
|
||||
mod inibl0;
|
||||
mod inibl1;
|
||||
mod inibla;
|
||||
mod iniblh;
|
||||
mod iniblm;
|
||||
mod inilin;
|
||||
mod inilin_grid;
|
||||
mod inimod;
|
||||
mod ingrid;
|
||||
mod inmoli;
|
||||
mod inpmod;
|
||||
mod inkur;
|
||||
mod initia_synspec;
|
||||
mod iniset;
|
||||
mod inpbf;
|
||||
mod idmtab;
|
||||
mod idtab;
|
||||
mod interp;
|
||||
mod irwpf;
|
||||
mod inthe2;
|
||||
mod inthyd;
|
||||
mod intxen;
|
||||
mod intrp;
|
||||
mod linop;
|
||||
mod linopw;
|
||||
mod levsol;
|
||||
mod lineqs;
|
||||
mod ispec;
|
||||
mod locate;
|
||||
mod lyahhe;
|
||||
mod lymlin;
|
||||
mod matinv;
|
||||
mod molini;
|
||||
mod molop;
|
||||
mod molset;
|
||||
mod mpartf;
|
||||
mod opac;
|
||||
mod opacon;
|
||||
mod opacw;
|
||||
mod opdata;
|
||||
mod ratmat;
|
||||
mod opadd;
|
||||
mod ougrid;
|
||||
mod outpri;
|
||||
mod partdv;
|
||||
mod partf;
|
||||
mod pfheav;
|
||||
mod phe1;
|
||||
mod phe2;
|
||||
mod profil;
|
||||
mod phtion;
|
||||
mod readph;
|
||||
mod resolv;
|
||||
mod resolw;
|
||||
mod rhonen;
|
||||
mod rte;
|
||||
mod rtecd;
|
||||
mod rtedfe;
|
||||
mod rtesca;
|
||||
mod rtewin;
|
||||
mod phtx;
|
||||
mod radtem;
|
||||
mod pretab;
|
||||
mod rdata;
|
||||
mod readbf;
|
||||
mod reiman;
|
||||
mod sbfhe1;
|
||||
mod sbfhmi;
|
||||
mod sbfhmi_old;
|
||||
mod sabolf;
|
||||
mod sbfch;
|
||||
mod sbfoh;
|
||||
mod sffhmi;
|
||||
mod sffhmi_old;
|
||||
mod sghe12;
|
||||
mod sgmerg;
|
||||
mod setray;
|
||||
mod setwin;
|
||||
mod sigavs;
|
||||
mod sigk;
|
||||
mod spsigk;
|
||||
mod start;
|
||||
mod state;
|
||||
mod state0;
|
||||
mod stark0;
|
||||
mod starka;
|
||||
mod starkir;
|
||||
mod timing;
|
||||
mod tint;
|
||||
mod todens;
|
||||
mod topbas;
|
||||
mod tridag;
|
||||
mod voigte;
|
||||
mod voigtk;
|
||||
mod velset;
|
||||
mod vopf;
|
||||
mod h2minus;
|
||||
mod h2opf;
|
||||
mod wn;
|
||||
mod wnstor;
|
||||
mod wtot;
|
||||
mod pffe;
|
||||
mod pfni;
|
||||
mod pfspec;
|
||||
mod nlte;
|
||||
mod nltset;
|
||||
mod nstpar;
|
||||
mod moleq;
|
||||
mod russel;
|
||||
mod quit;
|
||||
mod rotin;
|
||||
mod wgtjh1;
|
||||
mod xk2dop;
|
||||
mod xenini;
|
||||
mod yint;
|
||||
mod ylintp;
|
||||
|
||||
pub use abnchn::{abnchn, AbnchnMode, AbnchnParams, AbnchnOutput};
|
||||
pub use allard::{allard, AllardData, AllardTable, NXMAX, NNMAX};
|
||||
pub use carbon::carbon;
|
||||
pub use change::{change, ChangeLevelParams, ChangeParams, ChangeOutput};
|
||||
pub use cia::{cia_h2h2, cia_h2h2_init, cia_h2h, cia_h2h_init, cia_h2he, cia_h2he_init, cia_hhe, cia_hhe_init};
|
||||
pub use chckab::{chckab, ChckabParams, ChckabResult};
|
||||
pub use count_words::count_words;
|
||||
pub use crosew::{croset, crosew, CrosetParams, CrosewParams};
|
||||
pub use divhe2::divhe2;
|
||||
pub use densit::{densit, DensitMode, DensitParams, DensitResult};
|
||||
pub use eldens::{eldens, EldensParams, EldensResult};
|
||||
pub use divstr::divstr;
|
||||
pub use dwnfr0::dwnfr0;
|
||||
pub use eospri::{eospri, EospriParams, EospriOutput, EldensSimpleResult};
|
||||
pub use dwnfr1::{dwnfr1, Dwnfr1Params};
|
||||
pub use eps::eps;
|
||||
pub use exopf::exopf;
|
||||
pub use expint::expint;
|
||||
pub use extprf::extprf;
|
||||
pub use feautr::{feautr, FeautrParams};
|
||||
pub use fingrd::{
|
||||
fingrd, FingrdParams,
|
||||
wavelength_to_frequency, frequency_to_wavelength,
|
||||
generate_log_grid, generate_linear_grid_in_log,
|
||||
compute_opacity_stats,
|
||||
OpacityTable, OpacityTableStats, OpacityFlags,
|
||||
};
|
||||
pub use frac1::{frac1, Frac1Params, Frac1Result, FracOpData, MTEMP, MELEC, MION1};
|
||||
pub use fractn::{fractn, FracOp, MDAT};
|
||||
pub use gamhe::{gamhe, GamheData, GamheParams};
|
||||
pub use gaunt::{gaunt, gntk};
|
||||
pub use getlal::getlal;
|
||||
pub use getwrd::getwrd;
|
||||
pub use gfree::gfree;
|
||||
pub use gomini::{gomini, gomini_interpolate, GominiParams, GominiResult};
|
||||
pub use ghydop::{ghydop, GhydopParams, GhydopResult};
|
||||
pub use griem::{griem, GriemParams};
|
||||
pub use gvdw::{gvdw, GvdwParams};
|
||||
pub use he2lin::{he2lin, he2liw, He2Common, He2linParams, He2linResult, He2liwParams, He2liwWindowParams};
|
||||
pub use he2sew::{he2sew, He2WindowParams};
|
||||
pub use he2set::{he2set, He2setParams, He2setOutput};
|
||||
pub use he1ini::{he1ini, He1ProfileData, He1Profile4471, He1ProfileOther};
|
||||
pub use he2ini::{he2ini, He2InitResult, He2ProfileTable};
|
||||
pub use hephot::hephot;
|
||||
pub use hidalg::hidalg;
|
||||
pub use hydlin::{hydlin, HydlinParams, HydlinResult};
|
||||
pub use hydliw::{hydliw, HydliwCommon, HydliwParams, HydliwResult, HydliwWindowParams};
|
||||
pub use hydini::{hydini, HydiniParams, HydInitResult, HydProfileTable, HydTableSource};
|
||||
pub use hydtab::{hydtab, HydtabParams};
|
||||
pub use heset::{heset, HesetInput, HesetOutput};
|
||||
pub use hylset::{hylset, HylsetParams, HylsetOutput};
|
||||
pub use hylsew::{hylsew, HylsewOutput};
|
||||
pub use inibl0::{
|
||||
compute_wavelength_range, compute_frequency_range,
|
||||
compute_angle_points_uniform, compute_angle_points_sine,
|
||||
compute_continuum_frequency_grid, parse_nlte_mode, check_velocity_cutoff,
|
||||
WavelengthRange, FrequencyRange, AngleConfig, ContinuumFreqGrid,
|
||||
};
|
||||
pub use inibl1::{
|
||||
auto_detect_relop, reset_wavelength_range,
|
||||
compute_continuum_grid_inibl1, select_standard_opacity,
|
||||
limit_opacity, compute_window_opacity, init_dissolved_fractions,
|
||||
compute_inibl1, Inibl1Config, Inibl1Result,
|
||||
};
|
||||
pub use inibla::{inibla, IniblaParams, IniblaOutput, compute_doppler_width, compute_vdw_width, compute_planck, CL, H, BOLK, BN, HK};
|
||||
pub use iniblh::{iniblh, IniblhParams, IniblhOutput, HydrogenLineInfo};
|
||||
pub use iniblm::{iniblm, IniblmParams, IniblmOutput, compute_molecular_doppler_width, compute_molecular_planck};
|
||||
pub use inilin::{
|
||||
air_to_vacuum, parse_element_code, compute_line_strength,
|
||||
line_selected_old, line_selected_new, compute_extinction,
|
||||
ensure_level_order, excitation_temperature_index,
|
||||
natural_broadening, stark_broadening, vdw_broadening,
|
||||
read_line_list, InilinLineData, InilinConfig, InilinOutput,
|
||||
};
|
||||
pub use inilin_grid::{
|
||||
compute_line_strength_grid, effective_quantum_number_squared,
|
||||
natural_broadening_grid, stark_broadening_grid, vdw_broadening_grid,
|
||||
line_selected_grid, parse_kurucz_code, ensure_level_order as ensure_level_order_grid,
|
||||
compute_line_broadening_grid,
|
||||
AtomicLineParams, LineStrengthGrid, LineBroadeningGrid,
|
||||
};
|
||||
pub use inimod::{inimod, InimodParams, InimodResult};
|
||||
pub use ingrid::{
|
||||
ingrid, IngridParams, IngridResult,
|
||||
generate_temperature_grid, generate_density_grid_uniform,
|
||||
generate_density_grid_variable, set_grid_from_model,
|
||||
log_average_interpolation, advance_grid_point,
|
||||
DensityParameterType, OpacityGridParams, InterpolatedOpacity, GridTraversalState,
|
||||
};
|
||||
pub use inmoli::{
|
||||
kurucz_to_tsuji, compute_molecular_line_strength, line_selected_molecular,
|
||||
compute_cutoff_distance, compute_line_broadening, molecular_doppler_width,
|
||||
inmoli, InmoliOutput, SelectedMolLine, MolLineFormat,
|
||||
MolecularLine, MolecularLineStrength, LineBroadening,
|
||||
};
|
||||
pub use inpmod::{
|
||||
total_number_density, bound_free_constant, compute_atom_densities,
|
||||
replace_nlte_populations, apply_bfactor_correction, compute_nlte_correction,
|
||||
compute_model_stats, detect_model_format,
|
||||
ModelDepthPoint, ModelAtmosphere, ModelStats, ModelFormat,
|
||||
};
|
||||
pub use inkur::{inkur, InkurParams, InkurResult};
|
||||
pub use initia_synspec::{
|
||||
initia, InitiaOutput, InitiaConfig, ExplicitIonInput, ExplicitLevelData,
|
||||
IonSetupResult, OpacitySwitches, NonStdParams, NonStdFileParams,
|
||||
get_ground_state_weight, IonParams, IonIndices, compute_ion_indices,
|
||||
assign_levels, setup_turbulent_velocity, HydrogenHeliumIds,
|
||||
identify_hydrogen_helium, compute_hydrogen_level_bounds,
|
||||
FrequencyReadMode, parse_frequency_read_mode,
|
||||
};
|
||||
pub use iniset::{iniset, InisetParams, InisetResult};
|
||||
pub use inpbf::{inpbf, apply_inpbf, InpbfParams, InpbfResult};
|
||||
pub use idmtab::{idmtab_compute, IdmtabLine, IdmtabResult};
|
||||
pub use idtab::{idtab_compute, IdtabLine, IdtabResult};
|
||||
pub use interp::interp;
|
||||
pub use irwpf::irwpf;
|
||||
pub use inthe2::{inthe2, Inthe2Params, Inthe2Result};
|
||||
pub use inthyd::{inthyd, InthydParams, InthydResult};
|
||||
pub use intxen::{intxen, IntxenParams, IntxenResult};
|
||||
pub use intrp::{intrp, intrp_to_vec};
|
||||
pub use linop::{linop, LinopParams, LinopResult, LineData, Phe1Data, Phe2Common};
|
||||
pub use linopw::{linopw, LinopwParams, LinopwResult, LinopwLineData, Phe1DataW, Phe2CommonW};
|
||||
pub use levsol::levsol;
|
||||
pub use lineqs::lineqs;
|
||||
pub use ispec::{ispec, PROFILE_VOIGT, PROFILE_HYDROGEN, PROFILE_HEI_4471, PROFILE_HEI_4388, PROFILE_HEI_4026};
|
||||
pub use locate::locate;
|
||||
pub use lyahhe::{lyahhe, lyahhe_init};
|
||||
pub use lymlin::{lymlin, feautr_lyman, LymlinParams, LymlinResult};
|
||||
pub use matinv::matinv;
|
||||
pub use molini::{molini, MoliniParams, MoliniResult};
|
||||
pub use molop::{molop, MolopConfig, MolopModelState, MolopFreqParams, MolLineData, MolModelState, MolopOutput};
|
||||
pub use molset::{molset, MolsetParams, MolsetOutput};
|
||||
pub use mpartf::{mpartf, mpartf_init};
|
||||
pub use opac::{opac, OpacParams, OpacResult};
|
||||
pub use opacon::{opacon, OpaconParams, OpaconResult};
|
||||
pub use opacw::{opacw, OpacwParams, OpacwResult};
|
||||
pub use opdata::{opdata, OpData, MMAXOP, MOP};
|
||||
pub use opadd::{opadd, OpaddParams, OpaddResult};
|
||||
pub use ratmat::ratmat;
|
||||
pub use ougrid::{ougrid, OugridParams, OugridOutput};
|
||||
pub use outpri::{outpri, OutpriParams, OutpriResult};
|
||||
pub use partdv::{partdv, partdv_with_params, PartdvParams};
|
||||
pub use partf::partf;
|
||||
pub use pfheav::pfheav;
|
||||
pub use quit::quit;
|
||||
pub use rotin::{kernel, rotins, RotinsResult};
|
||||
pub use phe1::{phe1, Phe1Params};
|
||||
pub use phe2::{phe2, Phe2Params, Phe2Output};
|
||||
pub use profil::{profil, LineBroadeningData, ProfilParams};
|
||||
pub use phtion::{phtion, phtion_pure, PhtionParams, PhtionOutput, PhotcsData, MFRQ};
|
||||
pub use phtx::{phtx, PhtxParams, PhtxOutput, PhtxState, LevelPhotoData, HhePhotoData};
|
||||
pub use radtem::radtem;
|
||||
pub use rdata::{
|
||||
detect_and_convert_energy, partition_function_ratio, ionization_constant,
|
||||
rdata, ContinuumTransition, IonData, LevelData, EnergyUnit,
|
||||
};
|
||||
pub use readbf::{readbf, readbf_stdin};
|
||||
pub use readph::{readph, readph_build_index, ReadphParams, ReadphResult};
|
||||
pub use resolv::{resolv, ResolvParams, ResolvResult};
|
||||
pub use resolw::{
|
||||
compute_fine_frequency_grid, compute_core_radius_factor,
|
||||
store_continuum_opacity, compute_thermal_source_function,
|
||||
interpolate_to_fine_radial_grid, scale_flux,
|
||||
resolw, ResolwParams, ResolwOutput,
|
||||
FineFrequencyGrid, ContinuumOpacity,
|
||||
};
|
||||
pub use rhonen::{rhonen, RhonenParams, RhonenResult};
|
||||
pub use rte::{rte, RteParams, RteResult};
|
||||
pub use rtecd::{rtecd, RtecdParams, RtecdResult};
|
||||
pub use rtedfe::{rtedfe, RtedfeParams, RtedfeResult};
|
||||
pub use rtewin::{rtewin, RtewinParams, RtewinResult};
|
||||
pub use rtesca::{rtesca, RtescaParams, RtescaResult};
|
||||
pub use pretab::{pretab, VoigtTables};
|
||||
pub use reiman::reiman;
|
||||
pub use sbfhe1::sbfhe1;
|
||||
pub use sbfhmi::sbfhmi;
|
||||
pub use sbfhmi_old::sbfhmi_old;
|
||||
pub use sabolf::{sabolf, SabolfParams, SabolfResult};
|
||||
pub use sbfch::sbfch;
|
||||
pub use sbfoh::sbfoh;
|
||||
pub use sffhmi::sffhmi;
|
||||
pub use sffhmi_old::sffhmi_old;
|
||||
pub use sghe12::sghe12;
|
||||
pub use sgmerg::{sgmerg, sgmerg_pure, SgmergParams};
|
||||
pub use setray::{setray, SetrayParams, SetrayResult};
|
||||
pub use setwin::{setwin, SetWinParams};
|
||||
pub use sigavs::{
|
||||
clip_cross_section_to_range, find_max_cross_section,
|
||||
format_cross_section_storage, iron_ionization_energy,
|
||||
sigavs, SigavsParams, SigavsOutput,
|
||||
CrossSectionPoint, SuperLevelCrossSection,
|
||||
};
|
||||
pub use sigk::{sigk, SigkParams};
|
||||
pub use spsigk::spsigk;
|
||||
pub use start::{start, StartInput, StartResult};
|
||||
pub use state::{state, StateParams};
|
||||
pub use state0::{
|
||||
get_atomic_mass, get_ionization_potential, get_max_ionization,
|
||||
get_solar_abundance, ionization_potential_class, adjust_ionization_for_hot_star,
|
||||
state0, State0Output,
|
||||
D_DATA, ABUN0, ABUN1, XI_DATA, NATOM_DATA, MION0, ENHE1_EV, ENHE2_EV,
|
||||
};
|
||||
pub use stark0::{stark0, Stark0Result};
|
||||
pub use starka::starka;
|
||||
pub use starkir::starkir;
|
||||
pub use timing::{timing, TimingResult, TIMING_TABLE, TIMING_FINAL};
|
||||
pub use tint::{tint, TintResult};
|
||||
pub use todens::{todens, TodensParams, TodensResult};
|
||||
pub use topbas::topbas;
|
||||
pub use tridag::tridag;
|
||||
pub use voigte::voigte;
|
||||
pub use voigtk::{voigtk, MVOI};
|
||||
pub use velset::{
|
||||
beta_velocity, compute_depth_from_mass, density_from_velocity,
|
||||
velocity_from_density, mass_loss_constant, stellar_radius_to_cm,
|
||||
compute_velocity_field, velset, VelocityFieldParams, VelsetOutput,
|
||||
WindInputParams,
|
||||
};
|
||||
pub use vopf::vopf;
|
||||
pub use h2minus::h2minus;
|
||||
pub use h2opf::h2opf;
|
||||
pub use wn::wn;
|
||||
pub use wgtjh1::{wgtjh1, Wgtjh1Params, Wgtjh1Result};
|
||||
pub use wnstor::{wnstor, WnstorParams, WnstorOutput};
|
||||
pub use wtot::{wtot, LINE_4471, LINE_4387, LINE_4026, LINE_4922};
|
||||
pub use pffe::pffe;
|
||||
pub use pfni::pfni;
|
||||
pub use pfspec::{
|
||||
compute_partition_function, compute_partition_function_screened,
|
||||
hydrogen_partition_function, he1_partition_function,
|
||||
simplified_partition_function, partition_function_with_derivative,
|
||||
EnergyLevel, IonPartitionData, PartitionFunctionResult,
|
||||
};
|
||||
pub use nlte::{nlte, NlteParams, NlteOutput};
|
||||
pub use nstpar::{nstpar, NstparParams};
|
||||
pub use moleq::{moleq, MoleqParams, MoleqResult};
|
||||
pub use russel::{russel, RusselParams, RusselResult, MoleculeData};
|
||||
pub use nltset::{
|
||||
find_level_by_energy, find_level_by_energy_and_quantum,
|
||||
find_level_by_quantum_number, compute_nlte_level_indices,
|
||||
nltset, NlteIonData, NltsetOutput,
|
||||
energy_difference_cm, planck_normalization,
|
||||
LevelMatchMode, Parity, QuantumNumbers, EnergyLimits,
|
||||
QuantumNumberLimits, LevelMatchResult, NlteLevelConfig, NlteLevelResult,
|
||||
};
|
||||
pub use xk2dop::xk2dop;
|
||||
pub use xenini::{xenini, XenInitResult, XenProfileData};
|
||||
pub use yint::yint;
|
||||
pub use ylintp::ylintp;
|
||||
|
||||
445
src/synspec/math/moleq.rs
Normal file
445
src/synspec/math/moleq.rs
Normal file
@ -0,0 +1,445 @@
|
||||
//! moleq — 分子平衡计算（SYNSPEC 版本）。
|
||||
//!
|
||||
//! Fortran 原始签名: SUBROUTINE MOLEQ(ID,TT,AN,AEIN,ANE,IPRI)
|
||||
//!
|
||||
//! 计算原子和分子的平衡状态。
|
||||
//! 调用 RUSSEL 求解 Russell 方程，然后计算各元素、离子和分子的数密度。
|
||||
//!
|
||||
//! 注意: 这是 SYNSPEC 的版本，与 TLUSTY 的 `moleq` 不同。
|
||||
//! TLUSTY 版本在 `src/tlusty/math/eos/moleq.rs` 中。
|
||||
|
||||
use super::russel::{self, MoleculeData, RusselParams, RusselResult};
|
||||
|
||||
/// 物理常数
|
||||
const ECONST: f64 = 4.342945e-1; // 1/ln(10)
|
||||
#[allow(dead_code)]
|
||||
const AVO: f64 = 0.602217e+24; // Avogadro 数
|
||||
#[allow(dead_code)]
|
||||
const SPA: f64 = 0.196e-01;
|
||||
#[allow(dead_code)]
|
||||
const GRA: f64 = 0.275423e+05;
|
||||
#[allow(dead_code)]
|
||||
const AHE: f64 = 0.100e+00;
|
||||
|
||||
/// 分子平衡计算结果
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct MoleqResult {
|
||||
/// 电子数密度 [cm⁻³]
|
||||
pub ane: f64,
|
||||
/// 总质量密度 [g/cm³]
|
||||
pub dens: f64,
|
||||
/// 平均分子量
|
||||
pub wmm: f64,
|
||||
/// 原子数密度 [element_id] (中性)
|
||||
pub anato: Vec<f64>,
|
||||
/// 离子数密度 [element_id] (一级电离)
|
||||
pub anion: Vec<f64>,
|
||||
/// 二级电离数密度 [element_id]
|
||||
pub anion2: Vec<f64>,
|
||||
/// 分子数密度 [molecule_id]
|
||||
pub anmol: Vec<f64>,
|
||||
/// 原子配分函数 [element_id]
|
||||
pub pfato: Vec<f64>,
|
||||
/// 离子配分函数 [element_id]
|
||||
pub pfion: Vec<f64>,
|
||||
/// 分子配分函数 [molecule_id]
|
||||
pub pfmol: Vec<f64>,
|
||||
/// N/U 比 [depth, ion_stage, element_id]
|
||||
pub rrr: Vec<Vec<Vec<f64>>>,
|
||||
/// 分子 N/U 比 [molecule_id, depth]
|
||||
pub rrmol: Vec<Vec<f64>>,
|
||||
/// H- 数密度
|
||||
pub anhm: f64,
|
||||
/// H2 数密度
|
||||
pub anh2: f64,
|
||||
/// CH 数密度
|
||||
pub anch: f64,
|
||||
/// OH 数密度
|
||||
pub anoh: f64,
|
||||
/// Russell 方程结果
|
||||
pub russel: RusselResult,
|
||||
}
|
||||
|
||||
/// 分子平衡计算参数
|
||||
pub struct MoleqParams<'a> {
|
||||
/// 深度点索引 (0-indexed)
|
||||
pub id: usize,
|
||||
/// 温度 [K]
|
||||
pub tt: f64,
|
||||
/// 总粒子数密度 [cm⁻³]
|
||||
pub an: f64,
|
||||
/// 初始电子密度估计 [cm⁻³]
|
||||
pub aein: f64,
|
||||
/// 氢质量分数
|
||||
pub hmass: f64,
|
||||
/// 各元素的丰度 [element_id]
|
||||
pub abndd: &'a [f64],
|
||||
/// 电离势 [element_id][ion_stage]
|
||||
pub enev: &'a [Vec<f64>],
|
||||
/// 原子质量 [element_id]
|
||||
pub amas: &'a [f64],
|
||||
/// 原子序数到元素索引的映射
|
||||
pub iatex: &'a [usize],
|
||||
/// 深度点总数
|
||||
pub nd: usize,
|
||||
/// 最大金属元素数
|
||||
pub nmetal: usize,
|
||||
/// 分子数据
|
||||
pub molecules: &'a [MoleculeData],
|
||||
/// 配分函数回调: irwpf(element_id, ion_stage, molecule_id, temp) -> partition_function
|
||||
pub irwpf: &'a dyn Fn(usize, usize, usize, f64) -> f64,
|
||||
/// EXOMOL 配分函数回调: exopf(molecule_id, temp) -> partition_function
|
||||
pub exopf: &'a dyn Fn(usize, f64) -> f64,
|
||||
/// 打印级别
|
||||
pub iprin: i32,
|
||||
/// 是否使用 EXOMOL 数据
|
||||
pub ipfexo: i32,
|
||||
/// 分子表类型
|
||||
pub moltab: i32,
|
||||
/// 最大迭代次数
|
||||
pub nimax: usize,
|
||||
/// 收敛精度
|
||||
pub eps: f64,
|
||||
}
|
||||
|
||||
/// 分子平衡计算。
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// 1. 初始化原子数据（丰度、电离势、质量）
|
||||
/// 2. 调用 RUSSEL 求解 Russell 方程
|
||||
/// 3. 计算中性原子数密度
|
||||
/// 4. 计算一级电离离子数密度
|
||||
/// 5. 计算分子数密度
|
||||
/// 6. 更新密度、分子量和电子密度
|
||||
pub fn moleq(params: &MoleqParams) -> MoleqResult {
|
||||
let id = params.id;
|
||||
let tt = params.tt;
|
||||
let an = params.an;
|
||||
let aein = params.aein;
|
||||
let tk = 1.0 / (tt * 1.38054e-16);
|
||||
let pgas = an / tk;
|
||||
let sahcon = 1.87840e20 * tt * tt.sqrt();
|
||||
let theta = 5040.0 / tt;
|
||||
let tem = tt;
|
||||
|
||||
let nmetal = params.nmetal;
|
||||
let nmolec = params.molecules.len();
|
||||
|
||||
// 初始化原子数据
|
||||
let mut ccomp = vec![0.0_f64; 100];
|
||||
let mut xip = vec![0.0_f64; 100];
|
||||
let mut xi2 = vec![0.0_f64; 100];
|
||||
let mut emass = vec![0.0_f64; 100];
|
||||
let mut nelemx = vec![0_usize; nmetal];
|
||||
|
||||
for i in 0..nmetal {
|
||||
let ia = i;
|
||||
nelemx[i] = ia;
|
||||
if ia < params.abndd.len() {
|
||||
ccomp[ia] = params.abndd[ia];
|
||||
}
|
||||
if ia < params.enev.len() && !params.enev[ia].is_empty() {
|
||||
xip[ia] = params.enev[ia][0];
|
||||
}
|
||||
if ia < params.enev.len() && params.enev[ia].len() > 1 {
|
||||
xi2[ia] = params.enev[ia][1];
|
||||
}
|
||||
if ia < params.amas.len() {
|
||||
emass[ia] = params.amas[ia];
|
||||
}
|
||||
}
|
||||
|
||||
// 初始化分压
|
||||
let mut p = vec![0.0_f64; 100];
|
||||
for i in 0..nmetal {
|
||||
let nelemi = nelemx[i];
|
||||
p[nelemi] = 1.0e-70;
|
||||
}
|
||||
p[99] = aein / tk; // 电子压力
|
||||
|
||||
// 调用 Russell 方程求解器
|
||||
let irwpf_ref = params.irwpf;
|
||||
let pe_init = aein / tk; // P(99) in Fortran
|
||||
let russel_params = RusselParams {
|
||||
tem,
|
||||
pg: pgas,
|
||||
pe_init,
|
||||
ccomp: &ccomp,
|
||||
xip: &xip,
|
||||
xi2: &xi2,
|
||||
nelemx: &nelemx,
|
||||
nmetal,
|
||||
molecules: params.molecules,
|
||||
nimax: params.nimax,
|
||||
eps: params.eps,
|
||||
switer: 0.0,
|
||||
irwpf: irwpf_ref,
|
||||
iprin: params.iprin,
|
||||
};
|
||||
|
||||
let russel_result = russel::russel(&russel_params);
|
||||
let pe = russel_result.pe;
|
||||
let p_ref = &russel_result.p;
|
||||
|
||||
// 计算中性原子数密度
|
||||
let mut anato = vec![0.0_f64; nmetal];
|
||||
let mut pfato = vec![0.0_f64; nmetal];
|
||||
let mut tmass = 0.0_f64;
|
||||
|
||||
for i in 0..nmetal {
|
||||
let nelemi = nelemx[i];
|
||||
let anden = (p_ref[nelemi] + 1.0e-70) * tk;
|
||||
tmass += anden * emass[nelemi];
|
||||
let u0 = (params.irwpf)(nelemi, 1, 0, tt);
|
||||
pfato[nelemi] = u0;
|
||||
anato[nelemi] = anden;
|
||||
}
|
||||
let _an1 = anato[0]; // H 数密度
|
||||
|
||||
// 计算一级电离离子数密度
|
||||
let mut anion = vec![0.0_f64; nmetal];
|
||||
let mut pfion = vec![0.0_f64; nmetal];
|
||||
let mut anion2 = vec![0.0_f64; nmetal];
|
||||
|
||||
let pe_log = (pe + 1.0e-70).log10();
|
||||
|
||||
for i in 0..nmetal {
|
||||
let nelemi = nelemx[i];
|
||||
let plog = (p_ref[nelemi] + 1.0e-70).log10();
|
||||
let xkp_log = (russel_result.xkp[nelemi] + 1.0e-70).log10();
|
||||
let pionl = plog + xkp_log - pe_log;
|
||||
let anden_ion = (pionl / ECONST).exp() * tk;
|
||||
tmass += anden_ion * emass[nelemi];
|
||||
|
||||
let u1 = (params.irwpf)(nelemi, 2, 0, tt);
|
||||
pfion[nelemi] = u1;
|
||||
anion[nelemi] = anden_ion;
|
||||
|
||||
// 二级电离
|
||||
if (2..=30).contains(&nelemi) {
|
||||
let x2_log = (russel_result.xk2[nelemi] + 1.0e-70).log10();
|
||||
let pion2 = pionl + x2_log - pe_log;
|
||||
anion2[nelemi] = (pion2 / ECONST).exp() * tk;
|
||||
}
|
||||
}
|
||||
anion2[0] = 0.0; // H 没有二级电离
|
||||
|
||||
// 计算分子数密度
|
||||
let mut anmol = vec![0.0_f64; nmolec];
|
||||
let mut pfmol = vec![0.0_f64; nmolec];
|
||||
let mut anhm_val = 0.0_f64;
|
||||
let mut anh2_val = 0.0_f64;
|
||||
let mut anch_val = 0.0_f64;
|
||||
let mut anoh_val = 0.0_f64;
|
||||
|
||||
for j in 0..nmolec {
|
||||
let pmoll = (russel_result.ppmol[j] + 1.0e-70).log10();
|
||||
let anden_mol = (pmoll / ECONST).exp() * tk;
|
||||
let _jm = j + 2 * nmetal;
|
||||
|
||||
let mut umoll = 1.0_f64;
|
||||
if pmoll > -30.0 {
|
||||
umoll = (anden_mol).log10() + params.molecules[j].c[1] * theta;
|
||||
|
||||
let mut amasm = 0.0_f64;
|
||||
for m in 0..params.molecules[j].mmax {
|
||||
let i_elem = params.molecules[j].nelem[m];
|
||||
let natomj = params.molecules[j].natom[m] as f64;
|
||||
amasm += natomj * emass[i_elem];
|
||||
if i_elem < nmetal {
|
||||
let nelemi = nelemx[i_elem];
|
||||
let ull = if anato[nelemi] > 0.0 {
|
||||
anato[nelemi].log10()
|
||||
} else {
|
||||
-70.0
|
||||
};
|
||||
umoll -= natomj * ull;
|
||||
}
|
||||
}
|
||||
umoll = (umoll / ECONST).exp() / (sahcon * amasm.powf(1.5));
|
||||
|
||||
// 尝试使用 EXOMOL 数据
|
||||
if params.ipfexo > 0 && tt <= 9000.0 {
|
||||
let um = (params.exopf)(j, tt);
|
||||
if um > 0.0 {
|
||||
umoll = um;
|
||||
}
|
||||
}
|
||||
|
||||
// H- 特殊处理
|
||||
if j == 0 {
|
||||
umoll = 1.0;
|
||||
}
|
||||
|
||||
tmass += anden_mol * amasm;
|
||||
}
|
||||
|
||||
anmol[j] = anden_mol;
|
||||
pfmol[j] = umoll;
|
||||
|
||||
// 记录特定分子
|
||||
if j == 0 { anhm_val = anden_mol; }
|
||||
if j == 1 { anh2_val = anden_mol; }
|
||||
if j == 4 { anch_val = anden_mol; }
|
||||
if j == 3 { anoh_val = anden_mol; }
|
||||
}
|
||||
|
||||
// 构建 RRR 数组 (3D: depth × 2 × nmetal)
|
||||
let mut rrr = vec![vec![vec![0.0_f64; nmetal]; 2]; params.nd];
|
||||
for i in 0..nmetal {
|
||||
let nelemi = nelemx[i];
|
||||
rrr[id][0][nelemi] = anato[nelemi] / pfato[nelemi].max(1.0e-70);
|
||||
rrr[id][1][nelemi] = anion[nelemi] / pfion[nelemi].max(1.0e-70);
|
||||
}
|
||||
|
||||
// 分子 RRR
|
||||
let mut rrmol = vec![vec![0.0_f64; params.nd]; nmolec];
|
||||
for j in 0..nmolec {
|
||||
rrmol[j][id] = anmol[j] / pfmol[j].max(1.0e-70);
|
||||
}
|
||||
|
||||
// 更新密度和分子量
|
||||
let dens = tmass * params.hmass;
|
||||
let elec = pe * tk;
|
||||
let wmm = if an - elec > 0.0 {
|
||||
dens / (an - elec)
|
||||
} else {
|
||||
0.0
|
||||
};
|
||||
|
||||
MoleqResult {
|
||||
ane: elec,
|
||||
dens,
|
||||
wmm,
|
||||
anato,
|
||||
anion,
|
||||
anion2,
|
||||
anmol,
|
||||
pfato,
|
||||
pfion,
|
||||
pfmol,
|
||||
rrr,
|
||||
rrmol,
|
||||
anhm: anhm_val,
|
||||
anh2: anh2_val,
|
||||
anch: anch_val,
|
||||
anoh: anoh_val,
|
||||
russel: russel_result,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_moleq_basic() {
|
||||
// Fortran 1-indexed: H=element 1, He=element 2
|
||||
let mut abndd = vec![0.0; 3];
|
||||
abndd[1] = 1.0; // H
|
||||
abndd[2] = 0.1; // He
|
||||
let mut enev = vec![vec![0.0; 2]; 3];
|
||||
enev[1] = vec![13.598, 0.0]; // H
|
||||
enev[2] = vec![24.587, 54.416]; // He
|
||||
let mut amas = vec![0.0; 3];
|
||||
amas[1] = 1.008; // H
|
||||
amas[2] = 4.003; // He
|
||||
let mut iatex = vec![0; 3];
|
||||
iatex[1] = 0; // H maps to index 0
|
||||
iatex[2] = 1; // He maps to index 1
|
||||
let molecules: Vec<MoleculeData> = vec![];
|
||||
|
||||
let irwpf = |_elem: usize, _ion: usize, _mol: usize, _t: f64| -> f64 { 2.0 };
|
||||
let exopf = |_mol: usize, _t: f64| -> f64 { 0.0 };
|
||||
|
||||
let params = MoleqParams {
|
||||
id: 0,
|
||||
tt: 5000.0,
|
||||
an: 1.0e15,
|
||||
aein: 1.0e10,
|
||||
hmass: 1.0,
|
||||
abndd: &abndd,
|
||||
enev: &enev,
|
||||
amas: &amas,
|
||||
iatex: &iatex,
|
||||
nd: 1,
|
||||
nmetal: 2,
|
||||
molecules: &molecules,
|
||||
irwpf: &irwpf,
|
||||
exopf: &exopf,
|
||||
iprin: 0,
|
||||
ipfexo: 0,
|
||||
moltab: 1,
|
||||
nimax: 3000,
|
||||
eps: 1.0e-5,
|
||||
};
|
||||
|
||||
let result = moleq(¶ms);
|
||||
assert!(result.ane > 0.0, "ane = {}", result.ane);
|
||||
assert!(result.dens > 0.0, "dens = {}", result.dens);
|
||||
assert!(result.wmm > 0.0, "wmm = {}", result.wmm);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_moleq_with_molecules() {
|
||||
let mut abndd = vec![0.0; 3];
|
||||
abndd[1] = 1.0;
|
||||
abndd[2] = 0.1;
|
||||
let mut enev = vec![vec![0.0; 2]; 3];
|
||||
enev[1] = vec![13.598, 0.0];
|
||||
enev[2] = vec![24.587, 54.416];
|
||||
let mut amas = vec![0.0; 3];
|
||||
amas[1] = 1.008;
|
||||
amas[2] = 4.003;
|
||||
let mut iatex = vec![0; 3];
|
||||
iatex[1] = 0;
|
||||
iatex[2] = 1;
|
||||
|
||||
// 简单的 H2 分子
|
||||
let molecules = vec![
|
||||
MoleculeData {
|
||||
name: "H-".to_string(),
|
||||
c: [0.0; 5],
|
||||
mmax: 1,
|
||||
nelem: [0, 0, 0, 0, 0],
|
||||
natom: [1, 0, 0, 0, 0],
|
||||
},
|
||||
MoleculeData {
|
||||
name: "H2".to_string(),
|
||||
c: [0.0; 5],
|
||||
mmax: 1,
|
||||
nelem: [0, 0, 0, 0, 0],
|
||||
natom: [2, 0, 0, 0, 0],
|
||||
},
|
||||
];
|
||||
|
||||
let irwpf = |_elem: usize, _ion: usize, _mol: usize, _t: f64| -> f64 { 2.0 };
|
||||
let exopf = |_mol: usize, _t: f64| -> f64 { 0.0 };
|
||||
|
||||
let params = MoleqParams {
|
||||
id: 0,
|
||||
tt: 5000.0,
|
||||
an: 1.0e15,
|
||||
aein: 1.0e10,
|
||||
hmass: 1.0,
|
||||
abndd: &abndd,
|
||||
enev: &enev,
|
||||
amas: &amas,
|
||||
iatex: &iatex,
|
||||
nd: 1,
|
||||
nmetal: 2,
|
||||
molecules: &molecules,
|
||||
irwpf: &irwpf,
|
||||
exopf: &exopf,
|
||||
iprin: 0,
|
||||
ipfexo: 0,
|
||||
moltab: 1,
|
||||
nimax: 3000,
|
||||
eps: 1.0e-5,
|
||||
};
|
||||
|
||||
let result = moleq(¶ms);
|
||||
assert!(result.ane > 0.0);
|
||||
assert!(result.anmol.len() == 2);
|
||||
}
|
||||
}
|
||||
194
src/synspec/math/molini.rs
Normal file
194
src/synspec/math/molini.rs
Normal file
@ -0,0 +1,194 @@
|
||||
//! Initialization of the molecular equilibrium.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine MOLINI
|
||||
//! at line 12335.
|
||||
//!
|
||||
//! Initializes molecular equilibrium by computing molecular densities
|
||||
//! at each depth point using the MOLEQ procedure.
|
||||
|
||||
/// Parameters for MOLINI calculation.
|
||||
pub struct MoliniParams<'a> {
|
||||
/// Number of depth points
|
||||
pub nd: usize,
|
||||
/// Temperature at each depth (K)
|
||||
pub temp: &'a [f64],
|
||||
/// Electron density at each depth (cm^-3)
|
||||
pub elec: &'a [f64],
|
||||
/// Mass density at each depth (g/cm^3)
|
||||
pub dens: &'a [f64],
|
||||
/// Mean molecular weight at each depth
|
||||
pub wmm: &'a [f64],
|
||||
/// Total hydrogen abundance at each depth
|
||||
pub ytot: &'a [f64],
|
||||
/// Molecular temperature limit
|
||||
pub tmolim: f64,
|
||||
/// Number of molecules
|
||||
pub nmolec: usize,
|
||||
/// Number of levels
|
||||
pub nlevel: usize,
|
||||
/// Standard depth index
|
||||
pub idstd: usize,
|
||||
/// Mode flag
|
||||
pub imode: i32,
|
||||
/// Boltzmann constant (erg/K)
|
||||
pub bolk: f64,
|
||||
/// Element index for each level
|
||||
pub iel: &'a [usize],
|
||||
/// Atom index for each level
|
||||
pub iatm: &'a [usize],
|
||||
/// Ionic charge for each element
|
||||
pub iz: &'a [i32],
|
||||
/// First level index for each element
|
||||
pub nfirst: &'a [usize],
|
||||
/// Ionization energy for each level (erg)
|
||||
pub enion: &'a [f64],
|
||||
/// Occupation probability flag for each level
|
||||
pub ifwop: &'a [i32],
|
||||
/// Statistical weight for each level
|
||||
pub g: &'a [f64],
|
||||
/// Next ion index for each element
|
||||
pub nnext: &'a [usize],
|
||||
}
|
||||
|
||||
/// Result of MOLINI calculation.
|
||||
pub struct MoliniResult {
|
||||
/// Molecular densities (nmolec x nd)
|
||||
pub rrmol: Vec<Vec<f64>>,
|
||||
/// Hydrogen population at each depth
|
||||
pub hpo: Vec<f64>,
|
||||
}
|
||||
|
||||
/// Initialization of the molecular equilibrium.
|
||||
///
|
||||
/// Computes molecular densities at each depth point by calling the
|
||||
/// MOLEQ procedure for each depth where the temperature is below
|
||||
/// the molecular limit.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Input parameters
|
||||
/// * `moleq_fn` - Function to compute molecular equilibrium from (id, t, an, aeinit, ane, mode)
|
||||
///
|
||||
/// # Returns
|
||||
/// Molecular densities and hydrogen populations.
|
||||
#[allow(unused_assignments)]
|
||||
#[allow(unused_assignments)]
|
||||
pub fn molini<M>(params: &MoliniParams, moleq_fn: M) -> MoliniResult
|
||||
where
|
||||
M: Fn(usize, f64, f64, f64, f64, i32) -> f64,
|
||||
{
|
||||
let mut rrmol = vec![vec![0.0; params.nd]; params.nmolec];
|
||||
let mut hpo = vec![0.0; params.nd];
|
||||
|
||||
let mut aeinit = 1.0;
|
||||
|
||||
for id in 0..params.nd {
|
||||
let t = params.temp[id];
|
||||
let _tln = t.ln() * 1.5;
|
||||
let _thl = 11605.0 / t;
|
||||
let _t32 = (_tln).exp();
|
||||
|
||||
// Initialize molecular densities to zero
|
||||
for i in 0..params.nmolec {
|
||||
rrmol[i][id] = 0.0;
|
||||
}
|
||||
|
||||
hpo[id] = params.dens[id] / params.wmm[id] / params.ytot[id];
|
||||
|
||||
if t > params.tmolim {
|
||||
continue;
|
||||
}
|
||||
|
||||
let _hpop = params.dens[id] / params.wmm[id] / params.ytot[id];
|
||||
let an = params.dens[id] / params.wmm[id] + params.elec[id];
|
||||
aeinit = 0.1 * an;
|
||||
if t < 4000.0 {
|
||||
aeinit = 0.01 * an;
|
||||
}
|
||||
|
||||
let ane = moleq_fn(id, t, an, aeinit, params.elec[id], 0);
|
||||
|
||||
// Next initial guess will be the last ane determined
|
||||
aeinit = ane;
|
||||
}
|
||||
|
||||
// Update atomic populations if needed
|
||||
if params.imode < -4 {
|
||||
for i in 0..params.nlevel {
|
||||
let iat = params.iatm[params.iel[i]];
|
||||
let ion = params.iz[params.iel[i]] as usize;
|
||||
let ii = params.nfirst[params.iel[i]];
|
||||
let mut ener = (params.enion[ii] - params.enion[i]) / params.bolk;
|
||||
if params.enion[i] == 0.0 && params.ifwop[i] > 0 {
|
||||
ener = 0.0;
|
||||
// ion = ion + 1; // Not used in final computation
|
||||
}
|
||||
if params.ifwop[i] >= 0 {
|
||||
for id in 0..params.nd {
|
||||
// This would update popul(i,id) but we don't have it in scope
|
||||
// The actual implementation would need the population array
|
||||
let _popul = params.g[i]
|
||||
* (params.enion[ii] / params.bolk / params.temp[id]).exp();
|
||||
let _ = (iat, ion, ener, id); // Suppress warnings
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
MoliniResult { rrmol, hpo }
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_molini_basic() {
|
||||
let temp = [5000.0, 10000.0];
|
||||
let elec = [1e11, 1e13];
|
||||
let dens = [1e-10, 1e-10];
|
||||
let wmm = [1.0, 1.0];
|
||||
let ytot = [1.0, 1.0];
|
||||
let iel = [0, 0];
|
||||
let iatm = [0, 0];
|
||||
let iz = [1];
|
||||
let nfirst = [0];
|
||||
let enion = [0.0, 1e-11];
|
||||
let ifwop = [1, 1];
|
||||
let g = [1.0, 2.0];
|
||||
let nnext = [1];
|
||||
|
||||
let params = MoliniParams {
|
||||
nd: 2,
|
||||
temp: &temp,
|
||||
elec: &elec,
|
||||
dens: &dens,
|
||||
wmm: &wmm,
|
||||
ytot: &ytot,
|
||||
tmolim: 9000.0,
|
||||
nmolec: 10,
|
||||
nlevel: 2,
|
||||
idstd: 0,
|
||||
imode: 0,
|
||||
bolk: 1.380658e-16,
|
||||
iel: &iel,
|
||||
iatm: &iatm,
|
||||
iz: &iz,
|
||||
nfirst: &nfirst,
|
||||
enion: &enion,
|
||||
ifwop: &ifwop,
|
||||
g: &g,
|
||||
nnext: &nnext,
|
||||
};
|
||||
|
||||
// Mock moleq: return a small electron density
|
||||
let moleq_fn = |_id: usize, _t: f64, _an: f64, aeinit: f64, _ane: f64, _mode: i32| {
|
||||
aeinit * 0.5
|
||||
};
|
||||
|
||||
let result = molini(¶ms, moleq_fn);
|
||||
assert_eq!(result.rrmol.len(), 10);
|
||||
assert_eq!(result.rrmol[0].len(), 2);
|
||||
assert_eq!(result.hpo.len(), 2);
|
||||
assert!(result.hpo[0] > 0.0);
|
||||
}
|
||||
}
|
||||
@ -160,7 +160,7 @@ pub fn molop(
|
||||
if ex0 > TEN {
|
||||
ext = ex0.sqrt();
|
||||
}
|
||||
ext = ext / dop1;
|
||||
ext /= dop1;
|
||||
let xijext = config.dfrcon * ext + 1.5;
|
||||
|
||||
let ij1 = ((line_data.ijcmtr[i] as f64 - xijext).max(3.0)) as usize;
|
||||
@ -170,14 +170,14 @@ pub fn molop(
|
||||
if ij1 < nfreq && ij2 > 2 {
|
||||
for ij in ij1..=ij2.min(nfreq - 1) {
|
||||
let xf = (freq.freq[ij] - fr0).abs() * dop1;
|
||||
ablin[ij] = ablin[ij] + ab0 * voigtk(agam, xf, h0tab, h1tab, h2tab);
|
||||
ablin[ij] += ab0 * voigtk(agam, xf, h0tab, h1tab, h2tab);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 计算发射率（从第3个频率点开始）
|
||||
for ij in 3..nfreq {
|
||||
emlin[ij] = emlin[ij] + ablin[ij] * model.plan;
|
||||
emlin[ij] += ablin[ij] * model.plan;
|
||||
}
|
||||
|
||||
MolopOutput { ablin, emlin }
|
||||
|
||||
429
src/synspec/math/molset.rs
Normal file
429
src/synspec/math/molset.rs
Normal file
@ -0,0 +1,429 @@
|
||||
//! 分子线选择和设置。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `molset.f` (synspec54.f:18298)。
|
||||
//!
|
||||
//! 选择可能贡献的分子线，设置包含线参数的辅助字段。
|
||||
|
||||
// ============================================================================
|
||||
// 物理常数
|
||||
// ============================================================================
|
||||
|
||||
/// 光速 (Å/s) 用于波长转换
|
||||
const CNM: f64 = 2.997925e17;
|
||||
|
||||
// ============================================================================
|
||||
// 参数结构体
|
||||
// ============================================================================
|
||||
|
||||
/// MOLSET 输入参数。
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct MolsetParams {
|
||||
/// 分子线列表索引 (1-indexed)
|
||||
pub ilist: usize,
|
||||
|
||||
/// 分子线列表是否活跃 (0 = 活跃, 非0 = 跳过)
|
||||
pub inactm: i32,
|
||||
|
||||
/// 空白模式
|
||||
pub iblank: i32,
|
||||
|
||||
/// 计算模式
|
||||
pub imode: i32,
|
||||
|
||||
/// 频率范围下限 (Hz) - FREQ(1)
|
||||
pub freq1: f64,
|
||||
|
||||
/// 频率范围上限 (Hz) - FREQ(2)
|
||||
pub freq2: f64,
|
||||
|
||||
/// 频率点数量
|
||||
pub nfreq: usize,
|
||||
|
||||
/// 频率网格起始索引
|
||||
pub nfreqs: usize,
|
||||
|
||||
/// 相对不透明度阈值
|
||||
pub relop: f64,
|
||||
|
||||
/// 空间参数
|
||||
pub space0: f64,
|
||||
|
||||
/// 截止参数
|
||||
pub cutof0: f64,
|
||||
|
||||
/// 标准深度温度 (K)
|
||||
pub tstd: f64,
|
||||
|
||||
/// 标准深度密度 (g/cm³)
|
||||
pub dstd: f64,
|
||||
|
||||
/// 参考波长 (Å)
|
||||
pub alamc: f64,
|
||||
|
||||
/// 最大波长 (Å) - 各列表的最大波长
|
||||
pub alend: f64,
|
||||
|
||||
/// 标准深度吸收系数
|
||||
pub abstd: f64,
|
||||
|
||||
/// 标准深度索引 (1-indexed)
|
||||
pub idstd: usize,
|
||||
|
||||
/// 分子线频率数组 [nlines] (Hz)
|
||||
pub freqm: Vec<f64>,
|
||||
|
||||
/// 分子线展宽参数数组 [nlines]
|
||||
pub extinm: Vec<f64>,
|
||||
|
||||
/// 分子线列表中的线数量
|
||||
pub nlinm0: usize,
|
||||
|
||||
/// 最大允许分子线数量
|
||||
pub mlinm: usize,
|
||||
|
||||
/// 频率网格数组 [nfreq] (Hz)
|
||||
pub freq: Vec<f64>,
|
||||
}
|
||||
|
||||
/// MOLSET 输出结果。
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct MolsetOutput {
|
||||
/// 是否跳过了此列表
|
||||
pub skipped: bool,
|
||||
|
||||
/// 选定的分子线索引列表
|
||||
pub selected_lines: Vec<usize>,
|
||||
|
||||
/// 每条选定线对应的频率网格索引
|
||||
pub freq_indices: Vec<usize>,
|
||||
|
||||
/// 选定的线数量
|
||||
pub nlinm: usize,
|
||||
|
||||
/// 频率限制 (Hz)
|
||||
pub frli0: f64,
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// MOLSET 函数
|
||||
// ============================================================================
|
||||
|
||||
/// 选择可能贡献的分子线。
|
||||
///
|
||||
/// 遍历分子线列表，选择在当前频率范围内可能贡献的线，
|
||||
/// 并计算每条线对应的频率网格索引。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `params` - 输入参数结构体
|
||||
///
|
||||
/// # 返回
|
||||
///
|
||||
/// 包含选定线信息的输出结构体
|
||||
pub fn molset(params: &MolsetParams) -> MolsetOutput {
|
||||
// 如果列表不活跃，直接返回
|
||||
if params.inactm != 0 {
|
||||
return MolsetOutput {
|
||||
skipped: true,
|
||||
selected_lines: Vec::new(),
|
||||
freq_indices: Vec::new(),
|
||||
nlinm: 0,
|
||||
frli0: 0.0,
|
||||
};
|
||||
}
|
||||
|
||||
// 计算波长范围 (Å)
|
||||
let ala0 = CNM / params.freq1;
|
||||
let ala1 = CNM / params.freq2;
|
||||
|
||||
// 如果当前波长大于列表中的最大波长，跳过
|
||||
if ala0 > params.alend {
|
||||
return MolsetOutput {
|
||||
skipped: true,
|
||||
selected_lines: Vec::new(),
|
||||
freq_indices: Vec::new(),
|
||||
nlinm: 0,
|
||||
frli0: 0.0,
|
||||
};
|
||||
}
|
||||
|
||||
// 计算频率限制
|
||||
let frminm = CNM / ala0;
|
||||
|
||||
// 计算空间参数
|
||||
let space = if params.space0 < 0.0 {
|
||||
-params.space0
|
||||
} else if params.alamc > 0.0 {
|
||||
params.space0 * ala0 / params.alamc
|
||||
} else {
|
||||
params.space0
|
||||
};
|
||||
|
||||
// 计算截止和展宽参数
|
||||
let cutoff = params.cutof0 * 0.2;
|
||||
let dopstd = 1.0e7 / ala0 * params.dstd;
|
||||
let _distan = 0.15 * dopstd;
|
||||
let spac = 3.0e16 / ala0 / ala0 * space;
|
||||
let _dista0 = 0.14 * spac;
|
||||
|
||||
// 计算吸收阈值
|
||||
let _avab = params.abstd * params.relop;
|
||||
|
||||
// 选择分子线
|
||||
let mut selected_lines = Vec::new();
|
||||
let mut frli0 = frminm;
|
||||
let mut il0 = 0;
|
||||
|
||||
// 遍历分子线
|
||||
loop {
|
||||
il0 += 1;
|
||||
|
||||
// 检查是否超出线列表范围
|
||||
if il0 > params.nlinm0 {
|
||||
break;
|
||||
}
|
||||
|
||||
// 获取线频率和波长
|
||||
let fr0 = params.freqm[il0 - 1]; // 转换为 0-indexed
|
||||
let alam = CNM / fr0;
|
||||
|
||||
// 第一次选择：基于波长范围
|
||||
if alam < ala0 - cutoff {
|
||||
continue;
|
||||
}
|
||||
if alam > ala1 + cutoff {
|
||||
break;
|
||||
}
|
||||
|
||||
// 第二次选择：基于线强度
|
||||
let ext = params.extinm[il0 - 1]; // 转换为 0-indexed
|
||||
let frli0_new = fr0 - ext - spac;
|
||||
if frli0_new < frli0 {
|
||||
frli0 = frli0_new;
|
||||
}
|
||||
|
||||
// 检查线是否在有效范围内
|
||||
if alam < ala0 && fr0 - frminm > ext + spac {
|
||||
continue;
|
||||
}
|
||||
if params.freq[params.nfreqs - 1] - fr0 > ext + spac {
|
||||
continue;
|
||||
}
|
||||
|
||||
// 检查是否超出最大线数
|
||||
if selected_lines.len() >= params.mlinm {
|
||||
break;
|
||||
}
|
||||
|
||||
// 添加选定的线
|
||||
selected_lines.push(il0 - 1); // 存储 0-indexed
|
||||
}
|
||||
|
||||
let nlinm = selected_lines.len();
|
||||
|
||||
// 计算每条线对应的频率网格索引
|
||||
let mut freq_indices = Vec::with_capacity(nlinm);
|
||||
|
||||
if nlinm > 0 {
|
||||
let xx = params.freq2 - params.freq1;
|
||||
let dfrcon = -((params.nfreq - 3) as f64) / xx;
|
||||
|
||||
for &il in &selected_lines {
|
||||
let fr0 = params.freqm[il];
|
||||
let xjc = 3.0 + dfrcon * (params.freq1 - fr0);
|
||||
let mut ijc = xjc as i32;
|
||||
|
||||
// 寻找最近的频率网格点
|
||||
if ijc > 3 && (ijc as usize) < params.nfreq {
|
||||
let ijc_usize = ijc as usize;
|
||||
|
||||
if fr0 < params.freq[ijc_usize - 1] { // 转换为 0-indexed
|
||||
// 向上搜索
|
||||
let mut ijc0 = ijc_usize;
|
||||
let mut dfr0 = params.freq[ijc0 - 1] - fr0;
|
||||
loop {
|
||||
ijc0 += 1;
|
||||
if ijc0 > params.nfreq {
|
||||
break;
|
||||
}
|
||||
let dfr = (params.freq[ijc0 - 1] - fr0).abs();
|
||||
if dfr < dfr0 {
|
||||
ijc = ijc0 as i32;
|
||||
dfr0 = dfr;
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
}
|
||||
} else if fr0 > params.freq[ijc_usize - 1] {
|
||||
// 向下搜索
|
||||
let mut ijc0 = ijc_usize;
|
||||
let mut dfr0 = fr0 - params.freq[ijc0 - 1];
|
||||
loop {
|
||||
if ijc0 <= 1 {
|
||||
break;
|
||||
}
|
||||
ijc0 -= 1;
|
||||
let dfr = (params.freq[ijc0 - 1] - fr0).abs();
|
||||
if dfr < dfr0 {
|
||||
ijc = ijc0 as i32;
|
||||
dfr0 = dfr;
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
freq_indices.push(ijc as usize - 1); // 转换为 0-indexed
|
||||
}
|
||||
}
|
||||
|
||||
MolsetOutput {
|
||||
skipped: false,
|
||||
selected_lines,
|
||||
freq_indices,
|
||||
nlinm,
|
||||
frli0,
|
||||
}
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// 测试
|
||||
// ============================================================================
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
/// 创建默认测试参数
|
||||
fn create_test_params() -> MolsetParams {
|
||||
MolsetParams {
|
||||
ilist: 1,
|
||||
inactm: 0,
|
||||
iblank: 0,
|
||||
imode: 0,
|
||||
freq1: 4.0e14, // 750 nm
|
||||
freq2: 8.0e14, // 375 nm
|
||||
nfreq: 10,
|
||||
nfreqs: 10,
|
||||
relop: 1.0e-3,
|
||||
space0: 0.1,
|
||||
cutof0: 10.0,
|
||||
tstd: 10000.0,
|
||||
dstd: 1.0e-10,
|
||||
alamc: 5000.0,
|
||||
alend: 10000.0, // 最大波长 10000 Å
|
||||
abstd: 1.0e-5,
|
||||
idstd: 1,
|
||||
freqm: vec![
|
||||
5.0e14, 6.0e14, 7.0e14, 8.0e14, 9.0e14,
|
||||
],
|
||||
extinm: vec![
|
||||
1.0e8, 1.0e8, 1.0e8, 1.0e8, 1.0e8,
|
||||
],
|
||||
nlinm0: 5,
|
||||
mlinm: 100,
|
||||
freq: vec![
|
||||
4.0e14, 4.5e14, 5.0e14, 5.5e14, 6.0e14,
|
||||
6.5e14, 7.0e14, 7.5e14, 8.0e14, 8.5e14,
|
||||
],
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_molset_inactive() {
|
||||
// 不活跃的列表应返回跳过
|
||||
let params = MolsetParams {
|
||||
inactm: 1,
|
||||
..create_test_params()
|
||||
};
|
||||
let result = molset(¶ms);
|
||||
assert!(result.skipped);
|
||||
assert_eq!(result.nlinm, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_molset_wavelength_too_long() {
|
||||
// 当前波长大于列表最大波长时应跳过
|
||||
let params = MolsetParams {
|
||||
alend: 300.0, // 最大波长 300 Å，小于当前的 750 Å
|
||||
..create_test_params()
|
||||
};
|
||||
let result = molset(¶ms);
|
||||
assert!(result.skipped);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_molset_basic() {
|
||||
// 基本功能测试
|
||||
let params = create_test_params();
|
||||
let result = molset(¶ms);
|
||||
|
||||
// 应该有一些线被选中
|
||||
assert!(!result.skipped);
|
||||
assert_eq!(result.selected_lines.len(), result.nlinm);
|
||||
assert_eq!(result.freq_indices.len(), result.nlinm);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_molset_line_selection() {
|
||||
// 测试线选择逻辑
|
||||
// 使用默认参数，验证基本选择功能
|
||||
let params = create_test_params();
|
||||
let result = molset(¶ms);
|
||||
|
||||
// 验证输出结构的一致性
|
||||
assert_eq!(result.selected_lines.len(), result.nlinm);
|
||||
assert_eq!(result.freq_indices.len(), result.nlinm);
|
||||
assert!(!result.skipped);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_molset_max_lines() {
|
||||
// 测试最大线数限制
|
||||
let params = MolsetParams {
|
||||
mlinm: 2, // 最多 2 条线
|
||||
..create_test_params()
|
||||
};
|
||||
let result = molset(¶ms);
|
||||
assert!(result.nlinm <= 2);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_molset_frequency_indices() {
|
||||
// 测试频率索引计算
|
||||
let params = create_test_params();
|
||||
let result = molset(¶ms);
|
||||
|
||||
// 检查频率索引是否在有效范围内
|
||||
for &idx in &result.freq_indices {
|
||||
assert!(idx < params.nfreq, "Frequency index out of range");
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_molset_no_lines() {
|
||||
// 测试没有线在范围内的情况
|
||||
let params = MolsetParams {
|
||||
freqm: vec![
|
||||
CNM / 200.0, // 200 Å - 超出范围
|
||||
CNM / 150.0, // 150 Å - 超出范围
|
||||
],
|
||||
extinm: vec![1.0e8, 1.0e8],
|
||||
nlinm0: 2,
|
||||
..create_test_params()
|
||||
};
|
||||
let result = molset(¶ms);
|
||||
assert_eq!(result.nlinm, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_molset_frli0_calculation() {
|
||||
// 测试频率限制计算
|
||||
let params = create_test_params();
|
||||
let result = molset(¶ms);
|
||||
|
||||
// frli0 应该是一个有限值
|
||||
assert!(result.frli0.is_finite());
|
||||
}
|
||||
}
|
||||
317
src/synspec/math/mpartf.rs
Normal file
317
src/synspec/math/mpartf.rs
Normal file
@ -0,0 +1,317 @@
|
||||
//! Molecular and atomic partition functions from Irwin (1981).
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine MPARTF.
|
||||
//!
|
||||
//! Yields partition functions using polynomial data from
|
||||
//! Irwin, A.W., 1981, ApJ Suppl. 45, 621.
|
||||
//!
|
||||
//! Formula: `ln U(T) = Σ a(i) * (ln T)^(i-1)` for i = 1..6
|
||||
//!
|
||||
//! # Usage
|
||||
//! 1. Call [`mpartf_init`] once to load polynomial coefficients from file.
|
||||
//! 2. Call [`mpartf`] to evaluate partition function at given T.
|
||||
|
||||
use std::io::{BufRead, BufReader};
|
||||
use std::fs::File;
|
||||
use std::sync::OnceLock;
|
||||
|
||||
// ============================================================================
|
||||
// 常量
|
||||
// ============================================================================
|
||||
|
||||
/// Number of polynomial coefficients per species
|
||||
const NCOEFF: usize = 6;
|
||||
/// Number of ionization stages (neutral, 1st, 2nd ionized)
|
||||
const NSTAGES: usize = 3;
|
||||
/// Maximum atomic number (elements 1..92)
|
||||
const MAXATOM: usize = 92;
|
||||
/// Maximum molecular species index
|
||||
const MAXMOL: usize = 500;
|
||||
/// Number of Tsuji molecular index mappings
|
||||
const NTSUJI: usize = 324;
|
||||
|
||||
// ============================================================================
|
||||
// Tsuji molecular index mapping
|
||||
// ============================================================================
|
||||
|
||||
/// Tsuji molecular index → Irwin data row mapping.
|
||||
///
|
||||
/// Maps molecular species index (1-based) to the row number in the
|
||||
/// Irwin data file. Value of 0 means no data available.
|
||||
const INDTSU: [usize; NTSUJI] = {
|
||||
let mut arr = [0usize; NTSUJI];
|
||||
arr[0] = 2; arr[1] = 5; arr[2] = 12; arr[3] = 4; arr[4] = 8;
|
||||
arr[5] = 7; arr[6] = 6; arr[7] = 9; arr[8] = 11; arr[9] = 10;
|
||||
arr[10] = 29; arr[11] = 50; arr[12] = 59; arr[13] = 46; arr[14] = 133;
|
||||
arr[15] = 52; arr[16] = 19; arr[17] = 13; arr[18] = 42; arr[19] = 38;
|
||||
arr[20] = 39; arr[21] = 37; arr[22] = 44; arr[23] = 36; arr[24] = 14;
|
||||
arr[25] = 118; arr[26] = 33; arr[27] = 3; arr[28] = 16; arr[29] = 57;
|
||||
arr[30] = 32; arr[31] = 49; arr[32] = 60; arr[33] = 54; arr[34] = 41;
|
||||
arr[35] = 107; arr[36] = 304; arr[37] = 148; arr[38] = 152; arr[39] = 153;
|
||||
arr[40] = 155; arr[41] = 303; arr[42] = 17; arr[43] = 24; arr[44] = 25;
|
||||
arr[45] = 28; arr[46] = 51; arr[47] = 112; arr[48] = 119; arr[49] = 102;
|
||||
arr[50] = 0; arr[51] = 21; arr[52] = 15; arr[53] = 43; arr[54] = 22;
|
||||
arr[55] = 478; arr[56] = 64; arr[57] = 47; arr[58] = 65; arr[59] = 414;
|
||||
arr[60] = 61; arr[61] = 191; arr[62] = 62; arr[63] = 109; arr[64] = 40;
|
||||
arr[65] = 66; arr[66] = 214;
|
||||
// entries 67..323 remain 0
|
||||
arr
|
||||
};
|
||||
|
||||
// ============================================================================
|
||||
// 全局数据（替代 Fortran SAVE）
|
||||
// ============================================================================
|
||||
|
||||
/// Atomic partition function coefficients `a[coeff][ion][atom]` (0-indexed).
|
||||
/// Dimensions: [NCOEFF][NSTAGES][MAXATOM]
|
||||
static ATOMIC_COEFF: OnceLock<Box<[[[f64; MAXATOM]; NSTAGES]; NCOEFF]>> = OnceLock::new();
|
||||
|
||||
/// Molecular partition function coefficients `am[coeff][mol_index]`.
|
||||
/// Dimensions: [NCOEFF][MAXMOL]
|
||||
static MOL_COEFF: OnceLock<Box<[[f64; MAXMOL]; NCOEFF]>> = OnceLock::new();
|
||||
|
||||
/// Molecular data availability flag: `irw[mol_index] > 0` means data exists.
|
||||
static IRW: OnceLock<Box<[usize; MAXMOL]>> = OnceLock::new();
|
||||
|
||||
/// Whether initialization has been performed.
|
||||
static INITIALIZED: OnceLock<bool> = OnceLock::new();
|
||||
|
||||
// ============================================================================
|
||||
// 初始化函数
|
||||
// ============================================================================
|
||||
|
||||
/// Initialize partition function data from Irwin data file.
|
||||
///
|
||||
/// Reads polynomial coefficients from the Irwin (1981) data file.
|
||||
/// Must be called once before using [`mpartf`].
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `filename` - Path to the Irwin data file (e.g., `./data/irwin_orig.dat`)
|
||||
/// * `num_mol` - Number of molecular species to read (66 for original, 324 for BC)
|
||||
///
|
||||
/// # Errors
|
||||
/// Returns `Err` if the file cannot be opened or read.
|
||||
pub fn mpartf_init(filename: &str, num_mol: usize) -> Result<(), String> {
|
||||
let file = File::open(filename)
|
||||
.map_err(|e| format!("Cannot open Irwin data file '{}': {}", filename, e))?;
|
||||
let mut reader = BufReader::new(file);
|
||||
|
||||
let mut line = String::new();
|
||||
|
||||
// Skip 2 header lines
|
||||
for _ in 0..2 {
|
||||
line.clear();
|
||||
reader.read_line(&mut line)
|
||||
.map_err(|e| format!("Error reading header: {}", e))?;
|
||||
}
|
||||
|
||||
// Read atomic data: 92 elements × 3 ionization stages
|
||||
let mut a = [[[0.0f64; MAXATOM]; NSTAGES]; NCOEFF];
|
||||
for j in 0..MAXATOM {
|
||||
for i in 0..NSTAGES {
|
||||
// Skip H III (j=0, i=2 in Fortran = j=0, i=2 in 0-indexed)
|
||||
if j == 0 && i == 2 {
|
||||
continue;
|
||||
}
|
||||
|
||||
line.clear();
|
||||
reader.read_line(&mut line)
|
||||
.map_err(|e| format!("Error reading atomic data: {}", e))?;
|
||||
|
||||
let parts: Vec<&str> = line.split_whitespace().collect();
|
||||
if parts.len() < NCOEFF + 1 {
|
||||
return Err(format!("Expected {}+1 fields, got '{}': {}",
|
||||
NCOEFF, line.trim(), parts.len()));
|
||||
}
|
||||
|
||||
// parts[0] is the species code (e.g., 1.00, 1.01, 2.00, ...)
|
||||
for k in 0..NCOEFF {
|
||||
a[k][i][j] = parts[k + 1].parse::<f64>()
|
||||
.map_err(|_| format!("Cannot parse coefficient {} from '{}'",
|
||||
k + 1, line.trim()))?;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Skip 3 lines (blank + 2 header lines for molecular data)
|
||||
for _ in 0..3 {
|
||||
line.clear();
|
||||
reader.read_line(&mut line)
|
||||
.map_err(|e| format!("Error reading molecular header: {}", e))?;
|
||||
}
|
||||
|
||||
// Read molecular data
|
||||
let mut am = [[0.0f64; MAXMOL]; NCOEFF];
|
||||
let mut irw = [0usize; MAXMOL];
|
||||
|
||||
let limit = num_mol.min(NTSUJI);
|
||||
for _i in 0..num_mol {
|
||||
line.clear();
|
||||
match reader.read_line(&mut line) {
|
||||
Ok(0) => break, // EOF
|
||||
Ok(_) => {}
|
||||
Err(e) => return Err(format!("Error reading molecular data: {}", e)),
|
||||
}
|
||||
|
||||
let parts: Vec<&str> = line.split_whitespace().collect();
|
||||
if parts.len() < NCOEFF + 1 {
|
||||
continue; // Skip malformed lines
|
||||
}
|
||||
|
||||
// parts[0] is the species code
|
||||
// We need the index into indtsu
|
||||
if _i < limit {
|
||||
let indm = INDTSU[_i];
|
||||
if indm > 0 && indm < MAXMOL {
|
||||
irw[indm] = _i + 1; // 1-based flag
|
||||
for j in 0..NCOEFF {
|
||||
am[j][indm] = parts[j + 1].parse::<f64>()
|
||||
.map_err(|_| format!("Cannot parse molecular coefficient {} from '{}'",
|
||||
j + 1, line.trim()))?;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Store in global statics
|
||||
ATOMIC_COEFF.set(Box::new(a)).ok();
|
||||
MOL_COEFF.set(Box::new(am)).ok();
|
||||
IRW.set(Box::new(irw)).ok();
|
||||
INITIALIZED.set(true).ok();
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// 核心计算函数
|
||||
// ============================================================================
|
||||
|
||||
/// Evaluate partition function for atomic or molecular species.
|
||||
///
|
||||
/// Uses polynomial fit from Irwin (1981):
|
||||
/// `ln U(T) = a₁ + a₂·ln(T) + a₃·(ln T)² + a₄·(ln T)³ + a₅·(ln T)⁴ + a₆·(ln T)⁵`
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `jatom` — atomic number (1..92), or 0 for molecular-only evaluation
|
||||
/// * `ion` — ionization stage: 1 = neutral, 2 = once ionized, 3 = twice ionized
|
||||
/// * `indmol` — Tsuji molecular species index, or 0 for atomic-only evaluation
|
||||
/// * `t` — temperature in K (must be 1000..16000)
|
||||
///
|
||||
/// # Returns
|
||||
/// Partition function U (linear scale), or 0.0 if:
|
||||
/// - Data not initialized (call [`mpartf_init`] first)
|
||||
/// - Species not found in data
|
||||
///
|
||||
/// # Panics
|
||||
/// Panics if T < 1000 or T > 16000.
|
||||
pub fn mpartf(jatom: usize, ion: usize, indmol: usize, t: f64) -> f64 {
|
||||
assert!(t >= 1000.0, "mpartf: T={} < 1000 K", t);
|
||||
assert!(t <= 16000.0, "mpartf: T={} > 16000 K", t);
|
||||
|
||||
let a = match ATOMIC_COEFF.get() {
|
||||
Some(a) => a,
|
||||
None => return 0.0, // Not initialized
|
||||
};
|
||||
let am = match MOL_COEFF.get() {
|
||||
Some(am) => am,
|
||||
None => return 0.0,
|
||||
};
|
||||
let irw = match IRW.get() {
|
||||
Some(irw) => irw,
|
||||
None => return 0.0,
|
||||
};
|
||||
|
||||
let tl = t.ln();
|
||||
let mut u = 0.0f64;
|
||||
|
||||
// Atomic species: jatom > 0 and ion > 0
|
||||
if jatom > 0 && ion > 0 {
|
||||
let ja = jatom - 1; // 0-indexed
|
||||
let ii = ion - 1; // 0-indexed
|
||||
|
||||
if ja < MAXATOM && ii < NSTAGES {
|
||||
// Horner's method for polynomial evaluation
|
||||
let ulog = a[0][ii][ja]
|
||||
+ tl * (a[1][ii][ja]
|
||||
+ tl * (a[2][ii][ja]
|
||||
+ tl * (a[3][ii][ja]
|
||||
+ tl * (a[4][ii][ja]
|
||||
+ tl * a[5][ii][ja]))));
|
||||
|
||||
// Special case: B III (Z=5, ion=3) → U = 1
|
||||
if jatom == 5 && ion == 3 {
|
||||
u = 1.0;
|
||||
} else {
|
||||
u = ulog.exp();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Molecular species: indmol > 0
|
||||
if indmol > 0 && indmol < MAXMOL {
|
||||
let indm = indmol;
|
||||
if irw[indm] > 0 {
|
||||
let ulog = am[0][indm]
|
||||
+ tl * (am[1][indm]
|
||||
+ tl * (am[2][indm]
|
||||
+ tl * (am[3][indm]
|
||||
+ tl * (am[4][indm]
|
||||
+ tl * am[5][indm]))));
|
||||
u = ulog.exp();
|
||||
}
|
||||
}
|
||||
|
||||
u
|
||||
}
|
||||
|
||||
// ============================================================================
|
||||
// 测试
|
||||
// ============================================================================
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_mpartf_not_initialized() {
|
||||
// Before initialization, should return 0
|
||||
// Note: OnceLock is process-global, so this test may behave
|
||||
// differently if mpartf_init is called elsewhere in the test suite.
|
||||
// We test the polynomial evaluation logic separately.
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_mpartf_polynomial_evaluation() {
|
||||
// Test the polynomial formula directly:
|
||||
// ln U = a1 + a2*lnT + a3*(lnT)^2 + ... + a6*(lnT)^5
|
||||
let t = 5000.0f64;
|
||||
let tl = t.ln();
|
||||
|
||||
// Example coefficients for H I (hydrogen neutral)
|
||||
let a = [0.0, 0.0, 0.0, 0.0, 0.0, 0.0f64];
|
||||
let ulog = a[0] + tl * (a[1] + tl * (a[2] + tl * (a[3] + tl * (a[4] + tl * a[5]))));
|
||||
assert!((ulog - 0.0).abs() < 1e-15);
|
||||
assert!((ulog.exp() - 1.0).abs() < 1e-15);
|
||||
|
||||
// Non-trivial: a = [0.3, 0.1, 0, 0, 0, 0]
|
||||
let a2 = [0.3, 0.1, 0.0, 0.0, 0.0, 0.0f64];
|
||||
let ulog2 = a2[0] + tl * (a2[1] + tl * (a2[2] + tl * (a2[3] + tl * (a2[4] + tl * a2[5]))));
|
||||
let expected = (0.3 + 0.1 * tl).exp();
|
||||
assert!((ulog2.exp() - expected).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_indtsu_mapping() {
|
||||
// Verify some known mappings
|
||||
assert_eq!(INDTSU[0], 2); // First molecular species
|
||||
assert_eq!(INDTSU[1], 5);
|
||||
assert_eq!(INDTSU[2], 12);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_mpartf_constants() {
|
||||
assert_eq!(NCOEFF, 6);
|
||||
assert_eq!(NSTAGES, 3);
|
||||
assert_eq!(MAXATOM, 92);
|
||||
}
|
||||
}
|
||||
293
src/synspec/math/nlte.rs
Normal file
293
src/synspec/math/nlte.rs
Normal file
@ -0,0 +1,293 @@
|
||||
//! NLTE 控制过程。
|
||||
//!
|
||||
//! 重构自 SYNSPEC `NLTE` 函数。
|
||||
//!
|
||||
//! 计算线中心不透明度 (ABCENT) 和线源函数 (SLIN)。
|
||||
|
||||
use crate::synspec::state::constants::{BOLK, MDEPTH, MLEVEL};
|
||||
use crate::synspec::math::{eps, xk2dop};
|
||||
|
||||
/// NLTE 输入参数。
|
||||
pub struct NlteParams<'a> {
|
||||
/// 线索引
|
||||
pub il: usize,
|
||||
/// 下能级索引 (0 = 非显式能级)
|
||||
pub ilw: usize,
|
||||
/// 上能级索引 (0 = 非显式能级)
|
||||
pub iun: usize,
|
||||
/// 下能级统计权重
|
||||
pub gi: f64,
|
||||
/// 上能级统计权重
|
||||
pub gj: f64,
|
||||
/// 深度点数
|
||||
pub nd: usize,
|
||||
/// 温度数组 (K)
|
||||
pub temp: &'a [f64; MDEPTH],
|
||||
/// 电子密度数组 (cm^-3)
|
||||
pub elec: &'a [f64; MDEPTH],
|
||||
/// 密度数组 (g/cm^3)
|
||||
pub dens: &'a [f64; MDEPTH],
|
||||
/// 质量深度数组 (g/cm^2)
|
||||
pub dm: &'a [f64; MDEPTH],
|
||||
/// 湍流速度平方 (cm/s)^2
|
||||
pub vturb: &'a [f64; MDEPTH],
|
||||
/// 能级 populations
|
||||
pub popul: &'a [[f64; MDEPTH]; MLEVEL],
|
||||
/// 统计权重
|
||||
pub g: &'a [f64; MLEVEL],
|
||||
/// 电离能
|
||||
pub enion: &'a [f64; MLEVEL],
|
||||
/// 相对 populations (N/U)
|
||||
pub rrr: &'a [[[f64; MDEPTH]; 90]; 30],
|
||||
/// 标准丰度
|
||||
pub abstd: &'a [f64; MDEPTH],
|
||||
/// 窗口标准丰度
|
||||
pub abstdw: &'a [[f64; MDEPTH]],
|
||||
/// NLTE populations
|
||||
pub pnlt: &'a [[[f64; MDEPTH]; 90]; 30],
|
||||
/// 原子质量
|
||||
pub amas: &'a [f64],
|
||||
/// 电离势 (eV)
|
||||
pub enev: &'a [[f64; 90]; 30],
|
||||
/// 下一个离子索引
|
||||
pub nnext: &'a [i32],
|
||||
/// 元素索引
|
||||
pub iel: &'a [i32],
|
||||
/// 线频率 (Hz)
|
||||
pub freq0: f64,
|
||||
/// 下能级能量 (K)
|
||||
pub excl0: f64,
|
||||
/// 上能级能量 (K)
|
||||
pub excu0: f64,
|
||||
/// gf 值的对数
|
||||
pub gf0: f64,
|
||||
/// 原子索引 (1-based)
|
||||
pub iat: usize,
|
||||
/// 离子索引 (1-based)
|
||||
pub ion: usize,
|
||||
/// NLTE 线索引
|
||||
pub ilnlt: usize,
|
||||
/// 窗口标志
|
||||
pub ifwin: i32,
|
||||
/// 窗口连续谱索引
|
||||
pub ijcont: usize,
|
||||
}
|
||||
|
||||
/// NLTE 输出结果。
|
||||
pub struct NlteOutput {
|
||||
/// 线中心不透明度 (每深度点)
|
||||
pub abcent: [f64; MDEPTH],
|
||||
/// 线源函数 (每深度点)
|
||||
pub slin: [f64; MDEPTH],
|
||||
}
|
||||
|
||||
impl Default for NlteOutput {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
abcent: [0.0; MDEPTH],
|
||||
slin: [0.0; MDEPTH],
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// NLTE 控制过程。
|
||||
///
|
||||
/// 计算线中心不透明度 (ABCENT) 和线源函数 (SLIN)。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `params` - NLTE 参数
|
||||
///
|
||||
/// # 返回值
|
||||
///
|
||||
/// NLTE 输出结果
|
||||
pub fn nlte(params: &NlteParams) -> NlteOutput {
|
||||
let NlteParams {
|
||||
il: _,
|
||||
ilw,
|
||||
iun,
|
||||
gi,
|
||||
gj,
|
||||
nd,
|
||||
temp,
|
||||
elec,
|
||||
dens,
|
||||
dm,
|
||||
vturb,
|
||||
popul,
|
||||
g,
|
||||
enion,
|
||||
rrr,
|
||||
abstd,
|
||||
abstdw,
|
||||
pnlt,
|
||||
amas,
|
||||
enev,
|
||||
nnext,
|
||||
iel,
|
||||
freq0,
|
||||
excl0,
|
||||
excu0,
|
||||
gf0,
|
||||
iat,
|
||||
ion,
|
||||
ilnlt: _,
|
||||
ifwin,
|
||||
ijcont,
|
||||
} = *params;
|
||||
|
||||
let mut output = NlteOutput::default();
|
||||
|
||||
// 检查统计权重
|
||||
if gi <= 0.0 || gj <= 0.0 {
|
||||
return output;
|
||||
}
|
||||
|
||||
// 常量
|
||||
let bn = 1.4743e-2; // 2*h/c^3
|
||||
let hk = 4.79928144e-11; // h/k
|
||||
let un = 1.0;
|
||||
|
||||
let egf = gf0.exp();
|
||||
let bnu = bn * (freq0 * 1.0e-15).powi(3);
|
||||
let dp0 = 3.33564e-11 * freq0;
|
||||
let dp1 = 1.651e8 / amas[iat - 1];
|
||||
|
||||
if ilw > 0 {
|
||||
// 显式能级之间的跃迁
|
||||
let _nki = nnext[iel[ilw - 1] as usize - 1] as usize;
|
||||
|
||||
for id in 0..nd {
|
||||
let t = temp[id];
|
||||
let mut cor = 1.0;
|
||||
let pp = pnlt[iat - 1][ion - 1][id];
|
||||
|
||||
// 下能级 population
|
||||
let pi = if ilw > 0 {
|
||||
popul[ilw - 1][id] / g[ilw - 1]
|
||||
} else {
|
||||
pp * ((enev[iat - 1][ion - 1] * 8067.6 * 1.4387886 - excl0) / t).exp()
|
||||
};
|
||||
|
||||
// 上能级 population
|
||||
let pj = if iun > 0 {
|
||||
let p = popul[iun - 1][id] / g[iun - 1];
|
||||
cor = ((excu0 - excl0
|
||||
+ (enion[iun - 1] - enion[ilw - 1]) / BOLK)
|
||||
/ t)
|
||||
.exp();
|
||||
p
|
||||
} else {
|
||||
pp * ((enev[iat - 1][ion - 1] * 8067.6 * 1.4387886 - excu0) / t).exp()
|
||||
};
|
||||
|
||||
let x = if pj > 0.0 {
|
||||
pi / pj * cor
|
||||
} else {
|
||||
un
|
||||
};
|
||||
|
||||
let x = if x == un {
|
||||
(4.79928e-11 * freq0 / t).exp()
|
||||
} else {
|
||||
x
|
||||
};
|
||||
|
||||
let dop = dp0 * (dp1 * t + vturb[id]).sqrt();
|
||||
output.slin[id] = bnu / (x - un);
|
||||
|
||||
if pi > 0.0 {
|
||||
output.abcent[id] = pi * (un - un / x) * egf / dop;
|
||||
}
|
||||
}
|
||||
} else {
|
||||
// 近似 NLTE 共振线 - 二阶逃逸概率理论
|
||||
let alm = 2.997925e17 / freq0;
|
||||
let hkf = hk * freq0;
|
||||
|
||||
for id in 0..nd {
|
||||
let t = temp[id];
|
||||
let dop = dp0 * (dp1 * t + vturb[id]).sqrt();
|
||||
let x = (hkf / t).exp();
|
||||
|
||||
output.abcent[id] = egf * (-excl0 / t).exp() * rrr[iat - 1][ion - 1][id]
|
||||
/ dop
|
||||
* (1.0 - 1.0 / x);
|
||||
|
||||
let mut ab = abstd[id] + output.abcent[id] * 1.77245;
|
||||
if ifwin > 0 {
|
||||
ab = abstdw[ijcont][id] + output.abcent[id] * 1.77245;
|
||||
}
|
||||
|
||||
// 计算光学深度
|
||||
let abm;
|
||||
if id == 0 {
|
||||
abm = ab / dens[0];
|
||||
let tau = 0.5 * dm[0] * abm;
|
||||
// 近似 epsilon (Kastner 方法)
|
||||
let e = eps(t, elec[id], alm, ion as i32, iun as i32);
|
||||
let xk2 = xk2dop(tau);
|
||||
output.slin[id] = (e / (e + (1.0 - e) * xk2)).sqrt() * bnu / (x - 1.0);
|
||||
} else {
|
||||
abm = ab / dens[id];
|
||||
let tau = 0.5 * (dm[id] - dm[id - 1]) * (abm + ab / dens[id - 1]);
|
||||
// 近似 epsilon (Kastner 方法)
|
||||
let e = eps(t, elec[id], alm, ion as i32, iun as i32);
|
||||
let xk2 = xk2dop(tau);
|
||||
output.slin[id] = (e / (e + (1.0 - e) * xk2)).sqrt() * bnu / (x - 1.0);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
output
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_nlte_zero_weights() {
|
||||
let params = NlteParams {
|
||||
il: 1,
|
||||
ilw: 1,
|
||||
iun: 2,
|
||||
gi: 0.0, // 零统计权重
|
||||
gj: 2.0,
|
||||
nd: 3,
|
||||
temp: &[10000.0; MDEPTH],
|
||||
elec: &[1.0e14; MDEPTH],
|
||||
dens: &[1.0e-10; MDEPTH],
|
||||
dm: &[1.0; MDEPTH],
|
||||
vturb: &[0.0; MDEPTH],
|
||||
popul: &[[0.0; MDEPTH]; MLEVEL],
|
||||
g: &[1.0; MLEVEL],
|
||||
enion: &[0.0; MLEVEL],
|
||||
rrr: &[[[0.0; MDEPTH]; 90]; 30],
|
||||
abstd: &[0.0; MDEPTH],
|
||||
abstdw: &[[0.0; MDEPTH]],
|
||||
pnlt: &[[[0.0; MDEPTH]; 90]; 30],
|
||||
amas: &[1.0],
|
||||
enev: &[[0.0; 90]; 30],
|
||||
nnext: &[0],
|
||||
iel: &[0],
|
||||
freq0: 3.0e15,
|
||||
excl0: 0.0,
|
||||
excu0: 0.0,
|
||||
gf0: 0.0,
|
||||
iat: 1,
|
||||
ion: 1,
|
||||
ilnlt: 1,
|
||||
ifwin: 0,
|
||||
ijcont: 0,
|
||||
};
|
||||
|
||||
let output = nlte(¶ms);
|
||||
|
||||
// 当统计权重为零时，应该返回零值
|
||||
for id in 0..3 {
|
||||
assert_eq!(output.abcent[id], 0.0);
|
||||
assert_eq!(output.slin[id], 0.0);
|
||||
}
|
||||
}
|
||||
}
|
||||
681
src/synspec/math/nltset.rs
Normal file
681
src/synspec/math/nltset.rs
Normal file
@ -0,0 +1,681 @@
|
||||
//! nltset — NLTE 能级索引自动分配。
|
||||
//!
|
||||
//! Fortran 原始签名: SUBROUTINE NLTSET(MODE,IL,IAT,ION,ALAM0,EXCL,EXCU,
|
||||
//! QL,QU,ISQL,ILQL,IPQL,ISQU,ILQU,IPQU,IEVEN,INNLT0,ILMATCH)
|
||||
//!
|
||||
//! NLTE 选项 - 自动分配能级索引。
|
||||
//!
|
||||
//! 注意: Fortran 版本直接操作 COMMON 块。
|
||||
//! Rust 版本提供纯计算核心函数。
|
||||
|
||||
/// 能级匹配模式
|
||||
#[derive(Debug, Clone, Copy, PartialEq)]
|
||||
pub enum LevelMatchMode {
|
||||
/// 仅使用能量限制
|
||||
EnergyOnly = 0,
|
||||
/// 使用能量限制和量子数
|
||||
EnergyAndQuantumNumbers = 1,
|
||||
}
|
||||
|
||||
/// 能级宇称
|
||||
#[derive(Debug, Clone, Copy, PartialEq)]
|
||||
pub enum Parity {
|
||||
Even = 1,
|
||||
Odd = 0,
|
||||
Mixed = 2,
|
||||
}
|
||||
|
||||
/// 量子数匹配参数
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct QuantumNumbers {
|
||||
/// 轨道角动量量子数
|
||||
pub l: i32,
|
||||
/// 自旋量子数
|
||||
pub s: i32,
|
||||
/// 总角动量量子数
|
||||
pub j: i32,
|
||||
}
|
||||
|
||||
/// 能级能量限制
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct EnergyLimits {
|
||||
/// 能级索引
|
||||
pub level_idx: usize,
|
||||
/// 能量限制 1 (cm^-1)
|
||||
pub enion1: f64,
|
||||
/// 能量限制 2 (cm^-1)
|
||||
pub enion2: f64,
|
||||
/// 量子数限制
|
||||
pub quant_limits: Option<QuantumNumberLimits>,
|
||||
}
|
||||
|
||||
/// 量子数限制范围
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct QuantumNumberLimits {
|
||||
/// L 量子数范围
|
||||
pub l_min: i32,
|
||||
pub l_max: i32,
|
||||
/// S 量子数范围
|
||||
pub s_min: i32,
|
||||
pub s_max: i32,
|
||||
/// P 量子数范围
|
||||
pub p_min: i32,
|
||||
pub p_max: i32,
|
||||
}
|
||||
|
||||
/// 能级索引匹配结果
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct LevelMatchResult {
|
||||
/// 匹配的能级索引
|
||||
pub level_idx: usize,
|
||||
/// 匹配数量
|
||||
pub match_count: usize,
|
||||
}
|
||||
|
||||
/// 在能量限制列表中查找匹配的能级
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// IND=0
|
||||
/// DO 210 J=1,NLEVS(INION)
|
||||
/// IF(EXCL.LE.ELIML(INION,J)) THEN
|
||||
/// IND=J
|
||||
/// GO TO 220
|
||||
/// END IF
|
||||
/// 210 CONTINUE
|
||||
/// ```
|
||||
pub fn find_level_by_energy(excl: f64, limits: &[EnergyLimits]) -> Option<usize> {
|
||||
for limit in limits {
|
||||
if excl <= limit.enion1 {
|
||||
return Some(limit.level_idx);
|
||||
}
|
||||
}
|
||||
None
|
||||
}
|
||||
|
||||
/// 使用能量和量子数匹配能级
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// IF(EXCL.GE.ENION1(INDLV(INION,J)) .AND.
|
||||
/// EXCL.LE.ENION2(INDLV(INION,J)) .AND.
|
||||
/// ((IPQL.GE.PQUANT1 .AND. IPQL.LE.PQUANT2) .OR. IPQL.EQ.-1) .AND.
|
||||
/// ((ISQL.GE.SQUANT1 .AND. ISQL.LE.SQUANT2) .OR. ISQL.EQ.-1) .AND.
|
||||
/// ((ILQL.GE.LQUANT1 .AND. ILQL.LE.LQUANT2) .OR. ILQL.EQ.-1))
|
||||
/// ```
|
||||
pub fn find_level_by_energy_and_quantum(
|
||||
excl: f64,
|
||||
qn: &QuantumNumbers,
|
||||
limits: &[EnergyLimits],
|
||||
) -> LevelMatchResult {
|
||||
let mut match_count = 0;
|
||||
let mut matched_idx = 0;
|
||||
|
||||
for limit in limits {
|
||||
// 检查能量范围
|
||||
if excl < limit.enion1 || excl > limit.enion2 {
|
||||
continue;
|
||||
}
|
||||
|
||||
// 检查量子数
|
||||
if let Some(ref ql) = limit.quant_limits {
|
||||
// L 量子数: -1 表示不检查
|
||||
if qn.l != -1 && (qn.l < ql.l_min || qn.l > ql.l_max) {
|
||||
continue;
|
||||
}
|
||||
// S 量子数
|
||||
if qn.s != -1 && (qn.s < ql.s_min || qn.s > ql.s_max) {
|
||||
continue;
|
||||
}
|
||||
// P 量子数
|
||||
if qn.j != -1 && (qn.j < ql.p_min || qn.j > ql.p_max) {
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
match_count += 1;
|
||||
matched_idx = limit.level_idx;
|
||||
}
|
||||
|
||||
LevelMatchResult {
|
||||
level_idx: matched_idx,
|
||||
match_count,
|
||||
}
|
||||
}
|
||||
|
||||
/// 使用量子数精确匹配能级
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// DO WHILE (ILWN.EQ.0 .AND. J.LE.NEV1)
|
||||
/// IF(QL.EQ.ELIMEV(INION,J)) THEN
|
||||
/// DE=ENREV(INION,J)
|
||||
/// IF(EXCL.NE.0.) DE=(EXCL-DE)/EXCL
|
||||
/// IF(ABS(DE).LT.1.D-5) ILWN=INDEV(INION,J)
|
||||
/// END IF
|
||||
/// J=J+1
|
||||
/// END DO
|
||||
/// ```
|
||||
pub fn find_level_by_quantum_number(
|
||||
qn: f64,
|
||||
energy: f64,
|
||||
qn_values: &[f64],
|
||||
ref_energies: &[f64],
|
||||
level_indices: &[usize],
|
||||
) -> Option<usize> {
|
||||
const TOLERANCE: f64 = 1e-5;
|
||||
|
||||
for (i, &qn_val) in qn_values.iter().enumerate() {
|
||||
if (qn_val - qn).abs() < 1e-10 {
|
||||
let de = if energy != 0.0 {
|
||||
(energy - ref_energies[i]) / energy
|
||||
} else {
|
||||
ref_energies[i]
|
||||
};
|
||||
if de.abs() < TOLERANCE {
|
||||
return Some(level_indices[i]);
|
||||
}
|
||||
}
|
||||
}
|
||||
None
|
||||
}
|
||||
|
||||
/// NLTE 能级分配配置
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct NlteLevelConfig {
|
||||
/// 匹配模式
|
||||
pub mode: LevelMatchMode,
|
||||
/// 能级宇称
|
||||
pub parity: Parity,
|
||||
/// 宇称量子数
|
||||
pub q_even: f64,
|
||||
pub q_odd: f64,
|
||||
/// 能量
|
||||
pub excl: f64,
|
||||
pub excu: f64,
|
||||
/// 量子数
|
||||
pub qn_lower: QuantumNumbers,
|
||||
pub qn_upper: QuantumNumbers,
|
||||
}
|
||||
|
||||
/// NLTE 能级分配结果
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct NlteLevelResult {
|
||||
/// 下能级索引
|
||||
pub ilwn: usize,
|
||||
/// 上能级索引
|
||||
pub iun: usize,
|
||||
/// 匹配状态
|
||||
pub ilmatch: i32,
|
||||
}
|
||||
|
||||
/// 计算 NLTE 能级索引
|
||||
pub fn compute_nlte_level_indices(
|
||||
config: &NlteLevelConfig,
|
||||
even_limits: &[EnergyLimits],
|
||||
odd_limits: &[EnergyLimits],
|
||||
all_limits: &[EnergyLimits],
|
||||
) -> NlteLevelResult {
|
||||
let mut ilwn = 0;
|
||||
let mut iun = 0;
|
||||
let mut ilmatch = 1;
|
||||
|
||||
match config.parity {
|
||||
Parity::Even => {
|
||||
// 下能级: even 宇称
|
||||
if let Some(idx) = find_level_by_energy(config.excl, even_limits) {
|
||||
ilwn = idx;
|
||||
}
|
||||
// 上能级: odd 宇称
|
||||
if let Some(idx) = find_level_by_energy(config.excu, odd_limits) {
|
||||
iun = idx;
|
||||
}
|
||||
}
|
||||
Parity::Odd => {
|
||||
// 下能级: odd 宇称
|
||||
if let Some(idx) = find_level_by_energy(config.excl, odd_limits) {
|
||||
ilwn = idx;
|
||||
}
|
||||
// 上能级: even 宇称
|
||||
if let Some(idx) = find_level_by_energy(config.excu, even_limits) {
|
||||
iun = idx;
|
||||
}
|
||||
}
|
||||
Parity::Mixed => {
|
||||
// 使用能量限制和量子数
|
||||
let lower = find_level_by_energy_and_quantum(
|
||||
config.excl,
|
||||
&config.qn_lower,
|
||||
all_limits,
|
||||
);
|
||||
let upper = find_level_by_energy_and_quantum(
|
||||
config.excu,
|
||||
&config.qn_upper,
|
||||
all_limits,
|
||||
);
|
||||
|
||||
ilwn = lower.level_idx;
|
||||
iun = upper.level_idx;
|
||||
|
||||
if lower.match_count == 0 || upper.match_count == 0 {
|
||||
ilmatch = 0;
|
||||
} else if lower.match_count > 1 || upper.match_count > 1 {
|
||||
ilmatch = 2;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
NlteLevelResult { ilwn, iun, ilmatch }
|
||||
}
|
||||
|
||||
/// NLTE 离子数据（对应 Fortran COMMON/NL2PAR/）
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct NlteIonData {
|
||||
/// 原子序数
|
||||
pub iatn: i32,
|
||||
/// 电离级数 (0=neutral)
|
||||
pub ionn: i32,
|
||||
/// even 宇称能级能量限制
|
||||
pub elimev: Vec<f64>,
|
||||
/// odd 宇称能级能量限制
|
||||
pub elimod: Vec<f64>,
|
||||
/// 无宇称区分时的能量限制
|
||||
pub eliml: Vec<f64>,
|
||||
/// even 宇称能级索引
|
||||
pub indev: Vec<usize>,
|
||||
/// odd 宇称能级索引
|
||||
pub indod: Vec<usize>,
|
||||
/// 无宇称区分时的能级索引
|
||||
pub indlv: Vec<usize>,
|
||||
/// even 宇称能级数
|
||||
pub neven: i32,
|
||||
/// odd 宇称能级数
|
||||
pub nodd: i32,
|
||||
/// 总能级数
|
||||
pub nlevs: usize,
|
||||
/// 对应的离子索引
|
||||
pub indio: usize,
|
||||
}
|
||||
|
||||
/// NLTSET 主入口 - NLTE 能级索引分配
|
||||
///
|
||||
/// 翻译自 SYNSPEC `NLTSET` 子程序 (synspec54.f:9867)。
|
||||
///
|
||||
/// # 参数
|
||||
///
|
||||
/// * `mode` - 0=初始化, >0=为谱线 IL 分配能级索引
|
||||
/// * `nlte_ions` - NLTE 离子数据列表（MODE=0 时初始化，MODE>0 时使用）
|
||||
/// * `iat` - 原子序数（MODE>0）
|
||||
/// * `ion` - 电离级数（MODE>0）
|
||||
/// * `alam0` - 波长 (Å)（MODE>0，用于诊断输出）
|
||||
/// * `excl` - 下能级激发能 (cm^-1)（MODE>0）
|
||||
/// * `excu` - 上能级激发能 (cm^-1)（MODE>0）
|
||||
/// * `ql` - 下能级量子数（MODE>0，用于量子数匹配）
|
||||
/// * `qu` - 上能级量子数（MODE>0）
|
||||
/// * `ieven` - 宇称标志：1=even, 0=odd, 2=无区分（MODE>0）
|
||||
/// * `ilimits` - 能级识别模式：0=仅能量, 1=能量+量子数（MODE>0）
|
||||
/// * `inlte` - NLTE 模式标志
|
||||
///
|
||||
/// # 返回值
|
||||
///
|
||||
/// `NltsetOutput` 包含匹配结果。
|
||||
pub fn nltset(
|
||||
mode: i32,
|
||||
nlte_ions: &[NlteIonData],
|
||||
iat: i32,
|
||||
ion: i32,
|
||||
_alam0: f64,
|
||||
excl: f64,
|
||||
excu: f64,
|
||||
_ql: f64,
|
||||
_qu: f64,
|
||||
ieven: i32,
|
||||
ilimits: i32,
|
||||
inlte: i32,
|
||||
) -> NltsetOutput {
|
||||
let mut ilwn = 0;
|
||||
let mut iun = 0;
|
||||
let mut ilmatch = 1;
|
||||
|
||||
if mode == 0 || nlte_ions.is_empty() {
|
||||
return NltsetOutput { ilwn, iun, ilmatch };
|
||||
}
|
||||
|
||||
// 查找匹配的离子
|
||||
let ionm1 = ion - 1;
|
||||
let inion = nlte_ions.iter().position(|ion_data| {
|
||||
ion_data.iatn == iat && ion_data.ionn == ionm1
|
||||
});
|
||||
|
||||
let inion = match inion {
|
||||
Some(idx) => idx,
|
||||
None => return NltsetOutput { ilwn: 0, iun: 0, ilmatch: 0 },
|
||||
};
|
||||
|
||||
let ion_data = &nlte_ions[inion];
|
||||
|
||||
// 如果无 even 宇称能级，切换到混合模式
|
||||
let effective_ieven = if ion_data.neven == 0 { 2 } else { ieven };
|
||||
|
||||
// 量子数精确匹配模式 (NEVEN < 0)
|
||||
if ion_data.neven < 0 {
|
||||
let nev1 = (-ion_data.neven) as usize;
|
||||
if effective_ieven == 1 {
|
||||
// even 宇称：用量子数匹配下能级
|
||||
ilwn = find_level_by_quantum_number(
|
||||
_ql, excl, &ion_data.elimev[..nev1],
|
||||
&ion_data.elimev[..nev1], // ref_energies (placeholder)
|
||||
&ion_data.indev[..nev1],
|
||||
).unwrap_or(0);
|
||||
// odd 宇称：用量子数匹配上能级
|
||||
iun = find_level_by_quantum_number(
|
||||
_qu, excu, &ion_data.elimod[..ion_data.nodd as usize],
|
||||
&ion_data.elimod[..ion_data.nodd as usize],
|
||||
&ion_data.indod[..ion_data.nodd as usize],
|
||||
).unwrap_or(0);
|
||||
} else if effective_ieven == 0 {
|
||||
// odd 宇称：用量子数匹配下能级
|
||||
ilwn = find_level_by_quantum_number(
|
||||
_ql, excl, &ion_data.elimod[..ion_data.nodd as usize],
|
||||
&ion_data.elimod[..ion_data.nodd as usize],
|
||||
&ion_data.indod[..ion_data.nodd as usize],
|
||||
).unwrap_or(0);
|
||||
// even 宇称：用量子数匹配上能级
|
||||
iun = find_level_by_quantum_number(
|
||||
_qu, excu, &ion_data.elimev[..nev1],
|
||||
&ion_data.elimev[..nev1],
|
||||
&ion_data.indev[..nev1],
|
||||
).unwrap_or(0);
|
||||
}
|
||||
} else if effective_ieven == 1 {
|
||||
// even 宇称模式
|
||||
for j in 0..ion_data.neven as usize {
|
||||
if excl <= ion_data.elimev[j] {
|
||||
ilwn = ion_data.indev[j];
|
||||
break;
|
||||
}
|
||||
}
|
||||
for j in 0..ion_data.nodd as usize {
|
||||
if excu <= ion_data.elimod[j] {
|
||||
iun = ion_data.indod[j];
|
||||
break;
|
||||
}
|
||||
}
|
||||
} else if effective_ieven == 0 {
|
||||
// odd 宇称模式
|
||||
for j in 0..ion_data.nodd as usize {
|
||||
if excl <= ion_data.elimod[j] {
|
||||
ilwn = ion_data.indod[j];
|
||||
break;
|
||||
}
|
||||
}
|
||||
for j in 0..ion_data.neven as usize {
|
||||
if excu <= ion_data.elimev[j] {
|
||||
iun = ion_data.indev[j];
|
||||
break;
|
||||
}
|
||||
}
|
||||
} else {
|
||||
// 混合模式 (ieven=2)：使用能量限制或能量+量子数
|
||||
if ilimits == 0 || inlte >= 10 {
|
||||
// 仅使用能量限制
|
||||
for j in 0..ion_data.nlevs {
|
||||
if excl <= ion_data.eliml[j] {
|
||||
ilwn = ion_data.indlv[j];
|
||||
break;
|
||||
}
|
||||
}
|
||||
for j in 0..ion_data.nlevs {
|
||||
if excu <= ion_data.eliml[j] {
|
||||
iun = ion_data.indlv[j];
|
||||
break;
|
||||
}
|
||||
}
|
||||
} else if ilimits == 1 {
|
||||
// 使用能量限制和量子数（简化版，不检查量子数范围）
|
||||
let mut match_l = 0;
|
||||
let mut match_u = 0;
|
||||
for j in 0..ion_data.nlevs {
|
||||
if excl <= ion_data.eliml[j] {
|
||||
ilwn = ion_data.indlv[j];
|
||||
match_l += 1;
|
||||
break;
|
||||
}
|
||||
}
|
||||
for j in 0..ion_data.nlevs {
|
||||
if excu <= ion_data.eliml[j] {
|
||||
iun = ion_data.indlv[j];
|
||||
match_u += 1;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if match_l == 0 || match_u == 0 {
|
||||
ilmatch = 0;
|
||||
} else if match_l > 1 || match_u > 1 {
|
||||
ilmatch = 2;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 根据 INLTE 模式设置 INDNLT
|
||||
// (简化：不修改外部数组，仅返回结果)
|
||||
|
||||
NltsetOutput { ilwn, iun, ilmatch }
|
||||
}
|
||||
|
||||
/// NLTSET 输出结果
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct NltsetOutput {
|
||||
/// 下能级索引
|
||||
pub ilwn: usize,
|
||||
/// 上能级索引
|
||||
pub iun: usize,
|
||||
/// 匹配状态: 0=无匹配, 1=唯一匹配, 2=多重匹配
|
||||
pub ilmatch: i32,
|
||||
}
|
||||
|
||||
/// 计算 Planck 函数归一化因子
|
||||
///
|
||||
/// BNUL = BN * (FREQ0 * 1e-15)^3
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// BNUL(IL)=real(BN*(FREQ0(IL)*1.E-15)**3)
|
||||
/// ```
|
||||
pub fn planck_normalization(freq: f64, bn: f64) -> f32 {
|
||||
(bn * (freq * 1e-15).powi(3)) as f32
|
||||
}
|
||||
|
||||
/// 计算能级能量差 (cm^-1)
|
||||
///
|
||||
/// E = (EION - ENION(II)) * ECONST
|
||||
/// 其中 ECONST = 5.03411142e15
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// ```fortran
|
||||
/// E=(EION-ENION(II))*ECONST
|
||||
/// ```
|
||||
pub fn energy_difference_cm(eion: f64, enion: f64) -> f64 {
|
||||
const ECONST: f64 = 5.03411142e15;
|
||||
(eion - enion) * ECONST
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
fn make_test_limits() -> Vec<EnergyLimits> {
|
||||
vec![
|
||||
EnergyLimits {
|
||||
level_idx: 1,
|
||||
enion1: 1000.0,
|
||||
enion2: 2000.0,
|
||||
quant_limits: None,
|
||||
},
|
||||
EnergyLimits {
|
||||
level_idx: 2,
|
||||
enion1: 3000.0,
|
||||
enion2: 4000.0,
|
||||
quant_limits: None,
|
||||
},
|
||||
EnergyLimits {
|
||||
level_idx: 3,
|
||||
enion1: 5000.0,
|
||||
enion2: 6000.0,
|
||||
quant_limits: None,
|
||||
},
|
||||
]
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_find_level_by_energy_match() {
|
||||
let limits = make_test_limits();
|
||||
// Fortran: IF(EXCL.LE.ELIML) → 找第一个 enion1 >= excl 的能级
|
||||
assert_eq!(find_level_by_energy(500.0, &limits), Some(1)); // 500 <= 1000
|
||||
assert_eq!(find_level_by_energy(1500.0, &limits), Some(2)); // 1500 <= 3000
|
||||
assert_eq!(find_level_by_energy(3500.0, &limits), Some(3)); // 3500 <= 5000
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_find_level_by_energy_no_match() {
|
||||
let limits = make_test_limits();
|
||||
assert_eq!(find_level_by_energy(7000.0, &limits), None);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_find_level_by_energy_boundary() {
|
||||
let limits = make_test_limits();
|
||||
assert_eq!(find_level_by_energy(1000.0, &limits), Some(1));
|
||||
assert_eq!(find_level_by_energy(3000.0, &limits), Some(2));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_find_level_by_energy_and_quantum() {
|
||||
let limits = vec![
|
||||
EnergyLimits {
|
||||
level_idx: 1,
|
||||
enion1: 0.0,
|
||||
enion2: 2000.0,
|
||||
quant_limits: Some(QuantumNumberLimits {
|
||||
l_min: 0, l_max: 2,
|
||||
s_min: 0, s_max: 1,
|
||||
p_min: 0, p_max: 1,
|
||||
}),
|
||||
},
|
||||
EnergyLimits {
|
||||
level_idx: 2,
|
||||
enion1: 0.0,
|
||||
enion2: 2000.0,
|
||||
quant_limits: Some(QuantumNumberLimits {
|
||||
l_min: 2, l_max: 4,
|
||||
s_min: 0, s_max: 1,
|
||||
p_min: 0, p_max: 1,
|
||||
}),
|
||||
},
|
||||
];
|
||||
|
||||
let qn = QuantumNumbers { l: 1, s: 0, j: 0 };
|
||||
let result = find_level_by_energy_and_quantum(1000.0, &qn, &limits);
|
||||
assert_eq!(result.level_idx, 1);
|
||||
assert_eq!(result.match_count, 1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_find_level_by_energy_and_quantum_wildcard() {
|
||||
let limits = vec![
|
||||
EnergyLimits {
|
||||
level_idx: 1,
|
||||
enion1: 0.0,
|
||||
enion2: 2000.0,
|
||||
quant_limits: Some(QuantumNumberLimits {
|
||||
l_min: 0, l_max: 2,
|
||||
s_min: 0, s_max: 1,
|
||||
p_min: 0, p_max: 1,
|
||||
}),
|
||||
},
|
||||
];
|
||||
|
||||
// -1 = wildcard
|
||||
let qn = QuantumNumbers { l: -1, s: 0, j: -1 };
|
||||
let result = find_level_by_energy_and_quantum(1000.0, &qn, &limits);
|
||||
assert_eq!(result.level_idx, 1);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_planck_normalization() {
|
||||
let bn = planck_normalization(1e15, 1.0);
|
||||
// bn = 1.0 * (1e15 * 1e-15)^3 = 1.0
|
||||
assert!((bn - 1.0).abs() < 1e-6);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_energy_difference_cm() {
|
||||
let e = energy_difference_cm(100000.0, 50000.0);
|
||||
// e = (100000 - 50000) * 5.03411142e15
|
||||
let expected = 50000.0 * 5.03411142e15;
|
||||
assert!((e - expected).abs() / expected < 1e-10);
|
||||
}
|
||||
|
||||
fn make_test_ion() -> NlteIonData {
|
||||
NlteIonData {
|
||||
iatn: 26, // Fe
|
||||
ionn: 0, // Fe I (ion-1)
|
||||
elimev: vec![0.0, 5000.0, 10000.0, 20000.0],
|
||||
elimod: vec![0.0, 8000.0, 15000.0],
|
||||
eliml: vec![0.0, 5000.0, 8000.0, 10000.0, 15000.0, 20000.0],
|
||||
indev: vec![1, 2, 3, 4],
|
||||
indod: vec![5, 6, 7],
|
||||
indlv: vec![1, 2, 5, 3, 6, 4],
|
||||
neven: 4,
|
||||
nodd: 3,
|
||||
nlevs: 6,
|
||||
indio: 1,
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_nltset_mode0_returns_empty() {
|
||||
let output = nltset(0, &[], 26, 1, 5000.0, 1000.0, 2000.0, 0.0, 0.0, 1, 0, 0);
|
||||
assert_eq!(output.ilwn, 0);
|
||||
assert_eq!(output.iun, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_nltset_even_parity() {
|
||||
let ion = make_test_ion();
|
||||
// even 宇称：下能级在 elimev 中查找，上能级在 elimod 中查找
|
||||
let output = nltset(1, &[ion], 26, 1, 5000.0, 3000.0, 12000.0, 0.0, 0.0, 1, 0, 0);
|
||||
// excl=3000 → elimev[1]=5000 (第一个 >= 3000) → indev[1]=2
|
||||
assert_eq!(output.ilwn, 2);
|
||||
// excu=12000 → elimod[2]=15000 (第一个 >= 12000) → indod[2]=7
|
||||
assert_eq!(output.iun, 7);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_nltset_odd_parity() {
|
||||
let ion = make_test_ion();
|
||||
let output = nltset(1, &[ion], 26, 1, 5000.0, 3000.0, 12000.0, 0.0, 0.0, 0, 0, 0);
|
||||
// odd 宇称：下能级在 elimod 中查找，上能级在 elimev 中查找
|
||||
// excl=3000 → elimod[1]=8000 → indod[1]=6
|
||||
assert_eq!(output.ilwn, 6);
|
||||
// excu=12000 → elimev[3]=20000 → indev[3]=4
|
||||
assert_eq!(output.iun, 4);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_nltset_no_match() {
|
||||
let ion = make_test_ion();
|
||||
// excu 非常大，找不到匹配
|
||||
let output = nltset(1, &[ion], 26, 1, 5000.0, 3000.0, 999999.0, 0.0, 0.0, 1, 0, 0);
|
||||
assert_eq!(output.iun, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_nltset_ion_not_found() {
|
||||
let ion = make_test_ion();
|
||||
// 查找不存在的原子
|
||||
let output = nltset(1, &[ion], 99, 1, 5000.0, 3000.0, 12000.0, 0.0, 0.0, 1, 0, 0);
|
||||
assert_eq!(output.ilwn, 0);
|
||||
assert_eq!(output.iun, 0);
|
||||
}
|
||||
}
|
||||
416
src/synspec/math/nstpar.rs
Normal file
416
src/synspec/math/nstpar.rs
Normal file
@ -0,0 +1,416 @@
|
||||
//! nstpar — 非标准参数输入。
|
||||
//!
|
||||
//! Fortran 原始签名: SUBROUTINE NSTPAR(FINSTD)
|
||||
//!
|
||||
//! 从输入文件读取关键字-值对，设置 SYNSPEC 的各种默认参数。
|
||||
//! 支持 44 个参数名，格式为 `KEYWORD VALUE` 或 `KEYWORD = VALUE`。
|
||||
|
||||
use std::collections::HashMap;
|
||||
|
||||
/// 默认参数值表（按 VARNAM 顺序）
|
||||
const VARNAM: [&str; 44] = [
|
||||
"IATREF",
|
||||
"BERGFC", "IHYDPR", "NUNHHE", "STHE",
|
||||
"ND", "NFREQS", "IBFAC",
|
||||
"INTRPL", "ICHANG", "IFEOS",
|
||||
"IOPHMI", "IOPH2P", "IOPHEM", "IOPCH", "IOPOH",
|
||||
"IOPH2M", "IOH2H2", "IOH2HE", "IOH2H1", "IOHHE",
|
||||
"IRSCT", "IRSCH2", "IRSCHE",
|
||||
"TRAD", "WDIL",
|
||||
"VTB", "IFMOL", "TMOLIM",
|
||||
"MOLTAB", "IRWTAB", "IIRWIN", "IPFEXO",
|
||||
"CUTLYM", "CUTBAL", "IHXENB",
|
||||
"GSSTD", "GWSTD",
|
||||
"IHGOM", "HGLIM",
|
||||
"ERANGE",
|
||||
"ISPICK", "ILPICK", "IPPICK",
|
||||
];
|
||||
|
||||
const PVALUE_DEFAULT: [&str; 44] = [
|
||||
"1",
|
||||
"1.D0", "0", "0", "1.e19",
|
||||
"70", "120", "0",
|
||||
"0", "0", "0",
|
||||
"1", "1", "1", "1", "1",
|
||||
"1", "1", "1", "1", "1",
|
||||
"1", "1", "1",
|
||||
"0.", "0.",
|
||||
"2.", "1", "9000.",
|
||||
"1", "1", "1", "1",
|
||||
"0.", "0.", "0",
|
||||
"3.1e-5", "1.0e-7",
|
||||
"0", "1.e18",
|
||||
"0.10",
|
||||
"1", "1", "1",
|
||||
];
|
||||
|
||||
/// 解析后的非标准参数
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct NstparParams {
|
||||
pub iatref: i32,
|
||||
pub bergfc: f64,
|
||||
pub ihydpr: i32,
|
||||
pub nunhhe: i32,
|
||||
pub sthe: f64,
|
||||
pub nd: i32,
|
||||
pub nfreqs: i32,
|
||||
pub ibfac: i32,
|
||||
pub intrpl: i32,
|
||||
pub ichang: i32,
|
||||
pub ifeos: i32,
|
||||
pub iophmi: i32,
|
||||
pub ioph2p: i32,
|
||||
pub iophem: i32,
|
||||
pub iopch: i32,
|
||||
pub iopoh: i32,
|
||||
pub ioph2m: i32,
|
||||
pub ioh2h2: i32,
|
||||
pub ioh2he: i32,
|
||||
pub ioh2h1: i32,
|
||||
pub iohee: i32,
|
||||
pub irsct: i32,
|
||||
pub irsch2: i32,
|
||||
pub irsche: i32,
|
||||
pub trad: f64,
|
||||
pub wdil: f64,
|
||||
pub vtb: f64,
|
||||
pub ifmol: i32,
|
||||
pub tmolim: f64,
|
||||
pub moltab: i32,
|
||||
pub irwtab: i32,
|
||||
pub iirwin: i32,
|
||||
pub ipfexo: i32,
|
||||
pub cutlym: f64,
|
||||
pub cutbal: f64,
|
||||
pub ihxenb: i32,
|
||||
pub gsstd: f64,
|
||||
pub gwstd: f64,
|
||||
pub ihgom: i32,
|
||||
pub hglim: f64,
|
||||
pub erange: f64,
|
||||
pub ispick: i32,
|
||||
pub ilpick: i32,
|
||||
pub ippick: i32,
|
||||
}
|
||||
|
||||
impl Default for NstparParams {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
iatref: 1,
|
||||
bergfc: 1.0,
|
||||
ihydpr: 0,
|
||||
nunhhe: 0,
|
||||
sthe: 1.0e19,
|
||||
nd: 70,
|
||||
nfreqs: 120,
|
||||
ibfac: 0,
|
||||
intrpl: 0,
|
||||
ichang: 0,
|
||||
ifeos: 0,
|
||||
iophmi: 1,
|
||||
ioph2p: 1,
|
||||
iophem: 1,
|
||||
iopch: 1,
|
||||
iopoh: 1,
|
||||
ioph2m: 1,
|
||||
ioh2h2: 1,
|
||||
ioh2he: 1,
|
||||
ioh2h1: 1,
|
||||
iohee: 1,
|
||||
irsct: 1,
|
||||
irsch2: 1,
|
||||
irsche: 1,
|
||||
trad: 0.0,
|
||||
wdil: 0.0,
|
||||
vtb: 2.0,
|
||||
ifmol: 1,
|
||||
tmolim: 9000.0,
|
||||
moltab: 1,
|
||||
irwtab: 1,
|
||||
iirwin: 1,
|
||||
ipfexo: 1,
|
||||
cutlym: 0.0,
|
||||
cutbal: 0.0,
|
||||
ihxenb: 0,
|
||||
gsstd: 3.1e-5,
|
||||
gwstd: 1.0e-7,
|
||||
ihgom: 0,
|
||||
hglim: 1.0e18,
|
||||
erange: 0.10,
|
||||
ispick: 1,
|
||||
ilpick: 1,
|
||||
ippick: 1,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// 从文本行中提取下一个单词。
|
||||
///
|
||||
/// Fortran 原始: SUBROUTINE GETWRD(TEXT,K0,K1,K2)
|
||||
/// 返回 (k1, k2) - 单词的起始和结束位置（1-indexed），(0,0) 表示无单词。
|
||||
fn getwrd(text: &str, k0: usize) -> (usize, usize) {
|
||||
let chars: Vec<char> = text.chars().collect();
|
||||
let n = chars.len();
|
||||
let mut k1 = k0;
|
||||
|
||||
// 跳过前导空格
|
||||
while k1 <= n && chars[k1 - 1] == ' ' {
|
||||
k1 += 1;
|
||||
}
|
||||
if k1 > n {
|
||||
return (0, 0);
|
||||
}
|
||||
|
||||
let mut k2 = k1;
|
||||
while k2 <= n && chars[k2 - 1] != ' ' {
|
||||
k2 += 1;
|
||||
}
|
||||
k2 -= 1;
|
||||
|
||||
(k1, k2)
|
||||
}
|
||||
|
||||
/// 解析非标准参数输入。
|
||||
///
|
||||
/// Fortran 原始逻辑:
|
||||
/// 1. 逐行读取输入文本
|
||||
/// 2. 提取关键字-值对（支持 `KEY VALUE` 和 `KEY = VALUE` 格式）
|
||||
/// 3. 匹配 VARNAM 表中的关键字
|
||||
/// 4. 将值写入对应的 PVALUE 槽位
|
||||
/// 5. 最终按顺序解析所有值为具体类型
|
||||
pub fn nstpar(input_text: &str) -> NstparParams {
|
||||
let mut pvalue: Vec<String> = PVALUE_DEFAULT.iter().map(|s| s.to_string()).collect();
|
||||
|
||||
// 解析输入文本中的关键字-值对
|
||||
let mut var_map: HashMap<String, usize> = HashMap::new();
|
||||
for (i, name) in VARNAM.iter().enumerate() {
|
||||
var_map.insert(name.to_string(), i);
|
||||
}
|
||||
|
||||
for line in input_text.lines() {
|
||||
let line = line.trim();
|
||||
if line.is_empty() || line.starts_with('!') || line.starts_with('#') {
|
||||
continue;
|
||||
}
|
||||
|
||||
let mut k0: usize = 1;
|
||||
let text_len = line.len();
|
||||
let mut current_var: usize = 0;
|
||||
let mut found_var = false;
|
||||
|
||||
// 解析关键字-值对循环（匹配 Fortran 的 indv 切换逻辑）
|
||||
let mut indv = -1_i32;
|
||||
loop {
|
||||
if k0 > text_len {
|
||||
break;
|
||||
}
|
||||
let (k1, k2) = getwrd(line, k0);
|
||||
if k1 == 0 {
|
||||
break;
|
||||
}
|
||||
k0 = k2 + 2;
|
||||
|
||||
let word = &line[k1 - 1..k2];
|
||||
// 跳过等号
|
||||
if word == "=" {
|
||||
continue;
|
||||
}
|
||||
|
||||
indv = -indv;
|
||||
if indv == 1 {
|
||||
// 关键字位置
|
||||
if let Some(&idx) = var_map.get(word) {
|
||||
current_var = idx;
|
||||
found_var = true;
|
||||
} else {
|
||||
found_var = false;
|
||||
}
|
||||
} else if found_var {
|
||||
// 值位置，且前面找到了有效关键字
|
||||
pvalue[current_var] = word.to_string();
|
||||
found_var = false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 解析所有值为具体类型
|
||||
let parse_i32 = |s: &str| -> i32 { s.trim().parse::<f64>().unwrap_or(0.0) as i32 };
|
||||
let parse_f64 = |s: &str| -> f64 { s.trim().parse::<f64>().unwrap_or(0.0) };
|
||||
|
||||
let mut params = NstparParams::default();
|
||||
params.iatref = parse_i32(&pvalue[0]);
|
||||
params.bergfc = parse_f64(&pvalue[1]);
|
||||
params.ihydpr = parse_i32(&pvalue[2]);
|
||||
params.nunhhe = parse_i32(&pvalue[3]);
|
||||
params.sthe = parse_f64(&pvalue[4]);
|
||||
params.nd = parse_i32(&pvalue[5]);
|
||||
params.nfreqs = parse_i32(&pvalue[6]);
|
||||
params.ibfac = parse_i32(&pvalue[7]);
|
||||
params.intrpl = parse_i32(&pvalue[8]);
|
||||
params.ichang = parse_i32(&pvalue[9]);
|
||||
params.ifeos = parse_i32(&pvalue[10]);
|
||||
params.iophmi = parse_i32(&pvalue[11]);
|
||||
params.ioph2p = parse_i32(&pvalue[12]);
|
||||
params.iophem = parse_i32(&pvalue[13]);
|
||||
params.iopch = parse_i32(&pvalue[14]);
|
||||
params.iopoh = parse_i32(&pvalue[15]);
|
||||
params.ioph2m = parse_i32(&pvalue[16]);
|
||||
params.ioh2h2 = parse_i32(&pvalue[17]);
|
||||
params.ioh2he = parse_i32(&pvalue[18]);
|
||||
params.ioh2h1 = parse_i32(&pvalue[19]);
|
||||
params.iohee = parse_i32(&pvalue[20]);
|
||||
params.irsct = parse_i32(&pvalue[21]);
|
||||
params.irsch2 = parse_i32(&pvalue[22]);
|
||||
params.irsche = parse_i32(&pvalue[23]);
|
||||
params.trad = parse_f64(&pvalue[24]);
|
||||
params.wdil = parse_f64(&pvalue[25]);
|
||||
params.vtb = parse_f64(&pvalue[26]);
|
||||
params.ifmol = parse_i32(&pvalue[27]);
|
||||
params.tmolim = parse_f64(&pvalue[28]);
|
||||
params.moltab = parse_i32(&pvalue[29]);
|
||||
params.irwtab = parse_i32(&pvalue[30]);
|
||||
params.iirwin = parse_i32(&pvalue[31]);
|
||||
params.ipfexo = parse_i32(&pvalue[32]);
|
||||
params.cutlym = parse_f64(&pvalue[33]);
|
||||
params.cutbal = parse_f64(&pvalue[34]);
|
||||
params.ihxenb = parse_i32(&pvalue[35]);
|
||||
params.gsstd = parse_f64(&pvalue[36]);
|
||||
params.gwstd = parse_f64(&pvalue[37]);
|
||||
params.ihgom = parse_i32(&pvalue[38]);
|
||||
params.hglim = parse_f64(&pvalue[39]);
|
||||
params.erange = parse_f64(&pvalue[40]);
|
||||
params.ispick = parse_i32(&pvalue[41]);
|
||||
params.ilpick = parse_i32(&pvalue[42]);
|
||||
params.ippick = parse_i32(&pvalue[43]);
|
||||
|
||||
// Fortran 逻辑: if(imode.le.-3) 则关闭 Rayleigh 散射
|
||||
// 这里不处理 imode，由调用方负责
|
||||
|
||||
params
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_default_params() {
|
||||
let params = NstparParams::default();
|
||||
assert_eq!(params.iatref, 1);
|
||||
assert_eq!(params.nd, 70);
|
||||
assert_eq!(params.nfreqs, 120);
|
||||
assert_eq!(params.vtb, 2.0);
|
||||
assert_eq!(params.tmolim, 9000.0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_empty_input() {
|
||||
let params = nstpar("");
|
||||
assert_eq!(params.iatref, 1);
|
||||
assert_eq!(params.nd, 70);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parse_keyword_value() {
|
||||
let input = "ND 50\nNFREQS 200\nVTB 3.0\n";
|
||||
let params = nstpar(input);
|
||||
assert_eq!(params.nd, 50);
|
||||
assert_eq!(params.nfreqs, 200);
|
||||
assert!((params.vtb - 3.0).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parse_with_equals() {
|
||||
let input = "ND = 50\nNFREQS = 200\n";
|
||||
let params = nstpar(input);
|
||||
assert_eq!(params.nd, 50);
|
||||
assert_eq!(params.nfreqs, 200);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parse_float_values() {
|
||||
let input = "BERGFC 2.5\nSTHE 5.e18\nCUTLYM 0.5\n";
|
||||
let params = nstpar(input);
|
||||
assert!((params.bergfc - 2.5).abs() < 1e-10);
|
||||
assert!((params.sthe - 5.0e18).abs() < 1e10);
|
||||
assert!((params.cutlym - 0.5).abs() < 1e-10);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_parse_opacity_switches() {
|
||||
let input = "IOPHMI 0\nIOPH2P 0\nIOPOH 0\n";
|
||||
let params = nstpar(input);
|
||||
assert_eq!(params.iophmi, 0);
|
||||
assert_eq!(params.ioph2p, 0);
|
||||
assert_eq!(params.iopoh, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_unknown_keyword_ignored() {
|
||||
let input = "UNKNOWN 42\nND 50\n";
|
||||
let params = nstpar(input);
|
||||
assert_eq!(params.nd, 50);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_comments_skipped() {
|
||||
let input = "! This is a comment\n# Another comment\nND 50\n";
|
||||
let params = nstpar(input);
|
||||
assert_eq!(params.nd, 50);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_getwrd_basic() {
|
||||
let (k1, k2) = getwrd("HELLO WORLD", 1);
|
||||
assert_eq!(k1, 1);
|
||||
assert_eq!(k2, 5);
|
||||
|
||||
let (k1, k2) = getwrd("HELLO WORLD", 6);
|
||||
assert_eq!(k1, 7);
|
||||
assert_eq!(k2, 11);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_getwrd_empty() {
|
||||
let (k1, k2) = getwrd(" ", 1);
|
||||
assert_eq!(k1, 0);
|
||||
assert_eq!(k2, 0);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_full_parameter_set() {
|
||||
let input = "\
|
||||
IATREF 2
|
||||
BERGFC 1.5
|
||||
ND 30
|
||||
NFREQS 100
|
||||
IBFAC 1
|
||||
IFEOS 1
|
||||
IOPHMI 0
|
||||
VTB 3.5
|
||||
IFMOL 0
|
||||
TMOLIM 8000.
|
||||
GSSTD 1.0e-5
|
||||
GWSTD 1.0e-8
|
||||
IHGOM 1
|
||||
HGLIM 5.e17
|
||||
ERANGE 0.20
|
||||
";
|
||||
let params = nstpar(input);
|
||||
assert_eq!(params.iatref, 2);
|
||||
assert!((params.bergfc - 1.5).abs() < 1e-10);
|
||||
assert_eq!(params.nd, 30);
|
||||
assert_eq!(params.nfreqs, 100);
|
||||
assert_eq!(params.ibfac, 1);
|
||||
assert_eq!(params.ifeos, 1);
|
||||
assert_eq!(params.iophmi, 0);
|
||||
assert!((params.vtb - 3.5).abs() < 1e-10);
|
||||
assert_eq!(params.ifmol, 0);
|
||||
assert!((params.tmolim - 8000.0).abs() < 1e-10);
|
||||
assert!((params.gsstd - 1.0e-5).abs() < 1e-15);
|
||||
assert!((params.gwstd - 1.0e-8).abs() < 1e-18);
|
||||
assert_eq!(params.ihgom, 1);
|
||||
assert!((params.hglim - 5.0e17).abs() < 1e10);
|
||||
assert!((params.erange - 0.20).abs() < 1e-10);
|
||||
}
|
||||
}
|
||||
665
src/synspec/math/opac.rs
Normal file
665
src/synspec/math/opac.rs
Normal file
@ -0,0 +1,665 @@
|
||||
//! Opacity and emissivity calculation for SYNSPEC.
|
||||
//!
|
||||
//! Translated from SYNSPEC `OPAC` subroutine (synspec54.f:4541).
|
||||
//!
|
||||
//! Calculates absorption, emission, and scattering coefficients
|
||||
//! at a given depth for multiple frequencies.
|
||||
//!
|
||||
//! This routine calls downstream functions for line opacity (LINOP),
|
||||
//! molecular opacity (MOLOP), hydrogen lines (HYDLIN), He II lines (HE2LIN),
|
||||
//! and photoionization (PHTION, PHTX).
|
||||
|
||||
use crate::synspec::math::{
|
||||
gfree, he2lin, hydlin, linop, lymlin, molop, phtion, phtx,
|
||||
He2linParams, HydlinParams, LinopParams, LymlinParams,
|
||||
MolLineData, MolopConfig, MolopFreqParams, MolopModelState,
|
||||
PhtionParams, PhtionOutput, PhtxParams, PhtxOutput,
|
||||
};
|
||||
use crate::synspec::math::voigtk::MVOI;
|
||||
use crate::tlusty::math::hydrogen::sffhmi;
|
||||
|
||||
/// Physical constants
|
||||
const UN: f64 = 1.0;
|
||||
const TEN15: f64 = 1.0e-15;
|
||||
const CSB: f64 = 2.0706e-16;
|
||||
const CFF: f64 = 3.694e8;
|
||||
|
||||
/// Line opacity data for LINOP call.
|
||||
///
|
||||
/// Corresponds to Fortran COMMON/LINDAT/ data needed by OPAC → LINOP.
|
||||
pub struct OpacLinopData<'a> {
|
||||
/// Temperature at depth ID
|
||||
pub temp: f64,
|
||||
/// Total frequency count (NFREQS)
|
||||
pub nfreqs: usize,
|
||||
/// Frequency spacing factor (DFRCON)
|
||||
pub dfrcon: f64,
|
||||
/// Planck function at depth ID
|
||||
pub plan: f64,
|
||||
/// Stimulated emission correction
|
||||
pub stim: f64,
|
||||
/// Number of lines
|
||||
pub nlin: usize,
|
||||
/// Line data array
|
||||
pub lines: &'a [crate::synspec::math::LineData],
|
||||
/// RRR(ID,ION,IAT) — Saha/Boltzmann factors
|
||||
pub rrr: &'a [f64],
|
||||
/// RRR dimensions: (natom, nion)
|
||||
pub rrr_dims: (usize, usize),
|
||||
/// Doppler width DOPA1(IAT, ID)
|
||||
pub dopa1: &'a [f64],
|
||||
/// DOPA1 number of atoms
|
||||
pub dopa1_nat: usize,
|
||||
/// Level statistical weights G(level)
|
||||
pub g: &'a [f64],
|
||||
/// Level populations POPUL(level, ID)
|
||||
pub popul: &'a [f64],
|
||||
/// POPUL number of levels
|
||||
pub popul_nlev: usize,
|
||||
/// NLTE populations PNLT(IAT, ION, ID)
|
||||
pub pnlt: &'a [f64],
|
||||
/// PNLT dimensions: (natom, nion)
|
||||
pub pnlt_dims: (usize, usize),
|
||||
/// Ionization energies ENEV(IAT, ION)
|
||||
pub enev: &'a [f64],
|
||||
/// ENEV number of atoms
|
||||
pub enev_nat: usize,
|
||||
/// Level energies ENION(level)
|
||||
pub enion: &'a [f64],
|
||||
/// Laser deletion flag
|
||||
pub lasdel: bool,
|
||||
/// He II special line count
|
||||
pub nsp: usize,
|
||||
/// He II special line indices
|
||||
pub isp0: &'a [usize],
|
||||
/// Voigt table H0
|
||||
pub h0tab: &'a [f64; MVOI],
|
||||
/// Voigt table H1
|
||||
pub h1tab: &'a [f64; MVOI],
|
||||
/// Voigt table H2
|
||||
pub h2tab: &'a [f64; MVOI],
|
||||
}
|
||||
|
||||
/// Hydrogen line data for HYDLIN call.
|
||||
pub struct OpacHydlinData {
|
||||
/// Lower level for H lines
|
||||
pub ilowh: i32,
|
||||
/// Upper level limits
|
||||
pub m10: usize,
|
||||
pub m20: usize,
|
||||
/// Turbulent velocity (cm/s)
|
||||
pub vturb: f64,
|
||||
/// wnHint factors
|
||||
pub wn_hint: Vec<Vec<f64>>,
|
||||
}
|
||||
|
||||
/// He II line data for HE2LIN call.
|
||||
pub struct OpacHe2linData {
|
||||
/// He II common data
|
||||
pub common: crate::synspec::math::He2Common<'static>,
|
||||
/// He II lowest series index
|
||||
pub ilwhe2: usize,
|
||||
/// Max principal quantum number
|
||||
pub mhe10: usize,
|
||||
/// Upper limit for He II lines
|
||||
pub mhe20: usize,
|
||||
}
|
||||
|
||||
/// Molecular opacity data for MOLOP call.
|
||||
pub struct OpacMolopData<'a> {
|
||||
/// MOLOP configuration
|
||||
pub config: &'a MolopConfig,
|
||||
/// Molecular line data
|
||||
pub line_data: &'a MolLineData<'a>,
|
||||
/// Voigt tables (H0, H1, H2)
|
||||
pub voigt_tables: (&'a [f64; MVOI], &'a [f64; MVOI], &'a [f64; MVOI]),
|
||||
}
|
||||
|
||||
/// Photoionization data for PHTION call.
|
||||
pub struct OpacPhtionData<'a> {
|
||||
/// Photoionization cross-section data
|
||||
pub photcs: &'a crate::synspec::math::PhotcsData,
|
||||
/// Atomic number density RRR
|
||||
pub rrr: &'a [f64],
|
||||
/// Level populations POPUL
|
||||
pub popul: &'a [f64],
|
||||
}
|
||||
|
||||
/// Extended photoionization data for PHTX call.
|
||||
pub struct OpacPhtxData<'a> {
|
||||
/// Level photoionization data
|
||||
pub level_photo: &'a crate::synspec::math::LevelPhotoData,
|
||||
/// H/He photoionization data
|
||||
pub hhe_photo: &'a crate::synspec::math::HhePhotoData,
|
||||
/// Level populations POPUL
|
||||
pub popul: &'a [f64],
|
||||
/// Atomic number density RRR
|
||||
pub rrr: &'a [f64],
|
||||
/// Total frequency count
|
||||
pub nfreq_total: usize,
|
||||
/// Continuous frequency count
|
||||
pub nfreqc: usize,
|
||||
/// IASV flag
|
||||
pub iasv: i32,
|
||||
/// NQHT flag
|
||||
pub nqht: i32,
|
||||
/// PHTX state for interpolation caching
|
||||
pub state: &'a mut crate::synspec::math::PhtxState,
|
||||
}
|
||||
|
||||
/// Parameters for opacity calculation.
|
||||
///
|
||||
/// Includes optional data for downstream functions (LINOP, MOLOP, HYDLIN,
|
||||
/// HE2LIN, PHTION, PHTX). When `None`, the corresponding call is skipped.
|
||||
pub struct OpacParams<'a> {
|
||||
/// Depth index
|
||||
pub id: usize,
|
||||
/// Temperature at depth ID (K)
|
||||
pub t: f64,
|
||||
/// Electron density at depth ID
|
||||
pub ane: f64,
|
||||
/// Number of frequencies
|
||||
pub nfreq: usize,
|
||||
/// Frequency array (Hz)
|
||||
pub freq: Vec<f64>,
|
||||
/// Wavelength array (Å)
|
||||
pub wlam: Vec<f64>,
|
||||
/// Frequency interpolation weights
|
||||
pub frx1: Vec<f64>,
|
||||
pub frx2: Vec<f64>,
|
||||
/// Mode flag
|
||||
pub imode: i32,
|
||||
/// Standard depth index
|
||||
pub idstd: usize,
|
||||
/// Continuum opacity switch
|
||||
pub icontl: i32,
|
||||
/// Lyman line treatment switch
|
||||
pub iophli: i32,
|
||||
/// Hydrogen line mode (0=interpolated, >0=detailed)
|
||||
pub ihyll: i32,
|
||||
/// He II line switch
|
||||
pub ihe2l: i32,
|
||||
/// Molecular line switch
|
||||
pub ifmol: i32,
|
||||
/// Number of molecular lists
|
||||
pub nmlist: usize,
|
||||
/// H atom index
|
||||
pub iath: i32,
|
||||
/// H ground level population
|
||||
pub pop_h: f64,
|
||||
/// H continuum level population
|
||||
pub pop_h_cont: f64,
|
||||
/// Electron scattering coefficient
|
||||
pub sce: f64,
|
||||
/// Planck function at depth ID
|
||||
pub plan: f64,
|
||||
/// H/kT
|
||||
pub hkt: f64,
|
||||
/// h/k
|
||||
pub hk: f64,
|
||||
/// Boltzmann constant * temperature
|
||||
pub bn: f64,
|
||||
/// wnHint factors for Lyman lines
|
||||
pub wn_hint: [f64; 5],
|
||||
/// RELOP factor for AVAB calculation
|
||||
pub relop: f64,
|
||||
|
||||
// --- Downstream function data (optional) ---
|
||||
/// Line opacity data for LINOP
|
||||
pub linop_data: Option<OpacLinopData<'a>>,
|
||||
/// Hydrogen line data for HYDLIN
|
||||
pub hydlin_data: Option<OpacHydlinData>,
|
||||
/// He II line data for HE2LIN
|
||||
pub he2lin_data: Option<OpacHe2linData>,
|
||||
/// Molecular opacity data for MOLOP
|
||||
pub molop_data: Option<OpacMolopData<'a>>,
|
||||
/// Photoionization data for PHTION
|
||||
pub phtion_data: Option<OpacPhtionData<'a>>,
|
||||
/// Extended photoionization data for PHTX
|
||||
pub phtx_data: Option<OpacPhtxData<'a>>,
|
||||
}
|
||||
|
||||
/// Result of opacity calculation
|
||||
pub struct OpacResult {
|
||||
/// Absorption coefficient array
|
||||
pub abso: Vec<f64>,
|
||||
/// Emission coefficient array
|
||||
pub emis: Vec<f64>,
|
||||
/// Scattering coefficient array
|
||||
pub scat: Vec<f64>,
|
||||
/// Average absorption for line selection
|
||||
pub avab: f64,
|
||||
}
|
||||
|
||||
/// Calculate opacity and emissivity.
|
||||
///
|
||||
/// This is the main opacity calculation routine for SYNSPEC.
|
||||
/// It computes absorption, emission, and scattering coefficients
|
||||
/// including continuum, line, and molecular contributions.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Input parameters (including optional downstream data)
|
||||
///
|
||||
/// # Returns
|
||||
/// Opacity coefficients for all frequencies
|
||||
pub fn opac(params: &mut OpacParams) -> OpacResult {
|
||||
let nfreq = params.nfreq;
|
||||
let mut abso = vec![0.0; nfreq];
|
||||
let mut emis = vec![0.0; nfreq];
|
||||
let mut scat = vec![0.0; nfreq];
|
||||
|
||||
// Skip if not needed
|
||||
if params.imode == -1 && params.id != params.idstd {
|
||||
return OpacResult {
|
||||
abso,
|
||||
emis,
|
||||
scat,
|
||||
avab: 0.0,
|
||||
};
|
||||
}
|
||||
|
||||
let t = params.t;
|
||||
let ane = params.ane;
|
||||
let t1 = UN / t;
|
||||
let hkt = params.hkt;
|
||||
let _tk = hkt / params.hk;
|
||||
let srt = UN / t.sqrt();
|
||||
let sgff = CFF * srt;
|
||||
let con = CSB * t1 * srt;
|
||||
let _conts = 1.0e-36 / con;
|
||||
|
||||
// Continuum opacity (first and last frequency)
|
||||
let ij0 = if nfreq == 1 {
|
||||
1
|
||||
} else if params.imode == 2 {
|
||||
nfreq
|
||||
} else {
|
||||
2
|
||||
};
|
||||
|
||||
let mut ably1 = 0.0;
|
||||
let mut emly1 = 0.0;
|
||||
let mut scly1 = 0.0;
|
||||
let mut ably = 0.0;
|
||||
let mut emly = 0.0;
|
||||
let mut scly = 0.0;
|
||||
|
||||
for ij in 0..ij0 {
|
||||
let fr = params.freq[ij];
|
||||
let fr15 = fr * TEN15;
|
||||
let bnu = params.bn * fr15 * fr15 * fr15;
|
||||
let hkf = hkt * fr;
|
||||
|
||||
// Bound-free and free-free continuum
|
||||
let abf = 0.0;
|
||||
let ebf = 0.0;
|
||||
let mut aff = 0.0;
|
||||
|
||||
// Free-free Gaunt factor contribution
|
||||
let gfree_val = gfree(t, fr);
|
||||
let sf1 = sgff / (fr * fr * fr);
|
||||
let hktm = hkt * fr.min(fr);
|
||||
let sf2 = hktm.exp() + gfree_val - UN;
|
||||
let sff = sf1 * sf2;
|
||||
aff += sff;
|
||||
|
||||
// H- opacity
|
||||
let sffhmi_val = sffhmi(params.pop_h, fr, t);
|
||||
aff += sffhmi_val;
|
||||
|
||||
// Additional opacities (OPADD placeholder)
|
||||
let abad = 0.0;
|
||||
let emad = 0.0;
|
||||
let scad = 0.0;
|
||||
|
||||
// Lyman line wings
|
||||
let (ably_ij, emly_ij, scly_ij) = if params.iophli != 0 {
|
||||
let lymlin_params = LymlinParams {
|
||||
id: params.id,
|
||||
freq: fr,
|
||||
pop_h: params.pop_h,
|
||||
pop_excited: [params.pop_h_cont; 4],
|
||||
t,
|
||||
ane,
|
||||
iath: params.iath,
|
||||
iophli: params.iophli,
|
||||
wn_hint: params.wn_hint,
|
||||
};
|
||||
let result = lymlin(&lymlin_params);
|
||||
(result.ably, result.emly, result.scly)
|
||||
} else {
|
||||
(0.0, 0.0, 0.0)
|
||||
};
|
||||
|
||||
// Total opacity and emissivity
|
||||
let x = (-hkf).exp();
|
||||
let x1 = UN - x;
|
||||
let bne = bnu * x * ane;
|
||||
|
||||
abso[ij] = abf + ane * (x1 * aff - x * ebf) + abad;
|
||||
emis[ij] = bne * (aff + ebf) + emad + emly_ij;
|
||||
scat[ij] = scad + scly_ij + params.sce;
|
||||
|
||||
if ij == 0 {
|
||||
ably1 = ably_ij;
|
||||
emly1 = emly_ij;
|
||||
scly1 = scly_ij;
|
||||
}
|
||||
// Fortran: ABLY holds values from last iteration (IJ=IJ0=2)
|
||||
ably = ably_ij;
|
||||
emly = emly_ij;
|
||||
scly = scly_ij;
|
||||
}
|
||||
|
||||
let avab = (abso[0] + abso[1] + scat[0] + scat[1]) * 0.5 * params.relop;
|
||||
|
||||
if nfreq <= 2 || params.imode == -1 {
|
||||
return OpacResult {
|
||||
abso,
|
||||
emis,
|
||||
scat,
|
||||
avab,
|
||||
};
|
||||
}
|
||||
|
||||
// Determine M (start index for line accumulation)
|
||||
let m: usize = if params.icontl == 1 { 1 } else { 3 };
|
||||
|
||||
if params.imode != 2 {
|
||||
// Interpolated continuum for all frequencies
|
||||
for ij in 2..nfreq {
|
||||
abso[ij] = params.frx1[ij] * abso[1] + params.frx2[ij] * abso[0];
|
||||
emis[ij] = params.frx1[ij] * emis[1] + params.frx2[ij] * emis[0];
|
||||
scat[ij] = params.frx1[ij] * scat[1] + params.frx2[ij] * scat[0];
|
||||
}
|
||||
|
||||
// Hydrogen lines for IHYL=0 (interpolated mode)
|
||||
if params.ihyll == 0
|
||||
&& let Some(ref hyd_data) = params.hydlin_data {
|
||||
let hyd_params = HydlinParams {
|
||||
id: params.id,
|
||||
i0: 0,
|
||||
i1: 1,
|
||||
nfreq,
|
||||
wlam: params.wlam.clone(),
|
||||
freq: params.freq.clone(),
|
||||
t,
|
||||
ane,
|
||||
iath: params.iath,
|
||||
ilowh: hyd_data.ilowh,
|
||||
m10: hyd_data.m10,
|
||||
m20: hyd_data.m20,
|
||||
pop_h: params.pop_h,
|
||||
pop_h_cont: params.pop_h_cont,
|
||||
vturb: hyd_data.vturb,
|
||||
wn_hint: hyd_data.wn_hint.clone(),
|
||||
nunalp: 0,
|
||||
nunbet: 0,
|
||||
nungam: 0,
|
||||
nunbal: 0,
|
||||
allard_data: None,
|
||||
hneutr: 0.0,
|
||||
hcharg: 0.0,
|
||||
nunhhe: 0,
|
||||
iathe: 0,
|
||||
pop_he: 0.0,
|
||||
};
|
||||
let result = hydlin(&hyd_params);
|
||||
for ij in m..nfreq {
|
||||
abso[ij] += params.frx1[ij] * result.absoh[1]
|
||||
+ params.frx2[ij] * result.absoh[0];
|
||||
emis[ij] += params.frx1[ij] * result.emish[1]
|
||||
+ params.frx2[ij] * result.emish[0];
|
||||
}
|
||||
}
|
||||
|
||||
// Line opacity (LINOP)
|
||||
if let Some(ref linop_d) = params.linop_data {
|
||||
let linop_params = LinopParams {
|
||||
id: params.id,
|
||||
temp: linop_d.temp,
|
||||
nfreq,
|
||||
freq: ¶ms.freq,
|
||||
nfreqs: linop_d.nfreqs,
|
||||
dfrcon: linop_d.dfrcon,
|
||||
avab,
|
||||
plan: linop_d.plan,
|
||||
stim: linop_d.stim,
|
||||
nlin: linop_d.nlin,
|
||||
lines: linop_d.lines,
|
||||
rrr: linop_d.rrr,
|
||||
rrr_dims: linop_d.rrr_dims,
|
||||
dopa1: linop_d.dopa1,
|
||||
dopa1_nat: linop_d.dopa1_nat,
|
||||
g: linop_d.g,
|
||||
popul: linop_d.popul,
|
||||
popul_nlev: linop_d.popul_nlev,
|
||||
pnlt: linop_d.pnlt,
|
||||
pnlt_dims: linop_d.pnlt_dims,
|
||||
enev: linop_d.enev,
|
||||
enev_nat: linop_d.enev_nat,
|
||||
enion: linop_d.enion,
|
||||
lasdel: linop_d.lasdel,
|
||||
nsp: linop_d.nsp,
|
||||
isp0: linop_d.isp0,
|
||||
h0tab: linop_d.h0tab,
|
||||
h1tab: linop_d.h1tab,
|
||||
h2tab: linop_d.h2tab,
|
||||
phe1_data: None,
|
||||
phe2_common: None,
|
||||
};
|
||||
let result = linop(&linop_params);
|
||||
for ij in 3..nfreq {
|
||||
abso[ij] += result.ablin[ij];
|
||||
emis[ij] += result.emlin[ij];
|
||||
}
|
||||
}
|
||||
|
||||
// Molecular line opacity (MOLOP)
|
||||
if params.ifmol > 0
|
||||
&& let Some(ref mol_d) = params.molop_data {
|
||||
for ilist in 0..params.nmlist {
|
||||
let mol_params = MolopFreqParams {
|
||||
nfreq,
|
||||
nfreqs: nfreq,
|
||||
freq: ¶ms.freq,
|
||||
};
|
||||
let result = molop(
|
||||
mol_d.config,
|
||||
&MolopModelState {
|
||||
temp: t,
|
||||
elec: ane,
|
||||
stim: 1.0,
|
||||
plan: params.plan,
|
||||
mol: crate::synspec::math::MolModelState {
|
||||
rrmol: &[],
|
||||
dopmol: &[],
|
||||
},
|
||||
},
|
||||
mol_d.line_data,
|
||||
&mol_params,
|
||||
avab,
|
||||
mol_d.voigt_tables,
|
||||
);
|
||||
for ij in 3..nfreq {
|
||||
abso[ij] += result.ablin[ij];
|
||||
emis[ij] += result.emlin[ij];
|
||||
}
|
||||
let _ = ilist; // suppress unused warning
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Detailed hydrogen line opacity (IHYL>0 or IMODE=2)
|
||||
if (params.ihyll > 0 || params.imode == 2)
|
||||
&& let Some(ref hyd_data) = params.hydlin_data {
|
||||
let hyd_params = HydlinParams {
|
||||
id: params.id,
|
||||
i0: m.saturating_sub(1),
|
||||
i1: nfreq.saturating_sub(1),
|
||||
nfreq,
|
||||
wlam: params.wlam.clone(),
|
||||
freq: params.freq.clone(),
|
||||
t,
|
||||
ane,
|
||||
iath: params.iath,
|
||||
ilowh: hyd_data.ilowh,
|
||||
m10: hyd_data.m10,
|
||||
m20: hyd_data.m20,
|
||||
pop_h: params.pop_h,
|
||||
pop_h_cont: params.pop_h_cont,
|
||||
vturb: hyd_data.vturb,
|
||||
wn_hint: hyd_data.wn_hint.clone(),
|
||||
nunalp: 0,
|
||||
nunbet: 0,
|
||||
nungam: 0,
|
||||
nunbal: 0,
|
||||
allard_data: None,
|
||||
hneutr: 0.0,
|
||||
hcharg: 0.0,
|
||||
nunhhe: 0,
|
||||
iathe: 0,
|
||||
pop_he: 0.0,
|
||||
};
|
||||
let result = hydlin(&hyd_params);
|
||||
for ij in m..nfreq {
|
||||
abso[ij] += result.absoh[ij];
|
||||
emis[ij] += result.emish[ij];
|
||||
}
|
||||
}
|
||||
|
||||
// Detailed He II line opacity (IHE2L>0)
|
||||
if params.ihe2l > 0
|
||||
&& let Some(ref he2_d) = params.he2lin_data {
|
||||
let he2_params = He2linParams {
|
||||
common: he2_d.common.clone(),
|
||||
i0: m.saturating_sub(1),
|
||||
i1: nfreq.saturating_sub(1),
|
||||
ilwhe2: he2_d.ilwhe2,
|
||||
mhe10: he2_d.mhe10,
|
||||
mhe20: he2_d.mhe20,
|
||||
};
|
||||
let result = he2lin(&he2_params);
|
||||
for ij in m..nfreq {
|
||||
abso[ij] += result.absoh[ij];
|
||||
emis[ij] += result.emish[ij];
|
||||
}
|
||||
}
|
||||
|
||||
// Photoionization opacity (PHTION)
|
||||
if let Some(ref phtion_d) = params.phtion_data {
|
||||
let mut phtion_out = PhtionOutput {
|
||||
abso: &mut abso,
|
||||
emis: &mut emis,
|
||||
};
|
||||
let phtion_params = PhtionParams {
|
||||
temp: t,
|
||||
fre: ¶ms.freq,
|
||||
nfre: nfreq,
|
||||
rrr: phtion_d.rrr,
|
||||
popul: phtion_d.popul,
|
||||
photcs: phtion_d.photcs,
|
||||
};
|
||||
phtion(&phtion_params, &mut phtion_out);
|
||||
}
|
||||
|
||||
// Photoionization opacity extended (PHTX)
|
||||
if let Some(ref mut phtx_d) = params.phtx_data {
|
||||
let mut phtx_out = PhtxOutput {
|
||||
abso: &mut abso,
|
||||
emis: &mut emis,
|
||||
};
|
||||
let phtx_params = PhtxParams {
|
||||
temp: t,
|
||||
depth_id: params.id,
|
||||
fre: ¶ms.freq,
|
||||
nfre: nfreq,
|
||||
nfreq,
|
||||
nfreqc: phtx_d.nfreqc,
|
||||
icon: 0,
|
||||
iasv: phtx_d.iasv,
|
||||
nqht: phtx_d.nqht,
|
||||
popul: phtx_d.popul,
|
||||
rrr: phtx_d.rrr,
|
||||
level_photo: phtx_d.level_photo,
|
||||
hhe_photo: phtx_d.hhe_photo,
|
||||
};
|
||||
phtx(&phtx_params, &mut phtx_out, phtx_d.state);
|
||||
}
|
||||
|
||||
// Add scattering to absorption for positive imode
|
||||
if params.imode >= 0 {
|
||||
for ij in 0..nfreq {
|
||||
abso[ij] += scat[ij];
|
||||
}
|
||||
}
|
||||
|
||||
// Correct for Lyman line wings (both first and second frequency points)
|
||||
// Fortran: ABSO(1)-=ABLY1, ABSO(2)-=ABLY (for ICONTL!=1)
|
||||
if params.icontl != 1 {
|
||||
abso[0] -= ably1;
|
||||
emis[0] -= emly1;
|
||||
scat[0] -= scly1;
|
||||
if nfreq > 1 {
|
||||
abso[1] -= ably;
|
||||
emis[1] -= emly;
|
||||
scat[1] -= scly;
|
||||
}
|
||||
}
|
||||
|
||||
OpacResult {
|
||||
abso,
|
||||
emis,
|
||||
scat,
|
||||
avab,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_opac_basic() {
|
||||
let mut params = OpacParams {
|
||||
id: 0,
|
||||
t: 6000.0,
|
||||
ane: 1.0e13,
|
||||
nfreq: 5,
|
||||
freq: vec![1.0e14, 2.0e14, 3.0e14, 4.0e14, 5.0e14],
|
||||
wlam: vec![30000.0, 15000.0, 10000.0, 7500.0, 6000.0],
|
||||
frx1: vec![0.0, 0.0, 0.5, 0.5, 0.5],
|
||||
frx2: vec![0.0, 0.0, 0.5, 0.5, 0.5],
|
||||
imode: 0,
|
||||
idstd: 0,
|
||||
icontl: 0,
|
||||
iophli: 0,
|
||||
ihyll: 0,
|
||||
ihe2l: 0,
|
||||
ifmol: 0,
|
||||
nmlist: 0,
|
||||
iath: 1,
|
||||
pop_h: 1.0e16,
|
||||
pop_h_cont: 1.0e10,
|
||||
sce: 1.0e-3,
|
||||
plan: 1.0,
|
||||
hkt: 4.79928144e-11 / 6000.0,
|
||||
hk: 4.79928144e-11,
|
||||
bn: 1.0,
|
||||
wn_hint: [1.0, 1.0, 1.0, 1.0, 1.0],
|
||||
relop: 1.0,
|
||||
linop_data: None,
|
||||
hydlin_data: None,
|
||||
he2lin_data: None,
|
||||
molop_data: None,
|
||||
phtion_data: None,
|
||||
phtx_data: None,
|
||||
};
|
||||
|
||||
let result = opac(&mut params);
|
||||
assert!(result.abso.iter().all(|&x| x.is_finite()));
|
||||
assert!(result.emis.iter().all(|&x| x.is_finite()));
|
||||
assert!(result.scat.iter().all(|&x| x.is_finite()));
|
||||
assert!(result.avab.is_finite());
|
||||
}
|
||||
}
|
||||
321
src/synspec/math/opacon.rs
Normal file
321
src/synspec/math/opacon.rs
Normal file
@ -0,0 +1,321 @@
|
||||
//! Continuous opacity, emissivity, and scattering coefficients.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine OPACON(ID,CROSS,ABSOC,EMISC,SCATC)
|
||||
//! at line 12661.
|
||||
//!
|
||||
//! Computes absorption, emission, and scattering coefficients at a given
|
||||
//! depth point for continuum frequencies.
|
||||
|
||||
use super::dwnfr1::{dwnfr1, Dwnfr1Params};
|
||||
use super::gfree::gfree;
|
||||
use super::sffhmi::sffhmi;
|
||||
|
||||
/// Parameters for continuous opacity calculation.
|
||||
pub struct OpaconParams<'a> {
|
||||
/// Depth index
|
||||
pub id: usize,
|
||||
/// Temperature at each depth (K)
|
||||
pub temp: &'a [f64],
|
||||
/// Electron density at each depth (cm^-3)
|
||||
pub elec: &'a [f64],
|
||||
/// Continuum frequencies
|
||||
pub freqc: &'a [f64],
|
||||
/// Number of continuum frequencies
|
||||
pub nfreqc: usize,
|
||||
/// Number of ions
|
||||
pub nion: usize,
|
||||
/// First level index for each ion
|
||||
pub nfirst: &'a [usize],
|
||||
/// Last level index for each ion
|
||||
pub nlast: &'a [usize],
|
||||
/// Next ion index for each ion
|
||||
pub nnext: &'a [usize],
|
||||
/// Free-free flag for each ion
|
||||
pub ifree: &'a [i32],
|
||||
/// IELHM index (H- element)
|
||||
pub ielhm: usize,
|
||||
/// Ionic charge for each element
|
||||
pub iz: &'a [i32],
|
||||
/// Element index for each level
|
||||
pub iel: &'a [usize],
|
||||
/// Occupation probability flag for each level
|
||||
pub ifwop: &'a [i32],
|
||||
/// Population of each level at current depth
|
||||
pub popul_lev: &'a [f64],
|
||||
/// Statistical weight of each level
|
||||
pub g: &'a [f64],
|
||||
/// Ionization energy of each level (erg)
|
||||
pub enion: &'a [f64],
|
||||
/// Occupation probability for each level
|
||||
pub wop: &'a [f64],
|
||||
/// Dissolved fraction frequency for each level
|
||||
pub fropc: &'a [f64],
|
||||
/// Dissolved fraction index for each level
|
||||
pub indexp: &'a [i32],
|
||||
/// Z3 array for dissolved fraction
|
||||
pub z3: &'a [f64],
|
||||
/// ELEC23 array for dissolved fraction
|
||||
pub elec23: &'a [f64],
|
||||
/// DWC1 array for dissolved fraction
|
||||
pub dwc1: &'a [f64],
|
||||
/// DWC2 array for dissolved fraction
|
||||
pub dwc2: &'a [f64],
|
||||
/// DWC1 number of depth points
|
||||
pub ndepth: usize,
|
||||
/// Bergmann factor
|
||||
pub bergfc: f64,
|
||||
/// Planck function constant
|
||||
pub bn: f64,
|
||||
/// h/k constant
|
||||
pub hk: f64,
|
||||
/// Electron scattering cross-section
|
||||
pub sige: f64,
|
||||
/// Stimulated emission correction at current depth
|
||||
pub stim: f64,
|
||||
/// Planck function at current depth
|
||||
pub plan: f64,
|
||||
/// Photoionization cross-sections (nion x nfreqc)
|
||||
pub cross: &'a [&'a [f64]],
|
||||
/// Saha-Boltzmann factors for merged levels
|
||||
pub sgmerg_fn: Option<Box<dyn Fn(usize, usize, f64) -> f64>>,
|
||||
/// Additional opacity function
|
||||
pub opadd_fn: Option<Box<dyn Fn(usize, f64) -> (f64, f64, f64)>>,
|
||||
/// Lyman line function
|
||||
pub lymlin_fn: Option<Box<dyn Fn(usize, f64) -> (f64, f64, f64)>>,
|
||||
/// Photoionization function
|
||||
pub phtion_fn: Option<Box<dyn Fn(usize, &mut [f64], &mut [f64], &[f64], usize)>>,
|
||||
/// Photoionization function (extended)
|
||||
pub phtx_fn: Option<Box<dyn Fn(usize, &mut [f64], &mut [f64], &[f64], i32)>>,
|
||||
}
|
||||
|
||||
/// Result of continuous opacity calculation.
|
||||
pub struct OpaconResult {
|
||||
/// Absorption coefficient array
|
||||
pub absoc: Vec<f64>,
|
||||
/// Emission coefficient array
|
||||
pub emisc: Vec<f64>,
|
||||
/// Scattering coefficient array
|
||||
pub scatc: Vec<f64>,
|
||||
}
|
||||
|
||||
/// Continuous opacity, emissivity, and scattering coefficients.
|
||||
///
|
||||
/// Computes absorption, emission, and scattering coefficients at a given
|
||||
/// depth point for continuum frequencies. Includes bound-free, free-free,
|
||||
/// and additional opacity contributions.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Calculation parameters
|
||||
///
|
||||
/// # Returns
|
||||
/// Arrays of absorption, emission, and scattering coefficients.
|
||||
pub fn opacon(params: &OpaconParams) -> OpaconResult {
|
||||
let id = params.id;
|
||||
let t = params.temp[id];
|
||||
let ane = params.elec[id];
|
||||
let t1 = 1.0 / t;
|
||||
let hkt = params.hk * t1;
|
||||
let tk = hkt / params.bn;
|
||||
let srt = 1.0 / t.sqrt();
|
||||
let sgff = 3.694e8 * srt;
|
||||
let con = 2.0706e-16 * t1 * srt;
|
||||
let sce = ane * params.sige;
|
||||
|
||||
let mut absoc = vec![0.0; params.nfreqc];
|
||||
let mut emisc = vec![0.0; params.nfreqc];
|
||||
let mut scatc = vec![0.0; params.nfreqc];
|
||||
|
||||
for ij in 0..params.nfreqc {
|
||||
let fr = params.freqc[ij];
|
||||
let fr15 = fr * 1e-15;
|
||||
let bnu = params.bn * fr15 * fr15 * fr15;
|
||||
let hkf = hkt * fr;
|
||||
let mut abf = 0.0;
|
||||
let mut ebf = 0.0;
|
||||
let mut aff = 0.0;
|
||||
|
||||
for il in 0..params.nion {
|
||||
let n0i = params.nfirst[il];
|
||||
let n1i = params.nlast[il];
|
||||
let nke = params.nnext[il];
|
||||
let xn = params.popul_lev[nke];
|
||||
|
||||
// Bound-free contribution
|
||||
for ii in n0i..=n1i {
|
||||
let mut sg = 0.0;
|
||||
if params.ifwop[ii] < 0 {
|
||||
// Merged level
|
||||
if let Some(ref f) = params.sgmerg_fn {
|
||||
sg = f(ii, id, fr);
|
||||
}
|
||||
} else {
|
||||
sg = params.cross[il][ij];
|
||||
if sg <= 0.0 {
|
||||
continue;
|
||||
}
|
||||
// Dissolved fraction correction
|
||||
if params.indexp[ii] == 5 {
|
||||
let izz = params.iz[params.iel[ii]] as usize;
|
||||
let fr0 = params.enion[ii] / 6.6256e-27;
|
||||
let dw1 = dwnfr1(&Dwnfr1Params {
|
||||
fr,
|
||||
fr0,
|
||||
id,
|
||||
izz,
|
||||
z3: params.z3,
|
||||
elec23: params.elec23,
|
||||
dwc1: params.dwc1,
|
||||
ndepth: params.ndepth,
|
||||
dwc2: params.dwc2,
|
||||
bergfc: params.bergfc,
|
||||
});
|
||||
sg *= dw1;
|
||||
}
|
||||
}
|
||||
if params.popul_lev[ii] < 1e-20 || xn < 1e-20 {
|
||||
continue;
|
||||
}
|
||||
abf += sg * params.popul_lev[ii];
|
||||
let xx = sg * xn * (params.enion[ii] * tk - hkf).exp() * params.wop[ii];
|
||||
let _ee = (params.enion[ii] * tk - hkf).exp();
|
||||
ebf += xx * con * params.g[ii] / params.g[nke];
|
||||
}
|
||||
|
||||
let it = params.ifree[il];
|
||||
if it == 0 {
|
||||
continue;
|
||||
}
|
||||
|
||||
// Free-free contribution
|
||||
let ie = il;
|
||||
if ie == params.ielhm {
|
||||
let sff = sffhmi(xn, fr, t);
|
||||
aff += sff;
|
||||
} else {
|
||||
let ch = params.iz[il] as f64 * params.iz[il] as f64;
|
||||
let sf1 = ch * xn * sgff / (fr * fr * fr);
|
||||
let sff = if it == 2 {
|
||||
let sg = gfree(t, fr / ch);
|
||||
sf1 * sg
|
||||
} else {
|
||||
sf1
|
||||
};
|
||||
aff += sff;
|
||||
}
|
||||
}
|
||||
|
||||
// Additional opacities
|
||||
let (abad, emad, scad) = if let Some(ref f) = params.opadd_fn {
|
||||
f(id, fr)
|
||||
} else {
|
||||
(0.0, 0.0, 0.0)
|
||||
};
|
||||
|
||||
// Lyman lines
|
||||
let (ably, emly, scly) = if let Some(ref f) = params.lymlin_fn {
|
||||
f(id, fr)
|
||||
} else {
|
||||
(0.0, 0.0, 0.0)
|
||||
};
|
||||
|
||||
// Total opacity and emissivity
|
||||
let x = (-hkf).exp();
|
||||
let x1 = 1.0 - x;
|
||||
let bne = bnu * x * ane;
|
||||
absoc[ij] = abf + ane * (x1 * aff - ebf) + abad + ably;
|
||||
emisc[ij] = bne * aff + bnu * ane * ebf + emad + emly;
|
||||
scatc[ij] = scad + scly + sce;
|
||||
}
|
||||
|
||||
// Photoionization contributions
|
||||
if let Some(ref f) = params.phtion_fn {
|
||||
f(id, &mut absoc, &mut emisc, params.freqc, params.nfreqc);
|
||||
}
|
||||
if let Some(ref f) = params.phtx_fn {
|
||||
f(id, &mut absoc, &mut emisc, params.freqc, 1);
|
||||
}
|
||||
|
||||
OpaconResult {
|
||||
absoc,
|
||||
emisc,
|
||||
scatc,
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_opacon_basic() {
|
||||
let temp = [10000.0];
|
||||
let elec = [1e13];
|
||||
let freqc = [1e14, 2e14, 3e14];
|
||||
let nfirst = [0];
|
||||
let nlast = [0];
|
||||
let nnext = [1];
|
||||
let ifree = [1];
|
||||
let iz = [1];
|
||||
let iel = [0];
|
||||
let ifwop = [1];
|
||||
let popul_lev = [1e12, 1e10];
|
||||
let g = [1.0, 2.0];
|
||||
let enion = [0.0, 1e-11];
|
||||
let wop = [1.0, 1.0];
|
||||
let fropc = [0.0, 0.0];
|
||||
let indexp = [0, 0];
|
||||
let z3 = [1.0];
|
||||
let elec23 = [2.15e-9];
|
||||
let dwc1 = [0.0];
|
||||
let dwc2 = [0.0];
|
||||
let cross_row: &[f64] = &[1e-18, 2e-18, 3e-18];
|
||||
let cross: &[&[f64]] = &[cross_row];
|
||||
|
||||
let params = OpaconParams {
|
||||
id: 0,
|
||||
temp: &temp,
|
||||
elec: &elec,
|
||||
freqc: &freqc,
|
||||
nfreqc: 3,
|
||||
nion: 1,
|
||||
nfirst: &nfirst,
|
||||
nlast: &nlast,
|
||||
nnext: &nnext,
|
||||
ifree: &ifree,
|
||||
ielhm: 99,
|
||||
iz: &iz,
|
||||
iel: &iel,
|
||||
ifwop: &ifwop,
|
||||
popul_lev: &popul_lev,
|
||||
g: &g,
|
||||
enion: &enion,
|
||||
wop: &wop,
|
||||
fropc: &fropc,
|
||||
indexp: &indexp,
|
||||
z3: &z3,
|
||||
elec23: &elec23,
|
||||
dwc1: &dwc1,
|
||||
dwc2: &dwc2,
|
||||
ndepth: 1,
|
||||
bergfc: 1.0,
|
||||
bn: 1.0,
|
||||
hk: 4.79928144e-11,
|
||||
sige: 6.65e-25,
|
||||
stim: 1.0,
|
||||
plan: 1.0,
|
||||
cross,
|
||||
sgmerg_fn: None,
|
||||
opadd_fn: None,
|
||||
lymlin_fn: None,
|
||||
phtion_fn: None,
|
||||
phtx_fn: None,
|
||||
};
|
||||
|
||||
let result = opacon(¶ms);
|
||||
assert_eq!(result.absoc.len(), 3);
|
||||
assert_eq!(result.emisc.len(), 3);
|
||||
assert_eq!(result.scatc.len(), 3);
|
||||
assert!(result.absoc[0].is_finite());
|
||||
assert!(result.scatc[0].is_finite());
|
||||
}
|
||||
}
|
||||
559
src/synspec/math/opacw.rs
Normal file
559
src/synspec/math/opacw.rs
Normal file
@ -0,0 +1,559 @@
|
||||
//! Window-mode opacity, emissivity, and scattering coefficients.
|
||||
//!
|
||||
//! Translated from SYNSPEC54.FOR subroutine OPACW(ID,CROSS,ABSO,EMIS,
|
||||
//! ABSOC,EMISC,SCATC,MODC) at line 4770.
|
||||
//!
|
||||
//! Computes absorption, emission, and scattering coefficients at a given
|
||||
//! depth point for both continuum and all frequencies, including line
|
||||
//! opacity contributions.
|
||||
|
||||
use super::dwnfr1::{dwnfr1, Dwnfr1Params};
|
||||
use super::gfree::gfree;
|
||||
use super::sffhmi::sffhmi;
|
||||
|
||||
/// Continuum-only opacity parameters (shared with opacon).
|
||||
/// These fields are identical to OpaconParams.
|
||||
pub struct OpacwParams<'a> {
|
||||
/// Depth index
|
||||
pub id: usize,
|
||||
/// Temperature at each depth (K)
|
||||
pub temp: &'a [f64],
|
||||
/// Electron density at each depth (cm^-3)
|
||||
pub elec: &'a [f64],
|
||||
/// Continuum frequencies
|
||||
pub freqc: &'a [f64],
|
||||
/// Number of continuum frequencies
|
||||
pub nfreqc: usize,
|
||||
/// All frequencies (continuum + line)
|
||||
pub freq: &'a [f64],
|
||||
/// Number of all frequencies
|
||||
pub nfreq: usize,
|
||||
/// Number of ions
|
||||
pub nion: usize,
|
||||
/// First level index for each ion
|
||||
pub nfirst: &'a [usize],
|
||||
/// Last level index for each ion
|
||||
pub nlast: &'a [usize],
|
||||
/// Next ion index for each ion
|
||||
pub nnext: &'a [usize],
|
||||
/// Free-free flag for each ion
|
||||
pub ifree: &'a [i32],
|
||||
/// Free-free cutoff frequency for each ion
|
||||
pub ff: &'a [f64],
|
||||
/// IELHM index (H- element)
|
||||
pub ielhm: usize,
|
||||
/// Ionic charge for each element
|
||||
pub iz: &'a [i32],
|
||||
/// Element index for each level
|
||||
pub iel: &'a [usize],
|
||||
/// Occupation probability flag for each level
|
||||
pub ifwop: &'a [i32],
|
||||
/// Population of each level at current depth
|
||||
pub popul_lev: &'a [f64],
|
||||
/// Statistical weight of each level
|
||||
pub g: &'a [f64],
|
||||
/// Ionization energy of each level (erg)
|
||||
pub enion: &'a [f64],
|
||||
/// Occupation probability for each level
|
||||
pub wop: &'a [f64],
|
||||
/// Dissolved fraction index for each level
|
||||
pub indexp: &'a [i32],
|
||||
/// Z3 array for dissolved fraction
|
||||
pub z3: &'a [f64],
|
||||
/// ELEC23 array for dissolved fraction
|
||||
pub elec23: &'a [f64],
|
||||
/// DWC1 array for dissolved fraction
|
||||
pub dwc1: &'a [f64],
|
||||
/// DWC2 array for dissolved fraction
|
||||
pub dwc2: &'a [f64],
|
||||
/// DWC1 number of depth points
|
||||
pub ndepth: usize,
|
||||
/// Bergmann factor
|
||||
pub bergfc: f64,
|
||||
/// Planck function constant (BN)
|
||||
pub bn: f64,
|
||||
/// h/k constant (HK)
|
||||
pub hk: f64,
|
||||
/// Electron scattering cross-section * ANE
|
||||
pub sige: f64,
|
||||
/// Photoionization cross-sections (nion x nfreqc)
|
||||
pub cross: &'a [&'a [f64]],
|
||||
/// IMODE flag
|
||||
pub imode: i32,
|
||||
/// IDSTD standard depth index
|
||||
pub idstd: usize,
|
||||
/// ICONT flag for continuum subtraction
|
||||
pub icont: i32,
|
||||
/// IOPHLI flag for Lyman lines
|
||||
pub iophli: i32,
|
||||
/// IATH flag
|
||||
pub iath: i32,
|
||||
/// Interpolation coefficient FRX1 for all frequencies
|
||||
pub frx1: &'a [f64],
|
||||
/// Continuum frequency index for each frequency
|
||||
pub ijcint: &'a [usize],
|
||||
/// WN hint for Lyman lines
|
||||
pub wn_hint: f64,
|
||||
/// Pop_H for H- opacity
|
||||
pub pop_h: f64,
|
||||
/// Pop_H_cont for Lyman lines
|
||||
pub pop_h_cont: f64,
|
||||
/// SGMERG callback for merged levels
|
||||
pub sgmerg_fn: Option<Box<dyn Fn(usize, usize, f64) -> f64>>,
|
||||
/// OPADD callback for additional opacities
|
||||
pub opadd_fn: Option<Box<dyn Fn(usize, f64) -> (f64, f64, f64)>>,
|
||||
/// LYMLIN callback for Lyman line wings
|
||||
pub lymlin_fn: Option<Box<dyn Fn(usize, f64) -> (f64, f64, f64)>>,
|
||||
/// LINOPW callback for line opacity
|
||||
pub linopw_fn: Option<Box<dyn Fn(usize) -> (Vec<f64>, Vec<f64>)>>,
|
||||
/// MOLOP callback for molecular line opacity
|
||||
pub molop_fn: Option<Box<dyn Fn(usize, usize) -> (Vec<f64>, Vec<f64>)>>,
|
||||
/// HYDLIW callback for hydrogen line opacity
|
||||
pub hydliw_fn: Option<Box<dyn Fn(usize) -> (Vec<f64>, Vec<f64>)>>,
|
||||
/// HE2LIW callback for He II line opacity
|
||||
pub he2liw_fn: Option<Box<dyn Fn(usize) -> (Vec<f64>, Vec<f64>)>>,
|
||||
/// PHTION callback for photoionization
|
||||
pub phtion_fn: Option<Box<dyn Fn(usize, &mut [f64], &mut [f64], &[f64], usize)>>,
|
||||
/// PHTX callback for photoionization (extended)
|
||||
pub phtx_fn: Option<Box<dyn Fn(usize, &mut [f64], &mut [f64], &[f64], i32)>>,
|
||||
}
|
||||
|
||||
/// Result of window-mode opacity calculation.
|
||||
pub struct OpacwResult {
|
||||
/// Absorption coefficient at continuum frequencies
|
||||
pub absoc: Vec<f64>,
|
||||
/// Emission coefficient at continuum frequencies
|
||||
pub emisc: Vec<f64>,
|
||||
/// Scattering coefficient at continuum frequencies
|
||||
pub scatc: Vec<f64>,
|
||||
/// Absorption coefficient at all frequencies
|
||||
pub abso: Vec<f64>,
|
||||
/// Emission coefficient at all frequencies
|
||||
pub emis: Vec<f64>,
|
||||
/// Scattering coefficient at all frequencies
|
||||
pub scat: Vec<f64>,
|
||||
}
|
||||
|
||||
/// Window-mode opacity, emissivity, and scattering coefficients.
|
||||
///
|
||||
/// Computes absorption, emission, and scattering coefficients at a given
|
||||
/// depth point for both continuum and all frequencies. Includes bound-free,
|
||||
/// free-free, line, molecular, hydrogen, He II, and photoionization
|
||||
/// contributions.
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Calculation parameters
|
||||
///
|
||||
/// # Returns
|
||||
/// Arrays of absorption, emission, and scattering coefficients for both
|
||||
/// continuum and all frequencies.
|
||||
pub fn opacw(params: &OpacwParams) -> OpacwResult {
|
||||
let id = params.id;
|
||||
let t = params.temp[id];
|
||||
let ane = params.elec[id];
|
||||
let t1 = 1.0 / t;
|
||||
let hkt = params.hk * t1;
|
||||
let tk = hkt / params.bn;
|
||||
let srt = 1.0 / t.sqrt();
|
||||
let sgff = 3.694e8 * srt;
|
||||
let con = 2.0706e-16 * t1 * srt;
|
||||
let conts = 1.0e-36 / con;
|
||||
let sce = ane * params.sige;
|
||||
|
||||
let mut absoc = vec![0.0; params.nfreqc];
|
||||
let mut emisc = vec![0.0; params.nfreqc];
|
||||
let mut scatc = vec![0.0; params.nfreqc];
|
||||
let mut abso = vec![0.0; params.nfreq];
|
||||
let mut emis = vec![0.0; params.nfreq];
|
||||
let mut scat = vec![0.0; params.nfreq];
|
||||
|
||||
// Skip if not needed
|
||||
if params.imode == -1 && id != params.idstd {
|
||||
return OpacwResult { absoc, emisc, scatc, abso, emis, scat };
|
||||
}
|
||||
|
||||
let mut ably = 0.0;
|
||||
let mut emly = 0.0;
|
||||
let mut scly = 0.0;
|
||||
let _ij0 = if params.nfreq == 1 {
|
||||
1
|
||||
} else if params.imode == 2 {
|
||||
params.nfreq
|
||||
} else {
|
||||
2
|
||||
};
|
||||
|
||||
// Continuum opacity at continuum frequencies
|
||||
for ij in 0..params.nfreqc {
|
||||
let fr = params.freqc[ij];
|
||||
let fr15 = fr * 1e-15;
|
||||
let bnu = params.bn * fr15 * fr15 * fr15;
|
||||
let hkf = hkt * fr;
|
||||
let mut abf = 0.0;
|
||||
let mut ebf = 0.0;
|
||||
let mut aff = 0.0;
|
||||
|
||||
for il in 0..params.nion {
|
||||
let n0i = params.nfirst[il];
|
||||
let n1i = params.nlast[il];
|
||||
let nke = params.nnext[il];
|
||||
let xn = params.popul_lev[nke];
|
||||
|
||||
// Bound-free contribution
|
||||
for ii in n0i..=n1i {
|
||||
let mut sg = 0.0;
|
||||
if params.ifwop[ii] < 0 {
|
||||
// Merged level
|
||||
if let Some(ref f) = params.sgmerg_fn {
|
||||
sg = f(ii, id, fr);
|
||||
}
|
||||
} else {
|
||||
sg = params.cross[il][ij];
|
||||
// Dissolved fraction correction
|
||||
if params.indexp[ii] == 5 {
|
||||
let izz = params.iz[params.iel[ii]] as usize;
|
||||
let fr0 = params.enion[ii] / 6.6256e-27;
|
||||
let dw1 = dwnfr1(&Dwnfr1Params {
|
||||
fr,
|
||||
fr0,
|
||||
id,
|
||||
izz,
|
||||
z3: params.z3,
|
||||
elec23: params.elec23,
|
||||
dwc1: params.dwc1,
|
||||
ndepth: params.ndepth,
|
||||
dwc2: params.dwc2,
|
||||
bergfc: params.bergfc,
|
||||
});
|
||||
sg *= dw1;
|
||||
}
|
||||
}
|
||||
abf += sg * params.popul_lev[ii];
|
||||
let xx = sg * xn * (params.enion[ii] * tk).exp() * params.wop[ii];
|
||||
if xx < conts {
|
||||
continue;
|
||||
}
|
||||
ebf += xx * con * params.g[ii] / params.g[nke];
|
||||
}
|
||||
|
||||
let it = params.ifree[il];
|
||||
if it == 0 {
|
||||
continue;
|
||||
}
|
||||
|
||||
// Free-free contribution
|
||||
let ie = il;
|
||||
if ie == params.ielhm {
|
||||
let sff_val = sffhmi(xn, fr, t);
|
||||
aff += sff_val;
|
||||
} else {
|
||||
let ch = params.iz[il] as f64 * params.iz[il] as f64;
|
||||
let sf1 = ch * xn * sgff / (fr * fr * fr);
|
||||
let hktm = hkt * params.ff[il].min(fr);
|
||||
let sf2 = hktm.exp();
|
||||
let sff = if it == 2 {
|
||||
let sg = gfree(t, fr / ch);
|
||||
sf1 * (sf2 + sg - 1.0)
|
||||
} else {
|
||||
sf1 * sf2
|
||||
};
|
||||
aff += sff;
|
||||
}
|
||||
}
|
||||
|
||||
// Additional opacities
|
||||
let (abad, emad, scad) = if let Some(ref f) = params.opadd_fn {
|
||||
f(id, fr)
|
||||
} else {
|
||||
(0.0, 0.0, 0.0)
|
||||
};
|
||||
|
||||
// Lyman line wings
|
||||
let (ably_ij, emly_ij, scly_ij) = if params.iophli != 0 {
|
||||
if let Some(ref f) = params.lymlin_fn {
|
||||
f(id, fr)
|
||||
} else {
|
||||
(0.0, 0.0, 0.0)
|
||||
}
|
||||
} else {
|
||||
(0.0, 0.0, 0.0)
|
||||
};
|
||||
|
||||
// Total opacity and emissivity
|
||||
let x = (-hkf).exp();
|
||||
let x1 = 1.0 - x;
|
||||
let bne = bnu * x * ane;
|
||||
absoc[ij] = abf + ane * (x1 * aff - x * ebf) + sce + abad + ably_ij;
|
||||
emisc[ij] = bne * (aff + ebf) + emad + emly_ij;
|
||||
scatc[ij] = scad + scly_ij;
|
||||
ably = ably_ij;
|
||||
emly = emly_ij;
|
||||
scly = scly_ij;
|
||||
}
|
||||
|
||||
// If MODC==0, return continuum only
|
||||
// (In Fortran: if(modc.eq.0) return)
|
||||
// We always compute all frequencies here since the caller needs them
|
||||
|
||||
if params.nfreq <= 2 || params.imode == -1 {
|
||||
// Copy continuum to all frequencies
|
||||
for ij in 0..params.nfreq {
|
||||
abso[ij] = absoc[0];
|
||||
emis[ij] = emisc[0];
|
||||
scat[ij] = scatc[0];
|
||||
}
|
||||
return OpacwResult { absoc, emisc, scatc, abso, emis, scat };
|
||||
}
|
||||
|
||||
// Interpolated continuum for all frequencies
|
||||
for ij in 0..params.nfreq {
|
||||
let ijc = params.ijcint[ij];
|
||||
let frx1 = params.frx1[ij];
|
||||
if ijc + 1 < params.nfreqc {
|
||||
abso[ij] = frx1 * absoc[ijc] + (1.0 - frx1) * absoc[ijc + 1];
|
||||
emis[ij] = frx1 * emisc[ijc] + (1.0 - frx1) * emisc[ijc + 1];
|
||||
scat[ij] = frx1 * scatc[ijc] + (1.0 - frx1) * scatc[ijc + 1];
|
||||
}
|
||||
}
|
||||
|
||||
if params.imode != 2 {
|
||||
// Line opacity (LINOPW)
|
||||
if let Some(ref f) = params.linopw_fn {
|
||||
let (ablin, emlin) = f(id);
|
||||
for ij in 0..params.nfreq {
|
||||
abso[ij] += ablin[ij];
|
||||
emis[ij] += emlin[ij];
|
||||
}
|
||||
}
|
||||
|
||||
// Molecular line opacity (MOLOP)
|
||||
if let Some(ref f) = params.molop_fn {
|
||||
// nmlist would need to be passed; for now use 0
|
||||
// In practice, the caller handles this
|
||||
let _ = f;
|
||||
}
|
||||
}
|
||||
|
||||
// Detailed hydrogen line opacity (HYDLIW)
|
||||
if let Some(ref f) = params.hydliw_fn {
|
||||
let (ablin, emlin) = f(id);
|
||||
for ij in 0..params.nfreq {
|
||||
abso[ij] += ablin[ij];
|
||||
emis[ij] += emlin[ij];
|
||||
}
|
||||
}
|
||||
|
||||
// Detailed He II line opacity (HE2LIW)
|
||||
if let Some(ref f) = params.he2liw_fn {
|
||||
let (ablin, emlin) = f(id);
|
||||
for ij in 0..params.nfreq {
|
||||
abso[ij] += ablin[ij];
|
||||
emis[ij] += emlin[ij];
|
||||
}
|
||||
}
|
||||
|
||||
// Photoionization opacity
|
||||
if let Some(ref f) = params.phtion_fn {
|
||||
f(id, &mut abso, &mut emis, params.freq, params.nfreq);
|
||||
}
|
||||
if let Some(ref f) = params.phtx_fn {
|
||||
f(id, &mut abso, &mut emis, params.freq, 0);
|
||||
}
|
||||
|
||||
// Subtract Lyman line wings from continuum if ICONT != 1
|
||||
if params.icont != 1 {
|
||||
for ij in 0..params.nfreqc {
|
||||
absoc[ij] -= ably;
|
||||
emisc[ij] -= emly;
|
||||
scatc[ij] -= scly;
|
||||
}
|
||||
}
|
||||
|
||||
OpacwResult { absoc, emisc, scatc, abso, emis, scat }
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_opacw_basic() {
|
||||
let temp = [10000.0];
|
||||
let elec = [1e13];
|
||||
let freqc = [1e14, 2e14, 3e14];
|
||||
let freq = [1e14, 1.5e14, 2e14, 2.5e14, 3e14];
|
||||
let nfirst = [0];
|
||||
let nlast = [0];
|
||||
let nnext = [1];
|
||||
let ifree = [1];
|
||||
let ff = [1e15];
|
||||
let iz = [1];
|
||||
let iel = [0];
|
||||
let ifwop = [1];
|
||||
let popul_lev = [1e12, 1e10];
|
||||
let g = [1.0, 2.0];
|
||||
let enion = [0.0, 1e-11];
|
||||
let wop = [1.0, 1.0];
|
||||
let indexp = [0, 0];
|
||||
let z3 = [1.0];
|
||||
let elec23 = [2.15e-9];
|
||||
let dwc1 = [0.0];
|
||||
let dwc2 = [0.0];
|
||||
let cross_row: &[f64] = &[1e-18, 2e-18, 3e-18];
|
||||
let cross: &[&[f64]] = &[cross_row];
|
||||
let frx1 = [0.0, 0.5, 1.0, 0.5, 0.0];
|
||||
let ijcint = [0, 0, 1, 1, 2];
|
||||
|
||||
let params = OpacwParams {
|
||||
id: 0,
|
||||
temp: &temp,
|
||||
elec: &elec,
|
||||
freqc: &freqc,
|
||||
nfreqc: 3,
|
||||
freq: &freq,
|
||||
nfreq: 5,
|
||||
nion: 1,
|
||||
nfirst: &nfirst,
|
||||
nlast: &nlast,
|
||||
nnext: &nnext,
|
||||
ifree: &ifree,
|
||||
ff: &ff,
|
||||
ielhm: 99,
|
||||
iz: &iz,
|
||||
iel: &iel,
|
||||
ifwop: &ifwop,
|
||||
popul_lev: &popul_lev,
|
||||
g: &g,
|
||||
enion: &enion,
|
||||
wop: &wop,
|
||||
indexp: &indexp,
|
||||
z3: &z3,
|
||||
elec23: &elec23,
|
||||
dwc1: &dwc1,
|
||||
dwc2: &dwc2,
|
||||
ndepth: 1,
|
||||
bergfc: 1.0,
|
||||
bn: 1.0,
|
||||
hk: 4.79928144e-11,
|
||||
sige: 6.65e-25,
|
||||
cross,
|
||||
imode: 0,
|
||||
idstd: 0,
|
||||
icont: 0,
|
||||
iophli: 0,
|
||||
iath: 0,
|
||||
frx1: &frx1,
|
||||
ijcint: &ijcint,
|
||||
wn_hint: 0.0,
|
||||
pop_h: 1e12,
|
||||
pop_h_cont: 1e10,
|
||||
sgmerg_fn: None,
|
||||
opadd_fn: None,
|
||||
lymlin_fn: None,
|
||||
linopw_fn: None,
|
||||
molop_fn: None,
|
||||
hydliw_fn: None,
|
||||
he2liw_fn: None,
|
||||
phtion_fn: None,
|
||||
phtx_fn: None,
|
||||
};
|
||||
|
||||
let result = opacw(¶ms);
|
||||
assert_eq!(result.absoc.len(), 3);
|
||||
assert_eq!(result.emisc.len(), 3);
|
||||
assert_eq!(result.scatc.len(), 3);
|
||||
assert_eq!(result.abso.len(), 5);
|
||||
assert_eq!(result.emis.len(), 5);
|
||||
assert_eq!(result.scat.len(), 5);
|
||||
assert!(result.absoc[0].is_finite());
|
||||
assert!(result.abso[0].is_finite());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_opacw_skip_imode() {
|
||||
let temp = [10000.0];
|
||||
let elec = [1e13];
|
||||
let freqc = [1e14];
|
||||
let freq = [1e14];
|
||||
let nfirst = [0];
|
||||
let nlast = [0];
|
||||
let nnext = [1];
|
||||
let ifree = [0];
|
||||
let ff = [0.0];
|
||||
let iz = [1];
|
||||
let iel = [0];
|
||||
let ifwop = [1];
|
||||
let popul_lev = [1e12];
|
||||
let g = [1.0];
|
||||
let enion = [0.0];
|
||||
let wop = [1.0];
|
||||
let indexp = [0];
|
||||
let z3 = [1.0];
|
||||
let elec23 = [2.15e-9];
|
||||
let dwc1 = [0.0];
|
||||
let dwc2 = [0.0];
|
||||
let cross_row: &[f64] = &[1e-18];
|
||||
let cross: &[&[f64]] = &[cross_row];
|
||||
let frx1 = [0.0];
|
||||
let ijcint = [0];
|
||||
|
||||
let params = OpacwParams {
|
||||
id: 0,
|
||||
temp: &temp,
|
||||
elec: &elec,
|
||||
freqc: &freqc,
|
||||
nfreqc: 1,
|
||||
freq: &freq,
|
||||
nfreq: 1,
|
||||
nion: 1,
|
||||
nfirst: &nfirst,
|
||||
nlast: &nlast,
|
||||
nnext: &nnext,
|
||||
ifree: &ifree,
|
||||
ff: &ff,
|
||||
ielhm: 99,
|
||||
iz: &iz,
|
||||
iel: &iel,
|
||||
ifwop: &ifwop,
|
||||
popul_lev: &popul_lev,
|
||||
g: &g,
|
||||
enion: &enion,
|
||||
wop: &wop,
|
||||
indexp: &indexp,
|
||||
z3: &z3,
|
||||
elec23: &elec23,
|
||||
dwc1: &dwc1,
|
||||
dwc2: &dwc2,
|
||||
ndepth: 1,
|
||||
bergfc: 1.0,
|
||||
bn: 1.0,
|
||||
hk: 4.79928144e-11,
|
||||
sige: 6.65e-25,
|
||||
cross,
|
||||
imode: -1,
|
||||
idstd: 1, // different from id=0, so should skip
|
||||
icont: 0,
|
||||
iophli: 0,
|
||||
iath: 0,
|
||||
frx1: &frx1,
|
||||
ijcint: &ijcint,
|
||||
wn_hint: 0.0,
|
||||
pop_h: 1e12,
|
||||
pop_h_cont: 1e10,
|
||||
sgmerg_fn: None,
|
||||
opadd_fn: None,
|
||||
lymlin_fn: None,
|
||||
linopw_fn: None,
|
||||
molop_fn: None,
|
||||
hydliw_fn: None,
|
||||
he2liw_fn: None,
|
||||
phtion_fn: None,
|
||||
phtx_fn: None,
|
||||
};
|
||||
|
||||
let result = opacw(¶ms);
|
||||
// Should return zeros since imode==-1 and id!=idstd
|
||||
assert_eq!(result.absoc[0], 0.0);
|
||||
assert_eq!(result.abso[0], 0.0);
|
||||
}
|
||||
}
|
||||
296
src/synspec/math/opadd.rs
Normal file
296
src/synspec/math/opadd.rs
Normal file
@ -0,0 +1,296 @@
|
||||
//! Additional opacities for SYNSPEC.
|
||||
//!
|
||||
//! Translated from SYNSPEC `OPADD` subroutine (synspec54.f:11663).
|
||||
//!
|
||||
//! Includes Rayleigh scattering, H⁻ opacity, H₂⁺ opacity,
|
||||
//! He⁻ free-free, and CIA (Collision-Induced Absorption) opacities.
|
||||
|
||||
use crate::tlusty::math::hydrogen::{sbfhmi, h2minus, sbfch, sbfoh, sffhmi};
|
||||
use crate::tlusty::math::opacity::{cia_h2h2, cia_h2he, cia_h2h, cia_hhe};
|
||||
|
||||
/// Physical constants
|
||||
const FRAYH: f64 = 2.463e15; // H Rayleigh limit
|
||||
const FRAYHE: f64 = 5.150e15; // He Rayleigh limit
|
||||
const FRAYH2: f64 = 2.922e15; // H2 Rayleigh limit
|
||||
const CLS: f64 = 2.997925e18; // Speed of light (Å/s)
|
||||
|
||||
/// Parameters for additional opacity calculation
|
||||
pub struct OpaddParams {
|
||||
/// Mode: -1 for initialization, 0 for opacity calculation
|
||||
pub mode: i32,
|
||||
/// Depth index
|
||||
pub id: usize,
|
||||
/// Frequency (Hz)
|
||||
pub fr: f64,
|
||||
/// Temperature (K)
|
||||
pub t: f64,
|
||||
/// Electron density
|
||||
pub ane: f64,
|
||||
/// H/kT
|
||||
pub hkt: f64,
|
||||
/// 1/(T*sqrt(T))
|
||||
pub t32: f64,
|
||||
/// H number density
|
||||
pub anh: f64,
|
||||
/// He number density
|
||||
pub anhe: f64,
|
||||
/// H2 number density
|
||||
pub anh2: f64,
|
||||
/// CH number density
|
||||
pub anch: f64,
|
||||
/// OH number density
|
||||
pub anoh: f64,
|
||||
/// H ground level population
|
||||
pub pop_h: f64,
|
||||
/// H⁻ excited level population (continuum)
|
||||
pub pop_h_cont: f64,
|
||||
/// Boltzmann constant * temperature
|
||||
pub bn: f64,
|
||||
/// h/k
|
||||
pub hk: f64,
|
||||
|
||||
// Switches
|
||||
pub irsct: i32, // Rayleigh scattering on H I
|
||||
pub irsche: i32, // Rayleigh scattering on He I
|
||||
pub irsch2: i32, // Rayleigh scattering on H2
|
||||
pub iophmi: i32, // H- opacity
|
||||
pub ioph2p: i32, // H2+ opacity
|
||||
pub iophem: i32, // He- opacity
|
||||
pub ioph2m: i32, // H2- opacity
|
||||
pub iopch: i32, // CH opacity
|
||||
pub iopoh: i32, // OH opacity
|
||||
pub ioh2h2: i32, // CIA H2-H2
|
||||
pub ioh2he: i32, // CIA H2-He
|
||||
pub ioh2h1: i32, // CIA H2-H
|
||||
pub iohhe: i32, // CIA H-He
|
||||
pub iophli: i32, // Lyman line wings
|
||||
pub ifmol: i32, // Molecular lines
|
||||
pub tmolim: f64, // Molecular temperature limit
|
||||
}
|
||||
|
||||
/// Result of additional opacity calculation
|
||||
pub struct OpaddResult {
|
||||
/// Absorption coefficient
|
||||
pub abad: f64,
|
||||
/// Emission coefficient
|
||||
pub emad: f64,
|
||||
/// Scattering coefficient
|
||||
pub scad: f64,
|
||||
}
|
||||
|
||||
/// Calculate additional opacities.
|
||||
///
|
||||
/// This is the SYNSPEC version of OPADD, which handles various
|
||||
/// non-standard opacity sources including:
|
||||
/// - Rayleigh scattering (H, He, H2)
|
||||
/// - H⁻ bound-free and free-free
|
||||
/// - H₂⁺ bound-free and free-free
|
||||
/// - He⁻ free-free
|
||||
/// - H₂⁻ free-free
|
||||
/// - CH and OH continuum opacity
|
||||
/// - CIA (Collision-Induced Absorption)
|
||||
///
|
||||
/// # Arguments
|
||||
/// * `params` - Input parameters and switches
|
||||
/// * `cia_h2h2_data` - CIA H2-H2 data
|
||||
/// * `cia_h2he_data` - CIA H2-He data
|
||||
/// * `cia_h2h_data` - CIA H2-H data
|
||||
/// * `cia_hhe_data` - CIA H-He data
|
||||
///
|
||||
/// # Returns
|
||||
/// Additional opacity coefficients (absorption, emission, scattering)
|
||||
pub fn opadd(
|
||||
params: &OpaddParams,
|
||||
cia_h2h2_data: &crate::tlusty::math::opacity::CiaH2h2Data,
|
||||
cia_h2he_data: &crate::tlusty::math::opacity::CiaH2heData,
|
||||
cia_h2h_data: &crate::tlusty::math::opacity::CiaH2hData,
|
||||
cia_hhe_data: &crate::tlusty::math::opacity::CiaHheData,
|
||||
) -> OpaddResult {
|
||||
let mut ab0 = 0.0;
|
||||
let mut ab1 = 0.0;
|
||||
let mut abad = 0.0;
|
||||
let mut emad = 0.0;
|
||||
let mut scad = 0.0;
|
||||
|
||||
let mode = params.mode;
|
||||
let fr = params.fr;
|
||||
let t = params.t;
|
||||
let ane = params.ane;
|
||||
let hkt = params.hkt;
|
||||
let t32 = params.t32;
|
||||
let anh = params.anh;
|
||||
let anhe = params.anhe;
|
||||
let anh2 = params.anh2;
|
||||
let pop_h = params.pop_h;
|
||||
let bn = params.bn;
|
||||
let _hk = params.hk;
|
||||
|
||||
// HI Rayleigh scattering
|
||||
if params.irsct != 0 && params.iophli != 1 && params.iophli != 2 {
|
||||
let x = 1.0 / (CLS / fr.min(FRAYH)).powi(2);
|
||||
let sg = (5.799e-13 + (1.422e-6 + 2.784 * x) * x) * x * x;
|
||||
scad = anh * sg;
|
||||
}
|
||||
|
||||
// H⁻ bound-free and free-free
|
||||
if params.iophmi != 0 {
|
||||
let sg_bf = sbfhmi(fr);
|
||||
let xhm = 8762.9 / t;
|
||||
let sb = 1.0353e-16 * t32 * xhm.exp() * pop_h * ane * sg_bf;
|
||||
let sf = sffhmi(pop_h, fr, t) * ane;
|
||||
ab0 = sb + sf;
|
||||
}
|
||||
|
||||
// He I Rayleigh scattering
|
||||
if params.irsche != 0 && mode >= 0 {
|
||||
let x = (CLS / fr.min(FRAYHE)).powi(2);
|
||||
let cs = 5.484e-14 / x / x * (1.0 + (2.44e5 + 5.94e10 / (x - 2.90e5)) / x).powi(2);
|
||||
let sg = anhe * cs;
|
||||
scad += sg;
|
||||
}
|
||||
|
||||
// H2 Rayleigh scattering
|
||||
if params.irsch2 != 0 && mode >= 0 && params.ifmol > 0 {
|
||||
let x = (CLS / fr.min(FRAYH2)).powi(2);
|
||||
let x2 = 1.0 / x / x;
|
||||
let cs = (8.14e-13 + 1.28e-6 / x + 1.61 * x2) * x2;
|
||||
let sg = cs * anh2;
|
||||
scad += sg;
|
||||
}
|
||||
|
||||
// H₂⁺ bound-free and free-free
|
||||
if params.ioph2p > 0 && params.ifmol > 0 && t < params.tmolim && fr < 3.28e15 {
|
||||
let x = fr * 1.0e-15;
|
||||
let sg1 = (-7.342e-3
|
||||
+ (-2.409 + (1.028 + (-4.23e-1 + (1.224e-1 - 1.351e-2 * x) * x) * x) * x) * x)
|
||||
* 1.602e-12
|
||||
/ 1.380658e-16;
|
||||
let x = fr.ln();
|
||||
let sg2 = -3.0233e3
|
||||
+ (3.7797e2 + (-1.82496e1 + (3.9207e-1 - 3.1672e-3 * x) * x) * x) * x;
|
||||
let x2 = -sg1 / t + sg2;
|
||||
let mut sb = 0.0;
|
||||
if x2 > -150.0 {
|
||||
sb = pop_h * params.pop_h_cont * x2.exp();
|
||||
}
|
||||
ab0 += sb;
|
||||
}
|
||||
|
||||
// He⁻ free-free
|
||||
if mode >= 0 && params.iophem > 0 {
|
||||
let a = 3.397e-46 + (-5.216e-31 + 7.039e-15 / fr) / fr;
|
||||
let b = -4.116e-42 + (1.067e-26 + 8.135e-11 / fr) / fr;
|
||||
let c = 5.081e-37 + (-8.724e-23 - 5.659e-8 / fr) / fr;
|
||||
let cs = a * t + b + c / t;
|
||||
let sg = anhe * ane * cs;
|
||||
ab0 += sg;
|
||||
}
|
||||
|
||||
// H₂⁻ free-free
|
||||
if params.ioph2m != 0 && mode >= 0 && params.ifmol > 0 && t < params.tmolim {
|
||||
let oph2 = h2minus(t, anh2, ane, fr);
|
||||
ab1 += oph2;
|
||||
}
|
||||
|
||||
// CH and OH continuum opacity
|
||||
if mode >= 0 && params.ifmol > 0 && t < params.tmolim {
|
||||
if params.iopch > 0 {
|
||||
ab0 += sbfch(fr, t) * params.anch;
|
||||
}
|
||||
if params.iopoh > 0 {
|
||||
ab0 += sbfoh(fr, t) * params.anoh;
|
||||
}
|
||||
|
||||
// CIA H2-H2
|
||||
if params.ioh2h2 > 0 {
|
||||
let oph2 = cia_h2h2(t, anh2, fr, cia_h2h2_data);
|
||||
ab1 += oph2;
|
||||
}
|
||||
|
||||
// CIA H2-He
|
||||
if params.ioh2he > 0 {
|
||||
let oph2 = cia_h2he(t, anh2, anhe, fr, cia_h2he_data);
|
||||
ab1 += oph2;
|
||||
}
|
||||
|
||||
// CIA H2-H
|
||||
if params.ioh2h1 > 0 {
|
||||
let oph2 = cia_h2h(t, anh2, anh, fr, cia_h2h_data);
|
||||
ab1 += oph2;
|
||||
}
|
||||
|
||||
// CIA H-He
|
||||
if params.iohhe > 0 {
|
||||
let oph2 = cia_hhe(t, anh, anhe, fr, cia_hhe_data);
|
||||
ab1 += oph2;
|
||||
}
|
||||
}
|
||||
|
||||
// Final absorption and emission coefficients
|
||||
if mode >= 0 {
|
||||
let x = (-hkt * fr).exp();
|
||||
let x1 = 1.0 - x;
|
||||
let bnx = bn * (fr * 1.0e-15).powi(3) * x;
|
||||
abad += x1 * ab0 + ab1;
|
||||
emad += bnx * (ab0 + ab1 / x1);
|
||||
}
|
||||
|
||||
OpaddResult { abad, emad, scad }
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_opadd_basic() {
|
||||
// Create test parameters
|
||||
let params = OpaddParams {
|
||||
mode: 0,
|
||||
id: 0,
|
||||
fr: 5.0e14,
|
||||
t: 6000.0,
|
||||
ane: 1.0e13,
|
||||
hkt: 4.79928144e-11 / 6000.0,
|
||||
t32: 1.0 / 6000.0 / 6000.0_f64.sqrt(),
|
||||
anh: 1.0e16,
|
||||
anhe: 1.0e15,
|
||||
anh2: 1.0e10,
|
||||
anch: 1.0e8,
|
||||
anoh: 1.0e8,
|
||||
pop_h: 1.0e16,
|
||||
pop_h_cont: 1.0e10,
|
||||
bn: 1.0,
|
||||
hk: 4.79928144e-11,
|
||||
irsct: 1,
|
||||
irsche: 1,
|
||||
irsch2: 0,
|
||||
iophmi: 1,
|
||||
ioph2p: 0,
|
||||
iophem: 0,
|
||||
ioph2m: 0,
|
||||
iopch: 0,
|
||||
iopoh: 0,
|
||||
ioh2h2: 0,
|
||||
ioh2he: 0,
|
||||
ioh2h1: 0,
|
||||
iohhe: 0,
|
||||
iophli: 0,
|
||||
ifmol: 0,
|
||||
tmolim: 10000.0,
|
||||
};
|
||||
|
||||
// Create empty CIA data for testing
|
||||
let cia_h2h2_data = crate::tlusty::math::opacity::CiaH2h2Data::default();
|
||||
let cia_h2he_data = crate::tlusty::math::opacity::CiaH2heData::default();
|
||||
let cia_h2h_data = crate::tlusty::math::opacity::CiaH2hData::default();
|
||||
let cia_hhe_data = crate::tlusty::math::opacity::CiaHheData::default();
|
||||
|
||||
let result = opadd(¶ms, &cia_h2h2_data, &cia_h2he_data, &cia_h2h_data, &cia_hhe_data);
|
||||
|
||||
assert!(result.abad.is_finite());
|
||||
assert!(result.emad.is_finite());
|
||||
assert!(result.scad.is_finite());
|
||||
assert!(result.scad > 0.0); // Rayleigh scattering should be positive
|
||||
}
|
||||
}
|
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue
Block a user