from matplotlib.pyplot import *
import math
import numpy
from mvtChampBaxial import MvtChampBaxial

mvt = MvtChampBaxial("../../elecmag/bobine/bobine-1")
            

figure(figsize=(8,5))
plot(mvt.r,mvt.f_ADr(0.0,mvt.r)[0])
xlabel("z")
ylabel("Atheta")
grid()
            

figure(figsize=(7,7))
mvt.lignesB(color='grey',density=1)
alpha=900.0
pz0=1.0
z0 = -0.7
r0=0.00
tmax=3.0
te=0.01
[r1,z1,theta1] = mvt.trajectoire(alpha,r0,z0,0.01,pz0,tmax,te)
plot(z1,r1)
[r2,z2,theta2] = mvt.trajectoire(alpha,r0,z0,0.02,pz0,tmax,te)
plot(z2,r2)
[r3,z3,theta3] = mvt.trajectoire(alpha,r0,z0,0.05,pz0,tmax,te)
plot(z3,r3)
[r4,z4,theta4] = mvt.trajectoire(alpha,r0,z0,0.1,pz0,tmax,te)
plot(z4,r4)
[r5,z5,theta5] = mvt.trajectoire(alpha,r0,z0,0.15,pz0,tmax,te)
plot(z5,r5)
axis([-1,1,-0.2,0.2])
grid()

            

figure(figsize=(8,5))
plot(z1,theta1)
plot(z2,theta2)
plot(z3,theta3)
plot(z4,theta4)
xlabel("z")
ylabel("theta")
grid()
            

mvt = MvtChampBaxial("../../elecmag/bobine/bobine-2")
            

figure(figsize=(8,5))
plot(mvt.r,mvt.f_ADr(0.0,mvt.r)[0])
xlabel("z")
ylabel("Atheta")
grid()
            

figure(figsize=(7,7))
mvt.lignesB(color='grey',density=1)
alpha=120.0
pz0=1.0
z0 = -0.7
r0=0.00
tmax=3.0
te=0.01
[r1,z1,theta1] = mvt.trajectoire(alpha,r0,z0,0.01,pz0,tmax,te)
plot(z1,r1)
[r2,z2,theta2] = mvt.trajectoire(alpha,r0,z0,0.02,pz0,tmax,te)
plot(z2,r2)
[r3,z3,theta3] = mvt.trajectoire(alpha,r0,z0,0.05,pz0,tmax,te)
plot(z3,r3)
[r4,z4,theta4] = mvt.trajectoire(alpha,r0,z0,0.1,pz0,tmax,te)
plot(z4,r4)
[r5,z5,theta5] = mvt.trajectoire(alpha,r0,z0,0.15,pz0,tmax,te)
plot(z5,r5)
axis([-1,1,-0.2,0.2])
grid()

            

figure(figsize=(8,5))
plot(z1,theta1)
plot(z2,theta2)
plot(z3,theta3)
plot(z4,theta4)
xlabel("z")
ylabel("theta")
grid()
            

alpha=120.0
pz0=0.05
z0 = -0.7
r0=0.00
tmax=50.0
te=0.01
[r1,z1,theta1] = mvt.trajectoire(alpha,r0,z0,0.01,pz0,tmax,te)
figure(figsize=(7,7))
plot(z1,r1)
xlabel("z")
ylabel("r")
axis([-1,1,-0.2,0.2])
grid()
            

figure(figsize=(8,5))
plot(z1,theta1)
xlabel("z")
ylabel("theta")
grid()
            

alpha=120.0
pz0=0.03
z0 = -0.7
r0=0.00
tmax=50.0
te=0.01
[r1,z1,theta1] = mvt.trajectoire(alpha,r0,z0,0.01,pz0,tmax,te)
figure(figsize=(7,7))
plot(z1,r1)
xlabel("z")
ylabel("r")
axis([-1,1,-0.2,0.2])
grid()