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 =0.  # pociatocna velkost rychlosti rychlost v m/s
omega0=-1.
g=9.81  # gravitacne zrychlenie
alpha = 0.1 
f=0.4;
m=1.  #hmotnost
R=0.1
absft=m*g*cos(alpha)*f
gsin=-m*g*sin(alpha)
t=empty(1001) # nenaplnene array casovych momentov indexovanych cislami od 0 po 1000
s=empty(1001) # nenaplnene array suradnice x
v=empty(1001) # nenaplnene array suradnice y
omega=empty(1001) # nenaplnene array x komponenta rychosti
I=0.5*m*R**2

dt=0.0001 

t[0]=0  # pociatocny cas
s[0]=0  # pociatocna suradnica
v[0]=v0
omega[0]=omega0
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
	vrel=v[i]+R*omega[i]
	if (vrel<0):
		ft=absft
	else:
		ft=-absft
	a=(gsin+ft)/m
	s[i+1]=s[i]+v[i]*dt
	v[i+1]=v[i]+a*dt
	omega[i+1]=omega[i]+(ft*R/I)*dt
	print(i,vrel, ft, v[i],omega[i])
plot(t,s)     # vykresli trajektoriu
show()  #okno s grafom treba zatvorit aby to pokracovalo
plot(t,omega)
show()