File mplot/mart.py changed (mode: 100644) (index b7bc751..439531b) |
... |
... |
def rbf_colormesh(ax, ngrid=500, **kwargs): |
437 |
437 |
return ax.pcolormesh(X, Y, C, cmap=cmap, shading='nearest', zorder=-100, rasterized=True, **kwargs) |
return ax.pcolormesh(X, Y, C, cmap=cmap, shading='nearest', zorder=-100, rasterized=True, **kwargs) |
438 |
438 |
|
|
439 |
439 |
|
|
|
440 |
|
|
440 |
441 |
def gcircle(ax, r=1, nrod=200, **kwargs): |
def gcircle(ax, r=1, nrod=200, **kwargs): |
441 |
442 |
if ax.space == 'G': |
if ax.space == 'G': |
442 |
443 |
circle = mpatches.Circle((0,0), r, **kwargs) |
circle = mpatches.Circle((0,0), r, **kwargs) |
443 |
444 |
else: |
else: |
444 |
|
phi = np.linspace(0, 6.283185307, nrod, endpoint=True) |
|
445 |
|
cos_phi = np.cos(phi) |
|
446 |
|
sin_phi = np.sin(phi) |
|
447 |
|
|
|
448 |
|
sample_G = np.array((cos_phi, sin_phi)).T * r |
|
449 |
|
|
|
450 |
|
f_model = ax.sample_box.f_model |
|
451 |
|
sample = f_model.new_sample(sample_G, space='G', extend=True) |
|
|
445 |
|
sample = wmisc.get_isodistances(ax.sample_box.f_model, r, nrod=200) |
452 |
446 |
xy = getattr(sample, ax.space)[:,:2] |
xy = getattr(sample, ax.space)[:,:2] |
453 |
447 |
circle = mpatches.Polygon(xy, **kwargs) |
circle = mpatches.Polygon(xy, **kwargs) |
454 |
448 |
|
|
|
... |
... |
def uframe(ax, **kwargs): |
465 |
459 |
|
|
466 |
460 |
|
|
467 |
461 |
def isocurves(ax, ngrid=200, limits=None, ncurves=5, **kwargs): |
def isocurves(ax, ngrid=200, limits=None, ncurves=5, **kwargs): |
468 |
|
if limits is None: |
|
469 |
|
#xmin, xmax = ax.get_xlim() |
|
470 |
|
#ymin, ymax = ax.get_ylim() |
|
|
462 |
|
r = ncurves + 1 #č jakási rezerva, mám už plné zuby s těmi limity |
|
463 |
|
sample = wmisc.get_isodistances(ax.sample_box.f_model, r, nrod=ngrid) |
|
464 |
|
xy = getattr(sample, ax.space)[:,:2] |
471 |
465 |
|
|
472 |
|
sample_G = np.array([[-ncurves, -ncurves], [ncurves, ncurves]]) |
|
473 |
|
sample = ax.sample_box.f_model.new_sample(sample_G, space='G', extend=True) |
|
474 |
|
xy = getattr(sample, ax.space)[:,:2] |
|
475 |
|
xmin, ymin = np.min(xy, axis=0) |
|
476 |
|
xmax, ymax = np.max(xy, axis=0) |
|
|
466 |
|
xmin, ymin = np.min(xy, axis=0) |
|
467 |
|
xmax, ymax = np.max(xy, axis=0) |
477 |
468 |
|
|
478 |
|
else: # G-čko zlobí |
|
479 |
|
xmin, xmax, ymin, ymax = limits |
|
480 |
|
|
|
481 |
|
|
|
|
469 |
|
if limits is not None: # G-čko zlobí |
|
470 |
|
lxmin, lxmax, lymin, lymax = limits |
|
471 |
|
xmin = min(xmin, lxmin) |
|
472 |
|
xmax = max(xmax, lxmax) |
|
473 |
|
ymin = min(ymin, lymin) |
|
474 |
|
ymax = max(ymax, lymax) |
482 |
475 |
|
|
483 |
476 |
x = np.linspace(xmin, xmax, ngrid) |
x = np.linspace(xmin, xmax, ngrid) |
484 |
477 |
y = np.linspace(ymin, ymax, ngrid) |
y = np.linspace(ymin, ymax, ngrid) |
|
... |
... |
def curly(ax, linewidths=[0.7, 0.5, 0.4, 0.3, 0.2, 0.1], nrid=200, color='k', ** |
557 |
550 |
if ax.space in ('U', 'aU'): |
if ax.space in ('U', 'aU'): |
558 |
551 |
return None |
return None |
559 |
552 |
|
|
560 |
|
elif (ax.sample_box.nvar==2) and (ax.space in ('Rn', 'aRn', 'R', 'aR', 'P', 'aP')): |
|
561 |
|
isocurves(ax, ngrid=nrid, limits=None, ncurves=len(linewidths),\ |
|
|
553 |
|
elif (ax.sample_box.nvar == 2) and (ax.space in ('Rn', 'aRn', 'R', 'aR', 'P', 'aP')): |
|
554 |
|
limits = (*ax.get_xlim(), *ax.get_ylim()) |
|
555 |
|
isocurves(ax, ngrid=nrid, limits=limits, ncurves=len(linewidths),\ |
562 |
556 |
linewidths=np.flip(linewidths), colors=[color], **kwargs) |
linewidths=np.flip(linewidths), colors=[color], **kwargs) |
563 |
557 |
|
|
564 |
558 |
else: |
else: |
File mplot/maxes.py changed (mode: 100644) (index 7cd68b1..288a7be) |
... |
... |
def tri_nodes_plot(ax, tri_space=None, tn_scheme=None, ms=3, mew=0.6, lw=0.7, |
279 |
279 |
on_delete_simplex=None) |
on_delete_simplex=None) |
280 |
280 |
|
|
281 |
281 |
mart.setup(ax) |
mart.setup(ax) |
282 |
|
mart.curly(ax, nrid=nrid, linewidths=linewidths) |
|
283 |
282 |
Tri.integrate() |
Tri.integrate() |
284 |
283 |
if tri_space == ax.space: |
if tri_space == ax.space: |
285 |
284 |
mart.triplot(ax, color="#B2B2B2", lw=lw/1.4, zorder=100) |
mart.triplot(ax, color="#B2B2B2", lw=lw/1.4, zorder=100) |
|
... |
... |
def tri_nodes_plot(ax, tri_space=None, tn_scheme=None, ms=3, mew=0.6, lw=0.7, |
287 |
286 |
mart.tri_plot(ax, Tri=Tri, color="#B2B2B2", lw=lw/1.4, zorder=100, rasterized=True) |
mart.tri_plot(ax, Tri=Tri, color="#B2B2B2", lw=lw/1.4, zorder=100, rasterized=True) |
288 |
287 |
|
|
289 |
288 |
mart.plot_points(ax, ms=ms, mew=mew, zorder=100500) |
mart.plot_points(ax, ms=ms, mew=mew, zorder=100500) |
|
289 |
|
#č curly a boundaries raději nakreslíme co nejpozději |
|
290 |
|
mart.curly(ax, nrid=nrid, linewidths=linewidths) |
290 |
291 |
try: |
try: |
291 |
292 |
mart.plot_boundaries(ax, lw=lw, zorder=1050, nrod=nrid) |
mart.plot_boundaries(ax, lw=lw, zorder=1050, nrod=nrid) |
292 |
293 |
except: |
except: |
|
... |
... |
def tri_plot(ax, tri_space=None, linewidths=[0.7, 0.5, 0.4, 0.3, 0.2, 0.1], |
317 |
318 |
on_delete_simplex=None) |
on_delete_simplex=None) |
318 |
319 |
|
|
319 |
320 |
mart.setup(ax) |
mart.setup(ax) |
320 |
|
mart.curly(ax, nrid=nrid, linewidths=linewidths) |
|
321 |
321 |
if tri_space == ax.space: |
if tri_space == ax.space: |
322 |
322 |
mart.triplot(ax, color="#B2B2B2", lw=lw/1.4, zorder=100) |
mart.triplot(ax, color="#B2B2B2", lw=lw/1.4, zorder=100) |
323 |
323 |
else: |
else: |
324 |
324 |
mart.tri_plot(ax, Tri=Tri, color="#B2B2B2", lw=lw/1.4, zorder=100, rasterized=True) |
mart.tri_plot(ax, Tri=Tri, color="#B2B2B2", lw=lw/1.4, zorder=100, rasterized=True) |
325 |
325 |
|
|
326 |
326 |
mart.plot_points(ax, ms=ms, mew=mew, zorder=100500) |
mart.plot_points(ax, ms=ms, mew=mew, zorder=100500) |
|
327 |
|
#č curly a boundaries raději nakreslíme co nejpozději |
|
328 |
|
mart.curly(ax, nrid=nrid, linewidths=linewidths) |
327 |
329 |
try: |
try: |
328 |
330 |
mart.plot_boundaries(ax, lw=lw, zorder=1050, nrod=nrid) |
mart.plot_boundaries(ax, lw=lw, zorder=1050, nrod=nrid) |
329 |
331 |
except: |
except: |
|
... |
... |
def convex_hull_plot(ax, tri_space=None, ms=3, mew=0.6, lw=0.7, nrid=200, |
348 |
350 |
tri_space = ax.space |
tri_space = ax.space |
349 |
351 |
|
|
350 |
352 |
mart.setup(ax) |
mart.setup(ax) |
351 |
|
mart.curly(ax, nrid=nrid, linewidths=linewidths) |
|
352 |
353 |
qhull = khull.QHull(ax.sample_box, space=tri_space, incremental=False) |
qhull = khull.QHull(ax.sample_box, space=tri_space, incremental=False) |
353 |
354 |
mart.qhull_plot(ax, qhull, color="#B2B2B2", lw=lw, zorder=100) |
mart.qhull_plot(ax, qhull, color="#B2B2B2", lw=lw, zorder=100) |
354 |
355 |
mart.plot_points(ax, ms=ms, mew=mew, zorder=100500) |
mart.plot_points(ax, ms=ms, mew=mew, zorder=100500) |
|
356 |
|
#č curly a boundaries raději nakreslíme co nejpozději |
|
357 |
|
mart.curly(ax, nrid=nrid, linewidths=linewidths) |
355 |
358 |
try: |
try: |
356 |
359 |
mart.plot_boundaries(ax, lw=lw, zorder=1050, nrod=nrid) |
mart.plot_boundaries(ax, lw=lw, zorder=1050, nrod=nrid) |
357 |
360 |
except: |
except: |