Update and standardize documentation plots

doc/gallery_src/equivalent_sources/block_averaged_sources.py

@@ -106,17 +106,15 @@
 
 title = "Observed magnetic anomaly data"
 
-# Make colormap of data
-# Get the 95 percentile of the maximum absolute value between the original and
-# gridded data so we can use the same color scale for both plots and have 0
-# centered at the white color.
-maxabs = vd.maxabs(data.total_field_anomaly_nt, grid.magnetic_anomaly.values) * 0.95
-pygmt.makecpt(
-    cmap="vik",
-    series=(-maxabs, maxabs),
-    background=True,
+# Get the 99th percentile of the absolute value to use as color scale limits
+maxabs = vd.maxabs(
+    data.total_field_anomaly_nt,
+    percentile=99,
 )
 
+# Make colormap of data
+pygmt.makecpt(cmap="balance+h0", series=[-maxabs, maxabs], background=True)
+
 with pygmt.config(FONT_TITLE="12p"):
     fig.plot(
         projection=fig_proj,
@@ -129,8 +127,6 @@
         cmap=True,
     )
 
-fig.colorbar(cmap=True, frame=["a400f100", "x+lnT"])
-
 fig.shift_origin(xshift=fig_width + 1)
 
 title = "Gridded and upward-continued"
@@ -142,6 +138,10 @@
         cmap=True,
     )
 
-fig.colorbar(cmap=True, frame=["a400f100", "x+lnT"])
+fig.colorbar(
+    cmap=True,
+    frame=["a400f100", "x+lnT"],
+    position="n0/0+jTC+w8c/0.25c+h+o-0.5c/0.8c",
+)
 
 fig.show()

doc/gallery_src/equivalent_sources/cartesian.py

@@ -95,17 +95,15 @@
 
 title = "Observed magnetic anomaly data"
 
-# Make colormap of data
-# Get the 95 percentile of the maximum absolute value between the original and
-# gridded data so we can use the same color scale for both plots and have 0
-# centered at the white color.
-maxabs = vd.maxabs(data.total_field_anomaly_nt, grid.magnetic_anomaly.values) * 0.95
-pygmt.makecpt(
-    cmap="vik",
-    series=(-maxabs, maxabs),
-    background=True,
+# Get the 99th percentile of the absolute value to use as color scale limits
+maxabs = vd.maxabs(
+    data.total_field_anomaly_nt,
+    percentile=99,
 )
 
+# Make colormap of data
+pygmt.makecpt(cmap="balance+h0", series=[-maxabs, maxabs], background=True)
+
 with pygmt.config(FONT_TITLE="12p"):
     fig.plot(
         projection=fig_proj,
@@ -118,8 +116,6 @@
         cmap=True,
     )
 
-fig.colorbar(cmap=True, frame=["a400f100", "x+lnT"])
-
 fig.shift_origin(xshift=fig_width + 1)
 
 title = "Gridded and upward-continued"
@@ -131,6 +127,10 @@
         cmap=True,
     )
 
-fig.colorbar(cmap=True, frame=["a400f100", "x+lnT"])
+fig.colorbar(
+    cmap=True,
+    frame=["a400f100", "x+lnT"],
+    position="n0/0+jTC+w8c/0.25c+h+o-0.5c/0.8c",
+)
 
 fig.show()

doc/gallery_src/equivalent_sources/gradient_boosted.py

@@ -115,16 +115,15 @@
 
 title = "Observed gravity disturbance data"
 
-# Make colormap of data
-pygmt.makecpt(
-    cmap="vik",
-    series=(
-        -data.gravity_disturbance.quantile(0.99),
-        data.gravity_disturbance.quantile(0.99),
-    ),
-    background=True,
+# Get the 99th percentile of the absolute value to use as color scale limits
+maxabs = vd.maxabs(
+    data.gravity_disturbance,
+    percentile=99,
 )
 
+# Make colormap of data
+pygmt.makecpt(cmap="balance+h0", series=[-maxabs, maxabs], background=True)
+
 with pygmt.config(FONT_TITLE="14p"):
     fig.plot(
         projection=fig_proj,
@@ -137,8 +136,6 @@
         cmap=True,
     )
 
-fig.colorbar(cmap=True, frame=["a50f25", "x+lmGal"])
-
 fig.shift_origin(xshift=fig_width + 1)
 
 title = "Gridded with gradient-boosted equivalent sources"
@@ -150,6 +147,10 @@
         cmap=True,
     )
 
-fig.colorbar(cmap=True, frame=["a50f25", "x+lmGal"])
+fig.colorbar(
+    cmap=True,
+    frame=["x+lmGal"],
+    position="n0/0+jTC+w10c/0.5c+h+o-0.5c/0.9c",
+)
 
 fig.show()

doc/gallery_src/equivalent_sources/spherical.py

@@ -80,38 +80,41 @@
 # Plot observed and gridded gravity disturbance
 fig = pygmt.Figure()
 
