simulations/material.cpp at master · bodnartibi/simulations · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
#include "material.h"

Material::Material(double mass,
                   double pos_x, double pos_y,
                   double velocity_x, double velocity_y,
                   double accelerate_x, double accelerate_y,
                   int border_x_min, int border_x_max, int border_y_min, int border_y_max,
                   double radius)
{
  this->mass = mass;

  this->pos_x = pos_x;
  this->pos_y = pos_y;

  this->velocity_x = velocity_x;
  this->velocity_y = velocity_y;

  this->accelerate_x = accelerate_x;
  this->accelerate_y = accelerate_y;

  this->border_x_min = border_x_min;
  this->border_x_max = border_x_max;
  this->border_y_min = border_y_min;
  this->border_y_max = border_y_max;

  this->radius = radius;
}

void Material::setAccelerate(double accelerate_x, double accelerate_y)
{
  this->accelerate_x = accelerate_x;
  this->accelerate_y = accelerate_y;
}

void Material::setPosition(double pos_x, double pos_y)
{
  this->pos_x = pos_x;
  this->pos_y = pos_y;
}

void Material::elasticCollision(Material *other)
{
  double n_v_x, n_v_y;

  n_v_x = ((this->mass - other->mass)/(this->mass + other->mass)) * this->velocity_x + ((2 * other->mass)/(this->mass + other->mass)) * other->velocity_x;
  n_v_y = ((this->mass - other->mass)/(this->mass + other->mass)) * this->velocity_y + ((2 * other->mass)/(this->mass + other->mass)) * other->velocity_y;

  other->velocity_x = ((other->mass - this->mass)/(other->mass + this->mass)) * other->velocity_x + ((2 * this->mass)/(other->mass + this->mass)) * this->velocity_x;
  other->velocity_y = ((other->mass - this->mass)/(other->mass + this->mass)) * other->velocity_y + ((2 * this->mass)/(other->mass + this->mass)) * this->velocity_y;

  this->velocity_x = n_v_x;
  this->velocity_y = n_v_y;
}

void Material::reboundFromBorder()
{
  if (pos_x < border_x_min)
  {
    pos_x = border_x_min + (border_x_min - pos_x);
    velocity_x = -velocity_x;
  }
  else if (pos_x > border_x_max)
  {
    pos_x = border_x_max - (pos_x - border_x_max);
    velocity_x = -velocity_x;
  }


  if (pos_y < border_y_min)
  {
    pos_y = border_y_min + (border_y_min - pos_y);
    velocity_y = -velocity_y;
  }
  else if (pos_y > border_y_max)
  {
    pos_y = border_y_max - (pos_y - border_y_max);
    velocity_y = -velocity_y;
  }
}

void Material::move(const double step)
{
  pos_x += step * velocity_x;
  pos_y += step * velocity_y;

  this->reboundFromBorder();

  velocity_x += step * accelerate_x;
  velocity_y += step * accelerate_y;
}

void Material::getPosition(double &pos_x, double &pos_y)
{
  pos_x = this->pos_x;
  pos_y = this->pos_y;
}

void Material::getRadius(double &r)
{
  r = this->radius;
}