ArabDesert/Assets/harry_t/quest_stuff.cs

using System.Collections;
using System.Collections.Generic;
using UnityEditor;
using UnityEngine;

[System.Serializable]
public class mat_replacement
{
	[SerializeField] public string name;
	[SerializeField] public bool shown;

	[SerializeField] public Shader shader_default;
	[SerializeField] public Shader shader_quest;

	[SerializeField] public List< Material > materials;
}

[System.Serializable]
public class obj_toggly
{
	[SerializeField] public string name;
	[SerializeField] public bool shown;

	[SerializeField] public bool pc;
	[SerializeField] public bool quest;

	[SerializeField] public List< GameObject > objs;
}

public enum EQuestStuffUI
{
   k_EQuestStuffUI_noaction,
   k_EQuestStuffUI_set_pc,
   k_EQuestStuffUI_set_quest
};

[System.Serializable]
public struct quest_stuff_data
{
   [SerializeField] public List< mat_replacement > replacements;
	[SerializeField] public List< obj_toggly > objs;
}

public class quest_stuff : MonoBehaviour
{
   [SerializeField]
   public quest_stuff_data data;

   public static void ApplyReplacement( ref quest_stuff_data qst, bool target_pc )
   {
      if( qst.replacements != null )
      {
         for( int i = 0; i < qst.replacements.Count; i ++ )
         {
            mat_replacement replacement = qst.replacements[ i ];

            if( replacement.shader_default == null || replacement.shader_quest == null )
               continue;

            for( int j = 0; j < replacement.materials.Count; j ++ )
            {
               if( replacement.materials[ j ] != null )
               {
                  replacement.materials[ j ].shader = target_pc? replacement.shader_default: replacement.shader_quest;
               }
            }
         }
      }

      if( qst.objs != null )
      {
         for( int i = 0; i < qst.objs.Count; i ++ )
         {
            obj_toggly toggle = qst.objs[ i ];
            for( int j = 0; j < toggle.objs.Count; j ++ )
            {
               if( toggle.objs[ j ] != null )
               {
                  toggle.objs[ j ].SetActive( toggle.pc == target_pc || toggle.quest == !target_pc );
               }
            }
         }
      }
   }

   public static void ApplyReplacement( ref quest_stuff_data qst, EQuestStuffUI action )
   {
      if( action == EQuestStuffUI.k_EQuestStuffUI_set_pc )
      {
         ApplyReplacement( ref qst, true );
      }
      if( action == EQuestStuffUI.k_EQuestStuffUI_set_quest )
      {
         ApplyReplacement( ref qst, false );
      }
   }

   #if UNITY_EDITOR 

   public static EQuestStuffUI DrawQuestStuffGUI( ref quest_stuff_data qst )
   {
      EQuestStuffUI action = EQuestStuffUI.k_EQuestStuffUI_noaction;

      if( qst.replacements == null )
         qst.replacements = new List<mat_replacement>();

      if( qst.objs == null )
         qst.objs = new List<obj_toggly>();

      for( int i = 0; i < qst.replacements.Count; i ++ )
      {
         mat_replacement replacement = qst.replacements[ i ];

         replacement.shown = EditorGUILayout.Foldout( replacement.shown, replacement.name, true, EditorStyles.foldout );

         if( replacement.shown )
         {
            EditorGUI.indentLevel ++;
            replacement.name = EditorGUILayout.DelayedTextField( replacement.name );

            if( GUILayout.Button( "delete" ) )
            {
               qst.replacements.RemoveAt( i );
               continue;
            }

            int newCount = Mathf.Max( 0, EditorGUILayout.DelayedIntField( "size", replacement.materials.Count ));
            while( newCount < replacement.materials.Count)
               replacement.materials.RemoveAt( replacement.materials.Count - 1 );
            while( newCount > replacement.materials.Count)
               replacement.materials.Add( null );

            replacement.shader_default = (Shader)EditorGUILayout.ObjectField( replacement.shader_default, typeof( Shader ), false );
            replacement.shader_quest = (Shader)EditorGUILayout.ObjectField( replacement.shader_quest, typeof( Shader ), false );
 
            for( int j = 0; j < replacement.materials.Count; j++)
            {
               replacement.materials[ j ] = (Material)EditorGUILayout.ObjectField( replacement.materials[ j ], typeof(Material), false );
            }
            EditorGUI.indentLevel --;
         }
      }

      if( GUILayout.Button( "add shader replacement" ) )
      {
         mat_replacement replacement = new mat_replacement();
         replacement.materials = new List<Material>();
         qst.replacements.Add( replacement );
      } 

      for( int i = 0; i < qst.objs.Count; i ++ )
      {
         obj_toggly toggle = qst.objs[ i ];

         toggle.shown = EditorGUILayout.Foldout( toggle.shown, toggle.name, true, EditorStyles.foldout );

         if( toggle.shown )
         {
            EditorGUI.indentLevel ++;
            toggle.name = EditorGUILayout.DelayedTextField( toggle.name );

            if( GUILayout.Button( "delete" ) )
            {
               qst.objs.RemoveAt( i );
               continue;
            }

            int newCount = Mathf.Max( 0, EditorGUILayout.DelayedIntField( "size", toggle.objs.Count ));
            while( newCount < toggle.objs.Count)
               toggle.objs.RemoveAt( toggle.objs.Count - 1 );
            while( newCount > toggle.objs.Count)
               toggle.objs.Add( null );

            EditorGUILayout.BeginHorizontal();
            toggle.pc = EditorGUILayout.Toggle( "PC", toggle.pc );
            toggle.quest = EditorGUILayout.Toggle( "Quest", toggle.quest );
            EditorGUILayout.EndHorizontal();

            for( int j = 0; j < toggle.objs.Count; j++)
            {
               toggle.objs[ j ] = (GameObject)EditorGUILayout.ObjectField( toggle.objs[ j ], typeof(GameObject), true  );
            }
            EditorGUI.indentLevel --;
         }
      }

      if( GUILayout.Button( "add toggle" ) )
      {
         obj_toggly toggle = new obj_toggly();
         toggle.objs = new List<GameObject>();
         qst.objs.Add( toggle );
      }

      EditorGUILayout.BeginHorizontal();

      if( GUILayout.Button( "Set PC" ) )
      {
         ApplyReplacement( ref qst, true );
         action = EQuestStuffUI.k_EQuestStuffUI_set_pc;
      }

      if( GUILayout.Button( "Set Quest" ) )
      {
         ApplyReplacement( ref qst, false );

         action = EQuestStuffUI.k_EQuestStuffUI_set_quest;
      }

      EditorGUILayout.EndHorizontal();

      return action;
   }

   #endif
}