jE ddlmZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z ddlZddlmZmZddlmZmZddlmZddlZGd d eZGd d Zd eefdZd eefdZdee e e fe fdee e e fe fdededef dZeZeD]Ze edS))EnumN)Path)DictListSetOptionalTextTupleTypeUnion) PreprocessorsProtocol)create_protocolget_init)DatabaseceZdZdZdZdZdS) LoadingModerrN)__name__ __module__ __qualname__OVERRIDEKEEPERROR}/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/lib/python3.11/site-packages/pyannote/database/registry.pyrr,sH D EEErrceZdZdZddZejfdeee fdefdZ ej e fdeee fded e e fd Zdefd Z dd eedefd ZdZdejfdedeee fdefdZdZdS)RegistryzDatabase and experimental protocols registry Usage ----- >>> from pyannote.database import registry >>> registry.load_database("/path/to/first/database.yml") >>> registry.load_database("/path/to/second/database.yml") returnNcxt|_t|_t|_dSN)dictconfigssources databases)selfs r__init__zRegistry.__init__qs.B*. 04vv ,066rpathmodecv|||t|dS)aLoad YAML configuration file into the registry Parameters ---------- path : str or Path Path to YAML configuration file. mode : LoadingMode, optional Controls how to handle conflicts in protocol names. Defaults to overriding the existing protocol. Usage ----- >>> from pyannote.database import registry >>> registry.load_database("/path/to/database.yml") r+loadingN)_load_database_helperset_reload_meta_protocols)r(r*r+s r load_databasezRegistry.load_databases;* ""4dCEE"BBB ##%%%%%r database_ymlr.ct|}||vrdS||t |d5}t j|t j}dddn #1swxYwY|dt}t|ts|g}|D]G}t|}| s |j |z }| |||H|dt}|dd} | | |d<|D]\} } || | |||d t} | D]\} } t| ts| g} t}| D]Q}t|}| s |j |z }|t'|R||jt'| <||j|<dS) aBHelper function for recursive loading Parameters ---------- database_yml : Union[Text, Path] Path to the database.yml mode : LoadingMode, optional Controls how to handle conflicts in protocol names. Defaults to overriding the existing protocol. Nr)Loader Requirementsr- ProtocolsXr+ Databases)r expanduserresolveaddopenyamlload SafeLoaderpoplist isinstance is_absoluteparentr/getr$items_load_protocolsappendstrr&r%)r(r3r+r.fconfig requirementsrequirement_yaml protocolsxdb_name db_entriesr'value path_listpr*s rr/zRegistry._load_database_helpers$L))4466>>@@  7 " " F  L!!!, $ $ :Yq999F : : : : : : : : : : : : : : :zz.$&&99 ,-- *(>L , U U #$455 #//11 J#/#69I#I  & &'7dG & T T T TJJ{DFF33  MM#t $ $ =IcN$-??#4#4 O O GZ  *l N N N NJJ{DFF33 'oo// 3 3NGUeT** #'66I , ,Aww''))6'.5D  T++++)2DLW & &&, \"""s!B  BBc |j|}n>#t$r1|dkrd}nd}||}t|wxYw|di|S)zGet database by name Parameters ---------- database_name : str Database name. Returns ------- database : Database Database instance r9zCould not find any meta-protocol. Please refer to pyannote.database documentation to learn how to define them: https://github.com/pyannote/pyannote-databasezCould not find any protocol for "{name}" database. Please refer to pyannote.database documentation to learn how to define them: https://github.com/pyannote/pyannote-database)namer)r'KeyErrorformat ValueError)r( database_namekwargsdatabasemsgs r get_databasezRegistry.get_database s "~m4HH " " "##DQ jjmj44S// ! " x!!&!!!s ;A  preprocessorsc|d\}}}||}||||}||_|S)a]Get protocol by full name Parameters ---------- name : str Protocol full name (e.g. "Etape.SpeakerDiarization.TV") preprocessors : dict or (key, preprocessor) iterable When provided, each protocol item (dictionary) are preprocessed, such that item[key] = preprocessor(item). In case 'preprocessor' is not callable, it should be a string containing placeholder for item keys (e.g. {'audio': '/path/to/{uri}.wav'}) Returns ------- protocol : Protocol Protocol instance .)rb)splitra get_protocolrY)r(rYrbr] task_name protocol_namer_protocols rrfzRegistry.get_protocol-s\*37**S/// y-$$]33(( }M)   rc#K|jD]R}||}|D]&}||D]}|d|d|V'SdS)Nrd)r'ra get_tasks get_protocols)r(r]r_rgrhs r__iter__zRegistry.__iter__Ks!^ I IM((77H%//11 I I %-%;%;I%F%FIIM*HHYHHHHHHHHI I I IrrTc t|}t}|D]K\}}|D]1\}} t|}t|||| |} | *| |||f<2L||jvr%|j|j} t || |||d|D} t|tft| |d|j|<dS)aLoad all protocols from this database into the registry. Parameters ---------- db_name : _type_ Name of the database db_entries : dict Dict of all entries under this database (this should be tasks) database_yml : Union[Text, Path], optional Path to the database.yml file. Not required for X protocols, by default None Nc$g|] \\}}}|||fSrr).0taskp_namep_types r z,Registry._load_protocols..}s4   '=~fvT66 "   r)r) _protocols) rLr$rIrr'ru_merge_protocols_inplacetyperr) r(rSrTr3r+rQrg task_entriesriprotocol_entriesCustomProtocol old_protocols protocol_lists rrJzRegistry._load_protocolsRsD&g,,4866 '1'7'7'9'9 B B #I|.:.@.@.B.B B B**x==!0Y2BL""")3A 9h/00 B dn $ $ N73>M $=$      AJARAR   #'  K!-00 J J# # wrc|jdd|jD]T\}}|dt }d|vr)|d|d|tjUdS)z>Reloads all meta protocols from all database.yml files loaded.r9Nr8r:) r'rCr%rIrHr$rJrr)r(db_ymlrNr's rr1zRegistry._reload_meta_protocolss 3%%%"l0022  NFF ;77Ii$$3k6J%  r)r!Nr#)rrr__doc__r)rrr r rr2rr0rr/rrarr rrfrmr$rJr1rrrr r gs62626262v(0&&D$J&&&&&6(, SUU G,G,D$J'G,G,T G,G,G,G,R!"x!"!"!"!"H>B#+M#: <III+/'0 2 2 2 D$J' 2  2 2 2 2 h     rr r!ctjdd}g}|dD]L}t |}|r||M|S)zParse PYANNOTE_DATABASE_CONFIG environment variable PYANNOTE_DATABASE_CONFIG may contain multiple paths separation by ";". Returns ------- paths : list of Path List of all YAML database file defined in PYANNOTE_DATABASE_CONF PYANNOTE_DATABASE_CONFIG;)osenvironrHrerr<is_filerK)contentpathsr*s r_env_config_pathsrsyjnn7<>  LL    Lrc>g}td}|r||tjdz }|r|||t z }|S)aGet paths to default YAML configuration files * $HOME/.pyannote/database.yml * $CWD/database.yml * PYANNOTE_DATABASE_CONFIG environment variable Returns ------- paths : list of Path List of existing default YAML configuration files z~/.pyannote/database.ymlz database.yml)rr<rrKcwdr)r home_db_yml cwd_db_ymls r_find_default_ymlsrsE122==??K" [!!!n,J! Z      E Lr new_protocolsr{r+rSr3cp|D]\}}||vr|\}}|d|d|} |tjkrtd| d|d|tjkrt jd| d|dj|tjkr t jd| d |d|||<|||<d S) aMerge new and old protocols inplace into the passed new_protocol. Warning, merging order might be counterintuitive : "KEEP" strategy keeps element from the OLD protocol and MODIFIES the new protocol. TODO: make it intuitive :) Parameters ---------- new_protocols : Dict[Tuple[Text, Text], Type] New protocol dict Maps (task,protocol) tuples to custom protocol classes old_protocols : Dict[Tuple[Text, Text], Type] Old protocols dict Maps (task,protocol) tuples to custom protocol classes. mode : LoadingMode How to handle override db_name : _type_ Name of the database (for logging/warning purposes) database_yml : str Path of the database.yml file (for logging/warning purposes) rdz Cannot load z protocol from 'z' as it already exists.zReplacing existing z! protocol by the one defined in 'z'.z Skipping z protocol defined in 'N)rIrr RuntimeErrorrwarningswarnr) rr{r+rSr3p_idold_pt_namerrrealnames rrvrvs*>%**,,(( e = !NFF!55F55V55H{((("b8bb\bbb --- e(eeUaeee))) eee eee', d##(M$  9((r)!enumrrpathlibrtypingrrrrr r r r r#pyannote.database.protocol.protocolr rcustomrrr_rr@rr rrrLrvregistryymlr2rrrrs> FFFFFFFFFFFFFFFFFFFFGGGGGGGG-------- $vkkkkkkkk\ 4:*DJ8;(dDj)4/0;(dDj)4/0;( ;( ;(  ;(;(;(;(~ 8::      C 3  r