예제 #1
0
#
# 2014/10/29    First issue

import numpy as np
from Box import Box
import matplotlib.pyplot as plt
import matplotlib.animation as animation

dt = 0.01

#positions = np.array([[5, 5], [6, 6], [4, 6], [3, 6]])
#velocity = np.array([[1, 1], [2, 0], [3, 2], [1, 2]])
#masses = np.array([1, 1, 1, 1])
#radius = np.array([0.2, 0.5, 1, 0.1])

positions = np.array([[5, 5], [6, 6]])
velocity = np.array([[1, 3], [2, 0]])
masses = np.array([1, 1])
radius = np.array([0.2, 1])

box = Box(positions, velocity, masses, radius, dt = dt, debug=False)

positions = np.array([[5, 5], [6, 6], [4, 4]])
velocity = np.array([[1, 3], [2, 0], [0, 2]])
masses = np.array([1, 1, 1])
radius = np.array([0.2, 1, 1])

box = Box(positions, velocity, masses, radius, dt = dt, debug=False)
print(box.getPositions())
print(box.getVelocities())