From c685177a86e18a477131610fd4667aa5627c7e4e Mon Sep 17 00:00:00 2001
From: "a.yamaev" <a.yamaev@smartengines.com>
Date: Wed, 6 Dec 2023 15:13:09 +0300
Subject: [PATCH] Sheep counter

---
 src/debug_diagnostic.rs | 45 +++++++++++++++++++++++++++++++++++------
 src/lib.rs              |  3 +++
 src/safe_area.rs        | 45 +++++++++++++++++++++++++++++++++++++++++
 src/sheep.rs            |  1 +
 4 files changed, 88 insertions(+), 6 deletions(-)

diff --git a/src/debug_diagnostic.rs b/src/debug_diagnostic.rs
index b9bfdf0..a627a91 100644
--- a/src/debug_diagnostic.rs
+++ b/src/debug_diagnostic.rs
@@ -1,14 +1,19 @@
 use bevy::prelude::*;
 
+use crate::safe_area::SheepCounter;
+
 pub struct DiagnosticPlugin;
 
 impl Plugin for DiagnosticPlugin {
     fn build(&self, app: &mut App) {
         app.add_systems(
             Startup,
-            (setup_diagnostic_panel, apply_deferred, setup_counter),
+            (setup_diagnostic_panel, apply_deferred, setup_counter, setup_sheep_counter).chain(),
         )
-        .add_systems(Update, (fps_counting,))
+        .add_systems(Update, (
+            fps_counting,
+            sheep_counter_text,
+        ))
         .add_plugins(bevy_inspector_egui::quick::WorldInspectorPlugin::default());
     }
 }
@@ -22,16 +27,18 @@ pub fn setup_diagnostic_panel(mut commands: Commands) {
             top: Val::Px(0.0),
             left: Val::Px(0.0),
             width: Val::Px(100.0),
-            height: Val::Percent(100.0),
             position_type: PositionType::Absolute,
-            flex_direction: FlexDirection::Row,
-            align_items: AlignItems::Center,
+            flex_direction: FlexDirection::Column,
             justify_content: JustifyContent::Center,
+            align_items: AlignItems::Center,
+
+            align_self: AlignSelf::Stretch,
+
             ..default()
         },
         background_color: BackgroundColor(Color::rgba(0.0, 0.0, 0.0, 0.5)),
         ..default()
-    });
+    }).insert(DiagnosticPanel);
 }
 
 #[derive(Component)]
@@ -53,3 +60,29 @@ fn fps_counting(mut query: Query<&mut Text, With<FrameCounter>>, time: Res<Time>
         text.sections[0].value = format!("FPS: {:.0}", 1.0 / time.delta_seconds());
     }
 }
+
+#[derive(Component)]
+pub struct ShipDebugCounter;
+
+pub fn setup_sheep_counter(
+    mut commands: Commands,
+    panels: Query<Entity, With<DiagnosticPanel>>,
+) {
+    let sheep_counter = commands
+        .spawn(TextBundle::from_section("Sheep in safe area: ", TextStyle::default()))
+        .insert(ShipDebugCounter)
+        .id();
+
+    if let Ok(panel) = panels.get_single() {
+        commands.entity(panel).add_child(sheep_counter);
+    }
+}
+
+pub fn sheep_counter_text(
+    mut query: Query<&mut Text, With<ShipDebugCounter>>,
+    sheep_counter : Res<SheepCounter>,
+) {
+    for mut text in &mut query {
+        text.sections[0].value = format!("Sheep in safe area: {}", sheep_counter.count);
+    }
+}
\ No newline at end of file
diff --git a/src/lib.rs b/src/lib.rs
index 17eb283..afa1e45 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -53,6 +53,9 @@ impl Plugin for GamePlugin {
             sprite_material::SpriteMaterialPlugin,
         ));
 
+        //For long term updates
+        app.insert_resource(Time::<Fixed>::from_seconds(1.0));
+
         app.add_systems(Startup, (test_level::setup, sheep::setup));
         // TODO: Move to plugin
         app.add_systems(
diff --git a/src/safe_area.rs b/src/safe_area.rs
index 5700ac1..e74fc61 100644
--- a/src/safe_area.rs
+++ b/src/safe_area.rs
@@ -4,11 +4,17 @@ use std::f32::consts::PI;
 
 use bevy::prelude::*;
 
+use crate::sheep::Sheep;
+
 pub struct SafeAreaPlugin;
 
 impl Plugin for SafeAreaPlugin {
     fn build(&self, app: &mut App) {
         app.add_systems(Update, draw_safe_area);
+
+        app.add_systems(FixedUpdate, count_sheeps);
+
+        app.init_resource::<SheepCounter>();
     }
 }
 
@@ -18,6 +24,24 @@ pub enum SafeArea {
     Ellipse { pos1: Vec2, pos2: Vec2, radius: f32 },
 }
 
+impl SafeArea {
+    fn in_area(&self, sheep_pos: Vec2) -> bool {
+        match self {
+            SafeArea::Rect { pos, size } => {
+                let dx = sheep_pos.x - pos.x;
+                let dy = sheep_pos.y - pos.y;
+
+                dx.abs() < size.x / 2.0 && dy.abs() < size.y / 2.0
+            },
+            SafeArea::Ellipse { pos1, pos2, radius } => {
+                let d = (*pos1 - *pos2).length();
+                let r = (*pos1 - sheep_pos).length();
+                r * r <= d * d
+            },
+        }
+    }
+}
+
 fn draw_safe_area(mut gizmos: Gizmos, query: Query<&SafeArea>) {
     for safe_area in query.iter() {
         match safe_area {
@@ -33,3 +57,24 @@ fn draw_safe_area(mut gizmos: Gizmos, query: Query<&SafeArea>) {
         }
     }
 }
+
+#[derive(Resource, Default)]
+pub struct SheepCounter {
+    pub count: u32,
+}
+
+fn count_sheeps(
+    mut safe_areas : Query<&SafeArea>,
+    mut sheep : Query<&Transform, With<Sheep>>,
+    mut counter : ResMut<SheepCounter>,
+) {
+    let mut count = 0;
+    for safe_area in safe_areas.iter() {
+        for sheep in sheep.iter() {
+            if safe_area.in_area(Vec2::new(sheep.translation.x, sheep.translation.z)) {
+                count += 1;
+            }
+        }
+    }
+    counter.count = count;
+}
\ No newline at end of file
diff --git a/src/sheep.rs b/src/sheep.rs
index c429419..a77f30a 100644
--- a/src/sheep.rs
+++ b/src/sheep.rs
@@ -52,6 +52,7 @@ pub fn scared_sheeps(
             if sheep.1.translation.distance(bark_origin) <= bark.radius {
                 sheep.3 .0 = true;
                 sheep.2 .0 = sheep.1.translation - bark_origin;
+                sheep.2 .0.y = 0.0; //sheep must not fly and be in fixed height
             }
         }
     }