|
|
import re |
|
|
from typing import Tuple |
|
|
|
|
|
def parse_camera_params( |
|
|
text: str, |
|
|
mode: str = "base" |
|
|
) -> Tuple[float, float, float]: |
|
|
""" |
|
|
Extract roll, pitch, fov from text using one of two patterns: |
|
|
- 'base' mode: ... are: roll, pitch, fov. |
|
|
- 'cot' mode: <answer>roll, pitch, fov</answer> |
|
|
|
|
|
Args: |
|
|
text: The full text to search. |
|
|
mode: One of {"base", "cot"}. |
|
|
|
|
|
Returns: |
|
|
roll, pitch, fov as floats. |
|
|
|
|
|
Raises: |
|
|
ValueError if the chosen pattern is not found, or mode is invalid. |
|
|
""" |
|
|
|
|
|
pat_base = re.compile( |
|
|
r"are:\s*([+-]?\d+(?:\.\d+)?)\s*,\s*" |
|
|
r"([+-]?\d+(?:\.\d+)?)\s*,\s*" |
|
|
r"([+-]?\d+(?:\.\d+)?)[\.\s]*$" |
|
|
) |
|
|
pat_cot = re.compile( |
|
|
r"<answer>\s*([+-]?\d+(?:\.\d+)?)\s*,\s*" |
|
|
r"([+-]?\d+(?:\.\d+)?)\s*,\s*" |
|
|
r"([+-]?\d+(?:\.\d+)?)\s*</answer>" |
|
|
) |
|
|
|
|
|
m = None |
|
|
if mode == "base": |
|
|
m = pat_base.search(text) |
|
|
elif mode == "cot": |
|
|
m = pat_cot.search(text) |
|
|
else: |
|
|
raise ValueError(f"Invalid mode: {mode!r}. Choose 'base', 'cot', or 'auto'.") |
|
|
|
|
|
if not m: |
|
|
raise ValueError(f"No camera parameters found using mode '{mode}'.") |
|
|
|
|
|
roll_s, pitch_s, fov_s = m.group(1), m.group(2), m.group(3) |
|
|
return float(roll_s), float(pitch_s), float(fov_s) |
