ichni_Official/Assets/Wwise/API/Editor/WwiseWindows/AkWwisePicker.cs

#if UNITY_EDITOR
/*******************************************************************************
The content of this file includes portions of the proprietary AUDIOKINETIC Wwise
Technology released in source code form as part of the game integration package.
The content of this file may not be used without valid licenses to the
AUDIOKINETIC Wwise Technology.
Note that the use of the game engine is subject to the Unity(R) Terms of
Service at https://unity3d.com/legal/terms-of-service
 
License Usage
 
Licensees holding valid licenses to the AUDIOKINETIC Wwise Technology may use
this file in accordance with the end user license agreement provided with the
software or, alternatively, in accordance with the terms contained
in a written agreement between you and Audiokinetic Inc.
Copyright (c) 2025 Audiokinetic Inc.
*******************************************************************************/

public class AkWwisePicker : UnityEditor.EditorWindow
{
	[UnityEngine.SerializeField] UnityEditor.IMGUI.Controls.TreeViewState m_treeViewState;

	public static AkWwiseTreeView m_treeView;
	UnityEditor.IMGUI.Controls.SearchField m_SearchField;

	[UnityEditor.MenuItem("Window/Wwise Picker", false, (int)AkWwiseWindowOrder.WwisePicker)]
	public static void InitPickerWindow()
	{
		GetWindow<AkWwisePicker>("Wwise Picker", true,
		   typeof(UnityEditor.EditorWindow).Assembly.GetType("UnityEditor.ConsoleWindow"));
	}

	public void OnEnable()
	{
		if (m_treeViewState == null)
		{
			m_treeViewState = new UnityEditor.IMGUI.Controls.TreeViewState();
		}

		var multiColumnHeaderState = AkWwiseTreeView.CreateDefaultMultiColumnHeaderState();
		var multiColumnHeader = new UnityEditor.IMGUI.Controls.MultiColumnHeader(multiColumnHeaderState);
		m_treeView = new AkWwiseTreeView(m_treeViewState, multiColumnHeader, AkWwiseProjectInfo.GetTreeData());
		m_treeView.SetDoubleClickFunction(PlayPauseItem);

		m_treeView.dirtyDelegate = RequestRepaint;

		if (m_treeView.dataSource.Data.ItemDict.Count == 0)
		{
			Refresh();
			RequestRepaint();
		}
		
		m_SearchField = new UnityEditor.IMGUI.Controls.SearchField();
		m_SearchField.downOrUpArrowKeyPressed += m_treeView.SetFocusAndEnsureSelectedItem;
		m_SearchField.SetFocus();
	}

	public void OnDisable()
	{
		m_treeView.SaveExpansionStatus();
	}

	public static void Refresh(bool ignoreIfWaapi = false)
	{
		if (AkWwiseProjectInfo.GetData().currentDataSource == AkWwiseProjectInfo.DataSourceType.WwiseAuthoring && ignoreIfWaapi)
		{
			return;
		}

		if (m_treeView != null)
		{
			m_treeView.dataSource.FetchData();
		};
	}

	private void PlayPauseItem(AkWwiseTreeViewItem item)
	{
		if (m_treeView != null && m_treeView.CheckWaapi())
		{
			AkWaapiUtilities.TogglePlayEvent(item.objectType, item.objectGuid);
		}
	}

	private bool isDirty;
	public void RequestRepaint()
	{
		isDirty = true;
	}

	void Update()
	{
		if (isDirty)
		{
			Repaint();
			m_treeView.Reload();
			isDirty = false;
		}

		if (AkWwiseEditorSettings.Instance.UseWaapi)
		{
			AkWwiseProjectInfo.WaapiPickerData.Update();
		}
	}

