Hex table

src/subplate_automation.rs

@@ -1,16 +1,18 @@
-use rayon::iter::ParallelIterator;
-use std::time::Instant;
-use crate::SeededRng;
 use crate::basic::{IntoImage, Table};
+use crate::hex_table::HexTable;
+use crate::{RECTANGLE_SIDE, SeededRng};
 use bevy::asset::{Assets, RenderAssetUsages};
 use bevy::image::{BevyDefault, Image};
 use bevy::input::ButtonInput;
-use bevy::prelude::{Color, Commands, Component, Entity, KeyCode, Query, Res, ResMut, Sprite};
+use bevy::math::Vec2;
+use bevy::prelude::{Color, Commands, Component, Entity, KeyCode, Query, Res, ResMut, Sprite, With};
 use bevy::render::render_resource::{Extent3d, TextureDimension, TextureFormat};
 use bevy::tasks::futures_lite::StreamExt;
-use rand::{ RngExt};
+use rand::RngExt;
 use rand_chacha::ChaCha8Rng;
 use rayon::iter::IntoParallelIterator;
+use rayon::iter::ParallelIterator;
+use std::time::Instant;
 
 pub fn update_automation_view(
     query: Query<(&Sprite, &SubPlateAutomation)>,
@@ -43,7 +45,6 @@ pub fn update_automation(
     mut seeded_rng: ResMut<SeededRng>,
     keys: Res<ButtonInput<KeyCode>>,
     commands: Commands,
-
 ) {
     if query.is_empty() {
         return;
@@ -78,15 +79,21 @@ impl SubPlateAutomation {
 
         for index in 0..self.world.side * self.world.side {
             let current_color = *self.world.get(index);
-            if current_color == Color::BLACK || current_color != Color::WHITE {
+            if current_color == Color::NONE || current_color != Color::WHITE {
                 continue;
             }
 
-            if rng.random::<f32>() < 0.7 {
+            if rng.random::<f32>() < 0.8 {
                 continue;
             }
 
-            let around = self.world.around_line(index).iter().filter(|v| ***v != Color::WHITE && ***v != Color::BLACK).map(|color| (*color).clone()).collect::<Vec<_>>();
+            let around = self
+                .world
+                .around_line(index)
+                .iter()
+                .filter(|v| ***v != Color::WHITE && ***v != Color::NONE)
+                .map(|color| (*color).clone())
+                .collect::<Vec<_>>();
 
             if around.len() == 0 {
                 continue;
@@ -97,38 +104,108 @@ impl SubPlateAutomation {
         }
 
         println!("Time elapsed: {:?} ms", time.elapsed().as_millis());
-
     }
 
     fn smooth(&mut self) {
         let range = 0..self.world.side * self.world.side;
-        let update = range.into_iter().map(|index| {
+        for index in range {
             let around = self.world.around_line(index);
             let mut counts: Vec<(Color, u8)> = vec![];
             for p in around {
                 for count_index in 0..counts.len() {
                     if counts[count_index].0 == *p {
                         counts[count_index].1 += 1;
-                        continue
+                        continue;
                     }
                 }
                 counts.push((p.clone(), 1));
             }
-            counts.iter().max_by_key(|(_, c)| {c}).unwrap().0
-        }).collect::<Vec<Color>>();
-        self.world.data = update;
+            self.world
+                .set(index, counts.iter().max_by_key(|(_, c)| c).unwrap().0);
+        }
     }
 
-    fn seed(&mut self, rng: &mut ChaCha8Rng){
+    fn seed(&mut self, rng: &mut ChaCha8Rng) {
         let len = (self.world.side * self.world.side) as f32;
         loop {
             let index = (rng.random::<f32>() * len) as usize;
-            if *self.world.get(index) == Color::BLACK {
+            if *self.world.get(index) == Color::NONE {
                 continue;
             }
 
-            self.world.set(index, Color::hsv(rng.random::<f32>() * 360.0, 1.0, 1.0));
+            self.world
+                .set(index, Color::hsv(rng.random::<f32>() * 360.0, 1.0, 1.0));
             break;
         }
     }
 }
+
+pub fn setup_hex_matrix(
+    mut request_query: Query<(&HexMatrixRequest, Entity)>,
+    mut view_query: Query<&Sprite, With<HexMatrixView>>,
+    mut automation_query: Query<&SubPlateAutomation>,
+    mut commands: Commands,
+    mut images: ResMut<Assets<Image>>,
+) {
+    if request_query.is_empty() {
+        return;
+    }
+
+    let entity = request_query.iter().next().unwrap().1;
+    let automation = automation_query.iter_mut().next().unwrap();
+    commands.entity(entity).despawn();
+
+    println!("Seed hex matrix");
+
+    let resolution = automation.world.side / 4;
+
+    let generator = HexTable::new(resolution);
+    let mut table = Table::new(false, automation.world.side);
+    generator.calculate().iter().for_each(|(x, y)| {
+        let x = (x * 4.0) as usize;
+        let y = (y * 4.0) as usize;
+
+        if x + y * table.side >= table.data.len() {
+            return;
+        }
+
+        if *automation.world.get_dim(x, y) == Color::NONE {
+            return;
+        }
+
+        table.set_dim(x, y, true);
+    });
+
+    let data = table.get_image_data();
+    let image = Image::new(
+        Extent3d {
+            width: table.side as u32,
+            height: table.side as u32,
+            depth_or_array_layers: 1,
+        },
+        TextureDimension::D2,
+        data,
+        TextureFormat::bevy_default(),
+        RenderAssetUsages::default(),
+    );
+
+    let sprite = view_query.iter().next().unwrap();
+
+
+    images.remove(sprite.image.id());
+    images.insert(sprite.image.id(), image).unwrap();
+    commands.spawn((HexMatrixBuild{
+        points: table
+    },));
+}
+
+#[derive(Component)]
+pub struct HexMatrixRequest;
+
+#[derive(Component)]
+pub struct HexMatrixView;
+
+#[derive(Component)]
+pub struct HexMatrixBuild{
+    points: Table<bool>
+}