mariadb-columnstore-engine/utils/funcexp/func_add_time.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_add_time.cpp 3923 2013-06-19 21:43:06Z bwilkinson $
*
*
****************************************************************************/

#include <cstdlib>
#include <string>
#include <sstream>
using namespace std;

#include "functioncolumn.h"
#include "rowgroup.h"
using namespace execplan;

#include "dataconvert.h"
using namespace dataconvert;

#include "functor_dtm.h"
#include "funchelpers.h"

namespace
{
using namespace funcexp;

int64_t addTime(DateTime& dt1, Time& dt2)
{
    DateTime dt;
    dt.year = 0;
    dt.month = 0;
    dt.day = 0;
    dt.hour = 0;
    dt.minute = 0;
    dt.second = 0;
    dt.msecond = 0;

    int64_t month, day, hour, min, sec, msec, tmp;
    msec = (signed)(dt1.msecond + dt2.msecond);
    dt.msecond = tmp = msec % 1000000;

    if (tmp < 0)
    {
        dt.msecond = tmp + 1000000;
        dt2.second--;
    }

    sec = (signed)(dt1.second + dt2.second + msec / 1000000);
    dt.second = tmp = sec % 60;

    if (tmp < 0)
    {
        dt.second = tmp + 60;
        dt2.minute--;
    }

    min = (signed)(dt1.minute + dt2.minute + sec / 60);
    dt.minute = tmp = min % 60;

    if (tmp < 0)
    {
        dt.minute = tmp + 60;
        dt2.hour--;
    }

    hour = (signed)(dt1.hour + dt2.hour + min / 60);

    if ((hour < 0) || (hour > 23))
    {
        dt2.day = hour / 24;
        hour = hour % 24;
    }

    if (hour < 0)
    {
        dt.hour = hour + 24;
        dt2.day--;
    }
    else
    {
        dt.hour = hour;
    }

    day = (signed)(dt1.day + dt2.day);


    if (isLeapYear(dt1.year) && dt1.month == 2)
        day--;

    month = dt1.month;
    int addyear = 0;

    if (day < 0)
    {
        int monthSave = month;

        while (day <= 0)
        {
            month = (month == 1 ? 12 : month - 1);

            for (; day <= 0 && month > 0; month--)
                day += getDaysInMonth(month, dt1.year);

            month++;
//			month=12;
        }

        if ( month > monthSave )
            addyear--;
    }
    else
    {
        int monthSave = month;

        while (day > getDaysInMonth(month, dt1.year))
        {
            for (; day > getDaysInMonth(month, dt1.year) && month <= 12; month++)
                day -= getDaysInMonth(month, dt1.year);

            if (month > 12)
                month = 1;
        }

        if ( month < monthSave )
            addyear++;
    }

    dt.day = day;
    dt.month = month;
    dt.year = dt1.year + addyear;

    return *(reinterpret_cast<int64_t*>(&dt));
}

int64_t addTime(Time& dt1, Time& dt2)
{
    Time dt;
    dt.is_neg = false;
    dt.hour = 0;
    dt.minute = 0;
    dt.second = 0;
    dt.msecond = 0;

    int64_t min, sec, msec, tmp;
    msec = (signed)(dt1.msecond + dt2.msecond);
    dt.msecond = tmp = msec % 1000000;

    if (tmp < 0)
    {
        dt.msecond = tmp + 1000000;
        dt2.second--;
    }

    sec = (signed)(dt1.second + dt2.second + msec / 1000000);
    dt.second = tmp = sec % 60;

    if (tmp < 0)
    {
        dt.second = tmp + 60;
        dt2.minute--;
    }

    min = (signed)(dt1.minute + dt2.minute + sec / 60);
    dt.minute = tmp = min % 60;

    if (tmp < 0)
    {
        dt.minute = tmp + 60;
        dt2.hour--;
    }

    dt.hour = tmp = (signed)(dt1.hour + dt2.hour + min / 60);

    // Saturation
    if (tmp > 838)
    {
        dt.hour = 838;
        dt.minute = 59;
        dt.second = 59;
        dt.msecond = 999999;
    }
    else if (tmp < -838)
    {
        dt.is_neg = true;
        dt.hour = -838;
        dt.minute = 59;
        dt.second = 59;
        dt.msecond = 999999;
    }

    return *(reinterpret_cast<int64_t*>(&dt));
}


}

