HDRobotics/Assets/Oculus/SampleFramework/Usage/HandsTrainExample/Scripts/TrainCrossingController.cs

/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 * All rights reserved.
 *
 * Licensed under the Oculus SDK License Agreement (the "License");
 * you may not use the Oculus SDK except in compliance with the License,
 * which is provided at the time of installation or download, or which
 * otherwise accompanies this software in either electronic or hard copy form.
 *
 * You may obtain a copy of the License at
 *
 * https://developer.oculus.com/licenses/oculussdk/
 *
 * Unless required by applicable law or agreed to in writing, the Oculus SDK
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */


using System.Collections;
using UnityEngine;
using UnityEngine.Assertions;

namespace OculusSampleFramework
{
    public class TrainCrossingController : MonoBehaviour
    {
        [SerializeField] private AudioSource _audioSource = null;
        [SerializeField] private AudioClip[] _crossingSounds = null;
        [SerializeField] private MeshRenderer _lightSide1Renderer = null;
        [SerializeField] private MeshRenderer _lightSide2Renderer = null;
        [SerializeField] private SelectionCylinder _selectionCylinder = null;

        private Material _lightsSide1Mat;
        private Material _lightsSide2Mat;
        private int _colorId = Shader.PropertyToID("_Color");

        private Coroutine _xingAnimationCr = null;
        private InteractableTool _toolInteractingWithMe = null;

        private void Awake()
        {
            Assert.IsNotNull(_audioSource);
            Assert.IsNotNull(_crossingSounds);
            Assert.IsNotNull(_lightSide1Renderer);
            Assert.IsNotNull(_lightSide2Renderer);
            Assert.IsNotNull(_selectionCylinder);

            _lightsSide1Mat = _lightSide1Renderer.material;
            _lightsSide2Mat = _lightSide2Renderer.material;
        }

        private void OnDestroy()
        {
            if (_lightsSide1Mat != null)
            {
                Destroy(_lightsSide1Mat);
            }

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

        public void CrossingButtonStateChanged(InteractableStateArgs obj)
        {
            bool inActionState = obj.NewInteractableState == InteractableState.ActionState;
            if (inActionState)
            {
                ActivateTrainCrossing();
            }

            _toolInteractingWithMe = obj.NewInteractableState > InteractableState.Default ? obj.Tool : null;
        }

        private void Update()
        {
            if (_toolInteractingWithMe == null)
            {
                _selectionCylinder.CurrSelectionState = SelectionCylinder.SelectionState.Off;
            }
            else
            {
                _selectionCylinder.CurrSelectionState = (
                    _toolInteractingWithMe.ToolInputState == ToolInputState.PrimaryInputDown ||
                    _toolInteractingWithMe.ToolInputState == ToolInputState.PrimaryInputDownStay)
                    ? SelectionCylinder.SelectionState.Highlighted
                    : SelectionCylinder.SelectionState.Selected;
            }
        }

        private void ActivateTrainCrossing()
        {
            int maxSoundIndex = _crossingSounds.Length - 1;
            var audioClip = _crossingSounds[(int)(Random.value * maxSoundIndex)];
            _audioSource.clip = audioClip;
            _audioSource.timeSamples = 0;
            _audioSource.Play();
            if (_xingAnimationCr != null)
            {
                StopCoroutine(_xingAnimationCr);
            }

            _xingAnimationCr = StartCoroutine(AnimateCrossing(audioClip.length * 0.75f));
        }

        private IEnumerator AnimateCrossing(float animationLength)
        {
            ToggleLightObjects(true);

            float animationEndTime = Time.time + animationLength;

            float lightBlinkDuration = animationLength * 0.1f;
            float lightBlinkStartTime = Time.time;
            float lightBlinkEndTime = Time.time + lightBlinkDuration;
            Material lightToBlinkOn = _lightsSide1Mat;
            Material lightToBlinkOff = _lightsSide2Mat;
            Color onColor = new Color(1.0f, 1.0f, 1.0f, 1.0f);
            Color offColor = new Color(1.0f, 1.0f, 1.0f, 0.0f);

            while (Time.time < animationEndTime)
            {
                float t = (Time.time - lightBlinkStartTime) / lightBlinkDuration;
                lightToBlinkOn.SetColor(_colorId, Color.Lerp(offColor, onColor, t));
                lightToBlinkOff.SetColor(_colorId, Color.Lerp(onColor, offColor, t));

                // switch which lights blink on and off when time runs out
                if (Time.time > lightBlinkEndTime)
                {
                    Material temp = lightToBlinkOn;
                    lightToBlinkOn = lightToBlinkOff;
                    lightToBlinkOff = temp;
                    lightBlinkStartTime = Time.time;
                    lightBlinkEndTime = Time.time + lightBlinkDuration;
                }

                yield return null;
            }

            ToggleLightObjects(false);
        }

        private void AffectMaterials(Material[] materials, Color newColor)
        {
            foreach (var material in materials)
            {
                material.SetColor(_colorId, newColor);
            }
        }

        private void ToggleLightObjects(bool enableState)
        {
            _lightSide1Renderer.gameObject.SetActive(enableState);
            _lightSide2Renderer.gameObject.SetActive(enableState);
        }
    }
}