feat(base.collection): methods for finding index of an item that matc…

…hes a predicate with specified ordinal offset [skip ci]
auioc · Oct 11, 2022 · 4565a89 · 4565a89
1 parent 3c5e67c
commit 4565a89
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/src/main/java/org/auioc/mcmod/arnicalib/base/collection/ListUtils.java b/src/main/java/org/auioc/mcmod/arnicalib/base/collection/ListUtils.java
@@ -12,24 +12,34 @@ public static <T> T getLast(List<T> list) {
         return list.get(list.size() - 1);
     }
 
-    public static <T> int indexOf(List<T> list, Predicate<T> predicate) {
+    public static <T> int indexOf(List<T> list, Predicate<T> predicate, int ordinal) {
         for (int i = 0, l = list.size(); i < l; ++i) {
             if (predicate.test(list.get(i))) {
-                return i;
+                if (ordinal <= 0) return i;
+                ordinal--;
             }
         }
         return -1;
     }
 
-    public static <T> int lastIndexOf(List<T> list, Predicate<T> predicate) {
+    public static <T> int indexOf(List<T> list, Predicate<T> predicate) {
+        return indexOf(list, predicate, 0);
+    }
+
+    public static <T> int lastIndexOf(List<T> list, Predicate<T> predicate, int ordinal) {
         for (int i = list.size() - 1; i >= 0; --i) {
             if (predicate.test(list.get(i))) {
-                return i;
+                if (ordinal <= 0) return i;
+                ordinal--;
             }
         }
         return -1;
     }
 
+    public static <T> int lastIndexOf(List<T> list, Predicate<T> predicate) {
+        return lastIndexOf(list, predicate, 0);
+    }
+
     public <T> int[] allIndexesOf(List<T> list, Predicate<T> predicate) {
         return IntStream
             .range(0, list.size())