udf-builtins.h

Go to the documentation of this file.

// Copyright 2012 Cloudera Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#ifndef IMPALA_EXPRS_UDF_BUILTINS_H
#define IMPALA_EXPRS_UDF_BUILTINS_H

#include "udf/udf.h"

using namespace impala_udf;

namespace impala {

class UdfBuiltins {
 public:
  static DoubleVal Abs(FunctionContext* context, const DoubleVal& val);
  static DoubleVal Pi(FunctionContext* context);
  static StringVal Lower(FunctionContext* context, const StringVal& str);

  static IntVal MaxInt(FunctionContext* context);
  static TinyIntVal MaxTinyInt(FunctionContext* context);
  static SmallIntVal MaxSmallInt(FunctionContext* context);
  static BigIntVal MaxBigInt(FunctionContext* context);
  static IntVal MinInt(FunctionContext* context);
  static TinyIntVal MinTinyInt(FunctionContext* context);
  static SmallIntVal MinSmallInt(FunctionContext* context);
  static BigIntVal MinBigInt(FunctionContext* context);

  static BooleanVal IsNan(FunctionContext* context, const DoubleVal& val);
  static BooleanVal IsInf(FunctionContext* context, const DoubleVal& val);

  //
  static TimestampVal Trunc(FunctionContext* context, const TimestampVal& date,
      const StringVal& unit_str);
  static void TruncPrepare(FunctionContext* context,
      FunctionContext::FunctionStateScope scope);
  static void TruncClose(FunctionContext* context,
      FunctionContext::FunctionStateScope scope);

  //
  static IntVal Extract(FunctionContext* context, const StringVal& field_str,
      const TimestampVal& date);

  static IntVal Extract(FunctionContext* context, const TimestampVal& date,
      const StringVal& field_str);
  static void ExtractPrepare(FunctionContext* context,
      FunctionContext::FunctionStateScope scope);

  static void SwappedExtractPrepare(FunctionContext* context,
      FunctionContext::FunctionStateScope scope);
  static void ExtractClose(FunctionContext* context,
      FunctionContext::FunctionStateScope scope);

  static StringVal ToVector(FunctionContext* context, int n, const DoubleVal* values);

  static StringVal PrintVector(FunctionContext* context, const StringVal& arr);

  static DoubleVal VectorGet(FunctionContext* context, const BigIntVal& n,
      const StringVal& arr);

  static StringVal EncodeVector(FunctionContext* context, const StringVal& arr);

  static StringVal DecodeVector(FunctionContext* context, const StringVal& arr);

 private:
  static void ExtractPrepare(FunctionContext* context,
      FunctionContext::FunctionStateScope scope, int unit_idx);
};

} // namespace impala
#endif