com.unity.uiwidgets/com.unity.uiwidgets/Runtime/ui/renderer/cmdbufferCanvas/rendering/canvas_computebuffer_utils.cs

using System.Collections.Generic;
using System.Runtime.InteropServices;
using UnityEngine;

namespace Unity.UIWidgets.uiOld{
    public partial class PictureFlusher {
        
        struct uiVertex
        {
            public Vector2 position;
            public Vector2 uv;
        }

        ComputeBuffer _computeBuffer;
        List<uiVertex> _vertices;

        ComputeBuffer _indexBuffer;
        List<int> _indices;
        
        int _startVertex;
        int _startIndex;

        public const int COMPUTE_BUFFER_MAX_ITEM_NUM = 1024 * 1024;   // maxsize = 1M vertex/index

        bool supportComputeBuffer {
            get { return CanvasShader.supportComputeBuffer; }
        }

        void _releaseComputeBuffer() {
            if (!supportComputeBuffer) {
                return;
            }

            if (_computeBuffer == null) {
                return;
            }

            _computeBuffer.Dispose();
            _indexBuffer.Dispose();
            _vertices = null;
            _indices = null;
            _computeBuffer = null;
            _indexBuffer = null;
        }

        void _initComputeBuffer() {
            var stride = Marshal.SizeOf(typeof(uiVertex));
            var strideIndex = Marshal.SizeOf(typeof(int));
            _computeBuffer = new ComputeBuffer(COMPUTE_BUFFER_MAX_ITEM_NUM, stride);
            _vertices = new List<uiVertex>();
                
            _indexBuffer = new ComputeBuffer(COMPUTE_BUFFER_MAX_ITEM_NUM, strideIndex);
            _indices = new List<int>();
        }

        void _resetComputeBuffer() {
            if (!supportComputeBuffer) return;

            if (_computeBuffer == null) {
                _initComputeBuffer();
            }
            
            _vertices.Clear();
            _indices.Clear();
            _startVertex = 0;
            _startIndex = 0;
        }

        void _bindComputeBuffer() {
            if (!supportComputeBuffer) return;
            
            _computeBuffer.SetData(_vertices);
            _indexBuffer.SetData(_indices);
        }

        void _addMeshToComputeBuffer(List<Vector3> vertex, List<Vector2> uv, List<int> triangles) {
            if (!supportComputeBuffer) return;
            
            _startVertex = _vertices.Count;
            _startIndex = _indices.Count;

            var hasUv = uv != null;

            for (int i = 0; i < vertex.Count; i++) {
                _vertices.Add(new uiVertex {
                    position = new Vector2(vertex[i].x, vertex[i].y),
                    uv = hasUv ? uv[i] : Vector2.zero
                });
            }

            foreach (var triangleId in triangles) {
                _indices.Add(triangleId + _startVertex);
            }
        }
    }
}