Spaces:
Sleeping
Sleeping
File size: 752 Bytes
aa651cf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
# -*- coding: utf-8 -*-
"""
Simple example comparing different projectiles.
"""
from shotshaper.projectile import _Particle, ShotPutBall, SoccerBall
import matplotlib.pyplot as pl
import numpy as np
U = 10.0
angle = 20.0
p = _Particle()
shot = p.shoot(speed=U, pitch=angle)
pl.plot(shot.position[0,:],shot.position[2,:])
p = ShotPutBall('M')
shot = p.shoot(speed=U, pitch=angle)
pl.plot(shot.position[0,:],shot.position[2,:])
p = SoccerBall()
spin = np.array((0,0,0))
shot = p.shoot(speed=U, pitch=angle, spin=spin)
pl.plot(shot.position[0,:],shot.position[2,:])
spin = np.array((0,-10,0))
shot = p.shoot(speed=U, pitch=angle, spin=spin)
pl.plot(shot.position[0,:],shot.position[2,:])
pl.legend(('vacuum','air', 'no spin','spin'))
pl.show()
|