UnrealEngineUWP/Engine/Plugins/Runtime/Metasound/Source/MetasoundFrontend/Private/MetasoundFrontendInterfaceRegistryPrivate.h

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

#pragma once

#include "MetasoundFrontendArchetypeRegistry.h"
#include "MetasoundFrontendController.h"
#include "MetasoundFrontendDocument.h"
#include "MetasoundFrontendRegistryTransaction.h"
#include "UObject/NoExportTypes.h"

namespace Metasound
{
	namespace Frontend
	{
		using FInterfaceTransactionStream = TTransactionStream<FInterfaceRegistryTransaction>;

		class FInterfaceRegistry : public IInterfaceRegistry
		{
		public:
			static FInterfaceRegistry& Get();

			FInterfaceRegistry();

			virtual ~FInterfaceRegistry() = default;

			virtual void RegisterInterface(TUniquePtr<IInterfaceRegistryEntry>&& InEntry) override;

			virtual const IInterfaceRegistryEntry* FindInterfaceRegistryEntry(const FInterfaceRegistryKey& InKey) const override;

			virtual bool FindInterface(const FInterfaceRegistryKey& InKey, FMetasoundFrontendInterface& OutInterface) const override;

			virtual void AddInterfaceOutputBindings(const FMetasoundFrontendVersion& InInputInterfaceVersion, TArray<FMetasoundFrontendInterfaceBinding>&& InBinding) override;

			virtual bool RemoveInterfaceOutputBinding(const FMetasoundFrontendVersion& InInputInterfaceVersion, const FMetasoundFrontendVersion& InOutputInterfaceVersion) override;

			virtual bool RemoveAllInterfaceOutputBindings(const FMetasoundFrontendVersion& InInputInterfaceVersion) override;

			TUniquePtr<FInterfaceTransactionStream> CreateTransactionStream();

			virtual void ForEachRegistryTransactionSince(FRegistryTransactionID InSince, FRegistryTransactionID* OutCurrentRegistryTransactionID, TFunctionRef<void(const FInterfaceRegistryTransaction&)> InFunc) const override;

		private:
			using FInterfaceTransactionBuffer = TTransactionBuffer<FInterfaceRegistryTransaction>;

			TMap<FInterfaceRegistryKey, TUniquePtr<IInterfaceRegistryEntry>> Entries;
			TSharedRef<FInterfaceTransactionBuffer> TransactionBuffer;
		};
	}
}