def plot_data(geo_data,
direction="y",
data_type='all',
series="all",
legend_font_size=6,
**kwargs):
"""
Plot the projection of the raw data (interfaces and orientations) in 2D following a
specific directions
Args:
direction(str): xyz. Caartesian direction to be plotted
series(str): series to plot
**kwargs: seaborn lmplot key arguments. (TODO: adding the link to them)
Returns:
None
"""
warnings.warn(
"gempy plotting functionality will be moved in version 1.2, use gempy.plotting module instead",
FutureWarning)
plot = PlotData2D(geo_data)
# TODO saving options
return plot.plot_data(direction=direction,
data_type=data_type,
series=series,
legend_font_size=legend_font_size,
**kwargs)
def plot_section(geo_data, block, cell_number, direction="y", **kwargs):
"""
Plot a section of the block model
Args:
cell_number(int): position of the array to plot
direction(str): xyz. Caartesian direction to be plotted
interpolation(str): Type of interpolation of plt.imshow. Default 'none'. Acceptable values are 'none'
,'nearest', 'bilinear', 'bicubic',
'spline16', 'spline36', 'hanning', 'hamming', 'hermite', 'kaiser',
'quadric', 'catrom', 'gaussian', 'bessel', 'mitchell', 'sinc',
'lanczos'
**kwargs: imshow keywargs
Returns:
None
"""
warnings.warn(
"gempy plotting functionality will be moved in version 1.2, "
"use gempy.plotting module instead", FutureWarning)
plot = PlotData2D(geo_data)
sec_plot = plot.plot_block_section(cell_number,
block=block,
direction=direction,
**kwargs)
def plot_scalar_field(geo_data,
potential_field,
cell_number,
N=20,
direction="y",
plot_data=True,
series="all",
*args,
**kwargs):
"""
Plot a potential field in a given direction.
Args:
cell_number(int): position of the array to plot
potential_field(str): name of the potential field (or series) to plot
n_pf(int): number of the potential field (or series) to plot
direction(str): xyz. Caartesian direction to be plotted
serie: *Deprecated*
**kwargs: plt.contour kwargs
Returns:
None
"""
warnings.warn(
"gempy plotting functionality will be moved in version 1.2, "
"use gempy.plotting module instead", FutureWarning)
plot = PlotData2D(geo_data)
plot.plot_scalar_field(potential_field,
cell_number,
N=N,
direction=direction,
plot_data=plot_data,
series=series,
*args,
**kwargs)
def plot_topology(geo_data, G, centroids, direction="y"):
"""
Plot the topology adjacency graph in 2-D.
Args:
geo_data (gempy.data_management.InputData):
G (skimage.future.graph.rag.RAG):
centroids (dict): Centroid node coordinates as a dictionary with node id's (int) as keys and (x,y,z) coordinates
as values.
Keyword Args
direction (str): "x", "y" or "z" specifying the slice direction for 2-D topology analysis. Default None.
Returns:
Nothing, it just plots.
"""
warnings.warn("gempy plotting functionality will be moved in version 1.2, "
"use gempy.plotting module instead", FutureWarning)
PlotData2D.plot_topo_g(geo_data, G, centroids, direction=direction)
def plot_gradient(geo_data,
scalar_field,
gx,
gy,
gz,
cell_number,
q_stepsize=5,
direction="y",
plot_scalar=True,
**kwargs):
"""
Plot the gradient of the scalar field in a given direction.
Args:
geo_data (gempy.DataManagement.InputData): Input data of the model
scalar_field(numpy.array): scalar field to plot with the gradient
gx(numpy.array): gradient in x-direction
gy(numpy.array): gradient in y-direction
gz(numpy.array): gradient in z-direction
cell_number(int): position of the array to plot
q_stepsize(int): step size between arrows to indicate gradient
direction(str): xyz. Caartesian direction to be plotted
plot_scalar(bool): boolean to plot scalar field
**kwargs: plt.contour kwargs
Returns:
None
"""
plot = PlotData2D(geo_data)
plot.plot_gradient(scalar_field,
gx,
gy,
gz,
cell_number,
q_stepsize=q_stepsize,
direction=direction,
plot_scalar=plot_scalar,
**kwargs)
