comparison scripts/PitchBihist.py @ 38:e5e8e8a96948 branch-tests

before refactoring OPMEllin

     def get_melody_from_file(self, melodia_file, stop_sec=None):
         if not os.path.exists(melodia_file):
             return []
         data = np.loadtxt(melodia_file, delimiter=',')
         times, freqs = (data[:, 0], data[:, 1])
-        #melody_sr = 1. / (times[1] - times[0])
         if stop_sec is not None:
             stop_idx = np.where(times < stop_sec)[0]
             times, freqs = times[stop_idx], freqs[stop_idx]
         freqs[freqs<=0] = np.nan
         melody = freqs
-        return melody#, melody_sr
+        return melody
 
 
     def get_melody_matrix(self, melody):
         n_bins = 60
         n_frames = len(melody)

         melody_matrix = np.zeros((n_bins, n_frames))
         for time, pitch in enumerate(melody_octave):
             if not np.isnan(pitch):
                 melody_matrix[int(pitch), time] = 1
         return melody_matrix
-
-
-    def bihist_from_melodia(self, filename='sample_melodia.csv', secondframedecomp=True, stop_sec=None):
-        #melody, melody_sr = self.get_melody_from_file(filename, stop_sec=stop_sec)
-        melody = self.get_melody_from_file(filename, stop_sec=stop_sec)
-        if len(melody) == 0:
-            return []
-        melody_matrix = self.get_melody_matrix(melody)
-        bihist = []
-        if secondframedecomp:
-            nbins, norigframes = melody_matrix.shape
-            win2 = int(round(self.win2sec * self.melody_sr))
-            hop2 = int(round(self.hop2sec * self.melody_sr))
-            if norigframes<=win2:
-                nframes = 1
-                win2 = norigframes
-            else:
-                nframes = int(np.ceil((norigframes-win2)/float(hop2)))
-            bihistframes = np.empty((nbins*nbins, nframes))
-            for i in range(nframes):  # loop over all 8-sec frames
-                frame = melody_matrix[:, (i*hop2):(i*hop2+win2)]
-                bihist = self.bihistogram(frame)
-                bihist = np.reshape(bihist, -1)
-                bihistframes[:, i] = bihist
-            bihist = bihistframes
-        else:
-            bihist = self.bihistogram(melody_matrix)
-        return bihist
 
 
     def bihistogram(self, spec, spec_sr=None, winsec=0.5, align=True):
         if spec_sr is None:
             # assume spec is melody_matrix with default sr

             B = np.roll(B, -ref, axis=0)
             B = np.roll(B, -ref, axis=1)
         return B
 
 
+    def bihist_from_melodia(self, filename='sample_melodia.csv', secondframedecomp=True, stop_sec=None):
+        melody = self.get_melody_from_file(filename, stop_sec=stop_sec)
+        if len(melody) == 0:
+            return []
+        melody_matrix = self.get_melody_matrix(melody)
+        bihist = []
+        if secondframedecomp:
+            nbins, norigframes = melody_matrix.shape
+            win2 = int(round(self.win2sec * self.melody_sr))
+            hop2 = int(round(self.hop2sec * self.melody_sr))
+            if norigframes<=win2:
+                nframes = 1
+                win2 = norigframes
+            else:
+                nframes = int(np.ceil((norigframes-win2)/float(hop2)))
+            bihistframes = np.empty((nbins*nbins, nframes))
+            for i in range(nframes):  # loop over all 8-sec frames
+                frame = melody_matrix[:, (i*hop2):(i*hop2+win2)]
+                bihist = self.bihistogram(frame)
+                bihist = np.reshape(bihist, -1)
+                bihistframes[:, i] = bihist
+            bihist = bihistframes
+        else:
+            bihist = self.bihistogram(melody_matrix)
+        return bihist
+
+
 if __name__ == '__main__':
     pb = PitchBihist()
-    pb.bihist_from_melodia(filename='vamp_melodia.csv')
+    melody = np.concatenate([660 * np.ones(250), 440 * np.ones(500)])
+    melody_matrix = pb.get_melody_matrix(melody)
+    pb.bihistogram(melody_matrix, align=True)
+