package org.matheclipse.core.builtin.function;

import static org.matheclipse.core.expression.F.List;

import org.matheclipse.core.eval.exception.Validate;
import org.matheclipse.core.eval.interfaces.AbstractCoreFunctionEvaluator;
import org.matheclipse.core.expression.F;
import org.matheclipse.core.generic.Functors;
import org.matheclipse.core.generic.Predicates;
import org.matheclipse.core.interfaces.IAST;
import org.matheclipse.core.interfaces.IExpr;
import org.matheclipse.core.interfaces.ISymbol;

import com.google.common.base.Function;
import com.google.common.base.Predicate;

public class NestWhileList extends AbstractCoreFunctionEvaluator {

	public NestWhileList() {
	}

	@Override
	public IExpr evaluate(final IAST ast) {
		Validate.checkSize(ast, 4);

		return nestList(ast.arg2(), F.eval(ast.arg3()), Functors.append(F.ast(ast.arg1())), List());
	}

	public static IAST nestList(final IExpr expr, final IExpr test, final Function fn, final IAST resultList) {
		IExpr temp = expr;
		Predicate predicate = Predicates.isTrue(test);

		while (predicate.apply(temp)) {
			resultList.add(temp);
			temp = F.eval(fn.apply(temp));
		}
		resultList.add(temp);
		return resultList;
	}

	@Override
	public void setUp(final ISymbol symbol) {
		symbol.setAttributes(ISymbol.HOLDALL);
	}
}