New plotting methods for VertexOnlyMesh

[achanbour]

I have extended Firedrake's plotting routines to support the VertexOnlyMesh. In summary:

• I added a new pointplot method that calls matplotlib.scatter for visualising the points of a VertexOnlyMesh,
• I extended the existing quiver method to plot vector fields on a VertexOnlyMesh.

I've also added unit tests to make sure these changes work as expected. Unless anyone has anything to add or comment, I will edit Firedrake's documentation to include a demo of these new functionalities.

[connorjward]

Nice!

I'd add something small to the manual here.

[danshapero]

This will be really useful, thanks! My one suggestion is that you call it scatter instead of pointplot. The other plotting routines have the same names as their counterparts in matplotlib so that people could guess the names of the function they wanted without having to consult the documentation.

[achanbour]

This will be really useful, thanks! My one suggestion is that you call it scatter instead of pointplot. The other plotting routines have the same names as their counterparts in matplotlib so that people could guess the names of the function they wanted without having to consult the documentation.

This makes sense, thank you! I've now modified it to scatter.

[connorjward]

I'm basically happy now I think. Really nice.

[achanbour]

Added serial run checks in all plotting methods.

[JHopeCollins]

This is great, thanks! Will it allow plotting points values when the parent mesh is 1D?

[achanbour]

This is great, thanks! Will it allow plotting points values when the parent mesh is 1D?

Matplotlib's scatter assumes at least 2D so the code currently doesn't support 1D and raises an Exception.

firedrake/firedrake/pyplot/mpl.py

Lines 131 to 138 in 3d41501

	if axes is None:
	fig = plt.figure()
	if gdim == 3:
	axes = fig.add_subplot(111, projection="3d")
	elif gdim == 2:
	axes = fig.add_subplot(111)
	else:
	raise ValueError("Scatter is only supported for 2D and 3D meshes.")

[JHopeCollins]

Matplotlib's scatter assumes at least 2D so the code currently doesn't support 1D and raises an Exception.

Of course, I'd forgotten there were separate calls for the 1D and 2D plotting