ByteRawIndexer.java
/*
* Copyright (C) 2016-2019 Samuel Audet
*
* Licensed either under the Apache License, Version 2.0, or (at your option)
* under the terms of the GNU General Public License as published by
* the Free Software Foundation (subject to the "Classpath" exception),
* either version 2, or any later version (collectively, the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.gnu.org/licenses/
* http://www.gnu.org/software/classpath/license.html
*
* or as provided in the LICENSE.txt file that accompanied this code.
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.bytedeco.javacpp.indexer;
import org.bytedeco.javacpp.BytePointer;
import org.bytedeco.javacpp.Pointer;
/**
* An indexer for a {@link BytePointer} using the {@link Raw} instance.
*
* @author Samuel Audet
*/
public class ByteRawIndexer extends ByteIndexer {
/** The instance for the raw memory interface. */
protected static final Raw RAW = Raw.getInstance();
/** The backing pointer. */
protected BytePointer pointer;
/** Base address and number of elements accessible. */
final long base, size;
/** Calls {@code ByteRawIndexer(pointer, Index.create(pointer.limit() - pointer.position()))}. */
public ByteRawIndexer(BytePointer pointer) {
this(pointer, Index.create(pointer.limit() - pointer.position()));
}
/** Calls {@code ByteRawIndexer(pointer, Index.create(sizes))}. */
public ByteRawIndexer(BytePointer pointer, long... sizes) {
this(pointer, sizes, strides(sizes));
}
/** Calls {@code ByteRawIndexer(pointer, Index.create(sizes, strides))}. */
public ByteRawIndexer(BytePointer pointer, long[] sizes, long[] strides) {
this(pointer, Index.create(sizes, strides));
}
/** Constructor to set the {@link #pointer} and {@link #index}. */
public ByteRawIndexer(BytePointer pointer, Index index) {
super(index);
this.pointer = pointer;
this.base = pointer.address() + pointer.position();
this.size = pointer.limit() - pointer.position();
}
@Override public Pointer pointer() {
return pointer;
}
@Override public ByteIndexer reindex(Index index) {
return new ByteRawIndexer(pointer, index);
}
public byte getRaw(long i) {
return RAW.getByte(base + checkIndex(i, size));
}
@Override public byte get(long i) {
return getRaw(index(i));
}
@Override public ByteIndexer get(long i, byte[] b, int offset, int length) {
for (int n = 0; n < length; n++) {
b[offset + n] = getRaw(index(i) + n);
}
return this;
}
@Override public byte get(long i, long j) {
return getRaw(index(i, j));
}
@Override public ByteIndexer get(long i, long j, byte[] b, int offset, int length) {
for (int n = 0; n < length; n++) {
b[offset + n] = getRaw(index(i, j) + n);
}
return this;
}
@Override public byte get(long i, long j, long k) {
return getRaw(index(i, j, k));
}
@Override public byte get(long... indices) {
return getRaw(index(indices));
}
@Override public ByteIndexer get(long[] indices, byte[] b, int offset, int length) {
for (int n = 0; n < length; n++) {
b[offset + n] = getRaw(index(indices) + n);
}
return this;
}
public ByteIndexer putRaw(long i, byte b) {
RAW.putByte(base + checkIndex(i, size), b);
return this;
}
@Override public ByteIndexer put(long i, byte b) {
return putRaw(index(i), b);
}
@Override public ByteIndexer put(long i, byte[] b, int offset, int length) {
for (int n = 0; n < length; n++) {
putRaw(index(i) + n, b[offset + n]);
}
return this;
}
@Override public ByteIndexer put(long i, long j, byte b) {
putRaw(index(i, j), b);
return this;
}
@Override public ByteIndexer put(long i, long j, byte[] b, int offset, int length) {
for (int n = 0; n < length; n++) {
putRaw(index(i, j) + n, b[offset + n]);
}
return this;
}
@Override public ByteIndexer put(long i, long j, long k, byte b) {
putRaw(index(i, j, k), b);
return this;
}
@Override public ByteIndexer put(long[] indices, byte b) {
putRaw(index(indices), b);
return this;
}
@Override public ByteIndexer put(long[] indices, byte[] b, int offset, int length) {
for (int n = 0; n < length; n++) {
putRaw(index(indices) + n, b[offset + n]);
}
return this;
}
@Override public byte getByte(long i) {
return RAW.getByte(base + checkIndex(i, size - 1));
}
@Override public ByteIndexer putByte(long i, byte b) {
RAW.putByte(base + checkIndex(i, size - 1), b);
return this;
}
@Override public short getShort(long i) {
return RAW.getShort(base + checkIndex(i, size - 1));
}
@Override public ByteIndexer putShort(long i, short s) {
RAW.putShort(base + checkIndex(i, size - 1), s);
return this;
}
@Override public int getInt(long i) {
return RAW.getInt(base + checkIndex(i, size - 3));
}
@Override public ByteIndexer putInt(long i, int j) {
RAW.putInt(base + checkIndex(i, size - 3), j);
return this;
}
@Override public long getLong(long i) {
return RAW.getLong(base + checkIndex(i, size - 7));
}
@Override public ByteIndexer putLong(long i, long j) {
RAW.putLong(base + checkIndex(i, size - 7), j);
return this;
}
@Override public float getFloat(long i) {
return RAW.getFloat(base + checkIndex(i, size - 3));
}
@Override public ByteIndexer putFloat(long i, float f) {
RAW.putFloat(base + checkIndex(i, size - 3), f);
return this;
}
@Override public double getDouble(long i) {
return RAW.getDouble(base + checkIndex(i, size - 7));
}
@Override public ByteIndexer putDouble(long i, double d) {
RAW.putDouble(base + checkIndex(i, size - 7), d);
return this;
}
@Override public char getChar(long i) {
return RAW.getChar(base + checkIndex(i, size - 1));
}
@Override public ByteIndexer putChar(long i, char c) {
RAW.putChar(base + checkIndex(i, size - 1), c);
return this;
}
@Override public void release() { pointer = null; }
}