From 56b04d4ea7d142c88fafa04cd242045a0768b2e3 Mon Sep 17 00:00:00 2001
From: Armyke <mic.rovigatti@gmail.com>
Date: Mon, 4 Mar 2019 14:05:25 +0000
Subject: [PATCH] Added the same --tmp_dir flag to tesstrain_utils.sh

---
 src/training/tesstrain_utils.sh | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/training/tesstrain_utils.sh b/src/training/tesstrain_utils.sh
index b7d6153f83..c92cffba7c 100644
--- a/src/training/tesstrain_utils.sh
+++ b/src/training/tesstrain_utils.sh
@@ -149,6 +149,9 @@ parse_flags() {
             --fonts_dir)
                 parse_value "FONTS_DIR" ${ARGV[$j]:-}
                 i=$j ;;
+	    --tmp_dir)
+		parse_value "TMP_DIR"   ${ARGV[$j]:-}
+		i=$j ;;
             --lang)
                 parse_value "LANG_CODE" ${ARGV[$j]:-}
                 i=$j ;;
@@ -215,7 +218,11 @@ parse_flags() {
 
     # Location where intermediate files will be created.
     TIMESTAMP=$(date +%Y-%m-%d)
-    TMP_DIR=$(mktemp -d -t ${LANG_CODE}-${TIMESTAMP}.XXX)
+    if [[ -z ${TMP_DIR:-} ]]; then
+        TMP_DIR=$(mktemp -d -t ${LANG_CODE}-${TIMESTAMP}.XXX)
+    else
+	TMP_DIR=$(mktemp -d -p ${TMP_DIR} -t ${LANG_CODE}-${TIMESTAMP}.XXX)
+    fi
     TRAINING_DIR=${TMP_DIR}
     # Location of log file for the whole run.
     LOG_FILE=${TRAINING_DIR}/tesstrain.log