René's URL Explorer Experiment

{"@context":"https://schema.org","@type":"DiscussionForumPosting","headline":"[MNT]: Turn ContourSet into a (nearly) plain Collection","articleBody":"### Summary\n\nCurrently, ContourSets are not Artists, which causes issues such as #6139 (essentially, the API is not uniform, and they do not appear directly in the draw tree).\r\n\r\nAt a low level, a ContourSet is represented as a list of PathCollections (the `.collections` attribute), with one such PathCollection per contour level value; the individual paths in the PathCollection are the connected components of that contour level.  But we could instead \"lift\" things up one level and make ContourSet directly inherit from Collection (actually inheriting from PathCollection doesn't really help; despite the name, PathCollection is more designed for \"resizable\" paths such as scatter plots with variable sizes), and use a single Path object for each contour level (using MOVETOs as needed if the Path needs to be broken in multiple connected components.  While slightly tedious, temporary backcompat with the old API could be provided via on-access creation of the \"old\" attributes, similarly to what was done in #24455.\r\n\r\nThere's actually (AFAICT) only one main difficulty with this approach, which is that draw_path_collection (called by Collection.draw) doesn't support hatching individual paths.  However, this could be implemented by still implementing a ContourSet.draw which directly calls draw_path_collection if no hatching is needed, and in the other case re-decomposes the collection as needed to emit the right set of draw_path with the right hatching set.\r\n\r\nattn @ianthomas23\n\n### Proposed fix\n\nA first patch (which goes on top of #25121) is provided below.\r\nQuite a few things (hatching, labeling, legends, etc.) are not implemented yet, but this at least allows one to call e.g. `contour(np.random.rand(5, 5))` and get a contour plot.\r\n```patch\r\ndiff --git a/doc/api/next_api_changes/deprecations/XXXXX-AL.rst b/doc/api/next_api_changes/deprecations/XXXXX-AL.rst\r\nnew file mode 100644\r\nindex 0000000000..caf6506209\r\n--- /dev/null\r\n+++ b/doc/api/next_api_changes/deprecations/XXXXX-AL.rst\r\n@@ -0,0 +1,4 @@\r\n+``QuadContourSet.allsegs`` and ``QuadContourSet.allkinds``\r\n+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~\r\n+These attributes are deprecated; directly retrieve the vertices and codes of\r\n+the Path objects in ``QuadContourSet.collections`` if necessary.\r\ndiff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py\r\nindex db25af57ac..d46238bd97 100644\r\n--- a/lib/matplotlib/axes/_base.py\r\n+++ b/lib/matplotlib/axes/_base.py\r\n@@ -2179,10 +2179,7 @@ class _AxesBase(martist.Artist):\r\n         _api.check_isinstance(\r\n             (mpl.contour.ContourSet, mcoll.Collection, mimage.AxesImage),\r\n             im=im)\r\n-        if isinstance(im, mpl.contour.ContourSet):\r\n-            if im.collections[0] not in self._children:\r\n-                raise ValueError(\"ContourSet must be in current Axes\")\r\n-        elif im not in self._children:\r\n+        if im not in self._children:\r\n             raise ValueError(\"Argument must be an image, collection, or \"\r\n                              \"ContourSet in this Axes\")\r\n         self._current_image = im\r\ndiff --git a/lib/matplotlib/contour.py b/lib/matplotlib/contour.py\r\nindex 060d9990b0..f24b888cc0 100644\r\n--- a/lib/matplotlib/contour.py\r\n+++ b/lib/matplotlib/contour.py\r\n@@ -3,7 +3,6 @@ Classes to support contour plotting and labelling for the Axes class.\r\n \"\"\"\r\n \r\n import functools\r\n-import itertools\r\n from numbers import Integral\r\n \r\n import numpy as np\r\n@@ -474,6 +473,7 @@ class ContourLabeler:\r\n \r\n         # calc_label_rot_and_inline() requires that (xmin, ymin)\r\n         # be a vertex in the path. So, if it isn't, add a vertex here\r\n+        # FIXME: Adapt to now API.\r\n         paths = self.collections[conmin].get_paths()  # paths of correct coll.\r\n         lc = paths[segmin].vertices  # vertices of correct segment\r\n         # Where should the new vertex be added in data-units?\r\n@@ -524,8 +524,9 @@ class ContourLabeler:\r\n                 self.labelCValueList,\r\n         )):\r\n \r\n+            # FIXME: Adapt to new API.\r\n             con = self.collections[icon]\r\n-            trans = con.get_transform()\r\n+            trans = self.get_transform()\r\n             lw = self._get_nth_label_width(idx)\r\n             additions = []\r\n             paths = con.get_paths()\r\n@@ -627,7 +628,7 @@ layers : array\r\n \r\n \r\n @_docstring.dedent_interpd\r\n-class ContourSet(cm.ScalarMappable, ContourLabeler):\r\n+class ContourSet(mcoll.Collection, ContourLabeler):\r\n     \"\"\"\r\n     Store a set of contour lines or filled regions.\r\n \r\n@@ -721,23 +722,28 @@ class ContourSet(cm.ScalarMappable, ContourLabeler):\r\n             Keyword arguments are as described in the docstring of\r\n             `~.Axes.contour`.\r\n         \"\"\"\r\n+        if antialiased is None and filled:\r\n+            # Eliminate artifacts; we are not stroking the boundaries.\r\n+            antialiased = False\r\n+            # The default for line contours will be taken from the\r\n+            # LineCollection default, which uses :rc:`lines.antialiased`.\r\n+        super().__init__(\r\n+            antialiaseds=antialiased,\r\n+            alpha=alpha,\r\n+            transform=transform,\r\n+        )\r\n         self.axes = ax\r\n         self.levels = levels\r\n         self.filled = filled\r\n-        self.linewidths = linewidths\r\n-        self.linestyles = linestyles\r\n+        # FIXME: We actually need a new draw() method which detects whether\r\n+        # hatches is set (for filled contours) and, if so, performs the right\r\n+        # decomposition to perform the draw as a series of draw_path (with\r\n+        # hatching) instead of the single call to draw_path_collection.\r\n         self.hatches = hatches\r\n-        self.alpha = alpha\r\n         self.origin = origin\r\n         self.extent = extent\r\n         self.colors = colors\r\n         self.extend = extend\r\n-        self.antialiased = antialiased\r\n-        if self.antialiased is None and self.filled:\r\n-            # Eliminate artifacts; we are not stroking the boundaries.\r\n-            self.antialiased = False\r\n-            # The default for line contours will be taken from the\r\n-            # LineCollection default, which uses :rc:`lines.antialiased`.\r\n \r\n         self.nchunk = nchunk\r\n         self.locator = locator\r\n@@ -758,8 +764,6 @@ class ContourSet(cm.ScalarMappable, ContourLabeler):\r\n         if self.origin == 'image':\r\n             self.origin = mpl.rcParams['image.origin']\r\n \r\n-        self._transform = transform\r\n-\r\n         self.negative_linestyles = negative_linestyles\r\n         # If negative_linestyles was not defined as a keyword argument, define\r\n         # negative_linestyles with rcParams\r\n@@ -803,85 +807,50 @@ class ContourSet(cm.ScalarMappable, ContourLabeler):\r\n                 if self._extend_max:\r\n                     cmap.set_over(self.colors[-1])\r\n \r\n-        self.collections = cbook.silent_list(None)\r\n-\r\n         # label lists must be initialized here\r\n         self.labelTexts = []\r\n         self.labelCValues = []\r\n \r\n-        kw = {'cmap': cmap}\r\n+        self.set_cmap(cmap)\r\n         if norm is not None:\r\n-            kw['norm'] = norm\r\n-        # sets self.cmap, norm if needed;\r\n-        cm.ScalarMappable.__init__(self, **kw)\r\n+            self.set_norm(norm)\r\n         if vmin is not None:\r\n             self.norm.vmin = vmin\r\n         if vmax is not None:\r\n             self.norm.vmax = vmax\r\n         self._process_colors()\r\n \r\n-        if getattr(self, 'allsegs', None) is None:\r\n-            self.allsegs, self.allkinds = self._get_allsegs_and_allkinds()\r\n-        elif self.allkinds is None:\r\n-            # allsegs specified in constructor may or may not have allkinds as\r\n-            # well.  Must ensure allkinds can be zipped below.\r\n-            self.allkinds = [None] * len(self.allsegs)\r\n-\r\n-        # Each entry in (allsegs, allkinds) is a list of (segs, kinds) which\r\n-        # specifies a list of Paths; but kinds can be None too in which case\r\n-        # all paths in that list are codeless.\r\n-        allpaths = [\r\n-            [*map(mpath.Path,\r\n-                  segs,\r\n-                  kinds if kinds is not None else itertools.repeat(None))]\r\n-            for segs, kinds in zip(self.allsegs, self.allkinds)]\r\n+        if self._paths is None:\r\n+            self._paths = self._make_paths_from_contour_generator()\r\n \r\n         if self.filled:\r\n-            if self.linewidths is not None:\r\n+            if linewidths is not None:\r\n                 _api.warn_external('linewidths is ignored by contourf')\r\n             # Lower and upper contour levels.\r\n             lowers, uppers = self._get_lowers_and_uppers()\r\n             # Default zorder taken from Collection\r\n             self._contour_zorder = kwargs.pop('zorder', 1)\r\n+            self.set(\r\n+                edgecolor=\"none\",\r\n+                zorder=self._contour_zorder,\r\n+            )\r\n \r\n-            self.collections[:] = [\r\n-                mcoll.PathCollection(\r\n-                    paths,\r\n-                    antialiaseds=(self.antialiased,),\r\n-                    edgecolors='none',\r\n-                    alpha=self.alpha,\r\n-                    transform=self.get_transform(),\r\n-                    zorder=self._contour_zorder)\r\n-                for level, level_upper, paths\r\n-                in zip(lowers, uppers, allpaths)]\r\n         else:\r\n-            self.tlinewidths = tlinewidths = self._process_linewidths()\r\n-            tlinestyles = self._process_linestyles()\r\n-            aa = self.antialiased\r\n-            if aa is not None:\r\n-                aa = (self.antialiased,)\r\n             # Default zorder taken from LineCollection, which is higher than\r\n             # for filled contours so that lines are displayed on top.\r\n             self._contour_zorder = kwargs.pop('zorder', 2)\r\n+            self.set(\r\n+                facecolor=\"none\",\r\n+                linewidths=self._process_linewidths(linewidths),\r\n+                linestyle=self._process_linestyles(linestyles),\r\n+                zorder=self._contour_zorder,\r\n+                label=\"_nolegend_\",\r\n+            )\r\n \r\n-            self.collections[:] = [\r\n-                mcoll.PathCollection(\r\n-                    paths,\r\n-                    facecolors=\"none\",\r\n-                    antialiaseds=aa,\r\n-                    linewidths=width,\r\n-                    linestyles=[lstyle],\r\n-                    alpha=self.alpha,\r\n-                    transform=self.get_transform(),\r\n-                    zorder=self._contour_zorder,\r\n-                    label='_nolegend_')\r\n-                for level, width, lstyle, paths\r\n-                in zip(self.levels, tlinewidths, tlinestyles, allpaths)]\r\n \r\n-        for col in self.collections:\r\n-            self.axes.add_collection(col, autolim=False)\r\n-            col.sticky_edges.x[:] = [self._mins[0], self._maxs[0]]\r\n-            col.sticky_edges.y[:] = [self._mins[1], self._maxs[1]]\r\n+        self.axes.add_collection(self, autolim=False)\r\n+        self.sticky_edges.x[:] = [self._mins[0], self._maxs[0]]\r\n+        self.sticky_edges.y[:] = [self._mins[1], self._maxs[1]]\r\n         self.axes.update_datalim([self._mins, self._maxs])\r\n         self.axes.autoscale_view(tight=True)\r\n \r\n@@ -893,6 +862,15 @@ class ContourSet(cm.ScalarMappable, ContourLabeler):\r\n                 \", \".join(map(repr, kwargs))\r\n             )\r\n \r\n+    allsegs = _api.deprecated(\"3.8\")(property(lambda self: [\r\n+        p.vertices for c in self.collections for p in c.get_paths()]))  # FIXME\r\n+    allkinds = _api.deprecated(\"3.8\")(property(lambda self: [\r\n+        p.codes for c in self.collections for p in c.get_paths()]))  # FIXME\r\n+\r\n+    # FIXME: Provide backcompat aliases.\r\n+    tlinewidths = ...\r\n+    tcolors = ...\r\n+\r\n     def get_transform(self):\r\n         \"\"\"Return the `.Transform` instance used by this ContourSet.\"\"\"\r\n         if self._transform is None:\r\n@@ -935,9 +913,10 @@ class ContourSet(cm.ScalarMappable, ContourLabeler):\r\n         artists = []\r\n         labels = []\r\n \r\n+        # FIXME: Adapt to new API.\r\n         if self.filled:\r\n             lowers, uppers = self._get_lowers_and_uppers()\r\n-            n_levels = len(self.collections)\r\n+            n_levels = len(self._path_collection.get_paths())\r\n \r\n             for i, (collection, lower, upper) in enumerate(\r\n                     zip(self.collections, lowers, uppers)):\r\n@@ -977,51 +956,63 @@ class ContourSet(cm.ScalarMappable, ContourLabeler):\r\n         Must set self.levels, self.zmin and self.zmax, and update axes limits.\r\n         \"\"\"\r\n         self.levels = args[0]\r\n-        self.allsegs = args[1]\r\n-        self.allkinds = args[2] if len(args) \u003e 2 else None\r\n+        allsegs = args[1]\r\n+        allkinds = args[2] if len(args) \u003e 2 else None\r\n         self.zmax = np.max(self.levels)\r\n         self.zmin = np.min(self.levels)\r\n \r\n+        if allkinds is None:\r\n+            allkinds = [[None] * len(segs) for segs in allsegs]\r\n+\r\n         # Check lengths of levels and allsegs.\r\n         if self.filled:\r\n-            if len(self.allsegs) != len(self.levels) - 1:\r\n+            if len(allsegs) != len(self.levels) - 1:\r\n                 raise ValueError('must be one less number of segments as '\r\n                                  'levels')\r\n         else:\r\n-            if len(self.allsegs) != len(self.levels):\r\n+            if len(allsegs) != len(self.levels):\r\n                 raise ValueError('must be same number of segments as levels')\r\n \r\n         # Check length of allkinds.\r\n-        if (self.allkinds is not None and\r\n-                len(self.allkinds) != len(self.allsegs)):\r\n+        if len(allkinds) != len(allsegs):\r\n             raise ValueError('allkinds has different length to allsegs')\r\n \r\n         # Determine x, y bounds and update axes data limits.\r\n-        flatseglist = [s for seg in self.allsegs for s in seg]\r\n+        flatseglist = [s for seg in allsegs for s in seg]\r\n         points = np.concatenate(flatseglist, axis=0)\r\n         self._mins = points.min(axis=0)\r\n         self._maxs = points.max(axis=0)\r\n \r\n+        # Each entry in (allsegs, allkinds) is a list of (segs, kinds) which\r\n+        # gets concatenated to construct a Path.\r\n+        self._paths = [\r\n+            mpath.Path.make_compound_path(*map(mpath.Path, segs, kinds))\r\n+            for segs, kinds in zip(allsegs, allkinds)]\r\n+\r\n         return kwargs\r\n \r\n-    def _get_allsegs_and_allkinds(self):\r\n-        \"\"\"Compute ``allsegs`` and ``allkinds`` using C extension.\"\"\"\r\n-        allsegs = []\r\n-        allkinds = []\r\n+    def _make_paths_from_contour_generator(self):\r\n+        \"\"\"Compute ``paths`` using C extension.\"\"\"\r\n+        if self._paths is not None:\r\n+            return self._paths\r\n+        paths = []\r\n+        empty_path = mpath.Path(np.empty((0, 2)))\r\n         if self.filled:\r\n             lowers, uppers = self._get_lowers_and_uppers()\r\n             for level, level_upper in zip(lowers, uppers):\r\n                 vertices, kinds = \\\r\n                     self._contour_generator.create_filled_contour(\r\n                         level, level_upper)\r\n-                allsegs.append(vertices)\r\n-                allkinds.append(kinds)\r\n+                paths.append(\r\n+                    mpath.Path(np.concatenate(vertices), np.concatenate(kinds))\r\n+                    if len(vertices) else empty_path)\r\n         else:\r\n             for level in self.levels:\r\n                 vertices, kinds = self._contour_generator.create_contour(level)\r\n-                allsegs.append(vertices)\r\n-                allkinds.append(kinds)\r\n-        return allsegs, allkinds\r\n+                paths.append(\r\n+                    mpath.Path(np.concatenate(vertices), np.concatenate(kinds))\r\n+                    if len(vertices) else empty_path)\r\n+        return paths\r\n \r\n     def _get_lowers_and_uppers(self):\r\n         \"\"\"\r\n@@ -1048,18 +1039,11 @@ class ContourSet(cm.ScalarMappable, ContourLabeler):\r\n         # so if vmin/vmax are not set yet, this would override them with\r\n         # content from *cvalues* rather than levels like we want\r\n         self.norm.autoscale_None(self.levels)\r\n-        tcolors = [(tuple(rgba),)\r\n-                   for rgba in self.to_rgba(self.cvalues, alpha=self.alpha)]\r\n-        self.tcolors = tcolors\r\n-        hatches = self.hatches * len(tcolors)\r\n-        for color, hatch, collection in zip(tcolors, hatches,\r\n-                                            self.collections):\r\n-            if self.filled:\r\n-                collection.set_facecolor(color)\r\n-                # update the collection's hatch (may be None)\r\n-                collection.set_hatch(hatch)\r\n-            else:\r\n-                collection.set_edgecolor(color)\r\n+        tcolors = self.to_rgba(self.cvalues, alpha=self.alpha)\r\n+        if self.filled:\r\n+            self.set_facecolors(tcolors)\r\n+        else:\r\n+            self.set_edgecolors(tcolors)\r\n         for label, cv in zip(self.labelTexts, self.labelCValues):\r\n             label.set_alpha(self.alpha)\r\n             label.set_color(self.labelMappable.to_rgba(cv))\r\n@@ -1214,16 +1198,13 @@ class ContourSet(cm.ScalarMappable, ContourLabeler):\r\n         if self.extend in ('both', 'max', 'min'):\r\n             self.norm.clip = False\r\n \r\n-        # self.tcolors are set by the \"changed\" method\r\n-\r\n-    def _process_linewidths(self):\r\n-        linewidths = self.linewidths\r\n+    def _process_linewidths(self, linewidths):\r\n         Nlev = len(self.levels)\r\n         if linewidths is None:\r\n             default_linewidth = mpl.rcParams['contour.linewidth']\r\n             if default_linewidth is None:\r\n                 default_linewidth = mpl.rcParams['lines.linewidth']\r\n-            tlinewidths = [(default_linewidth,)] * Nlev\r\n+            tlinewidths = [default_linewidth] * Nlev\r\n         else:\r\n             if not np.iterable(linewidths):\r\n                 linewidths = [linewidths] * Nlev\r\n@@ -1234,11 +1215,10 @@ class ContourSet(cm.ScalarMappable, ContourLabeler):\r\n                     linewidths = linewidths * nreps\r\n                 if len(linewidths) \u003e Nlev:\r\n                     linewidths = linewidths[:Nlev]\r\n-            tlinewidths = [(w,) for w in linewidths]\r\n+            tlinewidths = [w for w in linewidths]\r\n         return tlinewidths\r\n \r\n-    def _process_linestyles(self):\r\n-        linestyles = self.linestyles\r\n+    def _process_linestyles(self, linestyles):\r\n         Nlev = len(self.levels)\r\n         if linestyles is None:\r\n             tlinestyles = ['solid'] * Nlev\r\n@@ -1319,7 +1299,7 @@ class ContourSet(cm.ScalarMappable, ContourLabeler):\r\n             raise ValueError(\"Method does not support filled contours.\")\r\n \r\n         if indices is None:\r\n-            indices = range(len(self.collections))\r\n+            indices = range(len(self._paths))\r\n \r\n         d2min = np.inf\r\n         conmin = None\r\n@@ -1330,6 +1310,7 @@ class ContourSet(cm.ScalarMappable, ContourLabeler):\r\n \r\n         point = np.array([x, y])\r\n \r\n+        # FIXME: Adapt to new API.\r\n         for icon in indices:\r\n             con = self.collections[icon]\r\n             trans = con.get_transform()\r\n@@ -1352,11 +1333,6 @@ class ContourSet(cm.ScalarMappable, ContourLabeler):\r\n \r\n         return (conmin, segmin, imin, xmin, ymin, d2min)\r\n \r\n-    def remove(self):\r\n-        super().remove()\r\n-        for coll in self.collections:\r\n-            coll.remove()\r\n-\r\n \r\n @_docstring.dedent_interpd\r\n class QuadContourSet(ContourSet):\r\n```","author":{"url":"https://github.com/anntzer","@type":"Person","name":"anntzer"},"datePublished":"2023-02-02T00:23:56.000Z","interactionStatistic":{"@type":"InteractionCounter","interactionType":"https://schema.org/CommentAction","userInteractionCount":9},"url":"https://github.com/25128/matplotlib/issues/25128"}