detectingPoints.py

"""
Please run THIS program first!

This program is responsible for detecting EBP (entire board points) and writing that data into a file
as well as showing an x and y coordinates for a clicked pixel.
"""

import numpy as np
import cv2
import sys
import os
from PcbFunctions import *

print("~~~---STARTING - Detecting Points---~~~")


# Mouse click handler TODO: move to pcb or OpenCV functions
def click_event(event, x, y, flags, params):
    # checking for left mouse clicks
    if event == cv2.EVENT_LBUTTONDOWN:

        # displaying the coordinates
        # on the Shell

        colorsB = img[y, x, 0]
        colorsG = img[y, x, 1]
        colorsR = img[y, x, 2]
        colors = img[y, x]

        print("Coordinates: X: ", x, "Y: ", y, "[ B: ",
              colorsB, "G: ", colorsG, "R: ", colorsR, " ]")
        # print("BRG Format: ", colors)
        # print("B: ", colorsB, "G: ", colorsG, "R: ", colorsR)

        # displaying the coordinates
        # on the image window
        font = cv2.FONT_HERSHEY_SIMPLEX
        cv2.putText(img, str(x) + ',' +
                    str(y), (x, y), font,
                    0.7, (255, 255, 255), 1, cv2.LINE_AA)

        cv2.circle(img, (x, y), 2, (0, 0, 0), -1)

        cv2.imshow('original image', img)

###########
# Golbal Variables
###########

# variable that stores the entire board x,y coordinates
EntireBoardPoints = np.array([[1, 2], [3, 4]])
# clearing the array for inputs
EntireBoardPoints = np.delete(EntireBoardPoints, [0, 1], axis=0)

# SET BEFORE USE!
ImageName = "Board8.png"
write_to_file = True
Debugging_Enable = False

# Write all the points to a file for further analysis by ConnectionFinding.py
if(write_to_file):
    try:
        POINTS_FILE = open(
            "output/Files/PointsFileFor_{}.txt".format(ImageName), "w")
        POINTS_FILE.write("EntireBoardPoints program:\n")
    except OSError as e:
        sys.exit("Could not open directory to output the Points file to. error: ", e)


img = cv2.imread('assets/Example_images/Board_images/{}'.format(ImageName), cv2.IMREAD_COLOR)
if img is None:
    sys.exit("Could not read the image.")

"""
print('Original Dimensions : ',img.shape)
 
scale_percent = 220 # percent of original size
width = int(img.shape[1] * scale_percent / 100)
height = int(img.shape[0] * scale_percent / 100)
dim = (width, height)
  
# resize image
img = cv2.resize(img, dim, interpolation = cv2.INTER_AREA)
 
print('Resized Dimensions : ',img.shape)
 
cv2.imshow("Resized image", img)
cv2.waitKey(0)

"""

hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
rows = gray.shape[0]  # 93.625
original_img = img.copy()

# detecting circles V2
EntireBoardPoints, img = DetectPointsV2(img, Debugging_Enable)

print("EntireBoardPoints:")
[print(f"({BoardPoint.x}, {BoardPoint.y})") for BoardPoint in EntireBoardPoints]
cv2.imshow('original image', img)

# write to a file all the points x,y
if(write_to_file):
    for PCBpoint in EntireBoardPoints:
        POINTS_FILE.write(f"\nPoint: [{PCBpoint.x},{PCBpoint.y}]")

    POINTS_FILE.close()

# handling clicked on a pixel in the image to show the x and y cords
cv2.setMouseCallback('original image', click_event)

# input handler
input = cv2.waitKey(0)

while(input != 27 and input != -1):  # input != esc
    print(input)

    if(input == 114):  # input == R; ==> Reset Pic
        img = original_img.copy()
        cv2.imshow("board image", img)

    input = cv2.waitKey(0)  # waits for any input


cv2.destroyAllWindows()

print("~~~---FINISH---~~~")