|
|
import json |
|
|
import pathlib |
|
|
from collections import OrderedDict |
|
|
|
|
|
import click |
|
|
|
|
|
|
|
|
@click.command(help='Eliminate short slur notes in DS files') |
|
|
@click.argument('ds_dir', metavar='DS_DIR') |
|
|
@click.argument('threshold', type=float, metavar='THRESHOLD') |
|
|
def eliminate_short( |
|
|
ds_dir, |
|
|
threshold: float |
|
|
): |
|
|
ds_dir = pathlib.Path(ds_dir).resolve() |
|
|
assert ds_dir.exists(), 'The directory of DS files does not exist.' |
|
|
|
|
|
for ds in ds_dir.iterdir(): |
|
|
if not ds.is_file() or ds.suffix != '.ds': |
|
|
continue |
|
|
|
|
|
with open(ds, 'r', encoding='utf8') as f: |
|
|
params = json.load(f) |
|
|
if not isinstance(params, list): |
|
|
params = [params] |
|
|
params = [OrderedDict(p) for p in params] |
|
|
|
|
|
for param in params: |
|
|
note_list = [ |
|
|
(note, float(dur), bool(int(slur))) |
|
|
for note, dur, slur |
|
|
in zip(param['note_seq'].split(), param['note_dur'].split(), param['note_slur'].split()) |
|
|
] |
|
|
word_note_div = [] |
|
|
cache = [] |
|
|
for note in note_list: |
|
|
if len(cache) == 0 or note[2]: |
|
|
cache.append(note) |
|
|
else: |
|
|
word_note_div.append(cache) |
|
|
cache = [note] |
|
|
if len(cache) > 0: |
|
|
word_note_div.append(cache) |
|
|
|
|
|
word_note_div_new = [] |
|
|
for i in range(len(word_note_div)): |
|
|
word_note_seq = word_note_div[i] |
|
|
if len(word_note_seq) == 1 or all(n[1] < threshold for n in word_note_seq): |
|
|
word_note_div_new.append(word_note_seq) |
|
|
continue |
|
|
|
|
|
word_note_seq_new = [] |
|
|
j = 0 |
|
|
prev_merge = 0. |
|
|
while word_note_seq[j][1] < threshold: |
|
|
|
|
|
prev_merge += word_note_seq[j][1] |
|
|
j += 1 |
|
|
|
|
|
while j < len(word_note_seq): |
|
|
k = j + 1 |
|
|
while k < len(word_note_seq) and word_note_seq[k][1] < threshold: |
|
|
k += 1 |
|
|
post_merge = sum(n[1] for n in word_note_seq[j + 1: k]) |
|
|
if k < len(word_note_seq): |
|
|
post_merge /= 2 |
|
|
word_note_seq_new.append( |
|
|
(word_note_seq[j][0], prev_merge + word_note_seq[j][1] + post_merge, False) |
|
|
) |
|
|
prev_merge = post_merge |
|
|
j = k |
|
|
|
|
|
word_note_div_new.append(word_note_seq_new) |
|
|
|
|
|
note_seq_new = [] |
|
|
note_dur_new = [] |
|
|
note_slur_new = [] |
|
|
for word_note_seq in word_note_div_new: |
|
|
note_seq_new += [n[0] for n in word_note_seq] |
|
|
note_dur_new += [n[1] for n in word_note_seq] |
|
|
note_slur_new += [pos > 0 for pos in range(len(word_note_seq))] |
|
|
param['note_seq'] = ' '.join(note_seq_new) |
|
|
param['note_dur'] = ' '.join(str(round(d, 6)) for d in note_dur_new) |
|
|
param['note_slur'] = ' '.join(str(int(s)) for s in note_slur_new) |
|
|
|
|
|
with open(ds, 'w', encoding='utf8') as f: |
|
|
json.dump(params, f, ensure_ascii=False, indent=2) |
|
|
|
|
|
|
|
|
if __name__ == '__main__': |
|
|
eliminate_short() |
|
|
|
