Robust/src/Analysis/Disjoint/RefSrcNode.java

   1 package Analysis.Disjoint;
   2
   3 import IR.*;
   4 import IR.Flat.*;
   5 import java.util.*;
   6
   7 public abstract class RefSrcNode {
   8
   9   protected HashSet<RefEdge> referencees;
  10
  11   public RefSrcNode() {
  12     referencees = new HashSet<RefEdge>();
  13   }
  14
  15
  16   public Iterator<RefEdge> iteratorToReferencees() {
  17     return referencees.iterator();
  18   }
  19
  20   public Iterator<RefEdge> iteratorToReferenceesClone() {
  21     HashSet<RefEdge> clone = (HashSet<RefEdge>)referencees.clone();
  22     return clone.iterator();
  23   }
  24
  25   public int getNumReferencees() {
  26     return referencees.size();
  27   }
  28
  29   public void addReferencee( RefEdge edge ) {
  30     assert edge != null;
  31     referencees.add( edge );
  32   }
  33
  34   public void removeReferencee( RefEdge edge ) {
  35     assert edge != null;
  36     assert referencees.contains( edge );
  37     referencees.remove( edge );
  38   }
  39
  40   public RefEdge getReferenceTo( HeapRegionNode hrn,
  41                                  TypeDescriptor type,
  42                                  String         field
  43                                  ) {
  44     assert hrn != null;
  45
  46     Iterator<RefEdge> itrEdge = referencees.iterator();
  47     while( itrEdge.hasNext() ) {
  48       RefEdge edge = itrEdge.next();
  49       if( edge.getDst().equals( hrn ) &&
  50           edge.typeEquals( type )     &&
  51           edge.fieldEquals( field )
  52           ) {
  53         return edge;
  54       }
  55     }
  56
  57     return null;
  58   }
  59 }