From 858f169024b7b81f309c206d6e99c8bedddd5bea Mon Sep 17 00:00:00 2001
From: "Knollmueller, Jakob (kjako)" <jakob@knollmueller.de>
Date: Sat, 15 Dec 2018 21:00:23 +0100
Subject: [PATCH] bbugfix
---
nifty5/plot.py | 72 +++++++++++++++++++++++++-------------------------
1 file changed, 36 insertions(+), 36 deletions(-)
diff --git a/nifty5/plot.py b/nifty5/plot.py
index d6fb0b168..6fd18c940 100644
--- a/nifty5/plot.py
+++ b/nifty5/plot.py
@@ -217,44 +217,44 @@ def _plot(f, ax, **kwargs):
ax.set_ylabel(kwargs.pop("ylabel", ""))
cmap = kwargs.pop("colormap", plt.rcParams['image.cmap'])
if isinstance(dom, DomainTuple):
- if isinstance(dom[1], RGSpace):
- if isinstance(dom[0],RGSpace):
- if len(dom[0].shape) == 2:
- nx, ny = dom[0].shape
- dx, dy = dom[0].distances
+ if len(dom) == 2:
+ if isinstance(dom[1], RGSpace):
+ if isinstance(dom[0],RGSpace):
+ if len(dom[0].shape) == 2:
+ nx, ny = dom[0].shape
+ dx, dy = dom[0].distances
+ rgb = _rgb_data(f[0].to_global_data())
+ im = ax.imshow(
+ rgb, extent=[0, nx * dx, 0, ny * dy], origin="lower", **norm)
+ # from mpl_toolkits.axes_grid1 import make_axes_locatable
+ # divider = make_axes_locatable(ax)
+ # cax = divider.append_axes("right", size="5%", pad=0.05)
+ # plt.colorbar(im,cax=cax)
+ _limit_xy(**kwargs)
+ return
+ if isinstance(dom[0],(HPSpace, GLSpace)):
+ import pyHealpix
+ xsize = 800
+ res, mask, theta, phi = _mollweide_helper(xsize)
+ res = np.full(shape=res.shape+(3,), fill_value=1., dtype=np.float64)
rgb = _rgb_data(f[0].to_global_data())
- im = ax.imshow(
- rgb, extent=[0, nx * dx, 0, ny * dy], origin="lower", **norm)
- # from mpl_toolkits.axes_grid1 import make_axes_locatable
- # divider = make_axes_locatable(ax)
- # cax = divider.append_axes("right", size="5%", pad=0.05)
- # plt.colorbar(im,cax=cax)
- _limit_xy(**kwargs)
+ if isinstance(dom[0], HPSpace):
+ ptg = np.empty((phi.size, 2), dtype=np.float64)
+ ptg[:, 0] = theta
+ ptg[:, 1] = phi
+ base = pyHealpix.Healpix_Base(int(np.sqrt(dom[0].size // 12)), "RING")
+ res[mask] = rgb[base.ang2pix(ptg)]
+ else:
+ ra = np.linspace(0, 2 * np.pi, dom[0].nlon + 1)
+ dec = pyHealpix.GL_thetas(dom[0].nlat)
+ ilat = _find_closest(dec, theta)
+ ilon = _find_closest(ra, phi)
+ ilon = np.where(ilon == dom[0].nlon, 0, ilon)
+ res[mask] = rgb[ilat * dom[0].nlon + ilon]
+ plt.axis('off')
+ plt.imshow(res, origin="lower")
return
- if isinstance(dom[0],(HPSpace, GLSpace)):
- import pyHealpix
- xsize = 800
- res, mask, theta, phi = _mollweide_helper(xsize)
- res = np.full(shape=res.shape+(3,), fill_value=1., dtype=np.float64)
- rgb = _rgb_data(f[0].to_global_data())
- if isinstance(dom[0], HPSpace):
- ptg = np.empty((phi.size, 2), dtype=np.float64)
- ptg[:, 0] = theta
- ptg[:, 1] = phi
- base = pyHealpix.Healpix_Base(int(np.sqrt(dom[0].size // 12)), "RING")
- res[mask] = rgb[base.ang2pix(ptg)]
- else:
- ra = np.linspace(0, 2 * np.pi, dom[0].nlon + 1)
- dec = pyHealpix.GL_thetas(dom[0].nlat)
- ilat = _find_closest(dec, theta)
- ilon = _find_closest(ra, phi)
- ilon = np.where(ilon == dom[0].nlon, 0, ilon)
- res[mask] = rgb[ilat * dom[0].nlon + ilon]
- plt.axis('off')
- plt.imshow(res, origin="lower")
- return
- else:
- dom = dom[0]
+ dom = dom[0]
if isinstance(dom, RGSpace):
if len(dom.shape) == 1:
npoints = dom.shape[0]
--
GitLab