[ Web Proxy ]
URL:
Viewing: https://raw.githubusercontent.com/gnuplot/gnuplot/master/src/internal.c [Back]  [Original]

/* GNUPLOT - internal.c */

/*[
 * Copyright 1986 - 1993, 1998, 2004   Thomas Williams, Colin Kelley
 *
 * Permission to use, copy, and distribute this software and its
 * documentation for any purpose with or without fee is hereby granted,
 * provided that the above copyright notice appear in all copies and
 * that both that copyright notice and this permission notice appear
 * in supporting documentation.
 *
 * Permission to modify the software is granted, but not the right to
 * distribute the complete modified source code.  Modifications are to
 * be distributed as patches to the released version.  Permission to
 * distribute binaries produced by compiling modified sources is granted,
 * provided you
 *   1. distribute the corresponding source modifications from the
 *    released version in the form of a patch file along with the binaries,
 *   2. add special version identification to distinguish your version
 *    in addition to the base release version number,
 *   3. provide your name and address as the primary contact for the
 *    support of your modified version, and
 *   4. retain our contact information in regard to use of the base
 *    software.
 * Permission to distribute the released version of the source code along
 * with corresponding source modifications in the form of a patch file is
 * granted with same provisions 2 through 4 for binary distributions.
 *
 * This software is provided "as is" without express or implied warranty
 * to the extent permitted by applicable law.
]*/


#include "internal.h"

#include "stdfn.h"
#include "alloc.h"
#include "util.h"	/* for int_error() */
#include "gplocale.h"	/* for locale handling */
#include "gp_time.h"	/* for str(p|f)time */
#include "command.h"	/* for do_system_func */
#include "datablock.h"	/* for datablock_size() */
#include "encoding.h"	/* for advance_one_utf8_char */
#include "parse.h"	/* for string_result_only */
#include "datafile.h"	/* for evaluate_inside_using */

#include 

#ifdef _WIN32
# define WIN32_LEAN_AND_MEAN
# include 
#endif

/*
 * Any platforms that still want support for matherr should
 * add appropriate definitions here.  Everyone else can now ignore it.
 *
 * Use of matherr is out of date on linux, since the matherr
 * mechanism has been replaced by math_error() and supposedly is only
 * enabled via an explicit declaration #define _SVID_SOURCE.
 */
/*
 * Excerpt from the Solaris man page for matherr():
 *
 *   The The System V Interface Definition, Third Edition (SVID3)
 *   specifies  that  certain  libm functions call matherr() when
 *   exceptions are detected. Users may define their own  mechan-
 *   isms  for handling exceptions, by including a function named
 *   matherr() in their programs.
 */
/* Watcom compiler */
#ifdef __WATCOMC__
int
matherr(struct _exception *x)
{
    return (undefined = TRUE);	/* don't print error message */
}
#endif


static enum DATA_TYPES sprintf_specifier(const char *format);

#define BADINT_DEFAULT int_error(NO_CARET, "error: bit shift applied to non-INT");
#define BAD_TYPE(type) int_error(NO_CARET, (type==NOTDEFINED) ? "uninitialized user variable" : "internal error : type neither INT nor CMPLX");

static const char *nonstring_error = "internal error : STRING operator applied to undefined or non-STRING variable";

static int recursion_depth = 0;
void
eval_reset_after_error()
{
    reset_stack();
    recursion_depth = 0;
    undefined = FALSE;
}

void
f_push(union argument *x)
{
    struct udvt_entry *udv;

    udv = x->udv_arg;
    if (udv->udv_value.type == NOTDEFINED) {
	if (string_result_only)
	/* We're only here to check whether this is a string. It isn't. */
	    udv = udv_NaN;
	else
	    int_error(NO_CARET, "undefined variable: %s", udv->udv_name);
    }
    push(&(udv->udv_value));
}

void
f_pushc(union argument *x)
{
    push(&(x->v_arg));
}


void
f_pushd1(union argument *x)
{
    push(&(x->udf_arg->dummy_values[0]));
}

void
f_pop(union argument *x)
{
    struct value dummy;
    pop(&dummy);
    if (dummy.type == STRING)
	gpfree_string(&dummy);
    if (dummy.type == ARRAY && dummy.v.value_array[0].type == TEMP_ARRAY)
	gpfree_array(&dummy);
}

void
f_nop(union argument *x)
{
    /* Presence of NOP in an action table serves as a barrier
     * at the end of a ternary operation sequence.
     */
}

void
f_pushd2(union argument *x)
{
    push(&(x->udf_arg->dummy_values[1]));
}


void
f_pushd(union argument *x)
{
    struct value param;
    (void) pop(¶m);
    push(&(x->udf_arg->dummy_values[param.v.int_val]));
}


/* execute a udf */
void
f_call(union argument *x)
{
    struct udft_entry *udf;
    struct value save_dummy;

    udf = x->udf_arg;
    if (!udf->at) {
	if (string_result_only) {
	    /* We're only here to check whether this is a string. It isn't. */
	    f_pop(x);
	    push(&(udv_NaN->udv_value));
	    return;
	}
	int_error(NO_CARET, "undefined function: %s", udf->udf_name);
    }

    save_dummy = udf->dummy_values[0];
    pop(&(udf->dummy_values[0]));

    if (udf->dummy_num != 1)
	int_error(NO_CARET, "function %s requires %d variables", udf->udf_name, udf->dummy_num);

    if (recursion_depth++ > STACK_DEPTH)
	int_error(NO_CARET, "recursion depth limit exceeded");

    /* User-defined functions need help to clean up temporary arrays.
     * If the action table contains hard-coded array functions they may
     * delete these before return, which would lead to a double-free below.
     * We replace the TEMP_ARRAY flag during evaluation to prevent this.
     */
    if (udf->dummy_values[0].type == ARRAY
    &&  udf->dummy_values[0].v.value_array[0].type == TEMP_ARRAY)
	udf->dummy_values[0].v.value_array[0].type = ARRAY;

    execute_at(udf->at);

    if (udf->dummy_values[0].type == ARRAY
    &&  udf->dummy_values[0].v.value_array[0].type == ARRAY) {
	/* Free TEMP_ARRAY passed as a parameter unless it is also the return value. */
	struct value top_of_stack;
	pop(&top_of_stack);
	if (udf->dummy_values[0].type == top_of_stack.type
	&&  udf->dummy_values[0].v.value_array == top_of_stack.v.value_array) {
	    top_of_stack.v.value_array[0].type = TEMP_ARRAY;
	} else {
	    gpfree_array(&udf->dummy_values[0]);
	}
	push(&top_of_stack);
	gpfree_string(&top_of_stack);
    }
    gpfree_string(&udf->dummy_values[0]);
    udf->dummy_values[0] = save_dummy;

    recursion_depth--;
}


