package com.dragome.compiler.ast;
import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.Map;
import org.apache.bcel.classfile.Code;
import org.apache.bcel.generic.Type;
import com.dragome.compiler.Project;
import com.dragome.compiler.generators.AbstractVisitor;
import com.dragome.compiler.parser.LineNumberCursor;
public class MethodDeclaration extends ASTNode
{
private String tempPrefix= "_";
private Block block;
private Map<String, VariableDeclaration> parameters= new LinkedHashMap<String, VariableDeclaration>();
private Map<String, VariableDeclaration> localVariables= new LinkedHashMap<String, VariableDeclaration>();
private int accessFlags;
private Code code;
private MethodBinding methodBinding;
private LineNumberCursor lineNumberCursor;
private Map<String, String> annotationsValues;
public Map<String, String> getAnnotationsValues()
{
return annotationsValues;
}
public void setAnnotationsValues(Map<String, String> annotationsValues)
{
this.annotationsValues= annotationsValues;
}
public MethodDeclaration(MethodBinding theMethodBinding, int theAccessFlags, Code theCode, Map<String, String> annotationsValues)
{
methodBinding= theMethodBinding;
accessFlags= theAccessFlags;
code= theCode;
this.annotationsValues= annotationsValues;
lineNumberCursor= new LineNumberCursor(code);
Project.getSingleton().getOrCreateProcedureUnit(methodBinding);
}
public int getAccess()
{
return accessFlags;
}
public void visit(AbstractVisitor visitor)
{
visitor.visit(this);
}
public boolean isInstanceConstructor()
{
return methodBinding.getName().equals("<init>");
}
public Block getBody()
{
return block;
}
public void setBody(Block theBlock)
{
block= theBlock;
theBlock.setParentNode(this);
}
public VariableBinding createVariableBinding(String name, Type type, boolean isWrite)
{
if (type == null)
throw new NullPointerException();
VariableDeclaration decl= getParameter(name);
if (decl == null)
{
decl= getLocalVariable(name);
}
if (decl == null)
{
decl= new VariableDeclaration(!isWrite);
decl.setName(name);
decl.setType(type);
addLocalVariable(decl);
}
VariableBinding binding= new VariableBinding(decl);
return binding;
}
private int vbCount= 0;
public VariableBinding createAnonymousVariableBinding(Type type, boolean isWrite)
{
String name= tempPrefix + (vbCount++);
VariableBinding vb= createVariableBinding(name, type, isWrite);
vb.setTemporary(true);
return vb;
}
public void addParameter(VariableDeclaration variableDecl)
{
parameters.put(variableDecl.getName(), variableDecl);
}
public Collection<VariableDeclaration> getParameters()
{
return parameters.values();
}
public VariableDeclaration getParameter(String name)
{
return parameters.get(name);
}
public void addLocalVariable(VariableDeclaration variableDecl)
{
localVariables.put(variableDecl.getName(), variableDecl);
}
public void removeLocalVariable(String name)
{
localVariables.remove(name);
}
public Collection<VariableDeclaration> getLocalVariables()
{
return localVariables.values();
}
public VariableDeclaration getLocalVariable(String name)
{
return localVariables.get(name);
}
public MethodBinding getMethodBinding()
{
return methodBinding;
}
public String toString()
{
return methodBinding.toString();
}
public Code getCode()
{
return code;
}
public LineNumberCursor getLineNumberCursor()
{
return lineNumberCursor;
}
}