VIS code

1 files changed, 23 insertions, 8 deletions

diff --git a/util/view.py b/util/view.py
index b6e95fe..60b3a47 100644
--- a/util/view.py
+++ b/util/view.py
@@ -1,11 +1,26 @@
-import sys
-import numpy as np
 import matplotlib.pyplot as plt
+import numpy as np
+import wave
+import sys
+
+
+spf = wave.open('test.wav','r')
+
+#Extract Raw Audio from Wav File
+signal = spf.readframes(-1)
+signal = np.fromstring(signal, 'Int16')
+signal = signal.astype(np.float32)
+
+
+#If Stereo
+if spf.getnchannels() == 2:
+    print('Just mono files')
+    sys.exit(0)
 
-f = open('test.csv')
-lines = [int(l.replace('\n', '')) for l in f.readlines()[:-2]]
-f.close()
+#plt.figure(1)
+#plt.title('Signal Wave...')
+#plt.plot(signal)
+#plt.show()
 
-s = np.array(lines)
-plt.plot(s)
-plt.show()
+plt.specgram(signal, Fs=48000)
+plt.show()
\ No newline at end of file