From 965ccb56c4c58066939d74bea22c2a9b59416d0d Mon Sep 17 00:00:00 2001
From: Tom Barrett <tom@tombarrett.xyz>
Date: Sun, 21 Mar 2021 17:51:55 +0100
Subject: moved cell stuff to another file

---
 src/main.rs | 215 ++----------------------------------------------------------
 1 file changed, 6 insertions(+), 209 deletions(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index f0a7eb6..566ded5 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,181 +1,8 @@
 use bevy::math::Vec3;
 use bevy::prelude::*;
+use gems::cell::{self, Cell};
 use gems::constants;
-use rand::{
-    distributions::{Distribution, Standard},
-    thread_rng, Rng,
-};
-
-#[derive(Debug, PartialEq, Clone, Copy)]
-enum Occupant {
-    None,
-    Green,
-    Yellow,
-    Red,
-    Diamond,
-    Explosion,
-}
-
-impl Default for Occupant {
-    fn default() -> Occupant {
-        Occupant::None
-    }
-}
-
-impl Occupant {
-    pub fn to_index(&self) -> u32 {
-        match self {
-            Occupant::Green => 0,
-            Occupant::Yellow => 1,
-            Occupant::Red => 2,
-            Occupant::Diamond => 3,
-            Occupant::Explosion => 4,
-            Occupant::None => 13,
-        }
-    }
-}
-
-impl Distribution<Occupant> for Standard {
-    fn sample<R: Rng + ?Sized>(&self, rng: &mut R) -> Occupant {
-        match rng.gen_range(0..=3) {
-            0 => Occupant::Green,
-            1 => Occupant::Yellow,
-            2 => Occupant::Diamond,
-            3 => Occupant::Red,
-            _ => Occupant::None,
-        }
-    }
-}
-
-#[derive(Debug, Clone, Default, Copy)]
-struct Cell {
-    x: usize,
-    y: usize,
-    occupant: Occupant,
-    selected: bool,
-    hovered: bool,
-}
-
-impl Cell {
-    pub fn new(x: usize, y: usize) -> Cell {
-        Cell {
-            x,
-            y,
-            occupant: Occupant::None,
-            selected: false,
-            hovered: false,
-        }
-    }
-}
-
-fn cell_insert_system(mut cell_query: Query<(&mut Cell, &mut TextureAtlasSprite)>) {
-    for (mut cell, mut sprite) in cell_query.iter_mut() {
-        if cell.occupant == Occupant::None && cell.y == constants::ROWS - 1 {
-            cell.occupant = rand::random();
-            sprite.index = cell.occupant.to_index();
-        }
-    }
-}
-
-fn explosion_animation_system(
-    time: Res<Time>,
-    commands: &mut Commands,
-    mut cell_query: Query<(
-        Entity,
-        &mut Cell,
-        &mut TextureAtlasSprite,
-        Option<&mut Timer>,
-    )>,
-) {
-    for (entity, mut cell, mut sprite, timer) in cell_query.iter_mut() {
-        if cell.occupant == Occupant::Explosion {
-            if let Some(mut timer) = timer {
-                timer.tick(time.delta_seconds());
-                if timer.finished() {
-                    if sprite.index < 7 && sprite.index >= 4 {
-                        sprite.index += 1;
-                    } else {
-                        cell.occupant = Occupant::None;
-                        sprite.index = cell.occupant.to_index();
-                        commands.remove_one::<Timer>(entity);
-                    }
-                }
-            } else {
-                commands.insert(entity, (Timer::from_seconds(0.1, true),));
-            }
-        }
-    }
-}
-
-fn cell_check_system(mut cell_query: Query<(&mut Cell, &mut TextureAtlasSprite)>) {
-    let mut cells = [[Cell::default(); constants::COLUMNS]; constants::ROWS];
-    for (cell, _) in cell_query.iter_mut() {
-        cells[cell.x][cell.y] = *cell;
-    }
-
-    let mut last = Occupant::None;
-    let mut connected = Vec::new();
-
-    for (i, _) in cells.iter().enumerate() {
-        let mut c = Vec::new();
-        for j in 0..constants::ROWS {
-            if cells[i][j].occupant == last && last != Occupant::None {
-                c.push((i, j));
-                c.push((i, j - 1));
-            } else {
-                connected.push(c.clone());
-                c.clear();
-            }
-            last = cells[i][j].occupant;
-        }
-        connected.push(c);
-        last = Occupant::None;
-    }
-
-    for c in connected.iter() {
-        if c.len() > 4 {
-            for (i, j) in c.iter() {
-                for (mut cell, mut sprite) in cell_query.iter_mut() {
-                    if &cell.x == i && &cell.y == j && cell.occupant != Occupant::Explosion {
-                        cell.occupant = Occupant::Explosion;
-                        sprite.index = cell.occupant.to_index();
-                    }
-                }
-            }
-        }
-    }
-
-    connected.clear();
-
-    for (i, row) in cells.iter().enumerate() {
-        let mut c = Vec::new();
-        for (j, _) in row.iter().enumerate() {
-            if cells[j][i].occupant == last && last != Occupant::None {
-                c.push((j, i));
-                c.push((j - 1, i));
-            } else {
-                connected.push(c.clone());
-                c.clear();
-            }
-            last = cells[j][i].occupant;
-        }
-        connected.push(c);
-        last = Occupant::None;
-    }
-
-    for c in connected.iter() {
-        if c.len() > 4 {
-            for (i, j) in c.iter() {
-                for (mut cell, mut sprite) in cell_query.iter_mut() {
-                    if &cell.x == i && &cell.y == j && cell.occupant != Occupant::Explosion {
-                        cell.occupant = Occupant::Explosion;
-                        sprite.index = cell.occupant.to_index();
-                    }
-                }
-            }
-        }
-    }
-}
+use rand::{thread_rng, Rng};
 
 fn star_spawning_system(commands: &mut Commands, time: Res<Time>, mut q: Query<&mut Timer>) {
     for mut timer in q.iter_mut() {
@@ -204,36 +31,6 @@ fn star_spawning_system(commands: &mut Commands, time: Res<Time>, mut q: Query<&
     }
 }
 
-fn cell_falling_system(mut cell_query: Query<(&mut Cell, &mut TextureAtlasSprite)>) {
-    let mut have_gems = Vec::new();
-    for (cell, _sprite) in cell_query.iter_mut() {
-        if cell.occupant != Occupant::None {
-            have_gems.push(*cell);
-        }
-    }
-
-    let mut moved_gems = Vec::new();
-    for (mut cell, mut sprite) in cell_query.iter_mut() {
-        if cell.occupant == Occupant::None {
-            if let Some(c) = have_gems
-                .iter()
-                .find(|&c| (c.x, c.y) == (cell.x, cell.y + 1))
-            {
-                cell.occupant = c.occupant;
-                sprite.index = cell.occupant.to_index();
-                moved_gems.push(c);
-            }
-        }
-    }
-
-    for (mut cell, mut sprite) in cell_query.iter_mut() {
-        if moved_gems.iter().any(|c| (c.x, c.y) == (cell.x, cell.y)) {
-            cell.occupant = Occupant::None;
-            sprite.index = cell.occupant.to_index();
-        }
-    }
-}
-
 pub fn setup(
     commands: &mut Commands,
     asset_server: Res<AssetServer>,
@@ -444,10 +241,10 @@ pub struct GemsPlugin;
 impl Plugin for GemsPlugin {
     fn build(&self, app: &mut AppBuilder) {
         app.add_startup_system(setup.system());
-        app.add_system(cell_insert_system.system());
-        app.add_system(cell_falling_system.system());
-        app.add_system(cell_check_system.system());
-        app.add_system(explosion_animation_system.system());
+        app.add_system(cell::insert.system());
+        app.add_system(cell::falling.system());
+        app.add_system(cell::check.system());
+        app.add_system(cell::explosion_animation.system());
         app.add_system(mouse_system.system());
         app.add_system(cosmonaut_detect_system.system());
         app.add_system(animation_system.system());
-- 
cgit v1.2.3