Selection modes

src/main.rs

@@ -18,7 +18,6 @@ use std::io::{ Write, Read, Cursor };
 use std::fs::File;
 
 
-
 //////////////////// Data Layout ////////////////////
 
 #[derive(Serialize, Deserialize, Debug)]
@@ -39,11 +38,11 @@ impl Board {
 	Board { nodes }
     }
 
-    pub fn add_node(&mut self, x: f32, y: f32) {
+    pub fn add_node(&mut self, x: f32, y: f32, name: String) {
 	self.nodes.insert(self.next_id(), Node {
 	    x,
 	    y,
-	    name: "Canada".to_string(),
+	    name,
 	    edges: HashSet::new(),
 	});
     }
@@ -134,29 +133,35 @@ impl CoordTransformer for frame::Frame {
     }
 }
 
-fn menu_cb(_m: &mut impl MenuExt) {
-    todo!();
-}
-
 //////////////////// App State ////////////////////
 
+enum EditMode {
+    Node,
+    Edge,
+}
 struct AppState {
     pub board: Board,
     pub image_raw: Option<DynamicImage>,
     pub image: Option<PngImage>,
+    pub selected_node: Option<usize>,
+    pub edit_mode: EditMode,
 }
-
 impl AppState {
     fn new() -> Self {
 	AppState {
 	    board: Board::new(),
 	    image_raw: None,
-	    image: None
+	    image: None,
+	    selected_node: None,
+	    edit_mode: EditMode::Node
 	}
     }
 }
 
+//////////////////// Procedural App Creation ////////////////////
+
 fn main() {
+    // App setup
     let app = app::App::default()
 	.with_scheme(app::Scheme::Gtk);
     let mut win = window::Window::default()
@@ -172,12 +177,12 @@ fn main() {
     // Menu
     let mut menubar = menu::MenuBar::default();
     let state_clone = Rc::clone(&state);
-    menubar.add("File/New"		, Shortcut::None, menu::MenuFlag::Normal, move |_| {
+    menubar.add("File/New", Shortcut::None, menu::MenuFlag::Normal, move |_| {
 	state_clone.replace(AppState::new());
 	app::redraw();
     });
     let state_clone = Rc::clone(&state);
-    menubar.add("File/Open..."		, Shortcut::None, menu::MenuFlag::Normal, move |_| {
+    menubar.add("File/Open...", Shortcut::None, menu::MenuFlag::Normal, move |_| {
 	let mut fc = dialog::NativeFileChooser::new(dialog::NativeFileChooserType::BrowseFile);
 	fc.show();
 
@@ -199,7 +204,7 @@ fn main() {
 	state_clone.replace(state);
     });
     let state_clone = Rc::clone(&state);
-    menubar.add("File/Open Image..."	, Shortcut::None, menu::MenuFlag::Normal, move |_| {
+    menubar.add("File/Open Image...", Shortcut::None, menu::MenuFlag::Normal, move |_| {
 	let mut fc = dialog::NativeFileChooser::new(dialog::NativeFileChooserType::BrowseFile);
 	fc.show();
 	let filename = fc.filename();
@@ -216,7 +221,7 @@ fn main() {
 	}
     });
     let state_clone = Rc::clone(&state);
-    menubar.add("File/Save As ..."	, Shortcut::None, menu::MenuFlag::Normal, move |_| {
+    menubar.add("File/Save As ...", Shortcut::None, menu::MenuFlag::Normal, move |_| {
 	let mut fc = dialog::NativeFileChooser::new(dialog::NativeFileChooserType::BrowseSaveFile);
 	fc.show();
 
@@ -235,8 +240,17 @@ fn main() {
 	}
 	ar.finish().ok();
     });
-    menubar.add("Edit/Edit Nodes"	, Shortcut::None, menu::MenuFlag::Normal, menu_cb);
-    menubar.add("Edit/Edit Edges"	, Shortcut::None, menu::MenuFlag::Normal, menu_cb);
+    let state_clone = Rc::clone(&state);
+    menubar.add("Edit/Edit Nodes", Shortcut::None, menu::MenuFlag::Normal, move |_| {
+	let mut state = state_clone.borrow_mut();
+	state.edit_mode = EditMode::Node;
+    });
+    let state_clone = Rc::clone(&state);
+    menubar.add("Edit/Edit Edges", Shortcut::None, menu::MenuFlag::Normal, move |_| {
+	let mut state = state_clone.borrow_mut();
+	state.edit_mode = EditMode::Edge;
+    });
+    
     flex.set_size(&menubar, 40);
 
     // Canvas
@@ -262,8 +276,14 @@ fn main() {
 	    Event::Push => {
 		let mut state = state_clone.borrow_mut();
 		let (pos_x, pos_y) = f.to_coords(app::event_x(), app::event_y());
+
+		match state.edit_mode {
+		    EditMode::Node => {
 			if app::event_button() == 1 {
-		    state.board.add_node(pos_x, pos_y);
+			    let name = dialog::input_default("Node", "");
+			    if let Some(name) = name {
+				state.board.add_node(pos_x, pos_y, name);
+			    }
 			} else if app::event_button() == 3 {
 			    let id = state.board.nearest_node(pos_x, pos_y);
 			    if let Some(id) = id {
@@ -271,6 +291,11 @@ fn main() {
 			    }
 			}
 			app::redraw();
+		    },
+		    EditMode::Edge => {
+			todo!();
+		    }
+		}
 		true
 	    }
 	    _ => false,
@@ -278,10 +303,8 @@ fn main() {
     });
 
     flex.end();
-
     win.end();
     win.make_resizable(true);
     win.show();
-
     app.run().unwrap();
 }