From 675215417f1858dca6b1b4c1de58372804f27461 Mon Sep 17 00:00:00 2001
From: Jack Walker <7463599+Jack-Walker@users.noreply.github.com>
Date: Tue, 31 Mar 2020 16:41:02 -0400
Subject: [PATCH] Updated script to check for multiple ROM extensions.

---
 fixbaserom.py | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/fixbaserom.py b/fixbaserom.py
index 88c6840249..3cc67be0a8 100644
--- a/fixbaserom.py
+++ b/fixbaserom.py
@@ -4,10 +4,18 @@ import sys
 import struct
 import hashlib
 
-# Read in the original ROM
+
+# Determine if we have a ROM file
+romFileName = ""
 if (path.exists("baserom_original.z64")):
-    print("File 'baserom_original.z64' found.")
-    with open("baserom_original.z64", mode='rb') as file:
+    romFileName = "baserom_original.z64"
+elif (path.exists("baserom_original.n64")):
+    romFileName = "baserom_original.n64"
+
+# Read in the original ROM
+if (romFileName != ""):
+    print("File '" + romFileName + "' found.")
+    with open(romFileName, mode='rb') as file:
         fileContent = bytearray(file.read())
 
         # Check if ROM needs to be byte swapped
@@ -28,9 +36,9 @@ if (path.exists("baserom_original.z64")):
         
             print("Byte swapping done.")
 else:
-    print("Error: Could not find baserom_original.z64.")
+    print("Error: Could not find baserom_original.z64/baserom_original.n64.")
     sys.exit(1)
-
+    
 # Strip the overdump
 print("Stripping overdump...")
 strippedContent = list(fileContent[0:0x3600000])