plotConservative.py

#!/usr/bin/env python3
 

 
# Authors: Caleb Voss, Wilson Beebe
 
 
from matplotlib import cm
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
 
 
def potential(x, y):
    return 1 + np.sin(x) * np.sin(y)
 
 
def potentialSurface():
    X = np.arange(-8, 8, 0.25)
    Y = np.arange(-8, 8, 0.25)
    X, Y = np.meshgrid(X, Y)
    Z = potential(X, Y)
    return X, Y, Z
 
 
fig = plt.figure()
ax = fig.gca(projection="3d", aspect="equal")
X, Y, Z = potentialSurface()
ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm, linewidth=0)
 
x = np.loadtxt("vfrrt-conservative.path")
ax.plot(x[:, 0], x[:, 1], potential(x[:, 0], x[:, 1]), color="b")
 
x = np.loadtxt("trrt-conservative.path")
ax.plot(x[:, 0], x[:, 1], potential(x[:, 0], x[:, 1]), color="r")
 
x = np.loadtxt("rrtstar-conservative.path")
ax.plot(x[:, 0], x[:, 1], potential(x[:, 0], x[:, 1]), color="g")
 
plt.show()