package mycompiler.mystatement;

import java.util.Hashtable;
import java.util.Vector;

import typinferenz.JavaCodeResult;
import typinferenz.SingleConstraint;
import typinferenz.ConstraintsSet;
import typinferenz.FreshTypeVariable;
import typinferenz.FunN;
import typinferenz.ResultSet;
import typinferenz.Typeable;
import typinferenz.assumptions.ParameterAssumption;
import typinferenz.assumptions.TypeAssumptions;
import typinferenz.exceptions.TypinferenzException;
import mycompiler.SyntaxTreeNode;
import mycompiler.mybytecode.ClassFile;
import mycompiler.mybytecode.CodeAttribute;
import mycompiler.myclass.Class;
import mycompiler.myclass.ClassHelper;
import mycompiler.myclass.FormalParameter;
import mycompiler.myclass.Method;
import mycompiler.myclass.ParameterList;
import mycompiler.myexception.CTypeReconstructionException;
import mycompiler.myexception.JVMCodeException;
import mycompiler.myexception.SCStatementException;
import mycompiler.mytype.DoubleType;
import mycompiler.mytype.GenericTypeVar;
import mycompiler.mytype.Type;
import mycompiler.mytype.TypePlaceholder;
import mycompiler.mytypereconstruction.CSupportData;
import mycompiler.mytypereconstruction.CTriple;
import mycompiler.mytypereconstruction.set.CSubstitutionSet;
import mycompiler.mytypereconstruction.set.CTripleSet;
import mycompiler.mytypereconstruction.set.CTypeAssumptionSet;
import mycompiler.mytypereconstruction.typeassumption.CParaTypeAssumption;
import mycompiler.mytypereconstruction.typeassumption.CTypeAssumption;

/**
 * @author A10023 - Andreas Stadelmeier
 * Momentan erweitert LambdaExpression noch Expr und erbt dadurch auch von ExprStatement ist also auch ein Statement
 * 
 * LambdaExpression Aufbau:
 * ( ParameterList ) -> { method_body };
 */
public class LambdaExpression extends Expr{ 
	
	private Block method_body;
	private ParameterList params;
	
	public LambdaExpression(int offset, int variableLength) {
		super(offset, variableLength);
		setParameterList(new ParameterList());//default is empty parameterlist
	}
	
	public void setBody(Block block){
		method_body = block;
	}
	
	public void setExpr(Expr expression){
		Block bl = new Block();
		Return returnStmt = new Return(0, 0);
		returnStmt.retexpr = expression;
		bl.set_Statement(returnStmt);
		this.setBody(bl);
	}
	
	public void setParameterList(ParameterList params){
		this.params = params;
	}


	@Override
	public void codegen(ClassFile classfile, CodeAttribute code, Vector paralist)
			throws JVMCodeException {
		// TODO Auto-generated method stub
		
	}


	@Override
	public void wandleRefTypeAttributes2GenericAttributes(
			Vector<Type> paralist,
			Vector<GenericTypeVar> genericMethodParameters) {
		Block block = this.method_body;
		// Zuerst Returntype untersuchen
        Type returnType=getType();
        GenericTypeVar pendantReturnType=ClassHelper.findGenericType(returnType, paralist,genericMethodParameters);
        if(pendantReturnType!=null){    //Wenn generisch, dann modifizieren
            setType(pendantReturnType);
        }                                       
        
        // Dann parameterlist untersuchen
        for(FormalParameter fp : params){
            Type fpType=fp.getType();
            // Nur wenn es sich um ein RefType-Field handelt
            GenericTypeVar pendantPara=ClassHelper.findGenericType(fpType,paralist,genericMethodParameters);
            if(pendantPara!=null){  //Wenn generisch, dann modifizieren
                fp.setType(pendantPara);
            }                       
        }
        
        // Zuletzt alle Lokalen Variablendeklarationen durchgehen
        if(block!=null){
            block.wandleRefTypeAttributes2GenericAttributes(paralist,genericMethodParameters);
        }
	}

	@Override
	public boolean addOffsetsToStatement(CTypeAssumption localAssumption,
			String NameVariable, boolean isMemberVariable) {
		// TODO Auto-generated method stub
		return false;
	}

	@Override
	public String get_Name() {
		// TODO Auto-generated method stub
		return null;
	}


	@Override
	public void addOffsetsToExpression(CTypeAssumption localAssumption,
			String NameVariable, boolean isMemberVariable) {
		// TODO Auto-generated method stub
		
	}

	
	/**
	  * Spezifikation:
	  * 
	  * TYPEExpr( Ass, Lambda( (x1 , . . . , xN ), expr|stmt ) ) =
	  * let
	  * AssArgs = { xi : ai | ai fresh type variables }
	  * (exprt : rty, ConS) = TYPEExpr( Ass ∪ AssArgs, expr )
	  * | (stmtt : rty, ConS) = TYPEStmt( Ass u AssArgs, stmt )
	  * in
	  * (Lambda( (x1 : a1 , . . . , xN : aN ), exprt : rty|stmtt : rty ) : a,
	  * ConS ∪ { (FunN<rty, a1 , . . . , aN > a) }),
	  * where a is a fresh type variable 
	 */
	@Override
	public ConstraintsSet TYPEExpr(TypeAssumptions assumptions) {
		ConstraintsSet ret = new ConstraintsSet();
		//Die Assumptions für die Parameter der LambdaExpression
		TypeAssumptions ArgumentAssumptions = new TypeAssumptions(this.getParentClass().getName());
		Vector<Type> paramTypes =  new Vector<Type>();
		
		for(FormalParameter param : params.formalparameter){
			if(param.getType()==null)param.setType(TypePlaceholder.fresh(this));
			int offset = 0;
			//Jeder Parameter der LambdaExpression wird als CParaTypeAssumption der Assumption liste hinzugefügt:
			ArgumentAssumptions.addAssumption(new ParameterAssumption(param));
			paramTypes.add(param.getType());
		}
		this.setType(TypePlaceholder.fresh(this));
		//ArgumentAssumptions + assumptions ergeben die Assumptions für die Statements innerhalb des Lambda-Bodys:
		ret.add(method_body.TYPEStmt(ArgumentAssumptions.add(assumptions))); //Es gibt die LambdaExpression nur mit einem Block als Method Body, nicht mit einer einzelnen Expression
		ret.add(new SingleConstraint(new FunN(method_body.getType(), paramTypes),this.getType()));
		return ret;
	}
	
	@Override
	public ConstraintsSet TYPEStmt(TypeAssumptions ass){
		throw new TypinferenzException("Eine LambdaExpression darf nicht als Statement verwendet werden.");
	}
	
	@Override
	public String getTypeInformation(){
		return this.getType().toString()+" :: ("+this.params.getTypeInformation()+ ") -> " +this.method_body.getTypeInformation();
	}
	
	@Override
	public String toString(){
		//return "LambdaExpression, Parameter: "+this.params+ ", Body: " +this.method_body;
		return this.getType() +  " (("+this.params+ ") -> "+this.method_body + ")";
	}
	
	@Override
	public JavaCodeResult printJavaCode(ResultSet resultSet){
		JavaCodeResult ret = new JavaCodeResult();
		ret.attach( "(").attach(this.params.printJavaCode(resultSet)).attach(")");
		ret.attach( " -> ").attach( this.method_body.printJavaCode(resultSet));
		return ret;
	}

	@Override
	public Vector<SyntaxTreeNode> getChildren() {
		Vector<SyntaxTreeNode> ret = new Vector<SyntaxTreeNode>();
		ret.add(this.method_body);
		return ret;
	}

}