import requests
import numpy as np
import cv2
import random
import matplotlib.pyplot as plt
import threading
from threading import Thread
ans = list()
URL = 'https://api.arstand-lab.ru'

token = 'Token 581314b5dc6db4dd928dc249bf390affbec4df75'
HEADERS = {'Authorization': token,
           'Content-Type': 'application/json'}
z=0
def get_markers():
    res = requests.get(f'{URL}/api/0/marker/get_markers/', headers=HEADERS)
    if res.ok:
        with open('file.npz', 'wb') as f:
            f.write(res.content)
        return True
    else:
        return False


def post_ids():
    global ans
    content = {'markers': str(ans)}
    res = requests.post(f'{URL}/api/0/marker/check_markers/ids/',
                        headers=HEADERS,
                        json=content)
    if res.ok:
        return True
    else:
        return False


if __name__ == '__main__':
    print('GET successful: ', get_markers())
    data = np.load('ss.npz')
    test = np.load("file.npz")
    z = 0
    for t in range(len(test['markers'])):
        rows = test['markers'][t].shape[0]
        circles = cv2.HoughCircles(test['markers'][t], cv2.HOUGH_GRADIENT, 1, rows / 8,
                                       param1=80, param2=30,
                                       minRadius=1, maxRadius=40)
        r = 0
        if circles is not None:
            circles = np.uint16(np.around(circles))
            for i in circles[0, :]:
                center = (i[0], i[1])
                    # circle center
                    # circle outline
                radius = i[2]
            if center[0] > 150 and center[1] > 150:
                r = 180
            elif center[0] < 150 and center[1] > 150:
                r = 270
            elif center[0] < 150 and center[1] < 150:
                r = 0
            else:
                r = 90
        (h, w) = test['markers'][t].shape[:2]
        center = (w / 2, h / 2)
        M = cv2.getRotationMatrix2D(center, r, 1.0)
        rotated = cv2.warpAffine(test['markers'][t], M, (w, h))
        s1 = 0
        id = 0
        ret, im_bw1 = cv2.threshold(cv2.medianBlur(rotated, 9), 127, 255, 0)
        for g in range(7):
            s = 0
            # im_bw1 = cv2.medianBlur(data['data'][i],5)
            # ret, im_bw1 = cv2.threshold(im_bw1, 110, 255, 0)
            ret, im_bw2 = cv2.threshold(cv2.medianBlur(data['data'][g], 9), 127, 255, 0)
            s = np.sum(np.equal(im_bw1,im_bw2))
            if s > s1:
                s1 = s
                id = data['ids'][g]
        ans.append([test['coords'][t][0], test['coords'][t][1], id])
print('POST ids successful: ', post_ids())