diff options
Diffstat (limited to 'src/func.c')
| -rw-r--r-- | src/func.c | 135 |
1 files changed, 135 insertions, 0 deletions
diff --git a/src/func.c b/src/func.c index 49f6c892b..9bee8576b 100644 --- a/src/func.c +++ b/src/func.c @@ -19,6 +19,9 @@ #include "sqliteInt.h" #include <stdlib.h> #include <assert.h> +#ifndef SQLITE_OMIT_FLOATING_POINT +# include <math.h> +#endif #include "vdbeInt.h" /* @@ -966,6 +969,134 @@ static void quoteFunc(sqlite3_context *context, int argc, sqlite3_value **argv){ } /* +** EXPERIMENTAL - This is not an official function. The interface may +** change. This function may disappear. Do not write code that depends +** on this function. +** +** Implementation of the tointeger() function. This function takes a +** single argument. If the argument is an integer or is a double that +** can be losslessly converted to an integer, the return value is the +** same as the argument. If the argument is a double that cannot be +** losslessly represented as an integer, the return value is NULL. +** If the argument is NULL, the return value is NULL. Otherwise, an +** attempt is made to convert the argument to an integer. If the +** conversion is successful, the integer value is returned; otherwise, +** NULL is returned. +*/ +static void tointegerFunc( + sqlite3_context *context, + int argc, + sqlite3_value **argv +){ + assert( argc==1 ); + UNUSED_PARAMETER(argc); + switch( sqlite3_value_type(argv[0]) ){ + case SQLITE_FLOAT: +#ifndef SQLITE_OMIT_FLOATING_POINT + { + double rVal = sqlite3_value_double(argv[0]); + double rIntVal = 0.0; + if( !sqlite3IsNaN(rVal) && modf(rVal, &rIntVal)==0.0 && + rIntVal>=SMALLEST_INT64 && rIntVal<=LARGEST_INT64 ){ + sqlite3_result_int64(context, (i64)rIntVal); + return; + } + sqlite3_result_null(context); + break; + } +#endif + case SQLITE_INTEGER: { + sqlite3_result_int64(context, sqlite3_value_int64(argv[0])); + break; + } + case SQLITE_BLOB: + case SQLITE_TEXT: { + const unsigned char *zStr = sqlite3_value_text(argv[0]); + if( zStr ){ + int nStr = sqlite3_value_bytes(argv[0]); + if( nStr ){ + i64 iVal; + if( !sqlite3Atoi64(zStr, &iVal, nStr, SQLITE_UTF8) ){ + sqlite3_result_int64(context, iVal); + return; + } + } + } + sqlite3_result_null(context); + break; + } + default: { + assert( sqlite3_value_type(argv[0])==SQLITE_NULL ); + sqlite3_result_null(context); + break; + } + } +} + +/* +** EXPERIMENTAL - This is not an official function. The interface may +** change. This function may disappear. Do not write code that depends +** on this function. +** +** Implementation of the toreal() function. This function takes a +** single argument. If the argument is a double or is an integer that +** can be losslessly converted to a double, the return value is the +** same as the argument. If the argument is an integer that cannot be +** losslessly represented as a double, the return value is NULL. +** If the argument is NULL, the return value is NULL. Otherwise, an +** attempt is made to convert the argument to a double. If the +** conversion is successful, the double value is returned; otherwise, +** NULL is returned. +*/ +#ifndef SQLITE_OMIT_FLOATING_POINT +static void torealFunc( + sqlite3_context *context, + int argc, + sqlite3_value **argv +){ + assert( argc==1 ); + UNUSED_PARAMETER(argc); + switch( sqlite3_value_type(argv[0]) ){ + case SQLITE_FLOAT: { + sqlite3_result_double(context, sqlite3_value_double(argv[0])); + break; + } + case SQLITE_INTEGER: { + i64 iVal = sqlite3_value_int64(argv[0]); + double rVal = (double)iVal; + if( iVal==rVal ){ + sqlite3_result_double(context, rVal); + return; + } + sqlite3_result_null(context); + break; + } + case SQLITE_BLOB: + case SQLITE_TEXT: { + const unsigned char *zStr = sqlite3_value_text(argv[0]); + if( zStr ){ + int nStr = sqlite3_value_bytes(argv[0]); + if( nStr ){ + double rVal; + if( sqlite3AtoF(zStr, &rVal, nStr, SQLITE_UTF8) ){ + sqlite3_result_double(context, rVal); + return; + } + } + } + sqlite3_result_null(context); + break; + } + default: { + assert( sqlite3_value_type(argv[0])==SQLITE_NULL ); + sqlite3_result_null(context); + break; + } + } +} +#endif + +/* ** The unicode() function. Return the integer unicode code-point value ** for the first character of the input string. */ @@ -1673,6 +1804,10 @@ void sqlite3RegisterGlobalFunctions(void){ FUNCTION(sqlite_compileoption_get, 1, 0, 0, compileoptiongetFunc ), #endif /* SQLITE_OMIT_COMPILEOPTION_DIAGS */ FUNCTION(quote, 1, 0, 0, quoteFunc ), + FUNCTION(tointeger, 1, 0, 0, tointegerFunc ), +#ifndef SQLITE_OMIT_FLOATING_POINT + FUNCTION(toreal, 1, 0, 0, torealFunc ), +#endif FUNCTION(last_insert_rowid, 0, 0, 0, last_insert_rowid), FUNCTION(changes, 0, 0, 0, changes ), FUNCTION(total_changes, 0, 0, 0, total_changes ), |