/* execute a udf of n variables */
void
f_calln(union argument *x)
{
    struct udft_entry *udf;
    struct value save_dummy[MAX_NUM_VAR];

    int i;
    int num_pop;
    struct value num_params;
    struct value top_of_stack;

    udf = x->udf_arg;
    if (!udf->at)		/* undefined */
	int_error(NO_CARET, "undefined function: %s", udf->udf_name);

    (void) pop(&num_params);

    num_pop = num_params.v.int_val;

    if (num_pop != udf->dummy_num)
	int_error(NO_CARET, "function %s requires %d variable%c",
	    udf->udf_name, udf->dummy_num, (udf->dummy_num == 1)?'\0':'s');
    if (num_pop > MAX_NUM_VAR)
	int_error(NO_CARET, "too many parameters passed to function %s",
	    udf->udf_name);

    if (recursion_depth++ > STACK_DEPTH)
	int_error(NO_CARET, "recursion depth limit exceeded");

    for (i = 0; i < num_pop; i++)
	save_dummy[i] = udf->dummy_values[i];

    /* pop parameters we can use */
    for (i = num_pop - 1; i >= 0; i--) {
	pop(&(udf->dummy_values[i]));
	/* User-defined functions need help to clean up temporary arrays.
	 * If the action table contains hard-coded array functions they may
	 * delete these before return, which would lead to a double-free below.
	 * We replace the TEMP_ARRAY flag during evaluation to prevent this.
	 */
	if (udf->dummy_values[i].type == ARRAY
	&&  udf->dummy_values[i].v.value_array[0].type == TEMP_ARRAY)
	    udf->dummy_values[i].v.value_array[0].type = ARRAY;
    }

    execute_at(udf->at);

    /* Free TEMP_ARRAY passed as a parameter unless it is also the return value */
    pop(&top_of_stack);
    for (i = 0; i < num_pop; i++) {
	if (udf->dummy_values[i].type == ARRAY
	&&  udf->dummy_values[i].v.value_array[0].type == ARRAY) {
	    if (udf->dummy_values[i].type == top_of_stack.type
	    &&  udf->dummy_values[i].v.value_array == top_of_stack.v.value_array) {
		top_of_stack.v.value_array[0].type = TEMP_ARRAY;
	    } else {
		gpfree_array(&udf->dummy_values[i]);
	    }
	}
	gpfree_string(&udf->dummy_values[i]);
	udf->dummy_values[i] = save_dummy[i];
    }
    push(&top_of_stack);
    gpfree_string(&top_of_stack);

    recursion_depth--;
}


/* Evaluate expression   sum [i=beg:end] f(i)
 * Return an integer if f(i) are all integral, complex otherwise.
 * Save and restore value of i.
 * This is a change from versions 5.0 and 5.2 which always returned
 * a complex value and clobbered the iteration variable.
 */
void
f_sum(union argument *arg)
{
    struct value beg, end, varname; /* [ = :] */
    udft_entry *udf;                /* function to evaluate */
    udvt_entry *udv;                /* iteration variable */
    struct value result;            /* accummulated sum */
    struct value f_i;
    struct value save_i;	    /* previous value of iteration variable */
    int i;
    intgr_t llsum;		    /* integer sum */
    TBOOLEAN integer_terms = TRUE;

    (void) pop(&end);
    (void) pop(&beg);
    (void) pop(&varname);

    /* Initialize sum to 0 */
    Gcomplex(&result, 0, 0);
    llsum = 0;

    if (beg.type != INTGR || end.type != INTGR)
	int_error(NO_CARET, "range specifiers of sum must have integer values");

    udv = add_udv_by_name(varname.v.string_val);
    gpfree_string(&varname);
    save_i = udv->udv_value;

    udf = arg->udf_arg;
    if (!udf)
	int_error(NO_CARET, "internal error: lost expression to be evaluated during summation");

    for (i=beg.v.int_val; iudv_value, i);
	execute_at(udf->at);
	pop(&f_i);

	x = real(&result) + real(&f_i);
	y = imag(&result) + imag(&f_i);
	Gcomplex(&result, x, y);

	if (f_i.type != INTGR)
	    integer_terms = FALSE;
	if (!integer_terms)
	    continue;

	/* So long as the individual terms are integral */
	/* keep an integer sum as well.			*/
	llsum += f_i.v.int_val;

	/* Check for integer overflow */
	if (overflow_handling == INT64_OVERFLOW_IGNORE)
	    continue;
	if (sgn(result.v.cmplx_val.real) != sgn(llsum))  {
	    integer_terms = FALSE;
	    if (overflow_handling == INT64_OVERFLOW_TO_FLOAT)
		continue;
	    if (overflow_handling == INT64_OVERFLOW_UNDEFINED)
		undefined = TRUE;
	    if (overflow_handling == INT64_OVERFLOW_NAN)
		Gcomplex(&result, not_a_number(), 0.0);
	    break;
	}
    }

    if (integer_terms)
	push(Ginteger(&result, llsum));
    else
	push(&result);

    /* restore original value of iteration variable */
    udv->udv_value = save_i;
}


void
f_lnot(union argument *arg)
{
    struct value a;

    (void) arg;			/* avoid -Wunused warning */
    int_check(pop(&a));
    push(Ginteger(&a, !a.v.int_val));
}


void
f_bnot(union argument *arg)
{
    struct value a;

    (void) arg;			/* avoid -Wunused warning */
    int_check(pop(&a));
    push(Ginteger(&a, ~a.v.int_val));
}


