CoordinateDatasetIterator.java example

Explorer
scisoft-core-master
/*-
 * Copyright (c) 2013 Diamond Light Source Ltd.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 */

package uk.ac.diamond.scisoft.analysis.fitting.functions;

import org.eclipse.january.dataset.CompoundDataset;
import org.eclipse.january.dataset.DTypeUtils;
import org.eclipse.january.dataset.Dataset;
import org.eclipse.january.dataset.DatasetUtils;
import org.eclipse.january.dataset.IDataset;
import org.eclipse.january.dataset.PositionIterator;

/**
 * An iterator over a dataset whose items are coordinates
 */
public class CoordinateDatasetIterator extends CoordinatesIterator {
	CompoundDataset cvalue;

	/**
	 * A single, possibly compound, dataset
	 * @param value
	 */
	public CoordinateDatasetIterator(IDataset value) {
		if (!(value instanceof CompoundDataset)) {
			int dtype = DTypeUtils.getBestDType(Dataset.ARRAYINT8,
					DTypeUtils.getDTypeFromClass(value.getElementClass()));
			cvalue = (CompoundDataset) DatasetUtils.cast(value, dtype);
		} else {
			cvalue = (CompoundDataset) value;
		}
		shape = cvalue.getShape();

		coords = new double[cvalue.getElementsPerItem()];
		values = new IDataset[] { cvalue };
		it = new PositionIterator(shape);
		pos = it.getPos();
	}

	@Override
	public boolean hasNext() {
		if (!it.hasNext())
			return false;

		cvalue.getDoubleArray(coords, pos);
		return true;
	}
}