Support registering custom layouts

MyWorldLLC · Jan 12, 2024 · d2e3c03 · d2e3c03
1 parent e046acb
commit d2e3c03
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/src/main/java/com/myworldvw/buoy/NativeMapper.java b/lib/src/main/java/com/myworldvw/buoy/NativeMapper.java
@@ -324,6 +324,11 @@ public MemoryLayout makePadding(long structSize, MemoryLayout element){
         return remainder != 0 ? MemoryLayout.paddingLayout(element.byteAlignment() - remainder) : null;
     }
 
+    public <T> NativeMapper register(Class<T> targetType, MemoryLayout customLayout){
+        layouts.put(targetType, customLayout);
+        return this;
+    }
+
     public <T> NativeMapper register(Class<T> targetType){
         return register(targetType, false);
     }
@@ -404,8 +409,7 @@ public <T> NativeMapper register(Class<T> targetType, ObjectHandlers<T> handlers
     }
 
     public <T extends Enum<?>> NativeMapper registerEnum(Class<T> enumeration){
-        layouts.put(enumeration, ValueLayout.JAVA_INT);
-        return this;
+        return register(enumeration, ValueLayout.JAVA_INT);
     }
 
     public boolean isRegistered(Class<?> targetType){