Added default option to mtlGen.py

Author: TCA166

.gitignore

@@ -20,4 +20,5 @@ mtlGen
 *.ow
 *.txt
 *Check*
-tests/benchmark
+tests/benchmark
+mtlFiles

mtlGen.py

@@ -2,6 +2,16 @@
 from PIL import Image
 import sys
 import os
+from io import TextIOWrapper
+import shutil
+
+def writeMaterialSection(o:TextIOWrapper, name:str, rgb:tuple, d:float) -> None:
+    o.write("newmtl %s\n" % name)
+    o.write("Ka %.3f %.3f %.3f\n" % (rgb[0], rgb[1], rgb[2]))
+    o.write("Kd %.3f %.3f %.3f\n" % (rgb[0], rgb[1], rgb[2]))
+    o.write("Ks 0.000 0.000 0.000\n")
+    o.write("d %.1f\n" % d)
+    o.write("illum 1\n")
 
 #a python script for generating mtl files
 
@@ -22,41 +32,45 @@
             a = True
     if a and not s:
         print("-a ignored")
+    if not s:
+        try:
+            os.mkdir("./mtlFiles")
+        except FileExistsError:
+            pass
     with open("out.mtl", "w") as o:
         for filename in os.listdir(path):
             f = os.path.join(path, filename)
             #simplified functioning that doesn't require textures
-            if os.path.isfile(f) and filename[-4:] == ".png" and s:
-                rgb = [0, 0, 0, 0]
-                img = Image.open(f).convert('RGBA')
-                pixels = 0
-                for x in range(img.size[0]):
-                    for y in range(img.size[1]):
-                        pixel = img.getpixel((x, y))
-                        if not (pixel[3] == 0 and a):
-                            rgb[0] += pixel[0]
-                            rgb[1] += pixel[1]
-                            rgb[2] += pixel[2]
-                            rgb[3] += pixel[3]
-                            pixels += 1
-                if pixels > 0:
-                    rgb[0] /= (pixels * 255)
-                    rgb[1] /= (pixels * 255)
-                    rgb[2] /= (pixels * 255)
-                    rgb[3] /= (pixels * 255)
-                if "grass" in filename or "leaves" in filename:
-                    # #8eb971
-                    rgb[0] = 142 / 255
-                    rgb[1] = 185 / 255
-                    rgb[2] = 113 / 255
-                if "top" in filename and t:
-                    filename = filename.replace("_top", "", 1)
-                o.write("newmtl %s\n" % filename[:-4])
-                o.write("Ka %.3f %.3f %.3f\n" % (rgb[0], rgb[1], rgb[2]))
-                o.write("Kd %.3f %.3f %.3f\n" % (rgb[0], rgb[1], rgb[2]))
-                o.write("Ks 0.000 0.000 0.000\n")
-                o.write("d %.1f\n" % rgb[3])
-                o.write("illum 1\n")
-
-
-            
+            if os.path.isfile(f) and filename[-4:] == ".png":
+                if s:
+                    rgb = [0, 0, 0, 0]
+                    img = Image.open(f).convert('RGBA')
+                    pixels = 0
+                    for x in range(img.size[0]):
+                        for y in range(img.size[1]):
+                            pixel = img.getpixel((x, y))
+                            if not (pixel[3] == 0 and a):
+                                rgb[0] += pixel[0]
+                                rgb[1] += pixel[1]
+                                rgb[2] += pixel[2]
+                                rgb[3] += pixel[3]
+                                pixels += 1
+                    if pixels > 0:
+                        rgb[0] /= (pixels * 255)
+                        rgb[1] /= (pixels * 255)
+                        rgb[2] /= (pixels * 255)
+                        rgb[3] /= (pixels * 255)
+                    if "grass" in filename or "leaves" in filename:
+                        # #8eb971
+                        rgb[0] = 142 / 255
+                        rgb[1] = 185 / 255
+                        rgb[2] = 113 / 255
+                    if "top" in filename and t:
+                        filename = filename.replace("_top", "", 1)
+                    writeMaterialSection(o, filename[:-4], rgb, rgb[3])
+                else:
+                    writeMaterialSection(o, filename[:-4], (1, 1, 1), 1)
+                    locFilename = "./mtlFiles/" + filename
+                    o.write("map_Ka %s\n" % locFilename)
+                    o.write("map_Kd %s\n" % locFilename)
+                    shutil.copyfile(f, locFilename)