From c632f3d8057a5ff3bd05a5693c43753fedc59fc4 Mon Sep 17 00:00:00 2001
From: Haifeng Shi <shihaifeng1998@gmail.com>
Date: Thu, 1 Dec 2022 17:21:45 -0500
Subject: [PATCH] addDeepPreference to cfi

---
 src/checkers/inference/InferenceVisitor.java | 43 ++++++++++++++++++++
 1 file changed, 43 insertions(+)

diff --git a/src/checkers/inference/InferenceVisitor.java b/src/checkers/inference/InferenceVisitor.java
index df07527f..7eeda197 100644
--- a/src/checkers/inference/InferenceVisitor.java
+++ b/src/checkers/inference/InferenceVisitor.java
@@ -22,6 +22,7 @@
 import java.lang.annotation.Annotation;
 import java.util.Arrays;
 import java.util.HashSet;
+import java.util.LinkedList;
 import java.util.List;
 import java.util.Set;
 import java.util.logging.Logger;
@@ -261,6 +262,48 @@ public void mainIsNoneOf(AnnotatedTypeMirror ty, AnnotationMirror[] mods, String
         }
     }
 
+    private void addDeepPreferenceImpl(AnnotatedTypeMirror ty, AnnotationMirror goal, int weight,
+                                       java.util.List<AnnotatedTypeMirror> visited, Tree node) {
+        if (infer) {
+            if (visited.contains(ty)) {
+                return;
+            }
+            visited.add(ty);
+
+            final SlotManager slotManager = InferenceMain.getInstance().getSlotManager();
+            Slot el = slotManager.getSlot(ty);
+
+            if (el == null) {
+                logger.warning("InferenceVisitor::addDeepPreferenceImpl: no annotation in type: " + ty);
+            } else {
+                addPreference(ty, goal, weight);
+            }
+
+            if (ty.getKind() == TypeKind.DECLARED) {
+                AnnotatedDeclaredType declaredType = (AnnotatedDeclaredType) ty;
+                for (AnnotatedTypeMirror typearg : declaredType.getTypeArguments()) {
+                    addDeepPreferenceImpl(typearg, goal, weight, visited, node);
+                }
+            } else if (ty.getKind() == TypeKind.ARRAY) {
+                AnnotatedArrayType arrayType = (AnnotatedArrayType) ty;
+                addDeepPreferenceImpl(arrayType.getComponentType(), goal, weight, visited, node);
+            } else if (ty.getKind() == TypeKind.TYPEVAR) {
+                AnnotatedTypeVariable atv = (AnnotatedTypeVariable) ty;
+                if (atv.getUpperBound()!=null) {
+                    addDeepPreferenceImpl(atv.getUpperBound(), goal, weight, visited, node);
+                }
+                if (atv.getLowerBound()!=null) {
+                    addDeepPreferenceImpl(atv.getLowerBound(), goal, weight, visited, node);
+                }
+            }
+        }
+        // Else, do nothing
+    }
+
+    public void addDeepPreference(AnnotatedTypeMirror ty, AnnotationMirror goal, int weight, Tree node) {
+        addDeepPreferenceImpl(ty, goal, weight, new LinkedList<>(), node);
+    }
+
     public void addPreference(AnnotatedTypeMirror type, AnnotationMirror anno, int weight) {
         if (infer) {
             ConstraintManager cManager = InferenceMain.getInstance().getConstraintManager();