1 files changed, 15 insertions, 14 deletions

diff --git a/src/undo.rs b/src/undo.rs
index ae96a66..1044c63 100644
--- a/src/undo.rs
+++ b/src/undo.rs
@@ -1,19 +1,22 @@
-use crate::bitmap::MapPoint;
+use crate::{bitmap::MapPoint, brush::Brush};
 
-#[derive(Copy, Clone, Debug)]
-pub struct ModifyRecord {
+#[derive(Debug, Clone)]
+pub enum ModifyRecord {
+    Paint(Vec<PaintRecord>),
+    Invert,
+    Brush { old: Brush, new: Brush },
+}
+
+#[derive(Debug, Copy, Clone)]
+pub struct PaintRecord {
     pub point: MapPoint,
-    pub old_val: bool,
-    pub val: bool,
+    pub old: bool,
+    pub new: bool,
 }
 
-impl ModifyRecord {
-    pub fn new<P: Into<MapPoint>>(point: P, old_val: bool, val: bool) -> Self {
-        ModifyRecord {
-            point: point.into(),
-            old_val,
-            val,
-        }
+impl PaintRecord {
+    pub fn new(point: MapPoint, old: bool, new: bool) -> Self {
+        Self { point, old, new }
     }
 }
 
@@ -22,8 +25,6 @@ pub enum OpKind {
     Redo,
 }
 
-pub type Operation = Vec<ModifyRecord>;
-
 #[derive(Debug)]
 pub struct UndoStack<T> {
     operations: Vec<T>,