ObjVsBpFilter: add the disableBp() and disableObj() methods

stephengold · Sep 23, 2024 · 8ff1650 · 8ff1650
1 parent 9c6b2e3
commit 8ff1650
Show file tree

Hide file tree

Showing 2 changed files with 68 additions and 0 deletions.
diff --git a/src/main/java/com/github/stephengold/joltjni/ObjVsBpFilter.java b/src/main/java/com/github/stephengold/joltjni/ObjVsBpFilter.java
@@ -44,6 +44,32 @@ public ObjVsBpFilter(int numObjectLayers, int numBpLayers) {
     // *************************************************************************
     // new methods exposed
 
+    /**
+     * Disable interactions with the specified layer.
+     *
+     * @param bpLayer the index of the broad-phase layer (&lt; numBpLayers)
+     * @return the modified filter (for chaining)
+     */
+    public ObjVsBpFilter disableBp(int bpLayer) {
+        long filterVa = va();
+        disableBp(filterVa, bpLayer);
+
+        return this;
+    }
+
+    /**
+     * Disable interactions with the specified layer.
+     *
+     * @param objLayer the index of the object layer (&lt; numObjectLayers)
+     * @return the modified filter (for chaining)
+     */
+    public ObjVsBpFilter disableObj(int objLayer) {
+        long filterVa = va();
+        disableObj(filterVa, objLayer);
+
+        return this;
+    }
+
     /**
      * Disable interactions between the specified layers.
      *
@@ -61,6 +87,10 @@ public ObjVsBpFilter disablePair(int objLayer, int bpLayer) {
     native private static long createObjVsBpFilter(
             int numObjectLayers, int numBpLayers);
 
+    native private static void disableBp(long filterVa, int bpLayer);
+
+    native private static void disableObj(long filterVa, int objLayer);
+
     native private static void disablePair(
             long filterVa, int objLayer, int bpLayer);
 

diff --git a/src/main/native/glue/o/ObjVsBpFilter.cpp b/src/main/native/glue/o/ObjVsBpFilter.cpp
@@ -58,6 +58,20 @@ class ObjVsBpFilter final : public ObjectVsBroadPhaseLayerFilter {
         return result;
     }
 
+    void DisableBp(BroadPhaseLayer inBpLayer) {
+        JPH_ASSERT(inBpLayer.GetValue() < mNumBpLayers);
+        for (ObjectLayer i = 0; i < mNumObjectLayers; ++i) {
+            mppEnable[i][inBpLayer.GetValue()] = false;
+        }
+    }
+
+    void DisableObj(ObjectLayer inObjectLayer) {
+        JPH_ASSERT(inObjectLayer < mNumObjectLayers);
+        for (uint j = 0; j < mNumBpLayers; ++j) {
+            mppEnable[inObjectLayer][j] = false;
+        }
+    }
+
     void DisablePair(ObjectLayer inObjectLayer, BroadPhaseLayer inBpLayer) {
         JPH_ASSERT(inObjectLayer < mNumObjectLayers);
         JPH_ASSERT(inBpLayer.GetValue() < mNumBpLayers);
@@ -87,6 +101,30 @@ JNIEXPORT jlong JNICALL Java_com_github_stephengold_joltjni_ObjVsBpFilter_create
     return reinterpret_cast<jlong> (pFilter);
 }
 
+/*
+ * Class:     com_github_stephengold_joltjni_ObjVsBpFilter
+ * Method:    disableBp
+ * Signature: (JI)V
+ */
+JNIEXPORT void JNICALL Java_com_github_stephengold_joltjni_ObjVsBpFilter_disableBp
+  (JNIEnv *, jclass, jlong filterVa, jint bpLayer) {
+    ObjVsBpFilter * const pFilter
+            = reinterpret_cast<ObjVsBpFilter *> (filterVa);
+    pFilter->DisableBp(BroadPhaseLayer(bpLayer));
+}
+
+/*
+ * Class:     com_github_stephengold_joltjni_ObjVsBpFilter
+ * Method:    disableObj
+ * Signature: (JI)V
+ */
+JNIEXPORT void JNICALL Java_com_github_stephengold_joltjni_ObjVsBpFilter_disableObj
+  (JNIEnv *, jclass, jlong filterVa, jint objLayer) {
+    ObjVsBpFilter * const pFilter
+            = reinterpret_cast<ObjVsBpFilter *> (filterVa);
+    pFilter->DisableObj(objLayer);
+}
+
 /*
  * Class:     com_github_stephengold_joltjni_ObjVsBpFilter
  * Method:    disablePair