| | 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) |
