"""
Module of functions to return diagram dictionary for LaTeX
"""

import random


def get_coordinates_dict(num=None):
    coord_dict = get_points_dict(num)
    coords_sections_dict = dict()
    coords_sections_dict["points_to_plot_without_xy"] = format_latex_coordinates_for_plotting_without_xy(coord_dict)
    coords_sections_dict["points_to_plot"] = format_latex_coordinates_for_plotting(coord_dict)
    return coords_sections_dict



def get_points_dict(num=None):
    # limit to max of 20
    if num is None or num > 20:
        num = random.randint(5, 20)  # Randomly choose how many points to generate
    coord_labels_list = ["A", "B", "C", "D", "E", "F", "G", "H", "I", "J", "K", "L", "M", "N", "O", "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z"]
    coord_dict = {}
    used_x_values = set()  # Track used x-values
    used_y_values = set()  # Track used y-values
    for i in range(num):
        label = coord_labels_list[i]  # Get the next available letter
        x, y = random_coordinate(used_x_values, used_y_values)  # Ensure unique values
        coord_dict[label] = (x, y)
    return coord_dict


def random_coordinate(used_x_values, used_y_values):
    while True:
        x = random.randint(-10, 10)
        y = random.randint(-10, 10)
        # Check constraints
        if y == -1:  # Avoid y = -1
            continue
        if y % 2 == 0 and (x == -1 or x == -2):  # Avoid x = 1 or -2 when y is even
            continue
        if y in used_y_values:  # Avoid reused y-values
            continue
        if x in used_x_values:  # Avoid reused x-values
            continue

        # If valid, mark x and y as used
        used_x_values.add(x)
        used_y_values.add(y)
        break
    return x, y


def format_latex_coordinates_for_plotting_without_xy(coord_dict):
    latex_str = ""
    for name, (x, y) in coord_dict.items():
        latex_str += f"\\fill ({x},{y}) circle (4pt);\n"
        latex_str += f"\\node[xshift=1.9em, yshift=7pt] at ({x},{y}) {{\\small {name}(\\hspace{{0.6cm}},\\hspace{{0.6cm}})}};\n"
    return latex_str




def format_latex_coordinates_for_plotting(coord_dict):
    latex_str = ""
    for name, (x, y) in coord_dict.items():
        latex_str += f"\\fill ({x},{y}) circle (4pt);\n"
        latex_str += f"\\node[xshift=1.9em, yshift=7pt] at ({x},{y}) {{\\small {name}({x},{y})}};\n"
    return latex_str

# print(get_coordinates_dict(20)["points_to_plot"])