HK_LongZhiMeng/Assets/Scripts/EditMesh/ModelMeshEditor.cs

using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using System.Text;
using System;
using UnityEngine.AddressableAssets;

public class ModelMeshEditor : MonoBehaviour
{
    public bool isCreatePointOnStart;
    //控制点的大小
    public float pointScale = 1.0f;
    private float lastPointScale = 1.0f;

    public Mesh mesh;
    private MeshCollider meshCollider;

    //顶点列表
    public List<Vector3> positionList = new List<Vector3>();

    //顶点控制物体列表
    public List<GameObject> positionObjList = new List<GameObject>();
    
    /// <summary>
    /// key:顶点字符串
    /// value:顶点在列表中的位置
    /// </summary>
    public Dictionary<string, List<int>> pointmap = new Dictionary<string, List<int>>();

   
    // Use this for initialization
    void Start()
    {
        lastPointScale = pointScale;
        if(isCreatePointOnStart) CreateEditorPoint();
        //CreateEditorPoint();
    }


    //删除控制点
    public void DeleteEditorPoint()
    {
        foreach(var point in positionObjList)
        {
            Destroy(point);
        }

        positionObjList.Clear();
        positionList.Clear();
        pointmap.Clear();
    }

    //创建控制点
    public void CreateEditorPoint()
    {
        if (GetComponent<MeshFilter>())
            mesh = GetComponent<MeshFilter>().mesh;
        else if (transform.Find("Scene/Plane"))
        {
            mesh = transform.Find("Scene/Plane").GetComponent<MeshFilter>().mesh;
        }
        
            
        meshCollider = GetComponent<MeshCollider>();
        positionList = new List<Vector3>(mesh.vertices);

        for (int i = 0; i < mesh.vertices.Length; i++)
        {
            string vstr = Vector2String(mesh.vertices[i]);

            if (!pointmap.ContainsKey(vstr))
            {
                pointmap.Add(vstr, new List<int>());
            }
            pointmap[vstr].Add(i);
        }

        foreach (string key in pointmap.Keys)
        {

            //GameObject editorpoint = (GameObject)Resources.Load("Props/MeshEditorPoint");
            var editorpoint = CreatePoint();
            editorpoint.transform.parent = transform;
            editorpoint.transform.localPosition = String2Vector(key);
            editorpoint.transform.localScale = new Vector3(pointScale, pointScale, pointScale);

            MeshEditorPoint editorPoint = editorpoint.GetComponent<MeshEditorPoint>();
            editorPoint.onMove = PointMove;
            editorPoint.pointid = key;

            positionObjList.Add(editorpoint);

        }
    }

    /// <summary>
    /// 动态生成控制点
    /// </summary>
    /// <returns></returns>
    private GameObject CreatePoint()
    {
        GameObject point = GameObject.CreatePrimitive(PrimitiveType.Sphere);
        point.GetComponent<MeshRenderer>().material = new Material(Shader.Find("Unlit/Color"));
        point.GetComponent<MeshRenderer>().material.color = Color.green;

        point.AddComponent<DragObject>();
        point.AddComponent<MeshEditorPoint>();

        return point;

    }

    //顶点物体被移动时调用此方法
    public void PointMove(string pointid, Vector3 position)
    {
        if (!pointmap.ContainsKey(pointid))
        {
            return;
        }

        List<int> _list = pointmap[pointid];

        for (int i = 0; i < _list.Count; i++)
        {
            positionList[_list[i]] = position;
        }

        mesh.vertices = positionList.ToArray();
        mesh.RecalculateNormals();
        meshCollider.sharedMesh = mesh;
    }

    // Update is called once per frame
    void Update()
    {
        //检测控制点尺寸是否改变
        if (Math.Abs(lastPointScale - pointScale) > 0.1f)
        {
            lastPointScale = pointScale;
            for (int i = 0; i < positionObjList.Count; i++)
            {
                positionObjList[i].transform.localScale = new Vector3(pointScale, pointScale, pointScale);
            }
        }
    }

    string Vector2String(Vector3 v)
    {
        StringBuilder str = new StringBuilder();
        str.Append(v.x).Append(",").Append(v.y).Append(",").Append(v.z);
        return str.ToString();
    }

    Vector3 String2Vector(string vstr)
    {
        try
        {
            string[] strings = vstr.Split(',');
            return new Vector3(float.Parse(strings[0]), float.Parse(strings[1]), float.Parse(strings[2]));
        }
        catch (Exception e)
        {
            Debug.LogError(e.ToString());
            return Vector3.zero;
        }
    }
}