From 4fae04f5cd934a63fda0c46f2168de2882603ada Mon Sep 17 00:00:00 2001 From: sparshg <43041139+sparshg@users.noreply.github.com> Date: Fri, 13 Jan 2023 01:32:58 +0530 Subject: [PATCH] play as human --- models/model.json | 2 +- src/main.rs | 127 ++++++++++++++++++++++++++++++++-------------- src/player.rs | 10 ++-- src/population.rs | 11 ++-- src/world.rs | 3 +- 5 files changed, 104 insertions(+), 49 deletions(-) diff --git a/models/model.json b/models/model.json index 8539cbd..033c971 100644 --- a/models/model.json +++ b/models/model.json @@ -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} \ No newline at end of file +{"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} \ No newline at end of file diff --git a/src/main.rs b/src/main.rs index 4150f9f..5910300 100644 --- a/src/main.rs +++ b/src/main.rs @@ -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 = 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,47 +403,47 @@ 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) { - ui.push_skin(&skin3); - ui.button(None, "Current #1"); - ui.pop_skin(); - } else { - if ui.button(None, "Current #1") { - auto_switch = Some(AutoSwitch::Best); - pop.auto_switch = auto_switch; - } - } if auto_switch == Some(AutoSwitch::BestAlive) { ui.push_skin(&skin3); ui.button(None, "Best Alive"); @@ -438,6 +454,16 @@ async fn main() { pop.auto_switch = auto_switch; } } + if auto_switch == Some(AutoSwitch::Best) { + ui.push_skin(&skin3); + ui.button(None, "Current #1"); + ui.pop_skin(); + } else { + if ui.button(None, "Current #1") { + auto_switch = Some(AutoSwitch::Best); + pop.auto_switch = auto_switch; + } + } if auto_switch.is_none() { ui.push_skin(&skin3); ui.button(None, "Do Nothing"); @@ -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(); }, diff --git a/src/player.rs b/src/player.rs index 9c17709..6458aab 100644 --- a/src/player.rs +++ b/src/player.rs @@ -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; diff --git a/src/population.rs b/src/population.rs index ce5ae2a..ff4baa2 100644 --- a/src/population.rs +++ b/src/population.rs @@ -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); diff --git a/src/world.rs b/src/world.rs index 9105271..6b9fc2f 100644 --- a/src/world.rs +++ b/src/world.rs @@ -23,6 +23,7 @@ impl World { activ: Option, ) -> 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 }