JavaTXCompilerInJavaTX/src/de/dhbwstuttgart/syntaxtree/statement/MethodCall.java

package de.dhbwstuttgart.syntaxtree.statement;

import de.dhbwstuttgart.exceptions.TypeinferenceException;
import de.dhbwstuttgart.typeinference.assumptions.TypeInferenceBlockInformation;
import de.dhbwstuttgart.typeinference.constraints.Constraint;
import de.dhbwstuttgart.typeinference.constraints.ConstraintSet;
import de.dhbwstuttgart.typeinference.constraints.Pair;
import de.dhbwstuttgart.typeinference.assumptions.MethodAssumption;
import org.antlr.v4.runtime.Token;

import de.dhbwstuttgart.syntaxtree.type.TypePlaceholder;

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


public class MethodCall extends Statement
{
    private final String name;
    private Receiver receiver;
    private ArgumentList arglist;

    public MethodCall(Receiver receiver, String methodName, ArgumentList argumentList, Token offset){
		super(TypePlaceholder.fresh(offset),offset);
		this.arglist = argumentList;
        this.name = methodName;
        this.receiver = receiver;
	}

    @Override
    public ConstraintSet getConstraints(TypeInferenceBlockInformation info) {
        ConstraintSet ret = receiver.getConstraints(info);
        ret.addAll(receiver.getConstraints(info));
        for(int i = 0;i<arglist.getArguments().size();i++){
            ret.addAll(arglist.getArguments().get(i).getConstraints(info));
        }
        //Overloading:
        Set<Constraint> methodConstraints = new HashSet<>();
        for(MethodAssumption m : info.getMethods(name, arglist)){
            Constraint methodConstraint = new Constraint();
            methodConstraint.add(new Pair(receiver.getType(), m.getReceiverType()));
            methodConstraint.add(new Pair(m.getReturnType(), this.getType()));
            for(int i = 0;i<arglist.getArguments().size();i++){
                methodConstraint.add(new Pair(m.getArgTypes().get(i),
                        arglist.getArguments().get(i).getType()));
            }
            methodConstraints.add(methodConstraint);
        }
        if(methodConstraints.size()<1){
            throw new TypeinferenceException("Methode "+name+" ist nicht vorhanden!",getOffset());
        }
        ret.addOderConstraint(methodConstraints);
        return ret;
    }
}