/*
** $Id: lfunc.c,v 1.45 2001/06/28 14:57:17 roberto Exp $
** Auxiliary functions to manipulate prototypes and closures
** See Copyright Notice in lua.h
*/


#include <stdlib.h>

#define LUA_PRIVATE
#include "lua.h"

#include "lfunc.h"
#include "lmem.h"
#include "lobject.h"
#include "lstate.h"


#define sizeCclosure(n)	(cast(int, sizeof(Closure)) + \
                         cast(int, sizeof(TObject)*((n)-1)))

#define sizeLclosure(n)	(cast(int, sizeof(Closure)) + \
                         cast(int, sizeof(TObject *)*((n)-1)))


Closure *luaF_newCclosure (lua_State *L, int nelems) {
  Closure *c = cast(Closure *, luaM_malloc(L, sizeCclosure(nelems)));
  c->isC = 1;
  c->next = G(L)->rootcl;
  G(L)->rootcl = c;
  c->mark = c;
  c->nupvalues = nelems;
  return c;
}


Closure *luaF_newLclosure (lua_State *L, int nelems) {
  Closure *c = cast(Closure *, luaM_malloc(L, sizeLclosure(nelems)));
  c->isC = 0;
  c->mark = c;
  c->u.l.isopen = 0;
  c->nupvalues = nelems;
  return c;
}


/*
** returns the open pointer in a closure that points higher into the stack
*/
static StkId uppoint (Closure *cl) {
  StkId lp = NULL;
  int i;
  lua_assert(cl->u.l.isopen);
  for (i=0; i<cl->nupvalues; i++) {
    if (!luaF_isclosed(cl, i))
      if (lp == NULL || cl->u.l.upvals[i] > lp)
        lp = cl->u.l.upvals[i];
  }
  lua_assert(lp != NULL);
  return lp;
}


void luaF_LConlist (lua_State *L, Closure *cl) {
  lua_assert(!cl->isC);
  if (cl->u.l.isopen == 0) {  /* no more open entries? */
    cl->next = G(L)->rootcl;  /* insert in final list */
    G(L)->rootcl = cl;
  }
  else {  /* insert in list of open closures, ordered by decreasing uppoints */
    StkId cli = uppoint(cl);
    Closure **p = &L->opencl;
    while (*p != NULL && uppoint(*p) > cli) p = &(*p)->next;
    cl->next = *p;
    *p = cl;
  }
}


static int closeCl (lua_State *L, Closure *cl, StkId level) {
  int got = 0;  /* flag: 1 if some pointer in the closure was corrected */
  int i;
  for  (i=0; i<cl->nupvalues; i++) {
    StkId var;
    if (!luaF_isclosed(cl, i) && (var=cl->u.l.upvals[i]) >= level) {
      if (ttype(var) != LUA_TUPVAL) {
        UpVal *v = luaM_new(L, UpVal);
        v->val = *var;
        v->marked = 0;
        v->next = G(L)->rootupval;
        G(L)->rootupval = v;
        setupvalue(var, v);
      }
      cl->u.l.upvals[i] = cast(TObject *, vvalue(var));
      luaF_closeentry(cl, i);
      got = 1;
    }
  }
  return got;
}


void luaF_close (lua_State *L, StkId level) {
  Closure *affected = NULL;  /* closures with open pointers >= level */
  Closure *cl;
  while ((cl=L->opencl) != NULL) {
    if (!closeCl(L, cl, level)) break;
    /* some pointer in `cl' changed; will re-insert it in original list */
    L->opencl = cl->next;  /* remove from original list */
    cl->next = affected;
    affected = cl;  /* insert in affected list */
  }
  /* re-insert all affected closures in original list */
  while ((cl=affected) != NULL) {
    affected = cl->next;
    luaF_LConlist(L, cl);
  }
}


Proto *luaF_newproto (lua_State *L) {
  Proto *f = luaM_new(L, Proto);
  f->k = NULL;
  f->sizek = 0;
  f->p = NULL;
  f->sizep = 0;
  f->code = NULL;
  f->sizecode = 0;
  f->nupvalues = 0;
  f->numparams = 0;
  f->is_vararg = 0;
  f->maxstacksize = 0;
  f->marked = 0;
  f->lineinfo = NULL;
  f->sizelocvars = 0;
  f->sizelineinfo = 0;
  f->locvars = NULL;
  f->lineDefined = 0;
  f->source = NULL;
  f->next = G(L)->rootproto;  /* chain in list of protos */
  G(L)->rootproto = f;
  return f;
}


void luaF_freeproto (lua_State *L, Proto *f) {
  luaM_freearray(L, f->code, f->sizecode, Instruction);
  luaM_freearray(L, f->locvars, f->sizelocvars, struct LocVar);
  luaM_freearray(L, f->k, f->sizek, TObject);
  luaM_freearray(L, f->p, f->sizep, Proto *);
  luaM_freearray(L, f->lineinfo, f->sizelineinfo, int);
  luaM_freelem(L, f);
}


void luaF_freeclosure (lua_State *L, Closure *c) {
  int size = (c->isC) ? sizeCclosure(c->nupvalues) : sizeLclosure(c->nupvalues);
  luaM_free(L, c, size);
}


/*
** Look for n-th local variable at line `line' in function `func'.
** Returns NULL if not found.
*/
const l_char *luaF_getlocalname (const Proto *f, int local_number, int pc) {
  int i;
  for (i = 0; i<f->sizelocvars && f->locvars[i].startpc <= pc; i++) {
    if (pc < f->locvars[i].endpc) {  /* is variable active? */
      local_number--;
      if (local_number == 0)
        return getstr(f->locvars[i].varname);
    }
  }
  return NULL;  /* not found */
}