void
f_lor(union argument *arg)
{
    struct value a, b;

    (void) arg;			/* avoid -Wunused warning */
    int_check(pop(&b));
    int_check(pop(&a));
    push(Ginteger(&a, a.v.int_val || b.v.int_val));
}

void
f_land(union argument *arg)
{
    struct value a, b;

    (void) arg;			/* avoid -Wunused warning */
    int_check(pop(&b));
    int_check(pop(&a));
    push(Ginteger(&a, a.v.int_val && b.v.int_val));
}


void
f_bor(union argument *arg)
{
    struct value a, b;

    (void) arg;			/* avoid -Wunused warning */
    int_check(pop(&b));
    int_check(pop(&a));
    push(Ginteger(&a, a.v.int_val | b.v.int_val));
}


void
f_xor(union argument *arg)
{
    struct value a, b;

    (void) arg;			/* avoid -Wunused warning */
    int_check(pop(&b));
    int_check(pop(&a));
    push(Ginteger(&a, a.v.int_val ^ b.v.int_val));
}


void
f_band(union argument *arg)
{
    struct value a, b;

    (void) arg;			/* avoid -Wunused warning */
    int_check(pop(&b));
    int_check(pop(&a));
    push(Ginteger(&a, a.v.int_val & b.v.int_val));
}


/*
 * Make all the following internal routines perform autoconversion
 * from string to numeric value.
 */
#define pop(x) pop_or_convert_from_string(x)

void
f_uminus(union argument *arg)
{
    struct value a;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&a);
    switch (a.type) {
    case INTGR:
	a.v.int_val = -a.v.int_val;
	break;
    case CMPLX:
	a.v.cmplx_val.real =
	    -a.v.cmplx_val.real;
	a.v.cmplx_val.imag =
	    -a.v.cmplx_val.imag;
	break;
    default:
	BAD_TYPE(a.type)
	break;
    }
    push(&a);
}


void
f_eq(union argument *arg)
{
    /* note: floating point equality is rare because of roundoff error! */
    struct value a, b;
    int result = 0;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&b);
    (void) pop(&a);

    switch (a.type) {
    case INTGR:
	switch (b.type) {
	case INTGR:
	    result = (a.v.int_val ==
		      b.v.int_val);
	    break;
	case CMPLX:
	    result = (a.v.int_val ==
		      b.v.cmplx_val.real &&
		      b.v.cmplx_val.imag == 0.0);
	    break;
	default:
	    BAD_TYPE(b.type)
	}
	break;
    case CMPLX:
	switch (b.type) {
	case INTGR:
	    result = (b.v.int_val == a.v.cmplx_val.real &&
		      a.v.cmplx_val.imag == 0.0);
	    break;
	case CMPLX:
	    result = (a.v.cmplx_val.real ==
		      b.v.cmplx_val.real &&
		      a.v.cmplx_val.imag ==
		      b.v.cmplx_val.imag);
	    break;
	default:
	    BAD_TYPE(b.type)
	}
	break;
    default:
	BAD_TYPE(a.type)
    }
    push(Ginteger(&a, result));
}


void
f_ne(union argument *arg)
{
    struct value a, b;
    int result = 0;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&b);
    (void) pop(&a);
    switch (a.type) {
    case INTGR:
	switch (b.type) {
	case INTGR:
	    result = (a.v.int_val !=
		      b.v.int_val);
	    break;
	case CMPLX:
	    result = (a.v.int_val !=
		      b.v.cmplx_val.real ||
		      b.v.cmplx_val.imag != 0.0);
	    break;
	default:
	    BAD_TYPE(b.type)
	}
	break;
    case CMPLX:
	switch (b.type) {
	case INTGR:
	    result = (b.v.int_val !=
		      a.v.cmplx_val.real ||
		      a.v.cmplx_val.imag != 0.0);
	    break;
	case CMPLX:
	    result = (a.v.cmplx_val.real !=
		      b.v.cmplx_val.real ||
		      a.v.cmplx_val.imag !=
		      b.v.cmplx_val.imag);
	    break;
	default:
	    BAD_TYPE(b.type)
	}
	break;
    default:
	BAD_TYPE(a.type)
    }
    push(Ginteger(&a, result));
}


void
f_gt(union argument *arg)
{
    struct value a, b;
    int result = 0;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&b);
    (void) pop(&a);
    switch (a.type) {
    case INTGR:
	switch (b.type) {
	case INTGR:
	    result = (a.v.int_val >
		      b.v.int_val);
	    break;
	case CMPLX:
	    result = (a.v.int_val >
		      b.v.cmplx_val.real);
	    break;
	default:
	    BAD_TYPE(b.type)
	}
	break;
    case CMPLX:
	switch (b.type) {
	case INTGR:
	    result = (a.v.cmplx_val.real >
		      b.v.int_val);
	    break;
	case CMPLX:
	    result = (a.v.cmplx_val.real >
		      b.v.cmplx_val.real);
	    break;
	default:
	    BAD_TYPE(b.type)
	}
	break;
    default:
	BAD_TYPE(a.type)
    }
    push(Ginteger(&a, result));
}


void
f_lt(union argument *arg)
{
    struct value a, b;
    int result = 0;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&b);
    (void) pop(&a);
    switch (a.type) {
    case INTGR:
	switch (b.type) {
	case INTGR:
	    result = (a.v.int_val <
		      b.v.int_val);
	    break;
	case CMPLX:
	    result = (a.v.int_val <
		      b.v.cmplx_val.real);
	    break;
	default:
	    BAD_TYPE(b.type)
	}
	break;
    case CMPLX:
	switch (b.type) {
	case INTGR:
	    result = (a.v.cmplx_val.real <
		      b.v.int_val);
	    break;
	case CMPLX:
	    result = (a.v.cmplx_val.real <
		      b.v.cmplx_val.real);
	    break;
	default:
	    BAD_TYPE(b.type)
	}
	break;
    default:
	BAD_TYPE(a.type)
    }
    push(Ginteger(&a, result));
}


