Implement graph components and archetypes

ARCHITECTURE.md

@@ -134,6 +134,7 @@ Update instructions:
 | re_space_view               | Types & utilities for defining Space View classes and communicating with the Viewport.                     |
 | re_space_view_bar_chart     | A Space View that shows a single bar chart.                                                                |
 | re_space_view_dataframe     | A Space View that shows the data contained in entities in a table.                                         |
+| re_space_view_graph         | A Space View that shows a graph (node-link diagram).                                                       |
 | re_space_view_map           | A Space View that shows geospatial data on a map.                                                          |
 | re_space_view_spatial       | Space Views that show entities in a 2D or 3D spatial relationship.                                         |
 | re_space_view_tensor        | A Space View dedicated to visualizing tensors with arbitrary dimensionality.                               |

Cargo.lock

@@ -2376,6 +2376,12 @@ version = "0.4.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "0ce7134b9999ecaf8bcd65542e436736ef32ddca1b3e06094cb6ec5755203b80"
 
+[[package]]
+name = "fjadra"
+version = "0.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ccc0416b27f53bba6c9bd564260f27d4784c5f430926eb16a519356be7d66bbc"
+
 [[package]]
 name = "flatbuffers"
 version = "23.5.26"
@@ -2754,6 +2760,25 @@ dependencies = [
  "bitflags 2.6.0",
 ]
 
+[[package]]
+name = "graph_binary_tree"
+version = "0.21.0-alpha.1+dev"
+dependencies = [
+ "anyhow",
+ "clap",
+ "rerun",
+]
+
+[[package]]
+name = "graph_lattice"
+version = "0.21.0-alpha.1+dev"
+dependencies = [
+ "anyhow",
+ "clap",
+ "itertools 0.13.0",
+ "rerun",
+]
+
 [[package]]
 name = "h2"
 version = "0.3.26"
@@ -6052,6 +6077,28 @@ dependencies = [
  "thiserror",
 ]
 
+[[package]]
+name = "re_space_view_graph"
+version = "0.21.0-alpha.1+dev"
+dependencies = [
+ "ahash",
+ "bytemuck",
+ "egui",
+ "fjadra",
+ "nohash-hasher",
+ "re_chunk",
+ "re_format",
+ "re_log_types",
+ "re_query",
+ "re_renderer",
+ "re_space_view",
+ "re_tracing",
+ "re_types",
+ "re_ui",
+ "re_viewer_context",
+ "re_viewport_blueprint",
+]
+
 [[package]]
 name = "re_space_view_map"
 version = "0.21.0-alpha.1+dev"
