def new_visualization(core, gas):
converter = nbody_system.nbody_to_si(10.0 | units.AU,
core.total_mass() + gas.total_mass())
visualization = Asterisk(converter)
visualization.initialize_code()
#optional: set the zoom and rotation of the visualization
#visualization.parameters.rotation = (15, -15, 45)
#visualization.parameters.camera_distance = 100 | units.parsec
visualization.marker_particles.add_particles(core)
visualization.gas_particles.add_particles(gas)
return visualization
def new_visualization(stars, gas):
#creating visualization code
converter = nbody_system.nbody_to_si(10.0 | units.parsec, stars.total_mass() + gas.total_mass())
visualization = Asterisk(converter, redirection="none")
#optional: change OpenGL perspective settings
#visualization.set_field_of_view(45.0)
#visualization.set_z_near(0.1 | units.parsec)
#visualization.set_z_far(3000.0 | units.parsec)
#initialize code (creates gui windows)
visualization.initialize_code()
visualization.parameters.use_octree_for_gas = True
#optional: set the zoom and rotation of the visualization
#visualization.parameters.rotation = (15, -15, 45)
#visualization.parameters.camera_distance = 100 | units.parsec
#add (now colored) particles to visualization
visualization.star_particles.add_particles(stars)
visualization.gas_particles.add_particles(gas)
return visualization
# create channels to and from the local particle set and the simulations
from_gravity_to_local = gravity.particles.new_channel_to(particles)
from_stellar_evolution_to_local = \
stellar_evolution.particles.new_channel_to(particles)
from_stellar_evolution_to_local.copy()
# creating colored particles
particles = new_particles_with_blackbody_color(particles)
particles.alpha = 1.0
particles.radius = stellar_evolution.particles.radius.sqrt() * \
(1e4 | units.parsec).sqrt()
# creating visualization code
converter = nbody_system.nbody_to_si(10.0 | units.parsec, masses.sum())
visualization = Asterisk(converter, redirection="none")
# optional: change OpenGL perspective settings
# visualization.set_field_of_view(45.0)
# visualization.set_z_near(0.1 | units.parsec)
# visualization.set_z_far(3000.0 | units.parsec)
# initialize code (creates gui windows)
visualization.initialize_code()
# optional: set the zoom and rotation of the visualization
# visualization.parameters.rotation = (15, -15, 45)
# visualization.parameters.camera_distance = 100 | units.parsec
# add (now colored) particles to visualization
visualization.particles.add_particles(particles)
# create channels to and from the local particle set and the simulations
from_gravity_to_local = gravity.particles.new_channel_to(particles)
from_stellar_evolution_to_local = \
stellar_evolution.particles.new_channel_to(particles)
from_stellar_evolution_to_local.copy()
# creating colored particles
particles = new_particles_with_blackbody_color(particles)
particles.alpha = 1.0
particles.radius = (stellar_evolution.particles.radius.sqrt() *
(1e4 | units.parsec).sqrt())
# creating visualization code
converter = nbody.nbody_to_si(10.0 | units.parsec, masses.sum())
visualization = Asterisk(converter, redirection="none")
visualization.initialize_code()
# optional: set the zoom and rotation of the visualization
# visualization.parameters.rotation = (15, -15, 45)
# visualization.parameters.camera_distance = 100 | units.parsec
# add (now colored) particles to visualization
visualization.particles.add_particles(particles)
from_local_to_viz = particles.new_channel_to(visualization.particles)
visualization.store_view(0 | units.Myr)
# evolve module for some time
for i in range(1, 100):
target_time = i * 0.05 | units.Myr
print('starting evolve to time = ', target_time)
gravity = new_gravity(particles, converter)
stellar_evolution = new_stellar_evolution(particles)
#create channels to and from the local particle set and the simulations
from_gravity_to_local = gravity.particles.new_channel_to(particles)
from_stellar_evolution_to_local = stellar_evolution.particles.new_channel_to(particles)
from_stellar_evolution_to_local.copy()
#creating colored particles
particles = new_particles_with_blackbody_color(particles)
particles.alpha = 1.0
particles.radius = stellar_evolution.particles.radius.sqrt() * (1e4 | units.parsec).sqrt()
#creating visualization code
converter = nbody.nbody_to_si(10.0 | units.parsec, masses.sum())
visualization = Asterisk(converter, redirection="none")
visualization.initialize_code()
#optional: set the zoom and rotation of the visualization
#visualization.parameters.rotation = (15, -15, 45)
#visualization.parameters.camera_distance = 100 | units.parsec
#add (now colored) particles to visualization
visualization.particles.add_particles(particles)
from_local_to_viz = particles.new_channel_to(visualization.particles)
visualization.store_view(0|units.Myr)
#evolve module for some time
for i in range(1, 100):
target_time = i * 0.05 | units.Myr
print 'starting evolve to time = ', target_time
