CrowdControl/Assets/3rd/Epic Toon FX/Demo/Scripts/ETFXEffectController.cs

using UnityEngine;
using System.Collections;
using UnityEngine.UI;

namespace EpicToonFX
{
    public class ETFXEffectController : MonoBehaviour
    {
        public GameObject[] effects;
        private int effectIndex = 0;

        [Space(10)]

        [Header("Spawn Settings")]
        public bool disableLights = true;
        public bool disableSound = true;
        public float startDelay = 0.2f;
        public float respawnDelay = 0.5f;
        public bool slideshowMode = false;
        public bool autoRotation = false;
        [Range(0.001f, 0.5f)]
        public float autoRotationSpeed = 0.1f;

        private GameObject currentEffect;
        private Text effectNameText;
        private Text effectIndexText;

        private ETFXMouseOrbit etfxMouseOrbit;

        private void Awake()
        {
            effectNameText = GameObject.Find("EffectName").GetComponent<Text>();
            effectIndexText = GameObject.Find("EffectIndex").GetComponent<Text>();

            etfxMouseOrbit = Camera.main.GetComponent<ETFXMouseOrbit>();
            etfxMouseOrbit.etfxEffectController = this;
        }

        void Start()
        {
            etfxMouseOrbit = Camera.main.GetComponent<ETFXMouseOrbit>();
            etfxMouseOrbit.etfxEffectController = this;

            Invoke("InitializeLoop", startDelay);
        }

        void Update()
        {
            if (Input.GetKeyDown(KeyCode.RightArrow) || Input.GetKeyDown(KeyCode.D))
            {
                NextEffect();
            }

            if (Input.GetKeyDown(KeyCode.A) || Input.GetKeyDown(KeyCode.LeftArrow))
            {
                PreviousEffect();
            }
        }

        private void FixedUpdate()
        {
            if (autoRotation)
            {
                etfxMouseOrbit.SetAutoRotationSpeed(autoRotationSpeed);

                if (!etfxMouseOrbit.isAutoRotating)
                    etfxMouseOrbit.InitializeAutoRotation();
            }
        }

        public void InitializeLoop()
        {
            StartCoroutine(EffectLoop());
        }

        public void NextEffect()
        {
            if (effectIndex < effects.Length - 1)
            {
                effectIndex++;
            }
            else
            {
                effectIndex = 0;
            }

            CleanCurrentEffect();
        }

        public void PreviousEffect()
        {
            if (effectIndex > 0)
            {
                effectIndex--;
            }
            else
            {
                effectIndex = effects.Length - 1;
            }

            CleanCurrentEffect();
        }

        private void CleanCurrentEffect()
        {
            StopAllCoroutines();

            if (currentEffect != null)
            {
                Destroy(currentEffect);
            }

            StartCoroutine(EffectLoop());
        }

        private IEnumerator EffectLoop()
        {
            //Instantiating effect
            GameObject effect = Instantiate(effects[effectIndex], transform.position, Quaternion.identity);
            currentEffect = effect;

            if (disableLights && effect.GetComponent<Light>())
            {
                effect.GetComponent<Light>().enabled = false;
            }

            if (disableSound && effect.GetComponent<AudioSource>())
            {
                effect.GetComponent<AudioSource>().enabled = false;
            }

            //Update GUIText with effect name
            effectNameText.text = effects[effectIndex].name;
            effectIndexText.text = (effectIndex + 1) + " of " + effects.Length;

            ParticleSystem particleSystem = effect.GetComponent<ParticleSystem>();

            while (true)
            {
                yield return new WaitForSeconds(particleSystem.main.duration + respawnDelay);

                if (!slideshowMode)
                {
                    if (!particleSystem.main.loop)
                    {
                        currentEffect.SetActive(false);
                        currentEffect.SetActive(true);
                    }
                }
                else
                {
                    //Double delay for looping effects
                    if (particleSystem.main.loop)
                    {
                        yield return new WaitForSeconds(respawnDelay);
                    }

                    NextEffect();
                }
            }
        }
    }
}