From 91c406b2a00882d37d9b931f2feae3eb7ccd03b7 Mon Sep 17 00:00:00 2001
From: de4dot <de4dot@gmail.com>
Date: Wed, 28 Dec 2011 13:22:41 +0100
Subject: [PATCH] Add findAny() methods

---
 blocks/MemberReferenceHelper.cs | 45 +++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/blocks/MemberReferenceHelper.cs b/blocks/MemberReferenceHelper.cs
index 971f8605..8b10e452 100644
--- a/blocks/MemberReferenceHelper.cs
+++ b/blocks/MemberReferenceHelper.cs
@@ -66,6 +66,15 @@ namespace de4dot.blocks {
 		}
 
 		public TValue find(TypeReference typeReference) {
+			TValue value;
+			if (typeReference is TypeDefinition)
+				tokenToValue.TryGetValue(getTokenKey(typeReference), out value);
+			else
+				refToValue.TryGetValue(getReferenceKey(typeReference), out value);
+			return value;
+		}
+
+		public TValue findAny(TypeReference typeReference) {
 			TValue value;
 			if (tokenToValue.TryGetValue(getTokenKey(typeReference), out value))
 				return value;
@@ -128,6 +137,15 @@ namespace de4dot.blocks {
 		protected abstract IFieldReferenceKey getReferenceKey(FieldReference fieldReference);
 
 		public TValue find(FieldReference fieldReference) {
+			TValue value;
+			if (fieldReference is FieldDefinition)
+				tokenToValue.TryGetValue(getTokenKey(fieldReference), out value);
+			else
+				refToValue.TryGetValue(getReferenceKey(fieldReference), out value);
+			return value;
+		}
+
+		public TValue findAny(FieldReference fieldReference) {
 			TValue value;
 			if (tokenToValue.TryGetValue(getTokenKey(fieldReference), out value))
 				return value;
@@ -202,6 +220,15 @@ namespace de4dot.blocks {
 		protected abstract IMethodReferenceKey getReferenceKey(MethodReference methodReference);
 
 		public TValue find(MethodReference methodReference) {
+			TValue value;
+			if (methodReference is MethodDefinition)
+				tokenToValue.TryGetValue(getTokenKey(methodReference), out value);
+			else
+				refToValue.TryGetValue(getReferenceKey(methodReference), out value);
+			return value;
+		}
+
+		public TValue findAny(MethodReference methodReference) {
 			TValue value;
 			if (tokenToValue.TryGetValue(getTokenKey(methodReference), out value))
 				return value;
@@ -275,6 +302,15 @@ namespace de4dot.blocks {
 		protected abstract IPropertyReferenceKey getReferenceKey(PropertyReference propertyReference);
 
 		public TValue find(PropertyReference propertyReference) {
+			TValue value;
+			if (propertyReference is PropertyDefinition)
+				tokenToValue.TryGetValue(getTokenKey(propertyReference), out value);
+			else
+				refToValue.TryGetValue(getReferenceKey(propertyReference), out value);
+			return value;
+		}
+
+		public TValue findAny(PropertyReference propertyReference) {
 			TValue value;
 			if (tokenToValue.TryGetValue(getTokenKey(propertyReference), out value))
 				return value;
@@ -327,6 +363,15 @@ namespace de4dot.blocks {
 		protected abstract IEventReferenceKey getReferenceKey(EventReference eventReference);
 
 		public TValue find(EventReference eventReference) {
+			TValue value;
+			if (eventReference is EventDefinition)
+				tokenToValue.TryGetValue(getTokenKey(eventReference), out value);
+			else
+				refToValue.TryGetValue(getReferenceKey(eventReference), out value);
+			return value;
+		}
+
+		public TValue findAny(EventReference eventReference) {
 			TValue value;
 			if (tokenToValue.TryGetValue(getTokenKey(eventReference), out value))
 				return value;