sus

gas_sim.py

@@ -1,30 +1,23 @@
 from math import hypot
 
 
-class Particle:
-    def __init__(self, pos, vel):
-        self.pos = pos
-        self.vel = vel
-
-    def update(self, dt):
-        self.pos = (self.pos[0] + self.vel[0] * dt, self.pos[1] + self.vel[1] * dt)
-
-
 class GasSim:
-    def __init__(self, particle_radius, particle_compressibility, substeps):
+    def __init__(self, particle_radius, particle_compressibility, particle_friction, substeps):
         self.particle_radius = particle_radius
         self.particle_compressibility = particle_compressibility
+        self.particle_friction = particle_friction
         self.substeps = substeps
 
         self.particles = []
+        self.walls = []
 
     def add_particle(self, pos, vel = (0, 0)):
-        self.particles.append(Particle(pos, vel))
-
-    def return_particles_pos(self):
-        return tuple(particle.pos for particle in self.particles)
+        self.particles.append([pos, vel])
 
     def cursor_particle_collide_test(self, cursor_pos, cursor_radius):
         for particle in self.particles:
-            if hypot(particle.pos[0] - cursor_pos[0], particle.pos[1] - cursor_pos[1]) <= cursor_radius + self.particle_radius:
-                return True
+            if hypot(particle[0][0] - cursor_pos[0], particle[0][1] - cursor_pos[1]) <= cursor_radius + self.particle_radius:
+                return True
+            
+    def add_wall(self, start, end):
+        self.walls.append((start, end))