Make BRDF example more intresting

Author: michidk

crates/engine/examples/brdf/main.rs

@@ -1,14 +1,19 @@
-use std::{path::Path, rc::Rc};
+use std::{
+    cell::Cell,
+    path::Path,
+    rc::{Rc, Weak},
+};
 
 /// Renders a brdf example
 use gfx_maths::*;
 use vulkan_engine::{
-    core::engine::Engine,
+    core::{engine::Engine, input::Input},
     scene::{
         component::{
             camera_component::CameraComponent, debug_movement_component::DebugMovementComponent,
-            light_component::LightComponent, renderer::RendererComponent,
+            light_component::LightComponent, renderer::RendererComponent, Component,
         },
+        entity::Entity,
         light::{DirectionalLight, PointLight},
         material::MaterialPipeline,
         model::Model,
@@ -112,43 +117,24 @@ fn setup(engine: &mut Engine) {
         }
     }
 
-    scene
-        .new_entity_with_transform(
-            "DirLight1".to_string(),
-            Transform {
-                position: Vec3::zero(),
-                rotation: Quaternion::axis_angle(Vec3::new(1.0, 0.0, 0.0), -90.0f32.to_radians()),
-                scale: Vec3::one(),
-            },
-        )
-        .new_component::<LightComponent>()
-        .light
-        .set(
-            DirectionalLight {
-                direction: Vec4::zero(),
-                illuminance: Vec4::new(10.1, 10.1, 10.1, 0.0),
-            }
-            .into(),
-        );
+    let sun = scene.new_entity_with_transform(
+        "Sun".to_string(),
+        Transform {
+            position: Vec3::zero(),
+            rotation: Quaternion::axis_angle(Vec3::new(1.0, 0.0, 0.0), -90.0f32.to_radians()),
+            scale: Vec3::one(),
+        },
+    );
 
-    scene
-        .new_entity_with_transform(
-            "DirLight2".to_string(),
-            Transform {
-                position: Vec3::zero(),
-                rotation: Quaternion::axis_angle(Vec3::new(1.0, 0.0, 0.0), 90.0f32.to_radians()),
-                scale: Vec3::one(),
-            },
-        )
-        .new_component::<LightComponent>()
-        .light
-        .set(
-            DirectionalLight {
-                direction: Vec4::zero(),
-                illuminance: Vec4::new(1.6, 1.6, 1.6, 0.0),
-            }
-            .into(),
-        );
+    sun.new_component::<LightComponent>().light.set(
+        DirectionalLight {
+            direction: Vec4::zero(),
+            illuminance: Vec4::new(239.0, 245.0, 218.0, 0.0) / 50.0,
+        }
+        .into(),
+    );
+
+    sun.new_component::<RotationComponent>();
 
     scene
         .new_entity_with_transform(
@@ -188,46 +174,9 @@ fn setup(engine: &mut Engine) {
         );
     scene
         .new_entity_with_transform(
-            "PointLight White 3".to_string(),
-            Transform {
-                position: Vec3::new(0.1, -3.0, -3.0),
-                rotation: Quaternion::identity(),
-                scale: Vec3::one(),
-            },
-        )
-        .new_component::<LightComponent>()
-        .light
-        .set(
-            PointLight {
-                position: Vec4::zero(),
-                luminous_flux: Vec4::new(100.0, 100.0, 100.0, 0.0),
-            }
-            .into(),
-        );
-    scene
-        .new_entity_with_transform(
-            "PointLight White 4".to_string(),
-            Transform {
-                position: Vec3::new(0.1, -3.0, -3.0),
-                rotation: Quaternion::identity(),
-                scale: Vec3::one(),
-            },
-        )
-        .new_component::<LightComponent>()
-        .light
-        .set(
-            PointLight {
-                position: Vec4::zero(),
-                luminous_flux: Vec4::new(100.0, 100.0, 100.0, 0.0),
-            }
-            .into(),
-        );
-
-    scene
-        .new_entity_with_transform(
-            "PointLight Red".to_string(),
+            "PointLight Cyan 3".to_string(),
             Transform {
-                position: Vec3::new(0.0, 0.0, -3.0),
+                position: Vec3::new(0.0, 0.0, 8.0),
                 rotation: Quaternion::identity(),
                 scale: Vec3::one(),
             },
@@ -237,10 +186,49 @@ fn setup(engine: &mut Engine) {
         .set(
             PointLight {
                 position: Vec4::zero(),
-                luminous_flux: Vec4::new(100.0, 0.0, 0.0, 0.0),
+                luminous_flux: Vec4::new(0.0, 160.0, 145.0, 0.0) * 2.0,
             }
             .into(),
         );
 
     scene.load();
 }
+
+#[derive(Debug)]
+struct RotationComponent {
+    entity: Weak<Entity>,
+    rotation_speed: Cell<f32>,
+}
+
+impl Component for RotationComponent {
+    fn create(entity: &Rc<Entity>) -> Rc<Self>
+    where
+        Self: Sized,
+    {
+        Rc::new(Self {
+            entity: Rc::downgrade(entity),
+            rotation_speed: Cell::new(30.0),
+        })
+    }
+
+    fn load(&self) {}
+
+    fn start(&self) {}
+
+    fn update(&self, _: &Input, delta: f32) {
+        if let Some(entity) = self.entity.upgrade() {
+            let mut transform = entity.transform.borrow_mut();
+
+            let mut rotation = transform.rotation;
+            rotation = Quaternion::axis_angle(
+                Vec3::new(0.0, 0.0, 1.0),
+                self.rotation_speed.get().to_radians() * delta,
+            ) * rotation;
+            transform.rotation = rotation;
+        }
+    }
+
+    fn inspector_name(&self) -> &'static str {
+        "TimeOfDayComponent"
+    }
+}