@@ -6462,6 +6509,7 @@ dependencies = [
  "re_smart_channel",
  "re_space_view_bar_chart",
  "re_space_view_dataframe",
+ "re_space_view_graph",
  "re_space_view_map",
  "re_space_view_spatial",
  "re_space_view_tensor",

Cargo.toml

@@ -99,6 +99,7 @@ re_space_view = { path = "crates/viewer/re_space_view", version = "=0.21.0-alpha
 re_space_view_bar_chart = { path = "crates/viewer/re_space_view_bar_chart", version = "=0.21.0-alpha.1", default-features = false }
 re_space_view_spatial = { path = "crates/viewer/re_space_view_spatial", version = "=0.21.0-alpha.1", default-features = false }
 re_space_view_dataframe = { path = "crates/viewer/re_space_view_dataframe", version = "=0.21.0-alpha.1", default-features = false }
+re_space_view_graph = { path = "crates/viewer/re_space_view_graph", version = "=0.21.0-alpha.1", default-features = false }
 re_space_view_map = { path = "crates/viewer/re_space_view_map", version = "=0.21.0-alpha.1", default-features = false }
 re_space_view_tensor = { path = "crates/viewer/re_space_view_tensor", version = "=0.21.0-alpha.1", default-features = false }
 re_space_view_text_document = { path = "crates/viewer/re_space_view_text_document", version = "=0.21.0-alpha.1", default-features = false }
@@ -186,6 +187,7 @@ enumset = "1.0.12"
 env_logger = { version = "0.10", default-features = false }
 ffmpeg-sidecar = { version = "2.0.2", default-features = false }
 fixed = { version = "1.28", default-features = false }
+fjadra = "0.1"
 flatbuffers = "23.0"
 futures-channel = "0.3"
 futures-util = { version = "0.3", default-features = false }

crates/store/re_types/definitions/rerun/archetypes/graph_edges.fbs

@@ -0,0 +1,33 @@
+namespace rerun.archetypes;
+
+// ---
+
+// TODO(ab): Add images to snippets.
+
+/// A list of edges in a graph.
+///
+/// By default, edges are undirected.
+///
+/// \example archetypes/graph_undirected !api title="Simple undirected graph" image=""
+/// \example archetypes/graph_directed !api title="Simple directed graph" image=""
+table GraphEdges (
+  "attr.docs.category": "Graph",
+  "attr.docs.unreleased",
+  "attr.docs.view_types": "GraphView",
+  "attr.rust.derive": "PartialEq, Eq",
+  "attr.rerun.experimental"
+) {
+  // --- Required ---
+
+  /// A list of node tuples.
+  edges: [rerun.components.GraphEdge] ("attr.rerun.component_required", order: 1000);
+
+
+  // --- Recommended ---
+
+  /// Specifies if the graph is directed or undirected.
+  ///
+  /// If no `GraphType` is provided, the graph is assumed to be undirected.
+  graph_type: rerun.components.GraphType ("attr.rerun.component_recommended", nullable, order: 2000);
+
+}

crates/store/re_types/definitions/rerun/archetypes/graph_nodes.fbs

@@ -0,0 +1,39 @@
+namespace rerun.archetypes;
+
+// ---
+
+// TODO(ab): Add images to snippets.
+
+/// A list of nodes in a graph with optional labels, colors, etc.
+///
+/// \example archetypes/graph_undirected !api title="Simple undirected graph" image=""
+/// \example archetypes/graph_directed !api title="Simple directed graph" image=""
+table GraphNodes (
+  "attr.docs.category": "Graph",
+  "attr.docs.unreleased",
+  "attr.docs.view_types": "GraphView",
+  "attr.rust.derive": "PartialEq",
+  "attr.rerun.experimental"
+) {
+  // --- Required ---
+
+  /// A list of node IDs.
+  node_ids: [rerun.components.GraphNode] ("attr.rerun.component_required", order: 1000);
+
+  // --- Optional ---
+
+  /// Optional center positions of the nodes.
+  positions: [rerun.components.Position2D] ("attr.rerun.component_optional", nullable, order: 3000);
+
+  /// Optional colors for the boxes.
+  colors: [rerun.components.Color] ("attr.rerun.component_optional", nullable, order: 3100);
+
+  /// Optional text labels for the node.
+  labels: [rerun.components.Text] ("attr.rerun.component_optional", nullable, order: 3200);
+
+  /// Optional choice of whether the text labels should be shown by default.
+  show_labels: rerun.components.ShowLabels ("attr.rerun.component_optional", nullable, order: 3250);
+
+  /// Optional radii for nodes.
+  radii: [rerun.components.Radius] ("attr.rerun.component_optional", nullable, order: 3300);
+}

crates/store/re_types/definitions/rerun/blueprint/views/graph.fbs

@@ -0,0 +1,14 @@
+namespace rerun.blueprint.views;
+
+/// A graph view to display time-variying, directed or undirected graph visualization.
+///
+/// \example views/graph title="Use a blueprint to create a graph view." image="https://static.rerun.io/graph_lattice/f9169da9c3f35b7260c9d74cd5be5fe710aec6a8/1200w.png"
+table GraphView (
+    "attr.rerun.view_identifier": "Graph",
+    "attr.docs.unreleased"
+) {
+    /// Everything within these bounds is guaranteed to be visible.
+    ///
+    /// Somethings outside of these bounds may also be visible due to letterboxing.
+    visual_bounds: rerun.blueprint.archetypes.VisualBounds2D (order: 1000);
+}

crates/store/re_types/definitions/rerun/components/graph_edge.fbs

@@ -0,0 +1,12 @@
+namespace rerun.components;
+
+// ---
+
+/// An edge in a graph connecting two nodes.
+table GraphEdge (
+  "attr.docs.unreleased",
+  "attr.rust.derive": "Default, PartialEq, Eq, PartialOrd, Ord",
+  "attr.rust.repr": "transparent"
+) {
+  edge: rerun.datatypes.Utf8Pair (order: 100);
+}

crates/store/re_types/definitions/rerun/components/graph_node.fbs

@@ -0,0 +1,14 @@
+namespace rerun.components;
+
+// ---
+
+/// A string-based ID representing a node in a graph.
+table GraphNode (
+  "attr.docs.unreleased",
+  "attr.python.aliases": "str",
+  "attr.python.array_aliases": "str, Sequence[str]",
+  "attr.rust.derive": "Default, PartialEq, Eq, PartialOrd, Ord, Hash",
+  "attr.rust.repr": "transparent"
+) {
+  id: rerun.datatypes.Utf8 (order: 100);
+}

crates/store/re_types/definitions/rerun/components/graph_type.fbs

@@ -0,0 +1,18 @@
+namespace rerun.components;
+
+// --
+
+/// Specifies if a graph has directed or undirected edges.
+enum GraphType: ubyte (
+  "attr.docs.unreleased",
+  "attr.rust.derive": "Default, PartialEq, Eq"
+) {
+  /// Invalid value. Won't show up in generated types.
+  Invalid = 0,
+
+  /// The graph has undirected edges.
+  Undirected (default),
+
+  /// The graph has directed edges.
+  Directed,
+}

crates/store/re_types/definitions/rerun/datatypes/utf8_pair.fbs

@@ -0,0 +1,15 @@
+namespace rerun.datatypes;
+
+/// Stores a tuple of UTF-8 strings.
+table Utf8Pair (
+  "attr.docs.unreleased",
+  "attr.python.aliases": "Tuple[datatypes.Utf8Like, datatypes.Utf8Like]",
+  "attr.python.array_aliases": "npt.NDArray[np.str_]",
+  "attr.rust.derive": "Default, PartialEq, Eq, PartialOrd, Ord"
+) {
+  /// The first string.
+  first: rerun.datatypes.Utf8 (order: 100);
+
+  /// The second string.
+  second: rerun.datatypes.Utf8 (order: 200);
+}