char
/
sketch-gravity
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Trans

main
Charlotte Som 2021-08-28 05:58:04 +01:00
parent 6bb5d90d54
commit 344a979e92
1 changed files with 26 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/main.rs
View File

@ -16,6 +16,7 @@ struct Body {
pos: DVec2, pos: DVec2,
vel: DVec2, vel: DVec2,
mass: i64, mass: i64,
color: Rgb,
} }
struct Model { struct Model {
@ -25,12 +26,13 @@ struct Model {
fn model(_app: &App) -> Model { fn model(_app: &App) -> Model {
let mut bodies = Vec::new(); let mut bodies = Vec::new();
const MASS_SUN: i64 = 150000000000000000; const MASS_SUN: i64 = 50000000000000000;
bodies.push(Body { bodies.push(Body {
pos: (0.0, 0.0).into(), pos: (0.0, 0.0).into(),
vel: (0.0, 0.0).into(), vel: (0.0, 0.0).into(),
mass: MASS_SUN, mass: MASS_SUN,
color: rgb(1.0, 1.0, 1.0),
}); });
const G: f64 = 6.67408e-11; const G: f64 = 6.67408e-11;
@ -39,18 +41,28 @@ fn model(_app: &App) -> Model {
let d_theta: f64 = (360.0 / COPIES as f64).to_radians(); let d_theta: f64 = (360.0 / COPIES as f64).to_radians();
for i in 0..COPIES { let palette = &[
let theta = i as f64 * d_theta; rgb(85.0 / 255.0, 205.0 / 255.0, 252.0 / 255.0),
rgb(1.0, 1.0, 1.0),
rgb(247.0 / 255.0, 168.0 / 255.0, 184.0 / 255.0),
];
for i in 10..250 { for i in 0..COPIES {
let r = 5.0 * i as f64; let mut theta = i as f64 * d_theta;
let m = 10000000 / i;
for i in 1..=256 {
let r = 25.0 + 5.0 * i as f64;
let m = 50000000000u64 / i;
let curr_color = palette[i as usize % palette.len()];
bodies.push(Body { bodies.push(Body {
pos: (r * theta.cos(), r * theta.sin()).into(), pos: (r * theta.cos(), r * theta.sin()).into(),
vel: (0.0, (G * MASS_SUN as f64 / r).sqrt()).into(), vel: (0.0, (G * MASS_SUN as f64 / r).sqrt()).into(),
mass: m, mass: m as i64,
color: curr_color,
}); });
theta += 1.94161103873;
} }
} }
@ -78,7 +90,7 @@ fn update(app: &App, model: &mut Model, _update: Update) {
let delta = other_body.pos - body.pos; let delta = other_body.pos - body.pos;
let dist_sq = body.pos.distance_squared(other_body.pos); let dist_sq = body.pos.distance_squared(other_body.pos);
if dist_sq < 1.0 { if dist_sq <= (body.mass as f64).log10() {
return; return;
} }
@ -95,12 +107,15 @@ fn update(app: &App, model: &mut Model, _update: Update) {
fn view(app: &App, model: &Model, frame: Frame) { fn view(app: &App, model: &Model, frame: Frame) {
// let t = app.duration.since_start.as_secs_f64(); // let t = app.duration.since_start.as_secs_f64();
let draw = app.draw().scale(2.0); let draw = app.draw().scale(1.0);
draw.background().color(BLACK); draw.background().color(BLACK);
for body in model.bodies.iter() { for body in model.bodies.iter() {
let diameter = (body.mass as f32).log10(); // should be proportional to cuberoot but this is so we can actually see the small ones let diameter = (body.mass as f32).log10() * 2.0; // should be proportional to cuberoot but this is so we can actually see the small ones
draw.ellipse().w_h(diameter, diameter).xy(body.pos.as_f32()); draw.ellipse()
.w_h(diameter, diameter)
.xy(body.pos.as_f32())
.color(body.color);
} }
draw.to_frame(app, &frame).unwrap(); draw.to_frame(app, &frame).unwrap();