Support '-' argument as stdin input in log_mixer

fenrir-naru · Oct 3, 2023 · 3b025c0 · 3b025c0
1 parent ee2d58f
commit 3b025c0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tool/log_mixer.rb b/tool/log_mixer.rb
@@ -236,8 +236,10 @@ def read_chunk
   :imu => ARGV.shift,
 }
 $log_mix.call({:readers => [
-  IMU_CSV::new(open(src[:imu], 'r')),
-  GPS_UBX::new(open(src[:gps], 'rb')), 
+  IMU_CSV::new((src[:imu] == '-') ? $stdin : open(src[:imu], 'r')),
+  GPS_UBX::new((src[:gps] == '-') \
+      ? proc{STDIN.binmode; $stdin}.call \
+      : open(src[:gps], 'rb')), 
 ]})
 
 end