panorama.py

import cv2, threading
from datetime import datetime

cap = cv2.VideoCapture(0)
images = []
err_dict = {
    cv2.STITCHER_OK: "STITCHER_OK",
    cv2.STITCHER_ERR_NEED_MORE_IMGS: "STITCHER_ERR_NEED_MORE_IMGS",
}

Dirs = ['E', 'S', 'W']
m_Dirs = ['NE', 'SE', 'SW', 'NW']

class lastFrame(threading.Thread):
    def __init__(self, camera):
        self.camera = camera
        self.frame = None
        super().__init__()
        # Start thread
        self.start()

    def run(self):
        while True:
            ret, self.frame = self.camera.read()

def takePicture(num : int, img):
    time = datetime.now()
    img = cv2.resize(img, (1080,720))
    img = img[100:630, 0:1080]     #CROPOWANIE ZDJĘCIA, USTAWCIE POD SIEBIE
    images.append(img)
    cv2.imwrite("URC_PICS/pic_" + str(num) + "_" + str(time) + ".jpg", img)

def Panorama(imgs):
    stitcher = cv2.Stitcher.create()
    (retval, stitchedImage) = stitcher.stitch(imgs)
    
    if retval!= cv2.STITCHER_OK: 
        print(f"stitching falied with error code {retval} : {err_dict[retval]}")
        exit(retval)
    else:
        print('Your Panorama is ready!!!')  
        cv2.imwrite("picture.jpg", stitchedImage)    

def click_event(event, x, y, flags, params):

    if event == cv2.EVENT_LBUTTONDOWN:
        font = cv2.FONT_HERSHEY_SIMPLEX
        cv2.putText(image, "N", (x - 8,y - 30), font,
                        1, (0, 30, 255), 2)
        cv2.putText(image, "|", (x, y), font,
                        1, (255, 0, 255), 2)
        pos = x
       
        for dir in Dirs:
            pos += step
            pos %= panoWidth
            cv2.putText(image, dir, (pos - 8, y - 30), font,
                        1, (255, 80, 0), 2)
            cv2.putText(image, "|", (pos, y), font,
                            1, (255, 0, 255), 2)
            
        m_pos = int(x + step/2)
        for dir in m_Dirs:
            cv2.putText(image, dir, (m_pos - 12, y - 30), font,
                        0.6, (255, 150, 0), 2)
            cv2.putText(image, "|", (m_pos, y), font,
                            0.6, (255, 0, 255), 2)
            m_pos += step
            m_pos %= panoWidth  
            
        cv2.imwrite('PANORAMA.jpg', image)
        cv2.destroyAllWindows()

last = lastFrame(cap)
counter = 0
inp = input("Press ENTER to take pictures or 'q' TO STOP/QUIT: ")

while inp != 'q':
    inp = input('Picture(' + str(counter) + ")")
    if(inp == ''):
        img = last.frame
        takePicture(counter, img)
        counter+=1

Panorama(images)

image = cv2.imread('picture.jpg')
panoWidth = image.shape[1]
step = int(panoWidth / 4)

cv2.imshow('image', image)
cv2.setMouseCallback('image', click_event)
cv2.waitKey(0)

cv2.destroyAllWindows()