summaryrefslogtreecommitdiffstats
path: root/src/com/mbien/opencl/InternalBufferUtil.java
blob: a2573784f621a2bb2398d415ebccb9e379d56da8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
package com.mbien.opencl;

import java.lang.reflect.Field;
import java.nio.Buffer;
import sun.misc.Unsafe;

/**
 *
 * @author Michael Bien
 */
class InternalBufferUtil {

    private static final long addressFieldOffset;
    private static Unsafe unsafe;

    static {
        try {
            Field f = Buffer.class.getDeclaredField("address");

            Field[] fields = Unsafe.class.getDeclaredFields();
            for (int i = 0; i < fields.length; i++) {
                if (fields[i].getName().equals("theUnsafe")) {
                    fields[i].setAccessible(true);
                    unsafe = (Unsafe)fields[i].get(Unsafe.class);
                    break;
                }
            }

            addressFieldOffset = unsafe.objectFieldOffset(f);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    public static long getDirectBufferAddress(Buffer buffer) {
        return ((buffer == null) ? 0 : unsafe.getLong(buffer, addressFieldOffset));
    }

}