Merge pull request #21 from spacek531/menu-item

Menu item

Author: leicestersquare

rct-graphics-helper/__init__.py

@@ -16,6 +16,7 @@
 from .properties.general_properties import register_general_properties, unregister_general_properties
 from .properties.track_properties import register_track_properties, unregister_track_properties
 from .rct_graphics_helper_panel import GraphicsHelperPanel
+from .rct_graphics_render_button import try_register_button, unregister_button
 from . import developer_utils
 import importlib
 import bpy
@@ -24,7 +25,7 @@
     "name": "RCT Graphics Helper",
     "description": "Render tool to replicate RCT graphics",
     "author": "Olivier Wervers",
-    "version": (0, 6, 0),
+    "version": (0, 6, 1),
     "blender": (2, 79, 0),
     "location": "Render",
     "support": "COMMUNITY",
@@ -52,6 +53,7 @@ def register():
     register_vehicles_properties()
     register_walls_properties()
     register_track_properties()
+    try_register_button(bpy.context)
 
     print("Registered {} with {} modules".format(
         bl_info["name"], len(modules)))
@@ -68,5 +70,6 @@ def unregister():
     unregister_vehicles_properties()
     unregister_walls_properties()
     unregister_track_properties()
+    unregister_button()
 
     print("Unregistered {}".format(bl_info["name"]))

rct-graphics-helper/operators/init_operator.py

@@ -15,6 +15,7 @@
 
 from ..builders.scene_builder import SceneBuilder
 from ..builders.compositor_builder import CompositorBuilder
+from ..rct_graphics_render_button import register_button, unregister_button
 
 
 class Init(bpy.types.Operator):
@@ -71,6 +72,9 @@ def execute(self, context):
         materialsBuilder = MaterialsBuilder()
         materialsBuilder.build(context)
 
+        unregister_button()
+        register_button()
+
         return {'FINISHED'}
 
     def delete_default_render_layer(self, context):

rct-graphics-helper/operators/render_switch_operator.py

@@ -0,0 +1,33 @@
+'''
+Copyright (c) 2022 RCT Graphics Helper developers
+
+For a complete list of all authors, please refer to the addon's meta info.
+Interested in contributing? Visit https://github.com/oli414/Blender-RCT-Graphics
+
+RCT Graphics Helper is licensed under the GNU General Public License version 3.
+'''
+
+import bpy
+
+from .vehicle_render_operator import RenderVehicle
+from .walls_render_operator import RenderWalls
+from .render_tiles_operator import RenderTiles
+
+class RenderRCTSwitch(bpy.types.Operator):
+    bl_idname = "render.rct_switch"
+    bl_label = "Render RCT project"
+
+
+    def execute(self, context):
+        
+        properties = context.scene.rct_graphics_helper_general_properties
+        render_mode = properties.render_mode
+
+        if render_mode == "TILES":
+            bpy.ops.render.rct_static()
+        elif render_mode == "VEHICLE":
+            bpy.ops.render.rct_vehicle()
+        elif render_mode == "WALLS":
+            bpy.ops.render.rct_walls()
+        
+        return{'FINISHED'}

rct-graphics-helper/rct_graphics_helper_panel.py

@@ -13,13 +13,7 @@
 
 from .operators.init_operator import Init
 
-from .operators.vehicle_render_operator import RenderVehicle
-
-from .operators.walls_render_operator import RenderWalls
-
-from .operators.track_render_operator import RenderTrack
-
-from .operators.render_tiles_operator import RenderTiles
+from .operators.render_switch_operator import RenderRCTSwitch
 
 from .models.palette import palette_colors, palette_colors_details
 
@@ -40,6 +34,13 @@ def execute(self, context):
     def invoke(self, context, event):
         return context.window_manager.invoke_confirm(self, event)
 
+def addRenderButton(layout, hasFailed = False):
+    row = layout.row()
+    text = "Render"
+    if hasFailed:
+        text = "Render failed"
+    row.operator("render.rct_switch", text=text)
+
 class GraphicsHelperPanel(bpy.types.Panel):
     bl_label = "RCT Graphics Helper"
     bl_idname = "VIEW3D_PT_rct_graphics_helper"
@@ -169,11 +170,7 @@ def draw_tiles_panel(self, scene, layout):
         if properties.object_width > 1 or properties.object_length > 1:
             row.prop(properties, "invert_tile_positions")
 
-        row = layout.row()
-        text = "Render"
-        if general_properties.rendering:
-            text = "Failed"
-        row.operator("render.rct_static", text=text)
+        addRenderButton(layout, general_properties.rendering)
 
     def draw_walls_panel(self, scene, layout):
         properties = scene.rct_graphics_helper_walls_properties
@@ -188,11 +185,7 @@ def draw_walls_panel(self, scene, layout):
         row = layout.row()
         row.prop(properties, "doorway")
 
-        row = layout.row()
-        text = "Render"
-        if general_properties.rendering:
-            text = "Failed"
-        row.operator("render.rct_walls", text=text)
+        addRenderButton(layout, general_properties.rendering)
 
     def draw_track_panel(self, scene, layout):
         properties = scene.rct_graphics_helper_track_properties
@@ -237,8 +230,4 @@ def draw_vehicle_panel(self, scene, layout):
         row = layout.row()
         row.prop(properties, "inverted_set")
 
-        row = layout.row()
-        text = "Render"
-        if general_properties.rendering:
-            text = "Failed"
-        row.operator("render.rct_vehicle", text=text)
+        addRenderButton(layout, general_properties.rendering)

rct-graphics-helper/rct_graphics_render_button.py

@@ -0,0 +1,27 @@
+'''
+Copyright (c) 2022 RCT Graphics Helper developers
+
+For a complete list of all authors, please refer to the addon's meta info.
+Interested in contributing? Visit https://github.com/oli414/Blender-RCT-Graphics
+
+RCT Graphics Helper is licensed under the GNU General Public License version 3.
+'''
+
+import bpy
+
+def draw(self, context):
+    self.layout.operator("render.rct_switch")
+
+def register_button():
+    bpy.types.INFO_MT_render.append(draw)
+
+
+def unregister_button():
+    bpy.types.INFO_MT_render.remove(draw)
+
+
+def try_register_button(context):
+    if "Rig" in context.scene.objects:
+        register_button()
+    else:
+        unregister_button()