add meshc and surfc plot functions

This commit is contained in:
wyq 2023-06-01 14:51:23 +08:00
parent 6214af5b47
commit 62ff9c0fb9
9 changed files with 235 additions and 52 deletions

View File

@ -1075,18 +1075,13 @@ public class Axis implements Cloneable {
} }
updateTickValues(); updateTickValues();
// if (this.timeAxis) {
// this.updateTimeLabels();
// } else {
// tickValues = MIMath.getIntervalValues(minValue, maxValue);
// }
} }
/** /**
* Update tick values * Update tick values
*/ */
public void updateTickValues() { public void updateTickValues() {
List<Object> r = MIMath.getIntervalValues1(minValue, maxValue); List<Object> r = MIMath.getIntervalValuesAxis(minValue, maxValue);
this.tickValues = (double[]) r.get(0); this.tickValues = (double[]) r.get(0);
this.tickDeltaValue = (Double) r.get(1); this.tickDeltaValue = (Double) r.get(1);
} }

View File

@ -964,6 +964,103 @@ public class MIMath {
return getIntervalValues(min, max, false); return getIntervalValues(min, max, false);
} }
/**
* Create contour values by minimum and maximum values
*
* @param min Minimum value
* @param max Maximum value
* @return Contour values
*/
public static List<Object> getIntervalValuesAxis(double min, double max) {
return getIntervalValuesAxis(min, max, false);
}
/**
* Create contour values by minimum and maximum values
*
* @param min Minimum value
* @param max Maximum value
* @param isExtend If extend values
* @return Contour values
*/
public static List<Object> getIntervalValuesAxis(double min, double max, boolean isExtend) {
int i, cNum, aD, aE;
double cDelt, range, newMin;
String eStr;
List<Object> r = new ArrayList<>();
range = BigDecimalUtil.sub(max, min);
if (range == 0.0) {
r.add(new double[]{min});
r.add(0.0);
return r;
} else if (range < 0) {
range = -range;
double temp = min;
min = max;
max = temp;
}
eStr = String.format("%1$E", range);
aD = Integer.parseInt(eStr.substring(0, 1));
aE = (int) Math.floor(Math.log10(range));
int nMin = 4;
if (aD >= nMin) {
cDelt = BigDecimalUtil.pow(10, aE);
cNum = aD;
} else {
double cd = BigDecimalUtil.pow(10, aE - 1);
i = 5;
cDelt = BigDecimalUtil.mul(i, cd);
cNum = (int) (range / cDelt);
while (cNum < nMin && i > 1) {
i--;
cDelt = BigDecimalUtil.mul(i, cd);
cNum = (int) (range / cDelt);
}
}
int temp = (int) (min / cDelt + 1);
newMin = BigDecimalUtil.mul(temp, cDelt);
if (newMin - min >= cDelt) {
newMin = BigDecimalUtil.sub(newMin, cDelt);
cNum += 1;
}
if (newMin + (cNum - 1) * cDelt > max) {
cNum -= 1;
} else if (newMin + (cNum - 1) * cDelt + cDelt < max) {
cNum += 1;
}
//Get values
List<Double> values = new ArrayList<>();
double v;
for (i = 0; i < cNum; i++) {
v = BigDecimalUtil.add(newMin, BigDecimalUtil.mul(i, cDelt));
if (v >= min && v <= max)
values.add(v);
}
//Extend values
if (isExtend) {
if (values.get(0) > min) {
values.add(0, BigDecimalUtil.sub(newMin, cDelt));
}
if (values.get(values.size() - 1) < max) {
values.add(BigDecimalUtil.add(values.get(values.size() - 1), cDelt));
}
}
double[] cValues = new double[values.size()];
for (i = 0; i < values.size(); i++) {
cValues[i] = values.get(i);
}
r.add(cValues);
r.add(cDelt);
return r;
}
/** /**
* Create contour values by minimum and maximum values * Create contour values by minimum and maximum values
* *
@ -993,29 +1090,15 @@ public class MIMath {
eStr = String.format("%1$E", range); eStr = String.format("%1$E", range);
aD = Integer.parseInt(eStr.substring(0, 1)); aD = Integer.parseInt(eStr.substring(0, 1));
aE = (int) Math.floor(Math.log10(range)); aE = (int) Math.floor(Math.log10(range));
// int idx = eStr.indexOf("E");
// if (idx < 0) {
// aE = 0;
// } else {
// aE = Integer.parseInt(eStr.substring(eStr.indexOf("E") + 1));
// }
if (aD > 5) { if (aD > 5) {
//cDelt = Math.pow(10, aE);
cDelt = BigDecimalUtil.pow(10, aE); cDelt = BigDecimalUtil.pow(10, aE);
cNum = aD; cNum = aD;
//newMin = Convert.ToInt32((min + cDelt) / Math.Pow(10, aE)) * Math.Pow(10, aE);
//newMin = (int) (min / cDelt + 1) * cDelt;
} else if (aD == 5) { } else if (aD == 5) {
//cDelt = aD * Math.pow(10, aE - 1);
cDelt = aD * BigDecimalUtil.pow(10, aE - 1); cDelt = aD * BigDecimalUtil.pow(10, aE - 1);
cNum = 10; cNum = 10;
//newMin = Convert.ToInt32((min + cDelt) / Math.Pow(10, aE)) * Math.Pow(10, aE);
//newMin = (int) (min / cDelt + 1) * cDelt;
cNum++; cNum++;
} else { } else {
//cDelt = aD * Math.pow(10, aE - 1);
double cd = BigDecimalUtil.pow(10, aE - 1); double cd = BigDecimalUtil.pow(10, aE - 1);
//cDelt = BigDecimalUtil.mul(aD, cDelt);
cDelt = BigDecimalUtil.mul(5, cd); cDelt = BigDecimalUtil.mul(5, cd);
cNum = (int) (range / cDelt); cNum = (int) (range / cDelt);
if (cNum < 5) { if (cNum < 5) {
@ -1026,8 +1109,6 @@ public class MIMath {
cNum = (int) (range / cDelt); cNum = (int) (range / cDelt);
} }
} }
//newMin = Convert.ToInt32((min + cDelt) / Math.Pow(10, aE - 1)) * Math.Pow(10, aE - 1);
//newMin = (int) (min / cDelt + 1) * cDelt;
} }
int temp = (int) (min / cDelt + 1); int temp = (int) (min / cDelt + 1);
newMin = BigDecimalUtil.mul(temp, cDelt); newMin = BigDecimalUtil.mul(temp, cDelt);

View File

@ -1,8 +1,6 @@
<?xml version="1.0" encoding="UTF-8" standalone="no"?> <?xml version="1.0" encoding="UTF-8" standalone="no"?>
<MeteoInfo File="milconfig.xml" Type="configurefile"> <MeteoInfo File="milconfig.xml" Type="configurefile">
<Path OpenPath="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\contour"> <Path OpenPath="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\mesh">
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\plot"/>
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\volume"/>
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\patch"/> <RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\patch"/>
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\satellite"/> <RecentFolder Folder="D:\Working\MIScript\Jython\mis\satellite"/>
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\satellite\FY"/> <RecentFolder Folder="D:\Working\MIScript\Jython\mis\satellite\FY"/>
@ -10,25 +8,29 @@
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\3d"/> <RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\3d"/>
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\funny"/> <RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\funny"/>
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\io\awx"/> <RecentFolder Folder="D:\Working\MIScript\Jython\mis\io\awx"/>
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl"/>
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types"/> <RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types"/>
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\contour"/> <RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\contour"/>
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\io"/> <RecentFolder Folder="D:\Working\MIScript\Jython\mis\io"/>
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\io\grib"/> <RecentFolder Folder="D:\Working\MIScript\Jython\mis\io\grib"/>
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\contour"/> <RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\contour"/>
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\surf"/>
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl"/>
<RecentFolder Folder="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\mesh"/>
</Path> </Path>
<File> <File>
<OpenedFiles> <OpenedFiles>
<OpenedFile File="D:\MyProgram\java\MeteoInfoDev\toolbox\meteoview3d\_reload.py"/> <OpenedFile File="D:\MyProgram\java\MeteoInfoDev\toolbox\meteoview3d\_reload.py"/>
<OpenedFile File="D:\MyProgram\java\MeteoInfoDev\toolbox\meteoview3d\mainGUI.py"/> <OpenedFile File="D:\MyProgram\java\MeteoInfoDev\toolbox\meteoview3d\mainGUI.py"/>
<OpenedFile File="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\contour\contour_1.py"/> <OpenedFile File="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\surf\surfc_1.py"/>
<OpenedFile File="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\contour\contourf_1.py"/> <OpenedFile File="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\mesh\mesh_2.py"/>
<OpenedFile File="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\mesh\meshc_peaks.py"/>
</OpenedFiles> </OpenedFiles>
<RecentFiles> <RecentFiles>
<RecentFile File="D:\MyProgram\java\MeteoInfoDev\toolbox\meteoview3d\_reload.py"/> <RecentFile File="D:\MyProgram\java\MeteoInfoDev\toolbox\meteoview3d\_reload.py"/>
<RecentFile File="D:\MyProgram\java\MeteoInfoDev\toolbox\meteoview3d\mainGUI.py"/> <RecentFile File="D:\MyProgram\java\MeteoInfoDev\toolbox\meteoview3d\mainGUI.py"/>
<RecentFile File="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\contour\contour_1.py"/> <RecentFile File="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\surf\surfc_1.py"/>
<RecentFile File="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\contour\contourf_1.py"/> <RecentFile File="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\mesh\mesh_2.py"/>
<RecentFile File="D:\Working\MIScript\Jython\mis\plot_types\3d\jogl\mesh\meshc_peaks.py"/>
</RecentFiles> </RecentFiles>
</File> </File>
<Font> <Font>
@ -36,5 +38,5 @@
</Font> </Font>
<LookFeel DockWindowDecorated="true" LafDecorated="true" Name="FlatDarkLaf"/> <LookFeel DockWindowDecorated="true" LafDecorated="true" Name="FlatDarkLaf"/>
<Figure DoubleBuffering="true"/> <Figure DoubleBuffering="true"/>
<Startup MainFormLocation="-7,0" MainFormSize="1354,809"/> <Startup MainFormLocation="-7,0" MainFormSize="1429,860"/>
</MeteoInfo> </MeteoInfo>

View File

@ -1146,11 +1146,17 @@ class Axes3D(Axes):
x = np.arange(a.shape[1]) x = np.arange(a.shape[1])
y = np.arange(a.shape[0]) y = np.arange(a.shape[0])
args = args[1:] args = args[1:]
elif n <=4: else:
x = args[0] x = args[0]
y = args[1] y = args[1]
a = args[2] a = args[2]
args = args[3:] args = args[3:]
if x.ndim == 2:
x = x[0]
if y.ndim == 2:
y = y[:,0]
if len(args) > 0: if len(args) > 0:
level_arg = args[0] level_arg = args[0]
if isinstance(level_arg, int): if isinstance(level_arg, int):
@ -1214,11 +1220,17 @@ class Axes3D(Axes):
x = np.arange(a.shape[1]) x = np.arange(a.shape[1])
y = np.arange(a.shape[0]) y = np.arange(a.shape[0])
args = args[1:] args = args[1:]
elif n <=4: else:
x = args[0] x = args[0]
y = args[1] y = args[1]
a = args[2] a = args[2]
args = args[3:] args = args[3:]
if x.ndim == 2:
x = x[0]
if y.ndim == 2:
y = y[:,0]
if len(args) > 0: if len(args) > 0:
level_arg = args[0] level_arg = args[0]
if isinstance(level_arg, int): if isinstance(level_arg, int):

View File

@ -1138,6 +1138,40 @@ class Axes3DGL(Axes3D):
self.add_graphic(graphics) self.add_graphic(graphics)
return graphics return graphics
def meshc(self, *args, **kwargs):
"""
Contour plot under mesh surface plot.
:param x: (*array_like*) Optional. X coordinate array.
:param y: (*array_like*) Optional. Y coordinate array.
:param z: (*array_like*) 2-D z value array.
:param cmap: (*string*) Color map string.
:returns: 3D mesh and contour graphics.
"""
if len(args) <= 2:
z = args[0]
if isinstance(z, DimArray):
x = args[0].dimvalue(1)
y = args[0].dimvalue(0)
else:
ny, nx = z.shape
x = np.arange(nx)
y = np.arange(ny)
x, y = np.meshgrid(x, y)
args1 = [x, y, z] + args[1:]
else:
args1 = args
gmesh = self.mesh(*args1, **kwargs)
zmin = args1[2].min()
kwargs['offset'] = zmin
args1 = args1[:3]
gcontour = self.contour(*args1, **kwargs)
return gmesh, gcontour
def surf(self, *args, **kwargs): def surf(self, *args, **kwargs):
""" """
creates a three-dimensional surface plot creates a three-dimensional surface plot
@ -1253,6 +1287,41 @@ class Axes3DGL(Axes3D):
self.add_graphic(graphics) self.add_graphic(graphics)
return graphics return graphics
def surfc(self, *args, **kwargs):
"""
Contour plot under surface plot.
:param x: (*array_like*) Optional. X coordinate array.
:param y: (*array_like*) Optional. Y coordinate array.
:param z: (*array_like*) 2-D z value array.
:param cmap: (*string*) Color map string.
:param lighting: (*bool*) Using light or not.
:returns: 3D surface and contour graphics.
"""
if len(args) <= 2:
z = args[0]
if isinstance(z, DimArray):
x = args[0].dimvalue(1)
y = args[0].dimvalue(0)
else:
ny, nx = z.shape
x = np.arange(nx)
y = np.arange(ny)
x, y = np.meshgrid(x, y)
args1 = [x, y, z] + args[1:]
else:
args1 = args
gsurf = self.surf(*args1, **kwargs)
zmin = args1[2].min()
kwargs['offset'] = zmin
args1 = args1[:3]
gcontour = self.contour(*args1, **kwargs)
return gsurf, gcontour
def plot_surface(self, *args, **kwargs): def plot_surface(self, *args, **kwargs):
""" """
creates a three-dimensional surface plot creates a three-dimensional surface plot

View File

@ -40,25 +40,21 @@ g_axes = None
__all__ = [ __all__ = [
'annotate', 'antialias', 'arrow', 'arrowline', 'axes', 'axes3d', 'axes3dgl', 'axesm', 'caxes', 'axis', 'annotate', 'antialias', 'arrow', 'arrowline', 'axes', 'axes3d', 'axes3dgl', 'axesm', 'caxes', 'axis',
'axism', 'bar', 'bar3', 'barh', 'barbs', 'barbsm', 'bgcolor', 'box', 'axism', 'bar', 'bar3', 'barh', 'barbs', 'barbsm', 'bgcolor', 'box', 'boxplot', 'windrose', 'cla',
'boxplot', 'windrose', 'cla', 'clabel', 'clc', 'clear', 'clf', 'cll', 'cloudspec', 'colorbar', 'contour', 'clabel', 'clc', 'clear', 'clf', 'cll', 'cloudspec', 'colorbar', 'contour', 'contourf',
'contourf', 'contourfm', 'contourm', 'contourfslice', 'contourslice', 'delfig', 'draw', 'draw_if_interactive',
'contourfm', 'contourm', 'contourfslice', 'contourslice', 'delfig', 'draw', 'draw_if_interactive', 'errorbar', 'errorbar', 'figure', 'glfigure', 'figsize', 'patch', 'rectangle', 'fill', 'fill3', 'fill_between',
'figure', 'glfigure', 'figsize', 'patch', 'rectangle', 'fill', 'fill3', 'fill_between', 'fill_betweenx', 'fimplicit3', 'fill_betweenx', 'fimplicit3', 'webmap', 'gca', 'gcf', 'gc_collect', 'geoshow', 'get_figure',
'webmap', 'gca', 'gcf', 'gc_collect', 'geoshow', 'get_figure', 'gifaddframe', 'gifanimation', 'giffinish', 'gifaddframe', 'gifanimation', 'giffinish', 'grid', 'gridshow', 'gridshowm', 'hist', 'imshow',
'grid', 'gridshow', 'gridshowm', 'hist', 'imshow', 'imshowm', 'isosurface', 'legend', 'left_title', 'lighting', 'imshowm', 'isosurface', 'legend', 'left_title', 'lighting', 'loglog', 'makecolors', 'makelegend',
'loglog', 'makecolors', 'makesymbolspec', 'masklayer', 'material', 'mesh', 'meshc', 'model', 'particles', 'pcolor', 'pcolorm',
'makelegend', 'makesymbolspec', 'masklayer', 'material', 'mesh', 'model', 'particles', 'pcolor', 'pcolorm', 'pie', 'plot', 'pie', 'plot', 'plot3', 'plotm', 'quiver', 'quiver3', 'quiverkey', 'quiverm', 'readlegend',
'plot3', 'plotm', 'quiver', 'quiver3', 'right_title', 'refresh', 'savefig', 'savefig_jpeg', 'scatter', 'scatter3', 'scatterm', 'semilogx',
'quiverkey', 'quiverm', 'readlegend', 'right_title', 'refresh', 'savefig', 'savefig_jpeg', 'scatter', 'scatter3', 'semilogy', 'show', 'slice3', 'stationmodel', 'stem', 'stem3', 'step', 'streamplot', 'streamplot3',
'scatterm', 'streamplotm', 'streamslice', 'subplot', 'subplots', 'suptitle', 'supxlabel', 'supylabel', 'surf',
'semilogx', 'semilogy', 'show', 'slice3', 'stationmodel', 'stem', 'stem3', 'step', 'streamplot', 'streamplot3', 'surfc', 'taylor_diagram', 'text', 'text3', 'title', 'trisurf', 'twinx', 'twiny', 'violinplot',
'streamplotm', 'streamslice', 'subplot', 'subplots', 'suptitle', 'supxlabel', 'supylabel', 'volumeplot', 'weatherspec', 'xaxis', 'xlabel', 'xlim', 'xreverse', 'xticks', 'yaxis', 'ylabel',
'surf', 'taylor_diagram', 'text', 'text3', 'title', 'trisurf', 'twinx', 'twiny', 'violinplot', 'volumeplot', 'ylim', 'yreverse', 'yticks', 'zaxis', 'zlabel', 'zlim', 'zticks', 'isinteractive'
'weatherspec', 'xaxis',
'xlabel', 'xlim', 'xreverse', 'xticks', 'yaxis', 'ylabel', 'ylim', 'yreverse', 'yticks', 'zaxis', 'zlabel', 'zlim',
'zticks',
'isinteractive'
] ]
@ -1999,6 +1995,20 @@ def mesh(*args, **kwargs):
return r return r
@_copy_docstring_and_deprecators(Axes3DGL.mesh)
def meshc(*args, **kwargs):
global g_axes
if g_axes is None:
g_axes = axes3d()
else:
if not isinstance(g_axes, Axes3D):
g_axes = axes3d()
r = g_axes.meshc(*args, **kwargs)
draw_if_interactive()
return r
@_copy_docstring_and_deprecators(Axes3D.surf) @_copy_docstring_and_deprecators(Axes3D.surf)
def surf(*args, **kwargs): def surf(*args, **kwargs):
global g_axes global g_axes
@ -2013,6 +2023,20 @@ def surf(*args, **kwargs):
return r return r
@_copy_docstring_and_deprecators(Axes3DGL.surfc)
def surfc(*args, **kwargs):
global g_axes
if g_axes is None:
g_axes = axes3d()
else:
if not isinstance(g_axes, Axes3D):
g_axes = axes3d()
r = g_axes.surfc(*args, **kwargs)
draw_if_interactive()
return r
@_copy_docstring_and_deprecators(Axes3DGL.trisurf) @_copy_docstring_and_deprecators(Axes3DGL.trisurf)
def trisurf(T, x, y, z, normal=None, **kwargs): def trisurf(T, x, y, z, normal=None, **kwargs):
global g_axes global g_axes