In [1]:

    

%matplotlib inline
import matplotlib
import matplotlib.pyplot as plt
#import librosa
import os
import sys
from random import sample
import hdf5_getters
import numpy as np
from dtw import dtw
from numpy.linalg import norm

matplotlib.rcParams['figure.figsize'] = (10.0, 8.0)


    

In [2]:

    

def norm2(x, y):
    return norm(x - y, ord=2)

def get_song_info(h5):
    print '%s - %s | (%s) | %s bpm' % (hdf5_getters.get_artist_name(h5), hdf5_getters.get_title(h5), hdf5_getters.get_year(h5), hdf5_getters.get_tempo(h5))


    

In [3]:

    

h5list = []
pitches = []
rootdir = './data/'

for subdir, dirs, files in os.walk(rootdir):
    for f in files:
        if f.lower().endswith('.h5'):
            h5list.append(hdf5_getters.open_h5_file_read(os.path.join(subdir, f)))
#            get_song_info(h5list[-1])
            pitches.append(hdf5_getters.get_segments_pitches(h5list[-1]))
            h5list[-1].close()
h5sample = sample(pitches, 100)


    

In [6]:

    

[ dtw(x, y, dist=norm2) for i, x in enumerate(h5sample) for j, y in enumerate(h5sample) if i != j ]


    

    




---------------------------------------------------------------------------
KeyboardInterrupt                         Traceback (most recent call last)
<ipython-input-6-f94d2754505f> in <module>()
----> 1 [ dtw(x, y, dist=norm2) for i, x in enumerate(h5sample) for j, y in enumerate(h5sample) if i != j ]

/usr/local/lib/python2.7/site-packages/dtw.pyc in dtw(x, y, dist)
     32     for i in range(r):
     33         for j in range(c):
---> 34             D[i+1, j+1] += min(D[i, j], D[i, j+1], D[i+1, j])
     35 
     36     D = D[1:, 1:]

KeyboardInterrupt: