nora
/
asteroids-genetic
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

play as human

main
sparshg 11 months ago
parent
0796eda239
commit
4fae04f5cd
5 changed files with 102 additions and 47 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      models/model.json
  2. 123
      src/main.rs
  3. 10
      src/player.rs
  4. 11
      src/population.rs
  5. 3
      src/world.rs

2
models/model.json
View File

@ -1 +1 @@
{"config":[6,7,7,4],"weights":[[[-0.5567123,0.23674688,-0.8115287,0.73920447,-0.9627226,0.98842543,-0.08672536,0.11889564,0.19367097,0.84720945,0.10916865,0.19691469,0.34381783,-1.1452736,-0.7501932,0.65345323,1.1373248,-0.2888124,0.288057,0.09440576,-0.061547264,0.35609803,0.14600815,0.4276784,-0.8315937,-0.5220405,-0.7070066,-0.41821334,0.88325745,1.179611,-0.38369912,0.8185778,0.1276589,0.11540664,-0.2861217,-0.88901085],6,6],[[0.64678603,0.11032932,-0.17325084,0.4482577,-0.4668608,-0.061782226,-0.23166645,-1.189321,0.38457078,-0.040191226,-1.0519018,0.53179306,-0.27904406,0.18169245,-0.028529042,0.4248598,0.6224678,-0.48197106,-0.8517104,0.626093,-0.53116995,0.74109083,-0.4775906,-0.6105986,-0.11459249,0.54738194,0.53385884,0.05052872,0.56253725,0.6168962,0.3087382,1.4871372,-0.058942188,-0.13154379,0.6763109,-0.72106224,-0.13756883,-0.10921177,-0.12797734,-0.12984648,0.47288033,0.06618643],6,7],[[0.24218078,0.1842654,0.27074534,0.50006706,0.9494213,-0.7030016,-0.1101043,0.5054117,0.18234585,-1.429309,-0.14925024,0.47471818,-1.079322,1.0112062,-0.5279594,-0.13144909,1.0667706,-0.85650563,0.076754875,1.3830073,-0.50329316,0.61952496,-0.94159657,-0.27969155,-0.6343474,0.08405438,1.0401202,-1.3237122],4,7]],"activ_func":"ReLU","mut_rate":0.05}
{"config":[6,7,7,4],"weights":[[[0.35300237,0.4247243,-0.7875665,-1.6072044,1.5638344,1.0116522,0.91539335,0.72748256,0.10979596,0.916985,1.1620126,-0.879529,-0.008039723,-1.4823594,-0.7815808,-0.9131131,0.25484133,-0.8343034,-0.97603,0.5233137,-0.8282123,-0.20260507,0.8145971,0.688537,1.0713704,0.8801996,2.5356312,0.14526872,0.2973215,-1.1754538,0.007596435,0.3534879,-0.29665652,1.1665709,-1.9667246,-1.3233892],6,6],[[-0.52504927,1.3216119,1.0895479,0.84873104,0.64963293,-2.0617585,-0.24353297,2.0812314,-0.06842113,-1.2582295,2.5253322,-0.04738267,-0.14587633,0.37131867,-0.18781468,-2.2502813,2.5478582,0.04243115,1.1420923,0.5919767,0.5798261,0.43632478,0.5544543,0.14799483,-1.3338035,-0.23948197,0.9541234,0.7938923,0.30592227,-0.044575863,-1.4038266,-1.282537,0.9133283,-0.09069362,-1.2081418,-0.05495484,1.632303,1.0032419,1.1147637,0.46290535,-0.035745874,-1.9028112],6,7],[[-0.16735676,-0.28051043,-1.0052958,0.33411616,2.1245909,-0.4129878,1.1603845,1.0717844,-0.6117815,-0.6603155,0.82145613,-0.6706493,1.7019055,-1.4384344,-1.4858634,-0.5874535,1.6686711,0.2978619,-0.46680036,-0.41379213,-1.1143796,0.6741852,-0.6568759,0.20075007,0.050768528,-0.39008367,0.6281415,2.8667881],4,7]],"activ_func":"ReLU","mut_rate":0.05}

123
src/main.rs
View File

@ -61,7 +61,9 @@ async fn main() {
let mut speedup = 1;
let mut paused = false;
let mut bias = false;
let mut human = false;
let mut size: u32 = 100;
let mut world: World = World::new(None, None, None);
let mut hlayers: Vec<usize> = vec![6, 6, 0];
let mut prev_hlayers = hlayers.clone();
@ -76,7 +78,7 @@ async fn main() {
ActivationFunc::Sigmoid,
ActivationFunc::Tanh,
];
let mut auto_switch = Some(AutoSwitch::Best);
let mut auto_switch = Some(AutoSwitch::BestAlive);
let mut pop = Population::new(
size as usize,
@ -241,10 +243,19 @@ async fn main() {
}
if !paused {
for _ in 0..speedup {
pop.update();
if !human {
pop.update()
} else if !world.over {
world.update()
};
}
}
pop.draw();
if human {
world.draw(pop.debug);
pop.draw_borders();
} else {
pop.draw();
}
draw_rectangle_lines(-WIDTH * 0.5, -HEIGHT * 0.5, WIDTH, HEIGHT, 2., WHITE);
draw_rectangle_lines(
WIDTH * 0.5 + th,
@ -270,7 +281,12 @@ async fn main() {
bias,
);
set_camera(&statcam);
pop.worlds[pop.track].draw_stats(
let w = if human {
&world
} else {
&pop.worlds[pop.track]
};
w.draw_stats(
screen_width() - WIDTH - 3. * th,
(screen_height() - 7. * th) * 0.5 - 2. * ui_thick,
pop.worlds.iter().fold(1, |acc, w| {
@ -387,54 +403,64 @@ async fn main() {
};
ui.same_line(0.);
if widgets::Button::new(restart).ui(ui) {
pop = Population::new(
size as usize,
auto_switch,
hlayers.clone(),
mut_rate,
activs[activ],
);
if human {
world = World::new(None, None, None);
} else {
pop = Population::new(
size as usize,
auto_switch,
hlayers.clone(),
mut_rate,
activs[activ],
);
}
};
});
ui.push_skin(&skin2);
widgets::Group::new(
hash!(),
vec2(ui_width * 0.2, ui_height * 0.8 - 2. * th - 2. * ui_thick),
vec2(ui_width * 0.2, ui_height * 0.85 - 2. * th - 2. * ui_thick),
)
.position(vec2(ui_width * 0.6, ui_height * 0.23 + ui_thick + th))
.position(vec2(ui_width * 0.82, ui_height * 0.15 + ui_thick + th))
.ui(ui, |ui| {
ui.label(None, "Track Ship:");
ui.label(None, "(or click a");
ui.label(None, "ship in game)");
if ui.button(None, "Best Alive") {
pop.track_best(false);
}
if ui.button(None, "Current #1") {
pop.track_best(true);
auto_switch = Some(AutoSwitch::Best);
pop.auto_switch = auto_switch;
}
if ui.button(None, "LastGen #1") {
pop.track_prev_best();
auto_switch = None;
pop.auto_switch = auto_switch;
}
ui.label(None, " ");
ui.label(None, "Auto Switch");
ui.label(None, "When Dead to:");
if auto_switch == Some(AutoSwitch::Best) {
if auto_switch == Some(AutoSwitch::BestAlive) {
ui.push_skin(&skin3);
ui.button(None, "Current #1");
ui.button(None, "Best Alive");
ui.pop_skin();
} else {
if ui.button(None, "Current #1") {
auto_switch = Some(AutoSwitch::Best);
if ui.button(None, "Best Alive") {
auto_switch = Some(AutoSwitch::BestAlive);
pop.auto_switch = auto_switch;
}
}
if auto_switch == Some(AutoSwitch::BestAlive) {
if auto_switch == Some(AutoSwitch::Best) {
ui.push_skin(&skin3);
ui.button(None, "Best Alive");
ui.button(None, "Current #1");
ui.pop_skin();
} else {
if ui.button(None, "Best Alive") {
auto_switch = Some(AutoSwitch::BestAlive);
if ui.button(None, "Current #1") {
auto_switch = Some(AutoSwitch::Best);
pop.auto_switch = auto_switch;
}
}
@ -451,10 +477,47 @@ async fn main() {
});
widgets::Group::new(
hash!(),
vec2(ui_width * 0.2, ui_height * 0.8 - 2. * th - 2. * ui_thick),
vec2(ui_width * 0.2, ui_height * 0.85 - 2. * th - 2. * ui_thick),
)
.position(vec2(ui_width * 0.8 - th, ui_height * 0.22 + ui_thick + th))
.position(vec2(ui_width * 0.6 - th, ui_height * 0.15 + ui_thick + th))
.ui(ui, |ui| {
ui.label(None, " ");
ui.push_skin(&skin);
if ui.button(
None,
if human {
" Train AI "
} else {
"Play As Human"
},
) {
human = !human;
if human {
world = World::new(None, None, None);
} else {
pop = Population::new(
size as usize,
auto_switch,
hlayers.clone(),
mut_rate,
activs[activ],
);
}
}
ui.pop_skin();
ui.label(None, "Mutation Rate");
ui.drag(hash!(), "«Drag»", Some((0., 1.)), &mut mut_rate);
if prev_mut_rate != mut_rate {
pop.change_mut(mut_rate);
prev_mut_rate = mut_rate;
}
ui.label(None, "Activation Func");
ui.combo_box(hash!(), "«Select»", &["ReLU", "Sigm", "Tanh"], &mut activ);
if prev_activ != activ {
pop.change_activ(activs[activ]);
prev_activ = activ;
}
ui.label(None, " ");
ui.label(None, "Hidden Layers");
ui.label(None, "Neurons Config");
@ -471,20 +534,6 @@ async fn main() {
);
prev_hlayers = hlayers.clone();
}
ui.label(None, " ");
ui.label(None, "Mutation Rate");
ui.drag(hash!(), "«Drag»", Some((0., 1.)), &mut mut_rate);
if prev_mut_rate != mut_rate {
pop.change_mut(mut_rate);
prev_mut_rate = mut_rate;
}
ui.label(None, " ");
ui.label(None, "Activation Func");
ui.combo_box(hash!(), "«Select»", &["ReLU", "Sigm", "Tanh"], &mut activ);
if prev_activ != activ {
pop.change_activ(activs[activ]);
prev_activ = activ;
}
});
ui.pop_skin();
},

10
src/player.rs
View File

@ -96,7 +96,7 @@ impl Player {
// .unwrap();
// }
// }
if asteroid.check_collision(self.pos, 8.) || self.lifespan > 4000 {
if asteroid.check_collision(self.pos, 8.) || self.lifespan > 4000 && self.brain.is_some() {
self.alive = false;
return true;
}
@ -157,21 +157,21 @@ impl Player {
.collect();
}
}
if keys[0] {
if keys[0] || self.brain.is_none() && is_key_down(KeyCode::Right) {
// RIGHT
self.rot = (self.rot + 0.1 + TAU as f32) % TAU as f32;
self.dir = vec2(self.rot.cos(), self.rot.sin());
}
if keys[1] {
if keys[1] || self.brain.is_none() && is_key_down(KeyCode::Left) {
// LEFT
self.rot = (self.rot - 0.1 + TAU as f32) % TAU as f32;
self.dir = vec2(self.rot.cos(), self.rot.sin());
}
if is_key_down(KeyCode::Up) || keys[2] {
if keys[2] || self.brain.is_none() && is_key_down(KeyCode::Up) {
// THROTTLE
self.acc = 0.14;
}
if is_key_down(KeyCode::Space) || keys[3] {
if keys[3] || self.brain.is_none() && is_key_down(KeyCode::Space) {
if self.last_shot > self.shot_interval {
self.last_shot = 0;
self.shots += 1;

11
src/population.rs
View File

@ -32,15 +32,18 @@ impl Population {
mut_rate: f32,
activ: ActivationFunc,
) -> Self {
Self {
let mut s = Self {
size,
hlayers: hlayers.clone(),
worlds: (0..size)
.map(|_| World::new(Some(hlayers.clone()), Some(mut_rate), Some(activ)))
.collect(),
auto_switch,
focus: true,
..Default::default()
}
};
s.worlds[0].track(true);
s
}
pub fn update(&mut self) {
@ -126,8 +129,10 @@ impl Population {
world.draw(self.debug);
}
}
self.draw_borders();
}
// draw black background outside the screen
pub fn draw_borders(&self) {
let th = (screen_height() - HEIGHT) * 0.5;
draw_rectangle(-WIDTH * 0.5, -screen_height() * 0.5, WIDTH, th, BLACK);
draw_rectangle(-WIDTH * 0.5, screen_height() * 0.5 - th, WIDTH, th, BLACK);

3
src/world.rs
View File

@ -23,6 +23,7 @@ impl World {
activ: Option<ActivationFunc>,
) -> Self {
Self {
color: Color::new(1., 1., 1., if hlayers.is_none() { 0.8 } else { 0.4 }),
player: Player::new(hlayers, mut_rate, activ),
score: 1.,
asteroids: vec![
@ -32,13 +33,13 @@ impl World {
Asteroid::new(AsteroidSize::Large),
Asteroid::new(AsteroidSize::Large),
],
color: Color::new(1., 1., 1., 0.4),
..Default::default()
}
}
pub fn simulate(brain: NN) -> Self {
let mut w = World::new(None, None, None);
w.player.brain = Some(brain);
w.color = Color::new(1., 1., 1., 0.4);
w
}

Write Preview
Loading…
Cancel
Save