Introduce LongCollection interface for storing instrumentation traces

UnitTestBot · Dec 19, 2024 · f2d0afa · f2d0afa
1 parent c5d3fbf
commit f2d0afa
Showing 1 changed file with 44 additions and 5 deletions.
diff --git a/...entation/src/collectors/java/org/usvm/instrumentation/collector/trace/TraceCollector.java b/...entation/src/collectors/java/org/usvm/instrumentation/collector/trace/TraceCollector.java
@@ -3,7 +3,7 @@
 //DO NOT EDIT!
 //USE TRACER
 public class TraceCollector {
-    public static LongHashSet trace = new LongHashSet();
+    public static LongCollection trace = new LongHashSet();
     public static LongArrayWrapper statics = new LongArrayWrapper();
 
     public static void jcInstructionCovered(long jcInstructionId) {
@@ -13,12 +13,27 @@ public static void jcStaticFieldAccessed(long jcStaticFieldAccessId) {
         statics.add(jcStaticFieldAccessId);
     }
 
-    public static class LongArrayWrapper {
+    public interface LongCollection {
+        void add(long key);
+
+        long[] getAllValues();
+
+        boolean contains(long key);
+
+        int realSize();
+
+        boolean isEmpty();
+
+        void clear();
+    }
+
+    public static class LongArrayWrapper implements LongCollection {
         public long[] arr;
         public int size;
 
-        LongArrayWrapper() {
+        public LongArrayWrapper() {
             arr = new long[10];
+            size = 0;
         }
 
         public void add(long el) {
@@ -35,6 +50,31 @@ public void removeLast() {
             size--;
         }
 
+        public long[] getAllValues() {
+            long[] res = new long[size];
+            for (int i = 0; i < size; i++) {
+                res[i] = arr[i];
+            }
+            return res;
+        }
+
+        public boolean contains(long key) {
+            for (int i = 0; i < size; i++) {
+                if (arr[i] == key) {
+                    return true;
+                }
+            }
+            return false;
+        }
+
+        public int realSize() {
+            return size;
+        }
+
+        public boolean isEmpty() {
+            return size == 0;
+        }
+
         public void clear() {
             for (int i = 0; i < size; i++) {
                 arr[i] = 0;
@@ -52,8 +92,7 @@ private void resize() {
         }
     }
 
-
-    public static class LongHashSet {
+    public static class LongHashSet implements LongCollection {
         private static final int DEFAULT_CAPACITY = 1024;
         private static final double DEFAULT_LOAD_FACTOR = 0.75;