io.vproxy.pni.PNIBuf Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of pni-api-jdk22 Show documentation
panama native interface pni-api-jdk22
The newest version!
package io.vproxy.pni;

import io.vproxy.pni.array.*;
import io.vproxy.pni.hack.VarHandleW;

import java.lang.foreign.MemoryLayout;
import java.lang.foreign.MemorySegment;
import java.lang.foreign.ValueLayout;
import java.nio.ByteBuffer;
import java.util.HashSet;
import java.util.Set;

public class PNIBuf implements NativeObject {
    public static final MemoryLayout LAYOUT = MemoryLayout.structLayout(
        ValueLayout.ADDRESS_UNALIGNED.withName("buf"),
        ValueLayout.JAVA_LONG_UNALIGNED.withName("len")
    );

    public final MemorySegment MEMORY;

    @Override
    public MemorySegment MEMORY() {
        return MEMORY;
    }

    public PNIBuf(MemorySegment memory) {
        MEMORY = memory.reinterpret(LAYOUT.byteSize());
    }

    public PNIBuf(Allocator allocator) {
        this(allocator.allocate(LAYOUT.byteSize()));
    }

    private PNIBuf(Allocator allocator, MemorySegment buf) {
        this(allocator);
        set(buf);
    }

    public static PNIBuf of(Allocator allocator, MemorySegment v) {
        if (v == null) return null;
        return new PNIBuf(allocator, v);
    }

    public static MemorySegment memoryOf(Allocator allocator, MemorySegment v) {
        if (v == null) return MemorySegment.NULL;
        return of(allocator, v).MEMORY;
    }

    public static PNIBuf of(Allocator allocator, BoolArray v) {
        if (v == null) return null;
        return new PNIBuf(allocator, v.MEMORY);
    }

    public static MemorySegment memoryOf(Allocator allocator, BoolArray v) {
        if (v == null) return MemorySegment.NULL;
        return of(allocator, v).MEMORY;
    }

    public static PNIBuf of(Allocator allocator, CharArray v) {
        if (v == null) return null;
        return new PNIBuf(allocator, v.MEMORY);
    }

    public static MemorySegment memoryOf(Allocator allocator, CharArray v) {
        if (v == null) return MemorySegment.NULL;
        return of(allocator, v).MEMORY;
    }

    public static PNIBuf of(Allocator allocator, DoubleArray v) {
        if (v == null) return null;
        return new PNIBuf(allocator, v.MEMORY);
    }

    public static MemorySegment memoryOf(Allocator allocator, DoubleArray v) {
        if (v == null) return MemorySegment.NULL;
        return of(allocator, v).MEMORY;
    }

    public static PNIBuf of(Allocator allocator, FloatArray v) {
        if (v == null) return null;
        return new PNIBuf(allocator, v.MEMORY);
    }

    public static MemorySegment memoryOf(Allocator allocator, FloatArray v) {
        if (v == null) return MemorySegment.NULL;
        return of(allocator, v).MEMORY;
    }

    public static PNIBuf of(Allocator allocator, IntArray v) {
        if (v == null) return null;
        return new PNIBuf(allocator, v.MEMORY);
    }

    public static MemorySegment memoryOf(Allocator allocator, IntArray v) {
        if (v == null) return MemorySegment.NULL;
        return of(allocator, v).MEMORY;
    }

    public static PNIBuf of(Allocator allocator, LongArray v) {
        if (v == null) return null;
        return new PNIBuf(allocator, v.MEMORY);
    }

    public static MemorySegment memoryOf(Allocator allocator, LongArray v) {
        if (v == null) return MemorySegment.NULL;
        return of(allocator, v).MEMORY;
    }

    public static PNIBuf of(Allocator allocator, ShortArray v) {
        if (v == null) return null;
        return new PNIBuf(allocator, v.MEMORY);
    }

    public static MemorySegment memoryOf(Allocator allocator, ShortArray v) {
        if (v == null) return MemorySegment.NULL;
        return of(allocator, v).MEMORY;
    }

    public static PNIBuf of(Allocator allocator, PointerArray v) {
        if (v == null) return null;
        return new PNIBuf(allocator, v.MEMORY);
    }

    public static MemorySegment memoryOf(Allocator allocator, PointerArray v) {
        if (v == null) return MemorySegment.NULL;
        return of(allocator, v).MEMORY;
    }

    public static PNIBuf of(Allocator allocator, RefArray v) {
        if (v == null) return null;
        return new PNIBuf(allocator, v.MEMORY);
    }

    public static MemorySegment memoryOf(Allocator allocator, RefArray v) {
        if (v == null) return MemorySegment.NULL;
        return of(allocator, v).MEMORY;
    }

    private static final VarHandleW bufVH = VarHandleW.of(LAYOUT.varHandle(
        MemoryLayout.PathElement.groupElement("buf")
    ));

    private static final VarHandleW lenVH = VarHandleW.of(LAYOUT.varHandle(
        MemoryLayout.PathElement.groupElement("len")
    ));

    public MemorySegment get() {
        var seg = bufVH.getMemorySegment(MEMORY);
        if (seg.address() == 0) {
            return null;
        }
        long len = lenVH.getLong(MEMORY);
        return seg.reinterpret(len);
    }

    public boolean isNull() {
        var seg = bufVH.getMemorySegment(MEMORY);
        return seg.address() == 0;
    }

    public void set(MemorySegment buf) {
        bufVH.set(MEMORY, buf);
        lenVH.set(MEMORY, buf.byteSize());
    }

    public void set(ByteBuffer buf) {
        set(MemorySegment.ofBuffer(buf));
    }

    public void setToNull() {
        lenVH.set(MEMORY, 0L);
        bufVH.set(MEMORY, MemorySegment.NULL);
    }

    public BoolArray toBoolArray() {
        var seg = get();
        return seg == null ? null : new BoolArray(seg);
    }

    public CharArray toCharArray() {
        var seg = get();
        return seg == null ? null : new CharArray(seg);
    }

    public DoubleArray toDoubleArray() {
        var seg = get();
        return seg == null ? null : new DoubleArray(seg);
    }

    public FloatArray toFloatArray() {
        var seg = get();
        return seg == null ? null : new FloatArray(seg);
    }

    public IntArray toIntArray() {
        var seg = get();
        return seg == null ? null : new IntArray(seg);
    }

    public LongArray toLongArray() {
        var seg = get();
        return seg == null ? null : new LongArray(seg);
    }

    public ShortArray toShortArray() {
        var seg = get();
        return seg == null ? null : new ShortArray(seg);
    }

    public PointerArray toPointerArray() {
        var seg = get();
        return seg == null ? null : new PointerArray(seg);
    }

    public ByteBuffer toByteBuffer() {
        var seg = get();
        if (seg == null) {
            return null;
        }
        return seg.asByteBuffer();
    }

    @Override
    public String toString() {
        var sb = new StringBuilder();
        toString(sb, 0, new HashSet<>(), false);
        return sb.toString();
    }

    @Override
    public void toString(StringBuilder sb, int indent, Set visited, boolean corrupted) {
        if (corrupted) {
            sb.append("@").append(Long.toString(MEMORY.address(), 16));
        } else {
            sb.append("PNIBuf(").append(PanamaUtils.memorySegmentToString(get())).append(")@")
                .append(Long.toString(MEMORY.address(), 16));
        }
    }
}