/* ========================================================================== */
/*            capaMapExpr.c    created by Isaac Tsai                       */
/*                                1998, 1999 copyrighted by Isaac Tsai     */
/*                                                                         */
/*  this is the main code to handle /MAP() function call in capa */
#include <stdio.h>
#include <string.h>
#include <math.h>

#include "capaParser.h"
#include "capaToken.h"
#include "ranlib.h"


            
/* |>|===============================================================|<| */
int
do_map(seed, varp, argp, argc, dir) 
char *seed; ArgNode_t *varp; ArgNode_t *argp; int argc; int dir;
{
  long        orig_gen, current_gen, seed1, seed2;
  long       *idx_array, *ridx_array;
  int         idx, val;
  Symbol      val_array[ONE_K];
  ArgNode_t  *tmpArgp;
  

  for(idx=0,tmpArgp=argp;idx<argc;idx++) {
    switch( FIRST_ARGTYPE(tmpArgp) ) {
    case I_VAR:
    case I_CONSTANT:
         (val_array[idx]).s_type = I_VAR;
         (val_array[idx]).s_int = FIRST_ARGINT(tmpArgp); break;
    case R_VAR:
    case R_CONSTANT:
         (val_array[idx]).s_type = R_VAR;
         (val_array[idx]).s_real = FIRST_ARGREAL(tmpArgp); break;
    case S_VAR:
    case S_CONSTANT:
         (val_array[idx]).s_type = S_VAR;
         (val_array[idx]).s_str = strsave(FIRST_ARGSTR(tmpArgp)); break;
    default:
         return -1;
	 break;
    }
    tmpArgp =  (tmpArgp->a_next);
  }
  idx_array = (long *)capa_malloc(sizeof(long), argc);
  for(idx=0;idx<argc;idx++) idx_array[idx] = idx;
  gscgn(GET_GENERATOR, &orig_gen);
  current_gen = PERMUTATION_G;
  gscgn(SET_GENERATOR, &current_gen);
  phrtsd(seed, &seed1, &seed2);
  setsd(seed1, seed2);
  genprm(idx_array, (long)argc);
  if(dir == REVERSE_MAP) {
    ridx_array = (long *)capa_malloc(sizeof(long), argc);
    for(idx=0;idx<argc;idx++) {
      ridx_array[ idx_array[idx] ] = idx;
    }
    for(idx=0;idx<argc;idx++) {
      idx_array[idx] = ridx_array[idx];
    }
    capa_mfree((char *)ridx_array);
  }
  for(idx=0,tmpArgp=varp;idx<argc;idx++) {
    val = idx_array[idx];
    switch( FIRST_ARGTYPE(tmpArgp) ) {
      case IDENTIFIER:
      case I_VAR: case I_CONSTANT:
      case R_VAR: case R_CONSTANT: break;
      case S_VAR: case S_CONSTANT: capa_mfree((char *)FIRST_ARGSTR(tmpArgp)); break;
    }
    FIRST_ARGTYPE(tmpArgp) = (val_array[val]).s_type;
    
    switch( val_array[val].s_type ) {
     case I_VAR: FIRST_ARGINT(tmpArgp) =  (val_array[val]).s_int;   break;
     case R_VAR: FIRST_ARGREAL(tmpArgp) = (val_array[val]).s_real;  break;
     case S_VAR: FIRST_ARGSTR(tmpArgp) =  (val_array[val]).s_str;   break;
    }
    tmpArgp = tmpArgp->a_next;
  }
  gscgn(SET_GENERATOR, &orig_gen);
  capa_mfree((char *)idx_array);
  return (0);
}