102 lines
3.3 KiB
C#
102 lines
3.3 KiB
C#
using System.Collections.Generic;
|
|
using UnityEngine;
|
|
using Object = UnityEngine.Object;
|
|
|
|
namespace Unity.WebRTC.Samples
|
|
{
|
|
internal static class AudioSettingsUtility
|
|
{
|
|
static Dictionary<AudioSpeakerMode, int> pairs =
|
|
new Dictionary<AudioSpeakerMode, int>()
|
|
{
|
|
{AudioSpeakerMode.Mono, 1},
|
|
{AudioSpeakerMode.Stereo, 2},
|
|
{AudioSpeakerMode.Quad, 4},
|
|
{AudioSpeakerMode.Surround, 5},
|
|
{AudioSpeakerMode.Mode5point1, 6},
|
|
{AudioSpeakerMode.Mode7point1, 8},
|
|
{AudioSpeakerMode.Prologic, 2},
|
|
};
|
|
public static int SpeakerModeToChannel(AudioSpeakerMode mode)
|
|
{
|
|
return pairs[mode];
|
|
}
|
|
}
|
|
|
|
class AudioSpectrumView : MonoBehaviour
|
|
{
|
|
[SerializeField] AudioSource target;
|
|
[SerializeField] LineRenderer line;
|
|
[SerializeField] Color[] lineColors;
|
|
[SerializeField] RectTransform rectTransform;
|
|
[SerializeField] float xRatio = 1f;
|
|
[SerializeField] float yRatio = 1f;
|
|
|
|
const int positionCount = 256;
|
|
float[] spectrum = new float[2048];
|
|
|
|
Vector3[] array;
|
|
List<LineRenderer> lines = new List<LineRenderer>();
|
|
|
|
void Start()
|
|
{
|
|
array = new Vector3[positionCount];
|
|
|
|
// This line object is used as a template.
|
|
if (line.gameObject.activeInHierarchy)
|
|
line.gameObject.SetActive(false);
|
|
|
|
var conf = AudioSettings.GetConfiguration();
|
|
int count = AudioSettingsUtility.SpeakerModeToChannel(conf.speakerMode);
|
|
ResetLines(count);
|
|
|
|
AudioSettings.OnAudioConfigurationChanged += OnAudioConfigurationChanged;
|
|
}
|
|
|
|
private void OnDestroy()
|
|
{
|
|
AudioSettings.OnAudioConfigurationChanged -= OnAudioConfigurationChanged;
|
|
}
|
|
|
|
void OnAudioConfigurationChanged(bool deviceChanged)
|
|
{
|
|
var conf = AudioSettings.GetConfiguration();
|
|
int count = AudioSettingsUtility.SpeakerModeToChannel(conf.speakerMode);
|
|
ResetLines(count);
|
|
}
|
|
|
|
void ResetLines(int channelCount)
|
|
{
|
|
foreach (var line in lines)
|
|
{
|
|
Object.Destroy(line.gameObject);
|
|
}
|
|
lines.Clear();
|
|
for (int i = 0; i < channelCount; i++)
|
|
{
|
|
var line_ = GameObject.Instantiate(line, line.transform.parent);
|
|
line_.gameObject.SetActive(true);
|
|
line_.positionCount = positionCount;
|
|
line_.startColor = lineColors[i];
|
|
line_.endColor = lineColors[i];
|
|
lines.Add(line_);
|
|
}
|
|
}
|
|
|
|
void Update()
|
|
{
|
|
for (int lineIndex = 0; lineIndex < lines.Count; lineIndex++)
|
|
{
|
|
target.GetSpectrumData(spectrum, lineIndex, FFTWindow.Rectangular);
|
|
for (int i = 1; i < array.Length; i++)
|
|
{
|
|
float x = rectTransform.rect.width * i / array.Length * xRatio;
|
|
float y = rectTransform.rect.height * Mathf.Log(spectrum[i] + 1) * yRatio;
|
|
array[i] = new Vector3(x, y, 0);
|
|
}
|
|
lines[lineIndex].SetPositions(array);
|
|
}
|
|
}
|
|
}
|
|
}
|