From 0bf151940966d937a1e34380ef6eea0bc9ba61f2 Mon Sep 17 00:00:00 2001
From: ml server <ml@spacecruft.org>
Date: Tue, 28 Jan 2020 18:17:41 -0700
Subject: [PATCH] prep txmode waterfalls into respective subdirs

---
 wut-dl-sort-txmode | 23 ++++++++++++++---------
 1 file changed, 14 insertions(+), 9 deletions(-)

diff --git a/wut-dl-sort-txmode b/wut-dl-sort-txmode
index ff4eb5a..b1805b6 100755
--- a/wut-dl-sort-txmode
+++ b/wut-dl-sort-txmode
@@ -25,27 +25,31 @@
 #
 # Possible vetted_status: bad, failed, good, null, unknown.
 
-cd /srv/satnogs
 
 OBSENC="$1"
 OBSIDMIN="$2"
 OBSIDMAX="$3"
 OBSID=$OBSIDMIN
+DATADIR="/srv/satnogs/data/txmodes/$OBSENC"
+DOWNDIR="/srv/satnogs/download"
+
+mkdir -p $DATADIR
+cd $DATADIR || exit
 
 # Enable the following if you want to download waterfalls in this range:
 #echo "Downloading Waterfalls"
 #./wut-water-range $OBSIDMIN $OBSIDMAX
 
 # XXX remove data/train and data/val directories XXX
-echo "Removing data/ subdirectories"
-rm -rf data/train data/val
+echo "Removing subdirectories"
+rm -rf train/ val/
 # Create new empty dirs
-mkdir -p data/train/good data/train/bad data/train/failed
-mkdir -p data/val/good   data/val/bad   data/val/failed
+mkdir -p train/good/ train/bad/ train/failed/
+mkdir -p val/good/   val/bad/   val/failed/
 
 # Then parse each file and link appropriately
 echo "Parsing download/ directory for observation IDs $OBSIDMIN to $OBSIDMAX"
-cd download/ || exit
+cd $DOWNDIR || exit
 
 while [ $OBSID -lt $OBSIDMAX ]
         do cd $OBSID
@@ -59,11 +63,11 @@ while [ $OBSID -lt $OBSIDMAX ]
 			CLASS_DIR="val"
 		fi
 		case "$VET" in
-			bad) ln waterfall_$OBSID_*.png ../../data/$CLASS_DIR/$VET/
+			bad) ln waterfall_$OBSID_*.png $DATADIR/$CLASS_DIR/$VET/
 			;;
-			good) ln waterfall_$OBSID_*.png ../../data/$CLASS_DIR/$VET/
+			good) ln waterfall_$OBSID_*.png $DATADIR/$CLASS_DIR/$VET/
 			;;
-			failed) ln waterfall_$OBSID_*.png ../../data/$CLASS_DIR/$VET/
+			failed) ln waterfall_$OBSID_*.png $DATADIR/$CLASS_DIR/$VET/
 			;;
 			null) echo "null, not copying"
 			;;
@@ -74,3 +78,4 @@ while [ $OBSID -lt $OBSIDMAX ]
         let OBSID=$OBSID+1
 	cd ..
 done
+