1 files changed, 39 insertions, 0 deletions
diff --git a/src/sim/rb_manager/rb_entity.rs b/src/sim/rb_manager/rb_entity.rs
new file mode 100644
index 0000000..18a6693
--- /dev/null
+++ b/src/sim/rb_manager/rb_entity.rs
@@ -0,0 +1,39 @@
+use rapier2d::prelude;
+
+use crate::sim::{
+    cell::{cell::Cell, materials::MaterialId},
+    lib::marching_squares::Marchable,
+};
+
+pub struct RbEntity {
+    pub id: u32,
+    pub width: i32,
+    pub height: i32,
+    pub cells: Vec<Cell>,
+    pub rb: prelude::RigidBodyHandle,
+    pub collider: Option<prelude::ColliderHandle>,
+}
+
+impl RbEntity {
+    #[inline]
+    pub fn get_cell_at_local_position(&self, x: u8, y: u8) -> Cell {
+        self.cells[x as usize + y as usize * self.width as usize]
+    }
+    #[inline]
+    pub fn set_cell_at_local_position(&mut self, x: u8, y: u8, cell: Cell) {
+        self.cells[x as usize + y as usize * self.width as usize] = cell;
+    }
+}
+
+impl Marchable for RbEntity {
+    fn occupied(&self, x: i32, y: i32) -> bool {
+        if x < 0 || x >= self.width || y < 0 || y >= self.height {
+            false
+        } else {
+            self.get_cell_at_local_position(x as u8, y as u8).material != MaterialId::Void
+        }
+    }
+    fn size(&self) -> (i32, i32) {
+        (self.width, self.height)
+    }
+}