org.bytedeco.javacpp.indexer.FloatRawIndexer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of javacpp Show documentation
The missing bridge between Java and native C++
There is a newer version: 1.5.11
/*
 * 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.FloatPointer;
import org.bytedeco.javacpp.Pointer;

/**
 * An indexer for a {@link FloatPointer} using the {@link Raw} instance.
 *
 * @author Samuel Audet
 */
public class FloatRawIndexer extends FloatIndexer {
    /** The instance for the raw memory interface. */
    protected static final Raw RAW = Raw.getInstance();
    /** The backing pointer. */
    protected FloatPointer pointer;
    /** Base address and number of elements accessible. */
    final long base, size;

    /** Calls {@code FloatRawIndexer(pointer, Index.create(pointer.limit() - pointer.position()))}. */
    public FloatRawIndexer(FloatPointer pointer) {
        this(pointer, Index.create(pointer.limit() - pointer.position()));
    }

    /** Calls {@code FloatRawIndexer(pointer, Index.create(sizes))}. */
    public FloatRawIndexer(FloatPointer pointer, long... sizes) {
        this(pointer, Index.create(sizes));
    }

    /** Calls {@code FloatRawIndexer(pointer, Index.create(sizes, strides))}. */
    public FloatRawIndexer(FloatPointer pointer, long[] sizes, long[] strides) {
        this(pointer, Index.create(sizes, strides));
    }

    /** Constructor to set the {@link #pointer} and {@link #index}. */
    public FloatRawIndexer(FloatPointer pointer, Index index) {
        super(index);
        this.pointer = pointer;
        this.base = pointer.address() + pointer.position() * VALUE_BYTES;
        this.size = pointer.limit() - pointer.position();
    }

    @Override public Pointer pointer() {
        return pointer;
    }

    @Override public FloatIndexer reindex(Index index) {
        return new FloatRawIndexer(pointer, index);
    }

    public float getRaw(long i) {
        return RAW.getFloat(base + checkIndex(i, size) * VALUE_BYTES);
    }
    @Override public float get(long i) {
        return getRaw(index(i));
    }
    @Override public FloatIndexer get(long i, float[] f, int offset, int length) {
        for (int n = 0; n < length; n++) {
            f[offset + n] = getRaw(index(i) + n);
        }
        return this;
    }
    @Override public float get(long i, long j) {
        return getRaw(index(i, j));
    }
    @Override public FloatIndexer get(long i, long j, float[] f, int offset, int length) {
        for (int n = 0; n < length; n++) {
            f[offset + n] = getRaw(index(i, j) + n);
        }
        return this;
    }
    @Override public float get(long i, long j, long k) {
        return getRaw(index(i, j, k));
    }
    @Override public float get(long... indices) {
        return getRaw(index(indices));
    }
    @Override public FloatIndexer get(long[] indices, float[] f, int offset, int length) {
        for (int n = 0; n < length; n++) {
            f[offset + n] = getRaw(index(indices) + n);
        }
        return this;
    }

    public FloatIndexer putRaw(long i, float f) {
        RAW.putFloat(base + checkIndex(i, size) * VALUE_BYTES, f);
        return this;
    }
    @Override public FloatIndexer put(long i, float f) {
        return putRaw(index(i), f);
    }
    @Override public FloatIndexer put(long i, float[] f, int offset, int length) {
        for (int n = 0; n < length; n++) {
            putRaw(index(i) + n, f[offset + n]);
        }
        return this;
    }
    @Override public FloatIndexer put(long i, long j, float f) {
        putRaw(index(i, j), f);
        return this;
    }
    @Override public FloatIndexer put(long i, long j, float[] f, int offset, int length) {
        for (int n = 0; n < length; n++) {
            putRaw(index(i, j) + n, f[offset + n]);
        }
        return this;
    }
    @Override public FloatIndexer put(long i, long j, long k, float f) {
        putRaw(index(i, j, k), f);
        return this;
    }
    @Override public FloatIndexer put(long[] indices, float f) {
        putRaw(index(indices), f);
        return this;
    }
    @Override public FloatIndexer put(long[] indices, float[] f, int offset, int length) {
        for (int n = 0; n < length; n++) {
            putRaw(index(indices) + n, f[offset + n]);
        }
        return this;
    }

    @Override public void release() { pointer = null; }
}