def main():
magnets_example = 5
if magnets_example == 1:
# Parameters: pos_x, pos_y, magnet strength constant
magnet1 = MagnetModel(1.0, 1.0, 1.5)
magnet2 = MagnetModel(-1.0, 1.0, 1.5)
magnet3 = MagnetModel(-1.0, -1.0, 1.5)
magnet4 = MagnetModel(1.0, -1.0, 1.5)
magnets = [magnet1, magnet2, magnet3, magnet4]
elif magnets_example == 2:
magnets = []
for i in range(1, 7):
angle = i * math.pi / 3
magnets.append(MagnetModel(math.cos(angle), math.sin(angle), 1.5))
magnets.append(MagnetModel(0.0, 0.0, 0.1))
elif magnets_example == 3:
magnet1 = MagnetModel(1.0, 1.0, 1.5)
magnet2 = MagnetModel(-2.0, -2.0, 1.5)
magnets = [magnet1, magnet2]
elif magnets_example == 4:
magnets = []
for i in range(1, 13):
angle = i * math.pi / 6
magnets.append(MagnetModel(math.cos(angle), math.sin(angle), 1.5))
magnets.append(MagnetModel(0.0, 0.0, 0.1))
elif magnets_example == 5:
magnet1 = MagnetModel(1.0, 0.0, 0.5)
magnet2 = MagnetModel(-1.0, -1.0, 0.5)
magnet3 = MagnetModel(-1.0, 1.0, 0.5)
magnets = [magnet1, magnet2, magnet3]
elif magnets_example == 6:
magnet1 = MagnetModel(1.0, 0.0, 0.5)
magnet2 = MagnetModel(-2.0, -2.0, 0.5)
magnet3 = MagnetModel(-1.0, 1.0, 0.5)
magnets = [magnet1, magnet2, magnet3]
elif magnets_example == 7:
magnet1 = MagnetModel(1.0, 0.0, 0.5)
magnet2 = MagnetModel(-1.0, 0.866, 0.5)
magnet3 = MagnetModel(-1.0, -0.866, 0.5)
magnets = [magnet1, magnet2, magnet3]
# Parameters: friction, gravity_pullback, plane_distance
pendulum = PendulumModel(0.2, 0.5, 0.1)
pendulum.magnets = magnets
# Parameters: time_step
integrator = EulerIntegrator(0.01)
# Paramaters: r, g, b - first color definition
#image_generator = BasicImageGenerator(255, 0, 0)
image_generator = AdvancedImageGenerator(255, 0, 0)
image_generator.draw_grid = False
image_generator.antialiasing = True # image will be 2x smaller
# Parameters: size, image size
basins_generator = BasinsGenerator(5, 1200)
basins_generator.pendulum_model = pendulum
basins_generator.integrator = integrator
basins_generator.image_generator = image_generator
# Parameters: initial velocity vect, simulation time, delta x/y, kernel sim time
basins_generator.calculate_basins([0, 0], 30, 0.2, 5)
# Parameters: file_name
basins_generator.draw_basins("basins")
if __name__ == "__main__":
print "Hit ESC key to quit, 'r' to render, 'c' to hide / display control points,"
print "'a' to turn on / off autorendering, 's' save screenshot,"
print "'g' to turn on / off grid"
print "Drag magnet control point to change magnet position"
print ""
magnet1 = MagnetModel(1.0, 0.0, 0.5)
magnet2 = MagnetModel(-1.0, -1.0, 0.5)
magnet3 = MagnetModel(-1.0, 1.0, 0.5)
magnets = [magnet1, magnet2, magnet3]
pendulum = PendulumModel(0.2, 0.5, 0.1)
pendulum.magnets = magnets
integrator = EulerIntegrator(0.01)
basins_generator = BasinsGenerator(5, 2 * SCREEN_SIZE)
basins_generator.pendulum_model = pendulum
basins_generator.integrator = integrator
visualizer = OpenGLvisualizer(basins_generator, len(magnets))
glEnd()
def _disable_gl(self):
glDisable(GL_MULTISAMPLE)
glDisable(GL_BLEND)
if __name__ == "__main__":
print "Hit ESC key to quit, 'r' to render, 'c' to hide / display control points,"
print "'a' to turn on / off autorendering, 's' save screenshot,"
print "'g' to turn on / off grid"
print "Drag magnet control point to change magnet position"
print ""
magnet1 = MagnetModel(1.0, 0.0, 0.5)
magnet2 = MagnetModel(-1.0, -1.0, 0.5)
magnet3 = MagnetModel(-1.0, 1.0, 0.5)
magnets = [magnet1, magnet2, magnet3]
pendulum = PendulumModel(0.2, 0.5, 0.1)
pendulum.magnets = magnets
integrator = EulerIntegrator(0.01)
basins_generator = BasinsGenerator(5, 2 * SCREEN_SIZE)
basins_generator.pendulum_model = pendulum
basins_generator.integrator = integrator
visualizer = OpenGLvisualizer(basins_generator, len(magnets))
def main():
magnets_example = 5
if magnets_example == 1:
# Parameters: pos_x, pos_y, magnet strength constant
magnet1 = MagnetModel(1.0, 1.0, 1.5)
magnet2 = MagnetModel(-1.0, 1.0, 1.5)
magnet3 = MagnetModel(-1.0, -1.0, 1.5)
magnet4 = MagnetModel(1.0, -1.0, 1.5)
magnets = [magnet1, magnet2, magnet3, magnet4]
elif magnets_example == 2:
magnets = []
for i in range(1, 7):
angle = i * math.pi / 3
magnets.append(MagnetModel(math.cos(angle), math.sin(angle), 1.5))
magnets.append(MagnetModel(0.0, 0.0, 0.1))
elif magnets_example == 3:
magnet1 = MagnetModel(1.0, 1.0, 1.5)
magnet2 = MagnetModel(-2.0, -2.0, 1.5)
magnets = [magnet1, magnet2]
elif magnets_example == 4:
magnets = []
for i in range(1, 13):
angle = i * math.pi / 6
magnets.append(MagnetModel(math.cos(angle), math.sin(angle), 1.5))
magnets.append(MagnetModel(0.0, 0.0, 0.1))
elif magnets_example == 5:
magnet1 = MagnetModel(1.0, 0.0, 0.5)
magnet2 = MagnetModel(-1.0, -1.0, 0.5)
magnet3 = MagnetModel(-1.0, 1.0, 0.5)
magnets = [magnet1, magnet2, magnet3]
elif magnets_example == 6:
magnet1 = MagnetModel(1.0, 0.0, 0.5)
magnet2 = MagnetModel(-2.0, -2.0, 0.5)
magnet3 = MagnetModel(-1.0, 1.0, 0.5)
magnets = [magnet1, magnet2, magnet3]
elif magnets_example == 7:
magnet1 = MagnetModel(1.0, 0.0, 0.5)
magnet2 = MagnetModel(-1.0, 0.866, 0.5)
magnet3 = MagnetModel(-1.0, -0.866, 0.5)
magnets = [magnet1, magnet2, magnet3]
# Parameters: friction, gravity_pullback, plane_distance
pendulum = PendulumModel(0.2, 0.5, 0.1)
pendulum.magnets = magnets
# Parameters: time_step
integrator = EulerIntegrator(0.01)
# Paramaters: r, g, b - first color definition
#image_generator = BasicImageGenerator(255, 0, 0)
image_generator = AdvancedImageGenerator(255, 0, 0)
image_generator.draw_grid = False
image_generator.antialiasing = True # image will be 2x smaller
# Parameters: size, image size
basins_generator = BasinsGenerator(5, 1200)
basins_generator.pendulum_model = pendulum
basins_generator.integrator = integrator
basins_generator.image_generator = image_generator
# Parameters: initial velocity vect, simulation time, delta x/y, kernel sim time
basins_generator.calculate_basins([0, 0], 30, 0.2, 5)
# Parameters: file_name
basins_generator.draw_basins("basins")
