audio_toolbox/plot_spec.py at master · amilo/audio_toolbox · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
#!/usr/bin/env python
# coding: utf-8


from librosa import load
import matplotlib.pyplot as plt
import librosa
import librosa.display
import numpy as np
import scipy as scipy
from scipy.signal import lfilter
import IPython.display as ipd
from math import pi
from math import tan
from math import pow
import seaborn as sns
from numpy import sin, cos, pi
import sys
import csv

from cycler import cycler

fs = 44100
# NFFT=0
# integ= 'slow'
#
#
# if integ == 'slow':
#     NFFT= 65436
# elif integ == 'fast':
#     NFFT= 8192


def plot_spectrogram(the_file, durata, P, integ):

    plt.figure(figsize=(20, 5))
    fig, ax = plt.subplots(figsize=(20, 5))

    NFFT=8192

    fft_size = NFFT
    sr=44100

    frame = NFFT/4

    ax = plt.gca()

    the_transform_mag = np.abs(librosa.core.stft(the_file, n_fft=fft_size, hop_length=None, win_length=None, window='hann'))
    librosa.display.specshow(librosa.amplitude_to_db(the_transform_mag,ref=-90),cmap='gray_r',sr=44100, y_axis='log',x_axis='frames')
    ax.xaxis.set_major_locator(plt.MultipleLocator(sr/frame*10))

    def format_func(value, tick_number):

        return int((value * frame)/fs)

    ax.xaxis.set_major_formatter(plt.FuncFormatter(format_func))


    plt.xlabel('Seconds')

    plt.title(P +'_'+ str(NFFT))

    # print("hey you")

    fig.tight_layout()


    plt.savefig(P+'_'  + str(NFFT)+'_'+ str(durata) +'.eps', dpi=150)


    # plt.show()


def main(files):

    try:
        for f in files:

            print('executing on: ', f)

            filebase = f[:-4]
            print('root is = ', filebase)

            mono, fs = load(f, sr=44100, mono=True)

            durata = len(mono)-1

            # partic = 'P01_D-mono_A.wav'
            partic = f[-16:-4]

            print(partic)


            plot_spectrogram(mono, durata, partic, 'fast')


        return 0
    except:
        return 1

if __name__ == "__main__":
    files = sys.argv[1:] # slices off the first argument (executable itself)
    exit (main(files))