From 7107b64d3b7e23fa700b2db3025be6059cf2a81f Mon Sep 17 00:00:00 2001
From: Gregor Billing <suushiemaniac@gmail.com>
Date: Sat, 5 Oct 2019 00:01:17 +0200
Subject: [PATCH] Always try to dump pruning files at first start

---
 threephase/src/main/java/cs/threephase/Tools.java | 11 +++++++----
 webscrambles/build.gradle.kts                     |  2 ++
 .../server/webscrambles/WebscramblesServer.kt     | 15 +++++++++++++++
 3 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/threephase/src/main/java/cs/threephase/Tools.java b/threephase/src/main/java/cs/threephase/Tools.java
index 94f0cfa38..59713206f 100644
--- a/threephase/src/main/java/cs/threephase/Tools.java
+++ b/threephase/src/main/java/cs/threephase/Tools.java
@@ -21,7 +21,7 @@ private static void read(int[][] arr, DataInput in) throws IOException {
 			for (int j=0, len=arr[i].length; j<len; j++) {
 				arr[i][j] = in.readInt();
 			}
-		}	
+		}
 	}
 
 	private static void write(int[][] arr, DataOutput out) throws IOException {
@@ -29,7 +29,7 @@ private static void write(int[][] arr, DataOutput out) throws IOException {
 			for (int j=0, len=arr[i].length; j<len; j++) {
 				out.writeInt(arr[i][j]);
 			}
-		}	
+		}
 	}
 
 	public synchronized static void initFrom(DataInput in) throws IOException {
@@ -37,6 +37,7 @@ public synchronized static void initFrom(DataInput in) throws IOException {
 			return;
 		}
 
+		System.out.println("Loading pruning tables from file...");
 		System.out.println("Initialize Center1 Solver...");
 
 		Center1.initSym();
@@ -58,7 +59,7 @@ public synchronized static void initFrom(DataInput in) throws IOException {
 		Edge3.initRaw2Sym();
 		read(Edge3.eprun, in);
 
-		System.out.println("OK");		
+		System.out.println("OK");
 
 		Search.inited = true;
 	}
@@ -67,7 +68,9 @@ public synchronized static void saveTo(DataOutput out) throws IOException {
 		if (!Search.inited) {
 			Search.init();
 		}
+		System.out.println("Writing pruning tables to file...");
+
 		write(Center1.ctsmv, out);
 		write(Edge3.eprun, out);
 	}
-}
\ No newline at end of file
+}
diff --git a/webscrambles/build.gradle.kts b/webscrambles/build.gradle.kts
index 520619ea0..7055de363 100644
--- a/webscrambles/build.gradle.kts
+++ b/webscrambles/build.gradle.kts
@@ -40,6 +40,8 @@ dependencies {
     implementation(project(":scrambles"))
     implementation(project(":server-ktor"))
 
+    implementation(project(":threephase"))
+
     implementation(ZIP4J)
     implementation(ITEXTPDF)
     implementation(BATIK_TRANSCODER)
diff --git a/webscrambles/src/main/kotlin/org/worldcubeassociation/tnoodle/server/webscrambles/WebscramblesServer.kt b/webscrambles/src/main/kotlin/org/worldcubeassociation/tnoodle/server/webscrambles/WebscramblesServer.kt
index c97624073..bc1ace9e3 100644
--- a/webscrambles/src/main/kotlin/org/worldcubeassociation/tnoodle/server/webscrambles/WebscramblesServer.kt
+++ b/webscrambles/src/main/kotlin/org/worldcubeassociation/tnoodle/server/webscrambles/WebscramblesServer.kt
@@ -1,6 +1,7 @@
 package org.worldcubeassociation.tnoodle.server.webscrambles
 
 import com.xenomachina.argparser.ArgParser
+import cs.threephase.Tools
 import io.ktor.application.Application
 import io.ktor.routing.routing
 import io.ktor.server.engine.commandLineEnvironment
@@ -23,6 +24,8 @@ import org.worldcubeassociation.tnoodle.server.webscrambles.routing.*
 import org.worldcubeassociation.tnoodle.server.webscrambles.server.LocalServerEnvironmentConfig
 import org.worldcubeassociation.tnoodle.server.webscrambles.server.MainLauncher
 import org.worldcubeassociation.tnoodle.server.webscrambles.server.OfflineJarUtils
+import java.io.DataInputStream
+import java.io.DataOutputStream
 
 class WebscramblesServer(val environmentConfig: ServerEnvironmentConfig) : ApplicationHandler {
     private val baseServer = TNoodleServer(environmentConfig)
@@ -36,6 +39,16 @@ class WebscramblesServer(val environmentConfig: ServerEnvironmentConfig) : Appli
         val scrambleHandler = ScrambleHandler(environmentConfig)
         val scrambleViewHandler = ScrambleViewHandler(environmentConfig)
 
+        if (environmentConfig.pruningTableExists(THREEPHASE_PRUNING)) {
+            DataInputStream(environmentConfig.getPruningTableInput(THREEPHASE_PRUNING)).use {
+                Tools.initFrom(it)
+            }
+        } else {
+            DataOutputStream(environmentConfig.getPruningTableOutput(THREEPHASE_PRUNING)).use {
+                Tools.saveTo(it)
+            }
+        }
+
         app.routing {
             PuzzleListHandler.install(this)
             RouteRedirectHandler.install(this)
@@ -52,6 +65,8 @@ class WebscramblesServer(val environmentConfig: ServerEnvironmentConfig) : Appli
     companion object {
         const val MIN_HEAP_SIZE_MEGS = 512
 
+        const val THREEPHASE_PRUNING = "444-threephase"
+
         val LOG = LoggerFactory.getLogger(WebscramblesServer::class.java)
 
         fun spinUpLocally(app: Application) =