UnrealEngineUWP/Engine/Source/Programs/AutomationTool/Scripts/BenchmarkBuild/BenchmarkCookTask.cs

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

using AutomationTool;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Tools.DotNETCommon;
using UnrealBuildTool;

namespace AutomationTool.Benchmark
{


	class BenchmarkCookTask : BenchmarkEditorTaskBase
	{
		string			ProjectName;

		FileReference	ProjectFile;

		string			CookPlatformName;

		EditorTaskOptions		Options;

		string			CookArgs;

		public BenchmarkCookTask(string InProject, UnrealTargetPlatform InPlatform, EditorTaskOptions InOptions, string InCookArgs)
		{
			ProjectName = InProject;
			Options = InOptions;
			CookArgs = InCookArgs;

			ProjectFile = ProjectUtils.FindProjectFileFromName(ProjectName);

			var PlatformToCookPlatform = new Dictionary<UnrealTargetPlatform, string> {
				{ UnrealTargetPlatform.Win64, "WindowsClient" },
				{ UnrealTargetPlatform.Mac, "MacClient" },
				{ UnrealTargetPlatform.Linux, "LinuxClient" },
				{ UnrealTargetPlatform.Android, "Android_ASTCClient" }
			};

			CookPlatformName = InPlatform.ToString();

			if (PlatformToCookPlatform.ContainsKey(InPlatform))
			{
				CookPlatformName = PlatformToCookPlatform[InPlatform];
			}

			TaskName = string.Format("Cook {0} {1}", InProject, CookPlatformName);

			if (Options.HasFlag(EditorTaskOptions.NoDDC))
			{
				TaskModifiers.Add("noddc");
			}

			if (Options.HasFlag(EditorTaskOptions.NoShaderDDC))
			{
				TaskModifiers.Add("noshaderddc");
			}

			if (Options.HasFlag(EditorTaskOptions.ColdDDC))
			{
				TaskModifiers.Add("coldddc");
			}

			if (!string.IsNullOrEmpty(CookArgs))
			{
				TaskModifiers.Add(CookArgs);
			}
		}

		protected override bool PerformPrequisites()
		{
			// build editor
			BuildTarget Command = new BuildTarget();
			Command.ProjectName = ProjectName;
			Command.Platforms = BuildHostPlatform.Current.Platform.ToString();
			Command.Targets = "Editor";
			
			if (Command.Execute() != ExitCode.Success)
			{
				return false;
			}

			// Do a cook to make sure the remote ddc is warm?
			if (Options.HasFlag(EditorTaskOptions.HotDDC))
			{
				// will throw an exception if it fails
				CommandUtils.RunCommandlet(ProjectFile, "UE4Editor-Cmd.exe", "Cook", String.Format("-TargetPlatform={0} ", CookPlatformName));
			}

			if (Options.HasFlag(EditorTaskOptions.ColdDDC))
			{
				DeleteLocalDDC(ProjectFile);
			}

			return true;
		}

		protected override bool PerformTask()
		{
			List<string> ExtraArgsList = new List<string>();
			
			if (Options.HasFlag(EditorTaskOptions.CookClient))
			{
				ExtraArgsList.Add("client");
			}

			if (Options.HasFlag(EditorTaskOptions.NoShaderDDC))
			{
				ExtraArgsList.Add("noshaderddc");
				//ExtraArgs.Add("noxgeshadercompile");
			}

			if (Options.HasFlag(EditorTaskOptions.NoDDC))
			{
				ExtraArgsList.Add("ddc=noshared");
			}

			string ExtraArgs = "";

			if (ExtraArgsList.Any())
			{
				ExtraArgs = "-" + string.Join(" -", ExtraArgsList);
			}

			if (CookArgs.Length > 0)
			{
				ExtraArgs += " " + CookArgs;
			}

			// will throw an exception if it fails
			CommandUtils.RunCommandlet(ProjectFile, "UE4Editor-Cmd.exe", "Cook", String.Format("-TargetPlatform={0} {1}", CookPlatformName, ExtraArgs));

			return true;
		}
	}
}