more util

src/main/java/me/cortex/voxy/common/util/ThreadUtils.java

@@ -1,6 +1,8 @@
 package me.cortex.voxy.common.util;
 
 import org.lwjgl.system.JNI;
+import org.lwjgl.system.MemoryStack;
+import org.lwjgl.system.MemoryUtil;
 import org.lwjgl.system.Platform;
 import org.lwjgl.system.windows.Kernel32;
 
@@ -12,11 +14,50 @@ public class ThreadUtils {
     public static final int WIN32_THREAD_MODE_BACKGROUND_END = 0x00020000;
     private static final boolean isWindows = Platform.get() == Platform.WINDOWS;
     private static final long SetThreadPriority;
+    private static final long SetThreadSelectedCpuSetMasks;
     static {
         if (isWindows) {
             SetThreadPriority = Kernel32.getLibrary().getFunctionAddress("SetThreadPriority");
+            SetThreadSelectedCpuSetMasks = Kernel32.getLibrary().getFunctionAddress("SetThreadSelectedCpuSetMasks");
         } else {
             SetThreadPriority = 0;
+            SetThreadSelectedCpuSetMasks = 0;
+        }
+    }
+
+    public static boolean SetThreadSelectedCpuSetMasksWin32(long mask) {
+        return SetThreadSelectedCpuSetMasksWin32(new long[]{mask}, new short[]{0});
+    }
+
+    public static boolean SetThreadSelectedCpuSetMasksWin32(long[] masks, short[] groups) {
+        if (SetThreadSelectedCpuSetMasks == 0 || !isWindows) {
+            return false;
+        }
+
+        if (masks == null) {
+            int retVal = JNI.invokePPCI(Kernel32.GetCurrentThread(), 0, (short) 0, SetThreadSelectedCpuSetMasks);
+            if (retVal == 0) {
+                throw new IllegalStateException();
+            }
+            return true;
+        }
+
+        if (masks.length != groups.length) {
+            throw new IllegalArgumentException();
+        }
+        try (var stack = MemoryStack.stackPush()) {
+            long ptr = stack.ncalloc(16, masks.length, 16);
+            MemoryUtil.memSet(ptr, 0, masks.length*16L);
+            for (int i = 0; i < masks.length; i++) {
+                MemoryUtil.memPutLong(ptr+i*16L, masks[i]);
+                MemoryUtil.memPutShort(ptr+i*16L+8L, groups[i]);
+            }
+
+            int retVal = JNI.invokePPCI(Kernel32.GetCurrentThread(), ptr, (short)masks.length, SetThreadSelectedCpuSetMasks);
+            if (retVal == 0) {
+                throw new IllegalStateException();
+            }
+            return true;
         }
     }