From ecb33e24479ccbf2851c1625e789b38826ffb3eb Mon Sep 17 00:00:00 2001 From: Lucas Marais Date: Thu, 23 Mar 2023 15:44:12 +0100 Subject: [PATCH] first implementation --- Position-Robot/extractcoord.py | 51 ++++++++++++++++++++++++++++++++++ 1 file changed, 51 insertions(+) create mode 100644 Position-Robot/extractcoord.py diff --git a/Position-Robot/extractcoord.py b/Position-Robot/extractcoord.py new file mode 100644 index 0000000..f57c7a7 --- /dev/null +++ b/Position-Robot/extractcoord.py @@ -0,0 +1,51 @@ +import math +import matplotlib.image as mpimg +import numpy as np +import matplotlib.pyplot as plt +import cv2 +import time +import os +import pygame +from PIL import Image +import random + + +class getpos(): + def __init__(self): + self.win=pygame.display.set_mode((1600, 720)) + self.background = (0,0,0) + self.Done = True + self.coords = [] + self.pos = np.array([]) + + while self.Done : + self.img = Image.open(('board.jpg')) + image = self.img.resize((int(1600), int(720))) + size = image.size + mode = image.mode + data = image.tobytes() + self.displayed = pygame.image.fromstring(data, size, mode) + self.rect = self.displayed.get_rect() + self.win.blit(self.displayed, self.rect) + self.checkEvent() + pygame.display.flip() + + def checkEvent(self): + for event in pygame.event.get(): + if event.type == pygame.QUIT: + self.Done = False + if event.type == pygame.MOUSEBUTTONDOWN: + mouse_presses = pygame.mouse.get_pressed() + if mouse_presses[0]: + x,y = pygame.mouse.get_pos() + if len(self.coords) < 4: + self.coords.append(np.array([x,y])) + if len(self.coords) == 4: + self.pos = np.array([x,y]) + + print('\n',self.coords,'\n', self.pos,'\n') + + print(x,y) + + +extract = getpos()