Add two math operations: floor & ceil (#72)

* Inital commit * Add ceil function
oap-project · Jan 7, 2022 · c2f31f5 · c2f31f5
1 parent 1162e63
commit c2f31f5
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/cpp/src/gandiva/function_registry_arithmetic.cc b/cpp/src/gandiva/function_registry_arithmetic.cc
@@ -115,6 +115,12 @@ std::vector<NativeFunction> GetArithmeticFunctionRegistry() {
       // normalize for nan and zero
       UNARY_SAFE_NULL_IF_NULL(normalize, {}, float32, float32),
       UNARY_SAFE_NULL_IF_NULL(normalize, {}, float64, float64),
+      // floor
+      UNARY_SAFE_NULL_IF_NULL(floor, {}, float64, int64),
+      UNARY_SAFE_NULL_IF_NULL(floor, {}, int64, int64),
+      // ceil
+      UNARY_SAFE_NULL_IF_NULL(ceil, {}, float64, int64),
+      UNARY_SAFE_NULL_IF_NULL(ceil, {}, int64, int64),
       // bitwise functions
       BINARY_GENERIC_SAFE_NULL_IF_NULL(shift_left, {}, int32, int32, int32),
       BINARY_GENERIC_SAFE_NULL_IF_NULL(shift_left, {}, int64, int32, int64),

diff --git a/cpp/src/gandiva/precompiled/arithmetic_ops.cc b/cpp/src/gandiva/precompiled/arithmetic_ops.cc
@@ -158,6 +158,30 @@ NORMALIZE(float32, float32)
 
 #undef NORMALIZE
 
+// floor
+#define FLOOR(IN_TYPE, OUT_TYPE)                                        \
+  FORCE_INLINE                                                          \
+  gdv_##OUT_TYPE floor_##IN_TYPE(gdv_##IN_TYPE in) {                    \
+    return static_cast<gdv_##OUT_TYPE>(std::floor(in));                 \
+  }
+
+FLOOR(float64, int64)
+FLOOR(int64, int64)
+
+#undef FLOOR
+
+// ceil
+#define CEIL(IN_TYPE, OUT_TYPE)                                        \
+  FORCE_INLINE                                                         \
+  gdv_##OUT_TYPE ceil_##IN_TYPE(gdv_##IN_TYPE in) {                    \
+    return static_cast<gdv_##OUT_TYPE>(std::ceil(in));                 \
+  }
+
+CEIL(float64, int64)
+CEIL(int64, int64)
+
+#undef CEIL
+
 // cast fns : takes one param type, returns another type.
 #define CAST_UNARY(NAME, IN_TYPE, OUT_TYPE)           \
   FORCE_INLINE                                        \