jlddlmZddlmZddlmZmZmZm Z m Z m Z m Z m Z GddejZdS)) annotations)spatial_upsamplers)DEPTH_RESOLUTION_DIVISORSMOTION_VECTOR_METHODSNR_MODESUSE_DEPTH_GUIDEconfigure_depth_estimator neural_renderrelease_flow_modelunavailable_reasonceZdZdZgdZddgZdZddZedZ ed Z d Z dd Z dddd d Z edZedZdddZedZdZddddddZdZdS)DLSS5SpatialUpsamplerdlss5))zFull Resfull)zHalf Res (default)half)z Quarter Resquarter)zOriginal (default, faster)original)zRAFT (better quality)raftTNc:||_|t|dSdSN) server_configr )selfr files_locators NC:\Users\johnn\OneDrive\Documents\AI\postprocessing\dlss5\spatial_upsampler.py__init__zDLSS5SpatialUpsampler.__init__ s,*  $ %m 4 4 4 4 4 % $c dddS)Nrrdepth_resolution motion_vector)clss rdefault_configz$DLSS5SpatialUpsampler.default_configs$*ZHHHrcji|t|pi}t|d}t|d}|t vr|nd|t vr|nddS)Nr r!rrr)r$dictstrstriplowerrr)r#configr r!s rnormalize_config_sectionz.DLSS5SpatialUpsampler.normalize_config_sections?C&&((?D2,>,>?v&89::@@BBHHJJF?344::<<BBDD 4DHa4a4a 0 0gm.;?T.T.T]]Zd   rc6tj|j|Sr)apiread_config_sectionrrs rr*zDLSS5SpatialUpsampler.config s&t'94@@@rcddlm}tr|d|jgng}|ddkr$||d|j|S)Nr)query_preprocessor_download_defdepthr!rflow)preprocessing.processorsr1r rr*append)r enabled_onlyr1 definitionss rquery_download_defsz)DLSS5SpatialUpsampler.query_download_defs#s{LLLLLLXgo66w@RSSTTmo ;;== )V 3 3   >>vtGYZZ [ [ [r)send_cmd status_textgenc Vddlm}||||pd||S)Nr)process_files_def_if_neededzDownloading DLSS 5 Model Files)r9r:r; process_files)shared.utils.downloadr=r8)rr>r9r:r;_kwargsr=s rdownloadzDLSS5SpatialUpsampler.download+sbEEEEEE**4+C+C+E+EPXfqgVvV\_o|}}} }rc@|jr tjn tjSr)reason_disabledr-PROCESSOR_STATUS_DISABLEDPROCESSOR_STATUS_ENABLEDr/s rstatuszDLSS5SpatialUpsampler.status0s040Dfs,,#Jffrc"tdS)NFtemporal)r r/s rrCz%DLSS5SpatialUpsampler.reason_disabled4s!51111rF) lock_configc Z|5|5||j|ddd| }||j|ddd| }dddn #1swxYwYdddn #1swxYwYd|fd|fgS)Nr z!DLSS 5 Depth Resolution PrecisionzRun depth extraction at the render resolution, or resize each frame to half/quarter width and height for lower latency and memory use.)choicesvaluelabelinfo interactiver!zDLSS 5 Motion VectorzeOriginal uses the faster OpenCV DIS estimator. RAFT generally produces higher-quality motion vectors.)GroupRowDropdownDEPTH_RESOLUTION_CHOICESMOTION_VECTOR_CHOICES)rgrr*rJr r!s rcreate_config_uiz&DLSS5SpatialUpsampler.create_config_ui8s XXZZ X X X X#%;;t7T\bcu\vbiqCNN;$O$O " D4NV\]lVmvLSzLWHW !X!X  X X X X X X X X X X X X X X X X X X X X X X X X X X X X X X$%56-8XYYs5BA B7 BB B B BB!BcVtd}d|rd|dndz}dtjfdtjdd |jd i||jfg|jt t id tjd |jd d ddtjtj tj fdddddddd gid S)NFrHzDLSS 5 Neural Renderingz ())videoimageri'zdlss5*1zDLSS 5 Neural Rendering. At x1 it refines at native resolution; higher modes refine and upscale. Requires Windows and a compatible NVIDIA RTX GPU. WSL can use the Windows worker through Windows interoperability; this is not native Linux Neural Rendering.spatial_upsampler_dlss_strength intensitynumberslider?gg@g?zDLSS 5 NR IntensityznControls the native Neural Rendering intensity. 1 is the native default; values up to 2 strengthen the effect.) namesettingtype componentuirequireddefaultminimummaximumsteprN description) rbupsampler_typesmediaprofile config_keypos method_posmethods multipliersdefault_spatial_upsamplingpostprocessing_categoryrlmethod_parameters) r r-UPSAMPLER_TYPE_POSTPROCESSINGUPSAMPLER_PROFILE_VIDEOMETHODtupler!POSTPROCESSING_CATEGORY_UPSAMPLERPARAMETER_UI_POSTPROCESSING PARAMETER_UI_LATE_POSTPROCESSINGPARAMETER_UI_MEDIA_FLOW)r#reasonrNs rquery_upsampler_defz)DLSS5SpatialUpsampler.query_upsampler_def?s#U333)v-M^&^^^^2N- # AC'2!:v. +,Jh8*3'*'L\"%*:{\ds{EHEdfifJLOLgDhv{HKX[hkuyDYjZ[[/"   rc||}||dtvrd|Std}|rd|dndS)Nrz&Unknown DLSS 5 Neural Rendering mode: FrHz(DLSS 5 Neural Rendering is unavailable: z. See docs/DLSS5.md.rZ) split_valuerr )rspatial_upsampling image_modesplitrs rvalidate_upsamplingz)DLSS5SpatialUpsampler.validate_upsamplingUse  !344 =E!HH44Prs""""""444444ssssssssssssssssssss^^^^^C6^^^^^r