more changes

[IRC.git] / Robust / src / IR / Flat / FlatMethod.java

diff --git a/Robust/src/IR/Flat/FlatMethod.java b/Robust/src/IR/Flat/FlatMethod.java
index be3c8182877704b38da955111c77576b073a4458..4f1e0c9f3aa7661f1148a6acffcf29508880d5a9 100644 (file)
--- a/Robust/src/IR/Flat/FlatMethod.java
+++ b/Robust/src/IR/Flat/FlatMethod.java
@@ -73,12 +73,12 @@ public class FlatMethod extends FlatNode {
 
     /** This method returns a set of the nodes in this flat representation */
 
-    public Set getNodeSet() {
-       HashSet tovisit=new HashSet();
-       HashSet visited=new HashSet();
+    public Set<FlatNode> getNodeSet() {
+       HashSet<FlatNode> tovisit=new HashSet<FlatNode>();
+       HashSet<FlatNode> visited=new HashSet<FlatNode>();
        tovisit.add(this);
        while(!tovisit.isEmpty()) {
-           FlatNode fn=(FlatNode)tovisit.iterator().next();
+           FlatNode fn=tovisit.iterator().next();
            tovisit.remove(fn);
            visited.add(fn);
            for(int i=0;i<fn.numNext();i++) {