import cv2
import numpy as np
import matplotlib.pyplot as plt

# Apply CLAHE to the grayscale image
clahe = cv2.createCLAHE(clipLimit=2.0, tileGridSize=(8, 8))
image_gray = cv2.imread("../data/results/phantom/capture_1_2024-07-09T15-42-41.png", cv2.IMREAD_GRAYSCALE)
image_gray = clahe.apply(image_gray)

# Threshold the image to create a binary image
_, binary = cv2.threshold(image_gray, 128, 255, cv2.THRESH_BINARY)

# Find contours in the binary image
contours, _ = cv2.findContours(binary, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

# Find the largest contour by area
largest_contour = max(contours, key=cv2.contourArea)

# Fit an ellipse to the largest contour if it has enough points
if len(largest_contour) >= 5:
    ellipse = cv2.fitEllipse(largest_contour)
    (x, y), (major_axis, minor_axis), angle = ellipse

    # Check if the ellipse dimensions are valid
    if major_axis > 0 and minor_axis > 0:
        # Draw the ellipse on the image
        image_with_ellipse = cv2.cvtColor(image_gray, cv2.COLOR_GRAY2BGR)
        cv2.ellipse(image_with_ellipse, ellipse, (0, 255, 0), 2)

        # Calculate the area of the ellipse
        semi_major_axis = major_axis / 2
        semi_minor_axis = minor_axis / 2
        ellipse_area = np.pi * semi_major_axis * semi_minor_axis
        print(f"Ellipse parameters: Center=({x:.2f}, {y:.2f}), "
              f"Major Axis={major_axis:.2f}, Minor Axis={minor_axis:.2f}, Angle={angle:.2f}")
        print(f"Calculated Ellipse Area: {ellipse_area:.2f} square pixels")

        # Display the images
        plt.figure(figsize=(10, 5))
        plt.subplot(1, 3, 1)
        plt.imshow(image_gray, cmap='gray')
        plt.title('CLAHE Image')
        plt.subplot(1, 3, 2)
        plt.imshow(binary, cmap='gray')
        plt.title('Binary Image')
        plt.subplot(1, 3, 3)
        plt.imshow(cv2.cvtColor(image_with_ellipse, cv2.COLOR_BGR2RGB))
        plt.title('Result Image with Ellipse')
        plt.show()
else:
    print("No valid ellipse found in the largest contour")