def step(self, step_t):
t = self.c.get_current_time()
print('Simulation time:', t)
T = time.time()
self.c.step(step_t)
print('Time:', time.time() - T)
image_buffer = tc_core.Array2DVector3(self.video_manager.width,
self.video_manager.height,
Vector(0, 0, 0.0))
particles = self.c.get_render_particles()
particles.write(
self.get_output_path('particles%05d.bin' % self.step_counter))
camera = Camera('pinhole',
origin=(0, 0, 50),
look_at=(0, 0, 0),
up=(0, 1, 0),
fov=70,
width=self.video_manager.width,
height=self.video_manager.height)
self.particle_renderer.set_camera(camera)
self.particle_renderer.render(image_buffer, particles)
img = image_buffer_to_ndarray(image_buffer)
# img = LDRDisplay(exposure=1, adaptive_exposure=False).process(img)
show_image('Vis', img)
self.video_manager.write_frame(img)
self.step_counter += 1
def step(self, step_t):
t = self.c.get_current_time()
print 'Simulation time:', t
T = time.time()
self.update_levelset(t, t + step_t)
print 'Update Leveset Time:', time.time() - T
T = time.time()
self.c.step(step_t)
print 'Step Time:', time.time() - T
image_buffer = tc_core.Array2DVector3(self.video_manager.width,
self.video_manager.height,
Vector(0, 0, 0.0))
particles = self.c.get_render_particles()
particles.write(self.directory + '/particles%05d.bin' % self.frame)
res = map(float, self.resolution)
camera = Camera('pinhole',
origin=(0, res[1] * 0.4, res[2] * 1.4),
look_at=(0, -res[1] * 0.5, 0),
up=(0, 1, 0),
fov=90,
width=10,
height=10)
self.particle_renderer.set_camera(camera)
self.particle_renderer.render(image_buffer, particles)
img = image_buffer_to_ndarray(image_buffer)
img = LDRDisplay(exposure=2.0, adaptive_exposure=False).process(img)
show_image('Vis', img)
self.video_manager.write_frame(img)
self.frame += 1
def step(self, step_t):
t = self.c.get_current_time()
print 'Simulation time:', t
T = time.time()
self.c.step(step_t)
print 'Time:', time.time() - T
image_buffer = tc_core.RGBImageFloat(self.video_manager.width,
self.video_manager.height,
Vector(0, 0, 0.0))
particles = self.c.get_render_particles()
particles.write(self.directory + '/particles%05d.bin' % self.frame)
res = map(float, self.resolution)
radius = res[0] * 2.5
theta = 0
camera = Camera('pinhole',
origin=(0, res[1], res[2] * 5),
look_at=(0, 0, 0),
up=(0, 1, 0),
fov=70,
width=10,
height=10)
self.particle_renderer.set_camera(camera)
self.particle_renderer.render(image_buffer, particles)
img = image_buffer_to_ndarray(image_buffer)
img = LDRDisplay(exposure=0.1).process(img)
show_image('Vis', img)
self.video_manager.write_frame(img)
self.frame += 1
def show(self):
# allow the user to opt out of the frame viewer by invoking the script
# with --no-viewer in the command line
if '--no-viewer' in sys.argv:
return
show_image('Taichi Renderer', self.get_output())
'''
def show(self,
title='Taichi Texture Renderer',
res=(512, 512),
post_processor=None):
from taichi.gui.image_viewer import show_image
img = self.rasterize_to_ndarray(res)
if post_processor:
img = post_processor.process(img)
show_image(title, img)
def view(self, frame, camera):
particles = tc_core.RenderParticles()
ret = particles.read(self.input_directory + 'particles%05d.bin' % frame)
if not ret:
print 'read file failed'
return False
image_buffer = tc_core.Array2DVector3(self.video_manager.width, self.video_manager.height, Vector(0, 0, 0.0))
self.particle_renderer.set_camera(camera)
self.particle_renderer.render(image_buffer, particles)
img = image_buffer_to_ndarray(image_buffer)
#img = LDRDisplay(exposure=1, adaptive_exposure=False).process(img)
show_image('Vis', img)
self.video_manager.write_frame(img)
return True
def step(self, step_t, camera=None):
t = self.c.get_current_time()
print('* Current t: %.3f' % t)
self.update_levelset(t, t + step_t)
T = time.time()
if not self.start_simulation_time:
self.start_simulation_time = T
if not self.simulation_total_time:
self.simulation_total_time = 0
self.c.step(step_t)
self.simulation_total_time += time.time() - T
print('* Step Time: %.2f [tot: %.2f per frame %.2f]' %
(time.time() - T, time.time() - self.start_simulation_time,
self.simulation_total_time / (self.c.frame + 1)))
image_buffer = tc_core.Array2DVector3(
Vectori(self.video_manager.width, self.video_manager.height),
Vector(0, 0, 0.0))
'''
particles = self.c.get_render_particles()
try:
os.mkdir(self.directory + '/particles')
except:
pass
particles.write(self.directory + '/particles/%05d.bin' % self.c.frame)
'''
res = list(map(float, self.res))
r = res[0]
if not camera:
camera = Camera('pinhole',
origin=(0, r * 0.4, r * 1.4),
look_at=(0, -r * 0.5, 0),
up=(0, 1, 0),
fov=90,
res=(10, 10))
if False:
self.particle_renderer.set_camera(camera)
self.particle_renderer.render(image_buffer, particles)
img = image_buffer_to_ndarray(image_buffer)
img = LDRDisplay(exposure=2.0,
adaptive_exposure=False).process(img)
show_image('Vis', img)
self.video_manager.write_frame(img)
def show(self):
if not self.visualize:
return
show_image('Taichi Renderer', self.get_output())
def show_image(window_name, img):
from taichi.gui.image_viewer import show_image
show_image(window_name, img)
from taichi.gui.image_viewer import show_image
if __name__ == '__main__':
resolution = [64] * 3
resolution[1] *= 2
particle_renderer = ParticleRenderer('shadow_map',
shadow_map_resolution=0.5, alpha=0.6, shadowing=0.07, ambient_light=0.3,
light_direction=(1, 3, 1))
smoke = Smoke3(resolution=tuple(resolution),
simulation_depth=resolution[2], delta_x=1.0 / resolution[0], gravity=(0, 0, 0),
advection_order=1, cfl=0.5, smoke_alpha=80.0, smoke_beta=800,
temperature_decay=0.05, pressure_tolerance=1e-6, density_scaling=2, initial_speed=(0, 0, 0),
tracker_generation=20, perturbation=0, pressure_solver='mgpcg', num_threads=2, open_boundary=True,
maximum_pressure_iterations=200)
for i in range(600):
smoke.step(0.03)
particles = smoke.c.get_render_particles()
width, height = 512, 1024
image_buffer = tc_core.Array2DVector3(width, height, Vector(0, 0, 0.0))
radius = resolution[0] * 4
camera = Camera('pinhole', origin=(0, radius * 0.3, radius),
look_at=(0, 0, 0), up=(0, 1, 0), fov=70,
width=width, height=height)
particle_renderer.set_camera(camera)
particle_renderer.render(image_buffer, particles)
img = image_buffer_to_ndarray(image_buffer)
show_image('Vis', img)
import time
import taichi as tc
from taichi.gui.image_viewer import show_image
tc.core.test_vector_and_matrix()
exit()
tc.core.print_all_units()
tc.core.test()
tex = tc.Texture('const', value=(0.1, 0.1, 0.1, 0.1))
tex = tex.flip(0)
try:
tc.core.test_raise_error()
except Exception as e:
print('Exception:', e)
img = tc.util.imread(tc.get_asset_path('textures/vegas.jpg'))
show_image('img', img)
time.sleep(1)
print('Testing passed.')
