mariadb-columnstore-engine/utils/funcexp/func_bitwise.cpp

/* Copyright (C) 2014 InfiniDB, Inc.

   This program is free software; you can redistribute it and/or
   modify it under the terms of the GNU General Public License
   as published by the Free Software Foundation; version 2 of
   the License.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
   MA 02110-1301, USA. */

/****************************************************************************
* $Id: func_bitwise.cpp 3616 2013-03-04 14:56:29Z rdempsey $
*
*
****************************************************************************/

#include <string>
using namespace std;

#include "functor_int.h"
#include "funchelpers.h"
#include "functioncolumn.h"
#include "predicateoperator.h"
using namespace execplan;

#include "rowgroup.h"
using namespace rowgroup;

#include "errorcodes.h"
#include "idberrorinfo.h"
#include "errorids.h"
using namespace logging;

#include "dataconvert.h"
using namespace dataconvert;

namespace
{
using namespace funcexp;

// @bug 4703 - the actual bug was only in the DATETIME case
// part of this statement below, but instead of leaving 5 identical
// copies of this code, extracted into a single utility function
// here.  This same method is potentially useful in other methods
// and could be extracted into a utility class with its own header
// if that is the case - this is left as future exercise
bool getUIntValFromParm(
    Row&  row,
    const execplan::SPTP& parm,
    uint64_t& value,
    bool& isNull)
{
    switch (parm->data()->resultType().colDataType)
    {
        case execplan::CalpontSystemCatalog::BIGINT:
        case execplan::CalpontSystemCatalog::INT:
        case execplan::CalpontSystemCatalog::MEDINT:
        case execplan::CalpontSystemCatalog::TINYINT:
        case execplan::CalpontSystemCatalog::SMALLINT:
        case execplan::CalpontSystemCatalog::DOUBLE:
        case execplan::CalpontSystemCatalog::FLOAT:
        case execplan::CalpontSystemCatalog::UDOUBLE:
        case execplan::CalpontSystemCatalog::UFLOAT:
        {
            value = parm->data()->getIntVal(row, isNull);
        }
        break;

        case execplan::CalpontSystemCatalog::UBIGINT:
        case execplan::CalpontSystemCatalog::UINT:
        case execplan::CalpontSystemCatalog::UMEDINT:
        case execplan::CalpontSystemCatalog::UTINYINT:
        case execplan::CalpontSystemCatalog::USMALLINT:
        {
            value = parm->data()->getUintVal(row, isNull);
        }
        break;

        case execplan::CalpontSystemCatalog::VARCHAR:
        case execplan::CalpontSystemCatalog::CHAR:
        case execplan::CalpontSystemCatalog::TEXT:
        {
            value = parm->data()->getIntVal(row, isNull);

            if (isNull)
            {
                isNull = true;
            }
            else
            {
                value = 0;
            }
        }
        break;

        case execplan::CalpontSystemCatalog::DECIMAL:
        case execplan::CalpontSystemCatalog::UDECIMAL:
        {
            IDB_Decimal d = parm->data()->getDecimalVal(row, isNull);

            if (parm->data()->resultType().colDataType == execplan::CalpontSystemCatalog::UDECIMAL &&
                    d.value < 0)
            {
                d.value = 0;
            }

            int64_t tmpval = d.value / helpers::power(d.scale);
            int lefto = (d.value - tmpval * helpers::power(d.scale)) / helpers::power(d.scale - 1);

            if ( tmpval >= 0 && lefto > 4 )
                tmpval++;

            if ( tmpval < 0 && lefto < -4 )
                tmpval--;

            value = tmpval;
        }
        break;

        case execplan::CalpontSystemCatalog::DATE:
        {
            int32_t time = parm->data()->getDateIntVal(row, isNull);

            Date d(time);
            value = d.convertToMySQLint();
        }
        break;

        case execplan::CalpontSystemCatalog::DATETIME:
        {
            int64_t time = parm->data()->getDatetimeIntVal(row, isNull);

            // @bug 4703 - missing year when convering to int
            DateTime dt(time);
            value = dt.convertToMySQLint();
        }
        break;

        default:
        {
            return false;
        }
    }

    return true;
}
}

