using UnityEngine;
using System.Collections;
using System.Collections.Generic;
 
public enum AvatarPart
{
    helmet,
    chest,
    shoulders,
    gloves,
    boots,
}
 
// 人物换装
public class ActorAvatar : MonoBehaviour
{
    // 换装的部件信息
    public class AvatarInfo
    {
        public string partName;
        public GameObject defaultPart;
        public GameObject avatarPart;
    }
 
    protected int _bodyModelId;
    protected GameObject _body;         // 基础模型动画
    protected Dictionary<string, AvatarInfo> _avatarInfo = new Dictionary<string, AvatarInfo>();        // 换装信息
 
    private List<int> _avatarLoadQueue = new List<int>();
 
    void Start()
    {
    }
 
    void Update()
    {
    }
 
    // 创建模型
    public void LoadModel(int modelId)
    {
        _bodyModelId = modelId;
        ResourceMgr.Instance.LoadModel(modelId, (GameObject obj) =>
        {
            _body = obj;
 
            // 换装请求
            if (_avatarLoadQueue.Count > 0) {
                foreach (var avatar in _avatarLoadQueue) {
                    LoadAvatar(avatar);
                }
                _avatarLoadQueue.Clear();
            }
        }, true);
    }
 
    // 给人物换装
    public void LoadAvatar(int avatarId)
    {
        // 如果还没有加载完基础模型，则等待
        if (_body == null) {
            _avatarLoadQueue.Add(avatarId);
            return;
        }
 
        AvatarData adata = DataMgr.Instance.GetAvatarData(avatarId);
        ResourceMgr.Instance.LoadModel(adata.model, (GameObject obj) => {
            ChangeAvatar(obj, adata.addpart);
        });
    }
 
    // 替换部件
    public void ChangeAvatar(GameObject avatarModel, string partName)
    {
        // 先卸载当前部件
        AvatarInfo currentInfo;
        if (_avatarInfo.TryGetValue(partName, out currentInfo)) {
            if (currentInfo.avatarPart != null) {
                Destroy(currentInfo.avatarPart);
                currentInfo.avatarPart = null;
            }
 
            if (currentInfo.defaultPart != null) {
                currentInfo.defaultPart.SetActive(true);
            }
        }
 
        // avatarModel是一个resource，并没有实例化
        if (avatarModel == null) {
            return;
        }
 
        // 需要替换的部件
        Transform avatarPart = GetPart(avatarModel.transform, partName);
        if (avatarPart == null) {
            Debug.LogError(string.Format("Avatar Part Not Found: ", partName));
            return;
        }
 
        // 将原始部件隐藏
        Transform bodyPart = GetPart(_body.transform, partName);
        if (bodyPart != null) {
            bodyPart.gameObject.SetActive(false);
        }
 
        // 设置到body上的新物件
        GameObject newPart = new GameObject(partName);
        newPart.transform.parent = _body.transform;
        SkinnedMeshRenderer newPartRender = newPart.AddComponent<SkinnedMeshRenderer>();
        SkinnedMeshRenderer avatarRender = avatarPart.GetComponent<SkinnedMeshRenderer>();
 
        // 刷新骨骼模型数据
        SetBones(newPart, avatarPart.gameObject, _body);
        newPartRender.sharedMesh = avatarRender.sharedMesh;
        newPartRender.sharedMaterials = avatarRender.sharedMaterials;
 
        // 记录换装信息
        AvatarInfo info = new AvatarInfo();
        info.partName = partName;
        if (bodyPart != null) {
            info.defaultPart = bodyPart.gameObject;
        } else {
            info.defaultPart = null;
        }
 
        info.avatarPart = newPart;
        _avatarInfo[partName] = info;
    }
 
     // 递归遍历子物体
    public static Transform GetPart(Transform t, string searchName)
    {
        foreach (Transform c in t) {
            string partName = c.name.ToLower();
            
            if (partName.IndexOf(searchName) != -1) {
                return c;
            } else {
                Transform r = GetPart(c, searchName);
                if (r != null) {
                    return r;
                }
            }
        }
        return null;
    }
 
    public static Transform FindChild(Transform t, string searchName)
    {
        foreach (Transform c in t) {
            string partName = c.name;
            if (partName == searchName) {
                return c;
            } else {
                Transform r = FindChild(c, searchName);
                if (r != null) {
                    return r;
                }
            }
        }
        return null;
    }
 
    // 刷新骨骼数据   将root物体的bodyPart骨骼更新为avatarPart
    public static void SetBones(GameObject goBodyPart, GameObject goAvatarPart, GameObject root)
    {
        var bodyRender = goBodyPart.GetComponent<SkinnedMeshRenderer>();
        var avatarRender = goAvatarPart.GetComponent<SkinnedMeshRenderer>();
        var myBones = new Transform[avatarRender.bones.Length];
        for (var i = 0; i < avatarRender.bones.Length; i++) {
            myBones[i] = FindChild(root.transform, avatarRender.bones[i].name);
        }
        bodyRender.bones = myBones;
    }
 
}