From d5e83f2f59c23569896e1945ccb2e81a3e3fa500 Mon Sep 17 00:00:00 2001
From: "Steinar H. Gunderson" <sgunderson@bigfoot.com>
Date: Mon, 15 Dec 2014 00:59:47 +0100
Subject: [PATCH] Cleanups in build-book.sh.

---
 build-book.sh | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/build-book.sh b/build-book.sh
index 772f3fb..d9c2876 100755
--- a/build-book.sh
+++ b/build-book.sh
@@ -2,10 +2,11 @@
 
 set -e
 
+export MAPPERS=40
+export REDUCERS=40
 export SHARDS=40
-export PARALLEL_MERGES=40
 
-rm -f pgnnames.txt tmp.mtbl* open.mtbl.new open.mtbl.part???? open.mtbl.part????.new 2>/dev/null
+rm -f pgnnames.txt tmp.mtbl* open.mtbl.part????.new 2>/dev/null
 
 for FILE in "$@"; do
 	echo "$FILE" >> pgnnames.txt
@@ -13,13 +14,13 @@ done
 
 TS_BEGIN=$( date +%s%3N )
 
-for X in $( seq 0 39 ); do
+for X in $( seq 0 $(( MAPPERS - 1 )) ); do
 	(
 		(
 			PGNNUM=0
 			for FILE in "$@"; do
-				START=$( ./find-pgn-split-point.sh "$FILE" $X 40 )
-				END=$( ./find-pgn-split-point.sh "$FILE" $(( X + 1 )) 40 )
+				START=$( ./find-pgn-split-point.sh "$FILE" $X $MAPPERS )
+				END=$( ./find-pgn-split-point.sh "$FILE" $(( X + 1 )) $MAPPERS )
 				~/nmu/pgn-extract/pgn-extract --startpos $START --endpos $END --startfilenum $PGNNUM -e -Wsessebin "$FILE" 2>/dev/null
 				PGNNUM=$(( PGNNUM + 1 ))
 			done
@@ -33,7 +34,7 @@ MAPOUTPUT=$( du -sch tmp.mtbl.mapper* | tail -n 1 | cut -f1 | sed 's/\([^0-9]\)$
 
 echo "Reducing..."
 
-parallel -i -j ${PARALLEL_MERGES} bash -c 'X={}; ./binmerger tmp.mtbl.mapper*.part$( printf %04d $X ).subshard* open.mtbl.part$( printf %04d $X ).new' -- $( seq 0 $(( SHARDS - 1 )) )
+parallel -i -j $REDUCERS bash -c 'X={}; ./binmerger tmp.mtbl.mapper*.part$( printf %04d $X ).subshard* open.mtbl.part$( printf %04d $X ).new' -- $( seq 0 $(( SHARDS - 1 )) )
 
 TS_REDUCEDONE=$( date +%s%3N )
 REDUCEOUTPUT=$( du -sch open.mtbl.part*.new | tail -n 1 | cut -f1 | sed 's/\([^0-9]\)$/ \1/' )
-- 
2.39.2