chore: use class reference to discover calling code instead of string

Author: Christopher-Chianelli

core/src/main/java/ai/timefold/solver/core/impl/bavet/common/AbstractNode.java

@@ -17,6 +17,10 @@ public abstract class AbstractNode {
 
     private long id;
     private long layerIndex = -1;
+    /**
+     * Used to map this node to where it was defined/used.
+     * Null when profiling is disabled.
+     */
     private @Nullable SortedSet<ConstraintNodeLocation> locationSet;
 
     /**

core/src/main/java/ai/timefold/solver/core/impl/bavet/common/BavetAbstractConstraintStream.java

@@ -1,15 +1,25 @@
 package ai.timefold.solver.core.impl.bavet.common;
 
+import java.lang.reflect.Method;
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.List;
 import java.util.Objects;
 import java.util.Set;
 import java.util.SortedSet;
 import java.util.TreeSet;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
 
 import ai.timefold.solver.core.api.score.Score;
 import ai.timefold.solver.core.api.score.constraint.ConstraintRef;
 import ai.timefold.solver.core.api.score.stream.Constraint;
+import ai.timefold.solver.core.api.score.stream.ConstraintFactory;
+import ai.timefold.solver.core.api.score.stream.ConstraintStream;
+import ai.timefold.solver.core.api.score.stream.bi.BiConstraintStream;
+import ai.timefold.solver.core.api.score.stream.quad.QuadConstraintStream;
+import ai.timefold.solver.core.api.score.stream.tri.TriConstraintStream;
+import ai.timefold.solver.core.api.score.stream.uni.UniConstraintStream;
 import ai.timefold.solver.core.impl.score.stream.bavet.BavetConstraint;
 import ai.timefold.solver.core.impl.score.stream.bavet.BavetConstraintFactory;
 import ai.timefold.solver.core.impl.score.stream.bavet.common.BavetScoringConstraintStream;
@@ -24,9 +34,14 @@ public abstract class BavetAbstractConstraintStream<Solution_>
         extends AbstractConstraintStream<Solution_>
         implements BavetStream {
 
-    private static final String BAVET_IMPL_PACKAGE = "ai.timefold.solver.core.impl.bavet";
-    private static final String BAVET_SCORE_IMPL_PACKAGE = "ai.timefold.solver.core.impl.score.stream";
-    private static final String BAVET_SCORE_API_PACKAGE = "ai.timefold.solver.core.api.score.stream";
+    private static final Set<String> constraintStreamApiMethodSet = Stream.of(
+            UniConstraintStream.class,
+            BiConstraintStream.class,
+            TriConstraintStream.class,
+            QuadConstraintStream.class)
+            .flatMap(clazz -> Arrays.stream(clazz.getMethods()))
+            .map(Method::getName)
+            .collect(Collectors.toUnmodifiableSet());
 
     protected final BavetConstraintFactory<Solution_> constraintFactory;
     protected final BavetAbstractConstraintStream<Solution_> parent;
@@ -54,10 +69,11 @@ protected BavetAbstractConstraintStream(BavetConstraintFactory<Solution_> constr
     public abstract StreamKind getStreamKind();
 
     private ConstraintNodeLocation determineStreamLocation() {
-        return StackWalker.getInstance().walk(stack -> stack
-                .dropWhile(stackFrame -> stackFrame.getClassName().startsWith(BAVET_IMPL_PACKAGE) ||
-                        stackFrame.getClassName().startsWith(BAVET_SCORE_IMPL_PACKAGE) ||
-                        stackFrame.getClassName().startsWith(BAVET_SCORE_API_PACKAGE))
+        return StackWalker.getInstance(StackWalker.Option.RETAIN_CLASS_REFERENCE).walk(stack -> stack
+                .dropWhile(stackFrame -> !ConstraintStream.class.isAssignableFrom(stackFrame.getDeclaringClass())
+                        && !constraintStreamApiMethodSet.contains(stackFrame.getMethodName()))
+                .dropWhile(stackFrame -> ConstraintStream.class.isAssignableFrom(stackFrame.getDeclaringClass()) ||
+                        ConstraintFactory.class.isAssignableFrom(stackFrame.getDeclaringClass()))
                 .map(stackFrame -> new ConstraintNodeLocation(
                         getStreamKind(),
                         stackFrame.getClassName(),