From eb40808a4084e4942b4cb979b44b0cce94d58e0c Mon Sep 17 00:00:00 2001
From: jjenista <jjenista>
Date: Wed, 22 Apr 2009 19:56:48 +0000
Subject: [PATCH 1/1] add a convenient method to retreive FlatMethod from task
 or method

---
 Robust/src/IR/State.java | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/Robust/src/IR/State.java b/Robust/src/IR/State.java
index 7a56b557..011a1756 100644
--- a/Robust/src/IR/State.java
+++ b/Robust/src/IR/State.java
@@ -188,6 +188,19 @@ public class State {
     return (FlatMethod)flatmethodmap.get(td);
   }
 
+  // The descriptor is either a method or task desc
+  // and should call one of the above methods
+  public FlatMethod getMethodFlat(Descriptor d) {
+    FlatMethod fm;
+    if( d instanceof MethodDescriptor ) {
+      fm = getMethodFlat( (MethodDescriptor) d);
+    } else {
+      assert d instanceof TaskDescriptor;
+      fm = getMethodFlat( (TaskDescriptor) d);
+    }
+    return fm;
+  }
+
   public void addTreeCode(MethodDescriptor md, BlockNode bn) {
     treemethodmap.put(md,bn);
   }
-- 
2.34.1