Source code for mastermind.utils.get_feedback

"""
This module defines the generate_feedback function, which takes a guess and a secret code, and returns the number of black and white pegs.
"""




def generate_feedback(guess: tuple, secret: tuple, number_of_colors: int) -> list:
    """
    Generates feedback based on the guess and the secret code.

    Args:
        guess (tuple): The player's guess.
        secret (tuple): The secret code.
        number_of_colors (int): The number of colors in the game.

    Returns:
        list: A list containing the number of black and white pegs.
    """
    # Optimized abstract algorithm (assuming correct input)
    list1 = [0] * (number_of_colors + 1)  # black pegs + color count of guess
    list2 = [0] * (number_of_colors + 1)  # white pegs + color count of secret

    # Count colors in guess and secret code
    for dot1, dot2 in zip(guess, secret):
        if dot1 == dot2:  # if exact match found
            list1[0] += 1  # black pegs count += 1
        else:  # otherwise increment the color count to find potential white pegs
            list1[dot1] += 1  # count of color in guess += 1
            list2[dot2] += 1  # count of color in secret += 1

    # Iterate through color count (skip pegs count) to count white pegs
    for count1, count2 in zip(list1[1:], list2[1:]):
        list2[0] += min(count1, count2)  # list2[0] is white pegs count

    return list1[0], list2[0]  # return black and white pegs count