prompto.value.IteratorValue 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.value;

import java.util.Iterator;

import prompto.intrinsic.PromptoSet;
import prompto.type.IType;
import prompto.type.IteratorType;

/* exposes a native IValue iterator as an IValue */
public class IteratorValue extends BaseValue implements Iterator {

	Iterator source;
	
	public IteratorValue(IType itemType, Iterator source) {
		super(new IteratorType(itemType));
		this.source = source;
	}

	@Override
	public boolean hasNext() {
		return source.hasNext();
	}
	
	@Override
	public IValue next() {
		return source.next();
	}
	
	public IValue toListValue() {
		ListValue values = new ListValue(((IteratorType)this.getType()).getItemType());
		while(source.hasNext())
			values.addItem(source.next());
		return values;
	}

	public IValue toSetValue() {
		PromptoSet items = new PromptoSet<>();
		while(source.hasNext())
			items.add(source.next());
		return new SetValue(((IteratorType)this.getType()).getItemType(), items);
	}

}