import sffLib
import os
import matplotlib.pyplot as plt
import numpy as np
np.set_printoptions(precision=3)


def plotSFF(filePath, show=False):

    # access SFF details and data like this
    sf = sffLib.SpectrogramFile(filePath)
    print(sf.getDescription())
    print(sf.values)

    # rotate values for display as a pseudocolor mesh
    rotatedValues = np.rot90(sf.values, 1)[::-1]

    # plot the spectrogram
    plt.figure()
    plt.pcolormesh(rotatedValues)
    plt.colorbar()
    plt.title(f"{os.path.basename(filePath)} Spectrogram")
    plt.savefig(os.path.basename(filePath)+".png")
    if show:
        plt.show()
    plt.close()


if __name__ == "__main__":
    plotSFF("../hal.sff", True)
    plotSFF("../halMel.sff", True)