using System.Collections;
using System.Collections.Generic;
using UnityEngine;

public class YogaConfig
{
    public static readonly Dictionary<string, int> BODY_PARTS = new Dictionary<string, int>()
        {
            { "Nose", 0 },  //{ "Nose", 0 },
            { "Neck", 1 },  //{ "Neck", 1 },
            { "LShoulder", 2 }, //{ "RShoulder", 2 },
            { "LElbow", 3 },    //{ "RElbow", 3 },
            { "LWrist", 4 },    //{ "RWrist", 4 },
            { "RShoulder",5 },  //{ "LShoulder",5 },
            { "RElbow", 6 },    //{ "LElbow", 6 },
            { "RWrist", 7 },    //{ "LWrist", 7 },
            { "LHip", 8 },  //{ "RHip", 8 },
            { "LKnee", 9 }, //{ "RKnee", 9 },
            { "LAnkle", 10 },   //{ "RAnkle", 10 },
            { "RHip", 11 }, //{ "LHip", 11 },
            { "RKnee", 12 },    //{ "LKnee", 12 },
            { "RAnkle", 13 },   //{ "LAnkle", 13 },
            { "LEye", 14 }, //{ "REye", 14 },
            { "REye", 15 }, //{ "LEye", 15 },
            { "LEar", 16 }, //{ "REar", 16 },
            { "REar", 17 }, //{ "LEar", 17 },
            { "Background", 18 }    //{ "Background", 18 }
        };

    public static readonly string[,] POSE_PAIRS = new string[,]
    {
            { "Neck", "RShoulder" },
            { "Neck", "LShoulder" },
            { "RShoulder", "RElbow" },
            { "RElbow", "RWrist" },
            { "LShoulder", "LElbow" },
            { "LElbow", "LWrist" },
            { "Neck", "RHip" },
            { "RHip", "RKnee" },
            { "RKnee", "RAnkle" },
            { "Neck", "LHip" },
            { "LHip", "LKnee" },
            { "LKnee", "LAnkle" },
            { "Neck", "Nose" },
            { "Nose", "REye" },
            { "REye", "REar" },
            { "Nose", "LEye" },
            { "LEye", "LEar" }
    };

    public const float InWidth = 256;
    public const float InHeight = 256;
    public const float InScale = 1.0f / 255;

    public static readonly Dictionary<ModelType, string> MODEL_PATHS = new Dictionary<ModelType, string>()
    {
        {ModelType.OpenPose,"OpenCVForUnity/dnn/lightweight_pose_estimation_201912.onnx" },
        //{"","" },
    };

    public static readonly double[] InMean = new double[] { 128.0, 128.0, 128.0 };
}

public enum ModelType
{
    OpenPose,
    YoloV7,
    //,
}