prompto.literal.SetLiteral 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.literal;

import java.util.Collection;

import prompto.compiler.CompilerUtils;
import prompto.compiler.Flags;
import prompto.compiler.MethodInfo;
import prompto.compiler.ResultInfo;
import prompto.error.PromptoError;
import prompto.expression.IExpression;
import prompto.intrinsic.PromptoSet;
import prompto.runtime.Context;
import prompto.transpiler.Transpiler;
import prompto.type.IType;
import prompto.type.MissingType;
import prompto.type.SetType;
import prompto.utils.CodeWriter;
import prompto.utils.ExpressionList;
import prompto.value.IValue;
import prompto.value.SetValue;

public class SetLiteral extends ContainerLiteral {

	public SetLiteral() {
		super(()->"<>", new SetValue(MissingType.instance()), null, false);
	}

	public SetLiteral(ExpressionList expressions) {
		super(()->"<" + expressions.toString() + ">", new SetValue(MissingType.instance()), expressions, false);
	}

	@Override
	public Collection getItems() {
		return value.getItems();
	}
	
	@Override
	protected IType newType(IType itemType) {
		return new SetType(itemType);
	}
	
	@Override
	public SetValue interpret(Context context) throws PromptoError {
		if(expressions==null || expressions.isEmpty())
			return value;
		else {
			check(context); // force computation of itemType
			PromptoSet set = new PromptoSet();
			for(IExpression exp : expressions) {
				IValue item = exp.interpret(context);
				item = interpretPromotion(item);
				set.add(item);
			}
			return new SetValue(itemType, set);
		}
	}

	@Override
	public void toDialect(CodeWriter writer) {
		if(expressions!=null) {
			writer.append('<');
			expressions.toDialect(writer);
			writer.append('>');
		} else
			writer.append("< >");
	}

	@Override
	public ResultInfo compile(Context context, MethodInfo method, Flags flags) {
		ResultInfo info = CompilerUtils.compileNewInstance(method, PromptoSet.class);
		if(expressions!=null)
			compileItems(context, method, PromptoSet.class);
		return info;
	}

	@Override
	public void declare(Transpiler transpiler) {
	    transpiler.require("StrictSet");
	    this.expressions.declare(transpiler);
	}
	
	@Override
	public boolean transpile(Transpiler transpiler) {
	    transpiler.append("new StrictSet([");
	    this.expressions.transpile(transpiler);
	    transpiler.append("])");
	    return false;
	}
}