def test_all_corner_coordinates_from_map(sub_smap):
coordinates = all_corner_coords_from_map(sub_smap)
shape = sub_smap.data.shape
assert coordinates.shape == (shape[0] + 1, shape[1] + 1)
assert isinstance(coordinates, SkyCoord)
assert isinstance(coordinates.frame, BaseCoordinateFrame)
assert coordinates.frame.name == sub_smap.coordinate_frame.name
def create_map(self, m):
"""
Create a mesh from a map.
Parameters
----------
m : sunpy.map.Map
Returns
-------
pyvista.StructuredGrid
"""
corner_coords = all_corner_coords_from_map(m)
nodes = self._coords_to_xyz(corner_coords.ravel())
return nodes
def test_all_corner_coordinates_from_map(sub_smap):
coordinates = all_corner_coords_from_map(sub_smap)
shape = sub_smap.data.shape
assert coordinates.shape == (shape[0] + 1, shape[1] + 1)
assert isinstance(coordinates, SkyCoord)
assert isinstance(coordinates.frame, BaseCoordinateFrame)
assert coordinates.frame.name == sub_smap.coordinate_frame.name
xpix, ypix = sub_smap.world_to_pixel(coordinates[0, 0])
assert_quantity_allclose(xpix, -0.5 * u.pix)
assert_quantity_allclose(ypix, -0.5 * u.pix)
xpix, ypix = sub_smap.world_to_pixel(coordinates[-1, -1])
assert_quantity_allclose(xpix, sub_smap.dimensions[0] - 0.5 * u.pix)
assert_quantity_allclose(ypix, sub_smap.dimensions[1] - 0.5 * u.pix)
def _pyvista_mesh(self, m):
"""
Create a mesh from a map.
Parameters
----------
m : sunpy.map.Map
Returns
-------
pyvista.StructuredGrid
"""
corner_coords = all_corner_coords_from_map(m)
nodes = self._coords_to_xyz(corner_coords.ravel())
grid = pv.StructuredGrid()
grid.points = nodes
grid.dimensions = [m.data.shape[0] + 1, m.data.shape[1] + 1, 1]
data = m.data.T.reshape(-1)
grid['data'] = m.plot_settings['norm'](data)
return grid