from Constantes import *

def S(N, p, a, b, S0):
    # V vaut 0 si T=1+a avec proba p
    # V vaut 1 si T=1+b avec proba 1-p
    V = np.random.rand(N) > p
    W = np.insert(np.cumsum(math.log(1+a) + math.log((1+b)/(1+a))*V), 0, 0)
    
    return S0 * ?????   

# Exemples de trajectoire

#plt.figure()
plt.scatter(np.linspace(0,N,N+1, dtype = int), S(N, 0.5, a, b, S0), label = 'p = 0.5')
plt.scatter(np.linspace(0,N,N+1, dtype = int), S(N, 0.1, a, b, S0), label = 'p = 0.1')
plt.scatter(np.linspace(0,N,N+1, dtype = int), S(N, 0.9, a, b, S0), label = 'p = 0.9')
plt.title("Exemples de trajectoire pour différentes valeurs de p")
plt.legend()
plt.show(block=False)