package spiglet.spiglet2kanga;

import java.util.HashMap;
import java.util.Vector;

public class SpgProc extends SpgSym {
	String name;
	int argCount;
	int stkCount;
	int maxCallArgCount;
	
	public Vector<SpgStmt> statements;
	public SpgSimpExpr retexp;
	
	public SpgProc(String s, int n) {
		name = s;
		argCount = n;
		statements = new Vector<SpgStmt>();
	}
	
	public void addStmt(SpgStmt s) {
		statements.addElement(s);
	}
	
	public void printProc() {
		System.err.println(name + "[" + argCount + "]");
		for (int i=0; i<statements.size(); i++) {
			System.err.println(statements.elementAt(i).toString());
			// statements.elementAt(i).printDefUse();
		}
		if (retexp!=null) {
			System.err.println("RETURN " + retexp.toString());
		}
		System.err.println();
	}
	
	public void setJmpTarget() {
		// convert all the jump labels to real statement
		
		// first scan all the labels
		HashMap<String, SpgStmt> s = new HashMap<String, SpgStmt>();
		for (int i=0; i<statements.size(); i++) {
			SpgStmt stmt = statements.elementAt(i);
			if (stmt.lb!=null) {
				s.put(stmt.lb, stmt);
			}
		}
		
		// then map the labels to statements
		for (int i=0; i<statements.size(); i++) {
			SpgStmt stmt = statements.elementAt(i);
			if (stmt.type==SpgStmt.StmtType.JUMP) {
				stmt.succ1 = s.get(stmt.jmptarget);
				stmt.succ2 = null;
			} else if (stmt.type==SpgStmt.StmtType.CJUMP) {
				if (i==statements.size()-1) {
					stmt.succ1 = null;
				} else {
					stmt.succ1 = statements.elementAt(i+1);
				}
				stmt.succ2 = s.get(stmt.jmptarget);
			} else {
				if (i==statements.size()-1) {
					stmt.succ1 = null;
				} else {
					stmt.succ1 = statements.elementAt(i+1);
				}
				stmt.succ2 = null;
			}
		}
	}
	
	public void getDefUse() {
		for (int i=0; i<statements.size(); i++) {
			statements.elementAt(i).getDefUse();
		}
	}
	
	public void getActiveVars() {
		boolean modified = true;
		
		// first, initialize the out[] of the last statement
		if (retexp!=null) {	
			statements.lastElement().out = retexp.getTmpUsed();
		}
		
		// then, loop in all statements for in[] and out[]
		while (modified) {
			modified = false;
			for (int i=statements.size()-1; i>=0; i--) {
				SpgStmt st = statements.elementAt(i);
				// first, reset the out[] set
				if (st.succ1!=null) {
					modified |= st.out.addAll(st.succ1.in);
				}
				if (st.succ2!=null) {
					modified |= st.out.addAll(st.succ2.in);
				}
				st.in.clear();
				st.in.addAll(st.out);
				if (st.def!=null) {
					st.in.removeAll(st.def);
				}
				if (st.use!=null) {
					st.in.addAll(st.use);
				}
			}
		}
	}
	
}