proton: add ntsync, fsr4, wayland, amdags options from emtaash's build

Author: GloriousEggroll

proton

@@ -17,6 +17,7 @@ import shlex
 import uuid
 import json
 import time
+import urllib.request
 
 from ctypes import CDLL
 from ctypes import CFUNCTYPE
@@ -732,6 +733,9 @@ class CompatData:
             #How can this happen??
             log("Prefix has an invalid version?! You may want to back up user files and delete this prefix.")
             #If it does, just let the Wine upgrade happen and hope it works...
+            with open(self.creation_sync_guard, "w"):
+                pass
+            os.sync()
             return
 
         try:
@@ -1009,6 +1013,19 @@ class CompatData:
                 os.remove(old)
                 os.symlink(src=link, dst=old)
 
+    def setup_fsr4(self):
+        fsr4files = ["amdxcffx64"]
+
+        for f in fsr4files:
+            dst = "drive_c/windows/system32/" + f + ".dll"
+            if not file_exists(self.prefix_dir + dst, follow_symlinks=False):
+                try:
+                    tracked_files.write(dst + '\n')
+                except Exception:
+                    pass
+            if file_exists(g_proton.lib_dir + "wine/amdprop/" + f + ".dll", follow_symlinks=False):
+                self.create_symlink(self.prefix_dir + dst, g_proton.lib_dir + "wine/amdprop/" + f + ".dll")
+
     def setup_prefix(self):
         with self.prefix_lock:
             if file_exists(self.version_file, follow_symlinks=True):
@@ -1235,6 +1252,8 @@ class CompatData:
                         tracked_files.write(dst + '\n')
                     self.create_symlink(self.prefix_dir + dst, g_proton.lib_dir + "wine/icu/i386-windows/" + f + ".dll")
 
+                self.setup_fsr4()
+
                 # If the user requested the NVAPI be available, copy it into place.
                 # If they didn't, clean up any stray nvapi DLLs.
                 if use_nvapi:
@@ -1611,6 +1630,13 @@ class Session:
             self.compat_config.discard(config_name)
         return True
 
+    def download_file(self, uri, dst):
+        try:
+            os.makedirs(os.path.dirname(dst), exist_ok=True)
+            urllib.request.urlretrieve(uri, dst)
+        except Exception:
+            pass
+
     def try_log_slr_versions(self):
         try:
             if "PRESSURE_VESSEL_RUNTIME_BASE" in self.env:
@@ -1687,10 +1713,6 @@ class Session:
         if not self.check_environment("PROTON_USE_WINED3D", "wined3d"):
             self.check_environment("PROTON_USE_WINED3D11", "wined3d")
         self.check_environment("PROTON_NO_WM_DECORATION", "nowmdecoration")
-        self.check_environment("PROTON_ENABLE_WAYLAND", "enablewayland")
-        self.check_environment("PROTON_ENABLE_HDR", "enablehdr")
-        self.check_environment("PROTON_PREFER_SDL", "sdlinput")
-        self.check_environment("PROTON_NO_STEAMINPUT", "sdlinput")
         self.check_environment("PROTON_DXVK_D3D8", "dxvkd3d8")
         self.check_environment("PROTON_NO_D3D11", "nod3d11")
         self.check_environment("PROTON_NO_D3D10", "nod3d10")
@@ -1709,6 +1731,19 @@ class Session:
         self.check_environment("PROTON_DISABLE_NVAPI", "disablenvapi")
         self.check_environment("PROTON_FORCE_NVAPI", "forcenvapi")
         self.check_environment("PROTON_HIDE_APU", "hideapu")
+        self.check_environment("PROTON_ENABLE_WAYLAND", "wayland")
+        self.check_environment("PROTON_USE_WAYLAND", "wayland")
+        self.check_environment("PROTON_PREFER_SDL", "sdlinput")
+        self.check_environment("PROTON_USE_SDL", "sdlinput")
+        self.check_environment("PROTON_FSR4_UPGRADE", "fsr4")
+        self.check_environment("PROTON_NATIVE_AGS", "nativeags")
+        self.check_environment("PROTON_ENABLE_HDR", "hdr")
+        self.check_environment("PROTON_USE_NTSYNC", "ntsync")
+
+        if "ntsync" in self.compat_config:
+            self.env["WINENTSYNC"] = "1"
+        else:
+            self.env.pop("WINENTSYNC", "")
 
         if "noesync" in self.compat_config:
             self.env.pop("WINEESYNC", "")
@@ -1718,24 +1753,6 @@ class Session:
         if "nowmdecoration" in self.compat_config:
             self.env["WINE_NO_WM_DECORATION"] = "1"
 
-        if "enablewayland" in self.compat_config:
-            if self.env.get("WAYLAND_DISPLAY", False):
-                self.dlloverrides["winex11.drv"] = "d"
-                self.dlloverrides["winewayland.drv"] = "b"
-                self.env["WINE_WAYLAND_HACKS"] = "1"
-                if "enablehdr" in self.compat_config:
-                    self.env["DXVK_HDR"] = "1"
-                """
-                    It is not possible for winewayland to support xalia since
-                    there is no way to position toplevel surfaces dynamically in wayland
-                """
-                self.env["PROTON_USE_XALIA"] = "0"
-                # Steam input does not work on wayland
-                self.env["PROTON_NO_STEAMINPUT"] = "1"
-                # Make sure SDL will still work with the controllers, steam sets this for some dumb reason
-                if "SDL_GAMECONTROLLER_IGNORE_DEVICES" in self.env:
-                    del self.env["SDL_GAMECONTROLLER_IGNORE_DEVICES"]
-
         if "noxim" not in self.compat_config:
             self.env.pop("WINE_ALLOW_XIM")
 
