Crizomb/Galaxy-sim
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

performances + blackhole

Browse source 
added __slots__ to the Node class
+
added central black hole at the center of the galaxy in create_stars
This commit is contained in:
Crizomb 2023-02-14 12:57:26 +01:00 • committed by GitHub
parent 0902bab7dd
commit 27a62277a6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
barnes-hut-galaxy-new.py
View file

@ -48,6 +48,7 @@ class Star:
class Node: class Node:
""""Node class""" """"Node class"""
__slots__ = "center", "center_of_mass", "mass", "children", "width"
def __init__(self, center: Vec3, center_of_mass: Vec3, mass: float, children, width: float): def __init__(self, center: Vec3, center_of_mass: Vec3, mass: float, children, width: float):
self.center = center self.center = center
self.center_of_mass = center_of_mass self.center_of_mass = center_of_mass
@ -132,6 +133,10 @@ def create_stars(nb_stars: int) -> List[Star]:
orthog = Vec3(1, 0, -x / z) if z > 0 else Vec3(-1, 0, x / z) orthog = Vec3(1, 0, -x / z) if z > 0 else Vec3(-1, 0, x / z)
speed = orthog.normalize()*pos.norm speed = orthog.normalize()*pos.norm
stars.append(Star(pos, speed, 1)) stars.append(Star(pos, speed, 1))
#add center blackhole
stars.append(Star(Vec3(0, 0, 0), Vec3(0, 0, 0), nb_stars/10))
return stars return stars
def create_tree(stars: List[Star]) -> Node: def create_tree(stars: List[Star]) -> Node: