1103 lines
21 KiB
C
1103 lines
21 KiB
C
/*
|
|
* $Id$
|
|
*/
|
|
|
|
#include "awk.h"
|
|
#include <ase/cmn/sio.h>
|
|
#include <ase/cmn/str.h>
|
|
#include <ase/cmn/time.h>
|
|
|
|
#include <math.h>
|
|
#include <stdarg.h>
|
|
#include <stdlib.h>
|
|
#include <ase/utl/stdio.h>
|
|
|
|
typedef struct xtn_t
|
|
{
|
|
ase_awk_prmfns_t prmfns;
|
|
} xtn_t;
|
|
|
|
typedef struct rxtn_t
|
|
{
|
|
unsigned int seed;
|
|
} rxtn_t;
|
|
|
|
static ase_real_t custom_awk_pow (void* custom, ase_real_t x, ase_real_t y)
|
|
{
|
|
#if defined(HAVE_POWL)
|
|
return powl (x, y);
|
|
#elif defined(HAVE_POW)
|
|
return pow (x, y);
|
|
#elif defined(HAVE_POWF)
|
|
return powf (x, y);
|
|
#else
|
|
#error ### no pow function available ###
|
|
#endif
|
|
}
|
|
|
|
static int custom_awk_sprintf (
|
|
void* custom, ase_char_t* buf, ase_size_t size,
|
|
const ase_char_t* fmt, ...)
|
|
{
|
|
int n;
|
|
|
|
va_list ap;
|
|
va_start (ap, fmt);
|
|
n = ase_vsprintf (buf, size, fmt, ap);
|
|
va_end (ap);
|
|
|
|
return n;
|
|
}
|
|
|
|
static int add_functions (ase_awk_t* awk);
|
|
|
|
ase_awk_t* ase_awk_opensimple (ase_size_t xtnsize)
|
|
{
|
|
ase_awk_t* awk;
|
|
xtn_t* xtn;
|
|
|
|
awk = ase_awk_open (ASE_MMGR_GETDFL(), xtnsize + ASE_SIZEOF(xtn_t));
|
|
ase_awk_setccls (awk, ASE_CCLS_GETDFL());
|
|
|
|
xtn = (xtn_t*)((ase_byte_t*)ase_awk_getxtn(awk) + xtnsize);
|
|
|
|
xtn->prmfns.pow = custom_awk_pow;
|
|
xtn->prmfns.sprintf = custom_awk_sprintf;
|
|
xtn->prmfns.data = ASE_NULL;
|
|
ase_awk_setprmfns (awk, &xtn->prmfns);
|
|
|
|
ase_awk_setoption (awk,
|
|
ASE_AWK_IMPLICIT | ASE_AWK_EXTIO | ASE_AWK_NEWLINE |
|
|
ASE_AWK_BASEONE | ASE_AWK_PABLOCK);
|
|
|
|
if (add_functions (awk) == -1)
|
|
{
|
|
ase_awk_close (awk);
|
|
return ASE_NULL;
|
|
}
|
|
|
|
return awk;
|
|
}
|
|
|
|
/*** PARSESIMPLE ***/
|
|
|
|
typedef struct sf_t sf_t;
|
|
|
|
struct sf_t
|
|
{
|
|
struct
|
|
{
|
|
int type;
|
|
union
|
|
{
|
|
const ase_char_t*const* files;
|
|
const ase_char_t* str;
|
|
} p;
|
|
ase_size_t index; /* current file index */
|
|
ase_sio_t* handle; /* the handle to an open file */
|
|
} in;
|
|
|
|
struct
|
|
{
|
|
const ase_char_t* file;
|
|
ase_sio_t* handle;
|
|
} out;
|
|
|
|
ase_awk_t* awk;
|
|
};
|
|
|
|
static ase_ssize_t sf_in (int cmd, void* arg, ase_char_t* data, ase_size_t size)
|
|
{
|
|
sf_t* sf = (sf_t*)arg;
|
|
ase_cint_t c;
|
|
|
|
if (cmd == ASE_AWK_IO_OPEN)
|
|
{
|
|
if (sf->in.type == ASE_AWK_PARSE_FILES)
|
|
{
|
|
if (sf->in.p.files[sf->in.index] == ASE_NULL) return 0;
|
|
|
|
if (sf->in.p.files[sf->in.index][0] == ASE_T('\0'))
|
|
{
|
|
sf->in.handle = ase_sio_in;
|
|
}
|
|
else
|
|
{
|
|
sf->in.handle = ase_sio_open (
|
|
ase_awk_getmmgr(sf->awk),
|
|
0,
|
|
sf->in.p.files[sf->in.index],
|
|
ASE_SIO_READ
|
|
);
|
|
if (sf->in.handle == ASE_NULL) return -1;
|
|
}
|
|
|
|
/*
|
|
ase_awk_setsinname ();
|
|
*/
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
else if (cmd == ASE_AWK_IO_CLOSE)
|
|
{
|
|
if (sf->in.handle != ASE_NULL &&
|
|
sf->in.handle != ase_sio_in &&
|
|
sf->in.handle != ase_sio_out &&
|
|
sf->in.handle != ase_sio_err)
|
|
{
|
|
ase_sio_close (sf->in.handle);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
else if (cmd == ASE_AWK_IO_READ)
|
|
{
|
|
ase_ssize_t n = 0;
|
|
|
|
if (sf->in.type == ASE_AWK_PARSE_FILES)
|
|
{
|
|
ase_sio_t* sio;
|
|
|
|
retry:
|
|
sio = sf->in.handle;
|
|
|
|
n = ase_sio_getsx (sio, data, size);
|
|
if (n == 0 && sf->in.p.files[++sf->in.index] != ASE_NULL)
|
|
{
|
|
if (sio != ase_sio_in) ase_sio_close (sio);
|
|
if (sf->in.p.files[sf->in.index][0] == ASE_T('\0'))
|
|
{
|
|
sf->in.handle = ase_sio_in;
|
|
}
|
|
else
|
|
{
|
|
sf->in.handle = ase_sio_open (
|
|
ase_awk_getmmgr(sf->awk),
|
|
0,
|
|
sf->in.p.files[sf->in.index],
|
|
ASE_SIO_READ
|
|
);
|
|
if (sf->in.handle == ASE_NULL) return -1;
|
|
}
|
|
|
|
/* TODO: reset internal line counters...
|
|
set new source name....
|
|
ase_awk_setsinname ();
|
|
*/
|
|
|
|
goto retry;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
while (n < size && sf->in.p.str[sf->in.index] != ASE_T('\0'))
|
|
{
|
|
data[n++] = sf->in.p.str[sf->in.index++];
|
|
}
|
|
}
|
|
|
|
return n;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
static ase_ssize_t sf_out (int cmd, void* arg, ase_char_t* data, ase_size_t size)
|
|
{
|
|
sf_t* sf = (sf_t*)arg;
|
|
|
|
if (cmd == ASE_AWK_IO_OPEN)
|
|
{
|
|
if (sf->out.file[0] == ASE_T('\0'))
|
|
{
|
|
sf->out.handle = ase_sio_out;
|
|
}
|
|
else
|
|
{
|
|
sf->out.handle = ase_sio_open (
|
|
ase_awk_getmmgr(sf->awk),
|
|
0,
|
|
sf->out.file,
|
|
ASE_SIO_WRITE|ASE_SIO_CREATE|ASE_SIO_TRUNCATE
|
|
);
|
|
if (sf->out.handle == ASE_NULL) return -1;
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
else if (cmd == ASE_AWK_IO_CLOSE)
|
|
{
|
|
if (sf->out.handle != ASE_NULL)
|
|
{
|
|
ase_sio_flush (sf->out.handle);
|
|
if (sf->out.handle != ase_sio_in &&
|
|
sf->out.handle != ase_sio_out &&
|
|
sf->out.handle != ase_sio_err)
|
|
{
|
|
ase_sio_close (sf->out.handle);
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
else if (cmd == ASE_AWK_IO_WRITE)
|
|
{
|
|
/*
|
|
ase_size_t left = size;
|
|
|
|
while (left > 0)
|
|
{
|
|
if (*data == ASE_T('\0'))
|
|
{
|
|
if (ase_fputc (*data, sf->out.handle) == ASE_CHAR_EOF) return -1;
|
|
left -= 1; data += 1;
|
|
}
|
|
else
|
|
{
|
|
int chunk = (left > ASE_TYPE_MAX(int))? ASE_TYPE_MAX(int): (int)left;
|
|
int n = ase_fprintf (sf->out.handle, ASE_T("%.*s"), chunk, data);
|
|
if (n < 0) return -1;
|
|
left -= n; data += n;
|
|
}
|
|
|
|
}
|
|
*/
|
|
|
|
return ase_sio_putsx (sf->out.handle, data, size);
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
int ase_awk_parsesimple (
|
|
ase_awk_t* awk, const void* isp, int ist, const ase_char_t* osf)
|
|
{
|
|
sf_t sf;
|
|
ase_awk_srcios_t sio;
|
|
|
|
if (isp == ASE_NULL)
|
|
{
|
|
ase_awk_seterrnum (awk, ASE_AWK_EINVAL);
|
|
return -1;
|
|
}
|
|
|
|
if (ist == ASE_AWK_PARSE_FILES) sf.in.p.files = isp;
|
|
else if (ist == ASE_AWK_PARSE_STRING) sf.in.p.str = isp;
|
|
else
|
|
{
|
|
ase_awk_seterrnum (awk, ASE_AWK_EINVAL);
|
|
return -1;
|
|
}
|
|
|
|
sf.in.type = ist;
|
|
sf.in.index = 0;
|
|
sf.in.handle = ASE_NULL;
|
|
|
|
sf.out.file = osf;
|
|
sf.out.handle = ASE_NULL;
|
|
sf.awk = awk;
|
|
|
|
sio.in = sf_in;
|
|
sio.out = (osf == ASE_NULL)? ASE_NULL: sf_out;
|
|
sio.data = &sf;
|
|
|
|
return ase_awk_parse (awk, &sio);
|
|
}
|
|
|
|
/*** RUNSIMPLE ***/
|
|
|
|
typedef struct runio_data_t
|
|
{
|
|
struct
|
|
{
|
|
ase_char_t** files;
|
|
ase_size_t index;
|
|
} ic; /* input console */
|
|
} runio_data_t;
|
|
|
|
static ase_ssize_t awk_extio_pipe (
|
|
int cmd, void* arg, ase_char_t* data, ase_size_t size)
|
|
{
|
|
ase_awk_extio_t* epa = (ase_awk_extio_t*)arg;
|
|
|
|
switch (cmd)
|
|
{
|
|
case ASE_AWK_IO_OPEN:
|
|
{
|
|
ase_sio_t* handle;
|
|
int mode;
|
|
|
|
if (epa->mode == ASE_AWK_EXTIO_PIPE_READ)
|
|
{
|
|
mode = ASE_SIO_READ;
|
|
}
|
|
else if (epa->mode == ASE_AWK_EXTIO_PIPE_WRITE)
|
|
{
|
|
mode = ASE_SIO_WRITE |
|
|
ASE_SIO_TRUNCATE |
|
|
ASE_SIO_CREATE;
|
|
}
|
|
else return -1; /* TODO: any way to set the error number? */
|
|
|
|
//dprint (ASE_T("opening %s of type %d (pipe)\n"), epa->name, epa->type);
|
|
|
|
// TOOD: pipe open....
|
|
handle = ase_sio_open (ase_awk_getrunmmgr(epa->run), 0, epa->name, mode);
|
|
if (handle == ASE_NULL) return -1;
|
|
epa->handle = (void*)handle;
|
|
return 1;
|
|
}
|
|
|
|
case ASE_AWK_IO_CLOSE:
|
|
{
|
|
//dprint (ASE_T("closing %s of type (pipe) %d\n"), epa->name, epa->type);
|
|
ase_sio_close ((ase_sio_t*)epa->handle);
|
|
epa->handle = ASE_NULL;
|
|
return 0;
|
|
}
|
|
|
|
case ASE_AWK_IO_READ:
|
|
{
|
|
return ase_sio_getsx (
|
|
(ase_sio_t*)epa->handle,
|
|
data,
|
|
size
|
|
);
|
|
}
|
|
|
|
case ASE_AWK_IO_WRITE:
|
|
{
|
|
return ase_sio_putsx (
|
|
(ase_sio_t*)epa->handle,
|
|
data,
|
|
size
|
|
);
|
|
}
|
|
|
|
case ASE_AWK_IO_FLUSH:
|
|
{
|
|
/*if (epa->mode == ASE_AWK_EXTIO_PIPE_READ) return -1;*/
|
|
return ase_sio_flush ((ase_sio_t*)epa->handle);
|
|
}
|
|
|
|
case ASE_AWK_IO_NEXT:
|
|
{
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
static ase_ssize_t awk_extio_file (
|
|
int cmd, void* arg, ase_char_t* data, ase_size_t size)
|
|
{
|
|
ase_awk_extio_t* epa = (ase_awk_extio_t*)arg;
|
|
|
|
switch (cmd)
|
|
{
|
|
case ASE_AWK_IO_OPEN:
|
|
{
|
|
ase_sio_t* handle;
|
|
int mode;
|
|
|
|
if (epa->mode == ASE_AWK_EXTIO_FILE_READ)
|
|
{
|
|
mode = ASE_SIO_READ;
|
|
}
|
|
else if (epa->mode == ASE_AWK_EXTIO_FILE_WRITE)
|
|
{
|
|
mode = ASE_SIO_WRITE |
|
|
ASE_SIO_CREATE |
|
|
ASE_SIO_TRUNCATE;
|
|
}
|
|
else if (epa->mode == ASE_AWK_EXTIO_FILE_APPEND)
|
|
{
|
|
mode = ASE_SIO_APPEND |
|
|
ASE_SIO_CREATE;
|
|
}
|
|
else return -1; /* TODO: any way to set the error number? */
|
|
|
|
//dprint (ASE_T("opening %s of type %d (file)\n"), epa->name, epa->type);
|
|
handle = ase_sio_open (
|
|
ase_awk_getrunmmgr(epa->run),
|
|
0,
|
|
epa->name,
|
|
mode
|
|
);
|
|
if (handle == ASE_NULL)
|
|
{
|
|
ase_cstr_t errarg;
|
|
|
|
errarg.ptr = epa->name;
|
|
errarg.len = ase_strlen(epa->name);
|
|
|
|
ase_awk_setrunerror (epa->run, ASE_AWK_EOPEN, 0, &errarg, 1);
|
|
return -1;
|
|
}
|
|
|
|
epa->handle = (void*)handle;
|
|
return 1;
|
|
}
|
|
|
|
case ASE_AWK_IO_CLOSE:
|
|
{
|
|
//dprint (ASE_T("closing %s of type %d (file)\n"), epa->name, epa->type);
|
|
ase_sio_close ((ase_sio_t*)epa->handle);
|
|
epa->handle = ASE_NULL;
|
|
return 0;
|
|
}
|
|
|
|
case ASE_AWK_IO_READ:
|
|
{
|
|
return ase_sio_getsx (
|
|
(ase_sio_t*)epa->handle,
|
|
data,
|
|
size
|
|
);
|
|
}
|
|
|
|
case ASE_AWK_IO_WRITE:
|
|
{
|
|
return ase_sio_putsx (
|
|
(ase_sio_t*)epa->handle,
|
|
data,
|
|
size
|
|
);
|
|
}
|
|
|
|
case ASE_AWK_IO_FLUSH:
|
|
{
|
|
return ase_sio_flush ((ase_sio_t*)epa->handle);
|
|
}
|
|
|
|
case ASE_AWK_IO_NEXT:
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
static int open_extio_console (ase_awk_extio_t* epa)
|
|
{
|
|
runio_data_t* rd = (runio_data_t*)epa->data;
|
|
|
|
//dprint (ASE_T("opening console[%s] of type %x\n"), epa->name, epa->type);
|
|
|
|
if (epa->mode == ASE_AWK_EXTIO_CONSOLE_READ)
|
|
{
|
|
if (rd->ic.files[rd->ic.index] == ASE_NULL)
|
|
{
|
|
/* no more input file */
|
|
//dprint (ASE_T("console - no more file\n"));;
|
|
return 0;
|
|
}
|
|
|
|
if (rd->ic.files[rd->ic.index][0] == ASE_T('\0'))
|
|
{
|
|
//dprint (ASE_T(" console(r) - <standard input>\n"));
|
|
epa->handle = ase_sio_in;
|
|
}
|
|
else
|
|
{
|
|
/* a temporary variable fp is used here not to change
|
|
* any fields of epa when the open operation fails */
|
|
ase_sio_t* fp;
|
|
|
|
fp = ase_sio_open (
|
|
ase_awk_getrunmmgr(epa->run),
|
|
0,
|
|
rd->ic.files[rd->ic.index],
|
|
ASE_SIO_READ
|
|
);
|
|
if (fp == ASE_NULL)
|
|
{
|
|
ase_cstr_t errarg;
|
|
|
|
errarg.ptr = rd->ic.files[rd->ic.index];
|
|
errarg.len = ase_strlen(rd->ic.files[rd->ic.index]);
|
|
|
|
ase_awk_setrunerror (epa->run, ASE_AWK_EOPEN, 0, &errarg, 1);
|
|
return -1;
|
|
}
|
|
|
|
//dprint (ASE_T(" console(r) - %s\n"), rd->ic.files[rd->ic.index]);
|
|
if (ase_awk_setfilename (
|
|
epa->run, rd->ic.files[rd->ic.index],
|
|
ase_strlen(rd->ic.files[rd->ic.index])) == -1)
|
|
{
|
|
ase_sio_close (fp);
|
|
return -1;
|
|
}
|
|
|
|
epa->handle = fp;
|
|
}
|
|
|
|
rd->ic.index++;
|
|
return 1;
|
|
}
|
|
else if (epa->mode == ASE_AWK_EXTIO_CONSOLE_WRITE)
|
|
{
|
|
//dprint (ASE_T(" console(w) - <standard output>\n"));
|
|
|
|
if (ase_awk_setofilename (epa->run, ASE_T(""), 0) == -1)
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
epa->handle = ase_sio_out;
|
|
return 1;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
static ase_ssize_t awk_extio_console (
|
|
int cmd, void* arg, ase_char_t* data, ase_size_t size)
|
|
{
|
|
ase_awk_extio_t* epa = (ase_awk_extio_t*)arg;
|
|
runio_data_t* rd = (runio_data_t*)epa->data;
|
|
|
|
if (cmd == ASE_AWK_IO_OPEN)
|
|
{
|
|
return open_extio_console (epa);
|
|
}
|
|
else if (cmd == ASE_AWK_IO_CLOSE)
|
|
{
|
|
//dprint (ASE_T("closing console of type %x\n"), epa->type);
|
|
|
|
if (epa->handle != ASE_NULL &&
|
|
epa->handle != ase_sio_in &&
|
|
epa->handle != ase_sio_out &&
|
|
epa->handle != ase_sio_err)
|
|
{
|
|
ase_sio_close ((ase_sio_t*)epa->handle);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
else if (cmd == ASE_AWK_IO_READ)
|
|
{
|
|
ase_ssize_t n;
|
|
|
|
while ((n = ase_sio_getsx((ase_sio_t*)epa->handle,data,size)) == 0)
|
|
{
|
|
/* it has reached the end of the current file.
|
|
* open the next file if available */
|
|
if (rd->ic.files[rd->ic.index] == ASE_NULL)
|
|
{
|
|
/* no more input console */
|
|
return 0;
|
|
}
|
|
|
|
if (rd->ic.files[rd->ic.index][0] == ASE_T('\0'))
|
|
{
|
|
if (epa->handle != ASE_NULL &&
|
|
epa->handle != ase_sio_in &&
|
|
epa->handle != ase_sio_out &&
|
|
epa->handle != ase_sio_err)
|
|
{
|
|
ase_sio_close ((ase_sio_t*)epa->handle);
|
|
}
|
|
|
|
epa->handle = ase_sio_in;
|
|
}
|
|
else
|
|
{
|
|
ase_sio_t* fp;
|
|
|
|
fp = ase_sio_open (
|
|
ase_awk_getrunmmgr(epa->run),
|
|
0,
|
|
rd->ic.files[rd->ic.index],
|
|
ASE_SIO_READ
|
|
);
|
|
|
|
if (fp == ASE_NULL)
|
|
{
|
|
ase_cstr_t errarg;
|
|
|
|
errarg.ptr = rd->ic.files[rd->ic.index];
|
|
errarg.len = ase_strlen(rd->ic.files[rd->ic.index]);
|
|
|
|
ase_awk_setrunerror (epa->run, ASE_AWK_EOPEN, 0, &errarg, 1);
|
|
return -1;
|
|
}
|
|
|
|
if (ase_awk_setfilename (
|
|
epa->run, rd->ic.files[rd->ic.index],
|
|
ase_strlen(rd->ic.files[rd->ic.index])) == -1)
|
|
{
|
|
ase_sio_close (fp);
|
|
return -1;
|
|
}
|
|
|
|
if (ase_awk_setglobal (
|
|
epa->run, ASE_AWK_GLOBAL_FNR, ase_awk_val_zero) == -1)
|
|
{
|
|
/* need to reset FNR */
|
|
ase_sio_close (fp);
|
|
return -1;
|
|
}
|
|
|
|
if (epa->handle != ASE_NULL &&
|
|
epa->handle != ase_sio_in &&
|
|
epa->handle != ase_sio_out &&
|
|
epa->handle != ase_sio_err)
|
|
{
|
|
ase_sio_close ((ase_sio_t*)epa->handle);
|
|
}
|
|
|
|
//dprint (ASE_T("open the next console [%s]\n"), rd->ic.files[rd->ic.index]);
|
|
epa->handle = fp;
|
|
}
|
|
|
|
rd->ic.index++;
|
|
}
|
|
|
|
return n;
|
|
}
|
|
else if (cmd == ASE_AWK_IO_WRITE)
|
|
{
|
|
return ase_sio_putsx (
|
|
(ase_sio_t*)epa->handle,
|
|
data,
|
|
size
|
|
);
|
|
}
|
|
else if (cmd == ASE_AWK_IO_FLUSH)
|
|
{
|
|
return ase_sio_flush ((ase_sio_t*)epa->handle);
|
|
}
|
|
else if (cmd == ASE_AWK_IO_NEXT)
|
|
{
|
|
int n;
|
|
ase_sio_t* fp = (ase_sio_t*)epa->handle;
|
|
|
|
//dprint (ASE_T("switching console[%s] of type %x\n"), epa->name, epa->type);
|
|
|
|
n = open_extio_console(epa);
|
|
if (n == -1) return -1;
|
|
|
|
if (n == 0)
|
|
{
|
|
/* if there is no more file, keep the previous handle */
|
|
return 0;
|
|
}
|
|
|
|
if (fp != ASE_NULL &&
|
|
fp != ase_sio_in &&
|
|
fp != ase_sio_out &&
|
|
fp != ase_sio_err)
|
|
{
|
|
ase_sio_close (fp);
|
|
}
|
|
|
|
return n;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
int ase_awk_runsimple (ase_awk_t* awk, ase_char_t** icf, ase_awk_runcbs_t* cbs)
|
|
{
|
|
ase_awk_runios_t ios;
|
|
runio_data_t rd;
|
|
rxtn_t rxtn;
|
|
ase_ntime_t now;
|
|
|
|
rd.ic.files = icf;
|
|
rd.ic.index = 0;
|
|
|
|
ios.pipe = awk_extio_pipe;
|
|
ios.file = awk_extio_file;
|
|
ios.console = awk_extio_console;
|
|
ios.data = &rd;
|
|
|
|
if (ase_gettime (&now) == -1) rxtn.seed = 0;
|
|
else rxtn.seed = (unsigned int)now;
|
|
srand (rxtn.seed);
|
|
|
|
return ase_awk_run (
|
|
awk,
|
|
ASE_NULL/*mfn*/,
|
|
&ios,
|
|
cbs,
|
|
ASE_NULL/*runarg*/,
|
|
&rxtn/*ASE_NULL*/
|
|
);
|
|
}
|
|
|
|
|
|
/*** EXTRA BUILTIN FUNCTIONS ***/
|
|
enum
|
|
{
|
|
BFN_MATH_LD,
|
|
BFN_MATH_D,
|
|
BFN_MATH_F
|
|
};
|
|
|
|
static int bfn_math_1 (
|
|
ase_awk_run_t* run, const ase_char_t* fnm, ase_size_t fnl, int type, void* f)
|
|
{
|
|
ase_size_t nargs;
|
|
ase_awk_val_t* a0;
|
|
ase_long_t lv;
|
|
ase_real_t rv;
|
|
ase_awk_val_t* r;
|
|
int n;
|
|
|
|
nargs = ase_awk_getnargs (run);
|
|
ASE_ASSERT (nargs == 1);
|
|
|
|
a0 = ase_awk_getarg (run, 0);
|
|
|
|
n = ase_awk_valtonum (run, a0, &lv, &rv);
|
|
if (n == -1) return -1;
|
|
if (n == 0) rv = (ase_real_t)lv;
|
|
|
|
if (type == BFN_MATH_LD)
|
|
{
|
|
long double (*rf) (long double) =
|
|
(long double(*)(long double))f;
|
|
r = ase_awk_makerealval (run, rf(rv));
|
|
}
|
|
else if (type == BFN_MATH_D)
|
|
{
|
|
double (*rf) (double) = (double(*)(double))f;
|
|
r = ase_awk_makerealval (run, rf(rv));
|
|
}
|
|
else
|
|
{
|
|
ASE_ASSERT (type == BFN_MATH_F);
|
|
float (*rf) (float) = (float(*)(float))f;
|
|
r = ase_awk_makerealval (run, rf(rv));
|
|
}
|
|
|
|
if (r == ASE_NULL)
|
|
{
|
|
ase_awk_setrunerrnum (run, ASE_AWK_ENOMEM);
|
|
return -1;
|
|
}
|
|
|
|
ase_awk_setretval (run, r);
|
|
return 0;
|
|
}
|
|
|
|
static int bfn_math_2 (
|
|
ase_awk_run_t* run, const ase_char_t* fnm, ase_size_t fnl, int type, void* f)
|
|
{
|
|
ase_size_t nargs;
|
|
ase_awk_val_t* a0, * a1;
|
|
ase_long_t lv0, lv1;
|
|
ase_real_t rv0, rv1;
|
|
ase_awk_val_t* r;
|
|
int n;
|
|
|
|
nargs = ase_awk_getnargs (run);
|
|
ASE_ASSERT (nargs == 2);
|
|
|
|
a0 = ase_awk_getarg (run, 0);
|
|
a1 = ase_awk_getarg (run, 1);
|
|
|
|
n = ase_awk_valtonum (run, a0, &lv0, &rv0);
|
|
if (n == -1) return -1;
|
|
if (n == 0) rv0 = (ase_real_t)lv0;
|
|
|
|
n = ase_awk_valtonum (run, a1, &lv1, &rv1);
|
|
if (n == -1) return -1;
|
|
if (n == 0) rv1 = (ase_real_t)lv1;
|
|
|
|
if (type == BFN_MATH_LD)
|
|
{
|
|
long double (*rf) (long double,long double) =
|
|
(long double(*)(long double,long double))f;
|
|
r = ase_awk_makerealval (run, rf(rv0,rv1));
|
|
}
|
|
else if (type == BFN_MATH_D)
|
|
{
|
|
double (*rf) (double,double) = (double(*)(double,double))f;
|
|
r = ase_awk_makerealval (run, rf(rv0,rv1));
|
|
}
|
|
else
|
|
{
|
|
ASE_ASSERT (type == BFN_MATH_F);
|
|
float (*rf) (float,float) = (float(*)(float,float))f;
|
|
r = ase_awk_makerealval (run, rf(rv0,rv1));
|
|
}
|
|
|
|
if (r == ASE_NULL)
|
|
{
|
|
ase_awk_setrunerrnum (run, ASE_AWK_ENOMEM);
|
|
return -1;
|
|
}
|
|
|
|
ase_awk_setretval (run, r);
|
|
return 0;
|
|
}
|
|
|
|
static int bfn_sin (ase_awk_run_t* run, const ase_char_t* fnm, ase_size_t fnl)
|
|
{
|
|
return bfn_math_1 (run, fnm, fnl,
|
|
#if defined(HAVE_SINL)
|
|
BFN_MATH_LD, sinl
|
|
#elif defined(HAVE_SIN)
|
|
BFN_MATH_D, sin
|
|
#elif defined(HAVE_SINF)
|
|
BFN_MATH_F, sinf
|
|
#else
|
|
#error ### no sin function available ###
|
|
#endif
|
|
);
|
|
}
|
|
|
|
static int bfn_cos (ase_awk_run_t* run, const ase_char_t* fnm, ase_size_t fnl)
|
|
{
|
|
return bfn_math_1 (run, fnm, fnl,
|
|
#if defined(HAVE_COSL)
|
|
BFN_MATH_LD, cosl
|
|
#elif defined(HAVE_COS)
|
|
BFN_MATH_D, cos
|
|
#elif defined(HAVE_COSF)
|
|
BFN_MATH_F, cosf
|
|
#else
|
|
#error ### no cos function available ###
|
|
#endif
|
|
);
|
|
}
|
|
|
|
static int bfn_tan (ase_awk_run_t* run, const ase_char_t* fnm, ase_size_t fnl)
|
|
{
|
|
return bfn_math_1 (run, fnm, fnl,
|
|
#if defined(HAVE_TANL)
|
|
BFN_MATH_LD, tanl
|
|
#elif defined(HAVE_TAN)
|
|
BFN_MATH_D, tan
|
|
#elif defined(HAVE_TANF)
|
|
BFN_MATH_F, tanf
|
|
#else
|
|
#error ### no tan function available ###
|
|
#endif
|
|
);
|
|
}
|
|
|
|
static int bfn_atan (ase_awk_run_t* run, const ase_char_t* fnm, ase_size_t fnl)
|
|
{
|
|
return bfn_math_1 (run, fnm, fnl,
|
|
#if defined(HAVE_ATANL)
|
|
BFN_MATH_LD, atanl
|
|
#elif defined(HAVE_ATAN)
|
|
BFN_MATH_D, atan
|
|
#elif defined(HAVE_ATANF)
|
|
BFN_MATH_F, atanf
|
|
#else
|
|
#error ### no atan function available ###
|
|
#endif
|
|
);
|
|
}
|
|
|
|
static int bfn_atan2 (ase_awk_run_t* run, const ase_char_t* fnm, ase_size_t fnl)
|
|
{
|
|
return bfn_math_2 (run, fnm, fnl,
|
|
#if defined(HAVE_ATAN2L)
|
|
BFN_MATH_LD, atan2l
|
|
#elif defined(HAVE_ATAN2)
|
|
BFN_MATH_D, atan2
|
|
#elif defined(HAVE_ATAN2F)
|
|
BFN_MATH_F, atan2f
|
|
#else
|
|
#error ### no atan2 function available ###
|
|
#endif
|
|
);
|
|
}
|
|
|
|
static int bfn_log (ase_awk_run_t* run, const ase_char_t* fnm, ase_size_t fnl)
|
|
{
|
|
return bfn_math_1 (run, fnm, fnl,
|
|
#if defined(HAVE_LOGL)
|
|
BFN_MATH_LD, logl
|
|
#elif defined(HAVE_LOG)
|
|
BFN_MATH_D, log
|
|
#elif defined(HAVE_LOGF)
|
|
BFN_MATH_F, logf
|
|
#else
|
|
#error ### no log function available ###
|
|
#endif
|
|
);
|
|
}
|
|
|
|
static int bfn_exp (ase_awk_run_t* run, const ase_char_t* fnm, ase_size_t fnl)
|
|
{
|
|
return bfn_math_1 (run, fnm, fnl,
|
|
#if defined(HAVE_EXPL)
|
|
BFN_MATH_LD, expl
|
|
#elif defined(HAVE_EXP)
|
|
BFN_MATH_D, exp
|
|
#elif defined(HAVE_EXPF)
|
|
BFN_MATH_F, expf
|
|
#else
|
|
#error ### no exp function available ###
|
|
#endif
|
|
);
|
|
}
|
|
|
|
static int bfn_sqrt (ase_awk_run_t* run, const ase_char_t* fnm, ase_size_t fnl)
|
|
{
|
|
return bfn_math_1 (run, fnm, fnl,
|
|
#if defined(HAVE_SQRTL)
|
|
BFN_MATH_LD, sqrtl
|
|
#elif defined(HAVE_SQRT)
|
|
BFN_MATH_D, sqrt
|
|
#elif defined(HAVE_SQRTF)
|
|
BFN_MATH_F, sqrtf
|
|
#else
|
|
#error ### no sqrt function available ###
|
|
#endif
|
|
);
|
|
}
|
|
|
|
static int bfn_int (ase_awk_run_t* run, const ase_char_t* fnm, ase_size_t fnl)
|
|
{
|
|
ase_size_t nargs;
|
|
ase_awk_val_t* a0;
|
|
ase_long_t lv;
|
|
ase_real_t rv;
|
|
ase_awk_val_t* r;
|
|
int n;
|
|
|
|
nargs = ase_awk_getnargs (run);
|
|
ASE_ASSERT (nargs == 1);
|
|
|
|
a0 = ase_awk_getarg (run, 0);
|
|
|
|
n = ase_awk_valtonum (run, a0, &lv, &rv);
|
|
if (n == -1) return -1;
|
|
if (n == 1) lv = (ase_long_t)rv;
|
|
|
|
r = ase_awk_makeintval (run, lv);
|
|
if (r == ASE_NULL)
|
|
{
|
|
ase_awk_setrunerrnum (run, ASE_AWK_ENOMEM);
|
|
return -1;
|
|
}
|
|
|
|
ase_awk_setretval (run, r);
|
|
return 0;
|
|
}
|
|
|
|
static int bfn_rand (ase_awk_run_t* run, const ase_char_t* fnm, ase_size_t fnl)
|
|
{
|
|
ase_awk_val_t* r;
|
|
|
|
r = ase_awk_makeintval (run, rand());
|
|
if (r == ASE_NULL)
|
|
{
|
|
ase_awk_setrunerrnum (run, ASE_AWK_ENOMEM);
|
|
return -1;
|
|
}
|
|
|
|
ase_awk_setretval (run, r);
|
|
return 0;
|
|
}
|
|
|
|
static int bfn_srand (ase_awk_run_t* run, const ase_char_t* fnm, ase_size_t fnl)
|
|
{
|
|
ase_size_t nargs;
|
|
ase_awk_val_t* a0;
|
|
ase_long_t lv;
|
|
ase_real_t rv;
|
|
ase_awk_val_t* r;
|
|
int n;
|
|
unsigned int prev;
|
|
rxtn_t* rxtn;
|
|
|
|
rxtn = ase_awk_getrundata (run);
|
|
nargs = ase_awk_getnargs (run);
|
|
ASE_ASSERT (nargs == 0 || nargs == 1);
|
|
|
|
prev = rxtn->seed;
|
|
|
|
if (nargs == 1)
|
|
{
|
|
a0 = ase_awk_getarg (run, 0);
|
|
|
|
n = ase_awk_valtonum (run, a0, &lv, &rv);
|
|
if (n == -1) return -1;
|
|
if (n == 1) lv = (ase_long_t)rv;
|
|
|
|
rxtn->seed = lv;
|
|
}
|
|
else
|
|
{
|
|
ase_ntime_t now;
|
|
|
|
if (ase_gettime(&now) == -1) rxtn->seed >>= 1;
|
|
else rxtn->seed = (unsigned int)now;
|
|
}
|
|
|
|
srand (rxtn->seed);
|
|
|
|
r = ase_awk_makeintval (run, prev);
|
|
if (r == ASE_NULL)
|
|
{
|
|
ase_awk_setrunerrnum (run, ASE_AWK_ENOMEM);
|
|
return -1;
|
|
}
|
|
|
|
ase_awk_setretval (run, r);
|
|
return 0;
|
|
}
|
|
|
|
static int bfn_systime (ase_awk_run_t* run, const ase_char_t* fnm, ase_size_t fnl)
|
|
{
|
|
ase_awk_val_t* r;
|
|
ase_ntime_t now;
|
|
int n;
|
|
|
|
if (ase_gettime(&now) == -1)
|
|
r = ase_awk_makeintval (run, ASE_TYPE_MIN(ase_long_t));
|
|
else
|
|
r = ase_awk_makeintval (run, now / ASE_MSEC_IN_SEC);
|
|
|
|
if (r == ASE_NULL)
|
|
{
|
|
ase_awk_setrunerrnum (run, ASE_AWK_ENOMEM);
|
|
return -1;
|
|
}
|
|
|
|
ase_awk_setretval (run, r);
|
|
return 0;
|
|
}
|
|
|
|
#define ADD_FUNC(awk,name,min,max,bfn) \
|
|
if (ase_awk_addfunc (\
|
|
(awk), (name), ase_strlen(name), \
|
|
0, (min), (max), ASE_NULL, (bfn)) == ASE_NULL) return -1;
|
|
|
|
static int add_functions (ase_awk_t* awk)
|
|
{
|
|
ADD_FUNC (awk, ASE_T("sin"), 1, 1, bfn_sin);
|
|
ADD_FUNC (awk, ASE_T("cos"), 1, 1, bfn_cos);
|
|
ADD_FUNC (awk, ASE_T("tan"), 1, 1, bfn_tan);
|
|
ADD_FUNC (awk, ASE_T("atan"), 1, 1, bfn_atan);
|
|
ADD_FUNC (awk, ASE_T("atan2"), 2, 2, bfn_atan2);
|
|
ADD_FUNC (awk, ASE_T("log"), 1, 1, bfn_log);
|
|
ADD_FUNC (awk, ASE_T("exp"), 1, 1, bfn_exp);
|
|
ADD_FUNC (awk, ASE_T("sqrt"), 1, 1, bfn_sqrt);
|
|
ADD_FUNC (awk, ASE_T("int"), 1, 1, bfn_int);
|
|
ADD_FUNC (awk, ASE_T("rand"), 0, 0, bfn_rand);
|
|
ADD_FUNC (awk, ASE_T("srand"), 0, 1, bfn_srand);
|
|
ADD_FUNC (awk, ASE_T("systime"), 0, 0, bfn_systime);
|
|
/*
|
|
ADD_FUNC (awk, ASE_T("strftime"), 0, 2, bfn_strftime);
|
|
ADD_FUNC (awk, ASE_T("strfgmtime"), 0, 2, bfn_strfgmtime);
|
|
ADD_FUNC (awk, ASE_T("system"), 1, 1, bfn_system);
|
|
*/
|
|
|
|
return 0;
|
|
}
|