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

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

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

namespace AutomationTool.Benchmark
{
	class BenchmarkRunEditorTask : BenchmarkEditorTaskBase
	{
		FileReference ProjectFile = null;

		string Project = "";
		string ProjectMap = "";
		string EditorArgs = "";

		EditorTaskOptions Options;

		public BenchmarkRunEditorTask(string InProject, string InProjectMap="", EditorTaskOptions InOptions = EditorTaskOptions.None, string InEditorArgs="")
		{
			Options = InOptions;
			EditorArgs = InEditorArgs;

			if (!InProject.Equals("UE4", StringComparison.OrdinalIgnoreCase))
			{
				Project = InProject;
				ProjectFile = ProjectUtils.FindProjectFileFromName(InProject);
				ProjectMap = InProjectMap;

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

			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(EditorArgs))
			{
				TaskModifiers.Add(EditorArgs);
			}

			TaskName = string.Format("Launch {0} Editor", InProject, BuildHostPlatform.Current.Platform);
		}

		protected override bool PerformPrequisites()
		{
			// build editor
			BuildTarget Command = new BuildTarget();
			Command.ProjectName = Project;
			Command.Platforms = BuildHostPlatform.Current.Platform.ToString();
			Command.Targets = "Editor";
			Command.NoTools = true;

			if (Command.Execute() != ExitCode.Success)
			{
				return false;

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

			// if they want a hot DDC thendo the test one time with no timing
			if (Options.HasFlag(EditorTaskOptions.HotDDC))
			{
				RunEditorAndWaitForMapLoad();
			}

			return true;
		}

		static IProcessResult CurrentProcess = null;
		static bool CurrentProcessWasKilled = false;

		/// <summary>
		/// A filter that suppresses all output od stdout/stderr
		/// </summary>
		/// <param name="Message"></param>
		/// <returns></returns>
		static string EndOnMapCheckFilter(string Message)
		{
			if (CurrentProcess != null)
			{
				if (Message.Contains("MapCheck: Map check complete"))
				{
					CurrentProcessWasKilled = true;
					CurrentProcess.ProcessObject.Kill();
				}
			}
			return Message;
		}

		protected bool RunEditorAndWaitForMapLoad()
		{
			string ProjectArg = ProjectFile != null ? ProjectFile.ToString() : "";
			string EditorPath = HostPlatform.Current.GetUE4ExePath("UE4Editor.exe");
			string Arguments = string.Format("{0} {1} {2} -stdout -AllowStdOutLogVerbosity -unattended", ProjectArg, ProjectMap, EditorArgs);

			var RunOptions = CommandUtils.ERunOptions.AllowSpew | CommandUtils.ERunOptions.NoWaitForExit;

			var SpewDelegate = new ProcessResult.SpewFilterCallbackType(EndOnMapCheckFilter);

			CurrentProcessWasKilled = false;
			CurrentProcess = CommandUtils.Run(EditorPath, Arguments, Options: RunOptions, SpewFilterCallback: SpewDelegate);

			DateTime StartTime = DateTime.Now;

			int MaxWaitMins = 30;

			while (!CurrentProcess.HasExited)
			{
				Thread.Sleep(5 * 1000);

				if ((DateTime.Now - StartTime).TotalMinutes >= MaxWaitMins)
				{
					Log.TraceError("Gave up waiting for task after {0} minutes", MaxWaitMins);
					CurrentProcess.ProcessObject.Kill();
				}
			}

			CurrentProcess = null;
			return CurrentProcessWasKilled;
		}

		protected override bool PerformTask()
		{
			return RunEditorAndWaitForMapLoad();
		}
	}
}