From 7e0aba78bac4c846f7935ad701fbaa765a8cd2e1 Mon Sep 17 00:00:00 2001
From: KoalaTrapLord <265133236+KoalaTrapLord@users.noreply.github.com>
Date: Mon, 23 Mar 2026 19:01:43 -0400
Subject: [PATCH] fix: add ML logging support to MultiThreadBatchRunner

---
 .../batch/MultiThreadBatchRunner.java         | 21 +++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/src/main/java/commanderailab/batch/MultiThreadBatchRunner.java b/src/main/java/commanderailab/batch/MultiThreadBatchRunner.java
index 3adebfb..d358e78 100644
--- a/src/main/java/commanderailab/batch/MultiThreadBatchRunner.java
+++ b/src/main/java/commanderailab/batch/MultiThreadBatchRunner.java
@@ -43,6 +43,11 @@ public class MultiThreadBatchRunner {
     private boolean useSimplifiedAi = false;
     private int aiThinkTimeMs = -1;
 
+    // ── ML Decision Logging ────────────────────────────────────
+    private boolean mlLogEnabled = false;
+    private String mlOutputDir = "results";
+    private String mlBatchId;
+
     public MultiThreadBatchRunner(String forgeJarPath, String forgeWorkDir,
                                   List<DeckInfo> decks, AiPolicy policy, int numThreads) {
         this(forgeJarPath, forgeWorkDir, decks, policy, numThreads, true, 120, "java");
@@ -93,6 +98,16 @@ public class MultiThreadBatchRunner {
         this.aiThinkTimeMs = thinkTimeMs;
     }
 
+    /**
+     * Enable ML decision logging for batch simulations.
+     * Each thread gets its own log file with a unique batch ID suffix.
+     */
+    public void enableMlLogging(String outputDir, String batchId) {
+        this.mlLogEnabled = true;
+        this.mlOutputDir = outputDir;
+        this.mlBatchId = batchId;
+    }
+
     /**
      * Auto-detect optimal thread count based on available CPU cores (Issue #5).
      *
@@ -193,6 +208,11 @@ public class MultiThreadBatchRunner {
 
             // Issue #4: Forward AI optimization settings
             runner.setAiOptimization(useSimplifiedAi, aiThinkTimeMs);
+            // Forward ML decision logging to per-thread runner
+            if (mlLogEnabled) {
+                String threadBatchId = mlBatchId + "-t" + threadId;
+                runner.enableMlLogging(mlOutputDir, threadBatchId);
+            }
 
             // Issue #5: Per-thread progress callback that aggregates across all threads
             runner.setProgressCallback((completed, total, pct, simsPerSec, lastResult) -> {
@@ -218,6 +238,7 @@ public class MultiThreadBatchRunner {
                 for (int i = 0; i < results.size(); i++) {
                     results.get(i).gameIndex = startIndex + i;
                 }
+                if (mlLogEnabled) { runner.closeMlLogger(); }
                 return results;
             }));