Fix for Issue #8

arguments.py

@@ -0,0 +1,15 @@
+import argparse
+import os
+
+
+def ensure_folder_exists(folder):
+    os.makedirs(folder, exist_ok=True)
+    return folder
+
+
+parser = argparse.ArgumentParser('SWE1R scripts ')
+parser.add_argument('--out', help='directory where the output files are written', default='output', type=ensure_folder_exists)
+parser.add_argument('input', help='input file')
+
+multiparser = argparse.ArgumentParser('SWE1R scripts ')
+multiparser.add_argument('input', nargs='+', help='input files')

extract-racer-tab.py

@@ -2,9 +2,13 @@
 
 import sys
 
+from arguments import multiparser
+
 strings = {}
 
-for path in sys.argv[1:]:
+args = multiparser.parse_args()
+
+for path in args.input:
   with open(path, 'rb') as in_file:
     data = in_file.read()
 

out_modelblock.py

@@ -1,9 +1,13 @@
 #!/usr/bin/env python3
 
-import sys
 import struct
+
+import os
 from PIL import Image
 
+from arguments import parser
+
+
 def read8(f):
   return f.read(1)[0]
 def read16(f):
@@ -121,7 +125,9 @@ def shifter(a1):
   #  return result;
   pass
 
-with open(sys.argv[1], 'rb') as f:
+args = parser.parse_args()
+
+with open(args.input, 'rb') as f:
   count = read32(f)
   for i in range(0, count - 1):
     f.seek(4 + 8 * i)
@@ -136,13 +142,15 @@ def shifter(a1):
     f.seek(a)
     buf = f.read(length1)
 
-    with open("/tmp/swep1r/model-%d-a.bin" % i, 'wb') as t:
+    model_a = os.path.join(args.out, 'model-%d-a.bin' % i)
+    with open(model_a, 'wb') as t:
       t.write(buf)
 
     f.seek(b)
     buf = f.read(length2)
 
-    with open("/tmp/swep1r/model-%d-b.bin" % i, 'wb') as t:
+    model_b = os.path.join(args.out, 'model-%d-b.bin' % i)
+    with open(model_b, 'wb') as t:
       t.write(buf)
 
     f.seek(a)

out_splineblock.py

@@ -1,9 +1,13 @@
 #!/usr/bin/env python3
 
-import sys
 import struct
+
+import os
 from PIL import Image
 
+from arguments import parser
+
+
 def read8(f):
   return f.read(1)[0]
 def read16(f):
@@ -29,7 +33,10 @@ def shifter(a1):
     result = 16
   return result
 
-with open(sys.argv[1], 'rb') as f:
+args = parser.parse_args()
+
+
+with open(args.input, 'rb') as f:
   count = read32(f)
   for i in range(0, count - 1):
     f.seek(4 + 4 * i)
@@ -42,10 +49,12 @@ def shifter(a1):
     f.seek(a)
     buf = f.read(length)
 
-    with open("/tmp/swep1r/spline-%d.bin" % i, 'wb') as t:
+    spline_bin = os.path.join(args.out, 'spline-%d.bin' % i)
+    with open(spline_bin, 'wb') as t:
       t.write(buf)
 
-    t = open("/tmp/swep1r/spline-%d.obj" % i, 'w')
+    spline_obj = os.path.join(args.out, 'spline-%d.obj' % i)
+    t = open(spline_obj, 'w')
 
     f.seek(a)
 

out_spriteblock.py

@@ -1,8 +1,11 @@
 #!/usr/bin/env python3
 
-import sys
+import os
 from PIL import Image
 
+from arguments import parser
+
+
 def read8(f):
   return f.read(1)[0]
 def read16(f):
@@ -26,7 +29,9 @@ def shifter(a1):
     result = 16
   return result
 
-with open(sys.argv[1], 'rb') as f:
+args = parser.parse_args()
+
+with open(args.input, 'rb') as f:
   count = read32(f)
   for i in range(0, count - 1):
     f.seek(4 + 4 * i)
@@ -174,7 +179,8 @@ def pixel_a8r8g8b8():
       reader = pixel_p8
       f.seek(section_a + unk4)
       buf = f.read(256 * 2)
-      with open("/tmp/swep1r/sprite-%d-palette.bin" % i, 'wb') as t:
+      sprite_palette = os.path.join(args.out, "sprite-%d-palette.bin" % i)
+      with open(sprite_palette, 'wb') as t:
         t.write(buf)
       f.seek(section_a + unk4)
       pixel_palette = []
@@ -246,4 +252,5 @@ def pixel_a8r8g8b8():
         if y == height:
           print("Complete")
           break
-    im.save("/tmp/swep1r/sprite-%d.png" % i, 'PNG')
+    sprite = os.path.join(args.out, 'sprite-%d.png' % i)
+    im.save(sprite, 'PNG')

out_textureblock.py

@@ -1,8 +1,11 @@
 #!/usr/bin/env python3
 
-import sys
+import os
 from PIL import Image
 
+from arguments import parser
+
+
 def read8(f):
   return f.read(1)[0]
 def read16(f):
@@ -26,7 +29,11 @@ def shifter(a1):
     result = 16
   return result
 
-with open(sys.argv[1], 'rb') as f:
+
+args = parser.parse_args()
+
+
+with open(args.input, 'rb') as f:
   count = read32(f)
   for i in range(0, count - 1):
     f.seek(4 + 8 * i)
@@ -43,7 +50,8 @@ def shifter(a1):
     f.seek(a)
     buf = f.read(length)
 
-    with open("/tmp/swep1r/texture-%d.bin" % i, 'wb') as t:
+    texture = os.path.join(args.out, "texture-%d.bin" % i)
+    with open(texture, 'wb') as t:
       f.read
       t.write(buf)
 
@@ -100,7 +108,8 @@ def shifter(a1):
           #g = 0 #((pixel >> 5) & 0x1F) * 0xFF // 0x1F
           #b = 0 #((pixel >> 10) & 0x1F) * 0xFF // 0x1F
           pixels[x, y] = (a, r, g, b)
-      im.save("/tmp/swep1r/texture-%d.png" % i, 'PNG')
+      texture = os.path.join(args.out, 'texture-%d.png' % i)
+      im.save(texture, 'PNG')
 
 
     if False:

parse-racer-tab.py

@@ -2,9 +2,13 @@
 
 import sys
 
+from arguments import multiparser
+
 strings = {}
 
-for path in sys.argv[1:]:
+args = multiparser.parse_args()
+
+for path in args.input:
   with open(path, 'rb') as in_file:
     data = in_file.read()
 

requirements.txt

@@ -0,0 +1,2 @@
+argparse
+Pillow

scr2wav.py

@@ -4,10 +4,13 @@
 # Basically it's just XOR'd with 0x55, 0x55, 0xEE, 0xEE and the name Caesar shifted.
 # This script undoes this mess, but the files will still be suffix'd with ".rbq"
 
-import sys
 import os
 
-for path in sys.argv[1:]:
+from arguments import multiparser
+
+args = multiparser.parse_args()
+
+for path in args.input:
   filename = os.path.basename(path)
   real_filename = ''
   for s in filename: