StationRadioTrackEditor.cs

using UnityEngine;
 
namespace RyleRadio.Editor
{
    using RyleRadio.Tracks;
 
#if UNITY_EDITOR
    using UnityEditor;
 
    /// <summary>
    /// The editor for a \ref StationRadioTrack
    /// <br><br>This mainly exists so that we can reset any newly added tracks in the station- much the same as \ref RadioDataEditor
    /// <br><b>See: </b>\ref StationRadioTrack
    /// </summary>
    [CustomPropertyDrawer(typeof(StationRadioTrack))]
    public class StationRadioTrackEditor : PropertyDrawer
    {
        /// The last recorded number of tracks in this station
        private int lastTrackWSize = -1;
 
        /// Whether or not the random threshold is currently visible in the inspector
        private bool isShowingThreshold = false;
 
        /// Whether or not this track is showing its child tracks- optional variable for cleanliness
        private bool showTrackInParent = true;
 
        /// <summary>
        /// Draws the track, its child tracks, and other variables, e.g: randomisation settings
        /// </summary>
        public override void OnGUI(Rect _position, SerializedProperty _property, GUIContent _label)
        {
            SerializedProperty trackWs = _property.FindPropertyRelative("stationTrackWs"); // the tracks
 
            SerializedProperty isRandom = _property.FindPropertyRelative("randomSequence"); // whether or not the station is random
            SerializedProperty threshold = _property.FindPropertyRelative("thresholdBeforeRepeats"); // the random threshold if it is
 
            _position.position += new Vector2(24, 0);
            _position.width -= 24;
 
            // visuals for randomization settings
            Rect dropdownRect = new(_position.position - new Vector2(24, 0), new Vector2(0, 20));
            Rect isRandomRect = dropdownRect;
 
            // put the threshold slider under the bool
            isRandomRect.x += 12;
 
            Rect thresholdSliderRect = isRandomRect;
 
            isRandomRect.width = _position.width;
            isRandomRect.y += 20;
 
            // put the threshold slider under the bool
            thresholdSliderRect.y += 20 + isRandomRect.height;
            thresholdSliderRect.width = _position.width;
 
            // if the station editor was just loaded, assign the default track count
            if (lastTrackWSize < 0)
                lastTrackWSize = trackWs.arraySize;
 
            showTrackInParent = EditorGUI.Foldout(dropdownRect, showTrackInParent, _label);
 
            if (showTrackInParent)
            { 
                // if the station is in a random order,
                if (EditorGUI.Toggle(isRandomRect, new GUIContent("Random Sequence"), isRandom.boolValue))
                {
                    isRandom.boolValue = true;
                    isShowingThreshold = true;
 
                    // show the threshold for that order
                    EditorGUI.Slider(thresholdSliderRect, threshold, 0, 1, new GUIContent("Threshold Before Repeats"));
                }
                else
                {
                    isRandom.boolValue = false;
                    isShowingThreshold = false;
                }
 
                // add the height of the random settings to the property height
                _position.position = new Vector2(
                    _position.position.x, 
                    _position.position.y + dropdownRect.height + isRandomRect.height + (
                        isShowingThreshold 
                        ? thresholdSliderRect.height + 4 
                        : 0
                    )
                );
 
                // show the list of tracks
                EditorGUI.PropertyField(_position, trackWs, new GUIContent("Station Tracks"), true);
 
                // if the station was changed
                if (GUI.changed)
                {
                    // and a new track was added
                    if (trackWs.arraySize > lastTrackWSize)
                    {
                        // get the new element
                        SerializedProperty newElement = trackWs.GetArrayElementAtIndex(trackWs.arraySize - 1);
 
                        // get the stored track and track eventType
                        SerializedProperty track = newElement.FindPropertyRelative("track");
                        SerializedProperty trackType = newElement.FindPropertyRelative("trackType");
 
                        // get the stored gain
                        SerializedProperty gain = newElement.FindPropertyRelative("gain");
 
                        // create an empty track for the station
                        track.managedReferenceValue = StationRadioTrackWrapper.CreateTrackEditor(trackType.stringValue);
 
                        // reset the track's gain
                        gain.floatValue = 100;
 
                        // store the new track list size
                        lastTrackWSize++;
                    }
                    // if a track was removed
                    else if (trackWs.arraySize < lastTrackWSize)
                    { 
                        // store the new track list size
                        lastTrackWSize--; 
                    }
                }
 
            }
            else
            {
                // add the height of the random settings to the property height
                _position.position = new Vector2(
                    _position.position.x,
                    _position.position.y + dropdownRect.height
                );
            }
 
                // apply any inspector changes
            _property.serializedObject.ApplyModifiedProperties();
 
        }
 
        /// <summary>
        /// Gets the total height of this track in the inspector, including child tracks
        /// </summary>
        public override float GetPropertyHeight(SerializedProperty property, GUIContent label)
        {
            if (showTrackInParent)
            {
                // if the random options are shown, add height accordingly
                float randomOptionsHeight = 48 + (isShowingThreshold ? 20 : 0);
 
                // get the height of the contained track list
                float listHeight = EditorGUI.GetPropertyHeight(property.FindPropertyRelative("stationTrackWs"));
 
                // combine the heights
                return randomOptionsHeight + listHeight;
            }
            else
            {
                return 18;
            }
        }
    }
#endif
 
}