package soot.jimple.toolkits.infoflow;

import soot.*;

import java.util.*;
import soot.toolkits.graph.*;
import soot.jimple.*;

// SmartMethodLocalObjectsAnalysis written by Richard L. Halpert, 2007-02-23
// Uses a SmartMethodInfoFlowAnalysis to determine if a Local or FieldRef is
// LOCAL or SHARED in the given method.

public class SmartMethodLocalObjectsAnalysis
{
	public static int counter = 0;
	static boolean printMessages;
	
	SootMethod method;
	InfoFlowAnalysis dfa;
	SmartMethodInfoFlowAnalysis smdfa;

	public SmartMethodLocalObjectsAnalysis(SootMethod method, InfoFlowAnalysis dfa)
	{
		this.method = method;
		this.dfa = dfa;
		this.smdfa = dfa.getMethodInfoFlowAnalysis(method);
		
		printMessages = dfa.printDebug();
		counter++;
	}
	
	public SmartMethodLocalObjectsAnalysis(UnitGraph g, InfoFlowAnalysis dfa)
	{
		this(g.getBody().getMethod(), dfa);
	}
	
	public Value getThisLocal()
	{
		return smdfa.getThisLocal();
	}
	
	// 
	public boolean isObjectLocal(Value local, CallLocalityContext context) // to this analysis of this method (which depends on context)
	{
		EquivalentValue localEqVal;
		if(local instanceof InstanceFieldRef)
			localEqVal = InfoFlowAnalysis.getNodeForFieldRef(method, ((FieldRef) local).getField());
		else
			localEqVal = new CachedEquivalentValue(local);
			
		List<EquivalentValue> sources = smdfa.sourcesOf(localEqVal);
		Iterator<EquivalentValue> sourcesIt = sources.iterator();
		while(sourcesIt.hasNext())
		{
			EquivalentValue source = sourcesIt.next();
			if(source.getValue() instanceof Ref)
			{
				if(!context.isFieldLocal(source))
				{
					if(printMessages)
						G.v().out.println("      Requested value " + local + " is SHARED in " + method + " ");
					return false;
				}
			} else if(source.getValue() instanceof Constant) {
				if(printMessages)
					G.v().out.println("      Requested value " + local + " is SHARED in " + method + " ");
				return false;
			}
		}
		if(printMessages)
			G.v().out.println("      Requested value " + local + " is LOCAL in " + method + " ");
		return true;
	}
	
	public static boolean isObjectLocal(InfoFlowAnalysis dfa, SootMethod method, CallLocalityContext context, Value local)
	{
		SmartMethodInfoFlowAnalysis smdfa = dfa.getMethodInfoFlowAnalysis(method);

		EquivalentValue localEqVal;
		if(local instanceof InstanceFieldRef)
			localEqVal = InfoFlowAnalysis.getNodeForFieldRef(method, ((FieldRef) local).getField());
		else
			localEqVal = new CachedEquivalentValue(local);
			
		List<EquivalentValue> sources = smdfa.sourcesOf(localEqVal);
		Iterator<EquivalentValue> sourcesIt = sources.iterator();
		while(sourcesIt.hasNext())
		{
			EquivalentValue source = sourcesIt.next();
			if(source.getValue() instanceof Ref)
			{
				if(!context.isFieldLocal(source))
				{
					if(printMessages)
						G.v().out.println("      Requested value " + local + " is LOCAL in " + method + " ");
					return false;
				}
			}
		}
		if(printMessages)
			G.v().out.println("      Requested value " + local + " is SHARED in " + method + " ");
		return true;
	}
}