'''
# -*- coding:utf-8 -*-
@author: TulLing
'''
import requests 
from json import JSONDecoder 
 
gesture_englist = ['big_v','fist','double_finger_up','hand_open','heart_d','index_finger_up','ok','phonecall','palm_up','rock','thumb_down','thumb_up','victory']
gesture_chinese = ["我最帥",
   "拳頭，停下",
   "我發(fā)誓",
   "數字5",
   "比心",
   "數字1",
   "好的呢，OK",
   "打電話",
   "手心向上",
   "愛你，520",
   "差評，不好的",
   "好評，Good，很棒",
   "勝利，開心"]
# 將字典排序
def sort_dict(adict):
 return sorted(adict.items(),key= lambda item:item[1])
 
class Gesture(object):
 def __init__(self):
 self.http_url = 'https://api-cn.faceplusplus.com/humanbodypp/beta/gesture'
 self.key = '*****'
 self.secret = '******'
 self.data = {"api_key":self.key,"api_secret":self.secret}
 
 
 # 獲取手勢信息
 def get_info(self,files):
 response = requests.post(self.http_url,data=self.data,files=files)
 req_con = response.content.decode('utf-8')
 req_dict = JSONDecoder().decode(req_con)
 #print(req_dict)
 if('error_message' not in req_dict.keys()) and (len(req_dict['hands'])):
 # 獲取
  hands_dict = req_dict['hands']
  #print(type(hands_dict))
  # 獲取到手的矩形的字典
  gesture_rectangle_dict = hands_dict[0]['hand_rectangle']
  # 獲取到手勢的字典
  gesture_dict = hands_dict[0]['gesture']
  
  return gesture_dict,gesture_rectangle_dict
 else:
  return [],[];
 
 # 獲取到手勢文本信息
 def get_text(self,index):
 return gesture_chinese[index]
 
 # 獲取到手勢對應的概率
 def get_pro(self,gesture_dict,index):
 # print(gesture_dict)
 if(gesture_dict is None or gesture_dict == []):
  return 0
 return gesture_dict[gesture_englist[index]]
 
 # 獲取到手勢的位置
 def get_rectangle(self,gesture_rectangle_dict):
 if(gesture_rectangle_dict is None or gesture_rectangle_dict == []):
  return (0,0,0,0)
 x = gesture_rectangle_dict['top']
 y = gesture_rectangle_dict['left']
 width = gesture_rectangle_dict['width']
 height = gesture_rectangle_dict['height']
 return (x,y,width,height)

'''
# -*- coding:utf-8 -*-
@author: TulLing
'''
import sys
sys.path.append("../gesture/")
 
import os
import random
import cv2 as cv
import time
import LearnGesture
 
def gestureLearning():
 os.system("cls")
 print("進入學習手勢模式！")
 print("我們有13個手勢，來和我學吧！(每次結束后可以選擇輸入 Q\q 退出！)")
 while(True):
 pic_num = random.randint(0,12) # 生成顯示的圖片的編號(隨機數： 0 - 13)
 print(pic_num)
 pic_path = '../gesture/pic/gesture' + str(pic_num) + ".jpg" # 生成圖片路徑
 
 pic = cv.imread(pic_path) # 加載圖片
 pic = cv.resize(pic,(120,120))
 cv.imshow("PIC",pic) # 顯示要學習的手勢
 
 print("即將打開攝像頭，你有5秒種的時間準備手勢，5秒種保持手勢！")
 write_path = "../gesture/pic/test.jpg"
 cap = cv.VideoCapture(1)
 while(True):
  _,frame = cap.read()
  cv.imshow("Frame",frame)
  key = cv.waitKey(10)
  if(key == ord('Q') or key == ord('q')):
  cv.imwrite(write_path,frame)
  cv.waitKey(200)
  cap.release()
  cv.destroyAllWindows()
  break
  
 # 此處應該有手勢識別
 files = {"image_file":open(write_path,'rb')}
 gesture = LearnGesture.Gesture()
 
 # 獲取到手勢文本
 ge_text = gesture.get_text(pic_num)
 # 獲取手勢信息
 gesture_dict,gesture_rectangle_dict = gesture.get_info(files)
 # 獲取手勢的概率
 ge_pro = gesture.get_pro(gesture_dict,pic_num)
 # 獲取到手勢的坐標
 ge_rect = gesture.get_rectangle(gesture_rectangle_dict)
 print("您學習的手勢是：",ge_text)
 print("相似度達到：",ge_pro)
 print("具體位置：",ge_rect)
 
 
 # print("一輪學習結束，是否繼續(xù)學習？(Y/N)")
 # 退出程序，回到主菜單或者繼續(xù)
 commend = input("一輪學習結束，是否繼續(xù)學習？(Y/N)：")
 print(commend)
 
 if( commend == 'N' or commend == "n"):
  break
gestureLearning()