prompto.declaration.DispatchMethodDeclaration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Core Show documentation
The newest version!
package prompto.declaration;

import java.util.HashSet;
import java.util.List;
import java.util.Set;

import prompto.compiler.ClassFile;
import prompto.param.AttributeParameter;
import prompto.param.CategoryParameter;
import prompto.param.IParameter;
import prompto.param.ParameterList;
import prompto.param.UnresolvedParameter;
import prompto.runtime.Context;
import prompto.statement.MethodCall;
import prompto.transpiler.Transpiler;
import prompto.type.IType;
import prompto.utils.CodeWriter;


public class DispatchMethodDeclaration extends BaseMethodDeclaration {

	Context context;
	MethodCall call;
	IMethodDeclaration declaration;
	List declarations;
	
	public DispatchMethodDeclaration(Context context, MethodCall call, IMethodDeclaration declaration, List declarations) {
		super(declaration.getId(), declaration.getParameters(), declaration.getReturnType());
		this.context = context;
		this.call = call;
		this.declaration = declaration;
		this.declarations = declarations;
	}

	@Override
	public String getTranspiledName(Context context) {
		return "$dispatch$" + this.declaration.getTranspiledName(context);
	}
	
	
	@Override
	public void declare(Transpiler transpiler) {
		// nothing to do
	}
	
	@Override
	public boolean transpile(Transpiler transpiler) {
	    this.registerParameters(transpiler.getContext());
	    this.transpileProlog(transpiler);
	    this.transpileDispatch(transpiler);
	    this.transpileEpilog(transpiler);
		return true;
	}

	private void transpileDispatch(Transpiler transpiler) {
		Set common = this.collectCommonArguments();
	    for(int i=0; i0)
	            transpiler.append("else ");
	        if(i {
	        transpiler.append(arg.getName());
	        transpiler.append(", ");
	    });
	    transpiler.trimLast(2);
	    transpiler.append(")");
	}

	private void transpileTest(Transpiler transpiler, Set common, IMethodDeclaration declaration) {
	    for(int i = 0, count = 0;i0)
	            transpiler.append(" && ");
	        count++;
	        if(incoming instanceof UnresolvedParameter)
	            incoming = ((UnresolvedParameter)incoming).getResolved();
	        IParameter outgoing = incoming==null ? declaration.getParameters().get(0) : findCorrespondingArg(transpiler.getContext(), declaration.getParameters(), common, incoming);
	        if(outgoing instanceof UnresolvedParameter)
	        	outgoing = ((UnresolvedParameter)incoming).getResolved();
	        if(incoming==null)
	            incoming = this.declaration.getParameters().get(0);
	        if(incoming instanceof UnresolvedParameter)
	            incoming = ((UnresolvedParameter)incoming).getResolved();
	        if(incoming instanceof CategoryParameter && outgoing instanceof CategoryParameter) {
	            transpiler.append(incoming.getName()).append(".instanceOf(").append(((CategoryParameter)outgoing).getType().getTypeName()).append(")");
	        } else if(incoming instanceof CategoryParameter && outgoing instanceof AttributeParameter) {
	            transpiler.append(incoming.getName()).append(".hasOwnProperty('").append(outgoing.getName()).append("')");
	        } else
	            throw new Error("Unsupported: " + incoming.getClass().getSimpleName() + " and " + outgoing.getClass().getSimpleName());
	    }
	}

	private IParameter findCorrespondingArg(Context context2, ParameterList arguments, Set common, IParameter incoming) {
	    for(int i=0;i collectCommonArguments() {
		Set common = null;
	    for(int i=0; i(declaration.getParameters());
	        else {
	        	Set current = new HashSet<>(declaration.getParameters());
	            common.retainAll(current);
	            if(common.isEmpty())
	                break;
	        }
	    }
	    return common;
	}

	@Override
	public boolean isAbstract() {
		return false;
	}

	@Override
	public boolean isTemplate() {
		return false;
	}

	@Override
	public void compile(Context context, boolean isStart, ClassFile classFile) {
		throw new UnsupportedOperationException("Should never get there!");
	}

	@Override
	public String compileTemplate(Context context, boolean isStart, ClassFile classFile) {
		throw new UnsupportedOperationException("Should never get there!");
	}

	@Override
	public IType check(Context context, boolean isStart) {
		throw new UnsupportedOperationException("Should never get there!");
	}
	
	@Override
	public IType checkChild(Context context) {
		throw new UnsupportedOperationException("Should never get there!");
	}

	@Override
	public void declarationToDialect(CodeWriter writer) {
		throw new UnsupportedOperationException("Should never get there!");
	}


}