From 6d080f7d578c5fbed3fed94349c0416072fe18ba Mon Sep 17 00:00:00 2001
From: Steve Streeting <steve@stevestreeting.com>
Date: Tue, 6 Oct 2020 15:57:54 +0100
Subject: [PATCH] WIP rebuild lightmaps script

---
 inc/uproject.ps1          |   2 +-
 ue4-rebuild-lightmaps.ps1 | 122 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 123 insertions(+), 1 deletion(-)
 create mode 100644 ue4-rebuild-lightmaps.ps1

diff --git a/inc/uproject.ps1 b/inc/uproject.ps1
index 2d006b3..d733ec8 100644
--- a/inc/uproject.ps1
+++ b/inc/uproject.ps1
@@ -8,7 +8,7 @@ function Get-Uproject-Filename {
     )
 
     $projfile = ""
-    if ($config.ProjectFile) {
+    if ($config -and $config.ProjectFile) {
         if (-not [System.IO.Path]::IsPathRooted($config.ProjectFile)) {
             $projfile = Join-Path $srcfolder $config.ProjectFile
         } else {
diff --git a/ue4-rebuild-lightmaps.ps1 b/ue4-rebuild-lightmaps.ps1
new file mode 100644
index 0000000..9a24177
--- /dev/null
+++ b/ue4-rebuild-lightmaps.ps1
@@ -0,0 +1,122 @@
+# Lightmap rebuild helper
+[CmdletBinding()] # Fail on unknown args
+param (
+    # Optional source folder, assumed current folder
+    [string]$src,
+    # quality level (Preview, Medium, High, Production), default = Production
+    [string]$quality,
+    # Explicit list of maps, if not supplied will use cooked maps in packageconfig.json
+    [array]$maps,
+    # Dry-run; does nothing but report what *would* have happened
+    [switch]$dryrun = $false,
+    [switch]$help = $false
+)
+
+. $PSScriptRoot\inc\platform.ps1
+. $PSScriptRoot\inc\packageconfig.ps1
+. $PSScriptRoot\inc\projectversion.ps1
+. $PSScriptRoot\inc\uproject.ps1
+. $PSScriptRoot\inc\ueinstall.ps1
+
+function Write-Usage {
+    Write-Output "Steve's UE4 lightmap rebuilding tool"
+    Write-Output "Usage:"
+    Write-Output "  ue4-rebuild-lightmaps.ps1 [-src:sourcefolder] [-quality:(preview|medium|high|production)]  [-maps Map1,Map2,Map3] [-dryrun]"
+    Write-Output " "
+    Write-Output "  -src          : Source folder (current folder if omitted)"
+    Write-Output "  -quality      : Lightmap quality, preview/medium/high/production"
+    Write-Output "                :   (Default: production)"
+    Write-Output "  -maps         : List of maps to rebuild. If omitted, will derive which ones to"
+    Write-Output "                  rebuild based on cooked maps in packageconfig.json"
+    Write-Output "  -dryrun       : Don't perform any actual actions, just report on what you would do"
+    Write-Output "  -help         : Print this help"
+}
+
+if ($src.Length -eq 0) {
+    $src = "."
+    Write-Verbose "-src not specified, assuming current directory"
+}
+
+$ErrorActionPreference = "Stop"
+
+if ($help) {
+    Write-Usage
+    Exit 0
+}
+
+# Detect Git
+if ($src -ne ".") { Push-Location $src }
+$isGit = Test-Path ".git"
+if ($src -ne ".") { Pop-Location }
+
+Write-Output "~-~-~ UE4 Lightmap Rebuild Start ~-~-~"
+
+try {
+    $config = Read-Package-Config -srcfolder:$src
+    $projfile = Get-Uproject-Filename -srcfolder:$src -config:$config
+    $proj = Read-Uproject $projfile
+    $ueVersion = Get-UE-Version $proj
+    $ueinstall = Get-UE-Install $ueVersion
+
+    if ($maps) {
+        # Explicit list of maps provided on command line
+        $mapsToRebuild = Find-File-Set -startDir:$(Join-Path $src "Content") -pattern:*.umap -includeByDefault:$false -includeBaseNames:$maps
+    } else {
+        # Derive maps from cook settings
+        $mapsToRebuild = Find-File-Set -startDir:$(Join-Path $src "Content") -pattern:*.umap -includeByDefault:$config.CookAllMaps -includeBaseNames:$config.MapsIncluded -excludeBaseNames:$config.MapsExcluded
+    }
+
+    # TODO check maps not found
+    # TODO lock map files if read-only
+
+    if (-not $quality) {
+        $quality = "Production"
+    }
+    if ($quality -notin @("Preview", "Medium", "High", "Production")) {
+        throw "Invalid quality level: $quality"
+    }
+
+    Write-Output ""
+    Write-Output "Project File : $projfile"
+    Write-Output "UE Version   : $ueVersion"
+    Write-Output "UE Install   : $ueinstall"
+    Write-Output ""
+    Write-Output "Maps         : $mapsToRebuild"
+    Write-Output "Quality      : $quality"
+    Write-Output ""
+
+    $argList = [System.Collections.ArrayList]@()
+    $argList.Add("`"$projfile`"") > $null
+    $argList.Add("-run=ResavePackages") > $null
+    $argList.Add("-buildtexturestreaming") > $null
+    $argList.Add("-buildlighting") > $null
+    $argList.Add("-buildreflectioncaptures") > $null
+    $argList.Add("-MapsOnly") > $null
+    $argList.Add("-ProjectOnly") > $null
+    $argList.Add("-AllowCommandletRendering") > $null
+    $argList.Add("-SkipSkinVerify") > $null
+    $argList.Add("-Quality=$quality") > $null
+    $argList.Add("-Map=$($mapsToRebuild -join "+")") > $null   
+
+    $ueEditorCmd = Join-Path $ueinstall "Engine/Binaries/Win64/UE4Editor-Cmd$exeSuffix"
+
+    if ($dryrun) {
+        Write-Output "Would have run:"
+        Write-Output "> $ueEditorCmd $($argList -join " ")"
+
+    } else {            
+        $proc = Start-Process $runUAT $ueEditorCmd -Wait -PassThru -NoNewWindow
+        if ($proc.ExitCode -ne 0) {
+            throw "Lightmap build failed!"
+        }
+
+    }
+
+} catch {
+    Write-Output $_.Exception.Message
+    Exit 9
+
+}
+
+
+Write-Output "~-~-~ UE4 Lightmap Rebuild OK ~-~-~"