ASTLet.java example

Explorer
JAME-master
/*
 * JAME 6.2.1
 * http://jame.sourceforge.net
 *
 * Copyright 2001, 2016 Andrea Medeghini
 *
 * This file is part of JAME.
 *
 * JAME is an application for creating fractals and other graphics artifacts.
 *
 * JAME is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * JAME is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with JAME.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
package net.sf.jame.contextfree.parser;

import org.antlr.v4.runtime.Token;

class ASTLet extends ASTUserFunction {
	private ASTRepContainer definitions;
	
	public ASTLet(ASTRepContainer args, ASTDefine func, Token location) {
		super(-1, null, func, location);
		this.definitions = args;
		isLet = true;
	}

	@Override
	public ASTExpression simplify() {
		getDefinition().compile(ECompilePhase.Simplify);
		if (isConstant()) {
			StringBuilder e = new StringBuilder();
			entropy(e);
			ASTParameter p = new ASTParameter(-1, getDefinition(), location);
			p.setDefinition(getDefinition());
			ASTExpression ret = p.constCopy(e.toString());
			if (ret != null) {
				return ret;
			}
		} else if (getArguments() == null) {
			ASTExpression ret = getDefinition().getExp();
			return ret;
		}
		return super.simplify();
	}

	@Override
	public ASTExpression compile(ECompilePhase ph) {
		switch (ph) {
			case TypeCheck: 
				{
					definitions.compile(ph, null, getDefinition());
					ASTExpression args = null;
					for (ASTReplacement rep : definitions.getBody()) {
						if (rep instanceof ASTDefine) {
							ASTDefine def = (ASTDefine)rep;
							if (def.getDefineType() == EDefineType.StackDefine) {
								getDefinition().incStackCount(def.getTupleSize());
								args = ASTExpression.append(args, def.getExp());
							}
						}
					}
					definitions.getParameters().remove(definitions.getParameters().size() - 1);
					for (ASTParameter param : definitions.getParameters()) {
						if (param.getDefinition() != null) {
							getDefinition().getParameters().add(param);
						}
					}
					definitions = null;//TODO controllare
					setArguments(args);
					isConstant = getArguments() == null && getDefinition().getExp().isConstant();
					isNatural = getDefinition().isNatural();
					locality = getDefinition().getExp() != null ? getDefinition().getExp().getLocality() : getDefinition().getChildChange().getLocality();
					type = getDefinition().getExpType();
				}
				break;
	
			case Simplify: 
				break;
	
			default:
				break;
		}
		return null;
	}
}