feat: Add Window Functions

Author: Pablo Pudgino

axiom/logical_plan/CMakeLists.txt

@@ -19,6 +19,7 @@ add_library(
   LogicalPlanNode.cpp
   PlanPrinter.cpp
   ExprApi.cpp
+  Utils.cpp
 )
 
 target_link_libraries(axiom_logical_plan velox_type)

axiom/logical_plan/Expr.h

@@ -576,7 +576,9 @@ class WindowExpr : public Expr {
         partitionKeys_{std::move(partitionKeys)},
         ordering_{std::move(ordering)},
         frame_{std::move(frame)},
-        ignoreNulls_{ignoreNulls} {}
+        ignoreNulls_{ignoreNulls} {
+    VELOX_USER_CHECK(!name_.empty());
+  }
 
   const std::string& name() const {
     return name_;

axiom/logical_plan/ExprPrinter.cpp

@@ -103,7 +103,59 @@ class ToTextVisitor : public ExprVisitor {
     out << expr.name();
     appendInputs(expr, out, context);
 
-    // TODO Add partitionKeys, ordering, frame, ignoreNulls.
+    if (expr.ignoreNulls()) {
+      out << " IGNORE NULLS";
+    }
+
+    out << " OVER (";
+
+    if (!expr.partitionKeys().empty()) {
+      out << "PARTITION BY ";
+      for (auto i = 0; i < expr.partitionKeys().size(); ++i) {
+        if (i > 0) {
+          out << ", ";
+        }
+        expr.partitionKeys()[i]->accept(*this, context);
+      }
+    }
+
+    if (!expr.ordering().empty()) {
+      if (!expr.partitionKeys().empty()) {
+        out << " ";
+      }
+      out << "ORDER BY ";
+      for (auto i = 0; i < expr.ordering().size(); ++i) {
+        if (i > 0) {
+          out << ", ";
+        }
+        expr.ordering()[i].expression->accept(*this, context);
+        out << " " << expr.ordering()[i].order.toString();
+      }
+    }
+
+    const auto& frame = expr.frame();
+    if (!expr.partitionKeys().empty() || !expr.ordering().empty()) {
+      out << " ";
+    }
+    out << WindowExpr::toName(frame.type) << " BETWEEN ";
+
+    // Start bound
+    if (frame.startValue != nullptr) {
+      frame.startValue->accept(*this, context);
+      out << " ";
+    }
+    out << WindowExpr::toName(frame.startType);
+
+    out << " AND ";
+
+    // End bound
+    if (frame.endValue != nullptr) {
+      frame.endValue->accept(*this, context);
+      out << " ";
+    }
+    out << WindowExpr::toName(frame.endType);
+
+    out << ")";
   }
 
   void visit(const ConstantExpr& expr, ExprVisitorContext& context)