using System;

namespace UnityEngine.Perception.GroundTruth
{
    /// <summary>
    /// A definition of a keypoint (joint).
    /// </summary>
    [Serializable]
    public class KeyPointDefinition
    {
        /// <summary>
        /// The name of the keypoint
        /// </summary>
        public string label;
        /// <summary>
        /// Does this keypoint map directly to a <see cref="Animator"/> <see cref="Avatar"/> <see cref="HumanBodyBones"/>
        /// </summary>
        public bool associateToRig = true;
        /// <summary>
        /// The associated <see cref="HumanBodyBones"/> of the rig
        /// </summary>
        public HumanBodyBones rigLabel = HumanBodyBones.Head;
        /// <summary>
        /// The color of the keypoint in the visualization
        /// </summary>
        public Color color;
    }

    /// <summary>
    /// A skeletal connection between two joints.
    /// </summary>
    [Serializable]
    public class SkeletonDefinition
    {
        /// <summary>
        /// The first joint
        /// </summary>
        public int joint1;
        /// <summary>
        /// The second joint
        /// </summary>
        public int joint2;
        /// <summary>
        /// The color of the skeleton in the visualization
        /// </summary>
        public Color color;
    }

    /// <summary>
    /// Template used to define the keypoints of a humanoid asset.
    /// </summary>
    [CreateAssetMenu(fileName = "KeypointTemplate", menuName = "Perception/Keypoint Template", order = 2)]
    public class KeyPointTemplate : ScriptableObject
    {
        /// <summary>
        /// The <see cref="Guid"/> of the template
        /// </summary>
        public string templateID = Guid.NewGuid().ToString();
        /// <summary>
        /// The name of the template
        /// </summary>
        public string templateName;
        /// <summary>
        /// Texture to use for the visualization of the joint.
        /// </summary>
        public Texture2D jointTexture;
        /// <summary>
        /// Texture to use for the visualization of the skeletal connection.
        /// </summary>
        public Texture2D skeletonTexture;
        /// <summary>
        /// Array of <see cref="KeyPointDefinition"/> for the template.
        /// </summary>
        public KeyPointDefinition[] keyPoints;
        /// <summary>
        /// Array of the <see cref="SkeletonDefinition"/> for the template.
        /// </summary>
        public SkeletonDefinition[] skeleton;
    }
}