#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Seal Text Unwarping with Tunable Parameters
Generates comparison of different unwarping schemes (Polar to Cartesian)
"""
import sys
import cv2
import numpy as np
import json
import os

def unwarp_seal(image_path, output_path):
    print(f"Loading {image_path}...", file=sys.stderr)
    img = cv2.imread(image_path)
    if img is None:
        print(f"Error: Could not load {image_path}", file=sys.stderr)
        return
        
    h, w = img.shape[:2]
    cx, cy = w // 2, h // 2
    radius_base = min(cx, cy)
    
    # Debug Grid: Rotation (Rows) x Variation (Cols)
    angles = [0, 90, 180, 270]
    flips = [
        ("Base", None),
        ("FlipV", 0),
        ("FlipH", 1),
        ("FlipVH", -1)
    ]
    
    # We'll use a fixed factor and r_scale for the grid
    factor = 2.0
    r_scale = 1.0
    
    rows = []
    for angle in angles:
        M = cv2.getRotationMatrix2D((cx, cy), angle, 1.0)
        rotated_img = cv2.warpAffine(img, M, (w, h), borderValue=(255, 255, 255))
        
        row_imgs = []
        for label, flip_code in flips:
            out_w = int(radius_base * 2 * np.pi * factor)
            out_h = radius_base
            
            unwarped = cv2.warpPolar(rotated_img, (out_w, out_h), (cx, cy), radius_base, cv2.INTER_LINEAR + cv2.WARP_FILL_OUTLIERS)
            
            # Show both outer and inner just in case text is inside
            # Outer 40%
            strip = unwarped[int(out_h * 0.6):out_h, :]
            
            if flip_code is not None:
                strip = cv2.flip(strip, flip_code)
                
            # Resize for grid
            disp_h = 150
            disp_w = 800
            strip_disp = cv2.resize(strip, (disp_w, disp_h))
            
            # Add label
            cv2.putText(strip_disp, f"Ang:{angle} {label}", (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1.0, (0, 0, 255), 2)
            row_imgs.append(strip_disp)
            
        rows.append(np.hstack(row_imgs))
        
    grid = np.vstack(rows)
    cv2.imwrite(output_path, grid)
    print(f"Saved debug grid to {output_path}")

if __name__ == "__main__":
    if len(sys.argv) < 3:
        print("Usage: python unwarp_seal.py <input> <output>")
        sys.exit(1)
    unwarp_seal(sys.argv[1], sys.argv[2])