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))  
        
        
        
        
        
                
        
Exemple #3
0
        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")