"""CSC110 Fall 2022 Assignment 1, Part 5: Working with Image Data

Instructions (READ THIS FIRST!)
===============================

Please follow the instructions in the assignment handout to complete this file.

Copyright and Usage Information
===============================

This file is provided solely for the personal and private use of students
taking CSC110 at the University of Toronto St. George campus. All forms of
distribution of this code, whether as given or with any changes, are
expressly prohibited. For more information on copyright for CSC110 materials,
please consult our Course Syllabus.

This file is Copyright (c) 2022 David Liu, Tom Fairgrieve, and Angela Zavaleta Bernuy.
"""
import a1_part4
import a1_helpers


###################################################################################################
# 0. Warmup
###################################################################################################
def warmup_part5() -> list:
    """Read an image file and return its pixels as a list of lists.

    This illustrates the use of the helper function load_image_pixels that we have provided you.
    """
    # 'images/spiderman.png' refers to a file called 'spiderman.jpg' inside the 'images' folder.
    example_image_data = a1_helpers.load_image_pixels('images/spiderman.png')

    # Display the pixels as squares in a pygame window
    a1_helpers.show_colour_rows_pygame(example_image_data)

    # Return the data (helpful for testing in this part of the assignment)
    return example_image_data


###################################################################################################
# 1. Transforming multiple colour rows
###################################################################################################
# This constant represents a 4-by-4 pixel grid. We have provided it to help
# with testing your work for this part.
EXAMPLE_PIXEL_GRID = [
    [[128, 128, 128], [35, 50, 65], [210, 32, 68], [32, 208, 43]],
    [[130, 20, 42], [43, 44, 45], [17, 243, 82], [61, 85, 92]],
    [[201, 23, 23], [23, 23, 23], [42, 180, 19], [16, 58, 29]],
    [[1, 52, 128], [26, 123, 128], [71, 234, 82], [23, 108, 34]]
]


def remove_red_in_image(image_data: list) -> list:
    """Return new image data with the same pixels as image_data, except each colour has its "red" value changed to 0.

    You may ASSUME that:
    - image_data is a non-empty list of colour rows, where each colour row has the same length
    """
    return [a1_part4.remove_red_in_row(i) for i in image_data]


def fade_red_in_image(image_data: list) -> list:
    """Return new image data with the same pixels as image_data, except each colour has its "red" value
    multipled by 0.25 and rounded to the nearest integer.

    You may ASSUME that:
    - image_data is a non-empty list of colour rows, where each colour row has the same length
    """
    return [a1_part4.fade_red_in_row(i) for i in image_data]


def fade_rows_in_image(image_data: list) -> list:
    """Return new image data where each row has been faded in the same process described in a1_part4.fade_row.

    You may ASSUME that:
    - image_data is a non-empty list of colour rows, where each colour row has the same length
    """
    return [a1_part4.fade_row(i) for i in image_data]


def blur_rows_in_image(image_data: list) -> list:
    """Return new image data where each row has been blurred in the same process described in a1_part4.blur_row.

    You may ASSUME that:
    - image_data is a non-empty list of colour rows, where each colour row has the same length
    - the length of the colour rows is >= 2
    """
    return [a1_part4.blur_row(i) for i in image_data]


def crop_rows_in_image(image_data: list, start: int, num_colours: int) -> list:
    """Return new image data where each colour row has been cropped in the same way as described
    in a1_part4.crop_row.

    Notes:
    1. start is the index of the first colour to take from each colour row.
    2. num_colours specifies the number of colours to take from each colour row.

    You may ASSUME that:
    - image_data is a non-empty list of colour rows, where each colour row has the same length
    - start >= 0
    - num_colours >= 0
    - start + num_colours <= len(colour_row[0])
    """
    return [a1_part4.crop_row(i, start, num_colours) for i in image_data]


###################################################################################################
# 2. Generalized cropping
###################################################################################################
def crop_image(image_data: list, start_row: int, start_col: int, crop_height: int, crop_width: int) -> list:
    """Return a cropped version of image_data.

    Notes:
    1. start_row is the index of the first colour row to take from image_data
    2. crop_height is the number of rows to take form image_data
    3. start_col is the index of the first colour to take from each colour row
    4. crop_width is the number of colours to take from each colour row

    You may ASSUME that:
    - every element of image_data is a valid colour row with the same length
    - start_row >= 0
    - start_col >= 0
    - crop_height >= 0
    - crop_width >= 0
    - start_row + crop_height <= len(image_data)
    - start_col + crop_width <= len(image_data[0])
    """
    return [a1_part4.crop_row(x, start_col, crop_width) for x in
            [image_data[i] for i in range(start_row, start_row + crop_height)]]


###################################################################################################
# Image transformation (exploration only, not for credit)
###################################################################################################
def transform_image(input_file_path: str, output_file_path: str) -> None:
    """Perform a series of transformations on the given input file image and save the result to the output_file_path.

    We have provided some sample function calls using the functions from this file. Feel free to try these out
    by uncommenting them and adding your own.

    Here is an example call for this function:

        transform_image('images/horses.png', 'images/new_horses.png')
    """
    original_image_data = a1_helpers.load_image_pixels(input_file_path)

    # You can comment out this line and switch it with one of the lines below
    new_image_data = original_image_data
    # new_image_data = remove_red_in_image(original_image_data)
    # new_image_data = fade_red_in_image(original_image_data)
    # new_image_data = fade_rows_in_image(original_image_data)
    # new_image_data = blur_rows_in_image(original_image_data)
    # new_image_data = crop_rows_in_image(original_image_data, 30, 10)
    # new_image_data = crop_image(original_image_data, 40, 70, 240, 200)

    a1_helpers.save_image(output_file_path, new_image_data)


###################################################################################################
# "Main block" (we'll discuss what this means in a future lecture)
###################################################################################################
if __name__ == '__main__':
    import doctest
    doctest.testmod(verbose=True)

    # When you are ready to check your work with python_ta, uncomment the following lines.
    # (In PyCharm, select the lines below and press Ctrl/Cmd + / to toggle comments.)
    import python_ta
    python_ta.check_all(config={
        'extra-imports': ['a1_part4', 'a1_helpers'],
        'max-line-length': 120
    })