Browse Source

Somewhat structure main.rs

main
Adrian Heine 3 years ago
parent
commit
2f06f9b0b7
  1. 86
      src/main.rs
  2. 3
      src/view.rs

86
src/main.rs

@ -1,5 +1,5 @@
use gntag::agent::{Agent, SimpleAgent}; use gntag::agent::{Agent, SimpleAgent};
use gntag::view::TerminalView;
use gntag::view::{Backend, TerminalView};
use gntag::world::ActualWorld; use gntag::world::ActualWorld;
use std::io; use std::io;
use std::io::Read; use std::io::Read;
@ -9,6 +9,8 @@ use std::thread;
fn main() { fn main() {
let view = Arc::new(Mutex::new(Some(TerminalView::try_new().unwrap()))); let view = Arc::new(Mutex::new(Some(TerminalView::try_new().unwrap())));
// Exit on q, ESC and Ctrl-C and reset terminal
let view2 = view.clone(); let view2 = view.clone();
thread::spawn(move || { thread::spawn(move || {
let stdin = io::stdin(); let stdin = io::stdin();
@ -22,48 +24,52 @@ fn main() {
} }
} }
}); });
loop { loop {
let (width, height) = (*view.lock().unwrap())
run_simulation(&view);
}
}
fn run_simulation(view: &Arc<Mutex<Option<TerminalView<impl Backend>>>>) {
let (width, height) = (*view.lock().unwrap())
.as_mut()
.unwrap()
.content_size()
.unwrap();
let mut agents: Vec<(_, Box<dyn Agent>)> = vec![];
for x in (0..width).step_by(10) {
for y in (0..height).step_by(10) {
agents.push(((x, y).into(), Box::new(SimpleAgent)));
}
}
let mut world = ActualWorld::new((width, height).into(), agents);
let mut gen = 0;
loop {
let resized = (*view.lock().unwrap())
.as_mut() .as_mut()
.unwrap() .unwrap()
.content_size()
.draw(
gen,
world
.state
.agent_positions
.get(&world.state.tagged)
.map(|pos| (pos.x, pos.y))
.unwrap(),
world
.state
.agent_positions
.iter()
.map(|(_id, pos)| (pos.x, pos.y))
.collect::<Vec<_>>()
.as_ref(),
)
.unwrap(); .unwrap();
let mut agents: Vec<(_, Box<dyn Agent>)> = vec![];
for x in (0..width).step_by(10) {
for y in (0..height).step_by(10) {
agents.push(((x, y).into(), Box::new(SimpleAgent)));
}
}
let mut world = ActualWorld::new((width, height).into(), agents);
let mut gen = 0;
loop {
let resized = (*view.lock().unwrap())
.as_mut()
.unwrap()
.draw(
gen,
world
.state
.agent_positions
.get(&world.state.tagged)
.map(|pos| (pos.x, pos.y))
.unwrap(),
world
.state
.agent_positions
.iter()
.map(|(_id, pos)| (pos.x, pos.y))
.collect::<Vec<_>>()
.as_ref(),
)
.unwrap();
if resized {
break;
};
world.do_step();
gen += 1;
//std::thread::sleep(std::time::Duration::from_millis(500));
}
if resized {
return;
};
world.do_step();
gen += 1;
} }
} }

3
src/view.rs

@ -3,7 +3,8 @@ use std::error::Error;
use std::io::{stdout, Stdout, Write}; use std::io::{stdout, Stdout, Write};
use termion::clear; use termion::clear;
use termion::raw::{IntoRawMode, RawTerminal}; use termion::raw::{IntoRawMode, RawTerminal};
use tui::backend::{Backend, TermionBackend};
pub use tui::backend::Backend;
use tui::backend::TermionBackend;
use tui::layout::{Alignment, Constraint, Direction, Layout, Rect}; use tui::layout::{Alignment, Constraint, Direction, Layout, Rect};
use tui::style::{Color, Modifier, Style}; use tui::style::{Color, Modifier, Style};
use tui::text::{Span, Spans}; use tui::text::{Span, Spans};

Loading…
Cancel
Save