Particles seem to work with compressible

Author: harpolea

compressible/_defaults

@@ -21,3 +21,6 @@ grav = 0.0                ; gravitational acceleration (in y-direction)
 
 riemann = HLLC            ; HLLC or CGF
 
+[particles]
+do_particles = 1
+particle_generator = grid

compressible/simulation.py

@@ -12,6 +12,7 @@
 import mesh.boundary as bnd
 from simulation_null import NullSimulation, grid_setup, bc_setup
 import util.plot_tools as plot_tools
+import particles.particles as particles
 
 
 class Variables(object):
@@ -141,6 +142,9 @@ def initialize(self, extra_vars=None, ng=4):
 
         self.cc_data = my_data
 
+        if self.rp.get_param("particles.do_particles") == 1:
+            self.particles = particles.Particles(self.cc_data, bc, self.rp)
+
         # some auxillary data that we'll need to fill GC in, but isn't
         # really part of the main solution
         aux_data = self.data_class(my_grid)
@@ -221,6 +225,13 @@ def evolve(self):
         ymom[:, :] += 0.5*self.dt*(dens[:, :] + old_dens[:, :])*grav
         ener[:, :] += 0.5*self.dt*(ymom[:, :] + old_ymom[:, :])*grav
 
+        if self.particles is not None:
+            limiter = self.rp.get_param("compressible.limiter")
+            u, v = self.cc_data.get_var("velocity")
+
+            self.particles.update_particles(u, v, self.dt, limiter)
+            self.particles.enforce_particle_boundaries()
+
         # increment the time
         self.cc_data.t += self.dt
         self.n += 1
@@ -282,6 +293,18 @@ def dovis(self):
             else:
                 ax.set_title(field_names[n])
 
+        if self.particles is not None:
+            ax = axes[0]
+            particle_positions = self.particles.get_positions()
+            # dye particles
+            colors = self.particles.get_init_positions()[:, 0]
+
+            # plot particles
+            ax.scatter(particle_positions[:, 0],
+                particle_positions[:, 1], s=5, c=colors, alpha=0.8, cmap="Greys")
+            ax.set_xlim([myg.xmin, myg.xmax])
+            ax.set_ylim([myg.ymin, myg.ymax])
+
         plt.figtext(0.05, 0.0125, "t = {:10.5g}".format(self.cc_data.t))
 
         plt.pause(0.001)

compressible/tests/test_compressible.py

@@ -23,6 +23,7 @@ def setup_method(self):
 
         self.rp.params["mesh.nx"] = 8
         self.rp.params["mesh.ny"] = 8
+        self.rp.params["particles.do_particles"] = 0
 
         self.rp.params["eos.gamma"] = 1.4
         self.rp.params["compressible.grav"] = 1.0

compressible_rk/tests/test_compressible_rk.py

@@ -19,6 +19,7 @@ def setup_method(self):
 
         self.rp.params["mesh.nx"] = 8
         self.rp.params["mesh.ny"] = 8
+        self.rp.params["particles.do_particles"] = 0
 
         self.rp.params["eos.gamma"] = 1.4
         self.rp.params["compressible.grav"] = 1.0