void
f_ge(union argument *arg)
{
    struct value a, b;
    int result = 0;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&b);
    (void) pop(&a);
    switch (a.type) {
    case INTGR:
	switch (b.type) {
	case INTGR:
	    result = (a.v.int_val >=
		      b.v.int_val);
	    break;
	case CMPLX:
	    result = (a.v.int_val >=
		      b.v.cmplx_val.real);
	    break;
	default:
	    BAD_TYPE(b.type)
	}
	break;
    case CMPLX:
	switch (b.type) {
	case INTGR:
	    result = (a.v.cmplx_val.real >=
		      b.v.int_val);
	    break;
	case CMPLX:
	    result = (a.v.cmplx_val.real >=
		      b.v.cmplx_val.real);
	    break;
	default:
	    BAD_TYPE(b.type)
	}
	break;
    default:
	BAD_TYPE(a.type)
    }
    push(Ginteger(&a, result));
}


void
f_le(union argument *arg)
{
    struct value a, b;
    int result = 0;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&b);
    (void) pop(&a);
    switch (a.type) {
    case INTGR:
	switch (b.type) {
	case INTGR:
	    result = (a.v.int_val = |d|
 *
 * and
 *
 * [a(c/d) + b] + i[b(c/d)  a] / [c(c/d) + d] for |c| <  |d|
 *
 */
void
cmplx_divide(double a, double b, double c, double d, struct value *result)
{
    double f1, f2, denom;

    /* The common case of pure real numbers has no spurious overflow */
    if (b == 0 && d == 0 && c != 0) {
	(void) Gcomplex(result, a/c, 0.0);

    } else if (fabs(c) + fabs(d)) {
	if(fabs(c) >= fabs(d)) {
	    f1 = 1;
	    f2 = d / c;
	} else {
	    f1 = c / d;
	    f2 = 1;
	}
	denom = c * f1 + d * f2;

	(void) Gcomplex(result,
		(a * f1 + b * f2) / denom,
		(b * f1 - a * f2) / denom);
    } else {
	(void) Gcomplex(result, 0.0, 0.0);
	undefined = TRUE;
    }
}

void
f_div(union argument *arg)
{
    struct value a, b, result;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&b);
    (void) pop(&a);		/* now do a/b */

    switch (a.type) {
    case INTGR:
	switch (b.type) {
	case INTGR:
	    if (b.v.int_val)
		(void) Ginteger(&result, a.v.int_val /
				b.v.int_val);
	    else {
		(void) Ginteger(&result, 0);
		undefined = TRUE;
	    }
	    break;
	case CMPLX:
	    cmplx_divide(
		(double)a.v.int_val, 0.0,
		b.v.cmplx_val.real, b.v.cmplx_val.imag,
		&result);
	    break;
	default:
	    BAD_TYPE(b.type)
	}
	break;
    case CMPLX:
	switch (b.type) {
	case INTGR:
	    cmplx_divide(
		a.v.cmplx_val.real, a.v.cmplx_val.imag,
		(double)b.v.int_val, 0.0,
		&result);
	    break;
	case CMPLX:
	    cmplx_divide(
		a.v.cmplx_val.real, a.v.cmplx_val.imag,
		b.v.cmplx_val.real, b.v.cmplx_val.imag,
		&result);
	    break;
	default:
	    BAD_TYPE(b.type)
	}
	break;
    default:
	BAD_TYPE(a.type)
    }
    push(&result);
}


void
f_mod(union argument *arg)
{
    struct value a, b;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&b);
    (void) pop(&a);		/* now do a%b */

    if (a.type != INTGR || b.type != INTGR)
	int_error(NO_CARET, "non-integer operand for %%");
    if (b.v.int_val)
	push(Ginteger(&a, a.v.int_val % b.v.int_val));
    else {
	push(Ginteger(&a, 0));
	undefined = TRUE;
    }
}


void
f_power(union argument *arg)
{
    struct value a, b, result;
    int i;
    double mag, ang;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&b);
    (void) pop(&a);		/* now find a**b */

    switch (a.type) {
    case INTGR:
	switch (b.type) {
	case INTGR:
	    if (a.v.int_val == 0) {
		if (b.v.int_val < 0)
		    undefined = TRUE;
		(void) Ginteger(&result, b.v.int_val == 0 ? 1 : 0);
		break;
	    } else if (b.v.int_val == 0) {
		(void) Ginteger(&result, 1);
		break;
	    } else if (b.v.int_val > 0) {
		/* deal with overflow by empirical check */
		intgr_t tprev, t;
		intgr_t tmag = labs(a.v.int_val);
		tprev = t = 1;
		for (i = 0; i < b.v.int_val; i++) {
		    tprev = t;
		    t *= tmag;
		    if (t < tprev)
			goto integer_power_overflow;
		}
		if (a.v.int_val < 0) {
		    if ((0x1 & b.v.int_val) == 0x1)
			t = -t;
		}
		(void) Ginteger(&result, t);
		break;
	    }
	integer_power_overflow:
	    if (overflow_handling == INT64_OVERFLOW_NAN) {
		/* result of integer overflow is NaN */
		(void) Gcomplex(&result, not_a_number(), 0.0);
	    } else if (overflow_handling == INT64_OVERFLOW_UNDEFINED) {
		/* result of integer overflow is undefined */
		undefined = TRUE;
	    } else {
		/* switch to double if overflow */
		mag = pow((double)a.v.int_val,(double)b.v.int_val);
		(void) Gcomplex(&result, mag, 0.0);
	    }
	    break;
	case CMPLX:
	    if (a.v.int_val == 0) {
		if (b.v.cmplx_val.imag != 0 || b.v.cmplx_val.real < 0) {
		    undefined = TRUE;
		}
		/* return 1.0 for 0**0 */
		Gcomplex(&result, b.v.cmplx_val.real == 0 ? 1.0 : 0.0, 0.0);
	    } else {
		mag =
		    pow(magnitude(&a), fabs(b.v.cmplx_val.real));
		if (b.v.cmplx_val.real < 0.0) {
		    if (mag != 0.0)
			mag = 1.0 / mag;
		    else
			undefined = TRUE;
		}
		mag *= gp_exp(-b.v.cmplx_val.imag * angle(&a));
		ang = b.v.cmplx_val.real * angle(&a) +
		    b.v.cmplx_val.imag * log(magnitude(&a));
		(void) Gcomplex(&result, mag * cos(ang),
				mag * sin(ang));
	    }
	    break;
	default:
	    BAD_TYPE(b.type)
	}
	break;
    case CMPLX:
	switch (b.type) {
	case INTGR:
	    if (a.v.cmplx_val.imag == 0.0) {
		mag = pow(a.v.cmplx_val.real, fabs((double)b.v.int_val));
		if (b.v.int_val < 0) {
		    if (mag != 0.0)
			mag = 1.0 / mag;
		    else
			undefined = TRUE;
		}
		(void) Gcomplex(&result, mag, 0.0);
	    } else {
		/* not so good, but...! */
		mag = pow(magnitude(&a), fabs((double)b.v.int_val));
		if (b.v.int_val < 0) {
		    if (mag != 0.0)
			mag = 1.0 / mag;
		    else
			undefined = TRUE;
		}
		ang = angle(&a) * b.v.int_val;
		(void) Gcomplex(&result, mag * cos(ang),
				mag * sin(ang));
	    }
	    break;
	case CMPLX:
	    if (a.v.cmplx_val.real == 0 && a.v.cmplx_val.imag == 0) {
		if (b.v.cmplx_val.imag != 0 || b.v.cmplx_val.real < 0) {
		    undefined = TRUE;
		}
		/* return 1.0 for 0**0 */
		Gcomplex(&result, b.v.cmplx_val.real == 0 ? 1.0 : 0.0, 0.0);
	    } else {
		mag = pow(magnitude(&a), fabs(b.v.cmplx_val.real));
		if (b.v.cmplx_val.real < 0.0) {
		    if (mag != 0.0)
			mag = 1.0 / mag;
		    else
			undefined = TRUE;
		}
		mag *= gp_exp(-b.v.cmplx_val.imag * angle(&a));
		ang = b.v.cmplx_val.real * angle(&a) +
		    b.v.cmplx_val.imag * log(magnitude(&a));
		(void) Gcomplex(&result, mag * cos(ang),
				mag * sin(ang));
	    }
	    break;
	default:
	    BAD_TYPE(b.type)
	}
	break;
    default:
	BAD_TYPE(a.type)
    }

    /* Catch underflow and return 0 */
    /* Note: fpclassify() is an ISOC99 macro found also in other libc implementations */
#ifdef fpclassify
    if (errno == ERANGE && result.type == CMPLX) {
	int fperror = fpclassify(result.v.cmplx_val.real);
	if (fperror == FP_ZERO || fperror == FP_SUBNORMAL) {
	    result.v.cmplx_val.real = 0.0;
	    result.v.cmplx_val.imag = 0.0;
	    errno = 0;
	}
    }
#endif

    push(&result);
}


