def _change_kernel_name(self, change):
if change["new"] == "flat":
self.kernel_table = None
flat = np.ones(256, dtype="f4")
flat[-1] = 0.0
self.interpolation_texture = Texture1D(data=flat)
return
self.kernel_table = SPHKernelInterpolationTable(change["new"])
values = self.kernel_table.interpolate_array(np.mgrid[0.0:1.0:256j])
self.interpolation_texture = Texture1D(data=values.astype("f4"))
def _generate_container_field_sph(self, field):
if field not in ["dts", "t"]:
raise KeyError(field)
length = unorm(self.vec)
pos = self[self.ds._sph_ptypes[0], "particle_position"]
r = pos - self.start_point
l = udot(r, self.vec / length)
if field == "t":
return l / length
hsml = self[self.ds._sph_ptypes[0], "smoothing_length"]
mass = self[self.ds._sph_ptypes[0], "particle_mass"]
dens = self[self.ds._sph_ptypes[0], "density"]
# impact parameter from particle to ray
b = np.sqrt(np.sum(r**2, axis=1) - l**2)
# Use an interpolation table to evaluate the integrated 2D
# kernel from the dimensionless impact parameter b/hsml.
itab = SPHKernelInterpolationTable(self.ds.kernel_name)
dl = itab.interpolate_array(b / hsml) * mass / dens / hsml**2
return dl / length
def _default_kernel_table(self):
if self.kernel_name is None:
return
return SPHKernelInterpolationTable(self.kernel_name)
class SPHRendering(SceneComponent):
name = "sph_rendering"
data = traitlets.Instance(ParticlePositions)
scale = traitlets.CFloat(1.0)
max_particle_size = traitlets.CFloat(1e-3)
kernel_name = traitlets.Unicode("cubic")
kernel_table = traitlets.Instance(SPHKernelInterpolationTable,
allow_none=True)
interpolation_texture = traitlets.Instance(Texture1D)
@traitlets.observe("kernel_name")
def _change_kernel_name(self, change):
if change["new"] == "flat":
self.kernel_table = None
flat = np.ones(256, dtype="f4")
flat[-1] = 0.0
self.interpolation_texture = Texture1D(data=flat)
return
self.kernel_table = SPHKernelInterpolationTable(change["new"])
values = self.kernel_table.interpolate_array(np.mgrid[0.0:1.0:256j])
self.interpolation_texture = Texture1D(data=values.astype("f4"))
@traitlets.default("kernel_table")
def _default_kernel_table(self):
if self.kernel_name is None:
return
return SPHKernelInterpolationTable(self.kernel_name)
@traitlets.default("interpolation_texture")
def _default_interpolation_texture(self):
if self.kernel_table is None and self.kernel_name == "flat":
flat = np.ones(256, dtype="f4")
flat[-1] = 0.0
return Texture1D(data=flat)
values = self.kernel_table.interpolate_array(np.mgrid[0.0:1.0:256j])
return Texture1D(data=values.astype("f4"))
def render_gui(self, imgui, renderer, scene):
changed = super(SPHRendering, self).render_gui(imgui, renderer, scene)
_, kernel_index = imgui.listbox("Kernel",
KERNEL_TYPES.index(self.kernel_name),
KERNEL_TYPES)
if _:
self.kernel_name = KERNEL_TYPES[kernel_index]
changed = changed or _
_, new_value = imgui.slider_float("Log Scale", math.log10(self.scale),
-8.0, 2.0)
if _:
self.scale = 10**new_value
changed = True
imgui.text("Filter Particle Max Size")
_, new_value = imgui.slider_float("", 1.0 / self.max_particle_size,
1.0, 100.0)
if _:
self.max_particle_size = 1.0 / new_value
changed = True
return changed
def draw(self, scene, program):
GL.glEnable(GL.GL_CULL_FACE)
GL.glCullFace(GL.GL_BACK)
with self.interpolation_texture.bind(0):
GL.glDrawArraysInstanced(GL.GL_TRIANGLE_STRIP, 0, 4,
self.data.size)
def _set_uniforms(self, scene, shader_program):
cam = scene.camera
shader_program._set_uniform("scale", self.scale)
shader_program._set_uniform("projection", cam.projection_matrix)
shader_program._set_uniform("modelview", cam.view_matrix)
shader_program._set_uniform("max_particle_size",
self.max_particle_size)
shader_program._set_uniform(
"inv_pmvm", np.linalg.inv(cam.projection_matrix @ cam.view_matrix))