/* makeimp.c

Author: Jay Han  (jayhan@cs.washington.edu)
Date: 3/22/94

This program automatically generates imp-spiral launch code.

Features:
   - Any number of points, any number of processes, in any size of core
   - Automatically computes imp-step
   - Two methods: jmp/add or binary
   - Works under both '88 and '94 Draft standards
   - Redefinable labels

Bugs:
   - Does not generate ultimately optimized code in some cases

*******************************************************************************/

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

#define LABELS		4
#define LABELSIZE	16

long i, p=0;
int std=94, *labels;
char labelname[LABELS][LABELSIZE] = {"launch", "imp", "step", "lbl"};

#define L_LAUNCH	0
#define L_IMP		1
#define L_STEP		2
#define L_LABEL		3

#define launch	(labelname[L_LAUNCH])
#define imp		(labelname[L_IMP])
#define step	(labelname[L_STEP])
#define label	(labelname[L_LABEL])

void makelabel (long l)
{
	if (labels[l])
		printf ("%s%ld\t", label, l);
	else printf ("\t");
}

void makeimp (long s, long q, int g)
{
	long n, z;

	if (g == 0)
	{
		z = q%(1<<i) ;
		makelabel (q);
		printf ("jmp\t%s+%ld*%s+%ld\n", imp, z%p, step, z/p);
		return;
	}

	if (s % 2 == 0)
	{
		for (n=0; s%2==0 && n<g; n++) s >>= 1;
		if (n == 1)
		{
			makelabel (q);
			printf ("jmp\t1\t\t; Idle 1 cycle\n");
		}
		else
		{
			if (std == 88)
			{
				makelabel (q);
				printf ("djn\t0,\t#%ld\t; Idle %ld cycles\n", n, n);
			}
			else
			{
				makelabel (q);
				printf ("djn.b\t0,\t#%ld\t; Idle %ld cycles\n", n, n);
			}
		}
		if (g-n == 0)
		{
			z = (q<<n)%(1<<i) ;
			printf ("\tjmp\t%s+%ld*%s+%ld\n", imp, z%p, step, z/p);
		}
		else makeimp (s, q<<n, g-n);
	}
	else
	{
		makelabel (q);
		printf ("spl\t%s%ld\n", label, (q<<1)+1);
		labels[(q<<1)+1] = 1;
		makeimp ((1<<g)-1, q<<1, g-1);
		makeimp (s>>1, (q<<1)+1, g-1);
	}
}

char *progname;

void error (void)
{
	fprintf (stderr,
			 "Syntax: %s <imp> [s #] [a|b] [p|x #] [r #] [l XXX=YYY]\n",
			 progname);
	fprintf (stderr, "     Options:                    Default:\n");
	fprintf (stderr, "imp  imp-number                  -- mandatory --\n");
	fprintf (stderr, "s #  coresize                    55440\n");
	fprintf (stderr, "a|b  mode: jmp/add or binary     binary\n");
	fprintf (stderr, "p #  number of processes         same as imp-number\n");
	fprintf (stderr, "x #  number of layers (p=x*imp)  1\n");
	fprintf (stderr, "r #  redcode standard            94\n");
	fprintf (stderr, "l #  change label XXX to YYY     launch imp step lbl\n");
	exit (-1);
}

int main (int argc, char *argv[])
{
	long m=55440, r, c, n, x, q, s=0, t=1;
	int j, k;
	char mode='b', *l;

	progname = argv[0];
	if (argc == 1) error();

	for (j=1; j<argc; j++)
	{
		switch (argv[j][0])
		{
		  case 's': sscanf (argv[++j], "%ld", &m); break;
		  case 'x':	sscanf (argv[++j], "%ld", &t); break;
		  case 'p': sscanf (argv[++j], "%ld", &s); break;
		  case 'a': case 'b': mode = argv[j][0]; break;
		  case 'r':	sscanf (argv[++j], "%d", &std); break;
		  case 'l': 
			l = index (argv[++j], '=');
			if (l==NULL) error();
			*l = 0; l++;
			for (k=0; k<LABELS; k++)
			{
				if (strcmp (argv[j], labelname[k]) == 0)
				{
					strncpy (labelname[k], l, LABELSIZE);
					break;
				}
			}
			if (k==LABELS)
			{
				fprintf (stderr, "Label `%s' not used.\n");
			}
			break;
		  default: sscanf (argv[j], "%ld", &p); break;
		}
	}

	if (s==0) s = t * p;
	if (p==0)
	{
		fprintf (stderr, "You must specify an imp-number!\n\n");
		error();
	}
	if (s<p)
	{
		fprintf (stderr,
				 "Number of processes must be at least equal to number of points!\n\n");
		error();
	}

	r = m % p;
	c = 0;
	for (n=1; n<p; n++) if ((c=(c+r)%p) == p-1) break;
	if (n >= p)
	{
		fprintf (stderr, "%ld is not an imp-number for coresize %ld!\n", p, m);
		exit (-1);
	}

	x = (m*n+1)/p;
	printf ("; %ld-process %ld-point spiral (coresize %ld, '%d standard)\n",
			s, p, m, std);
	printf ("%s\tequ\t%ld\n", step, x);
	s--;
	for (q=0, i=0; s; s/=2, i++) q = (q * 2) | (s % 2);

	printf ("%s\n", launch);
	switch (mode)
	{
	  case 'b':
		labels = (int*) calloc (2<<i, sizeof(int));
		makeimp (q, 1, i);
		break;

	  case 'a':
		for (; i; i--)
		{
			if (q % 2) printf ("\tspl\t1\n");
			else
			{
				if (std == 88) printf ("\tmov\t-1,\t0\n");
				else printf ("\tmov.i\t-1,\t0\n");
			}
			q >>= 1;
		}
		printf ("\tspl\t2\n");
		if (std == 88)
		{
			printf ("\tjmp\t@0,\t%s\n", imp);
			printf ("\tadd\t#%s,\t-1\n", step);
		}
		else
		{
			printf ("\tjmp\t%s\n", imp);
			printf ("\tadd.a\t#%s,\t-1\n", step);
		}
		printf ("\tdat\t#0,\t#0\t; This must be a DAT\n");
		break;

	  default: exit (-1);
	}

	if (std == 88) printf ("%s\tmov\t0,\t%s\n", imp, step);
	else printf ("%s\tmov.i\t0,\t%s\n", imp, step);

	return 0;
}