Fits small cameras

2026-07-29 16:25:16 +08:00
1 parent a773166c91
commit 98ed2576c6
@@ -0,0 +1,138 @@
+#!/usr/bin/env bash
+set -Eeuo pipefail
+
+ROOT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd)"
+CONTAINER_NAME="${SCAN_CALIBRATOR_CONTAINER:-smart-cam}"
+SCAN_STATUS_URL="http://127.0.0.1:18080/api/v1/scan/status"
+HELPER_SOURCE="$ROOT_DIR/src/measure_overlap.cpp"
+HELPER_CONTAINER_SOURCE="/tmp/scan-calibrator-measure.cpp"
+HELPER_CONTAINER_BIN="/tmp/scan-calibrator-measure"
+
+info() {
+    printf '[检查] %s\n' "$*"
+}
+
+ok() {
+    printf '[通过] %s\n' "$*"
+}
+
+fail() {
+    printf '[失败] %s\n' "$*" >&2
+    exit 1
+}
+
+install_host_packages() {
+    local -a packages=("$@")
+    local -a elevate=()
+
+    ((${#packages[@]} > 0)) || return 0
+    command -v apt-get >/dev/null 2>&1 ||
+        fail "缺少 ${packages[*]}，且系统没有apt-get，需先按当前发行版手动安装"
+
+    if ((EUID != 0)); then
+        command -v sudo >/dev/null 2>&1 ||
+            fail "安装 ${packages[*]} 需要root权限；请用root运行本脚本"
+        elevate=(sudo)
+    fi
+
+    info "安装宿主机依赖：${packages[*]}"
+    "${elevate[@]}" apt-get update
+    DEBIAN_FRONTEND=noninteractive "${elevate[@]}" apt-get install -y --no-install-recommends \
+        "${packages[@]}"
+}
+
+main() {
+    printf 'scan-calibrator 安装与运行环境检查\n\n'
+
+    local -a missing_packages=()
+    command -v python3 >/dev/null 2>&1 || missing_packages+=(python3)
+    if ! command -v identify >/dev/null 2>&1 ||
+       ! command -v montage >/dev/null 2>&1; then
+        missing_packages+=(imagemagick)
+    fi
+    install_host_packages "${missing_packages[@]}"
+
+    info "检查宿主机命令"
+    command -v bash >/dev/null 2>&1 || fail "未找到bash"
+    command -v python3 >/dev/null 2>&1 || fail "未找到python3"
+    command -v identify >/dev/null 2>&1 || fail "未找到ImageMagick identify"
+    command -v montage >/dev/null 2>&1 || fail "未找到ImageMagick montage"
+    command -v docker >/dev/null 2>&1 || fail "未找到docker；请先部署smart-cam运行环境"
+    docker info >/dev/null 2>&1 ||
+        fail "无法访问Docker daemon；请用有Docker权限的账号运行"
+    ok "Python、Docker和ImageMagick命令可用"
+
+    info "检查项目文件和Python语法"
+    [[ -f "$HELPER_SOURCE" ]] || fail "缺少 $HELPER_SOURCE"
+    [[ -f "$ROOT_DIR/interactive_calibrator.py" ]] ||
+        fail "缺少interactive_calibrator.py"
+    [[ -f "$ROOT_DIR/scan_calibrator.py" ]] || fail "缺少scan_calibrator.py"
+    [[ -f "$ROOT_DIR/scan-panorama/stitch_scan.py" ]] ||
+        fail "缺少scan-panorama/stitch_scan.py"
+    python3 -B -c '
+import ast
+import pathlib
+import sys
+for name in sys.argv[1:]:
+    path = pathlib.Path(name)
+    ast.parse(path.read_text(encoding="utf-8"), filename=str(path))
+' \
+        "$ROOT_DIR/interactive_calibrator.py" \
+        "$ROOT_DIR/scan_calibrator.py" \
+        "$ROOT_DIR/scan-panorama/stitch_scan.py"
+    chmod +x \
+        "$ROOT_DIR/install.sh" \
+        "$ROOT_DIR/interactive.sh" \
+        "$ROOT_DIR/calibrate.sh" \
+        "$ROOT_DIR/scan-panorama/stitch.sh"
+    mkdir -p "$ROOT_DIR/result" "$ROOT_DIR/scan-panorama/result"
+    [[ -w "$ROOT_DIR/result" && -w "$ROOT_DIR/scan-panorama/result" ]] ||
+        fail "当前账号不能写入result目录"
+    ok "程序文件完整且语法检查通过"
+
+    info "检查smart-cam容器"
+    [[ "$(docker inspect -f '{{.State.Running}}' "$CONTAINER_NAME" 2>/dev/null || true)" == "true" ]] ||
+        fail "容器 $CONTAINER_NAME 未运行"
+    docker exec "$CONTAINER_NAME" sh -lc '
+        command -v curl >/dev/null &&
+        command -v python3 >/dev/null &&
+        command -v g++ >/dev/null &&
+        command -v pkg-config >/dev/null &&
+        pkg-config --exists opencv4 &&
+        test -f /smart-cam/config/config.json &&
+        test -d /smart-cam/data/scan &&
+        test -w /smart-cam/data/scan
+    ' || fail "容器内缺少curl/python3/g++/OpenCV，或scan数据目录不可写"
+    ok "容器内编译环境、配置文件和scan数据目录正常"
+
+    info "编译并运行标定图像分析器自检"
+    docker cp "$HELPER_SOURCE" "$CONTAINER_NAME:$HELPER_CONTAINER_SOURCE" >/dev/null
+    docker exec "$CONTAINER_NAME" sh -lc \
+        "g++ -O2 -std=c++17 '$HELPER_CONTAINER_SOURCE' -o '$HELPER_CONTAINER_BIN' \
+\$(pkg-config --cflags --libs opencv4)"
+    docker exec "$CONTAINER_NAME" test -x "$HELPER_CONTAINER_BIN" ||
+        fail "标定图像分析器编译失败"
+    ok "OpenCV标定图像分析器可以编译"
+
+    info "检查scan HTTP服务"
+    local status
+    status="$(docker exec "$CONTAINER_NAME" curl -fsS --max-time 10 "$SCAN_STATUS_URL")" ||
+        fail "scan HTTP服务不可用：$SCAN_STATUS_URL"
+    python3 -c '
+import json
+import sys
+payload = json.load(sys.stdin)
+if not isinstance(payload.get("scan"), dict):
+    raise SystemExit(1)
+' <<<"$status" || fail "scan状态接口没有返回合法的scan JSON"
+    ok "scan HTTP服务正常"
+
+    info "检查ImageMagick拼图命令"
+    identify -version >/dev/null
+    montage -version >/dev/null
+    ok "ImageMagick identify/montage正常"
+
+    printf '\n全部检查通过。现在可以运行：\n  cd %s\n  ./interactive.sh\n' "$ROOT_DIR"
+}
+
+main "$@"