+# Get the 99th percentile of the absolute value to use as color scale limits
+maxabs = vd.maxabs(gravity_disturbance, percentile=99)
+
 # Make colormap of data
-# Get the 90% of the maximum absolute value between the original and gridded
-# data so we can use the same color scale for both plots and have 0 centered
-# at the white color.
-maxabs = vd.maxabs(gravity_disturbance, grid.gravity_disturbance.values) * 0.90
-pygmt.makecpt(
-    cmap="vik",
-    series=(-maxabs, maxabs),
-    background=True,
-)
+pygmt.makecpt(cmap="balance+h0", series=[-maxabs, maxabs], background=True)
+
+
+title = "Observed gravity disturbance data"
 
 fig.plot(
-    projection="M10c",
+    projection="M12c",
     region=region,
-    frame=["WSne", "xa5", "ya4"],
+    frame=[f"WSne+t{title}", "xa5", "ya4"],
     x=longitude,
     y=latitude,
     fill=gravity_disturbance,
     style="c0.1c",
     cmap=True,
 )
 
-fig.colorbar(cmap=True, frame=["a100f50", "x+lmGal"])
+fig.shift_origin(xshift="w+1c")
 
-fig.shift_origin(xshift="w+3c")
+title = "Gridded and upward-continued"
 
 fig.grdimage(
-    frame=["ESnw", "xa5", "ya4"],
+    frame=[f"ESnw+t{title}", "xa5", "ya4"],
     grid=grid.gravity_disturbance,
     cmap=True,
+    nan_transparent=True,
 )
 
-fig.colorbar(cmap=True, frame=["a100f50", "x+lmGal"])
+fig.colorbar(
+    cmap=True,
+    frame=["x+lmGal"],
+    position="n0/0+jTC+w10c/0.5c+h+o-0.5c/0.9c",
+)
 
 fig.show()

doc/gallery_src/forward/point_gravity.py

@@ -54,9 +54,9 @@
 
 title = "Gravitational acceleration (downward)"
 
-maxabs = vd.maxabs(gravity) * 0.80
+maxabs = vd.maxabs(gravity, percentile=100)
 
-pygmt.makecpt(cmap="vik", series=(-maxabs, maxabs, 0.3))
+pygmt.makecpt(cmap="vik", series=[-maxabs, maxabs, 0.3], background=True)
 
 with pygmt.config(FONT_TITLE="16p"):
     fig.grdimage(
@@ -68,8 +68,16 @@
         cmap=True,
     )
 
-fig.plot(x=easting, y=northing, style="c0.2c", fill="grey")
+fig.plot(
+    x=easting,
+    y=northing,
+    style="t0.15c",
+    fill="magenta",
+    label="Point masses",
+)
+
+fig.colorbar(cmap=True, position="JMR+e", frame=["x+lmGal"])
 
-fig.colorbar(cmap=True, position="JMR", frame=["a.6f.2", "x+lmGal"])
+fig.legend()
 
 fig.show()

doc/gallery_src/forward/prism_layer.py

@@ -51,17 +51,17 @@
 # Plot gravity field
 fig = pygmt.Figure()
 
-title = "Gravitational acceleration of a layer of prisms"
+title = "Gravitational acceleration of topography with prisms"
 
 with pygmt.config(FONT_TITLE="14p"):
     fig.grdimage(
         region=region,
         projection="X10c/10c",
         grid=grid.gravity,
-        frame=["a", f"+t{title}", 'x+l"easting (m)"', 'y+l"northing (m)"'],
+        frame=["a", f"+t{title}", "x+leasting (m)", "y+lnorthing (m)"],
         cmap="viridis",
     )
 
-fig.colorbar(cmap=True, position="JMR", frame=["a2f1", "x+lmGal"])
+fig.colorbar(cmap=True, position="JMR", frame=["x+lmGal"])
 
 fig.show()

doc/gallery_src/forward/prisms_topo_gravity.py

@@ -87,15 +87,21 @@
 
 title = "Gravitational acceleration of the topography"
 
+# Get the max absolute value to use as color scale limits
+cpt_lims = vd.maxabs(grid.gravity)
+
+# Make colormap of data
+pygmt.makecpt(cmap="balance+h0", series=[-cpt_lims, cpt_lims])
+
 with pygmt.config(FONT_TITLE="14p"):
     fig.grdimage(
         region=xy_region,
         projection=fig_proj,
         grid=grid.gravity,
         frame=["ag", f"+t{title}"],
-        cmap="vik",
+        cmap=True,
     )
 
-fig.colorbar(cmap=True, frame=["a100f50", "x+lmGal"])
+fig.colorbar(cmap=True, position="JMR", frame=["a100f50", "x+lmGal"])
 
 fig.show()

doc/gallery_src/forward/tesseroid.py

