UnrealEngineUWP/Engine/Plugins/Runtime/Metasound/Source/MetasoundStandardNodes/Public/MetasoundTrigger.h

// Copyright Epic Games, Inc. All Rights Reserved.

#pragma once

#include "MetasoundDataReference.h"
#include "MetasoundDataTypeRegistrationMacro.h"
#include "MetasoundOperatorSettings.h"

namespace Metasound
{
	/** FTrigger supports sample accurate triggering, sample accurate internal tracking,
	 * and a convenient interface for running trigger-aligned audio signal processing
	 * routines on buffers.. 
	 *
	 * FTriggers are triggered using FTrigger::TriggerTime or FTrigger::TriggerFrame.
	 * FTriggers track time internally by calling FTrigger::Advance.
	 * Executing audio signal processing on buffers can be performed by calling
	 * FTrigger::ExecuteBlock or FTrigger::LookAhead.
	 */
	class METASOUNDSTANDARDNODES_API FTrigger
	{
		public:
			/** FTrigger constructor. 
			 *
			 * @param InSettings - Operator settings.
			 * @param bShouldTrigger - If true, triggers first sample.
			 */
			explicit FTrigger(const FOperatorSettings& InSettings, bool bShouldTrigger);

			/** FTrigger constructor. 
			 *
			 * @param InFrameToTrigger - Set specific frame to trigger.
			 * @param InSettings - Operator settings.
			 */
			explicit FTrigger(const FOperatorSettings& InSettings, int32 InFrameToTrigger);

			/** FTrigger constructor. By default it is not triggered.
			 *
			 * @param InSettings - Operator settings.
			 */
			FTrigger(const FOperatorSettings& InSettings);

			/** Trigger a specific frame in the future.
			 *
			 * @param InFrameToTrigger - Index of frame to trigger.
			 */
			void TriggerFrame(int32 InFrameToTrigger);

			/** Trigger a specific time in the future.
			 *
			 * @param InTime - Time to set trigger.
			 */
			template<typename TimeType>
			void TriggerTime(const TimeType& InTime)
			{
				TriggerFrame(InTime.GetNumSamples(SampleRate));
			}

			/** Advance internal frame counters by block size. */
			void AdvanceBlock();

			/** Advance internal frame counters by specific frame count. */
			void Advance(int32 InNumFrames);

			/** Number of triggered frames. */
			int32 Num() const;

			/** Returns true if there is a trigger in the current block of audio */
			int32 NumTriggeredInBlock() const;

			/** Returns frame index for a given trigger index. 
			 *
			 * @param InIndex - Index of trigger. Must be a value between 0 and Num().
			 *
			 * @return The frame of the trigger.
			 */
			int32 operator[](int32 InTriggerIndex) const;

			/** Returns true if there are any triggered frames. */
			bool IsTriggered() const;

			/** Returns true there is a trigger in the current block of audio. */
			bool IsTriggeredInBlock() const;

			/** Implicit conversion of FTrigger into bool by calling IsTriggeredInBlock()
			 *
			 * @return Returns true if frame is triggered in current block.
			 */
			operator bool() const;

			/** Removes all triggered frames. */
			void Reset();

			/** Executes one block of frames and calls underlying InPreTrigger and InOnTrigger
			 * functions with frame indices.
			 *
			 * @param InPreTrigger - A function which handles frames before the first 
			 *                   trigger in the current block. The function must accept 
			 *                   the arguments (int32 StartFrame, int32 EndFrame).
			 * @param InOnTrigger - A function which handles frames starting with the 
			 *                  triggers index and ending the next trigger index or the
			 *                  number of frames in a block.. The function must
			 *                  accept the arguments (int32 StartFrame, int32 EndFrame).
			 */
			template<typename PreTriggerType, typename OnTriggerType>
			void ExecuteBlock(PreTriggerType InPreTrigger, OnTriggerType InOnTrigger) const
			{
				ExecuteFrames(NumFramesPerBlock, LastTriggerIndexInBlock, InPreTrigger, InOnTrigger);
			}

			/** Executes a desired number of frames and calls underlying InPreTrigger 
			 * and InOnTrigger functions with frame indices.
			 *
			 * @param InNumFrames - The number of frames to look ahead.
			 * @param InPreTrigger - A function which handles frames before the first 
			 *                   trigger in the range of frames. The function must accept 
			 *                   the arguments (int32 StartFrame, int32 EndFrame).
			 * @param InOnTrigger - A function which handles frames starting with the 
			 *                  triggers index and ending the next trigger index or InNumFrames
			 *             		The function must accept the arguments 
			 *             		(int32 StartFrame, int32 EndFrame).
			 */
			template<typename PreTriggerType, typename OnTriggerType>
			void LookAhead(int32 InNumFrames, PreTriggerType InPreTrigger, OnTriggerType InOnTrigger) const
			{
				if (InNumFrames > 0)
				{
					int32 LastTriggerIndexInLookAhead= Algo::LowerBound(TriggeredFrames, InNumFrames);
					
					ExecuteFrames(InNumFrames, LastTriggerIndexInLookAhead, InPreTrigger, InOnTrigger);
				}
			}

		private:
			void UpdateLastTriggerIndexInBlock();

			template<typename PreTriggerType, typename OnTriggerType>
			void ExecuteFrames(int32 InNumFrames, int32 InLastTriggerIndex, PreTriggerType InPreTrigger, OnTriggerType InOnTrigger) const
			{
				if (InLastTriggerIndex <= 0)
				{
					InPreTrigger(0, InNumFrames);
					return;
				}

				const int32* TriggeredFramesData = TriggeredFrames.GetData();

				if (TriggeredFramesData[0] > 0)
				{
					InPreTrigger(0, TriggeredFramesData[0]);
				}

				const int32 NextToLastTriggerIndex = InLastTriggerIndex - 1;

				for (int32 i = 0; i < NextToLastTriggerIndex; i++)
				{
					InOnTrigger(TriggeredFramesData[i], TriggeredFramesData[i + 1]);
				}

				InOnTrigger(TriggeredFramesData[NextToLastTriggerIndex], InNumFrames);
			}


			TArray<int32> TriggeredFrames;

			bool bHasTrigger = false;
			int32 NumFramesPerBlock = 0;
			float SampleRate;
			int32 LastTriggerIndexInBlock = 0;
	};

	DECLARE_METASOUND_DATA_REFERENCE_TYPES(FTrigger, METASOUNDSTANDARDNODES_API, FTriggerTypeInfo, FTriggerReadRef, FTriggerWriteRef);
}