Particle-Simulator/FundamentalForces.py at master · Learning-Python-Team/Particle-Simulator · 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
# Equations used in ParticleSimulator_2D.py
from Settings import *
import math as m


def calculate_gravity(a_pos, a_type, b_pos, b_type):
	x_diff = b_pos[0] - a_pos[0]
	y_diff = b_pos[1] - a_pos[1]
	hypotenuse = m.sqrt((x_diff ** 2) + (y_diff ** 2))
	sin = x_diff / hypotenuse
	cos = y_diff / hypotenuse
	a_mass = particle_mass[a_type]
	b_mass = particle_mass[b_type]

	f = (constant_G * a_mass * b_mass) / (hypotenuse ** 2)

	fx = f * sin
	fy = f * cos

	return fx, fy


def calculate_electromagnetic(a_pos, a_type, b_pos, b_type):
	if a_type or b_type == 'n':
		return [0, 0]

	x_diff = b_pos[0] - a_pos[0]
	y_diff = b_pos[1] - a_pos[1]
	hypotenuse = m.sqrt((x_diff ** 2) + (y_diff ** 2))
	sin = x_diff / hypotenuse
	cos = y_diff / hypotenuse

	a_mass = particle_mass[a_type]
	b_mass = particle_mass[b_type]

	f = (constant_coulombs_constant * a_mass * b_mass) / (hypotenuse ** 2)

	if a_type == b_type:
		fx = (f * sin) * -1
		fy = (f * cos) * -1
	else:
		fx = (f * sin)
		fy = (f * cos)
	return fx, fy

# def calculate_strong(forces_list, )