import json
import argparse
from datetime import datetime


def now_str():
    return datetime.now().strftime("%Y-%m-%d %H:%M:%S")


def load_json(path):
    with open(path, "r", encoding="utf-8") as f:
        return json.load(f)


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--camera_json", default="calibration/sensor_calibration.json")
    parser.add_argument("--fusion_json", default="calibration/manual_offsets.json")
    parser.add_argument("--out", default="calibration/module_params.json")
    args = parser.parse_args()

    cam_data = load_json(args.camera_json)
    fusion_data = load_json(args.fusion_json)

    camera_settings = cam_data.get("camera_settings")
    if not isinstance(camera_settings, dict):
        raise RuntimeError("camera_json sem camera_settings válido")

    fusion_config = {
        "alignment_mode": fusion_data.get("alignment_mode", "manual_affine"),
        "baseline_mm": fusion_data.get("baseline_mm", 75.0),
        "reference_camera": fusion_data.get("reference_camera", "cam2"),
        "manual_offsets": fusion_data.get("manual_offsets", {}),
        "homographies": fusion_data.get("homographies", {}),
        "crop_valid_common": fusion_data.get("crop_valid_common", True),
        "resize_after_crop": fusion_data.get("resize_after_crop", True),
        "target_size": fusion_data.get("target_size", None),
    }

    module_params = {
        "schema": "multispec_module_params_v1",
        "saved_at": now_str(),

        "frame_type": cam_data.get("frame_type", fusion_data.get("frame_type", "RAW_BRUTO")),
        "capture_mode_requested": cam_data.get("capture_mode_requested", "AUTO"),
        "capture_mode_effective": cam_data.get("capture_mode_effective", "AUTO"),
        "raw_policy": cam_data.get("raw_policy", "allow_single"),

        "sensor_width": cam_data.get("sensor_width", fusion_data.get("sensor_width")),
        "sensor_height": cam_data.get("sensor_height", fusion_data.get("sensor_height")),
        "bayer_pattern": cam_data.get("bayer_pattern", fusion_data.get("bayer_pattern", "GBRG")),

        "camera_settings": camera_settings,
        "fusion_config": fusion_config,
    }

    with open(args.out, "w", encoding="utf-8") as f:
        json.dump(module_params, f, ensure_ascii=False, indent=2)

    print(f"[OK] module_params gerado em: {args.out}")


if __name__ == "__main__":
    main()