from pylab import *
from numpy import *  #je dobre zvyknutis si aj na tento import
#uvedene dva importy v neinteraktivne spustanych .py suboroch robia nieco podobne
# ako ked sa interaktivny ipython spustkik s option --pylab 

v0 =7.  # pociatocna velkost rychlosti rychlost v m/s
g=-9.81  # gravitacne zrychlenie
alphadeg =45.   # uhol vystrelu v stupnoch
alpha = alphadeg/180.*pi  #doplnte premenu na radiany
K=1.  #koeficient trecej sily , vid zadanie vo formate .pdf
m=1.  #hmotnost
t=empty(1001) # nenaplnene array casovych momentov indexovanych cislami od 0 po 1000
x=empty(1001) # nenaplnene array suradnice x
z=empty(1001) # nenaplnene array suradnice y
vx=empty(1001) # nenaplnene array x komponenta rychosti
vz=empty(1001) # nenaplnene array y komponenta rychosti

dt=0.7/1000  # maly krok v case, pol sekunda rozdelena na 1000 intervalov

t[0]=0  # pociatocny cas
x[0]=0  # pociatocna suradnica
z[0]=0  # pociatocna suradnica
vx[0] = v0* cos(alpha) # doplnte vypocet x komponenty rychlosti na pociatku
vz[0] = v0* sin(alpha) # doplnte vypocet z komponenty rychlosti na pociatku
for i in range(0,1000):    #for statement, i bude z mnoziny {0,1,2,3,...,999}
	t[i+1]=t[i]+dt  #toto je posun v case o dt, novy casovy okamih ma index i+1
	#v nasledujucich riadkoch doplnte posuny v case
	x[i+1]=x[i]+vx[i]*dt
	z[i+1]=z[i]+vz[i]*dt 
	v=sqrt(vx[i]*vx[i]+vz[i]*vz[i])
	fx=-K*vx[i]*v
	fz=-K*vz[i]*v
	vx[i+1]=vx[i]+fx/m*dt
	vz[i+1]=vz[i]+fz/m*dt+g*dt
plot(x,z)     # vykresli trajektoriu
show()  #okno s grafom treba zatvorit aby to pokracovalo
plot(t,x)
show() #okno s grafom treba zatvorit aby to pokracovalo
plot(t,z)
show()