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

/*=============================================================================#
 # Copyright (c) 2009, 2020 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 java.io.IOException;
import java.util.Arrays;

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

import org.eclipse.statet.rj.data.RCharacterStore;
import org.eclipse.statet.rj.data.RJIO;


/**
 * {@link RCharacterStore} supporting long length.
 */
@NonNullByDefault
public class RCharacterFix64Store extends AbstractCharacterStore
		implements ExternalizableRStore {
	
	
	public static final int SEGMENT_LENGTH= DEFAULT_LONG_DATA_SEGMENT_LENGTH;
	
	
	private final long length;
	
	protected final @Nullable String [] @NonNull[] charValues;
	
	
	public RCharacterFix64Store(final long length) {
		this.length= length;
		this.charValues= new2dStringArray(length, SEGMENT_LENGTH);
		for (int i= 0; i < this.charValues.length; i++) {
			Arrays.fill(this.charValues[i], "");
		}
	}
	
	public RCharacterFix64Store(final @Nullable String [] @NonNull[] values) {
		this.length= check2dArrayLength(values, SEGMENT_LENGTH);
		this.charValues= values;
	}
	
	
	public RCharacterFix64Store(final RJIO io, final long length) throws IOException {
		this.length= length;
		this.charValues= new2dStringArray(length, SEGMENT_LENGTH);
		for (int i= 0; i < this.charValues.length; i++) {
			io.readStringData(this.charValues[i], this.charValues[i].length);
		}
	}
	
	@Override
	public void writeExternal(final RJIO io) throws IOException {
		for (int i= 0; i < this.charValues.length; i++) {
			io.writeStringData(this.charValues[i], this.charValues[i].length);
		}
	}
	
	
	@Override
	protected final boolean isStructOnly() {
		return false;
	}
	
	
	@Override
	public final long getLength() {
		return this.length;
	}
	
	@Override
	public boolean isNA(final int idx) {
		return (this.charValues[idx / SEGMENT_LENGTH][idx % SEGMENT_LENGTH] == null);
	}
	
	@Override
	public boolean isNA(final long idx) {
		return (this.charValues[(int)(idx / SEGMENT_LENGTH)][(int)(idx % SEGMENT_LENGTH)] == null);
	}
	
	@Override
	public void setNA(final int idx) {
		this.charValues[idx / SEGMENT_LENGTH][idx % SEGMENT_LENGTH]= null;
	}
	
	@Override
	public void setNA(final long idx) {
		this.charValues[(int)(idx / SEGMENT_LENGTH)][(int)(idx % SEGMENT_LENGTH)]= null;
	}
	
	@Override
	public boolean isMissing(final int idx) {
		return (this.charValues[idx / SEGMENT_LENGTH][idx % SEGMENT_LENGTH] == null);
	}
	
	@Override
	public boolean isMissing(final long idx) {
		return (this.charValues[(int)(idx / SEGMENT_LENGTH)][(int)(idx % SEGMENT_LENGTH)] == null);
	}
	
	@Override
	@SuppressWarnings("null")
	public String getChar(final int idx) {
		return this.charValues[idx / SEGMENT_LENGTH][idx % SEGMENT_LENGTH];
	}
	
	@Override
	@SuppressWarnings("null")
	public String getChar(final long idx) {
		return this.charValues[(int)(idx / SEGMENT_LENGTH)][(int)(idx % SEGMENT_LENGTH)];
	}
	
	@Override
	public void setChar(final int idx, final String value) {
//		assert (value != null);
		this.charValues[idx / SEGMENT_LENGTH][idx % SEGMENT_LENGTH]= value;
	}
	
	@Override
	public void setChar(final long idx, final String value) {
//		assert (value != null);
		this.charValues[(int)(idx / SEGMENT_LENGTH)][(int)(idx % SEGMENT_LENGTH)]= value;
	}
	
	
	@Override
	public @Nullable String get(final int idx) {
		if (idx < 0 || idx >= this.length) {
			throw new IndexOutOfBoundsException(Long.toString(idx));
		}
		return this.charValues[idx / SEGMENT_LENGTH][idx % SEGMENT_LENGTH];
	}
	
	@Override
	public @Nullable String get(final long idx) {
		if (idx < 0 || idx >= this.length) {
			throw new IndexOutOfBoundsException(Long.toString(idx));
		}
		return this.charValues[(int)(idx / SEGMENT_LENGTH)][(int)(idx % SEGMENT_LENGTH)];
	}
	
	@Override
	public @Nullable String [] toArray() {
		final int l= checkToArrayLength();
		final var array= new @Nullable String[l];
		for (int i= 0, destIdx= 0; i < this.charValues.length; i++) {
			final String[] chars= this.charValues[i];
			System.arraycopy(chars, 0, array, destIdx, chars.length);
			destIdx+= chars.length;
		}
		return array;
	}
	
	
	@Override
	public long indexOfNA(long fromIdx) {
		if (fromIdx < 0) {
			fromIdx= 0;
		}
		int i= (int)(fromIdx / SEGMENT_LENGTH);
		int j= (int)(fromIdx % SEGMENT_LENGTH);
		while (i < this.charValues.length) {
			final String[] chars= this.charValues[i];
			while (j < chars.length) {
				if (chars[j] == null) {
					return (i * (long)SEGMENT_LENGTH) + j;
				}
			}
			i++;
			j= 0;
		}
		return -1;
	}
	
	@Override
	public long indexOf(final @Nullable String character, long fromIdx) {
		if (character == null) {
			return -1;
		}
		if (fromIdx < 0) {
			fromIdx= 0;
		}
		int i= (int)(fromIdx / SEGMENT_LENGTH);
		int j= (int)(fromIdx % SEGMENT_LENGTH);
		while (i < this.charValues.length) {
			final String[] chars= this.charValues[i];
			while (j < chars.length) {
				if (chars[j] != null && chars[j].equals(character)) {
					return (i * (long)SEGMENT_LENGTH) + j;
				}
			}
			i++;
			j= 0;
		}
		return -1;
	}
	
}