Method _plot_points

phi/vis/_matplotlib/_matplotlib_plots.py:612–760

(axis: Axes, data: Field, dims: tuple, vector: Shape, color: Tensor, alpha: Tensor, err: Tensor, min_val, max_val, label)

    @staticmethod
    def _plot_points(axis: Axes, data: Field, dims: tuple, vector: Shape, color: Tensor, alpha: Tensor, err: Tensor, min_val, max_val, label):
        connected = spatial(data.points)
        if isinstance(data.sampled_elements, GeometryStack):
            for idx in data.sampled_elements.object_dims.meshgrid():
                PointCloud2D._plot_points(axis, data[idx], dims, vector, color[idx], alpha[idx], err[idx], min_val, max_val, label)
            return
        elif isinstance(data.sampled_elements, Intersection):
            math.assert_close(data.values, math.NAN, msg="Intersections can only be plotted as Geometries, not Fields.")
            sdf = CenteredGrid(data.sampled_elements.approximate_signed_distance, bounds=data.sampled_elements.bounding_box().scaled(1.).corner_representation(), **{d: 32 for d in dims})
            sdf_grid = SDFGrid(sdf.values, sdf.bounds, approximate_outside=False)
            data = Field(sdf_grid, math.NAN, 0)
        data = only_stored_elements(data)
        x, y = reshaped_numpy(data.points.vector[dims], ['vector', non_channel(data)])
        if wrap(color == 'cmap').all:
            values = reshaped_numpy(data.values, [non_channel(data)])
            mpl_colors = add_color_bar(axis, values, min_val, max_val)
            single_color = False
        elif non_channel(data).only(color.shape) and color.dtype.kind == float:  # use color map
            values = reshaped_numpy(color, [non_channel(data)])
            mpl_colors = add_color_bar(axis, values, None, None)
            single_color = False
        else:
            mpl_colors = matplotlib_colors(color, non_channel(data), default=0)
            single_color = True
        alphas = reshaped_numpy(alpha, [non_channel(data)])
        if isinstance(data.geometry, Point):
            if spatial(data.points).is_empty:
                axis.scatter(x, y, color=mpl_colors, s=6 ** 2, alpha=alphas)
                if (err != 0).any:
                    x_err = reshaped_numpy(err.vector[dims[0]], [instance(data)]) if dims[0] in err.vector.item_names else 0
                    y_err = reshaped_numpy(err.vector[dims[1]], [instance(data)]) if dims[1] in err.vector.item_names else 0
                    if all(np.all(c == mpl_colors[0]) for c in mpl_colors) and all(a == alphas[0] for a in alphas):
                        axis.errorbar(x, y, y_err, x_err, fmt=' ', color=mpl_colors[0], alpha=alphas[0])
                    else:
                        for x_, y_, y_err_, x_err_, col_, alpha_ in zip(x, y, y_err, x_err, mpl_colors, alphas):
                            axis.errorbar(x_, y_, y_err_, x_err_, fmt=' ', color=col_, alpha=alpha_)
        else:
            if isinstance(data.geometry, Sphere):
                rad = reshaped_numpy(data.geometry.bounding_radius(), [data.shape.non_channel])
                shapes = [plt.Circle((xi, yi), radius=ri, linewidth=0, alpha=a, facecolor=ci) for xi, yi, ri, ci, a in zip(x, y, rad, mpl_colors, alphas)]
                axis.add_collection(matplotlib.collections.PatchCollection(shapes, match_original=True))
            elif isinstance(data.geometry, BaseBox):
                half_size = data.geometry.half_size
                min_len = axis.get_ylim()[1] - axis.get_ylim()[0] + axis.get_xlim()[1] - axis.get_xlim()[0]
                half_size = math.where(math.is_finite(half_size), half_size, min_len)
                w2, h2 = reshaped_numpy(half_size, ['vector', data.shape.non_channel])
                if data.geometry.rotation_matrix is None:
                    angles = w2 * 0.
                    lower_x = x - w2
                    lower_y = y - h2
                else:
                    angles = reshaped_numpy(rotation_angles(data.geometry.rotation_matrix), [data.shape.non_channel])
                    lower_x, lower_y = reshaped_numpy(data.geometry.center - rotate(data.geometry.half_size, data.geometry.rotation_matrix), ['vector', data.shape.non_channel])
                shapes = [plt.Rectangle((lxi, lyi), w2i * 2, h2i * 2, angle=ang*180/np.pi, linewidth=1, edgecolor='white', alpha=a, facecolor=ci) for lxi, lyi, w2i, h2i, ang, ci, a in zip(lower_x, lower_y, w2, h2, angles, mpl_colors, alphas)]
                axis.add_collection(matplotlib.collections.PatchCollection(shapes, match_original=True))
            elif isinstance(data.geometry, Mesh):
                edgecolor = 'white' if single_color else None
                csr = data.mesh.elements.numpy().tocsr()