Loading...
Searching...
No Matches
plotConservative.py
1#!/usr/bin/env python
2
3
36
37# Authors: Caleb Voss, Wilson Beebe
38
39
40from matplotlib import cm
41import matplotlib.pyplot as plt
42from mpl_toolkits.mplot3d import Axes3D
43import numpy as np
44
45def potential(x, y):
46 return 1 + np.sin(x) * np.sin(y)
47
48def potentialSurface():
49 X = np.arange(-8, 8, 0.25)
50 Y = np.arange(-8, 8, 0.25)
51 X, Y = np.meshgrid(X, Y)
52 Z = potential(X, Y)
53 return X, Y, Z
54
55fig = plt.figure()
56ax = fig.gca(projection='3d', aspect='equal')
57X, Y, Z = potentialSurface()
58ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap=cm.coolwarm, linewidth=0)
59
60x = np.loadtxt("vfrrt-conservative.path")
61ax.plot(x[:, 0], x[:, 1], potential(x[:, 0], x[:, 1]), color='b')
62
63x = np.loadtxt("trrt-conservative.path")
64ax.plot(x[:, 0], x[:, 1], potential(x[:, 0], x[:, 1]), color='r')
65
66x = np.loadtxt("rrtstar-conservative.path")
67ax.plot(x[:, 0], x[:, 1], potential(x[:, 0], x[:, 1]), color='g')
68
69plt.show()