NhanNguyen1309/audio-separation-api
0
1o
2 �Kj- � @ s� d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ dZ dZ3e4fd'd d
�Z d(dd�Z d)dd�Z
d*dd�Zd*dd�Zd+d d �Zd d lmZ ed d �G d!d"� d"��Zd,d-d%d&�ZdS ).� )� annotationsN)�StringIO)�Path)z,vamp:qm-vamp-plugins:qm-barbeattracker:beatsz*vamp:qm-vamp-plugins:qm-tempotracker:beats� ^@� duration_sec�float�bpm�return� list[float]c C sT d| }dg}|}|| k r |� t|d�� ||7 }|| k s
|� t| d�� t|| �S )Ng N@� � )�append�round�sanitize_beats)r r Z
beat_duration�beatsZ next_beat� r �TD:\workspace\nextjs\music_toolkit_projects\audio-separation-api\app\beat_tracking.py�build_fallback_beat_grid s �5r r c sj t �fdd�| D ��� � r� d dkr� �dd� � d t�d�kr*� �t�d�� � fdd�t� �D �S ) Nc s" h | ]
}t tt|d �� �d��qS )r r )r �min�max��.0�beat)r r r � <setcomp> s " z!sanitize_beats.<locals>.<setcomp>r r �����r c s, g | ]\}}|d ks|� |d kr|�qS )r � r )r �indexr )�cleanedr r �6<listcomp>" s , z"sanitize_beats.<locals>.<listcomp>)�sorted�insertr r
� enumerate)r r r )r r r r s r �csv_text�strc 7 C s` g }t �t| ��}|D ] }t|�dk rq z |�t|d �� W q ttfy* Y q w t||�S )N� r ) �csv�readerr �lenr
r � TypeError�8ValueErrorr )r" r r r& �rowr r r �parse_sonic_annotator_beats% s �9r+ �10audio_pathr c 11 C s� t �d�sg S tD ];}dd|t| �dddg}z tj|ddddd �}W n ttjfy. Y q w |jd12krDt |j13|�}t |�d krD| S q g S ) Nzsonic-annotatorz-dz-wr% z --csv-stdoutT�x F)Zcapture_output�textZtimeout�checkr r$ ) �shutil�which�SONIC_BEAT_DESCRIPTORSr# �14subprocess�run�OSErrorZSubprocessError�15returncoder+ �stdoutr' )r, r Z16descriptorZcommand�resultr r r r �!detect_beats_with_sonic_annotator2 s 17�18 �r9 c C s� zdd l }W n ty g Y S w z |jt| �ddd�\}}|jj||dd�\}}|j||d�}W n ty= g Y S w dd � |D �}t||�S )19Nr �"V T��srZmonoZframes)�yr< Zunits�r< c S � g | ]}t |��qS r �r r r r r r R � z-detect_beats_with_librosa.<locals>.<listcomp>)�librosa� Exception�loadr# r �20beat_track�frames_to_timer ) r, r rB r= r< �_� beat_frames�21beat_timesr r r r �detect_beats_with_librosaE s � �22rJ �23str | Pathc C s< t | �}ttfD ]}|||�}t|�dkr| S qt|�S )Nr$ )r r9 rJ r' r )r, r �pathZdetectorr r r r � detect_beatsV s 24 �rM )� dataclassT)�slotsc @ s6 e Zd ZU ded< ded< ded< ded< ded < d25S ) �BeatTrackingResultr r r26 r �27confidencer# �sourcez list[str]�warningsN)�__name__�28__module__� __qualname__�__annotations__r r r r rP a s 29 rP �chords� list[ChordSegmentRecord] | Nonec C s� t | �}g }z30dd l}dd l}W n8 tyH |�d� d}zdd l}|�t|��j}W n ty5 Y nw d}t||�}t ||dd|d� Y S w zg|j31t|�dd d32�\} }33t |j | |34d ��}d } |j
j| |35| d
�} ddlm}
m} |
| |36| | |�}||�}|jj| |37| | |jd�\}}|j||38d�}dd� |D �}t||�}t t|jd�|t|jd�d|d�W S ty� } z@|�d� |�dt|�j� �� d}zdd l}|�t|��j}W n ty� Y nw d}t||�}t ||dd|d�W Y d }~S d }~ww )Nr Zbeat_tracking_fallback_usedg �r@r r �fallback)r r rQ rR rS r: Tr; )r= r< i )r= r< �39hop_length)�generate_bpm_candidates�choose_best_bpm_candidate)r= r< �onset_enveloper[ Z start_bpmr> c S r? r r@ r r r r r � rA z*detect_beats_and_tempo.<locals>.<listcomp>r$ r rB Zlibrosa_error_)r rB Z soundfilerC r
�infor# Zdurationr rP rD r Z get_durationZonsetZonset_strength�app.bpm_candidatesr\ r] r rE r rF r r �score�typerT )r, rX rL rS rB Zsfr r r r= r< r[ r^ r\ r] �40candidates�selected�temporH rI �excr r r �detect_beats_and_tempoj s� 41 �4243�� 44�454647�48 �49���rg )r r r r r r50 )r r51 r r r r52 )r" r# r r r r53 )r, r r r r r54 )r, rK r r r r55 )N)r, rK rX rY r rP )�56__future__r r% r0 r3 �ior Zpathlibr r2 Z FALLBACK_BPMr r r+ r9 rJ rM � dataclassesrN rP rg r r r r �<module> s$ 57 58 59
6061