Add upgraded GPU checks

Resources/Constants.py

@@ -30,6 +30,8 @@ class Constants:
         self.current_path = Path(__file__).parent.parent.resolve()
         self.payload_path = self.current_path / Path("payloads")
 
+        self.custom_model: str = None
+        self.custom_mxm_gpu: str = None
     # Payload Location
     # OpenCore
     @property

Resources/ModelArray.py

@@ -710,4 +710,52 @@ IHEHC2 = [
 
 IH = [
     "MacPro6,1"
+]
+
+upgradableMXMGPUs = [
+    "iMac10,1"
+    "iMac11,1",
+    "iMac11,2",
+    "iMac11,3",
+    "iMac12,1",
+    "iMac12,2",
+    "Xserve3,1",
+]
+
+# Reference: https://forums.macrumors.com/threads/2011-imac-graphics-card-upgrade.1596614/
+NVIDIAMXMGPUs = [
+    "0x12b9",#Quadro K610M
+    "0x0ff6",#Quadro K1100M
+    "0x11fc",#Quadro K2100M 
+    "0x0ffc",#Quadro K1000M 
+    "0x0ffb",#Quadro K2000M 
+    "0x11b6",#Quadro K3100M 
+    "0x11b7",#Quadro K4100M 
+    "0x11bc",#Quadro K5000M 
+    "0x11b8",#Quadro K5100M 
+    "0x11e1",#GTX 765M
+    "0x11e2",#GTX 765M 
+    "0x11e0",#GTX 770M
+    "0x119e",#GTX 780M Mac Edition
+    "0x119e",#GTX 780M
+    "0x119f",#GTX 880M
+    "0x119a",#GTX 860M
+    "0x1392",#GTX 860M
+    "0x1199",#GTX 870M
+    "0x11a9",#GTX 870M
+    "0x731f",
+]
+
+AMDMXMGPUs = [
+    "0x67EF",#AMD RX 460
+    "0x67e8",#AMD WX 4130/WX 4150 
+    "0x67e0",#AMD WX 4170 
+    "0x67c0",#AMD WX 7100
+    "0x731f",
+]
+
+nativeWifi = [
+    "0x43ba",#BCM43602
+    "0x43a3",#BCM4350
+    "0x43a0",#BCM4360
 ]

Resources/build.py

@@ -137,6 +137,28 @@ class BuildOpenCore:
         if self.model in ModelArray.HiDPIpicker:
             print("- Setting HiDPI picker")
             self.config["NVRAM"]["Add"]["4D1EDE05-38C7-4A6A-9CC6-4BCCA8B38C14"]["UIScale"] = binascii.unhexlify("02")
+        
+        # Check GPU Vendor
+        if self.constants.custom_model != "None":
+            #current_gpu: str = subprocess.run("system_profiler SPDisplaysDataType".split(), stdout=subprocess.PIPE, stderr=subprocess.STDOUT).stdout.decode()
+            #current_gpuv = [line.strip().split(": ", 1)[1] for line in current_gpu.split("\n") if line.strip().startswith(("Vendor"))][0]
+            #current_gpud = [line.strip().split(": ", 1)[1] for line in current_gpu.split("\n") if line.strip().startswith(("Device ID"))][0]
+            current_gpuv = "NVIDIA (0x10de)"
+            current_gpud = "0x11a9"
+            print(f"- Detected GPU: {current_gpuv} {current_gpud}")
+            if (current_gpuv == "AMD (0x1002)") & (current_gpud in ModelArray.AMDMXMGPUs):
+                self.constants.custom_mxm_gpu = True
+                print("- Adding AMD DRM patches")
+                self.config["NVRAM"]["Add"]["7C436110-AB2A-4BBB-A880-FE41995C9F82"]["boot-args"] += " shikigva=80 unfairgva=1"
+            elif (current_gpuv == "NVIDIA (0x10de)") & (current_gpud in ModelArray.NVIDIAMXMGPUs):
+                self.constants.custom_mxm_gpu = True
+                print("- Adding Brightness Control patches")
+                if self.model in ["iMac11,1", "iMac11,2", "iMac11,3"]:
+                    backlight_path = "PciRoot(0x0)/Pci(0x3,0x0)/Pci(0x0,0x0)"
+                    self.config["DeviceProperties"]["Add"][backlight_path] = {"@0,backlight-control": binascii.unhexlify("01000000"), "@0,built-in": binascii.unhexlify("01000000")}
+                elif self.model in ["iMac12,1", "iMac12,2"]:
+                    backlight_path = "PciRoot(0x0)/Pci(0x1,0x0)/Pci(0x0,0x0)"
+                    self.config["DeviceProperties"]["Add"][backlight_path] = {"@0,backlight-control": binascii.unhexlify("01000000"), "@0,built-in": binascii.unhexlify("01000000")}
 
         # Add OpenCanopy
         print("- Adding OpenCanopy GUI")
@@ -168,9 +190,14 @@ class BuildOpenCore:
             spoofed_model = "Macmini7,1"
             spoofed_board = "Mac-35C5E08120C7EEAF"
         elif self.model in ModelArray.iMac151:
-            print("- Spoofing to iMac15,1")
-            spoofed_model = "iMac15,1"
-            spoofed_board = "Mac-42FD25EABCABB274"
+            if self.constants.custom_mxm_gpu == True:
+                print("- Spoofing to iMacPro1,1")
+                spoofed_model = "iMacPro1,1"
+                spoofed_board = "Mac-7BA5B2D9E42DDD94"
+            else:
+                print("- Spoofing to iMac15,1")
+                spoofed_model = "iMac15,1"
+                spoofed_board = "Mac-42FD25EABCABB274"
         elif self.model in ModelArray.iMac144:
             print("- Spoofing to iMac14,4")
             spoofed_model = "iMac14,4"
@@ -198,6 +225,7 @@ class BuildOpenCore:
                 self.config["PlatformInfo"]["Automatic"] = True
                 self.config["PlatformInfo"]["UpdateDataHub"] = True
                 self.config["PlatformInfo"]["UpdateNVRAM"] = True
+                self.config["UEFI"]["ProtocolOverrides"]["DataHub"] = True
                 self.config["PlatformInfo"]["Generic"]["SystemProductName"] = spoofed_model
                 self.config["PlatformInfo"]["Generic"]["SystemSerialNumber"] = macserial_output[0]
                 self.config["PlatformInfo"]["Generic"]["MLB"] = macserial_output[1]