Reformated gravity.cc

This commit is contained in:
Karma Riuk 2023-05-17 13:25:36 +02:00
parent 487b5fe00b
commit 7df64d460c

View File

@ -1,10 +1,11 @@
#include <string> #include "gravity.h"
#include <sstream>
#include "game.h"
#include <cmath> #include <cmath>
#include <gtk/gtk.h> #include <gtk/gtk.h>
#include <sstream>
#include "gravity.h" #include <string>
#include "game.h"
static const double constant_field_increment = 1.0; static const double constant_field_increment = 1.0;
static const double newton_field_increment = 1000.0; static const double newton_field_increment = 1000.0;
@ -15,99 +16,102 @@ static double g_g = 10000000;
static int constant_field = 1; static int constant_field = 1;
void gravity_constant_field (double x, double y) { void gravity_constant_field(double x, double y) {
constant_field = 1; constant_field = 1;
g.x = x; g.x = x;
g.y = y; g.y = y;
} }
void gravity_newton_field (double r, double g) { void gravity_newton_field(double r, double g) {
constant_field = 0; constant_field = 0;
g_r = r; g_r = r;
g_g = g; g_g = g;
} }
void gravity_draw (cairo_t * cr) { void gravity_draw(cairo_t* cr) {
if (constant_field) { if (constant_field) {
cairo_save(cr); cairo_save(cr);
cairo_new_path(cr); cairo_new_path(cr);
cairo_move_to(cr, width/2, height/2); cairo_move_to(cr, width / 2, height / 2);
cairo_line_to(cr, width/2 + g.x, height/2 + g.y); cairo_line_to(cr, width / 2 + g.x, height / 2 + g.y);
cairo_set_source_rgb(cr, 1.0, 1.0, 1.0); cairo_set_source_rgb(cr, 1.0, 1.0, 1.0);
cairo_set_line_width(cr, 1.0); cairo_set_line_width(cr, 1.0);
cairo_stroke(cr); cairo_stroke(cr);
cairo_arc(cr, width/2 + g.x, height/2 + g.y, 3, 0, 2*M_PI); cairo_arc(cr, width / 2 + g.x, height / 2 + g.y, 3, 0, 2 * M_PI);
cairo_fill(cr); cairo_fill(cr);
cairo_restore(cr); cairo_restore(cr);
} else { } else {
std::ostringstream output; std::ostringstream output;
output << (g_g/1000) << "K"; output << (g_g / 1000) << "K";
cairo_save (cr); cairo_save(cr);
cairo_set_source_rgb (cr, 1.0, 1.0, 1.0); cairo_set_source_rgb(cr, 1.0, 1.0, 1.0);
cairo_select_font_face (cr, "sans-serif", CAIRO_FONT_SLANT_NORMAL, CAIRO_FONT_WEIGHT_NORMAL); cairo_select_font_face(cr,
cairo_set_font_size (cr, 16); "sans-serif",
CAIRO_FONT_SLANT_NORMAL,
CAIRO_FONT_WEIGHT_NORMAL);
cairo_set_font_size(cr, 16);
std::string output_str = output.str(); std::string output_str = output.str();
cairo_text_extents_t extent; cairo_text_extents_t extent;
cairo_text_extents (cr, output_str.c_str(), &extent); cairo_text_extents(cr, output_str.c_str(), &extent);
cairo_move_to (cr, width/2 - extent.width/2, height/2 + extent.height/2); cairo_move_to(
cairo_show_text (cr, output_str.c_str()); cr, width / 2 - extent.width / 2, height / 2 + extent.height / 2);
cairo_restore (cr); cairo_show_text(cr, output_str.c_str());
cairo_restore(cr);
} }
} }
void gravity_draw_visible_field (cairo_t * cr) { void gravity_draw_visible_field(cairo_t* cr) {
if (!constant_field) { if (!constant_field) {
cairo_save (cr); cairo_save(cr);
cairo_set_source_rgb(cr, 1.0, 1.0, 1.0); cairo_set_source_rgb(cr, 1.0, 1.0, 1.0);
cairo_new_path (cr); cairo_new_path(cr);
cairo_arc (cr, width/2, height/2, g_r, 0, 2*M_PI); cairo_arc(cr, width / 2, height / 2, g_r, 0, 2 * M_PI);
cairo_set_line_width(cr, 3.0); cairo_set_line_width(cr, 3.0);
cairo_stroke(cr); cairo_stroke(cr);
cairo_restore(cr); cairo_restore(cr);
} }
} }
void gravity_change (double dx, double dy) { void gravity_change(double dx, double dy) {
if (constant_field) { if (constant_field) {
g.x += dx*constant_field_increment; g.x += dx * constant_field_increment;
g.y += dy*constant_field_increment; g.y += dy * constant_field_increment;
} else { } else {
g_r += dx; g_r += dx;
g_g -= dy*newton_field_increment; g_g -= dy * newton_field_increment;
} }
} }
vec2d gravity_vector (const ball * b) { vec2d gravity_vector(const ball* b) {
if (constant_field) { if (constant_field) {
return g; return g;
} else { } else {
vec2d b_c = vec2d{width/2.0,height/2.0} - b->position; vec2d b_c = vec2d{width / 2.0, height / 2.0} - b->position;
double r2 = vec2d::dot(b_c,b_c); double r2 = vec2d::dot(b_c, b_c);
if (r2 < g_r*g_r) { if (r2 < g_r * g_r)
return vec2d{0,0}; return vec2d{0, 0};
} else { else
return g_g/r2/sqrt(r2)*b_c; return g_g / r2 / sqrt(r2) * b_c;
}
} }
} }
void gravity_collisions (ball * begin, ball * end) { void gravity_collisions(ball* begin, ball* end) {
if (constant_field) if (constant_field)
return; return;
for (ball * b = begin; b != end; ++b) { for (ball* b = begin; b != end; ++b) {
vec2d b_c = b->position - vec2d{width/2.0,height/2.0}; vec2d b_c = b->position - vec2d{width / 2.0, height / 2.0};
double d2 = vec2d::dot(b_c, b_c); double d2 = vec2d::dot(b_c, b_c);
double r = b->radius + g_r; double r = b->radius + g_r;
if (d2 <= r*r) { if (d2 <= r * r) {
double d = sqrt(d2); double d = sqrt(d2);
if (d <= b->radius) if (d <= b->radius)
b->position = vec2d{width/2.0,height/2.0+r}; b->position = vec2d{width / 2.0, height / 2.0 + r};
else else
b->position += (r - d)/d*b_c; b->position += (r - d) / d * b_c;
double f = vec2d::dot(b->velocity, b_c); double f = vec2d::dot(b->velocity, b_c);
if (f < 0) { if (f < 0) {
f /= d2; f /= d2;
b->velocity -= 2*f*b_c; b->velocity -= 2 * f * b_c;
} }
} }
} }