restructure Python SFF demo

#19
swharden · Jul 25, 2020 · 65116e7 · 65116e7
1 parent 3ea73cc
commit 65116e7
Show file tree

Hide file tree

Showing 6 changed files with 24 additions and 36 deletions.
diff --git a/dev/sff/complex.sff b/dev/sff/complex.sff
diff --git a/dev/sff/hal.png b/dev/sff/hal.png
diff --git a/dev/sff/halMel.png b/dev/sff/halMel.png
diff --git a/dev/sff/sffDemo.py b/dev/sff/sffDemo.py
@@ -0,0 +1,24 @@
+import os
+import matplotlib.pyplot as plt
+import numpy as np
+import sffLib
+
+def plotSFF(filePath, show = False):
+
+    sf = sffLib.SpectrogramFile(filePath)
+
+    # rotate values for display as a pseudocolor mesh
+    rotatedValues = np.rot90(sf.values, 1)[::-1]
+    plt.pcolormesh(rotatedValues)
+
+    plt.colorbar()
+    plt.title(f"{filePath} Spectrogram")
+    plt.savefig(filePath.replace(".sff", ".png"))
+    if show:
+        plt.show()
+
+if __name__ == "__main__":
+    plotSFF("hal.sff")
+    plotSFF("halMel.sff")
+
+
diff --git a/dev/sff/sffDemoComplex.py b/dev/sff/sffDemoComplex.py
diff --git a/dev/sff/sffDemoSimple.py b/dev/sff/sffDemoSimple.py