def do_3d_projection(self, renderer):
"""
Perform the 3D projection for this object.
"""
# FIXME: This may no longer be needed?
if self._A is not None:
self.update_scalarmappable()
self._facecolors3d = self._facecolors
txs, tys, tzs = proj3d.proj_transform_vec(self._vec, renderer.M)
xyzlist = [(txs[si:ei], tys[si:ei], tzs[si:ei])
for si, ei in self._segis]
# This extra fuss is to re-order face / edge colors
cface = self._facecolors3d
cedge = self._edgecolors3d
if len(cface) != len(xyzlist):
cface = cface.repeat(len(xyzlist), axis=0)
if len(cedge) != len(xyzlist):
if len(cedge) == 0:
cedge = cface
else:
cedge = cedge.repeat(len(xyzlist), axis=0)
# if required sort by depth (furthest drawn first)
if self._zsort:
z_segments_2d = sorted(
((self._zsortfunc(zs), np.column_stack([xs, ys]), fc, ec, idx)
for idx, ((xs, ys, zs), fc,
ec) in enumerate(zip(xyzlist, cface, cedge))),
key=lambda x: x[0],
reverse=True)
else:
raise ValueError("whoops")
segments_2d = [s for z, s, fc, ec, idx in z_segments_2d]
if self._codes3d is not None:
codes = [self._codes3d[idx] for z, s, fc, ec, idx in z_segments_2d]
PolyCollection.set_verts_and_codes(self, segments_2d, codes)
else:
PolyCollection.set_verts(self, segments_2d, self._closed)
self._facecolors2d = [fc for z, s, fc, ec, idx in z_segments_2d]
if len(self._edgecolors3d) == len(cface):
self._edgecolors2d = [ec for z, s, fc, ec, idx in z_segments_2d]
else:
self._edgecolors2d = self._edgecolors3d
# Return zorder value
if self._sort_zpos is not None:
zvec = np.array([[0], [0], [self._sort_zpos], [1]])
ztrans = proj3d.proj_transform_vec(zvec, renderer.M)
return ztrans[2][0]
elif tzs.size > 0:
# FIXME: Some results still don't look quite right.
# In particular, examine contourf3d_demo2.py
# with az = -54 and elev = -45.
return np.min(tzs)
else:
return np.nan
def test_proj3d_deprecated():
with pytest.warns(MatplotlibDeprecationWarning):
proj3d.line2d([0, 1], [0, 1])
with pytest.warns(MatplotlibDeprecationWarning):
proj3d.line2d_dist([0, 1, 3], [0, 1])
with pytest.warns(MatplotlibDeprecationWarning):
proj3d.mod([1, 1, 1])
vec = np.arange(4)
M = np.ones((4, 4))
with pytest.warns(MatplotlibDeprecationWarning):
proj3d.proj_transform_vec(vec, M)
with pytest.warns(MatplotlibDeprecationWarning):
proj3d.proj_transform_vec_clip(vec, M)
with pytest.warns(MatplotlibDeprecationWarning):
proj3d.vec_pad_ones(np.ones(3), np.ones(3), np.ones(3))
with pytest.warns(MatplotlibDeprecationWarning):
proj3d.proj_trans_clip_points(np.ones((4, 3)), M)
def test_proj3d_deprecated():
with pytest.warns(MatplotlibDeprecationWarning):
proj3d.line2d([0, 1], [0, 1])
with pytest.warns(MatplotlibDeprecationWarning):
proj3d.line2d_dist([0, 1, 3], [0, 1])
with pytest.warns(MatplotlibDeprecationWarning):
proj3d.mod([1, 1, 1])
vec = np.arange(4)
M = np.ones((4, 4))
with pytest.warns(MatplotlibDeprecationWarning):
proj3d.proj_transform_vec(vec, M)
with pytest.warns(MatplotlibDeprecationWarning):
proj3d.proj_transform_vec_clip(vec, M)
with pytest.warns(MatplotlibDeprecationWarning):
proj3d.vec_pad_ones(np.ones(3), np.ones(3), np.ones(3))
with pytest.warns(MatplotlibDeprecationWarning):
proj3d.proj_trans_clip_points(np.ones((4, 3)), M)