Method do_3d_projection

lib/mpl_toolkits/mplot3d/art3d.py:528–587

Project the points according to renderer matrix.

(self)

        super().set_segments([])

    def do_3d_projection(self):
        """
        Project the points according to renderer matrix.
        """
        segments = self._segments3d

        # Handle ragged inputs, but prefer a faster path for same-length segments
        segment_lengths = [len(segment) for segment in segments]
        ragged = len(set(segment_lengths)) > 1
        if ragged:
            # Branch masked / non-masked for speed
            if any(np.ma.isMA(segment) for segment in segments):
                segments = np.ma.concatenate(segments)
            else:
                segments = np.concatenate(segments)
        else:
            segments = np.asanyarray(segments)

        # Handle empty segments
        if segments.size == 0:
            LineCollection.set_segments(self, [])
            return np.nan

        mask = False
        if np.ma.isMA(segments) and segments.mask is not np.ma.nomask:
            mask = segments.mask

        scale_mask = _scale_invalid_mask(segments[..., 0],
                                         segments[..., 1],
                                         segments[..., 2],
                                         self.axes)
        if np.any(scale_mask):
            mask |= np.broadcast_to(scale_mask[..., np.newaxis],
                                    (*scale_mask.shape, 3))

        if self._axlim_clip:
            viewlim_mask = _viewlim_mask(segments[..., 0],
                                         segments[..., 1],
                                         segments[..., 2],
                                         self.axes)
            if np.any(viewlim_mask):
                # broadcast mask to 3D
                viewlim_mask = np.broadcast_to(viewlim_mask[..., np.newaxis],
                                               (*viewlim_mask.shape, 3))
                mask = mask | viewlim_mask

        xyzs = proj3d._scale_proj_transform_vectors(segments, self.axes)
        if mask is not False:
            xyzs = np.ma.array(xyzs, mask=mask)
        segments_2d = xyzs[..., 0:2]
        if ragged:
            segments_2d = np.split(segments_2d, np.cumsum(segment_lengths[:-1]))
        LineCollection.set_segments(self, segments_2d)

        # FIXME
        if len(xyzs) > 0:
            minz = min(xyzs[..., 2].min(), 1e9)
        else: