Update tempo.txt

tempo.txt

@@ -1,43 +1,61 @@
 import cv2
 import numpy as np
 
-# Load the mask image
-mask = cv2.imread('mask_image.png', cv2.IMREAD_GRAYSCALE)  # Replace 'mask_image.png' with the path to your mask image
-
-# Find contours in the mask
-contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-
-# Initialize variables to store the last two bottom-right points
-last_two_points = []
-
-# Iterate through each contour
-for contour in contours:
-    # Extract the bottom-right points of the contour
-    bottom_right_points = [point[0] for point in contour if point[0][0] == max(contour[:, :, 0]) and point[0][1] == max(contour[:, :, 1])]
+def extend_line_to_image_edges(x1, y1, x2, y2, img_width, img_height):
+    # Calculate the slope (m) and intercept (c) of the line
+    if x2 != x1:
+        m = (y2 - y1) / (x2 - x1)
+        c = y1 - m * x1
+    else:
+        # The line is vertical
+        m = None
     
-    # Sort the bottom-right points based on their x-coordinate
-    bottom_right_points.sort(key=lambda point: point[0])
+    # Find intersection points with image borders
+    points = []
     
-    # Choose the last two points
-    if len(bottom_right_points) >= 2:
-        last_two_points.extend(bottom_right_points[-2:])
-
-# Check if the distance between the last two points is more than 200 pixels
-if len(last_two_points) == 2 and np.linalg.norm(np.array(last_two_points[0]) - np.array(last_two_points[1])) > 200:
-    # Fit a line passing through the last two points
-    vx, vy, x0, y0 = cv2.fitLine(np.array(last_two_points), cv2.DIST_L2, 0, 0.01, 0.01)
-
-    # Calculate the start and end points of the line
-    start_point = (int(x0 - 1000*vx), int(y0 - 1000*vy))
-    end_point = (int(x0 + 1000*vx), int(y0 + 1000*vy))
-
-    # Draw the line on a blank image
-    line_image = np.zeros_like(mask)
-    cv2.line(line_image, start_point, end_point, (255, 255, 255), thickness=2)
-
-    # Display the line image
-    cv2.imshow('Line Passing Through Last Two Points', line_image)
-    cv2.waitKey(0)
-    cv2.destroyAllWindows()
-else:
-    print("Unable to find suitable points or distance constraint not met.")
+    # Intersection with the left border (x=0)
+    if m is not None:
+        y_at_left = c
+        if 0 <= y_at_left <= img_height:
+            points.append((0, int(y_at_left)))
+
+    # Intersection with the right border (x=img_width)
+    if m is not None:
+        y_at_right = m * img_width + c
+        if 0 <= y_at_right <= img_height:
+            points.append((img_width, int(y_at_right)))
+
+    # Intersection with the top border (y=0)
+    if m is not None:
+        x_at_top = -c / m
+        if 0 <= x_at_top <= img_width:
+            points.append((int(x_at_top), 0))
+    else:
+        # The line is vertical and intersects the top border if y1=0 or y2=0
+        if y1 == 0:
+            points.append((x1, 0))
+        elif y2 == 0:
+            points.append((x2, 0))
+        
+    # Intersection with the bottom border (y=img_height)
+    if m is not None:
+        x_at_bottom = (img_height - c) / m
+        if 0 <= x_at_bottom <= img_width:
+            points.append((int(x_at_bottom), img_height))
+    else:
+        # The line is vertical and intersects the bottom border if y1=img_height or y2=img_height
+        if y1 == img_height:
+            points.append((x1, img_height))
+        elif y2 == img_height:
+            points.append((x2, img_height))
+
+    # Choose the two valid intersection points that extend the line
+    if len(points) >= 2:
+        extended_points = points[:2]
+    else:
+        extended_points = points
+
+    return extended_points
+
+# Example usage
+img = np.zeros((500, 500, 3),