"""Command line interface for Copaint pdf generator.
"""
import argparse
import numpy as np
import logging

from copaint.copaint import image_to_pdf

# Configure logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(levelname)s - %(message)s',
    datefmt='%Y-%m-%d %H:%M:%S'
)
logger = logging.getLogger(__name__)

# Default identifiers for unique_identifier parameter
default_identifiers = [
    "Mauricette", "Gertrude", "Bernadette", "Henriette", "Georgette", "Antoinette", "Colette", "Suzette", "Yvette", "Paulette",
    "Juliette", "Odette", "Marinette", "Lucette", "Pierrette", "Cosette", "Rosette", "Claudette", "Violette", "Josette"
]

def default_identifier():
    """Return a random default identifier."""
    return np.random.choice(default_identifiers)

def get_grid_size(nparticipants, input_image):
    """Get grid size based on number of participants."""
    # Compute the grid size based on the number of participants
    # We want to find h_cells and w_cells such that h_cells * w_cells >= nparticipants
    # and h_cells/w_cells is close to the aspect ratio of the image
    from PIL import Image
    image = Image.open(input_image)
    w, h = image.size
    aspect_ratio = w/h
    
    # Find the smallest grid that can accommodate nparticipants
    min_cells = int(np.ceil(np.sqrt(nparticipants)))
    for i in range(min_cells, min_cells+10):
        for j in range(min_cells, min_cells+10):
            if i*j >= nparticipants:
                ratio = j/i
                if abs(ratio - aspect_ratio) < 0.2:
                    logger.info(f"Found grid size {i}x{j} for {nparticipants} participants")
                    return i, j
    
    # If no good ratio is found, just use the smallest grid that can accommodate nparticipants
    h_cells = min_cells
    w_cells = int(np.ceil(nparticipants/min_cells))
    logger.warning(f"No good aspect ratio found. Using {h_cells}x{w_cells} grid for {nparticipants} participants")
    return h_cells, w_cells

def main():
    parser = argparse.ArgumentParser(description='CoPaint')
    parser.add_argument('--input_image', type=str, default='./data/bear.png', help='input image')
    parser.add_argument('--copaint_logo', type=str, default='./data/logo_copaint.png', help='copaint logo')
    parser.add_argument('--outputfolder', type=str, default='output/', help='output image')
    parser.add_argument('--nparticipants', type=int, help='number of participants', default=None)
    parser.add_argument('--h_cells', type=int, help='number of cells in height', default=None)
    parser.add_argument('--w_cells', type=int, help='number of cells in width', default=None)
    parser.add_argument('--unique_identifier', type=str, help='unique identifier like Mauricette, in case users are doing multiple copaint at the same time, to avoid mixing the tiles', default=None)
    parser.add_argument('--cell_size_in_cm', type=float,  default=None, help='size of a cell in cm, for printing purposes')
    parser.add_argument('--use_presets', action='store_true', help='use a couple of presets cuts based on the size of the number of people attending')
    parser.add_argument('--a4', action='store_true',  help='use A4 format for the pdf')
    parser.add_argument('--high_res', action='store_true',  help='save in high resolution mode (PNG format without resizing)')
    parser.add_argument('--min_cell_size_in_cm', type=int, default=2, help='minimum size of cells in cm')
    parser.add_argument('--debug', action='store_true',  help='debug mode')
    
    args = parser.parse_args()

    if args.debug:
        logger.setLevel(logging.DEBUG)
    
    if args.unique_identifier is None:
        # select one at random
        idx = np.random.choice(len(default_identifiers), 1)
        args.unique_identifier = default_identifiers[idx[0]]
        logger.info(f"Using random identifier: {args.unique_identifier}")
    
    # generate a copaint pdf based on the number of participants
    if args.nparticipants:
        # assert other parameters are not set
        assert(args.h_cells is None), "When choosing via number of participants, the number of H cells can't be set"
        assert(args.w_cells is None ), "When choosing via number of participants, the number of W cells can't be set"
        
        # get the grid size based on the number of participants
        h_cells, w_cells = get_grid_size(args.nparticipants, args.input_image)
        image_to_pdf(args.input_image, args.copaint_logo, args.outputfolder, h_cells, w_cells, 
                            unique_identifier=args.unique_identifier, cell_size_in_cm=args.cell_size_in_cm, 
                            a4=args.a4, high_res=args.high_res, min_cell_size_in_cm=args.min_cell_size_in_cm, debug=args.debug)
    
    # Generate the copaint pdf using the specified number of cells
    else:        
        image_to_pdf(args.input_image, args.copaint_logo, args.outputfolder, args.h_cells, args.w_cells, 
                            unique_identifier=args.unique_identifier, cell_size_in_cm=args.cell_size_in_cm, 
                            a4=args.a4, high_res=args.high_res, min_cell_size_in_cm=args.min_cell_size_in_cm, debug=args.debug)

if __name__ == '__main__':
    main()