feat: add logarithmic functions with decimal type args (#669)

substrait-io · Aug 1, 2024 · d9fb1e3 · d9fb1e3
1 parent 2aa516b
commit d9fb1e3
Showing 1 changed file with 59 additions and 0 deletions.
diff --git a/extensions/functions_logarithmic.yaml b/extensions/functions_logarithmic.yaml
@@ -38,6 +38,17 @@ scalar_functions:
           on_log_zero:
             values: [NAN, ERROR, MINUS_INFINITY]
         return: fp64
+      - args:
+          - name: x
+            value: decimal<P,S>
+        options:
+          rounding:
+            values: [ TIE_TO_EVEN, TIE_AWAY_FROM_ZERO, TRUNCATE, CEILING, FLOOR ]
+          on_domain_error:
+            values: [ NAN, "NULL", ERROR ]
+          on_log_zero:
+            values: [ NAN, ERROR, MINUS_INFINITY ]
+        return: fp64
   -
     name: "log10"
     description: "Logarithm to base 10 of the value"
@@ -75,6 +86,17 @@ scalar_functions:
           on_log_zero:
             values: [NAN, ERROR, MINUS_INFINITY]
         return: fp64
+      - args:
+          - name: x
+            value: decimal<P,S>
+        options:
+          rounding:
+            values: [ TIE_TO_EVEN, TIE_AWAY_FROM_ZERO, TRUNCATE, CEILING, FLOOR ]
+          on_domain_error:
+            values: [ NAN, "NULL", ERROR ]
+          on_log_zero:
+            values: [ NAN, ERROR, MINUS_INFINITY ]
+        return: fp64
   -
     name: "log2"
     description: "Logarithm to base 2 of the value"
@@ -112,6 +134,17 @@ scalar_functions:
           on_log_zero:
             values: [NAN, ERROR, MINUS_INFINITY]
         return: fp64
+      - args:
+          - name: x
+            value: decimal<P,S>
+        options:
+          rounding:
+            values: [ TIE_TO_EVEN, TIE_AWAY_FROM_ZERO, TRUNCATE, CEILING, FLOOR ]
+          on_domain_error:
+            values: [ NAN, "NULL", ERROR ]
+          on_log_zero:
+            values: [ NAN, ERROR, MINUS_INFINITY ]
+        return: fp64
   -
     name: "logb"
     description: >
@@ -164,6 +197,21 @@ scalar_functions:
           on_log_zero:
             values: [NAN, ERROR, MINUS_INFINITY]
         return: fp64
+      - args:
+          - value: decimal<P1,S1>
+            name: "x"
+            description: "The number `x` to compute the logarithm of"
+          - value: decimal<P1,S1>
+            name: "base"
+            description: "The logarithm base `b` to use"
+        options:
+          rounding:
+            values: [ TIE_TO_EVEN, TIE_AWAY_FROM_ZERO, TRUNCATE, CEILING, FLOOR ]
+          on_domain_error:
+            values: [ NAN, "NULL", ERROR ]
+          on_log_zero:
+            values: [NAN, ERROR, MINUS_INFINITY]
+        return: fp64
   -
     name: "log1p"
     description: >
@@ -193,3 +241,14 @@ scalar_functions:
           on_log_zero:
             values: [NAN, ERROR, MINUS_INFINITY]
         return: fp64
+      - args:
+          - name: x
+            value: decimal<P,S>
+        options:
+          rounding:
+            values: [ TIE_TO_EVEN, TIE_AWAY_FROM_ZERO, TRUNCATE, CEILING, FLOOR ]
+          on_domain_error:
+            values: [ NAN, "NULL", ERROR ]
+          on_log_zero:
+            values: [NAN, ERROR, MINUS_INFINITY]
+        return: fp64