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, 0.7 sekundy 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]=???? #doplnte novu suradnicu
	z[i+1]=???? #doplnte novu suradnicu
	v= ???? #doplnte velkost rychlolsti v bode i
	fx= ???? #doplnte x-zlozku sily danej odporom vzduchu v bode i
	fz= ???? #doplnte z-zlozku sily danej odporom vzduchu a tiazou v bode i
	vx[i+1]= ???? #doplnte novu zlozku rychlosti
	vz[i+1]= ???? # doplnte novu zlosku rychlosti
plot(x,z)     # vykresli trajektoriu
show()  #okno s grafom treba zatvorit aby to pokracovalo
plot(t,x) # vykresli priebeh suradnicve x v case
show() #okno s grafom treba zatvorit aby to pokracovalo
plot(t,z) # vykresli priebeh suradnice z v case
show()