o
    à¥µi&  ã                   @   sf   d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl	Z	d dl
mZ eƒ Zdd„ ZdS )é    N)ÚOrderedDict)Úcopyfile)Ú
get_loggerc                 C   sÐ  t j |¡st  |¡ t d|› d|› ¡ | ¡  dd¡ d¡}t j 	| d¡}t j 	| d¡}t j 	|d¡}|› d	t j
d
< |› dt j
d< t j |¡sNJ ‚t  |tjtjB tjB ¡ t  |› d|› d|› ¡ tt j 	| d¡t j 	|d¡ƒ tt j 	| d¡t j 	|d¡ƒ tt j 	| d¡t j 	|d¡ƒ tt j 	| d¡t j 	|d¡ƒ tt j 	| d¡t j 	|d¡ƒ tt j 	| d¡t j 	|d¡ƒ tt j 	| d¡t j 	|d¡ƒ tt j 	| d¡t j 	|d¡ƒ tt j 	| d¡t j 	|d¡ƒ t j 	|d¡}i }	g |	d< |D ]@}
tƒ }|
|d< d|d< d|d< d|d< d|d < d|d!< d|d"< d#|d$< d%|d&< d'|d(< d)|d*< d+|d,< d-|d.< |	d  |¡ qþtj|d/d0d1}tj|	|d2d3d4 | ¡  W d   ƒ d S 1 saw   Y  d S )5Nzmaking runtime resource in z for ú Ú ú,Útrainztrain/nnet-copyzkwsr.netz:$PATHÚPATHz:$LD_LIBRARYPATHÚLD_LIBRARY_PATHz --binary=true zkwsr.cclzkwsr.cfgzkwsr.gbgzkwsr.lexzkwsr.mdlzkwsr.mvnzkwsr.phnz	kwsr.treez
kwsr.priorzkeywords.jsonÚ	word_listÚnameÚwakeupÚtypeTÚ
activationÚis_maing        Úlm_boostÚam_boostÚ
threshold1g      ð¿Ú
threshold2g333333ã¿Úsubseg_thresholdg     €V@Úhigh_thresholdgš™™™™™Ù?Úmin_durg      @Úmax_durÚcommonccÚcc_nameÚwzutf-8)Úencodingé   F)ÚindentÚensure_ascii)ÚosÚpathÚexistsÚmakedirsÚloggerÚinfoÚstripÚreplaceÚsplitÚjoinÚenvironÚchmodÚstatÚS_IRWXUÚS_IRGRPÚS_IROTHÚsystemr   r   ÚappendÚcodecsÚopenÚjsonÚdumpÚclose)Ú	model_dirÚ	dest_pathÚ
kaldi_textÚkeywordsÚkeywords_listÚ
kaldi_pathÚ
kaldi_toolÚ	kaldi_netÚkeywords_grammarÚkeywords_rootÚkeywordÚone_dictÚfh© rD   úe/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/trainers/audio/kws_utils/runtime_utils.pyÚmake_runtime_res   sŽ   
þþþþþþþþþ
$þrF   )r2   r    Úrer,   ÚsysÚcollectionsr   Úshutilr   r4   Úmodelscope.utils.loggerr   r$   rF   rD   rD   rD   rE   Ú<module>   s    