jR<ddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z mZmZmZddlmZmZmZmZeeZGddejZGdd Zd Zd Zd Z d'dZ d(dZdZ dZ!dZ" d)dZ#dZ$edkrdZ%dZ%ej&dZ'ej&d Z(ej&d!Z)ej&d"Z*ej&d#Z+ej,e'j-d$zd%zZ.ej/e*e)zdkZ0de.e0de0d&f<e#Z1e$e1e.Z2dSdS)*N)layers)NoteHeadConstant)get_bbox get_centermerge_nearby_bboxrm_merge_overlap_bbox to_rgb_img) get_unit_sizefind_closest_staffsget_global_unit_size get_loggerc6eZdZdZdZdZdZdZdZdZ dZ d Z d Z d S) NoteTyper N) __name__ __module__ __qualname__WHOLEHALFQUARTEREIGHTH SIXTEENTH THIRTY_SECOND SIXTY_FOURTHTRIPLETOTHERS HALF_OR_WHOLE/Users/tedpeters/Documents/Projects/Claude Projects/Misc Tools/notename-sheets/.venv/lib/python3.11/site-packages/oemer/notehead_extraction.pyrrsB E DG FIMLG FMMMr'rcteZdZdZedefdZejdefdZdefdZdZ dZ d Z d S) NoteHeadcg|_d|_d|_d|_d|_d|_d|_d|_d|_d|_ d|_ d|_ d|_ d|_ dS)NF)pointspitchhas_dotbboxstem_up stem_righttrackgroupstaff_line_posinvalidid note_group_idsfn_labelselfs r(__init__zNoteHead.__init__ sl#%  " !% ! $  #'" "&!% r'returnch|jr%td|jddS|jS)NzNote z is not a valid note.)r5loggerwarningr6r9r:s r(labelzNoteHead.label2s9 <  NNA47AAA B B B4{r'rAcl|j%td|jddS||_dS)NzThe label has been set to: zG. Use 'force_set_label' instead if you really want to modify the label.)r9r?debugr;rAs r(rAzNoteHead.label9sO ; " LLct{ccc d d d F r'ctd|jd|t|tsJ||_dS)NzForce set label from z to )r?rCrA isinstancerr9rDs r(force_set_labelzNoteHead.force_set_labelAsH DTZDDUDDEEE%***** r'c>|j||fdSN)r,append)r;xys r( add_pointzNoteHead.add_pointFs" Aq6"""""r'c"|j|jkSrI)r4)r;nts r(__lt__zNoteHead.__lt__Is"R%666r'cd|jdt|jd|jd|jd|jd|jd|jd|jd |j d |j d |j d |j d |j dS)Nz Notehead z ( Points: z Bounding box: z Stem up: z Track: z Group: z Pitch: z Dot: z Label: z Staff line pos: z Is valid: z Note group ID: z Sharp/Flat/Natural: z ) )r6lenr,r/r0r2r3r-r.rAr4r5r7r8r:s r(__repr__zNoteHead.__repr__Ms 47  T[))  #y  ,          l     "&!4   $|+  !% 2  &*X    r'N) rrrr<propertyrrAsetterrGrMrPrSr&r'r(r*r*s%%%$xX  \8\X ###777r'r*ctt|dz }tjtj||f}tjtj|tj ||}tt|tj ztt|tj zf}tjtj|}tj|tj |}tjtj|ddz|ddzf}tj||S)Nrrr) introundcv2getStructuringElement MORPH_ELLIPSEerodedilateastypenpuint8nhcNOTEHEAD_MORPH_WIDTH_FACTORNOTEHEAD_MORPH_HEIGHT_FACTOR)pred unit_size small_size small_kersizekernelimgs r(morph_noteheadrk^sU9q=))**J)#*;j*=UVVI 9SZ BH 5 5yAA9 M MD E)C;; < <== E)C<< = =>> D &s'8$ ? ?F )DKK))6 2 2C  &s'84719d1gai:P Q QF :c6 " ""r'cR||d|d|d|df}tj|dk\}}t|dkrdStj||dzdz }tj||dzdz}|d||d|fS)Nrrrr)r_whererRminmax)r/ noteheadsregionys_topbottoms r( adjust_bboxrvls tAwtAwQQ7 8F HVAX  EB 2ww!||t &**tAw  "C VBZZ$q' !A %F GS$q'6 **r'c |d|dz }|d|dz }t|\}}tj|z}|}g} t||z t||dzz kr|d|d||df} ||d|d|df} t | |} t | |} | $| t | ||| $| t | ||t| dkr.g} | D]&} | t | ||'| } ntt||z }|dkrs||z}t|D]^}|dt|d||zz|dt|d|dz|zzf}| |_| S)Nrrrr) rraNOTEHEAD_SIZE_RATIOabsrvextendcheck_bbox_sizerRrWrXrangerJ)r/rprewhcen_xrsnote_wnote_hnew_bboxleft_box right_boxtmp_newbox num_notessub_hisub_boxs r(r{r{ws  Q$q'A Q$q'A$HE1  $y 0F FH 1v:Q\****GT!WeT!W5DGT!Wd1g6 x33 955    OOOHiKK L L L  OOOIy)LL M M M 8}}q G GC NN?3 9EE F F F Fa&j))** q==NE9%% ) )G$q'AeG+,,G$q'QqS%K/00  (((( Or'皙?r333333?r?cltjd}|dd}|dd} g} |D]} t| \} } t| | }| |tj|zzks| | krA| d| dz }| d| dz }|||zks |||zkrv|||ztjzks|||ztjzkr|| d| d| d| df}||z|z}||dkj}||kr| | | S)Nzonesrrrr) r get_layerrr raCLEF_ZONE_WIDTH_UNIT_RATIOrxrhrJ)bboxesnotehead min_h_ratio max_h_ratio min_w_ratio max_w_ratiomin_area_ratiormin_xmax_x valid_bboxesr/rcen_yrer~r}rq min_countcounts r(filter_notehead_bboxrsx  W % %E !HQKE "IbMEL""!$'' u!%//  EC:9DD D D%RW--  Gd1g  Gd1g  +% % %1y/D+D+D  +%c&== = = +-c.EEEE $q'$q'/4747?:;EN* vax % 9   D!!!! r'333333?c tdt||}t|}t |}g} |D]>} t t | } t| || } | | ?tdt| tdt| ||||||}tdt||S)NzMorph noteheads)rezDetected noteheads: %dzFiltering noteheadsrrrrrz&Detected noteheads after filtering: %d) r?rCrkrrr rr{rzrRr) rdglobal_unit_sizerrrrrnoter result_bboxesrres r(get_notehead_bboxrs  LL"### $*: ; ; ;D d^^F "6 * *FM""!:c??3 c433S!!!! LL)3}+=+=>>> LL&''' ! %F LL93v;;GGG Mr'ctj|}|j\}}t|D]}d}g}g}||kr|||fdkrn |dz }||k||kr|||fdkrn |dz }||k||krD|||fdkrn5|||||dz }||kD||kr/tj|}tj|}d|||f<t|D]}d}g}g}||kr|||fdkrn |dz }||k||kr|||fdkrn |dz }||k||krD|||fdkrn5|||||dz }||kD||kr/tj|}tj|}d|||f<Ȑ|S)Nrr)r_copyshaper|rJarray) rqtarr~r}yicurcand_ycand_xxis r( fill_holersc '&//C 9DAqAhh-(-(Agg2s7|a 1HCAggAgg2s7|q   1HCAggAgg2s7|a MM"    MM#    1HC Agg 77Xf%%FXf%%F"#C (( ( (BCFF''sBw.assign_group_track;sxBHxBHHHr'cg|] }|j Sr&)y_center).0ls r( zgen_notes..Us===!1:===r'r)r*r/r_rmziprMrr ry_uppery_lowerrelines enumeraterJinsertargminryrrRr4rX)rsymbolsnotesr/rqrrxsrLrKrrrst1st2 st_masterup_stlo_ststs_censteppos_cen tmp_interidxceninterppos_idxdiffposrs @r( gen_notesr+s E>> ZZaa$q'$q'/9:&1*%%B d1g  d1g BKK  DAq LLA         "$'' u&ue44S LCL ( (K57777CK77777  s # # #II), )D)DC::3PS*LE5~6!;Gw""5)))! ""5)))! "Q&==yttt'<=== !'#2#,// % %HCGCEN*a/F   V $ $ $ $$Y// , ,KC NN3q57F + + + +1:d?#g-T1A0BB)BF28G#4#4U#:;;<< w ' ' ' 'Wa ' ' ' ' ' 'B   \\wqzE)**Dt $$C!$B  wr{U*++Dt $$s7||3a7C #B  R Lr'ctjd}tjdtj}t j|tj|}tj |\}}|D]}|j }||d|d|d|df}ttj |} d| vr| dt| dkr}| } tj|| k\}} tj| } |d|dzdz } | | k}||_dS)Nstems_rests_pred)rr)dtyperrrr)rrr_onesr`rYr]r^scipyndimagerAr/setuniqueremoverRpoprmmeanr1)rstemskerenhanced_stemsst_maprsrrrqllsrArst_cen_xron_rights r(parse_stem_inforosC  / 0 0E '& ) ) )CZ RX 6 6<Y4YdqrrrJIJ""hsmmaQAAs1v 56vax % A:: 6"""'% fs1vs1v.QQ0@+qQQQ FuNNSVAXs1v,>@XZ]_jlmnnn ) ) )   c " " " "   S ! ! ! ! KK&'''Iw//KZ11L LL.///S&&''99"*"8 S LL7888 , &EE Lr'c P|}tj|}|D]z}|j\}}}}d}d} t j|||z|| zf||z|| zfdd|jr9t j||jjd|dz|ftj ddd{|S)Nrrrg?rr) rr_rr/rYrrArnamer) rori_imgrjrx1y1x2y2x_offsety_offsets r( draw_notesrs ,,..C (3--CllBB cBxKH58 R[7QS^`abbb : l KTZ_Q/"Q$S=UWZ\gij k k k Jr'__main__wind2last staff_predrrroriginal_image)rrr)rrrrr)rrrrr)rrrrrrrr)3enumrYnumpyr_ scipy.ndimageroemerroemer.constantrra oemer.bboxrrrrr oemer.utilsr r r r rr?Enumrr*rkrvr{rrrrrrrf_namerstaffrrrrrraarmrrrrr&r'r(r s 222222aaaaaaaaaaaaaa\\\\\\\\\\\\ H       ty   <<<<<<<<~ # # #+++***`++++bD444nAAAH###H #BBBBJ    z F F F \ * *Ef~..G F / 0 0Ev00Hf/00G t# $ $s *B "(8E>A% & &CBs1vs1v~ GIIE E2  BBB-r'