Initial changes for airplane shadows

duncanspumpkin · duncanspumpkin · commit f7647f0020b0 · 2024-09-23T08:13:44.000+01:00
diff --git a/loco-graphics-helper/builders/materials_builder.py b/loco-graphics-helper/builders/materials_builder.py
@@ -28,6 +28,8 @@ def build(self, context):
         self.create_recolorable_material("Recolorable 1", (0.4, 0.3, 0.5), 1)
         self.create_recolorable_material("Recolorable 2", (0.9, 0.1, 0.4), 2)
 
+        self.create_shadow_capture_material("Shadow Capture")
+
     def create_recolorable_material(self, name, color, pass_index):
         material = self.create_material(name)
 
@@ -43,6 +45,17 @@ def create_recolorable_material(self, name, color, pass_index):
 
         return material
 
+    def create_shadow_capture_material(self, name):
+        material = self.create_material(name)
+
+        material.diffuse_color = (0.8, 0.8, 0.8)
+        material.diffuse_intensity = 1
+
+        material.use_only_shadow = True
+        material.shadow_only_type = 'SHADOW_ONLY'
+
+        return material
+
     def create_world_position_material(self, context):
         material = self.create_material("WorldPosition")
 
diff --git a/loco-graphics-helper/builders/scene_builder.py b/loco-graphics-helper/builders/scene_builder.py
@@ -71,6 +71,15 @@ def build(self, context):
 
         scene.objects.link(dome_light_obj)
         dome_light_obj.parent = vertical_joint_obj
+        
+        # Airplane shadows light
+        airplane_light_obj = self.create_airplane_shadow_light(context)
+
+        b = scene.objects.link(airplane_light_obj)
+        airplane_light_obj.parent = vertical_joint_obj
+        airplane_light_obj.layers[2] = True
+        airplane_light_obj.layers[0] = False
+        print(airplane_light_obj.layers[0])
 
         # Environment lighting
         light_settings = scene.world.light_settings
@@ -81,6 +90,15 @@ def build(self, context):
         light_settings.distance = 0
         light_settings.samples = 1
 
+        bpy.ops.mesh.primitive_plane_add(location=(0,0,0), radius=20)
+        shadow_catcher = bpy.context.active_object
+        shadow_catcher.name = "ShadowCatcher"
+        shadow_catcher.layers[2] = True
+        shadow_catcher.layers[0] = False
+        shadow_catcher.hide = True
+        shadow_catcher.hide_select = True
+        shadow_catcher.data.materials.append(bpy.data.materials.get("Shadow Capture"))
+
     def create_camera(self, context):
         name = self.prefix + "Camera" + self.suffix
         if name in bpy.data.cameras:
@@ -163,6 +181,27 @@ def create_light_dome(self, context):
 
         return lamp_object
 
+    def create_airplane_shadow_light(self, context):
+        lamp_data = self.create_lamp_data(context, "AirplaneShadowLight", "SUN")
+
+        lamp_data.energy = 1.3
+        lamp_data.use_specular = True
+        lamp_data.use_diffuse = True
+        lamp_data.shadow_method = "RAY_SHADOW"
+        lamp_data.shadow_ray_sample_method = "ADAPTIVE_QMC"
+        lamp_data.shadow_ray_samples = 4
+        lamp_data.shadow_soft_size = 0.5
+        lamp_data.shadow_adaptive_threshold = 0.001
+
+        lamp_object = self.create_scene_object(context, 'AirplaneShadowLight', lamp_data)
+
+        lamp_object.hide = True
+        lamp_object.hide_select = True
+        lamp_object.location = (0, 0, 0)
+        lamp_object.rotation_euler = (math.radians(0), 0, math.radians(90))
+
+        return lamp_object
+
     def create_scene_object(self, context, name, data=None):
         name = self.prefix + name + self.suffix
         if name in context.scene.objects:
