imageSeparator.py

import json
import os
from fnmatch import fnmatch
import numpy as np
import cv2
import math

path = "/home/n1m4/Desktop/tasks/OpenCV"
pathArray = []
counter = 2
isBackOn = False
subDirArray = []
counterNextDir = 0


def coord(x,y):
    "Convert world coordinates to pixel coordinates."
    return int(450+170*x), int(300-170*y)

# getting the name of dirs
current_directory = path
for dir in os.listdir(current_directory):
    if os.path.isdir(os.path.join(current_directory, dir)):
        subDirArray.append("/" + dir)

def getData(root, shouldShowField):
    global subDirArray
    print(subDirArray)
    for path, subdirs ,files in os.walk(root):
        for name in files:
            if fnmatch(name, "*.json"):
                pathJson = os.path.join(path, name)
                jsonSize = os.path.getsize(pathJson)
                if jsonSize > 0 and jsonSize < 300:
                    fh = open(pathJson)
                    data = json.load(fh)
                    detected = data["Ball"]["detected"]
                    with open(pathJson) as json_file:
                        data = json.load(json_file)
                        x_corBall = int(data["Ball"]["x"])
                        y_corBall = int(data["Ball"]["y"])
                        radius = int(data["Ball"]["r"])
                        x_corRobotPose = float(data["RobotPose"]["x"])
                        y_corRobotPose = float(data["RobotPose"]["y"])
                        angleRobotPose = float(data["RobotPose"]["angle"])
                        jpgFiles = (jsonToJPG(os.path.join(path, name)))
                        pathArray.append(jpgFiles)
                        if isBackOn is False:

                            showImage(jpgFiles,destPath, x_corBall, y_corBall, radius, detected)

                            if shouldShowField is True:
                                drawField(x_corRobotPose,y_corRobotPose,angleRobotPose) 
                        
                        
                        # # DEBUG
                        # print(path + name + " exists")
                        # print(x_corBall, y_corBall, radius)
                        # print(x_corRobotPose, y_corRobotPose, angleRobotPose)
                        # print(jsonToJPG(os.path.join(path, name)))
                        # jsonSize = os.path.getsize(pathJson)
                        # print("Size of json is : " + str(jsonSize))
                        # print("Size of image is : " + str(FileSize))
                        # fileExistsBool = os.path.isfile(jsonToJPG(os.path.join(path, name)))
                        # FileSize = os.path.getsize(jsonToJPG(os.path.join(path, name)))
                        # # DEBUG


                        
def drawField(x_corRobotPose,y_corRobotPose,angleRobotPose):
    greenField = np.full((600, 900, 3), 127, np.uint8) 
    greenField[:] = (5, 144, 51)
    width = 900
    height = 600
    center_coordinates = (900 - (width/2),600 - (height/2))
    color = (0, 0, 0) 
    thickness = 2

    cv2.circle(greenField, center_coordinates, (150/2), color, thickness)
    cv2.line(greenField, (450,0), (450,600), color, thickness) 

    x_corRobotPose = math.sin(x_corRobotPose)
    y_corRobotPose = math.cos(y_corRobotPose)
    angleInDegree = math.cos(math.radians(angleRobotPose))
    angleXSin = math.sin(angleRobotPose)
    angleXCos = math.cos(angleRobotPose)

    cv2.line(greenField, coord(x_corRobotPose,y_corRobotPose), coord(angleXCos,angleXSin), color, thickness) 
    cv2.circle(greenField, coord(angleXCos,angleXSin), 10, color, -1)
    cv2.rectangle(greenField,(0,0),(900,600),(0,0,0),15)
    cv2.imshow('Field', greenField)
    cv2.waitKey(0)
    cv2.destroyWindow('Field')


def jsonToJPG(path):
    path = path.strip('.json')
    path = path + ".jpg"
    return path

def writeToFile(path):
    file1 = open(destPath,"a")
    print("ADDDDDDDD")
    file1.write(path + "\n") 
    file1.close() 

def showImage(path, destPath, x_cor, y_cor, radius, ballDetected):
    global counter
    global current_directory
    global counterNextDir
    global isBackOn
    cv2.namedWindow(path,cv2.WINDOW_AUTOSIZE)
    img = cv2.imread(path)
    if ballDetected=="1":
        center_coordinates = (x_cor*2, y_cor*2) 
        color = (0, 0, 255) 
        thickness = 2
        cv2.circle(img, center_coordinates, radius*2, color, thickness)
    cv2.imshow(path,img)
    key = cv2.waitKey(0)
    if key == 27:
        cv2.destroyAllWindows()         # It's escape button to exit
        exit()
    if key == 102 or key == 70:         # It's F button to save
        print(path + " added succressfully")
        writeToFile(path)
        cv2.destroyAllWindows()
    if key == 106 or key == 74:         # It's J button to skip
        print(path + " dismissed")
        cv2.destroyAllWindows()
    if key == 39 or key == 83:         # It's Right Arrow button to next dir
        try:
            print(counter)
            cv2.destroyAllWindows()
            print(current_directory + subDirArray[counterNextDir] + " NEXT DIR")
            counterNextDir += 1
            cv2.destroyAllWindows()
            getData(current_directory + subDirArray[counterNextDir], shouldShowFieldState)
        except IndexError:
            counterNextDir = 0
            
        
    if key == 37 or key == 81:         # It's Left Arrow button to previus dir
        try:
            print(counter)
            cv2.destroyAllWindows()
            print(current_directory + subDirArray[counterNextDir] + " PREVIUS DIR")
            counterNextDir -= 1
            cv2.destroyAllWindows()
            getData(current_directory + subDirArray[counterNextDir], shouldShowFieldState)
        except IndexError:
            counterNextDir = 0
            print(counterNextDir)
            pass
            
    if key == 98 or key == 66:         # It's B button to back
        isBackOn = True
        counter = 2
        while isBackOn:
            lastIndex = pathArray[-(counter)]
            cv2.namedWindow('back '+lastIndex,cv2.WINDOW_AUTOSIZE)
            img = cv2.imread(lastIndex)
            cv2.imshow('back '+lastIndex,img)
            key = cv2.waitKey(0)
            counter += 1
            if key == 27:
                cv2.destroyAllWindows()         # It's escape button to exit
                exit()
            if key == 102 or key == 70:         # It's F button to save
                print(path + " added succressfully")
                writeToFile(lastIndex)
            cv2.destroyWindow('back'+lastIndex)
            print(lastIndex)
            if key == 106 or key == 74:         # It's J button to skip
                print(path + " dismissed")
                isBackOn = False


    cv2.destroyAllWindows()
                
# print("Enter the directory: ")
# p = raw_input() 

destPath = (path + "/file.txt")
print("PATH FOR SAVING IS " + path)
shouldShowFieldState = False

getData(path, shouldShowFieldState)