a è6“iÇ=ã @sUdZddlZddlZddlZddlZddlZddlZddlm Z ddl mZmZddl mZmZddlmZddlmZd Zdaeedfed <edœdd „Zd?eeedœdd„ZdZdZdZdZd@eedœdd„ZdAeedœdd„ZdBeedœdd„Z dCeedœdd„Z!eeeee"e"dœd d!„Z#eeee"e"d"œd#d$„Z$dDd%d&œeeeee"e"d'œd(d)„Z%eeeee e gdfee"d*œd+d,„Z&dd-œee eedfdd.œd/d0„Z'ee e dd1œd2d3„Z(ee e dd1œd4d5„Z)dEeeeedd6œd7d8„Z*dFeeeeeee"d9œd:d;„Z+e,eeedd<œd=d>„Z-dS)Gz4Download and setup DLLs to upgrade various upscalerséN)ÚPath)ÚCallableÚUnion)ÚunquoteÚurlparseé)Úlog)Úconfigzihttps://raw.githubusercontent.com/beeradmoore/dlss-swapper-manifest-builder/refs/heads/main/manifest.jsonÚ__manifest_json)Úreturnc CsœtdurtStjj d¡}|jddd| d¡}iazBtjjt dd}t | ¡¡aWdƒn1sl0YWnDt y¼}z,t dt ›d¡t t|ƒ¡WYd}~nHd}~00|jd d d }| t t¡¡Wdƒn1sò0YzPtsJ| ¡rJ|jd d}t | ¡¡aWdƒn1s@0YWnJt y–}z0t dt|ƒ›d¡t t|ƒ¡WYd}~n d}~00tS) NÚ upscalersT©ÚparentsÚexist_okz manifest.jsoné ©ÚtimeoutzFailed to download "ú"Úwúutf-8©Úencodingz Failed to read cached manifest ")r r ÚpathÚ cache_dirÚjoinpathÚmkdirÚurllibÚrequestÚurlopenÚ__manifest_urlÚjsonÚloadsÚreadÚ ExceptionrÚcritÚreprÚopenÚwriteÚdumpsÚexistsÚstr)rZcached_manifestÚurl_fdÚeZmanifest_fd©r-úb/home/james-whalen/.local/share/Steam/compatibilitytools.d/GE-Proton10-32/protonfixes/upscalers.pyÚ__get_manifests, 0$.2$r/Údefault)ÚupscalerÚversionrc CsŠtƒ|}ttdd„|ƒƒ}t|ƒD]4}||dvr$t d| ¡›d|›d¡|Sq$t d|›d| ¡›d |d d›¡|d S)NcSs |dS)NZis_dev_filer-)Údllr-r-r.Ú6óz$__get_dll_manifest..r2zFound "z" dll version "rz Version "z" for "z" not found, using éÿÿÿÿ)r/ÚtupleÚfilterÚreversedrÚdebugÚupper)r1r2Údllsr3r-r-r.Ú__get_dll_manifest4s ÿr=Zdlss_versionZxess_versionZfsr3_versionZfsr4_version)r2rcCstd|ƒtd|ƒtd|ƒdœS)NÚdlssZdlss_dZdlss_g)z'drive_c/windows/system32/nvngx_dlss.dllz(drive_c/windows/system32/nvngx_dlssd.dllz(drive_c/windows/system32/nvngx_dlssg.dll©r=©r2r-r-r.Ú__get_dlss_dllsGsÿÿûrAcCs&td|ƒtd|ƒtd|ƒtd|ƒdœS)NÚxessZ xess_dx11ZxellZxess_fg)z$drive_c/windows/system32/libxess.dllz)drive_c/windows/system32/libxess_dx11.dllz$drive_c/windows/system32/libxell.dllz'drive_c/windows/system32/libxess_fg.dllr?r@r-r-r.Ú__get_xess_dllsSsÿÿúrCcCstd|ƒtd|ƒdœS)NZ fsr_31_vkZfsr_31_dx12)z.drive_c/windows/system32/amd_fidelityfx_vk.dllz0drive_c/windows/system32/amd_fidelityfx_dx12.dllr?r@r-r-r.Ú__get_fsr3_dlls`sÿÿürDcCsNdddddœdddddœdddddœdœ}|d ks>|| ¡vrBd }d||iS)Nz4.0.0_67A4D2BC10ad000zNhttps://download.amd.com/dir/bin/amdxcffx64.dll/67A4D2BC10ad000/amdxcffx64.dll)r2Údownload_urlÚmd5_hashÚzip_md5_hashz4.0.1_67D435F7d97000zMhttps://download.amd.com/dir/bin/amdxcffx64.dll/67D435F7d97000/amdxcffx64.dllz4.0.2_68840348eb8000zMhttps://download.amd.com/dir/bin/amdxcffx64.dll/68840348eb8000/amdxcffx64.dll)ú4.0.0z4.0.1ú4.0.2r0rIz'drive_c/windows/system32/amdxcffx64.dll)Úkeys)r2Z__fsr4_dllsr-r-r.Ú__get_fsr4_dllsks(üüüóÿrK)Ú prefix_dirÚdstÚfiler2Úignore_versionrc Cs„tj ||¡}tj |¡r6t d|›d¡t |¡tj |¡rnt |¡j dkrnt d|›d¡t |¡tj |¡st d|›d¡dSt|dƒ&}t | ¡¡ ¡ ¡}Wdƒn1sÆ0Y|d}|dur|| ¡krt d |›d¡dS|s€|d |d kr4t d|›d¡dS| dd¡} | durn|| ¡krnt d|›d¡dSt d |›d¡dS)NzRemoving stale symlink "rézRemoving stale file "zMissing file from prefix "FÚrbrFz2MD5 checksum mismatch between version and prefix "r2z3Version mismatch between configuration and prefix "z3MD5 checksum mismatch between manifest and prefix "zFound matching file in prefix "T)ÚosrÚjoinÚislinkrr:Úunlinkr)ÚstatÚst_sizeÚwarnr&ÚhashlibÚmd5r"Ú hexdigestÚlowerÚget) rLrMrNr2rOÚtargetÚdst_fdÚdst_md5Zversion_md5Úfile_md5r-r-r.Ú__check_upscaler_fileˆs4 4rb)rLÚfilesÚversion_filerOrc sðtj |¡s"t d|›d¡dSz^t|dd}| ¡‰Wdƒn1sN0Yt ˆ¡‰ˆ ¡D]}ˆ| d¡}qjWnFtyÆ}z.t d|›d¡t t|ƒ¡WYd}~dSd}~00t ‡‡‡‡fdd „ˆ ¡Dƒƒ}t|ƒS) NzMissing version file "rFrrrFz"Error while reading version file "c3s&|]}tˆ|ˆ|ˆ|ˆƒVqdS)N)rb)Ú.0rM©rcrOrLr2r-r.Ú Æsÿz)__check_upscaler_files..)rRrÚisfilerrXr&r"r r!rJr]r#r%r7Úall) rLrcrdrOÚ version_fdrMÚ_r,Zvalid_filesr-rfr.Ú__check_upscaler_files³s"& þrlF©rO)ÚnameÚ compat_dirrLr2rOrc CsŽttfttfttfttfdœ}||\}}z||ƒ}Wn>tyt} z&t d¡t t| ƒ¡WYd} ~ dSd} ~ 00t||t j ||¡|ƒS)a Check for upscaler files and their versions name: the name of the upscaler, possible values dlss, xess, fsr3, fsr4 version: the version of the upscaler dll to download ignore_version: ignore version mismatch but still check if the dlls are present ©r>rBÚfsr3Úfsr4z)Failed to get file versions from manifestNF)rAÚ__dlss_version_filerCÚ__xess_version_filerDÚ__fsr3_version_filerKÚ__fsr4_version_filer#rr$r%rlrRrrS) rnrorLr2rOrÚ get_filesrdrcr,r-r-r.Úcheck_upscalerÎs$ü ürx)rLrcÚdlfuncrdrcCsdtjj d¡}i}| ¡D]}t dtj |¡›d¡t ||ƒ}t ||dƒ}z:| ¡sd| ¡rn| |¡|||||ƒ|j ddWnrtyþ} zXt d|j›d¡t t| ƒ¡|j dd| ¡sÜ| ¡ræ| |¡WYd} ~ d Sd} ~ 00||d ||ddœ||<qt|d dd } | t |¡¡Wdƒn1sV0YdS)a‡Download and install the required dlls. This function takes care of backing up, downloading, and installing the required dlls If the download fails, it will uses the backups to revert to the previous files, otherwise the backups are removed. The downloading, caching and installation of the dlls is facilitated in the callable passed through the `dlfunc` argument. rzDownloading upscaler file "rz.oldT©Ú missing_okzError while downloading file "NFr2rF)r2rFrrr)r rrrrJrÚinforRÚbasenamerr)Ú is_symlinkÚrenamerUr#r$rnr%r&r'r r()rLrcryrdrr2rMrNÚtempr,rjr-r-r.Ú__download_upscaler_filesñs0 þ0r©Úchecksum)ÚurlrMrƒrc Cs,|jjdddtjj|ddid}zÌ| d¡N}tjj|dd}| | ¡¡Wd ƒn1sf0YWd ƒn1s„0Yt | d ¡ ¡¡ ¡ ¡}| ¡r¼| ¡jnd}|d urÔ|| ¡ksÜ|dkrîtd t|ƒ›ƒ‚Wn6ty&}z|jdd|‚WYd }~n d }~00d S)zµDownloads a file and checks against a checksum. If the download fails or the checksums do not match, the file is removed and the exception is propagated to the caller. Tr z User-AgentzIMozilla/5.0 (Windows NT 10.0; WOW64; rv:40.0) Gecko/20100101 Firefox/40.0)ÚheadersÚwbrrNrQrrPzMalformed download rz)ÚparentrrrÚRequestr&rr'r"rYrZr[r\r)rVrWÚRuntimeErrorr*r#rU) r„rMrƒrr_r+r`Zdst_sizer,r-r-r.Ú__download_files"ÿþJrŠ)rNÚcacherMrcCsêttt|dƒjƒƒ}| |j¡}| dd¡}| ¡r†t | d¡ ¡¡ ¡ ¡}|dur†|| ¡kr†t d|j›d¡|jdd| ¡s t|d||d|jjddd t |¡}| |j¡Wdƒn1sÜ0YdS) NrErGrQz3MD5 checksum mismatch between manifest and cached "rTrzr‚r )rrrrrrnr]r)rYrZr&r"r[r\rr$rUrŠr‡rÚzipfileÚZipFileÚ extractall)rNr‹rMÚurl_pathÚcached_fileraZ cached_md5Zzip_fdr-r-r.Ú__download_extract_zip6sÿr‘cCsšttt|dƒjƒƒ}| |jd|d›|j¡}| dd¡}| ¡r`| ¡j dkr`| ¡| ¡szt|d||d|j jdddt ||¡dS) NrEZ_vr2rGrPr‚Tr )rrrrrÚstemÚsuffixr]r)rVrWrUrŠr‡rÚshutilÚcopy)rNr‹rMrrrar-r-r.Ú__download_fsr4Hsÿr–)rnrorLr2rc CsÖt||||ddrdSt d| ¡›d¡tttftttft tt fttt fdœ}||\}}}z(||ƒ}t|||tj ||¡ƒs†t‚WnHtyÐ} z0t d| ¡›d¡t t| ƒ¡WYd} ~ n d} ~ 00dS) zµCheck for upscaler files and their versions name: the name of the upscaler, possible values dlss, xess, fsr3, fsr4 version: the version of the upscaler dll to download FrmNzFailed to validate "z" files.rpzFailed to download z dlls.)rxrr|r;rAr‘rsrCrtrDrurKr–rvrrRrrSr‰r#r$r%) rnrorLr2rrwZ download_funcrdrcr,r-r-r.Údownload_upscalerWs*üür—)ÚenvÚkeyrnrorLr2rcCsZ| |d¡dvr||n|}t||||ƒt||||dd}|rVt d| ¡›d¡|S)NÚ0>Ú1ršTrmz Automatic z upgrade enabled.)r]r—rxrr|r;)r˜r™rnrorLr2Úenabledr-r-r.Ú__setup_upscalerxsr)Ú compat_configr˜rorLrcCs(tƒ}d|vr(t|dd||ƒr(| d¡d|vrJt|dd||ƒrJ| d¡d|vrlt|dd||ƒrl| d¡d|vr’t|dd ||d ƒr´| d ¡n"d |vr´t|dd ||ƒr´| d ¡d |vräd|d <d|vrÔd|d<d|vräd|d<d|vrü| dtdƒ¡d|vrd|vr|r$d |¡|d<dS)z“Setup configured upscalers usage: setup_upscalers(g_session.compat_config, g_session.env, g_compatdata.base_dir, g_compatdata.prefix_dir) r>ÚPROTON_DLSS_UPGRADErBÚPROTON_XESS_UPGRADErqÚPROTON_FSR3_UPGRADEÚ fsr4rdna3ÚPROTON_FSR4_RDNA3_UPGRADErrrHÚPROTON_FSR4_UPGRADEr›ZFSR4_UPGRADEZmlfgZMLFG_UPGRADEZwmma_rdna3_workaroundZDXIL_SPIRV_CONFIGZDXVK_NVAPI_DRS_SETTINGSz²ngx_dlss_sr_override=on,ngx_dlss_rr_override=on,ngx_dlss_fg_override=on,ngx_dlss_sr_override_render_preset_selection=default,ngx_dlss_rr_override_render_preset_selection=default,ú,ZWINE_LOADDLL_REPLACEN)ÚsetrÚaddÚ setdefaultr*rS)ržr˜rorLZloaddll_replacer-r-r.Úsetup_upscalersˆsF ÿ ÿþ r©)r0)r0)r0)r0)r0)r0)r0)r0).Ú__doc__rYr rRr”Úurllib.requestrrŒÚpathlibrÚtypingrrÚurllib.parserrÚloggerrr rr ÚdictÚ__annotations__r/r*r=rsrtrurvrArCrDrKÚboolrbrlrxrrŠr‘r–r—rr¦r©r-r-r-r.Ús† þ,þüúù$û*"ÿþ'úùþ