def plotting(s, chi, r0, phi0, alpha, beta, phiobs=np.pi/2, thetaobs=1.25, re=None):
r, t, p = re
fig = pl.figure()
ax = fig.add_subplot(111, projection='3d')
sol = solver.ODESolver(s, np.pi/2, phi0, chi, alpha, beta, r0, stop=45)
sol.r0 = r
sol.theta0 = t
sol.phi0 = p
plot.plot_3d(sol, ax, phiobs, thetaobs, r0, np.pi/2, phi0)
pl.show()
def plotting(s, chi, r0, phi0, alpha, beta, phiobs=np.pi/2, thetaobs=1.25, re=None, ax=None, c='orange'):
if not ax:
fig = pl.figure(figsize=(12, 12))
ax = fig.add_subplot(111, projection='3d')
sol = solver.ODESolver(s, np.pi/2, phi0, chi, alpha, beta, r0, stop=45)
if re:
r, t, p = re
sol.r0 = r
sol.theta0 = t
sol.phi0 = p
plot.plot_3d(sol, ax, phiobs, thetaobs, r0, np.pi/2, phi0, c)
return ax
def plotting(s, chi, r0, phi0, alpha, beta, phiobs=np.pi / 2, thetaobs=1.25):
fig = pl.figure()
ax = fig.add_subplot(111, projection='3d')
sol = solver.ODESolver(s, np.pi / 2, phi0, chi, alpha, beta, r0, stop=45)
plot.plot_3d(sol, ax, phiobs, thetaobs, r0, np.pi / 2, phi0)
pl.show()