G11-2024

import cv2 as cv

import numpy as np

import tkinter as tk

from tkinter import filedialog, messagebox

import os

from PIL import Image, ImageTk

class SharedData:

    def __init__(self):

        self.left_image = None

        self.right_image = None

        self.overlap_region = 0

class ImageIO:

    def __init__(self, shared_data):

        self.shared_data = shared_data

    def load_image(self, path):

        return cv.imread(path, cv.IMREAD_COLOR)

    def write_image(self, img, fileName="image.png"):

        if cv.imwrite(fileName, img):

            print(f"File {fileName} Written Successfully")

        else:

            print("File writing failed")

    def show_image(self, img, title="Image Title", canvas=None):

        img_rgb = cv.cvtColor(img, cv.COLOR_BGR2RGB)

        img_pil = Image.fromarray(img_rgb)

        img_tk = ImageTk.PhotoImage(img_pil)

        canvas.create_image(0, 0, anchor=tk.NW, image=img_tk)

        canvas.image = img_tk

class Commands:

    ## \brief A class to handle various commands related to image processing.

    #

    # The Commands class provides methods to load, display, and set overlap regions

    # for images using a graphical user interface (GUI) and shared data. It integrates

    # with an ImageIO class to handle image input and output operations.

    #

    # \author 

    # \version 1.0

    # \date 2024-10-24

    def __init__(self, gui, shared_data):

        ## \brief Initializes the Commands instance.

        #

        # Sets up the GUI interface and shared data. It also initializes

        # an ImageIO instance for handling image operations.

        #

        # \param gui The GUI instance used for displaying images and getting user input.

        # \param shared_data The shared data structure that stores images and other state information.

        self.gui = gui

        self.shared_data = shared_data

        self.io = ImageIO(shared_data)

    def load_left_image(self):

        ## \brief Loads the left image from a file.

        #

        # Opens a file dialog for the user to select a left image file. If a valid file path

        # is provided, the left image is loaded into the shared data. A message is printed

        # to confirm the loading process.

        # 

        # \warning This method does not check the file format or image content.

        filepath = filedialog.askopenfilename(title="Select Left Image")

        if filepath:

            self.shared_data.left_image = self.io.load_image(filepath)

            print("Left image loaded.")

    def load_right_image(self):

        ## \brief Loads the right image from a file.

        #

        # Opens a file dialog for the user to select a right image file. If a valid file path

        # is provided, the right image is loaded into the shared data. A message is printed

        # to confirm the loading process.

        #

        # \warning This method does not check the file format or image content.

        filepath = filedialog.askopenfilename(title="Select Right Image")

        if filepath:

            self.shared_data.right_image = self.io.load_image(filepath)

            print("Right image loaded.")

    def show_left_image(self):

        ## \brief Displays the left image on the GUI.

        #

        # If the left image is available in the shared data, it is shown on the GUI canvas.

        # Otherwise, an error message is displayed to the user.

        #

        # \exception Shows a message box error if no left image is loaded.

        if self.shared_data.left_image is not None:

            self.io.show_image(self.shared_data.left_image, canvas=self.gui.canvas)

            print("Left image displayed.")

        else:

            messagebox.showerror("Error", "No left image loaded!")

    def show_right_image(self):

        ## \brief Displays the right image on the GUI.

        #

        # If the right image is available in the shared data, it is shown on the GUI canvas.

        # Otherwise, an error message is displayed to the user.

        #

        # \exception Shows a message box error if no right image is loaded.

        if self.shared_data.right_image is not None:

            self.io.show_image(self.shared_data.right_image, canvas=self.gui.canvas)

            print("Right image displayed.")

        else:

            messagebox.showerror("Error", "No right image loaded!")

    def set_overlap_region(self):

        ## \brief Sets the overlap region for image processing.

        #

        # Retrieves the overlap region value from the GUI input and updates the shared data

        # accordingly. A message is printed to confirm the overlap region setting.

        #

        # \note This method assumes that the GUI input is a valid overlap region value.

        self.shared_data.overlap_region = self.gui.overlap.get()

        print(f"Overlap region set to {self.shared_data.overlap_region}")

class GUI:

    def __init__(self, master):

        self.master = master

        self.master.title("Image Projector")

        self.canvas = tk.Canvas(self.master, bg="black")

        self.canvas.pack(fill=tk.BOTH, expand=True)

        self.shared_data = SharedData()

        self.commands = Commands(self, self.shared_data)

        self.control_window()

    def control_window(self):

        self.control_window = tk.Toplevel(self.master)

        self.control_window.title("Controls")

        self.overlap = tk.IntVar()

        self.label = tk.Label(self.control_window, text='Overlap region Width', font=('calibre', 10, 'bold'))

        self.overlapBox = tk.Entry(self.control_window, textvariable=self.overlap)

        self.label.pack()

        self.overlapBox.pack()

        self.set_overlap_button = tk.Button(self.control_window, text="Set Overlap Region", command=self.commands.set_overlap_region)

        self.set_overlap_button.pack()

        self.button_load_left = tk.Button(self.control_window, text="Load Left Image", command=self.commands.load_left_image)

        self.button_load_left.pack()

        self.button_load_right = tk.Button(self.control_window, text="Load Right Image", command=self.commands.load_right_image)

        self.button_load_right.pack()

        self.button_show_left = tk.Button(self.control_window, text="Show Left Image", command=self.commands.show_left_image)

        self.button_show_left.pack()

        self.button_show_right = tk.Button(self.control_window, text="Show Right Image", command=self.commands.show_right_image)

        self.button_show_right.pack()

if __name__ == "__main__":

    root = tk.Tk()

    root.geometry("1280x800")

    app = GUI(root)

    root.mainloop()