Add range-query-engine derived interval information to the ir_viz csvs/protos.

These are useful for giving another perspective on what our analysis believes are valid values for nodes.

PiperOrigin-RevId: 720319354

Author: allight

xls/visualization/ir_viz/BUILD

@@ -121,7 +121,9 @@ cc_library(
         "//xls/ir:op",
         "//xls/passes:bdd_function",
         "//xls/passes:bdd_query_engine",
+        "//xls/passes:proc_state_range_query_engine",
         "//xls/passes:query_engine",
+        "//xls/passes:range_query_engine",
         "//xls/passes:ternary_query_engine",
         "//xls/passes:union_query_engine",
         "//xls/scheduling:pipeline_schedule",

xls/visualization/ir_viz/ir_to_csvs_main.cc

@@ -96,7 +96,8 @@ constexpr riegeli::CsvHeaderConstant kNodeHeader = {"name",
                                                     "initial_value",
                                                     "area_um",
                                                     "file",
-                                                    "line"};
+                                                    "line",
+                                                    "range"};
 riegeli::CsvRecord NodeRecord(const viz::Node& node) {
   return riegeli::CsvRecord(
       *kNodeHeader,
@@ -139,6 +140,7 @@ riegeli::CsvRecord NodeRecord(const viz::Node& node) {
               ? node.loc()[0].file()
               : "",
           node.loc_size() == 1 ? ToFieldValue(node.loc()[0].line()) : "",
+          node.attributes().has_ranges() ? node.attributes().ranges() : "",
       });
 }
 

xls/visualization/ir_viz/ir_to_proto.cc

@@ -47,7 +47,9 @@
 #include "xls/ir/proc.h"
 #include "xls/passes/bdd_function.h"
 #include "xls/passes/bdd_query_engine.h"
+#include "xls/passes/proc_state_range_query_engine.h"
 #include "xls/passes/query_engine.h"
+#include "xls/passes/range_query_engine.h"
 #include "xls/passes/ternary_query_engine.h"
 #include "xls/passes/union_query_engine.h"
 #include "xls/scheduling/pipeline_schedule.h"
@@ -130,6 +132,7 @@ absl::StatusOr<viz::NodeAttributes> NodeAttributes(
   }
   if (query_engine.IsTracked(node)) {
     attributes.set_known_bits(query_engine.ToString(node));
+    attributes.set_ranges(query_engine.GetIntervals(node).ToString());
   }
   if (std::optional<int64_t> state_index = MaybeGetStateReadIndex(node);
       state_index.has_value()) {
@@ -192,6 +195,8 @@ absl::StatusOr<viz::FunctionBase> FunctionBaseToVisualizationProto(
   engines.emplace_back(
       std::make_unique<BddQueryEngine>(BddFunction::kDefaultPathLimit));
   engines.emplace_back(std::make_unique<TernaryQueryEngine>());
+  engines.emplace_back(std::make_unique<RangeQueryEngine>());
+  engines.emplace_back(std::make_unique<ProcStateRangeQueryEngine>());
   UnionQueryEngine query_engine(std::move(engines));
   XLS_RETURN_IF_ERROR(query_engine.Populate(function).status());
 

xls/visualization/ir_viz/visualization.proto

@@ -73,6 +73,9 @@ message NodeAttributes {
   optional double state_param_index = 9;
   optional string initial_value = 10;
   optional double area_um = 11;
+
+  // If present the interval-set of the node.
+  optional string ranges = 32;
 }
 
 message SourceLocation {