jz=dZddlmZddlZddlmZddlmZddl Z ddl m Z ddl m Z mZmZmZmZddlZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZmZdedDZ de dedeegeffdZ!dZ"dededefdZ#de de de dededef dZ$dedede%fd Z& d+d!e d"e d#e d$ededef d%Z' d+d!e d"e d#e d$ededef d&Z(d'Z)d!e d"e d#e fd(Z*d!e d"e d#e d)ededee+dff d*Z,dS),zCustom protocols Protocols: MyDatabase: Collection: MyProtocol: train: uris: xxx.lst annotation: xxx.rttm annotated: xxx.uem )PathN)protocol ProtocolFile)Number)TextDictCallableAnyUnion)Subset)SegmentationProtocol)SpeakerDiarizationProtocol) entry_points) get_annotated)load_lst load_trialci|] }|j| S)name).0eps {/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib/python3.11/site-packages/pyannote/database/custom.py rFs,    GR   zpyannote.database.loader)grouptemplate database_ymlreturnct}|jtvrd|jd}t|t|jdt ffd }|S)afGet data loader based on template Parameters ---------- template : str Path format template (e.g. "/path/to/{uri}.csv"). Extension (here ".csv") determined which data loader to use. database_yml : Path Path to YAML configuration file, to which `template` is relative. Defaults to assume that `template` is absolute or relative to current working directory. Returns ------- data_loader : Callable[[ProtocolFile], Any] Callable that takes a ProtocolFile and returns some data. See also -------- pyannote.database.loader zNo loader for files with '' suffix current_filec ttjdit|}|sd|dd}t ||}||S)NNo such file or directory: 'z' (via 'z ' template).r) resolve_pathrformatabsis_fileFileNotFoundError)r#pathmsgloaderLoaderrrs rloadzTemplate..loadjsD!E!E3|3D3D!E!EFF UU||~~ )UUUxUUUC#C(( (vl###r)rsuffixLOADERS ValueErrorr/r)rrr+r,r/r.s`` @rTemplater3Ls. >>D {'!!@4;@@@oo T[ ! & & ( (F $< $ $ $ $ $ $ $ $ Krc"dtffd }|S)Nr#cSNr)r#values rr/zNumericValue..loadys rr)r7r/s` r NumericValuer8xs-< Krr+c|}|r|S|j|z }|r|Sd|d}t|)aVResolve path Parameters ---------- path : `Path` Path. Can be either absolute, relative to current working directory, or relative to `database_yml` parent directory. database_yml : `Path` Path to YAML configuration file. Returns ------- resolved_path: `Path` Resolved path. zCould not find file "z".) expanduserr)parentr*)r+r relative_pathr,s rr&r&~so" ??  D ||~~! %+d2  " " ! *$ * * *C C  r meta_database meta_task meta_protocol meta_subsetsubset_entriesc#Kddlm}|D]C\}}||} |D]&} | d} t | | D]} | V'DdS)a8Meta-protocol method that iterates over a subset Parameters ---------- meta_database : str "X" meta_task : str Task name (e.g. SpeakerDiarization, SpeakerVerification) meta_protocol : str Protocol name (e.g. MyProtocol) meta_subset : {"train", "development", "test"} Subset subset_entries : dict Subset entries. Etape.SpeakerDiarization.TV: [train] REPERE.SpeakerDiarization.Phase1: [train, development] REPERE.SpeakerDiarization.Phase2: [train, development] r)registry_iterN)rCitems get_protocolgetattr) r=r>r?r@rArrCrsubsetspartial_protocolsubset method_namefiles rmeta_subset_iterrNs8+1133'#00::  F#***K> 0+>>@@     rentriesc`t}|D]\}}|dks|dkrt|trt |||<;t t j|\}}}}tt|tdgz dk}|rI| dr d}tj ||dd}t||||<tt!||} | sd| d }t%|| jt(vrd | jd }t+|t(| j} | | ||<|S) aLoads all Loaders for data type specified in 'entries' into a dict. Parameters ---------- entries : Dict, optional Subset entries (eg 'uri', 'annotated', 'annotation', ...) database_yml : Path, optional Path to the 'database.yml' file Returns ------- dict A dictionary mapping each key of entry (except 'uri' and 'trial') to a function that given a ProtocolFile returns the data type related to this entry. uritrialNr_zSince version 4.1, pyannote.database is smart enough to know when paths defined in 'database.yml' contains placeholders. Remove the underscore (_) prefix to get rid of this warning.rr%'zNo loader for file with 'r")dictrF isinstancerr8zipstring Formatterparselenset startswithwarningswarnr3r&rr)r*r0r1 TypeErrorr/) rOr lazy_loaderkeyr7rS placeholders is_templater,r+r.s rgather_loadersres(&&Kmmoo5,5, U %<<3'>>  eV $ $ +E22K  !$V%5%7%7%=%=e%D%D E<A#l++c4&kk9::Q> ( ,$$ "S  c"""abb '|<> - en 7  fo I I I4 I I( I IV I I8D I I I   c-....X(WWTWWHWWvWWoo LcL99 : :D |LLLK  X L L8 LS^         rc #Kt||}t|d<t}tt t |d|D]} | d| d} } | |vr+|t| ||d||| <| |vr+|t| ||d||| <| d|| || d Vd S) a Parameters ---------- database : str Database name (e.g. MyDatabase) task : str Task name (e.g. SpeakerDiarization, SpeakerVerification) protocol : str Protocol name (e.g. MyProtocol) subset : {"train", "development", "test"} Subset entries : dict Subset entries. database_yml : `Path` Path to the 'database.yml' file rktry_withrRuri1uri2rlrm reference)rwfile1file2N)rerrUrr&r preprocessr) rprfrgrrKrOrrafilesrRrurvs r subset_trialr|Is:6!|LLLK+K '+ffELgg.>)?)?NNOO  6]E&Md u  // h&II$E$K u  // h&II$E$K{+4[4[      %  rcfd}|S)Nctt|j|D] }|j| dSr6)super __class____init__register_protocol)rpr protocolss rinitzget_init..initsM dnd##,,...! . .H "D "H - - - . .rr)rrs` rget_initrs#..... Krc|d|d|S)N__r)rfrgrs rget_custom_protocol_class_namers , ,$ , ,( , ,,rprotocol_entriesc tt|dkrdn|d}n/#t$r"d|d|d|d|d }t|YdSwxYw|dkrd |i}t }t |t rd |vr|d |d <t |trG|d d}|*|d kr$d |d|d|d|d }t|d|d <n||d <t } | D]\} } | dvr*d|d|d|d| d|d| d }tj |3| d} |d kr#tj t|||| | || | <atjt f|||| | |d|| | <d| vr&tjt$|||| | || | d<t'|||} t)| |f| }|t+| <|S)zCreate new protocol class Parameters ---------- database : str task : str protocol : str protocol_entries : dict Returns ------- CustomProtocol : type or None Protocolz Ignoring 'rjz' protocols found in z because 'z' tasks are not supported yet.N Collectionr{classesscopeXrTz ' found in za does not define the 'scope' of speaker labels (file, database, or global). Setting it to 'file'.rM)r{train developmenttest train_trialdevelopment_trial test_trialz ' entries are not supported yet.rD)rfrgrrKrOrrR_trial)rHprotocol_moduleAttributeErrorprintrU issubclassrpoprrFr^r_ functoolspartialrN partialmethodrrkeysr|rtypeglobals)rfrgrrr base_classr,rqrmethodsrKrArLcustom_protocol_class_nameCustomProtocolClasss rcreate_protocolrs{.  4:+=+=ZZdCTCTCT    = = =D = =| = = = = =  c tt $ |#%56vvH*233B ( ( ("2"6"6y"A"AHY *899 & $$Wd33 =X__dHddtddhdd<ddd  #JJJ &HW  !&HW ffG"2"8"8":":33    EXEEEExEE&EE\EE"EEE  M#    &&& s??#,#4 $$GK $-#: $!!&) $ $ $ $GK .--////-6-D %%!*!-...6)))*"@$""9J='RR,?GII() s #(AA)NNNNNN)-__doc__pathlibrrXrErr#pyannote.database.protocol.protocolrr^numbersrtypingr r r r r rprotocol.protocolrprotocol.segmentationrprotocol.speaker_diarizationrimportlib.metadatarutilrr-rrr1r3r8r&rNrUrerrr|rrrrrrrrsB   *)))))<<<<<<33333333333333%%%%%%777777DDDDDD++++++((((((((  l!;<<<    )t)4)Hl^S=P4Q))))X !t!4!D!!!!>#### #  #  ####LL LL LLLLb5 5 5  5  5  5  5 5 5 5 5 r8 8 8  8  8  8  8 8 8 8 8 v-T------BB BB B  B  4: BBBBBBr