forked from Ansh-U/ThomsonProblem
-
Notifications
You must be signed in to change notification settings - Fork 0
/
particle.py
33 lines (26 loc) · 1.03 KB
/
particle.py
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
import numpy as np
import math
class Particle:
def __init__(self, dt, mass=1, charge=1, force=[np.zeros(3)]):
self.mass = mass
self.charge = charge
self.force = force # net force on the particle
self.dt = dt
self.pos = [np.zeros(3)]
self.vel = [np.zeros(3)]
self.accel = [np.zeros(3)]
def velocity_constraint(self, vel, radius):
vel = 0.997*(vel - np.dot(vel, radius)*radius/np.linalg.norm(radius))
return vel
def position_constraint(self, pos):
pos = 1 * pos / (np.linalg.norm(pos))
return pos
def update(self):
self.accel = np.vstack((self.accel, self.force[-1] / self.mass))
radius = self.pos[-1]
new_vel = self.vel[-1] + self.accel[-1]*self.dt
new_vel = self.velocity_constraint(new_vel, radius)
self.vel = np.vstack((self.vel, new_vel))
new_pos = self.pos[-1] + (self.vel[-1] * self.dt)
new_pos = self.position_constraint(new_pos)
self.pos = np.vstack((self.pos, new_pos))