Backup_Repos/beets
1
0
Fork 0
mirror of https://github.com/beetbox/beets.git synced 2025-12-26 18:43:38 +01:00
Code Issues Projects Releases Wiki Activity

convert echonest bpm to value between 0-1

Browse source
This commit is contained in:
Adam M 2014-03-08 16:12:29 +00:00
parent 1e7fc76817
commit a0faa530f9
1 changed files with 17 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
beetsplug/echonest.py
View file

@ -60,7 +60,7 @@ def _splitstrip(string, delim=u','):
return [s.strip() for s in string.split(delim)]
def diff(item1, item2, attributes):
def diff(item1, item2, attributes, max_bpm):
result = 0.0
for attr in attributes:
try:
@ -70,13 +70,28 @@ def diff(item1, item2, attributes):
)
except TypeError:
result += 1.0
try:
result += abs(
float(item1.get('bpm', None)) / max_bpm -
float(item2.get('bpm', None)) / max_bpm
)
except TypeError:
result += 1.0
return result
def similar(lib, src_item, threshold=0.15):
max_bpm = reduce(lambda x, y: max(x, y),
[float(x.get('bpm', 0.0)) for x in lib.items()], 1.0)
attributes = ATTRIBUTES.values()
attributes.remove('bpm')
for item in lib.items():
if item.path != src_item.path:
d = diff(item, src_item, ATTRIBUTES.values())
d = diff(item, src_item, attributes, max_bpm)
if d < threshold:
print(u'{1:2.2f}: {0}'.format(item.path, d))