namespace funcexp
{

//
// BITAND
//


CalpontSystemCatalog::ColType Func_bitand::operationType( FunctionParm& fp, CalpontSystemCatalog::ColType& resultType )
{
    return resultType;
}

int64_t Func_bitand::getIntVal(Row& row,
                               FunctionParm& parm,
                               bool& isNull,
                               CalpontSystemCatalog::ColType& operationColType)
{
    if ( parm.size() < 2 )
    {
        isNull = true;
        return 0;
    }

    uint64_t val1 = 0;
    uint64_t val2 = 0;

    if (!getUIntValFromParm(row, parm[0], val1, isNull) ||
            !getUIntValFromParm(row, parm[1], val2, isNull))
    {
        std::ostringstream oss;
        oss << "bitand: datatype of " << execplan::colDataTypeToString(operationColType.colDataType);
        throw logging::IDBExcept(oss.str(), ERR_DATATYPE_NOT_SUPPORT);
    }

    return val1 & val2;
}


//
// LEFT SHIFT
//


CalpontSystemCatalog::ColType Func_leftshift::operationType( FunctionParm& fp, CalpontSystemCatalog::ColType& resultType )
{
    return resultType;
}

int64_t Func_leftshift::getIntVal(Row& row,
                                  FunctionParm& parm,
                                  bool& isNull,
                                  CalpontSystemCatalog::ColType& operationColType)
{
    if ( parm.size() < 2 )
    {
        isNull = true;
        return 0;
    }

    uint64_t val1 = 0;
    uint64_t val2 = 0;

    if (!getUIntValFromParm(row, parm[0], val1, isNull) ||
            !getUIntValFromParm(row, parm[1], val2, isNull))
    {
        std::ostringstream oss;
        oss << "leftshift: datatype of " << execplan::colDataTypeToString(operationColType.colDataType);
        throw logging::IDBExcept(oss.str(), ERR_DATATYPE_NOT_SUPPORT);
    }

    return val1 << val2;
}


//
// RIGHT SHIFT
//


CalpontSystemCatalog::ColType Func_rightshift::operationType( FunctionParm& fp, CalpontSystemCatalog::ColType& resultType )
{
    return resultType;
}

int64_t Func_rightshift::getIntVal(Row& row,
                                   FunctionParm& parm,
                                   bool& isNull,
                                   CalpontSystemCatalog::ColType& operationColType)
{
    if ( parm.size() < 2 )
    {
        isNull = true;
        return 0;
    }

    uint64_t val1 = 0;
    uint64_t val2 = 0;

    if (!getUIntValFromParm(row, parm[0], val1, isNull) ||
            !getUIntValFromParm(row, parm[1], val2, isNull))
    {
        std::ostringstream oss;
        oss << "rightshift: datatype of " << execplan::colDataTypeToString(operationColType.colDataType);
        throw logging::IDBExcept(oss.str(), ERR_DATATYPE_NOT_SUPPORT);
    }

    return val1 >> val2;
}


//
// BIT OR
//


CalpontSystemCatalog::ColType Func_bitor::operationType( FunctionParm& fp, CalpontSystemCatalog::ColType& resultType )
{
    return resultType;
}

int64_t Func_bitor::getIntVal(Row& row,
                              FunctionParm& parm,
                              bool& isNull,
                              CalpontSystemCatalog::ColType& operationColType)
{
    if ( parm.size() < 2 )
    {
        isNull = true;
        return 0;
    }

    uint64_t val1 = 0;
    uint64_t val2 = 0;

    if (!getUIntValFromParm(row, parm[0], val1, isNull) ||
            !getUIntValFromParm(row, parm[1], val2, isNull))
    {
        std::ostringstream oss;
        oss << "bitor: datatype of " << execplan::colDataTypeToString(operationColType.colDataType);
        throw logging::IDBExcept(oss.str(), ERR_DATATYPE_NOT_SUPPORT);
    }

    return val1 | val2;
}

uint64_t Func_bitor::getUintVal(rowgroup::Row& row,
                                FunctionParm& fp,
                                bool& isNull,
                                execplan::CalpontSystemCatalog::ColType& op_ct)
{
    return static_cast<uint64_t>(getIntVal(row, fp, isNull, op_ct));
}


//
// BIT XOR
//


CalpontSystemCatalog::ColType Func_bitxor::operationType( FunctionParm& fp, CalpontSystemCatalog::ColType& resultType )
{
    return resultType;
}

int64_t Func_bitxor::getIntVal(Row& row,
                               FunctionParm& parm,
                               bool& isNull,
                               CalpontSystemCatalog::ColType& operationColType)
{
    if ( parm.size() < 2 )
    {
        isNull = true;
        return 0;
    }

    uint64_t val1 = 0;
    uint64_t val2 = 0;

    if (!getUIntValFromParm(row, parm[0], val1, isNull) ||
            !getUIntValFromParm(row, parm[1], val2, isNull))
    {
        std::ostringstream oss;
        oss << "bitxor: datatype of " << execplan::colDataTypeToString(operationColType.colDataType);
        throw logging::IDBExcept(oss.str(), ERR_DATATYPE_NOT_SUPPORT);
    }

    return val1 ^ val2;
}


} // namespace funcexp
// vim:ts=4 sw=4: