Initial commit: Speckle-Scanner 3D pipeline with setup README

2026-06-10 03:09:05 +05:00
commit 1765934846
375 changed files with 123081 additions and 0 deletions
@@ -0,0 +1,93 @@
+"""Shared CLI helpers for calibration scripts."""
+
+from __future__ import annotations
+
+import argparse
+from pathlib import Path
+from typing import Dict, Optional, Tuple
+
+import config
+
+
+def parse_chessboard_size(value: str) -> Tuple[int, int]:
+    parts = value.split(",")
+    if len(parts) != 2:
+        raise argparse.ArgumentTypeError(
+            "chessboard size must be width,height (e.g. 8,7)"
+        )
+    return tuple(map(int, parts))
+
+
+def add_session_args(parser: argparse.ArgumentParser) -> None:
+    parser.add_argument("--project", required=True, help="Project name (e.g. Olsen_wings)")
+    parser.add_argument("--date", required=True, help="Date string (e.g. 2026-05-12)")
+    parser.add_argument(
+        "--calib_name", default="calib1", help="Calibration folder name (default: calib1)"
+    )
+
+
+def add_board_args(parser: argparse.ArgumentParser) -> None:
+    parser.add_argument(
+        "--chessboard_size",
+        type=parse_chessboard_size,
+        default="8,7",
+        help="Default inner corner grid width,height",
+    )
+    parser.add_argument(
+        "--square_size",
+        type=float,
+        default=0.045,
+        help="Default chessboard square size in metres",
+    )
+    parser.add_argument("--left_chessboard_size", type=parse_chessboard_size, default=None)
+    parser.add_argument("--right_chessboard_size", type=parse_chessboard_size, default=None)
+    parser.add_argument("--left_square_size", type=float, default=None)
+    parser.add_argument("--right_square_size", type=float, default=None)
+    parser.add_argument(
+        "--preprocessing",
+        type=str,
+        default="None",
+        help="Pre-detection chain: G=gray, C=CLAHE, T=threshold (e.g. C, GC)",
+    )
+
+
+def add_troubleshooting_arg(parser: argparse.ArgumentParser) -> None:
+    parser.add_argument(
+        "--troubleshooting",
+        action="store_true",
+        help=(
+            "Verbose logs and intermediate debug files (corners/, pairing_reports/, "
+            "rectified/). Default: minimal logs; step 2 writes only params/"
+        ),
+    )
+
+
+def resolve_input_path(args) -> Path:
+    return config.CALIB_DATA_DIR / args.project / args.date / args.calib_name
+
+
+def build_board_config(args) -> Tuple[Dict[str, Tuple[int, int]], Dict[str, float]]:
+    default_board = args.chessboard_size
+    default_square = args.square_size
+    left_board = args.left_chessboard_size or default_board
+    right_board = args.right_chessboard_size or default_board
+    left_square = args.left_square_size if args.left_square_size is not None else default_square
+    right_square = (
+        args.right_square_size if args.right_square_size is not None else default_square
+    )
+
+    board_sizes = {
+        "lc": left_board,
+        "lc-ir": left_board,
+        "rc": right_board,
+        "rg": right_board,
+        "ir": right_board,
+    }
+    square_sizes = {
+        "lc": left_square,
+        "lc-ir": left_square,
+        "rc": right_square,
+        "rg": right_square,
+        "ir": right_square,
+    }
+    return board_sizes, square_sizes