changes.

[IRC.git] / Robust / src / IR / TypeUtil.java

diff --git a/Robust/src/IR/TypeUtil.java b/Robust/src/IR/TypeUtil.java
index a2a76e397f44ee5bbf86f4fe6cca8e8ea51e2aab..fe802b0b5be4fdf971d3e3347a9d386649fb4b45 100644 (file)
--- a/Robust/src/IR/TypeUtil.java
+++ b/Robust/src/IR/TypeUtil.java
@@ -17,6 +17,7 @@ public class TypeUtil {
   State state;
   Hashtable supertable;
   Hashtable subclasstable;
+  Hashtable directSubClassTable;
   BuildIR bir;
 
   // for interfaces
@@ -284,6 +285,7 @@ NextMethod:
 
   public void createFullTable() {
     subclasstable=new Hashtable();
+    directSubClassTable=new Hashtable();
     HashSet tovisit=new HashSet();
     HashSet visited=new HashSet();
 
@@ -305,6 +307,14 @@ NextMethod:
         }
       }
 
+      if(tmp!=null){
+        if(!directSubClassTable.containsKey(tmp)){
+          directSubClassTable.put(tmp, new HashSet());
+        }
+        ((Set)directSubClassTable.get(tmp)).add(cd);        
+      }
+      
+
       while(tmp!=null) {
         if (!subclasstable.containsKey(tmp))
           subclasstable.put(tmp,new HashSet());
@@ -345,7 +355,12 @@ NextMethod:
       }
     }
   }
-
+  
+  public Set getDirectSubClasses(ClassDescriptor cd) {
+    System.out.println("$$$cd="+cd+"  children="+directSubClassTable.get(cd));
+    return (Set)directSubClassTable.get(cd);
+  }
+  
   public Set getSubClasses(ClassDescriptor cd) {
     return (Set)subclasstable.get(cd);
   }