diff --git a/loco-graphics-helper/builders/task_builder.py b/loco-graphics-helper/builders/task_builder.py
@@ -38,6 +38,8 @@ def __init__(self):
 
         self.recolorables = 0
 
+        self.cast_shadows = True
+
         self.layer = "Editor"
 
         self.palette = None
@@ -67,6 +69,8 @@ def add_frame(self, frame_index, number_of_viewing_angles, angle_index, animatio
 
         frame.set_recolorables(self.recolorables)
 
+        frame.set_cast_shadows(self.cast_shadows)
+
         frame.set_layer(self.layer)
 
         frame.set_base_palette(self.palette)
@@ -104,6 +108,8 @@ def add_viewing_angles(self, number_of_viewing_angles, animation_frame_index=0,
 
                 frame.set_recolorables(self.recolorables)
 
+                frame.set_cast_shadows(self.cast_shadows)
+
                 frame.set_layer(self.layer)
 
                 frame.set_base_palette(self.palette)
@@ -146,6 +152,10 @@ def add_viewing_angles(self, number_of_viewing_angles, animation_frame_index=0,
     def set_recolorables(self, number_of_recolorables):
         self.recolorables = number_of_recolorables
 
+    # Sets the number of recolorable materials
+    def set_cast_shadows(self, cast_shadows):
+        self.cast_shadows = cast_shadows
+
     # Sets the base palette to use
     def set_palette(self, palette):
         self.palette = palette
diff --git a/loco-graphics-helper/frame.py b/loco-graphics-helper/frame.py
@@ -42,6 +42,8 @@ def __init__(self, frame_index, task, view_angle, bank_angle=0, vertical_angle=0
         self.anti_alias_with_background = False
         self.maintain_aliased_silhouette = True
 
+        self.cast_shadows = True
+
         self.offset_x = 0
         self.offset_y = 0
 
@@ -150,3 +152,6 @@ def set_output_indices(self, indices):
     
     def set_target_object(self, object):
         self.target_object = object
+
+    def set_cast_shadows(self, cast_shadows):
+        self.cast_shadows = cast_shadows
diff --git a/loco-graphics-helper/loco_object_helper_panel.py b/loco-graphics-helper/loco_object_helper_panel.py
@@ -130,3 +130,6 @@ def draw_car_panel(self, context, layout):
         if vehicle_properties.braking_lights and vehicle_properties.roll_angle != 0:
             row.label("WARNING CANNOT HAVE BOTH BRAKING LIGHTS AND ROLL ANGLE SET") 
             row = layout.row()
+
+        row.prop(vehicle_properties, "is_airplane")
+        row = layout.row()
diff --git a/loco-graphics-helper/operators/init_operator.py b/loco-graphics-helper/operators/init_operator.py
@@ -57,18 +57,27 @@ def execute(self, context):
                                          True, False, False, False, False, False, False, False, False, False, False)
         braking_lights_layer.use = False
 
+        braking_lights_layer = self.create_render_layer(context, "Top Down Shadow")
+        braking_lights_layer.layers = (False, False, True, False, False, False, False, False,
+                                   False, False, False, False, False, False, False, False, False, False, False, False)
+        braking_lights_layer.layers_zmask = (False, False, True, False, False, False, False, False, False,
+                                         False, False, False, False, False, False, False, False, False, False, False)
+        braking_lights_layer.use = False
+
         self.delete_default_render_layer(context)
 
         # Create dependencies in the context
+
+        # Materials used in scene builder so do this first
+        materialsBuilder = MaterialsBuilder()
+        materialsBuilder.build(context)
+
         sceneBuilder = SceneBuilder()
         sceneBuilder.build(context)
 
         compositorBuilder = CompositorBuilder()
         compositorBuilder.build(context)
 
-        materialsBuilder = MaterialsBuilder()
-        materialsBuilder.build(context)
-
         return {'FINISHED'}
 
     def delete_default_render_layer(self, context):
diff --git a/loco-graphics-helper/operators/render_tiles_operator.py b/loco-graphics-helper/operators/render_tiles_operator.py
@@ -34,6 +34,9 @@ def create_task(self, context):
         self.task_builder.set_recolorables(
             general_props.number_of_recolorables)
 
+        self.task_builder.set_cast_shadows(
+            general_props.cast_shadows)
+
         self.task_builder.set_size(
             props.object_width, props.object_length, props.invert_tile_positions)
 
diff --git a/loco-graphics-helper/operators/vehicle_render_operator.py b/loco-graphics-helper/operators/vehicle_render_operator.py
@@ -31,6 +31,8 @@ def create_task(self, context):
         # Add vehicle frames
         self.task_builder.set_recolorables(
             general_props.number_of_recolorables)
+        self.task_builder.set_cast_shadows(
+            general_props.cast_shadows)
         self.task_builder.set_palette(self.palette_manager.get_base_palette(
             general_props.palette, general_props.number_of_recolorables, "FULL"))
 
@@ -39,10 +41,15 @@ def create_task(self, context):
         for car_object in cars:
             self.add_render_angles(car_object)
 
-        bogies = [x for x in context.scene.objects if x.loco_graphics_helper_object_properties.object_type == "BOGIE"]
-        bogies = sorted(bogies, key=lambda x: x.loco_graphics_helper_vehicle_properties.index)
-        for bogie_object in bogies:
-            self.add_render_angles(bogie_object)
+        if cars[0].loco_graphics_helper_vehicle_properties.is_airplane:
+            self.task_builder.set_cast_shadows(
+                True)
+            self.add_airplane_shadow_render_angles(cars[0])
+        else:
+            bogies = [x for x in context.scene.objects if x.loco_graphics_helper_object_properties.object_type == "BOGIE"]
+            bogies = sorted(bogies, key=lambda x: x.loco_graphics_helper_vehicle_properties.index)
+            for bogie_object in bogies:
+                self.add_render_angles(bogie_object)
 
         return self.task_builder.create_task(context)
 
@@ -122,3 +129,31 @@ def add_render_angles(self, car_object):
                                 self.task_builder.add_frame(
                                     frame_index, num_viewing_angles, i, 0, rotation_range, car_object)
                                 self.task_builder.set_layer("Editor")
+
+    def add_airplane_shadow_render_angles(self, car_object):
+        props = car_object.loco_graphics_helper_vehicle_properties
+        
+        track_sections = track_angle_sections["VEHICLE_SPRITE_FLAG_FLAT"]
+        for track_section in track_sections:
+
+            base_view_angle = 0
+            self.task_builder.set_rotation(
+                base_view_angle, 0, vertical_angle=track_section[2])
+
+            num_viewing_angles = int(props.flat_viewing_angles)
+
+            rotational_symmetry = props.rotational_symmetry
+
+            if rotational_symmetry:
+                num_viewing_angles = int(num_viewing_angles / 2)
+
+            rotation_range = 180 if rotational_symmetry else 360
+
+            start_output_index = self.task_builder.output_index
+
+            for i in range(num_viewing_angles):
+                self.task_builder.set_layer("Top Down Shadow")
+                frame_index = start_output_index + i
+                self.task_builder.add_frame(
+                    frame_index, num_viewing_angles, i, 0, rotation_range, car_object)
+                self.task_builder.set_layer("Editor")
diff --git a/loco-graphics-helper/operators/walls_render_operator.py b/loco-graphics-helper/operators/walls_render_operator.py
@@ -94,6 +94,9 @@ def create_task(self, context):
         self.task_builder.set_recolorables(
             general_props.number_of_recolorables)
 
+        self.task_builder.set_cast_shadows(
+            general_props.cast_shadows)
+
         sloped_a_anim_frame = general_props.number_of_animation_frames
         sloped_b_anim_frame = general_props.number_of_animation_frames * 2
 
diff --git a/loco-graphics-helper/processors/sub_processes/frame_processors/render_processor.py b/loco-graphics-helper/processors/sub_processes/frame_processors/render_processor.py
@@ -35,6 +35,8 @@ def process(self, frame, callback):
 
         self.renderer.set_aa(frame.use_anti_aliasing)
         self.renderer.set_aa_with_background(frame.anti_alias_with_background)
+        
+        self.renderer.set_cast_shadows(frame.cast_shadows)
 
         self.renderer.set_animation_frame(frame.animation_frame_index)
 
diff --git a/loco-graphics-helper/properties/object_properties.py b/loco-graphics-helper/properties/object_properties.py
@@ -19,6 +19,7 @@ def object_type_update_func(self, context):
         props.sloped_viewing_angles = "32"
         props.roll_angle = 0
         props.braking_lights = False
+        props.is_airplane = False
 
     # Reset to default for cars
     if object.loco_graphics_helper_object_properties.object_type == "CAR":
diff --git a/loco-graphics-helper/properties/vehicle_properties.py b/loco-graphics-helper/properties/vehicle_properties.py
@@ -110,6 +110,12 @@ class VehicleProperties(bpy.types.PropertyGroup):
         default=False
     )
 
+    is_airplane = bpy.props.BoolProperty(
+        name="Is an airplane",
+        description="If airplane will render airplane shadows (bogie)",
+        default=False
+    )
+
 
 def register_vehicles_properties():
     bpy.types.Object.loco_graphics_helper_vehicle_properties = bpy.props.PointerProperty(
diff --git a/loco-graphics-helper/rct_graphics_helper_panel.py b/loco-graphics-helper/rct_graphics_helper_panel.py
@@ -232,6 +232,9 @@ def get_number_of_sprites(object):
             num_sprites = num_sprites + (int(props.sloped_viewing_angles) * 2 + num_transition_sprites) * multiplier
         if props.sprite_track_flags[2]:
             num_sprites = num_sprites + (int(props.sloped_viewing_angles) * 2 + num_transition_sprites) * multiplier
+        
+        if props.is_airplane:
+            num_sprites = num_sprites + int(props.flat_viewing_angles) * multiplier
         return int(num_sprites)
 
     def draw_vehicle_panel(self, scene, layout):
diff --git a/loco-graphics-helper/renderer.py b/loco-graphics-helper/renderer.py
@@ -58,7 +58,6 @@ def __init__(self, context, palette_manager):
                                          context.scene.render.resolution_x)
 
         self.started_with_anti_aliasing = context.scene.render.use_antialiasing
-        context.scene.render.use_shadows = context.scene.loco_graphics_helper_general_properties.cast_shadows
 
         bpy.app.handlers.render_complete.append(self._render_finished)
         bpy.app.handlers.render_cancel.append(self._render_reset)
@@ -140,6 +139,7 @@ def set_aa_with_background(self, aa_with_background):
     def set_override_material(self, material):
         self.context.scene.render.layers["Editor"].material_override = material
         self.context.scene.render.layers["Braking Lights"].material_override = material
+        self.context.scene.render.layers["Top Down Shadow"].material_override = material
 
     def set_multi_tile_size(self, width, length):
         width_node = None
@@ -158,7 +158,7 @@ def set_multi_tile_size(self, width, length):
 
     # Sets the active render layer
     def set_layer(self, layer_name):
-        layers = ["Editor", "Braking Lights"]
+        layers = ["Editor", "Braking Lights", "Top Down Shadow"]
 
         for layer in layers:
             self.context.scene.render.layers[layer].use = False
@@ -176,6 +176,9 @@ def set_layer(self, layer_name):
     def set_animation_frame(self, animation_frame_index):
         self.context.scene.frame_set(animation_frame_index)
 
+    def set_cast_shadows(self, cast_shadows):
+        self.context.scene.render.use_shadows = cast_shadows
+
     # Sets the still render output path
     def set_output_path(self, path):
         self.context.scene.render.filepath = path
