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
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 )
- ~/nmu/pgn-extract/pgn-extract --startpos $START --endpos $END --startfilenum $PGNNUM -e -Wsessebin "$FILE" 2>/dev/null
+ COMP=0
+ if echo "$FILE" | grep -q ^comp:; then
+ COMP=1
+ FILE=$( echo "$FILE" | sed s/comp:// )
+ fi
+ START=$( ./find-pgn-split-point.sh "$FILE" $X $MAPPERS )
+ END=$( ./find-pgn-split-point.sh "$FILE" $(( X + 1 )) $MAPPERS )
+ ~/nmu/pgn-extract/pgn-extract --computerflag $COMP --startpos $START --endpos $END --startfilenum $PGNNUM -e -Wsessebin "$FILE" 2>/dev/null
PGNNUM=$(( PGNNUM + 1 ))
done
) | ./binloader - tmp.mtbl.mapper$X $SHARDS 500000
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/' )