1 files changed, 16 insertions, 3 deletions

diff --git a/src/brush.rs b/src/brush.rs
index 8557ba7..8d3ee1c 100644
--- a/src/brush.rs
+++ b/src/brush.rs
@@ -25,8 +25,9 @@ pub struct CircleBrush {
 
 #[derive(Debug, Copy, Clone)]
 pub struct RectSelectBrush {
-    pub start: MapPoint,
-    pub end: MapPoint,
+    pub start: Option<MapPoint>,
+    pub end: Option<MapPoint>,
+    pub active_end: bool,
 }
 
 impl Brush {
@@ -60,8 +61,20 @@ impl Brush {
         })
     }
 
+    pub fn rect() -> Self {
+        Brush::RectSelect(RectSelectBrush {
+            start: None,
+            end: None,
+            active_end: true,
+        })
+    }
+
     pub fn is_line(&self) -> bool {
-        matches!(self, Self::Line { .. })
+        matches!(self, Self::Line(_))
+    }
+
+    pub fn is_rect(&self) -> bool {
+        matches!(self, Self::RectSelect(_))
     }
 
     pub fn size(&self) -> Option<u8> {