noita_entangled_worlds/blob_guy/src/blob_guy.rs

use crate::chunk::{CellType, Chunk, ChunkPos};
use crate::{CHUNK_SIZE, State};
#[derive(Default, Debug)]
pub struct Pos {
    x: f64,
    y: f64,
}
impl Pos {
    pub fn new(x: f64, y: f64) -> Self {
        Self { x, y }
    }
    pub fn to_chunk(&self) -> ChunkPos {
        ChunkPos::new(self.x as i32, self.y as i32)
    }
    pub fn to_chunk_inner(&self) -> usize {
        self.y.rem_euclid(CHUNK_SIZE as f64) as usize * CHUNK_SIZE
            + self.x.rem_euclid(CHUNK_SIZE as f64) as usize
    }
}
impl State {
    pub fn update(&mut self) -> eyre::Result<()> {
        if self.blobs.is_empty() {
            self.blobs.push(Blob::new(16.0, -128.0))
        }
        if let Some(pws) = self.particle_world_state.as_mut() {
            'upper: for blob in self.blobs.iter_mut() {
                let c = blob.pos.to_chunk();
                let mut k = 0;
                for x in -1..=1 {
                    for y in -1..=1 {
                        if unsafe { !pws.encode_area(c.x + x, c.y + y, &mut self.world[k]) } {
                            continue 'upper;
                        }
                        k += 1;
                    }
                }
                blob.update(&mut self.world);
                let mut k = 0;
                for x in -1..=1 {
                    for y in -1..=1 {
                        if unsafe { !pws.decode_area(c.x + x, c.y + y, &self.world[k]) } {
                            continue 'upper;
                        }
                        k += 1;
                    }
                }
            }
        }
        Ok(())
    }
}
const SIZE: usize = 9;
pub struct Blob {
    pub pos: Pos,
    pixels: [Pos; SIZE * SIZE],
}
impl Blob {
    pub fn update(&mut self, map: &mut [Chunk; 9]) {
        let mut last = ChunkPos::new(i32::MAX, i32::MAX);
        let mut k = 0;
        let start = self.pos.to_chunk();
        for p in &self.pixels {
            let c = p.to_chunk();
            if c != last {
                k = ((c.x - start.x + 1) * 3 + c.y - start.y + 1) as usize;
                last = c;
            }
            let n = p.to_chunk_inner();
            map[k][n] = CellType::Blob
        }
    }
    pub fn new(x: f64, y: f64) -> Self {
        Blob {
            pos: Pos::new(x, y),
            pixels: std::array::from_fn(|i| {
                let a = (i / SIZE) as f64 - SIZE as f64 / 2.0;
                let b = (i % SIZE) as f64 - SIZE as f64 / 2.0;
                Pos::new(x + a, y + b)
            }),
        }
    }
}