/*
	Typical group example
*/

#include <stdio.h>
#include <strings.h>
#include "pvm3.h"

int
main(argc, argv)
int argc;
char *argv[];
{
	int mytid, mygid, ctid[32];
	int nproc, i, indx;
	char *gp;

	mytid = pvm_mytid();
	fprintf(stderr, "%s 0x%x enrolled\n", argv[0], mytid);
	if (mytid < 0) {
		pvm_perror(argv[0]);
		return -1;
		}

	if (argc != 2) goto usage;
	if ((nproc = atoi(argv[1])) < 1) goto usage;
	if (nproc > 32) goto usage;

	/* join a group */
	/* the group name is the last path of the path name */
	gp = rindex(argv[0],'/');
	if (gp == 0) 
		gp = argv[0];
	else
		gp++;
	mygid = pvm_joingroup(gp);

	fprintf(stderr, "%s 0x%x %d\n", gp, mytid, mygid);

	/* if I'm the first to join then start the others */
	if (mygid == 0) {
		/* start a bunch of children */
    	pvm_spawn(argv[0], argv+1, PvmTaskDefault, "", nproc-1, ctid);
		/* check them */
		for (i = 0; i < nproc-1; i++)
			fprintf(stderr, "0x%x\n", ctid[i]);
		}

	/* sync on a barrier */
	if (pvm_barrier(gp, nproc) < 0) {
		pvm_perror(argv[0]);
		pvm_lvgroup(argv[0]);
		pvm_exit();
		return -1;
		}

	fprintf(stderr, "%s %d: sync\n", argv[0], mygid);

	/* everyone broadcast their gids and tids */
	pvm_initsend(PvmDataDefault);
	pvm_pkint(&mygid, 1, 1);
	pvm_pkint(&mytid, 1, 1);
	pvm_bcast(gp, 63);

	/* recv all the gids and tids (except from myself) */
	ctid[mygid] = mytid;
	for (i = 0; i < nproc-1; i++) {
		pvm_recv(-1, 63);
		pvm_upkint(&indx, 1, 1);
		pvm_upkint(ctid+indx, 1, 1);
		}

	/* check to make sure the gids and tids are correct */
	for (i = 0; i < nproc; i++) {
		if (i != pvm_getinst(gp, ctid[i])) {
			fprintf(stderr, "gid %d doesn't match up!\n", i);
			}
		if (ctid[i] != pvm_gettid(gp, i)) {
			fprintf(stderr, "gid %d doesn't match up!\n", i);
			}
		}

	/* leave the group */
	if(pvm_lvgroup(gp) < 0)
			pvm_perror(gp);

	pvm_exit();
	return 0;
usage:
	fprintf(stderr, "usage: %s <nproc>\n", argv[0]);
	return -1;
}