@@ -50,7 +50,7 @@
 # Plot the gravitational field
 fig = pygmt.Figure()
 
-title = "Downward component of gravitational acceleration"
+title = "Gravitational acceleration of a tesseroid"
 
 with pygmt.config(FONT_TITLE="16p"):
     fig.grdimage(
@@ -61,7 +61,17 @@
         cmap="viridis",
     )
 
-fig.colorbar(cmap=True, frame=["a200f50", "x+lmGal"])
+# Plot edges of tesseroid
+fig.plot(
+    x=[tesseroid[0], tesseroid[1], tesseroid[1], tesseroid[0], tesseroid[0]],
+    y=[tesseroid[2], tesseroid[2], tesseroid[3], tesseroid[3], tesseroid[2]],
+    pen="1p,red",
+    label="Tesseroid boundaries",
+)
+
+fig.legend()
+
+fig.colorbar(cmap=True, position="JMR", frame=["a200f50", "x+lmGal"])
 
 fig.coast(shorelines="1p,black")
 

doc/gallery_src/forward/tesseroid_layer.py

@@ -86,9 +86,11 @@
     gravity.g_z,
     projection="M15c",
     nan_transparent=True,
+    cmap=True,
     frame="+tForward gravity",
 )
+
 fig.basemap(frame=True)
-fig.colorbar(frame="af+lGravity (mGal)")
+fig.colorbar(frame="af+lgravity (mGal)")
 fig.coast(shorelines="0.5p,black", borders=["1/0.5p,black"])
 fig.show()

doc/gallery_src/forward/tesseroid_variable_density.py

@@ -78,7 +78,7 @@ def density(radius):
 # Plot the gravitational field
 fig = pygmt.Figure()
 
-title = "Downward component of gravitational acceleration"
+title = "Gravitational acceleration of variable density tesseroids"
 
 with pygmt.config(FONT_TITLE="16p"):
     fig.grdimage(
@@ -89,7 +89,18 @@ def density(radius):
         cmap="viridis",
     )
 
-fig.colorbar(cmap=True, frame=["a200f50", "x+lmGal"])
+# Plot edges of tesseroids
+for i, tesseroid in enumerate(tesseroids):
+    label = "Tesseroid boundaries" if i == 0 else None
+    fig.plot(
+        x=[tesseroid[0], tesseroid[1], tesseroid[1], tesseroid[0], tesseroid[0]],
+        y=[tesseroid[2], tesseroid[2], tesseroid[3], tesseroid[3], tesseroid[2]],
+        pen="1p,red",
+        label=label,
+    )
+fig.legend()
+
+fig.colorbar(cmap=True, position="JMR", frame=["a200f50", "x+lmGal"])
 
 fig.coast(shorelines="1p,black")
 

doc/gallery_src/gravity_disturbance.py

@@ -19,6 +19,7 @@
 import boule as bl
 import ensaio
 import pygmt
+import verde as vd
 import xarray as xr
 
 # Load the global gravity grid
@@ -36,7 +37,11 @@
 # Make a plot of data using PyGMT
 fig = pygmt.Figure()
 
-pygmt.grd2cpt(grid=disturbance, cmap="polar", continuous=True)
+# Get the 99.9th percentile of the absolute value to use as color scale limits
+maxabs = vd.maxabs(disturbance, percentile=99.9)
+
+# Make colormap of data
+pygmt.makecpt(cmap="balance+h0", series=[-maxabs, maxabs], background=True)
 
 title = "Gravity disturbance of the Earth"
 
@@ -50,6 +55,6 @@
 
 fig.coast(shorelines="0.5p,black", resolution="crude")
 
-fig.colorbar(cmap=True, frame=["a100f50", "x+lmGal"])
+fig.colorbar(cmap=True, frame=["x+lmGal"], position="+e")
 
 fig.show()

doc/gallery_src/gravity_disturbance_topofree.py

@@ -29,6 +29,7 @@
 import boule as bl
 import ensaio
 import pygmt
+import verde as vd
 import xarray as xr
 
 import harmonica as hm
@@ -62,7 +63,11 @@
 # Make a plot of data using PyGMT
 fig = pygmt.Figure()
 
-pygmt.grd2cpt(grid=disturbance_topofree, cmap="vik+h0", continuous=True)
+# Get the 99th percentile of the absolute value to use as color scale limits
+maxabs = vd.maxabs(disturbance_topofree, percentile=99)
+
+# Make colormap of data
+pygmt.makecpt(cmap="balance+h0", series=[-maxabs, maxabs], background=True)
 
 title = "Topography-free (Bouguer) gravity disturbance of the Earth"
 
@@ -77,6 +82,6 @@
 
 fig.coast(shorelines="0.5p,black", resolution="crude")
 
-fig.colorbar(cmap=True, frame=["a200f50", "x+lmGal"])
+fig.colorbar(cmap=True, frame=["x+lmGal"], position="+e")
 
 fig.show()