VRCycling/Assets/BOXOPHOBIC/Utils/Editor/StyledMaterial/StyledButtonDrawer.cs

// Cristian Pop - https://boxophobic.com/

using UnityEngine;
using UnityEditor;
using System;

namespace Boxophobic.StyledGUI
{
    public class StyledButtonDrawer : MaterialPropertyDrawer
    {
        public string text;
        public string target = "";
        public float value = 1;
        public float top;
        public float down;

        public StyledButtonDrawer(string text)
        {
            this.text = text;
            this.value = 1;
            this.top = 0;
            this.down = 0;
        }

        public StyledButtonDrawer(string text, float value, float top, float down)
        {
            this.text = text;
            this.value = value;
            this.top = top;
            this.down = down;
        }

        public StyledButtonDrawer(string text, string target, float value, float top, float down)
        {
            this.text = text;
            this.target = target;
            this.value = value;
            this.top = top;
            this.down = down;
        }

        public override void OnGUI(Rect position, MaterialProperty prop, String label, MaterialEditor materialEditor)
        {
            Material material = materialEditor.target as Material;

            GUILayout.Space(top);

            if (GUILayout.Button(text))
            {
                if (target == "")
                {
                    prop.floatValue = value;
                }
                else
                {
                    if (material.HasProperty(target))
                    {
                        material.SetFloat(target, value);
                    }
                }
            }

            GUILayout.Space(down);
        }

        public override float GetPropertyHeight(MaterialProperty prop, string label, MaterialEditor editor)
        {
            return -2;
        }
    }
}