add rev_order anchor for Studio

Author: mattseddon

src/dvc_render/vega.py

@@ -132,6 +132,7 @@ def get_partial_filled_template(self):
             split_anchors=[
                 "color",
                 "data",
+                "rev_order",
                 "shape",
                 "stroke_dash",
                 "tooltip",
@@ -208,6 +209,7 @@ def _process_optional_anchors(self, split_anchors: List[str]):
                 "group_by_y",
                 "group_by",
                 "pivot_field",
+                "rev_order",
                 "row",
                 "shape",
                 "stroke_dash",
@@ -221,6 +223,7 @@ def _process_optional_anchors(self, split_anchors: List[str]):
 
         self._fill_color(split_anchors, optional_anchors)
         self._fill_zoom_and_pan(split_anchors, optional_anchors)
+        self._fill_rev_order(split_anchors, optional_anchors)
 
         y_definitions = self.properties.get("anchors_y_definitions", [])
         is_single_source = len(y_definitions) <= 1
@@ -245,17 +248,25 @@ def _fill_color(self, split_anchors: List[str], optional_anchors: List[str]):
 
     def _fill_zoom_and_pan(self, split_anchors: List[str], optional_anchors: List[str]):
         name = "zoom_and_pan"
-        encoding = {"name": "grid", "select": "interval", "bind": "scales"}
-        if "zoom_and_pan" not in optional_anchors:
+        if name not in optional_anchors:
             return
 
-        optional_anchors.remove("zoom_and_pan")
+        optional_anchors.remove(name)
 
-        if name in split_anchors:
-            self._set_split_content(name, encoding)
+        self._fill_split_anchor(
+            split_anchors,
+            name,
+            {"name": "grid", "select": "interval", "bind": "scales"},
+        )
+
+    def _fill_rev_order(self, split_anchors: List[str], optional_anchors: List[str]):
+        name = "rev_order"
+        if name not in optional_anchors:
             return
 
-        self.template.fill_anchor(name, encoding)
+        optional_anchors.remove(name)
+
+        self._fill_split_anchor(split_anchors, name, self.get_revs())
 
     def _process_single_source_plot(
         self, split_anchors: List[str], optional_anchors: List[str]
@@ -421,6 +432,9 @@ def _fill_optional_anchor_mapping(
 
         encoding = self._get_optional_anchor_mapping(field, name, domain)
 
+        self._fill_split_anchor(split_anchors, name, encoding)
+
+    def _fill_split_anchor(self, split_anchors: List[str], name: str, encoding: Any):
         if name in split_anchors:
             self._set_split_content(name, encoding)
             return

src/dvc_render/vega_templates.py

@@ -200,7 +200,10 @@ class ConfusionTemplate(Template):
         "$schema": "https://vega.github.io/schema/vega-lite/v5.json",
         "data": {"values": Template.anchor("data")},
         "title": Template.anchor("title"),
-        "facet": {"column": {"field": "rev"}, "row": Template.anchor("row")},
+        "facet": {
+            "column": {"field": "rev", "sort": Template.anchor("rev_order")},
+            "row": Template.anchor("row"),
+        },
         "spec": {
             "transform": [
                 {
@@ -310,7 +313,10 @@ class NormalizedConfusionTemplate(Template):
         "$schema": "https://vega.github.io/schema/vega-lite/v5.json",
         "data": {"values": Template.anchor("data")},
         "title": Template.anchor("title"),
-        "facet": {"column": {"field": "rev"}, "row": Template.anchor("row")},
+        "facet": {
+            "column": {"field": "rev", "sort": Template.anchor("rev_order")},
+            "row": Template.anchor("row"),
+        },
         "spec": {
             "transform": [
                 {