quakeforge/libs/gib/gib_execute.c

/*
	gib_execute.c

	GIB runtime execution functions

	Copyright (C) 2002 Brian Koropoff

	Author: Brian Koropoff
	Date: #DATE#

	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; either version 2
	of the License, or (at your option) any later version.

	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:

		Free Software Foundation, Inc.
		59 Temple Place - Suite 330
		Boston, MA  02111-1307, USA

*/

#ifdef HAVE_CONFIG_H
# include "config.h"
#endif

static __attribute__ ((unused)) const char rcsid[] =
        "$Id$";

#include <string.h>
#include <stdlib.h>

#include "QF/cbuf.h"
#include "QF/cmd.h"
#include "QF/gib_buffer.h"
#include "QF/gib_vars.h"
#include "QF/gib_process.h"
#include "QF/gib_builtin.h"
#include "QF/gib_function.h"
#include "QF/gib_execute.h"

static void
GIB_Execute_Generate_Composite (struct cbuf_s *cbuf)
{
	cbuf_args_t *args = cbuf->args;
	int i;
	
	dstring_clearstr (GIB_DATA (cbuf)->arg_composite);
	for (i = 0; i < args->argc; i++) {
		// ->str could be moved by realloc when a dstring is resized
		// so save the offset instead of the pointer
		args->args[i] = (const char *) strlen (GIB_DATA (cbuf)->arg_composite->str);
		dstring_appendstr (GIB_DATA (cbuf)->arg_composite, args->argv[i]->str);
		dstring_appendstr (GIB_DATA (cbuf)->arg_composite, " ");
	}
	
	// Get rid of trailing space
	GIB_DATA (cbuf)->arg_composite->str[strlen(GIB_DATA (cbuf)->arg_composite->str)-1] = 0;
	
	for (i = 0; i < args->argc; i++)
		// now that arg_composite is done we can add the pointer to the stored
		// offsets and get valid pointers. This *should* be portable.
		args->args[i] += (unsigned long int) GIB_DATA (cbuf)->arg_composite->str;
}		

static void
GIB_Execute_Split_Array (cbuf_t *cbuf)
{
	gib_var_t *var;
	unsigned int i;
	int start = 0, end = 0;
	char *c, *str = cbuf->args->argv[cbuf->args->argc-1]->str+1;
	void *m = cbuf->args->argm[cbuf->args->argc-1];

	i = strlen(str)-1;
	if (str[i] == ']')
		for (; i; i--)
			if (str[i] == '[') {
				str[i] = 0;
				start = atoi (str+i+1);
				if ((c = strchr (str+i+1, ':'))) {
					if (c[1] != ']')
						end = atoi (c+1);
					else
						end = (int) ((unsigned int)~0>>1);
				} else
					end = start+1;
				break;
			}
	cbuf->args->argc--;
	if (!(var = GIB_Var_Get_Complex (&GIB_DATA(cbuf)->locals, &GIB_DATA(cbuf)->globals, str, &i, false)))
		return;
	if (end < 0)
		end += var->size;
	else if (end > var->size)
		end = var->size;
	if (start < 0) {
		start += var->size;
		if (start < 0)
			start = 0;
	} else if (start >= var->size || start >= end)
		return;
	for (i = start; i < end; i++) {
		if (var->array[i])
			Cbuf_ArgsAdd (cbuf->args, var->array[i]->str);
		else
			Cbuf_ArgsAdd (cbuf->args, "");
		cbuf->args->argm[cbuf->args->argc-1] = m;
	}
}

