Tidy up.

src/product_gen/generate_product.py

@@ -0,0 +1,46 @@
+import os
+import bpy
+import product_gen.generate_lamp
+import rendering.setup_scene
+import rendering.setup_renderer
+
+def generate(shape, 
+             height, 
+             radius, 
+             color, 
+             output_prefix,
+             is_final=False):
+    
+    rendering.setup_scene.initialize_scene()
+    
+    # Generate models
+    models = product_gen.generate_lamp.generate_models(shape, height, radius)
+    
+    # Optimize geometries for render
+    product_gen.generate_lamp.optimize_for_render(models)
+
+    # Set up materials and textures
+    colormap = {"shade": color,
+                "base": (220.0/255.0, 220.0/255.0, 220.0/255.0),
+                "chord": (20.0/255.0, 20.0/255.0, 20.0/255.0)}
+    product_gen.generate_lamp.apply_textures(colormap, models)
+    
+    rendering.setup_scene.setup_scene()
+    
+    rendering.setup_renderer.setup_renderer()
+    
+    # Do rendering
+    prefix = os.getcwd() + "/" + output_prefix + "/" + output_prefix
+    if is_final:
+        bpy.context.scene.render.filepath = prefix + "_kitchen.png"
+        bpy.ops.render.render(write_still=True)
+        
+        bpy.context.scene.render.filepath = prefix + "_hall.png"
+        bpy.ops.render.render(write_still=True)
+        
+        bpy.context.scene.render.filepath = prefix + "_landing.png"
+        bpy.ops.render.render(write_still=True)
+        
+    else:
+        bpy.context.scene.render.filepath = prefix + ".png"
+        bpy.ops.render.render(write_still=True)