Add sdft test plot #5

Author: jurihock

python/stftpitchshift/main.py

@@ -1,7 +1,9 @@
 from stftpitchshift import StftPitchShift
 from stftpitchshift import __version__ as version
 from stftpitchshift.io import read, write
-from stftpitchshift.stft import stft, istft, spectrogram
+from stftpitchshift.plot import spectrogram
+from stftpitchshift.sdft import sdft
+from stftpitchshift.stft import stft
 
 import click
 import matplotlib.pyplot as plot
@@ -51,9 +53,18 @@ def cent(value): return pow(2, float(re.match('([+,-]?\\d+){1}([+,-]\\d+){0,1}',
 
         for channel in range(channels):
 
+            # STFT
             framesX = stft(x[:, channel], framesize, hopsize)
             framesY = stft(y[:, channel], framesize, hopsize)
 
+            # SDFT
+            # framesX = sdft(x[:, channel], framesize // 2)
+            # framesY = sdft(y[:, channel], framesize // 2)
+
+            # TEST
+            # framesX = stft(x[:, channel], framesize, hopsize)
+            # framesY = sdft(x[:, channel], framesize // 2)
+
             figure = plot.figure(f'Channel {channel+1}/{channels}')
 
             spectrogramX = figure.add_subplot(2, 1, 1, title='Input Spectrogram')