@@ -1867,6 +1884,11 @@ class Session:
         if "disablelibglesv2" in self.compat_config:
             self.dlloverrides["libglesv2"] = "d"
 
+        if "nativeags" in self.compat_config:
+            self.dlloverrides["atiadlxx"] = "b"
+            self.dlloverrides["atidxx64"] = "b"
+            self.dlloverrides["amd_ags_x64"] = "n"
+
         if "nomfdxgiman" in self.compat_config:
             self.env["WINE_DO_NOT_CREATE_DXGI_DEVICE_MANAGER"] = "1"
 
@@ -1893,6 +1915,44 @@ class Session:
             if "SDL_GAMECONTROLLER_IGNORE_DEVICES" in self.env:
                 del self.env["SDL_GAMECONTROLLER_IGNORE_DEVICES"]
 
+        if "fsr4" in self.compat_config:
+            self.env["FSR4_UPGRADE"] = "1"
+            versions = ["67D435F7d97000", "67A4D2BC10ad000"]
+            version_match = True
+
+            # TODO: make the following code more efficent
+
+            # 0. Check downloaded version
+            if file_exists(g_proton.lib_dir + "wine/amdprop/amdxcffx64_version", follow_symlinks=False):
+                with open(g_proton.lib_dir + "wine/amdprop/amdxcffx64_version", "r") as file:
+                    first_line = file.readline()
+                    version_match = False
+                    for version in versions:
+                        if first_line in version:
+                            version_match = True
+            else:
+                version_match = False
+
+            # 1. Download amdxcffx64 library
+            if not file_exists(g_proton.lib_dir + "wine/amdprop/amdxcffx64.dll", follow_symlinks=False) or not version_match:
+                self.download_file("https://download.amd.com/dir/bin/amdxcffx64.dll/" + versions[1] + "/amdxcffx64.dll", g_proton.lib_dir + "wine/amdprop/amdxcffx64.dll")
+
+            # 2. Update stored version
+            if file_exists(g_proton.lib_dir + "wine/amdprop/amdxcffx64.dll", follow_symlinks=False):
+                g_compatdata.setup_fsr4()
+                if not version_match:
+                    with open(g_proton.lib_dir + "wine/amdprop/amdxcffx64_version", "w") as file:
+                        file.write(versions[1] + "\n")
+
+        if "hdr" in self.compat_config:
+            self.env["DXVK_HDR"] = "1"
+
+        # igdext is loaded either using the LoadLibraryExW hack on 1.3 or lower
+        # or using the INTC_ALT_DRIVER_EXTENSIONS_PATH on 2.0 or newer
+        # TODO: Load it through driver store data like it is on windows
+        # TODO: D3D11 has a different loading mechanism because why not
+        self.env["INTC_ALT_DRIVER_EXTENSIONS_PATH"] = "C:\\Windows\\System32"
+
         # NVIDIA software may check for the "DriverStore" by querying the
         # NGXCore\NGXPath registry key via `D3DDDI_QUERYREGISTRY_SERVICEKEY` for
         # a given adapter. In the case where this path cannot be found, the
@@ -1937,6 +1997,12 @@ class Session:
                     if var in self.env:
                         self.log_file.write("Effective " + var + ": " + self.env[var] + "\n")
 
+                if "fsr4" in self.compat_config:
+                    if file_exists(g_proton.lib_dir + "wine/amdprop/amdxcffx64.dll", follow_symlinks=False):
+                        self.log_file.write("INFO: Automatic FSR4 upgrade enabled!\n")
+                    else:
+                        self.log_file.write("ERROR: Failed to download amdxcffx64.dll!\n")
+
                 # check for low fd limit
                 _soft_limit, hard_limit = resource.getrlimit(resource.RLIMIT_NOFILE)
                 if hard_limit < 524288:
@@ -1977,6 +2043,22 @@ class Session:
         if "disablenvapi" not in self.compat_config or "forcenvapi" in self.compat_config:
             self.env["DXVK_ENABLE_NVAPI"] = "1"
 
+        if "wayland" in self.compat_config:
+            if "WAYLAND_DISPLAY" in self.env:
+                self.dlloverrides["winex11.drv"] = "d"
+                self.dlloverrides["winewayland.drv"] = "b"
+                self.env["WINE_WAYLAND_HACKS"] = "1"
+                """
+                    It is not possible for winewayland to support xalia since
+                    there is no way to position toplevel surfaces dynamically in wayland
+                """
+                self.env["PROTON_USE_XALIA"] = "0"
+                # Steam input does not work on wayland
+                self.env["PROTON_NO_STEAMINPUT"] = "1"
+                # Make sure SDL will still work with the controllers, steam sets this for some dumb reason
+                if "SDL_GAMECONTROLLER_IGNORE_DEVICES" in self.env:
+                    del self.env["SDL_GAMECONTROLLER_IGNORE_DEVICES"]
+
         if "forcenvapi" in self.compat_config:
             self.env["DXVK_NVAPI_ALLOW_OTHER_DRIVERS"] = "1"
             self.env["DXVK_NVAPI_DRIVER_VERSION"] = "99999"