arm01.py

import cv2
import uuid
import mediapipe as mp
import numpy as np
import datetime
import requests


mp_drawing = mp.solutions.drawing_utils  # 有關於任何需要繪製的東西
mp_pose = mp.solutions.pose


# 初始化计数器和标志位
counter = 8
stage = None

id = "123123"
part = "arm"
playstage = "上肢上舉"
start_time = datetime.datetime.now()
print(f"{part}\n{playstage}\nStart time: {start_time}")

# 11:LEFT_SHOULDER；13:LEFT_ELBOW手肘；15:LEFT_WRIST手腕
# 計算角度


def calculate_angle(a, b, c):  # 分別為肩，手肘，手腕
    a = np.array(a)
    b = np.array(b)
    c = np.array(c)
    # 得到兩個向量之間的角度
    radains = np.arctan2(c[1]-b[1], c[0]-b[0])  # 弧度(y,x)
    angle = np.abs(radains*180.0/np.pi)  # 弧度->角度
    if angle > 180.0:
        angle = 360-angle  # >180度代表手是垂下來的
    return angle


# 定义MediaPipe姿势识别模型
with mp_pose.Pose(model_complexity=0, min_detection_confidence=0.5, min_tracking_confidence=0.5) as pose:  # 想要偵測得更具體可以提高信度
    # 建立視窗，並讓視窗可以動態調整
    cv2.namedWindow('Camera', cv2.WINDOW_NORMAL)
    # 打开摄像头
    cap = cv2.VideoCapture(0)

    # 讀取圖像
    # img = cv2.imread('giftbox.png')
    while cap.isOpened():
        # 读取帧
        success, image = cap.read()

        # mediapipe需要要RGB
        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
        image.flags.writeable = False  # 設定是否可寫

        # 检测姿势
        results = pose.process(image)

        # opencv需要BGR
        image.flags.writeable = True
        image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)

        # 提取關節點landmark
        try:

            landmarks = results.pose_landmarks.landmark
            # get coordinates
            shoulder = [landmarks[mp_pose.PoseLandmark.LEFT_SHOULDER.value].x,
                        landmarks[mp_pose.PoseLandmark.LEFT_SHOULDER.value].y]  # landmarks獲取座標(取左肩X值，取Y值)
            elbow = [landmarks[mp_pose.PoseLandmark.LEFT_ELBOW.value].x,
                     landmarks[mp_pose.PoseLandmark.LEFT_ELBOW.value].y]
            wrist = [landmarks[mp_pose.PoseLandmark.LEFT_WRIST.value].x,
                     landmarks[mp_pose.PoseLandmark.LEFT_WRIST.value].y]

            # calculate angle
            angle = calculate_angle(elbow, shoulder, wrist)

            # visualize視覺化 angle
            # cv2.putText(image, str(angle), tuple(np.multiply(elbow, [640, 480]).astype(int)), cv2.FONT_HERSHEY_SIMPLEX, 3, (255, 255, 255), 5, cv2.LINE_AA)#用中間點*頁面大小

            # counter:raise arm time
            if angle > 80:
                stage = "down"
            if angle < 50 and stage == 'down':  # 希望做動作時是由down的狀態往up的狀態，不希望手直接舉高
                stage = "up"
                counter += 1
                if counter == 10:  # 次數=10結束
                    end_time = datetime.datetime.now()
                    duration = end_time - start_time
                    duration_time_str = str(duration)

                    PlayDate = start_time.strftime('%Y-%m-%d')
                    StartTime = start_time.strftime('%H:%M:%S')
                    EndTime = end_time.strftime('%H:%M:%S')
                    UID = str(uuid.uuid4())
                    duration_formatted = duration.total_seconds()
                    hours, remainder = divmod(duration_formatted, 3600)
                    minutes, seconds = divmod(remainder, 60)
                    duration_str_formatted = "{:02}:{:02}:{:02}".format(
                        int(hours), int(minutes), int(seconds))

                    print("id:", id)
                    print("Counter:", counter)
                    print("part:", part)
                    print("UID:", UID)
                    print("PlayStage:", playstage)
                    print("Start Time:", StartTime)
                    print("End time:", EndTime)
                    print("Duration Time:", duration_str_formatted)

                    url = 'http://127.0.0.1:8000/add_gamerecord/'  # 替換為你的Django應用程式的URL和端點

                    data = {
                        'USER_UID': id,
                        'PlayDate': PlayDate,
                        'PlayPart': part,
                        'UID': UID,
                        'PlayStage': playstage,
                        'StartTime': StartTime,
                        'EndTime': EndTime,
                        'DurationTime': duration_str_formatted,
                        'AddCoin': "5",
                        'ExerciseCount': counter
                    }
                    header = {
                        "Content-Type": "application/json"
                    }
                    response = requests.post(url, json=data, headers=header)
                    print(response.text)
                    if response.status_code == 200:
                        print("Data sent successfully.")
                    else:
                        print("Failed to send data.")
                    break

            # if counter == 5:
            #     cv2.putText(image, 'DONE!!', (320, 240), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 0), 1, cv2.LINE_AA)

        except:
            pass  # 如果有錯誤不會影響整個循環

        # 在左上角放置一個狀態框
        # 畫方形(pic,左上點,右下點,顏色,粗細(使用填滿功能))
        cv2.rectangle(image, (0, 0), (170, 73), (245, 117, 16), -1)
        # cv2.rectangle(image,(277,0),(554,73),(245,117,16),-1)
        # cv2.putText(image, str('Raise left hand, parallel to shoulder.'), (10, 100), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 1, cv2.LINE_AA)
        # 顯示次數
        cv2.putText(image, 'number of times', (15, 12),
                    cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 0), 1, cv2.LINE_AA)  # 3:起始座標
        cv2.putText(image, str(counter), (10, 60),
                    cv2.FONT_HERSHEY_SIMPLEX, 2, (255, 255, 255), 2, cv2.LINE_AA)
        # 顯示上下
        # cv2.putText(image, 'STAGE', (65, 12), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 0), 1, cv2.LINE_AA)  # 3:起始座標
        # cv2.putText(image, stage, (60, 60), cv2.FONT_HERSHEY_SIMPLEX, 2, (255, 255, 255), 2, cv2.LINE_AA)

        # 關節點可視化
        # 分別可以印出後兩個參數試試，可以看到關節點座標與各關節點如何連接
        mp_drawing.draw_landmarks(
            image, results.pose_landmarks, mp_pose.POSE_CONNECTIONS)

        # 显示图像
        cv2.imshow('Camera', image)
        # 按下q键退出
        if cv2.waitKey(10) & 0xFF == ord('q'):
            end_time = datetime.datetime.now()

            duration = end_time - start_time
            duration_time_str = str(duration)

            PlayDate = start_time.strftime('%Y-%m-%d')
            StartTime = start_time.strftime('%H:%M:%S')
            EndTime = end_time.strftime('%H:%M:%S')
            UID = str(uuid.uuid4())
            duration_formatted = duration.total_seconds()
            hours, remainder = divmod(duration_formatted, 3600)
            minutes, seconds = divmod(remainder, 60)
            duration_str_formatted = "{:02}:{:02}:{:02}".format(
                int(hours), int(minutes), int(seconds))

            print("Counter:", counter)
            print("Start Time:", PlayDate)
            print("End time:", EndTime)
            print("Duration Time:", duration_str_formatted)

            url = 'http://127.0.0.1:8000/add_gamerecord/'  # 替換為你的Django應用程式的URL和端點

            data = {
                'USER_UID': id,
                'PlayDate': PlayDate,
                'PlayPart': part,
                'UID': UID,
                'PlayStage': playstage,
                'StartTime': StartTime,
                'EndTime': EndTime,
                'DurationTime': duration_str_formatted,
                'AddCoin': "5",
                'ExerciseCount': counter
            }
            header = {
                "Content-Type": "application/json"
            }
            response = requests.post(url, json=data, headers=header)

            if response.status_code == 200:
                print("Data sent successfully.")
            else:
                print("Failed to send data.")
            break
    # 释放摄像头和窗口
    cap.release()
    cv2.destroyAllWindows()