12345qiupeng
/
gtsam
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Address matplotlib deprecation warnings for fig.gca() and window.set_title().

release/4.3a0
vik748 2022-06-20 01:18:07 -07:00 committed by GitHub
parent 784f16fe75
commit 0984fea3f5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
python/gtsam/examples/Pose3ISAM2Example.py
View File

@ -34,7 +34,10 @@ def report_on_progress(graph: gtsam.NonlinearFactorGraph, current_estimate: gtsa
# Plot the newly updated iSAM2 inference. # Plot the newly updated iSAM2 inference.
fig = plt.figure(0) fig = plt.figure(0)
axes = fig.gca(projection='3d') if not fig.axes:
axes = fig.add_subplot(projection='3d')
else:
axes = fig.axes[0]
plt.cla() plt.cla()
i = 1 i = 1

5
python/gtsam/examples/VisualISAM2Example.py
View File

@ -33,7 +33,10 @@ def visual_ISAM2_plot(result):
fignum = 0 fignum = 0
fig = plt.figure(fignum) fig = plt.figure(fignum)
axes = fig.gca(projection='3d') if not fig.axes:
axes = fig.add_subplot(projection='3d')
else:
axes = fig.axes[0]
plt.cla() plt.cla()
# Plot points # Plot points

19
python/gtsam/utils/plot.py
View File

@ -333,7 +333,10 @@ def plot_point3(
""" """
fig = plt.figure(fignum) fig = plt.figure(fignum)
axes = fig.gca(projection='3d') if not fig.axes:
axes = fig.add_subplot(projection='3d')
else:
axes = fig.axes[0]
plot_point3_on_axes(axes, point, linespec, P) plot_point3_on_axes(axes, point, linespec, P)
axes.set_xlabel(axis_labels[0]) axes.set_xlabel(axis_labels[0])
@ -388,7 +391,7 @@ def plot_3d_points(fignum,
fig = plt.figure(fignum) fig = plt.figure(fignum)
fig.suptitle(title) fig.suptitle(title)
fig.canvas.set_window_title(title.lower()) fig.canvas.manager.set_window_title(title.lower())
def plot_pose3_on_axes(axes, pose, axis_length=0.1, P=None, scale=1): def plot_pose3_on_axes(axes, pose, axis_length=0.1, P=None, scale=1):
@ -490,7 +493,10 @@ def plot_trajectory(
axis_labels (iterable[string]): List of axis labels to set. axis_labels (iterable[string]): List of axis labels to set.
""" """
fig = plt.figure(fignum) fig = plt.figure(fignum)
axes = fig.gca(projection='3d') if not fig.axes:
axes = fig.add_subplot(projection='3d')
else:
axes = fig.axes[0]
axes.set_xlabel(axis_labels[0]) axes.set_xlabel(axis_labels[0])
axes.set_ylabel(axis_labels[1]) axes.set_ylabel(axis_labels[1])
@ -522,7 +528,7 @@ def plot_trajectory(
plot_pose3_on_axes(axes, pose, P=covariance, axis_length=scale) plot_pose3_on_axes(axes, pose, P=covariance, axis_length=scale)
fig.suptitle(title) fig.suptitle(title)
fig.canvas.set_window_title(title.lower()) fig.canvas.manager.set_window_title(title.lower())
def plot_incremental_trajectory(fignum: int, def plot_incremental_trajectory(fignum: int,
@ -545,7 +551,10 @@ def plot_incremental_trajectory(fignum: int,
Used to create animation effect. Used to create animation effect.
""" """
fig = plt.figure(fignum) fig = plt.figure(fignum)
axes = fig.gca(projection='3d') if not fig.axes:
axes = fig.add_subplot(projection='3d')
else:
axes = fig.axes[0]
poses = gtsam.utilities.allPose3s(values) poses = gtsam.utilities.allPose3s(values)
keys = gtsam.KeyVector(poses.keys()) keys = gtsam.KeyVector(poses.keys())