static int
GIB_Execute_Prepare_Line (cbuf_t *cbuf, gib_tree_t *line)
{
	gib_tree_t *cur;
	cbuf_args_t *args = cbuf->args;
	unsigned int pos;
	
	args->argc = 0;
	
	for (cur = line->children; cur; cur = cur->next) {
		if (cur->flags & TREE_CONCAT) {
			pos = args->argv[args->argc-1]->size-1;
			if (cur->flags & TREE_P_EMBED) {
				GIB_Process_Embedded (cur, cbuf->args);
			} else
				dstring_appendstr (args->argv[args->argc-1], cur->str);
		} else {
			pos = 0;
			if (cur->flags & TREE_P_EMBED) {
				Cbuf_ArgsAdd (args, "");
				GIB_Process_Embedded (cur, cbuf->args);
			} else
				Cbuf_ArgsAdd (args, cur->str);
			args->argm[args->argc-1] = cur;
		}
		if (cur->delim == '(' && GIB_Process_Math (args->argv[args->argc-1], pos))
			return -1;
		if (cur->flags & TREE_ASPLIT)
			GIB_Execute_Split_Array (cbuf);
	}
	return 0;
}

int
GIB_Execute_For_Next (cbuf_t *cbuf)
{
	unsigned int index;
	gib_var_t *var;
	struct gib_dsarray_s *array = GIB_DATA(cbuf)->stack.values+GIB_DATA(cbuf)->stack.p-1;
	if (array->size == 1) {
		GIB_Buffer_Pop_Sstack (cbuf);
		return 0;
	}
	array->size--;
	var = GIB_Var_Get_Complex (&GIB_DATA(cbuf)->locals, &GIB_DATA(cbuf)->globals, array->dstrs[0]->str, &index, true);
	dstring_clearstr (var->array[index]);
	dstring_appendstr (var->array[index], array->dstrs[array->size]->str);
	return 1;
}

void
GIB_Execute (cbuf_t *cbuf)
{
	gib_buffer_data_t *g = GIB_DATA (cbuf);
	gib_builtin_t *b;
	gib_function_t *f;
	int cond;
	
	if (!g->program)
		return;
	if (!g->ip)
		g->ip = g->program;
	while (!g->done) {
		if (GIB_Execute_Prepare_Line (cbuf, g->ip))
			return;
		if (g->ip->flags & TREE_COND) {
			cond = g->ip->flags & TREE_NOT ? atoi (cbuf->args->argv[1]->str) : !atoi (cbuf->args->argv[1]->str);
			if (cond)
				g->ip = g->ip->jump;
		} else if (g->ip->flags & TREE_FORNEXT) {
			if (GIB_Execute_For_Next (cbuf))
				g->ip = g->ip->jump;
		} else if (g->ip->flags & TREE_END) {
			g->ip = g->ip->jump;
			if (g->ip->flags & TREE_COND)
				continue;
		} else if (cbuf->args->argc) {
			if (g->ip->flags & TREE_EMBED) {
				// Get ready for return values
				g->waitret = true;
				GIB_Buffer_Push_Sstack (cbuf);
			} else
				g->waitret = false;
			if (cbuf->args->argc >= 2 && !strcmp (cbuf->args->argv[1]->str, "=") && ((gib_tree_t *)cbuf->args->argm[1])->delim == ' ') {
				unsigned int index;
				gib_var_t *var = GIB_Var_Get_Complex (&g->locals, &g->globals, cbuf->args->argv[0]->str, &index, true);
				GIB_Var_Assign (var, index, cbuf->args->argv+2, cbuf->args->argc-2);
			}
			else if ((b = GIB_Builtin_Find (cbuf->args->argv[0]->str))) {
				b->func ();
			} else if ((f = GIB_Function_Find (cbuf->args->argv[0]->str))) {
				cbuf_t *new = Cbuf_PushStack (&gib_interp);
				GIB_Function_Execute (new, f, cbuf->args->argv, cbuf->args->argc);
			} else {
				GIB_Execute_Generate_Composite (cbuf);
				if (Cmd_Command (cbuf->args))
					GIB_Error ("command", "No builtin, function, or console command named '%s' was found.", cbuf->args->argv[0]->str);
			}
		}
		if (!(g->ip = g->ip->next)) // No more commands
			g->done = true;
		if (cbuf->state) // Let the stack walker figure out what to do
			return;
	}
}