first implementation

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()