import copy
import dataclasses
import json

import cv2
import numpy as np

import shenzhen_solitaire.card_detection.adjustment as adjustment
import shenzhen_solitaire.card_detection.card_finder as card_finder
from shenzhen_solitaire.card_detection.configuration import Configuration
import argparse


def main() -> None:
    """Generate a configuration"""

    parser = argparse.ArgumentParser(description='Process some integers.')
    parser.add_argument('screenshot_path', metavar='screenshot_path', type=str,
                    help='Path to the screenshot')

    args = parser.parse_args()
    print(args.screenshot_path)
    image = cv2.imread(args.screenshot_path)

    border_adjustment = adjustment.adjust_squares(image, count_x=8, count_y=13)
    border_square_pos = adjustment.adjust_squares(
        image, count_x=1, count_y=1, adjustment=copy.deepcopy(border_adjustment)
    )
    border_square = card_finder.get_field_squares(image, border_square_pos, 1, 1)
    empty_square_pos = adjustment.adjust_squares(
        image, count_x=1, count_y=1, adjustment=copy.deepcopy(border_adjustment)
    )
    empty_square = card_finder.get_field_squares(image, empty_square_pos, 1, 1)

    cv2.imwrite("/tmp/border_square.png", border_square[0])
    cv2.imwrite("/tmp/empty_square.png", empty_square[0])
    print(json.dumps(dataclasses.asdict(border_adjustment)))


if __name__ == "__main__":
    main()