import numpy as np
import matplotlib.pyplot as plt
import calc_way
import get_data
import calc_slope_line
import cv2
import model

model = model.Model()
alpha = model.alpha
beta = model.beta
img_path = r'C:\Users\Administrator\Desktop\BYD\error\20250620\20250620\pic1.jpg'
def draw_lines(img_path):
    image = cv2.imread(img_path)
    if image is None:
            print("Error: 无法读取图像，请检查路径！")
            exit()

    x_zero, y_zero = calc_way.calc_zeros_xto0(862+60)
    print(x_zero)
    print(y_zero)
    point1 = (int(x_zero[0]),960-int(y_zero[0]))
    point2 = (int(x_zero[-1]),960-int(y_zero[-1]))
    cv2.line(image, point1, point2, (0, 0, 255), 1)

    x_zero, y_zero = calc_way.calc_zeros_yto0(0)
    print(x_zero)
    print(y_zero)
    point1 = (int(x_zero[0]),960-int(y_zero[0]))
    point2 = (int(x_zero[-1]),960-int(y_zero[-1]))
    cv2.line(image, point1, point2, (0, 0, 255), 1)

    x_zero, y_zero = calc_way.calc_zeros_xto0(862+300)
    print(x_zero)
    print(y_zero)
    point1 = (int(x_zero[0]),960-int(y_zero[0]))
    point2 = (int(x_zero[-1]),960-int(y_zero[-1]))
    cv2.line(image, point1, point2, (0, 0, 255), 1)

    x_zero, y_zero = calc_way.calc_zeros_yto0(0-480)
    print(x_zero)
    print(y_zero)
    point1 = (int(x_zero[0]),960-int(y_zero[0]))
    point2 = (int(x_zero[-1]),960-int(y_zero[-1]))
    cv2.line(image, point1, point2, (0, 0, 255), 1)

    cv2.imshow('image', image)
    cv2.waitKey(0)
    cv2.destroyAllWindows()

draw_lines(img_path)