#
# exclude jobs prior to a designated job from a DAX process graph
#
# 
#

minstage=$1
maxstage=$2

a1=`mktemp /tmp/awk.XXXXXX`
dax=`mktemp /tmp/dax.XXXXXX`
cat >$dax

maxlevel=`maxlevel <$dax`

cat <<EOF >$a1

BEGIN {
	maxlevel = $maxlevel;
}

/<adag / {
	print \$0;
}

/<job / {
	id = \$2;
	sub( /^.*ID/,"ID",id);
	sub( /"/,"",id);
}

/<job /,/job>/ {
	job = job \$0 "\n";
}

/job>/ {
	jobtext[id] = job;
	job = "";
}

/<child ref=/ {
	child = \$2;
	sub( /^.*ID/,"ID",child);
	sub( /".*/,"",child);
	parents = "";
}

/<parent ref=/ {
	parent = \$2;
	sub( /^.*ID/,"ID",parent);
	sub( /".*/,"",parent);
	parents = parents " " parent;
}

/<\\/child>/ {
        sub("^  *","",parents);
	parlist[child] = parents;
}

function selstage(minstage,maxstage) {

	maxlev = maxlevel - minstage + 1;
	minlev = maxlevel - maxstage + 1;

	for(j in jobtext) {
		found=match(jobtext[j],"level=\"[0-9][0-9]*");
		if(found) {
			level=substr(jobtext[j],RSTART,RLENGTH);
			sub("level=\"","",level);
			if ( (level < minlev) || (level > maxlev) ) {
				delete jobtext[j];
			}
		}
	}
}

END {
	selstage($minstage,$maxstage);

	print("<!-- part 2: definition of remaining jobs -->");

	for (j in jobtext) {
		print(jobtext[j]);
	}
	print "<!-- part 3: list of control-flow dependencies (may be empty) -->";
	for ( c in parlist ) {
		printf("  <child ref=\"%s\">\n", c);
		split(parlist[c],plist," ");
		for(p in plist) {
			printf("    <parent ref=\"%s\"/>\n", plist[p]);
		}
		print("  </child>\n");
	}
	print "</adag>";
}

EOF

awk -f $a1 $dax

rm $a1