namespace funcexp
{

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

int64_t Func_add_time::getIntVal(rowgroup::Row& row,
                                 FunctionParm& parm,
                                 bool& isNull,
                                 CalpontSystemCatalog::ColType& op_ct)
{
    if (parm[0]->data()->resultType().colDataType == execplan::CalpontSystemCatalog::TIME)
    {
        return getTimeIntVal(row, parm, isNull, op_ct);
    }
    else
    {
        return getDatetimeIntVal(row, parm, isNull, op_ct);
    }
}

string Func_add_time::getStrVal(rowgroup::Row& row,
                                FunctionParm& parm,
                                bool& isNull,
                                CalpontSystemCatalog::ColType& ct)
{
    return intToString(getIntVal(row, parm, isNull, ct));
}

int32_t Func_add_time::getDateIntVal(rowgroup::Row& row,
                                     FunctionParm& parm,
                                     bool& isNull,
                                     CalpontSystemCatalog::ColType& ct)
{
    return (((getDatetimeIntVal(row, parm, isNull, ct) >> 32) & 0xFFFFFFC0) | 0x3E);
}

int64_t Func_add_time::getDatetimeIntVal(rowgroup::Row& row,
        FunctionParm& parm,
        bool& isNull,
        CalpontSystemCatalog::ColType& ct)
{
    int64_t val1 = parm[0]->data()->getDatetimeIntVal(row, isNull);

    if (isNull)
        return -1;

    // Adding a zero date to a time is always NULL
    if (val1 == 0)
    {
        isNull = true;
        return -1;
    }

    const string& val2 = parm[1]->data()->getStrVal(row, isNull);
    int sign = parm[2]->data()->getIntVal(row, isNull);
    DateTime dt1;
    dt1.year = (val1 >> 48) & 0xffff;
    dt1.month = (val1 >> 44) & 0xf;
    dt1.day = (val1 >> 38) & 0x3f;
    dt1.hour = (val1 >> 32) & 0x3f;
    dt1.minute = (val1 >> 26) & 0x3f;
    dt1.second = (val1 >> 20) & 0x3f;
    dt1.msecond = val1 & 0xfffff;

    int64_t	time = DataConvert::stringToTime(val2);

    if (time == -1)
    {
        isNull = true;
        return -1;
    }

    Time t2 = *(reinterpret_cast<Time*>(&time));

    // MySQL TIME type range '-838:59:59' and '838:59:59'
    if (t2.minute > 59 || t2.second > 59 || t2.msecond > 999999)
    {
        isNull = true;
        return -1;
    }

    int val_sign = 1;

    if (t2.hour < 0)
    {
        val_sign = -1;
    }

    if (abs(t2.hour) > 838)
    {
        t2.hour = 838;
        t2.minute = 59;
        t2.second = 59;
        t2.msecond = 999999;
    }

    if (val_sign * sign < 0)
    {
        t2.hour = -abs(t2.hour);
        t2.minute = -abs(t2.minute);
        t2.second = -abs(t2.second);
        t2.msecond = -abs(t2.msecond);
    }
    else
    {
        t2.hour = abs(t2.hour);
        t2.minute = abs(t2.minute);
        t2.second = abs(t2.second);
        t2.msecond = abs(t2.msecond);
    }

    t2.day = 0;

    return addTime(dt1, t2);
}

int64_t Func_add_time::getTimeIntVal(rowgroup::Row& row,
                                     FunctionParm& parm,
                                     bool& isNull,
                                     CalpontSystemCatalog::ColType& ct)
{
    int64_t val1 = parm[0]->data()->getTimeIntVal(row, isNull);

    if (isNull)
        return -1;

    const string& val2 = parm[1]->data()->getStrVal(row, isNull);
    int sign = parm[2]->data()->getIntVal(row, isNull);
    Time dt1;
    dt1.day = 0;
    dt1.is_neg = val1 >> 63;
    dt1.hour = (val1 >> 40) & 0xfff;
    dt1.minute = (val1 >> 32) & 0xff;
    dt1.second = (val1 >> 24) & 0xff;
    dt1.msecond = val1 & 0xffffff;

    int64_t	time = DataConvert::stringToTime(val2);

    if (time == -1)
    {
        isNull = true;
        return -1;
    }

    Time t2 = *(reinterpret_cast<Time*>(&time));

    // MySQL TIME type range '-838:59:59' and '838:59:59'
    if (t2.minute > 59 || t2.second > 59 || t2.msecond > 999999)
    {
        isNull = true;
        return -1;
    }

    int val_sign = 1;

    if (t2.hour < 0)
    {
        val_sign = -1;
    }

    if (abs(t2.hour) > 838)
    {
        t2.hour = 838;
        t2.minute = 59;
        t2.second = 59;
        t2.msecond = 999999;
    }

    t2.day = 0;

    if (val_sign * sign < 0)
    {
        t2.hour = -abs(t2.hour);
        t2.minute = -abs(t2.minute);
        t2.second = -abs(t2.second);
        t2.msecond = -abs(t2.msecond);
    }
    else
    {
        t2.hour = abs(t2.hour);
        t2.minute = abs(t2.minute);
        t2.second = abs(t2.second);
        t2.msecond = abs(t2.msecond);
    }

    return addTime(dt1, t2);
}



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