void
f_factorial(union argument *arg)
{
    struct value a;
    intgr_t i;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&a);		/* find a! (factorial) */

    if (a.type != INTGR)
	int_error(NO_CARET, "factorial (!) argument must be an integer");

    if (((sizeof(int) == sizeof(intgr_t)) && a.v.int_val  1; i--)
	    val *= i;
	push(Gcomplex(&a, val, 0.0));
    }
}

/*
 * Terminate the autoconversion from string to numeric values
 */
#undef pop

void
f_concatenate(union argument *arg)
{
    struct value a, b, result;
    char *newstring;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&b);
    (void) pop(&a);

    if (b.type == INTGR) {
	int i = b.v.int_val;
	b.type = STRING;
	b.v.string_val = (char *)gp_alloc(32,"str_const");
	snprintf(b.v.string_val,32,"%d",i);
    }

    if (a.type != STRING || b.type != STRING)
	int_error(NO_CARET, nonstring_error);

    newstring = gp_alloc(strlen(a.v.string_val)+strlen(b.v.string_val)+1,"gp_stradd");
    strcpy(newstring, a.v.string_val);
    strcat(newstring, b.v.string_val);

    Gstring(&result, newstring);
    push(&result);

    gpfree_string(&a);
    gpfree_string(&b);
    gpfree_string(&result);
}

void
f_eqs(union argument *arg)
{
    struct value a, b, result;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&b);
    (void) pop(&a);

    if (a.type != STRING || b.type != STRING)
	int_error(NO_CARET, nonstring_error);

    (void) Ginteger(&result, !strcmp(a.v.string_val, b.v.string_val));
    gpfree_string(&a);
    gpfree_string(&b);
    push(&result);
}

void
f_nes(union argument *arg)
{
    struct value a, b, result;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&b);
    (void) pop(&a);

    if (a.type != STRING || b.type != STRING)
	int_error(NO_CARET, nonstring_error);

    (void) Ginteger(&result, (int)(strcmp(a.v.string_val, b.v.string_val)!=0));
    gpfree_string(&a);
    gpfree_string(&b);
    push(&result);
}

void
f_strlen(union argument *arg)
{
    struct value a, result;

    (void) arg;
    (void) pop(&a);

    if (a.type != STRING)
	int_error(NO_CARET, "internal error : strlen of non-STRING argument");

    (void) Ginteger(&result, (int)gp_strlen(a.v.string_val));
    gpfree_string(&a);
    push(&result);
}

void
f_strstrt(union argument *arg)
{
    struct value needle, haystack, result;
    char *start;
    int hit = 0;

    (void) arg;
    (void) pop(&needle);
    (void) pop(&haystack);

    if (needle.type != STRING || haystack.type != STRING)
	int_error(NO_CARET, "internal error : non-STRING argument to strstrt");

    start = strstr(haystack.v.string_val, needle.v.string_val);
    if (start == 0) {
	hit = -1;
    } else if (encoding == S_ENC_UTF8) {
	char *utfstring = haystack.v.string_val;
	while (utfstring < start) {
	    advance_one_utf8_char(utfstring);
	    hit++;
	}
    } else {
	hit = (start - haystack.v.string_val);
    }
    (void) Ginteger(&result, hit+1);
    gpfree_string(&needle);
    gpfree_string(&haystack);
    push(&result);
}

/*
 * f_range() handles both explicit calls to substr(string, beg, end)
 * and the short form string[beg:end].  The calls to gp_strlen() and
 * gp_strchrn() allow it to handle utf8 strings.
 * f_range() also handles requests for an array slice B = A[low:high].
 */
void
f_range(union argument *arg)
{
    struct value beg, end, full;
    struct value result;
    int ibeg, iend;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&end);
    (void) pop(&beg);
    (void) pop(&full);

    if (beg.type == INTGR)
	ibeg = beg.v.int_val;
    else if (beg.type == CMPLX)
	ibeg = floor(beg.v.cmplx_val.real);
    else
	int_error(NO_CARET, "internal error: non-numeric substring range specifier");
    if (end.type == INTGR)
	iend = end.v.int_val;
    else if (end.type == CMPLX)
	iend = floor(end.v.cmplx_val.real);
    else
	int_error(NO_CARET, "internal error: non-numeric substring range specifier");

    if (full.type == ARRAY) {
	result.v.value_array = array_slice( &full, ibeg, iend );
	result.type = ARRAY;
	if (full.v.value_array[0].type == TEMP_ARRAY)
	    gpfree_array(&full);
	push(&result);
	return;
    }

    if (full.type != STRING)
	int_error(NO_CARET, "internal error: substring range operator applied to non-STRING type");

    if (iend > gp_strlen(full.v.string_val))
	iend = gp_strlen(full.v.string_val);
    if (ibeg < 1)
	ibeg = 1;

    if (ibeg > iend) {
	push(Gstring(&result, ""));
    } else {
	char *begp = gp_strchrn(full.v.string_val,ibeg-1);
	char *endp = gp_strchrn(full.v.string_val,iend);
	*endp = '\0';
	push(Gstring(&result, begp));
    }
    gpfree_string(&full);
}


/*
 * f_index() extracts the value of a single element from an array.
 */
void
f_index(union argument *arg)
{
    struct value array, index;
    int i = -1;

    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&index);
    (void) pop(&array);

    if (index.type == INTGR)
	i = index.v.int_val;
    else if (index.type == CMPLX)
	i = floor(index.v.cmplx_val.real);
    else
	int_error(NO_CARET, "non-numeric array index");

    if (array.type == ARRAY) {
	if (i  array.v.value_array[0].v.int_val)
	    int_error(NO_CARET, "array index out of range");
	push( &array.v.value_array[i] );
	if (array.v.value_array[0].type == TEMP_ARRAY)
	    gpfree_array(&array);

    } else if (array.type == DATABLOCK) {
	i--;	/* line numbers run from 1 to nlines */
	if (i < 0 || i >= datablock_size(&array))
	    int_error(NO_CARET, "datablock index out of range");
	push( Gstring(&array, array.v.data_array[i]) );
    }

    else
	int_error(NO_CARET, "internal error: attempt to index a scalar variable");
}

/*
 * f_cardinality() extracts the number of elements in an array.
 */
void
f_cardinality(union argument *arg)
{
    struct value array;
    int size;
    (void) arg;			/* avoid -Wunused warning */
    (void) pop(&array);

    if (array.type == ARRAY) {
	size = array.v.value_array[0].v.int_val;
	if (array.v.value_array[0].type == TEMP_ARRAY)
	    gpfree_array(&array);
    } else if (array.type == DATABLOCK) {
	size = datablock_size(&array);
    } else {
	int_error(NO_CARET, "internal error: cardinality of a scalar variable");
    }

    push(Ginteger(&array, size));
}

/* Magic number! */
#define RETURN_WORD_COUNT (-17*23*61)

void
f_words(union argument *arg)
{
    struct value a;

    /* "words(s)" is implemented as "word(s,RETURN_WORD_COUNT)" */
    push(Ginteger(&a, RETURN_WORD_COUNT));
    f_word(arg);
}

void
f_word(union argument *arg)
{
    struct value a, b, result;
    int nwords = 0;
    int in_string = 0;
    int ntarget;
    char q = '\0';
    char *s;

    (void) arg;
    if (pop(&b)->type != INTGR)
	int_error(NO_CARET, "internal error : non-INTGR argument");
    ntarget = b.v.int_val;

    if (pop(&a)->type != STRING)
	int_error(NO_CARET, "internal error : non-STRING argument");
    s = a.v.string_val;

    Gstring(&result, "");
    while (*s) {
	while (isspace((unsigned char)*s)) s++;
	if (!*s)
	    break;
	nwords++;
	if (*s == '"' || *s == '\'') {
	    q = *s;
	    s++;
	    in_string = 1;
	}
	if (nwords == ntarget) { /* Found the one we wanted */
	    Gstring(&result,s);
	    s = result.v.string_val;
	}
	while (*s && ((!isspace((unsigned char)*s) && !in_string) || (in_string && *s != q))) s++;
	if (nwords == ntarget) { /* Terminate this word cleanly */
	    *s = '\0';
	    break;
	}
	if (in_string && (*s == q)) {
	    in_string = 0;
	    s++;
	}
    }

    /* words(s) = word(s,magic_number) = # of words in string */
    if (ntarget == RETURN_WORD_COUNT)
	Ginteger(&result, nwords);

    push(&result);
    gpfree_string(&a);
}
#undef RETURN_WORD_COUNT


/* EAM July 2004
 * revised to handle 64-bit integers April 2018
 * There are probably an infinite number of things that can
 * go wrong if the user mis-matches arguments and format strings
 * in the call to sprintf, but I hope none will do worse than
 * result in a garbage output string.
 */
void
f_sprintf(union argument *arg)
{
    struct value a[10], *args;
    struct value num_params;
    struct value result;
    int bufsize;
    char *next_start, *outpos, tempchar;
    int next_length;
    char *prev_start;
    int prev_pos;
    int i, remaining;
    int nargs = 0;
    enum DATA_TYPES spec_type;
    char *buffer = NULL;
    char *error_return_message = NULL;

    /* Retrieve number of parameters from top of stack */
    (void) arg;
    pop(&num_params);
    nargs = num_params.v.int_val;
    if (nargs > 10) {	/* Fall back to slow but sure allocation */
	args = gp_alloc(sizeof(struct value)*nargs, "sprintf args");
    } else
	args = a;

    for (i=0; i 0) {
	if (*next_start == '%' && *(next_start+1) == '%')
	    next_start++;
	*outpos++ = *next_start++;
    }
    *outpos = '\0';

    push(Gstring(&result, buffer));

f_sprintf_error_return:

    /* Free all locally allocated memory before returning */
    free(buffer);

    /* Free any strings from parameters we have now used */
    for (i=0; i= buflen)
	int_error(NO_CARET, "String produced by time format is too long");

    /* Remove trailing space */
    assert(buffer[length-1] == ' ');
    buffer[length-1] = NUL;

    gpfree_string(&val);
    gpfree_string(&fmt);
    free(fmtstr);

    push(Gstring(&val, buffer));
    free(buffer);
}

/* Convert string into seconds from ZERO_YEAR */
void
f_strptime(union argument *arg)
{
    struct value fmt, val;
    struct tm time_tm;
    double usec = 0.0;
    double result;

    (void) arg; /* Avoid compiler warnings */

    pop(&val);
    pop(&fmt);

    if ( fmt.type != STRING || val.type != STRING )
	int_error(NO_CARET,
		  "Both parameters to strptime must be strings");
    if ( !fmt.v.string_val || !val.v.string_val )
	int_error(NO_CARET, "Internal error: string not allocated");


    /* string -> time_tm  plus extra fractional second */
    if (gstrptime(val.v.string_val, fmt.v.string_val, &time_tm, &usec, &result)
	    == DT_TIMEDATE) {
	/* time_tm -> result */
	result = gtimegm(&time_tm);
	/* Add back any extra fractional second */
	result += usec;
    }
    FPRINTF((stderr," strptime result = %g seconds \n", result));

    gpfree_string(&val);
    gpfree_string(&fmt);
    push(Gcomplex(&val, result, 0.0));
}

/* Get current system time in seconds since ZERO_YEAR.
 * The type of the value popped from the stack
 * determines what is returned.
 * If integer, the result is also an integer.
 * If real (complex), the result is also real,
 * with microsecond precision (if available).
 * If string, it is assumed to be a format string,
 * and it is passed to strftime to get a formatted time string.
 */
void
f_time(union argument *arg)
{
    struct value val, val2;
    double time_now;
#ifdef HAVE_SYS_TIME_H
    struct timeval tp;

    gettimeofday(&tp, NULL);
    tp.tv_sec -= SEC_OFFS_SYS;
    time_now = tp.tv_sec + (tp.tv_usec/1000000.0);
#elif defined(_WIN32)
    SYSTEMTIME systime;
    FILETIME filetime;
    ULARGE_INTEGER itime;

    /* get current system time (UTC) */
    GetSystemTime(&systime);
    /* convert to integer value in 100ns steps */
    SystemTimeToFileTime(&systime, &filetime);
    itime.HighPart = filetime.dwHighDateTime;
    itime.LowPart = filetime.dwLowDateTime;
    /* reference of this value is 1601-01-01 (no typo!) */
    /* strptime("%Y-%m-%d", "1601-01-01") = -11644473600.0 */
    time_now = itime.QuadPart * 100e-9 - 11644473600.0 - SEC_OFFS_SYS;
#else
    time_now = (double) time(NULL);
    time_now -= SEC_OFFS_SYS;
#endif

    (void) arg; /* Avoid compiler warnings */
    pop(&val);

    switch(val.type) {
	case INTGR:
	    push(Ginteger(&val, (intgr_t) time_now));
	    break;
	case CMPLX:
	    push(Gcomplex(&val, time_now, 0.0));
	    break;
	case STRING:
	    push(&val); /* format string */
	    push(Gcomplex(&val2, time_now, 0.0));
	    f_strftime(arg);
	    gpfree_string(&val);
	    break;
	default:
	    int_error(NO_CARET,"internal error: invalid argument type");
    }
}


/* Return which argument type sprintf will need for this format string:
 *   char*       STRING
 *   int         INTGR
 *   double      CMPLX
 * Should call int_err for any other type.
 * format is expected to start with '%'
 */
static enum DATA_TYPES
sprintf_specifier(const char* format)
{
    const char string_spec[]  = "s";
    const char real_spec[]    = "aAeEfFgG";
    const char int_spec[]     = "cdiouxX";
    /* The following characters are used to reject invalid formats */
    const char illegal_spec[] = "hlLqjzZtCSpn*";

    int string_pos, real_pos, int_pos, illegal_pos;
    int nonascii_pos;

    /* check if really format specifier */
    if (format[0] != '%')
	int_error(NO_CARET,
		  "internal error: sprintf_specifier called without '%'\n");

    string_pos  = strcspn(format, string_spec);
    real_pos    = strcspn(format, real_spec);
    int_pos     = strcspn(format, int_spec);
    illegal_pos = strcspn(format, illegal_spec);

    /* Unfortunately snprintf can segfault on weird bytes in the format */
    for (nonascii_pos=0; format[nonascii_pos]; nonascii_pos++) {
	if (!isascii(format[nonascii_pos]))
	    break;
    }
    if ( nonascii_pos < int_pos && nonascii_pos < real_pos
	 && nonascii_pos < string_pos )
	return INVALID_NAME;

    if ( illegal_pos < int_pos && illegal_pos < real_pos
	 && illegal_pos < string_pos )
	return INVALID_NAME;
    if ( string_pos < real_pos && string_pos < int_pos )
	return STRING;
    if ( real_pos < int_pos )
	return CMPLX;
    if ( int_pos < strlen(format) )
	return INTGR;

    return INVALID_NAME;
}


/* execute a system call and return stream from STDOUT */
void
f_system(union argument *arg)
{
    struct value val, result;
    char *output;
    int output_len, ierr;

    /* Retrieve parameters from top of stack */
    (void) arg;
    pop(&val);

    /* Make sure parameters are of the correct type */
    if (val.type != STRING)
	int_error(NO_CARET, "non-string argument to system()");

    FPRINTF((stderr," f_system input = \"%s\"\n", val.v.string_val));

    ierr = do_system_func(val.v.string_val, &output);
    fill_gpval_integer("GPVAL_ERRNO", ierr);

    /* chomp result */
    output_len = strlen(output);
    if ( output_len > 0 && output[output_len-1] == '\n' )
	output[output_len-1] = NUL;

    FPRINTF((stderr," f_system result = \"%s\"\n", output));

    push(Gstring(&result, output));

    gpfree_string(&result); /* free output */
    gpfree_string(&val);    /* free command string */
}


/* Variable assignment operator */
void
f_assign(union argument *arg)
{
    struct udvt_entry *udv;
    struct value a, b, index;
    struct value *dest;
    (void) arg;
    (void) pop(&b);	/* new value */
    dest = pop(&a);	/* name of variable or pointer to array content */

    if (dest->type == ARRAY) {
	/* It's an assignment to an array element. We don't know the index yet */
	;

    } else {
	if (dest->type != STRING)
	    int_error(NO_CARET, "attempt to assign to something other than a named variable");
	if (!strncmp(dest->v.string_val,"GPVAL_",6) || !strncmp(dest->v.string_val,"MOUSE_",6))
	    int_error(NO_CARET, "attempt to assign to a read-only variable");

	udv = add_udv_by_name(a.v.string_val);
	gpfree_string(&a);	    /* This frees the name string, not the variable it names */
	dest = &(udv->udv_value);   /* Now dest points to where the new value will go */
    }

    if (b.type == ARRAY) {
	if (arg->v_arg.type == ARRAY)	/* Actually flags assignment to an array element */
	    int_error(NO_CARET, "cannot nest arrays");
	free_value(dest);
	*dest = b;
	make_array_permanent(dest);

    } else if (dest->type == ARRAY) {
	int i;
	pop(&index);
	if (index.type == INTGR)
	    i = index.v.int_val;
	else if (index.type == CMPLX)
	    i = floor(index.v.cmplx_val.real);
	else
	    int_error(NO_CARET, "non-numeric array index");
	if (i  dest->v.value_array[0].v.int_val)
	    int_error(NO_CARET, "array index out of range");
	gpfree_string(&dest->v.value_array[i]);
	dest->v.value_array[i] = b;

    } else {
	free_value(dest);
	*dest = b;
    }

    push(&b);
}

/*
 * Retrieve the current value of a user-defined variable whose name is known.
 * B = value("A") has the same result as B = A.
 */

void
f_value(union argument *arg)
{
    struct udvt_entry *p = first_udv;
    struct value a;
    struct value result;

    (void) arg;
    (void) pop(&a);

    if (a.type != STRING) {
	/* int_warn(NO_CARET,"non-string value passed to value()"); */
	push(&a);
	return;
    }

    while (p) {
	if (!strcmp(p->udv_name, a.v.string_val)) {
	    result = p->udv_value;
	    if (p->udv_value.type == NOTDEFINED)
		p = NULL;
	    else if (result.type == STRING)
		result.v.string_val = gp_strdup(result.v.string_val);
	    break;
	}
	p = p->next_udv;
    }
    gpfree_string(&a);
    if (!p) {
	/* int_warn(NO_CARET,"undefined variable name passed to value()"); */
	result.type = CMPLX;
	result.v.cmplx_val.real = not_a_number();
	result.v.cmplx_val.imag = 0;
    }
    push(&result);
}

/*
 * retrieve index of array entry with known value
 */
void
f_lookup(union argument *arg)
{
    struct value entry;
    struct value a;
    struct value *array;
    int i, n;
    int index = 0;

    /* what entry are we looking for? */
    pop(&entry);

    /* what array is it in? */
    pop(&a);
    if (a.type != ARRAY)
	int_error(NO_CARET, "index: expecting an array");
    array = a.v.value_array;
    n = array[0].v.int_val;

    for (i=1; i trim) && isspace((unsigned char) *s))
	*(s--) = '\0';

    s = a.v.string_val;
    a.v.string_val = trim;
    push(&a);
    free(s);
}

/*
 * split ( "string", {"sep"} )
 * Split string into an array of words.
 * The second parameter is optional.
 */
void
f_split(union argument *arg)
{
    struct value a;
    char *string, *sep;
    static char *whitespace = " ";

    /* Determine number of parameters passed (1 or 2) */
    (void)pop(&a);
    if (a.v.int_val == 1) {
	/* Separator defaults to whitespace, indicated by " " */
	sep = whitespace;
    } else if (a.v.int_val == 2) {
	(void)pop(&a);
	if (a.type != STRING)
	    int_error(NO_CARET, nonstring_error);
	sep = a.v.string_val;
	if (*sep == '\0')
	    sep = whitespace;
    } else
	int_error(NO_CARET, "too many parameters to split()");

    (void)pop(&a);
    if (a.type != STRING)
	int_error(NO_CARET, nonstring_error);
    string = a.v.string_val;

    a.v.value_array = split(string, sep);
    a.type = (a.v.value_array) ? ARRAY : NOTDEFINED;

    if (sep != whitespace)
	free(sep);
    free(string);
    push(&a);
}

/*
 * join ( array, "sep" )
 * Concatenate the string elements of array into a single longer string.
 * The character sequence in "sep" is inserted between each element.
 */
void
f_join(union argument *arg)
{
    struct value a;
    struct value *array;
    char *sep;
    char *concatenation;
    int i, n, size;

    (void)pop(&a);
    if (a.type != STRING)
	int_error(NO_CARET, "join: expecting join(array, \"separator\")");
    sep = a.v.string_val;

    (void)pop(&a);
    if (a.type != ARRAY)
	int_error(NO_CARET, "join: expecting join(array, \"separator\")");

    array = a.v.value_array;
    n = array[0].v.int_val;
    size = 0;
    for (i=1; i

Web Proxy Viewer  |  New URL  |  Original Page