UnrealEngineUWP/Engine/Plugins/FastBuildController/Source/Public/FastBuildControllerModule.h

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

#pragma once

#include "DistributedBuildInterface/Public/DistributedBuildControllerInterface.h"
#include "Containers/Array.h"
#include "Containers/Queue.h"

class FFastBuildJobProcessor;

class FFastBuildControllerModule : public IDistributedBuildController
{
	bool bSupported;
	bool bModuleInitialized;
	bool bControllerInitialized;

	FThreadSafeCounter NextFileID;
	FThreadSafeCounter NextTaskID;

	FThreadSafeCounter PendingTasksCounter;

	/** The thread spawned for shader compiling. */
	TUniquePtr<FFastBuildJobProcessor> JobDispatcherThread;
	
	// Taken when accessing the PendingTasks and DispatchedTasks members.
	TSharedPtr<FCriticalSection> TasksCS;

	// Queue of tasks submitted by the engine, but not yet dispatched to the controller.
	TQueue<FTask*> PendingTasks;

	// Map of tasks dispatched to the controller and running within FastBuild, that have not yet finished.
	TMap<uint32,FTask*> DispatchedTasks;

	const FString RootWorkingDirectory;

	const FString WorkingDirectory;

	bool bShutdown;

	void CleanWorkingDirectory();

public:
	FFastBuildControllerModule();
	virtual ~FFastBuildControllerModule();

	virtual void StartupModule() override final;
	virtual void ShutdownModule() override final;

	virtual bool RequiresRelativePaths() override final{ return true; };

	virtual void InitializeController() override final;

	virtual const FString GetName() override final { return FString("FastBuild Controller"); };

	virtual bool IsSupported() override final;

	virtual FString CreateUniqueFilePath() override final;
	virtual TFuture<FDistributedBuildTaskResult> EnqueueTask(const FTaskCommandData& CommandData) override final;
	void EnqueueTask(FTask* Task);
	FTask* DequeueTask();

	void RegisterDispatchedTask(FTask* DispatchedTask);
	void ReEnqueueDispatchedTasks();
	void DeRegisterDispatchedTasks(const TArray<uint32>& InTasksID);
	void ReportJobProcessed(FTask* CompletedTask, const FTaskResponse& InTaskResponse);
	
	static FFastBuildControllerModule& Get();

	inline bool AreTasksPending() const
	{
		FScopeLock Lock(TasksCS.Get());
		return  !PendingTasks.IsEmpty();
	}

	inline bool AreTasksDispatched() const
	{
		FScopeLock Lock(TasksCS.Get());
		return DispatchedTasks.Num() > 0;
	}

	inline bool AreTasksDispatchedOrPending() const
	{
		FScopeLock Lock(TasksCS.Get());
		return DispatchedTasks.Num() > 0 || !PendingTasks.IsEmpty();
	}

	int32 GetPendingTasksAmount() const { return PendingTasksCounter.GetValue();}

	FString GetRootWorkingDirectory() const { return RootWorkingDirectory;}

	FString GetWorkingDirectory() const { return WorkingDirectory;}

	const TMap<uint32, FTask*>& GetDispatchedTasks() const
	{
		return DispatchedTasks;
	}

	FCriticalSection* GetTasksCS() const
	{
		return TasksCS.Get();
	}
};