Mercurial > hg > vampy-host
comparison vamp/frames.py @ 84:d91a2285fbb2
naming: local variables snake_case
| author | Chris Cannam |
|---|---|
| date | Wed, 21 Jan 2015 13:04:01 +0000 |
| parents | a11b57e9fb0b |
| children | 3e5791890b65 |
comparison
equal
deleted
inserted
replaced
| 83:fbd084b81ece | 84:d91a2285fbb2 |
|---|---|
| 1 '''A high-level interface to the vampyhost extension module, for quickly and easily running Vamp audio analysis plugins on audio files and buffers.''' | 1 '''A high-level interface to the vampyhost extension module, for quickly and easily running Vamp audio analysis plugins on audio files and buffers.''' |
| 2 | 2 |
| 3 import numpy | 3 import numpy |
| 4 | 4 |
| 5 def frames_from_array(arr, stepSize, frameSize): | 5 def frames_from_array(arr, step_size, frameSize): |
| 6 """Generate a list of frames of size frameSize, extracted from the input array arr at stepSize intervals""" | 6 """Generate a list of frames of size frameSize, extracted from the input array arr at step_size intervals""" |
| 7 # presumably such a function exists in many places, but I need practice | 7 # presumably such a function exists in many places, but I need practice |
| 8 assert(stepSize > 0) | 8 assert(step_size > 0) |
| 9 if arr.ndim == 1: # turn 1d into 2d array with 1 channel | 9 if arr.ndim == 1: # turn 1d into 2d array with 1 channel |
| 10 arr = numpy.reshape(arr, (1, arr.shape[0])) | 10 arr = numpy.reshape(arr, (1, arr.shape[0])) |
| 11 assert(arr.ndim == 2) | 11 assert(arr.ndim == 2) |
| 12 n = arr.shape[1] | 12 n = arr.shape[1] |
| 13 i = 0 | 13 i = 0 |
| 16 w = frame.shape[1] | 16 w = frame.shape[1] |
| 17 if (w < frameSize): | 17 if (w < frameSize): |
| 18 pad = numpy.zeros((frame.shape[0], frameSize - w)) | 18 pad = numpy.zeros((frame.shape[0], frameSize - w)) |
| 19 frame = numpy.concatenate((frame, pad), 1) | 19 frame = numpy.concatenate((frame, pad), 1) |
| 20 yield frame | 20 yield frame |
| 21 i = i + stepSize | 21 i = i + step_size |
| 22 | 22 |
