core/org.eclipse.statet.rj.data/src/org/eclipse/statet/rj/data/impl/AbstractNumericStore.java

/*=============================================================================#
 # Copyright (c) 2009, 2021 Stephan Wahlbrink and others.
 # 
 # This program and the accompanying materials are made available under the
 # terms of the Eclipse Public License 2.0 which is available at
 # https://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
 # which is available at https://www.apache.org/licenses/LICENSE-2.0.
 # 
 # SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
 # 
 # Contributors:
 #     Stephan Wahlbrink <sw@wahlbrink.eu> - initial API and implementation
 #=============================================================================*/

package org.eclipse.statet.rj.data.impl;

import org.eclipse.statet.jcommons.lang.NonNullByDefault;
import org.eclipse.statet.jcommons.lang.Nullable;

import org.eclipse.statet.rj.data.RNumericStore;
import org.eclipse.statet.rj.data.RObject;
import org.eclipse.statet.rj.data.RStore;


@NonNullByDefault
public abstract class AbstractNumericStore extends AbstractRStore<Double>
		implements RNumericStore {
	
	
	protected static final String toChar(final double num) {
		if (num == Double.POSITIVE_INFINITY) {
			return "Inf"; //$NON-NLS-1$
		}
		else if (num == Double.NEGATIVE_INFINITY) {
			return "-Inf"; //$NON-NLS-1$
		}
		else {
			return Double.toString(num);
		}
	}
	
	
	@Override
	public final byte getStoreType() {
		return RStore.NUMERIC;
	}
	
	@Override
	public final String getBaseVectorRClassName() {
		return RObject.CLASSNAME_NUMERIC;
	}
	
	
	@Override
	public boolean getLogi(final int idx) {
		return getNum(idx) != 0.0;
	}
	
	@Override
	public final boolean getLogi(final long idx) {
		return getNum(idx) != 0.0;
	}
	
	@Override
	public final void setLogi(final int idx, final boolean logi) {
		setNum(idx, (logi) ? 1.0 : 0.0);
	}
	
	@Override
	public final void setLogi(final long idx, final boolean logi) {
		setNum(idx, (logi) ? 1.0 : 0.0);
	}
	
	@Override
	public final int getInt(final int idx) {
		return (int)getNum(idx);
	}
	
	@Override
	public final int getInt(final long idx) {
		return (int)getNum(idx);
	}
	
	@Override
	public final void setInt(final int idx, final int integer) {
		setNum(idx, integer);
	}
	
	@Override
	public final void setInt(final long idx, final int integer) {
		setNum(idx, integer);
	}
	
	@Override
	public final double getCplxRe(final int idx) {
		return getNum(idx);
	}
	
	@Override
	public final double getCplxRe(final long idx) {
		return getNum(idx);
	}
	
	@Override
	public final double getCplxIm(final int idx) {
		return 0.0;
	}
	
	@Override
	public final double getCplxIm(final long idx) {
		return 0.0;
	}
	
	@Override
	public final String getChar(final int idx) {
		return toChar(getNum(idx));
	}
	
	@Override
	public String getChar(final long idx) {
		return toChar(getNum(idx));
	}
	
	
	@Override
	public abstract @Nullable Double [] toArray();
	
	
	@Override
	public boolean allEqual(final RStore<?> other) {
		final long length= getLength();
		if (NUMERIC != other.getStoreType() || length != other.getLength()) {
			return false;
		}
		if (length < 0) {
			return true;
		}
		else if (length <= Integer.MAX_VALUE) {
			final int ilength= (int)length;
			for (int idx= 0; idx < ilength; idx++) {
				if (!(isMissing(idx) ?
						(isNA(idx) ? other.isNA(idx) : other.isMissing(idx)) :
						(Math.abs(getNum(idx) - other.getNum(idx)) < 2.220446e-16 )) ) {
					return false;
				}
			}
		}
		else {
			for (long idx= 0; idx < length; idx++) {
				if (!(isMissing(idx) ?
						(isNA(idx) ? other.isNA(idx) : other.isMissing(idx)) :
						(Math.abs(getNum(idx) - other.getNum(idx)) < 2.220446e-16 )) ) {
					return false;
				}
			}
		}
		return true;
	}
	
}