MultipleGeosModel.java

package org.geogebra.common.gui.dialog.options.model;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.TreeSet;

import org.geogebra.common.kernel.StringTemplate;
import org.geogebra.common.kernel.geos.GeoElement;
import org.geogebra.common.main.App;
import org.geogebra.common.main.Localization;

public abstract class MultipleGeosModel extends MultipleOptionsModel {
	private List<String> choices;

	public MultipleGeosModel(App app) {
		super(app);
		choices = new ArrayList<String>();
	}

	/**
	 * @param loc
	 *            localization
	 * @return list of points
	 */
	public List<GeoElement> getGeoChoiches(Localization loc) {
		TreeSet<GeoElement> points = app.getKernel().getPointSet();
		List<GeoElement> choices2 = new ArrayList<GeoElement>();
		choices2.add(null);
		Iterator<GeoElement> it = points.iterator();
		int count = 0;
		while (it.hasNext() || ++count > MAX_CHOICES) {
			GeoElement p = it.next();
			choices2.add(p);
		}
		return choices2;
	}

	@Override
	public List<String> getChoiches(Localization loc) {
		TreeSet<GeoElement> points = app.getKernel().getPointSet();
		choices.clear();
		choices.add("");
		Iterator<GeoElement> it = points.iterator();
		int count = 0;
		while (it.hasNext() || ++count > MAX_CHOICES) {
			GeoElement p = it.next();
			choices.add(p.getLabel(StringTemplate.editTemplate));
		}
		return choices;
	}

}