	public void OnGUI()
	{
		AkWwiseProjectInfo.DataSourceType ds;
		var buttonWidth = 150;
		using (new UnityEngine.GUILayout.HorizontalScope("box"))
		{
			ds = (AkWwiseProjectInfo.DataSourceType)UnityEditor.EditorGUILayout.EnumPopup(
				AkWwiseProjectInfo.GetData().currentDataSource, UnityEngine.GUILayout.Width(buttonWidth));
			UnityEngine.GUILayout.Space(5);

			var projectData = AkWwiseProjectInfo.GetData();

			if (ds != projectData.currentDataSource)
			{
				projectData.currentDataSource = ds;
				m_treeView.SetDataSource(AkWwiseProjectInfo.GetTreeData());
			}

			if (ds == AkWwiseProjectInfo.DataSourceType.FileSystem)
			{
				projectData.autoPopulateEnabled =
					UnityEngine.GUILayout.Toggle(projectData.autoPopulateEnabled, "Auto populate");
			}
			else
			{
				projectData.AutoSyncSelection =
					UnityEngine.GUILayout.Toggle(projectData.AutoSyncSelection, "Autosync selection");
				AkWwiseProjectInfo.WaapiPickerData.AutoSyncSelection = projectData.AutoSyncSelection;
			}

			UnityEngine.GUILayout.FlexibleSpace();

			if (UnityEngine.GUILayout.Button("Refresh Project", UnityEngine.GUILayout.Width(buttonWidth)))
			{
				if (ds == AkWwiseProjectInfo.DataSourceType.FileSystem)
				{
					AkWwiseProjectInfo.Populate();
				}
				Refresh();
			}


			if (UnityEngine.GUILayout.Button("Generate SoundBanks", UnityEngine.GUILayout.Width(buttonWidth)))
			{
#if UNITY_EDITOR_WIN
				string wwiseInstallationPath = AkWwiseEditorSettings.Instance.WwiseInstallationPathWindows;
#elif UNITY_EDITOR_OSX
				string wwiseInstallationPath = AkWwiseEditorSettings.Instance.WwiseInstallationPathMac;
#endif
				if (AkUtilities.IsSoundbankGenerationAvailable(wwiseInstallationPath))
				{
					AkUtilities.GenerateSoundbanks(wwiseInstallationPath, AkWwiseEditorSettings.WwiseProjectAbsolutePath);
				}
				else if(!AkUtilities.GeneratingSoundBanks)
				{
					UnityEngine.Debug.LogError("Access to Wwise is required to generate the SoundBanks. Please go to Edit > Project Settings... and set the Wwise Application Path found in the Wwise Integration view.");
				}
			}

			if (projectData.autoPopulateEnabled && AkUtilities.IsWwiseProjectAvailable)
			{
				AkWwiseWWUBuilder.StartWWUWatcher();
			}
			else
			{
				AkWwiseWWUBuilder.StopWWUWatcher();
			}
		}

		using (new UnityEngine.GUILayout.HorizontalScope("box"))
		{
			var search_width = System.Math.Max(position.width / 3, buttonWidth * 2);

			if (ds == AkWwiseProjectInfo.DataSourceType.FileSystem)
			{
				m_treeView.StoredSearchString = m_SearchField.OnGUI(UnityEngine.GUILayoutUtility.GetRect(search_width, 20), m_treeView.StoredSearchString);
				UnityEngine.GUILayout.FlexibleSpace();

			}

			else
			{
				m_treeView.StoredSearchString = m_SearchField.OnGUI(UnityEngine.GUILayoutUtility.GetRect(search_width, 20), m_treeView.StoredSearchString);
				UnityEngine.GUILayout.FlexibleSpace();

				var labelStyle = new UnityEngine.GUIStyle();
				labelStyle.richText = true;
				UnityEngine.GUILayout.Label(AkWaapiUtilities.GetStatusString(), labelStyle);
			}
		}

		UnityEngine.GUILayout.Space(UnityEditor.EditorGUIUtility.standardVerticalSpacing);


		UnityEngine.GUILayout.FlexibleSpace();
		UnityEngine.Rect lastRect = UnityEngine.GUILayoutUtility.GetLastRect();
		m_treeView.OnGUI(new UnityEngine.Rect(lastRect.x, lastRect.y, position.width, lastRect.height));

		if (UnityEngine.GUI.changed && AkUtilities.IsWwiseProjectAvailable)
		{
			UnityEditor.EditorUtility.SetDirty(AkWwiseProjectInfo.GetData());
		}
	}

	static void SelectInWwisePicker(System.Guid guid)
	{
		InitPickerWindow();
		m_treeView.SelectItem(guid);
	}

	[UnityEditor.MenuItem("CONTEXT/AkBank/Select in Wwise Picker")]
	[UnityEditor.MenuItem("CONTEXT/AkAmbient/Select in Wwise Picker")]
	[UnityEditor.MenuItem("CONTEXT/AkEvent/Select in Wwise Picker")]
	[UnityEditor.MenuItem("CONTEXT/AkState/Select in Wwise Picker")]
	[UnityEditor.MenuItem("CONTEXT/AkSwitch/Select in Wwise Picker")]
	[UnityEditor.MenuItem("CONTEXT/AkWwiseTrigger/Select in Wwise Picker")]
	static void SelectItemInWwisePicker(UnityEditor.MenuCommand command)
	{
		AkTriggerHandler component = (AkTriggerHandler)command.context;
		try
		{
			var data = component.GetType().GetField("data");
			var guid = (data.GetValue(component) as AK.Wwise.BaseType).ObjectReference.Guid;
			SelectInWwisePicker(guid);
		}
		catch { }
	}

	[UnityEditor.MenuItem("CONTEXT/AkRoom/Select Aux Bus in Wwise Picker")]
	static void SelectAkRoomAuxBusInWwisePicker(UnityEditor.MenuCommand command)
	{
		AkRoom component = (AkRoom)command.context;
		SelectInWwisePicker(component.reverbAuxBus.ObjectReference.Guid);
	}

	[UnityEditor.MenuItem("CONTEXT/AkRoom/Select Event in Wwise Picker")]
	static void SelectAkRoomEventInWwisePicker(UnityEditor.MenuCommand command)
	{
		AkRoom component = (AkRoom)command.context;
		SelectInWwisePicker(component.roomToneEvent.ObjectReference.Guid);
	}

	[UnityEditor.MenuItem("CONTEXT/AkSurfaceReflector/Select in Wwise Picker")]
	static void SelectReflectorTextureItemInWwisePicker(UnityEditor.MenuCommand command)
	{
		AkSurfaceReflector component = (AkSurfaceReflector)command.context;
		if (component.AcousticTextures.Length >0 && component.AcousticTextures[0].ObjectReference !=null)
		{
			SelectInWwisePicker(component.AcousticTextures[0].ObjectReference.Guid);
		}
	}

	[UnityEditor.MenuItem("CONTEXT/AkEnvironment/Select in Wwise Picker")]
	static void SelectEnvironmentItemInWwisePicker(UnityEditor.MenuCommand command)
	{
		AkEnvironment component = (AkEnvironment)command.context;
		SelectInWwisePicker(component.data.ObjectReference.Guid);
	}

	[UnityEditor.MenuItem("CONTEXT/AkEarlyReflections/Select in Wwise Picker")]
	static void SelectReflectionsItemInWwisePicker(UnityEditor.MenuCommand command)
	{
		AkEarlyReflections component = (AkEarlyReflections)command.context;
		SelectInWwisePicker(component.reflectionsAuxBus.ObjectReference.Guid);
	}
}
#endif