From a6be9a99d080e095a2bfdd46f832d66462f8a14f Mon Sep 17 00:00:00 2001
From: Keith Bloemer <32459398+GuitarML@users.noreply.github.com>
Date: Tue, 4 Oct 2022 13:01:53 -0500
Subject: [PATCH] Update prepare.py

---
 prepare.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/prepare.py b/prepare.py
index 7c04829..230ca87 100644
--- a/prepare.py
+++ b/prepare.py
@@ -15,6 +15,30 @@ def prepare(args):
     out_rate, out_data = wavfile.read(args.out_file)
     assert in_rate == out_rate, "in_file and out_file must have same sample rate"
 
+    # Trim the length of audio to equal the smaller wav file
+    if len(in_data) > len(out_data):
+      print("Trimming input audio to match output audio")
+      in_data = in_data[0:len(out_data)]
+    if len(out_data) > len(in_data): 
+      print("Trimming output audio to match input audio")
+      out_data = out_data[0:len(in_data)]
+
+    #If stereo data, use channel 0
+    if len(in_data.shape) > 1:
+        print("[WARNING] Stereo data detected for in_data, only using first channel (left channel)")
+        in_data = in_data[:,0]
+    if len(out_data.shape) > 1:
+        print("[WARNING] Stereo data detected for out_data, only using first channel (left channel)")
+        out_data = out_data[:,0]    
+        
+    # Convert PCM16 to FP32
+    if in_data.dtype == "int16":
+        in_data = in_data/32767
+        print("In data converted from PCM16 to FP32")
+    if out_data.dtype == "int16":
+        out_data = out_data/32767
+        print("Out data converted from PCM16 to FP32")
+    
     #normalize data
     if args.normalize == True:
         in_data = normalize(in_data)