PONOSAdvent Calendar 2024

Unityでシェーダーに渡すパラメーターを頂点情報に含めることでsetpass callを減らす試み

Last updated at 2024-12-05Posted at 2024-12-05

この記事はPONOS Advent Calendar 2024 6日目の記事です。
前回は@nissy_gpさんでした。

今回使用する画像はいらすとやからお借りしています。

目的

Unityでオブジェクトを大量に描画するときに、できるだけ単一のマテリアルで描画することでsetpass callを減らし描画負荷を減らすことを試みるかと思います。
しかし「オブジェクトごとに別パラメーターで描画したい」みたいなことがあると思います。そうなるとマテリアルにmaterial.SetFloat("key" value)のような処理でパラメーターを渡してやる必要があり、それによってマテリアルが複製されSetPass Callsが増えて負荷が増えます。
今回の記事はSetPass Callsを増やさずにシェーダーにパラメーターを渡すことを目的としています。

方法

いくつか方法が考えられると思いますが今回は頂点情報(UV)で渡す方法を考えます。

実装

今回は単一の画像を描画するSprite Rendererのようなものを実装します。
以下のように同じ画像のパラメーター違いを3つ描画します。

普通に描画
グレースケールで描画
色を変更して描画

実現のためにはSpriteを描画するC#スクリプトの実装とそれに対応するシェーダーの追加が必要です。

C#スクリプトの実装

コード全ては長いので折り畳みで書いてあります。
頂点4つのメッシュを作成し描画するコンポーネントです。

コード全文

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

namespace Block
{
    [RequireComponent(typeof(MeshRenderer), typeof(MeshFilter))]
    [ExecuteAlways]
    public class CustomSpriteRenderer : MonoBehaviour
    {
        static readonly Vector3[] MeshVertices = new Vector3[] {
                new Vector3(-1, -1, 0),
                new Vector3(-1, 1, 0),
                new Vector3(1, 1, 0),
                new Vector3(1, -1, 0),
            };
        static readonly Vector2[] UV = new Vector2[] {
                new Vector2(0, 0),
                new Vector2(0, 1),
                new Vector2(1, 1),
                new Vector2(1, 0),
            };
        static readonly int[] VertexIndeces = new int[] { 0, 1, 2, 3, 0, 2 };
        
        [SerializeField]
        Material material = null;
        
        bool isDirty = false;

        [field: SerializeField]
        public Color Color { get; private set; } = Color.white;

        [field: SerializeField]
        public bool IsGrayScale { get; private set; } = false;

        Mesh mesh = null;
        
        MeshRenderer meshRenderer = null;

        List<Color> colors = new() { Color.white, Color.white, Color.white, Color.white };

        List<Vector2> uv0 = UV.ToList(); 

        List<Vector4> uv1 = new (){ Vector4.zero, Vector4.zero, Vector4.zero, Vector4.zero, };
        
        void OnEnable()
        {
            TryGetComponent(out meshRenderer);
            if (!Application.isPlaying && mesh != null)
            {
                DestroyImmediate(mesh);
                mesh = null;
            }

            if (mesh == null)
            {
                mesh = GenerateMesh();
                meshRenderer.sharedMaterials = new Material[] { material };
                TryGetComponent<MeshFilter>(out var meshFilter);
                meshFilter.sharedMesh = mesh;
                SetColor(Color);
                SetGrayScale(IsGrayScale);
                mesh.SetUVs(0, uv0);
                mesh.SetUVs(1, uv1);
                mesh.SetColors(colors);
            }
            
            isDirty = true;
        }
        
        void OnValidate()
        {
            if (meshRenderer.sharedMaterial != material)
            {
                meshRenderer.sharedMaterial = material;
            }
            SetColor(Color);
            SetGrayScale(IsGrayScale);
            isDirty = true;
        }
        
        void LateUpdate()
        {
            if (!isDirty) { return; }
            mesh.SetColors(colors);
            mesh.SetUVs(0, uv0);
            mesh.SetUVs(1, uv1);
        }

        static Mesh GenerateMesh()
        {
            var mesh = new Mesh();
            mesh.vertices = MeshVertices;
            mesh.subMeshCount = 1;
            mesh.SetTriangles(VertexIndeces, 0);
            return mesh;
        }
        
        public void SetColor(Color color)
        {
            Color = color;
            colors[0] = color;
            colors[1] = color;
            colors[2] = color;
            colors[3] = color;
            isDirty = true;
        }
        
        public void SetGrayScale(bool isGrayScale)
        {
            IsGrayScale = isGrayScale;
            var v = uv1[0];
            v.x = isGrayScale ? 1.0f : 0.0f;
            uv1[0] = v;
            uv1[1] = v;
            uv1[2] = v;
            uv1[3] = v;
            isDirty = true;
        }
    }
}

エディタで実行しなくても確認できるようにするために処理が複雑になっている箇所があります。
重要なのは以下の処理です。

        void LateUpdate()
        {
            if (!isDirty) { return; }
            mesh.SetColors(colors);
            mesh.SetUVs(0, uv0);
            mesh.SetUVs(1, uv1);
        }

        public void SetColor(Color color)
        {
            Color = color;
            colors[0] = color;
            colors[1] = color;
            colors[2] = color;
            colors[3] = color;
            isDirty = true;
        }

        public void SetGrayScale(bool isGrayScale)
        {
            // uv.xにグレースケール化するかどうかの値をセットする
            IsGrayScale = isGrayScale;
            var v = uv1[0];
            v.x = isGrayScale ? 1.0f : 0.0f;
            uv1[0] = v;
            uv1[1] = v;
            uv1[2] = v;
            uv1[3] = v;
            isDirty = true;
        }