ui.py

import cv2
import numpy as np
import pygame.mixer
import time
import threading

MAX_WIDTH = 1824
MAX_HEIGHT = 984
SIGN_HEIGHT = (int)(MAX_HEIGHT / 3)
SIGN_WIDTH = SIGN_HEIGHT
GUIDE_WIDTH = MAX_WIDTH - SIGN_WIDTH
GUIDE_HEIGHT = MAX_HEIGHT
TEXT_WIDTH = GUIDE_WIDTH
TEXT_HEIGHT = MAX_HEIGHT - 98
LIST_MARGIN = 123
LIST_HEIGHT = 190
LIST_WIDTH = 495

def soundPlay():
    pygame.mixer.music.play(1)
    time.sleep(10)
    pygame.mixer.music.stop()

def windowInit(lang):
    #cv2.namedWindow('drive', cv2.WINDOW_AUTOSIZE)
    cv2.namedWindow('drive', cv2.WINDOW_NORMAL)
    cv2.setWindowProperty('drive', cv2.WND_PROP_FULLSCREEN, cv2.WINDOW_FULLSCREEN)
    
    pygame.init()
    pygame.mixer.music.load(lang + '/voice/turn.mp3')

    images = {}
    images['REFUEL'] = cv2.imread(lang + '/img/refuel.png')			#full size
    images['LEFT'] = cv2.imread('./img/turn_left.png')				#main size
    images['RIGHT'] = cv2.imread('./img/turn_right.png')			#main size
    images['STOP'] = cv2.imread(lang + '/img/stop.png')				#sign size
    images['SLOW'] = cv2.imread(lang + '/img/slow.png')				#sign size
    images['OVER'] = cv2.imread(lang + '/img/overtaking.png')		#sign size
    images['ATTENTION'] = cv2.imread(lang + '/img/attention.png')	#text size
    images['DOOR'] = cv2.imread(lang + '/img/door.png')				#full size
    images['BACK'] = cv2.imread(lang + '/img/back.png')				#full size
    images['DANGER'] = cv2.imread(lang + '/img/danger.png')			#text size

    return images

def pastePicture(background, src, x, y):
    
    row, col, channel = src.shape
    
    graySrc = cv2.cvtColor(src, cv2.COLOR_BGR2GRAY)
    ret, mask = cv2.threshold(graySrc, 10, 255, cv2.THRESH_BINARY)
    mask_inv = cv2.bitwise_not(mask)
    
    roi = background[0 + y : row + y, 0 + x : col + x]
    
    bg = cv2.bitwise_and(roi, roi, mask = mask_inv)
    srcFg = cv2.bitwise_and(src, src, mask = mask)
    dst = cv2.add(bg, srcFg)
    background[0 + y : row + y, 0 + x : col + x] = dst

def makeWindow(base, images, stateString):
    turn, text, stop, slow, over = stateString.split(',')
    
    #main area    
    if turn == 'left':
        pastePicture(base, images['LEFT'], SIGN_WIDTH, 0)
        soundThread = threading.Thread(target = soundPlay)
        soundThread.start()
        
    elif turn == 'right':
        pastePicture(base, images['RIGHT'], SIGN_WIDTH, 0)
        soundThread = threading.Thread(target = soundPlay)
        soundThread.start()
    
    #text area
    if text == 'danger':
        pastePicture(base, images['DANGER'], SIGN_WIDTH, TEXT_HEIGHT)

    else :
        pastePicture(base, images['ATTENTION'], SIGN_WIDTH, TEXT_HEIGHT)
    
    #sign area
    if stop == 'stop':
        pastePicture(base, images['STOP'], 0, 0)
    
    if over == 'over':
        pastePicture(base, images['OVER'], 0, SIGN_HEIGHT)
    
    if slow == 'slow':
        pastePicture(base, images['SLOW'], 0, SIGN_HEIGHT * 2)
    
    cv2.imshow('drive', base)
    
def langSelectWindow(bgOrigin, frame, lang):
    bg = bgOrigin.copy()
    if lang == 'en':
        pastePicture(bg, frame, LIST_WIDTH, LIST_MARGIN + LIST_HEIGHT * 0)
    elif lang == 'zh-tw':
        pastePicture(bg, frame, LIST_WIDTH, LIST_MARGIN + LIST_HEIGHT * 1)
    elif lang == 'zh-cn':
        pastePicture(bg, frame, LIST_WIDTH, LIST_MARGIN + LIST_HEIGHT * 2)
    elif lang == 'ko':
        pastePicture(bg, frame, LIST_WIDTH, LIST_MARGIN + LIST_HEIGHT * 3)
        
    cv2.imshow('drive', bg)
    cv2.waitKey(1)