j}|ddlmZddlmZddlmZddlmZm Z m Z m Z m Z m Z ddlmZGddejZd S) ) annotations)spatial_upsamplers)temporal_upsamplers) DLSSG_LINUXdlssg_capabilitiesframe_generateis_rtx_50_seriesrelease_flow_modelunavailable_reason)DLSS5SpatialUpsamplerceZdZdZdZddZdZddZddddd Ze d Z e d Z d Z d ddZ ddddZdZdS)DLSSGTemporalUpsamplerdlssg)@g@g@g@g@Nc>||_|ddlm}||dSdS)Nr)configure_depth_estimator) server_configruntimer)selfr files_locatorrs OC:\Users\johnn\OneDrive\Documents\AI\postprocessing\dlss5\temporal_upsampler.py__init__zDLSSGTemporalUpsampler.__init__ sA*  $ : : : : : : % %m 4 4 4 4 4 % $cZtjtj|jdS)Ndlss5)r normalize_config_section spatial_apiread_config_section_by_keyrrs rconfigzDLSSGTemporalUpsampler.configs8$=k>deiewzA?B?BCC CrTcrddlm}|ddkr|d|jgngS)Nr)query_preprocessor_download_def motion_vectorraftflow)preprocessing.processorsr#r!r)r enabled_onlyr#s rquery_download_defsz*DLSSGTemporalUpsampler.query_download_defssXLLLLLLPTP[P[P]P]^mPnrxPxPx//8JKKLLA Ar)send_cmd status_textgenc Vddlm}||||pd||S)Nr)process_files_def_if_neededz-Downloading DLSS Frame Generation Model Files)r*r+r, process_files)shared.utils.downloadr.r))rr/r*r+r,_kwargsr.s rdownloadzDLSSGTemporalUpsampler.downloadsbEEEEEE**4+C+C+E+EPXfqgevekn~KLLL Lrc@|jr tjn tjSN)reason_disabledapiPROCESSOR_STATUS_DISABLEDPROCESSOR_STATUS_ENABLEDr s rstatuszDLSSGTemporalUpsampler.status s040Dfs,,#Jffrc"tdS)NTtemporal)r r s rr5z&DLSSGTemporalUpsampler.reason_disabled$s!40000rc :td}|stni}t|dd}t|ddtrdnt rdnd t fd |jD}|r)|dkr!t fd |jDnd }|s |dkrd }n|sdzkrddz}d|rd|dndz}ddd|jdi||jfg|j|idddS)NTr;worker_versionrmulti_frame_count_maxrc3DK|]}|tdk|VdS)rBN)min).0scalehardware_limits r zFDLSSGTemporalUpsampler.query_temporal_upsampler_def...s9cce5CPQSaLbLbCbCbECbCbCbCbccrc3:K|]}|k|dz k|VdS)rN)rErFrGmaximums rrHzFDLSSGTemporalUpsampler.query_temporal_upsampler_def..0s@xx%uP^G^G^chklclpwcwcwcwcwcwcwxxr)rzworker v2 required for x3+zGPU/runtime supports up to xzDLSS Frame Generationz ()ri'zdlssg*2a NVIDIA DLSS Frame Generation using video motion estimation. Linux supports experimental x2 with the NVIDIA Vulkan driver and a compatible RTX GPU. Windows supports x2-x4 and capability-gated x5/x6 on RTX 50 series, and requires Windows 11, HAGS, and RTX 40 or newer.)name config_keypos method_posmethods multipliersdefault_temporal_upsampling description) r rintgetrr tuple MULTIPLIERSMETHOD)rreason capabilitiesr>rSlabelrGrKs @@rquery_temporal_upsampler_defz3DLSSGTemporalUpsampler.query_temporal_upsampler_def(s#T22239A)+++r \--.>BBCCl&&'>BBCC)O5E5G5G0NQcccct/?ccccc  \}KOP}P}P%xxxxx43CxxxxxxV\K B.1,,1FF BGaK.88AGaKAAF'V+K>>>>>L+!;/ ,- K5+4h   rF)source_is_imagec||}||d|jvrd|S|rdStr|ddkrdStd}|rd|d St }t |d d }t |d d}|dd krtsdS|ddkr|dkrdS|ddz |kr d|dzdSdS)Nrz$Unknown DLSS Frame Generation mode: z1Temporal Upsampling can not be used with an Imager@z7Linux DLSS Frame Generation currently supports x2 only.Tr;z&DLSS Frame Generation is unavailable: z. See docs/DLSS5.md.r>rr?rBzADLSS Frame Generation x5/x6 requires a GeForce RTX 50-series GPU.zBDLSS Frame Generation x3-x6 requires worker v2. See docs/DLSS5.md.z2DLSS Frame Generation on this GPU supports up to x.rM) split_valuerYrr rrVrWr )rtemporal_upsamplingr_splitr[r\r>rKs rvalidate_upsamplingz*DLSSGTemporalUpsampler.validate_upsamplingAs?  !455 =E!HD,<<<O:MOO O  GFF  M58q==LL#T222  YXFXXX X)++ \--.>BBCCl&&'>BBCC 8a<< 0 2 2<VV 8a<rzs""""""<<<<<<555555~~~~~~~~~~~~~~~~444444VVVVVS7VVVVVr