4 files changed, 23 insertions, 2 deletions

diff --git a/src/cuchaz/enigma/mapping/BehaviorEntry.java b/src/cuchaz/enigma/mapping/BehaviorEntry.java
new file mode 100644
index 0000000..99fdd28
--- /dev/null
+++ b/src/cuchaz/enigma/mapping/BehaviorEntry.java
@@ -0,0 +1,6 @@
+package cuchaz.enigma.mapping;
+
+public interface BehaviorEntry extends Entry
+{
+        public String getSignature();
+}
diff --git a/src/cuchaz/enigma/mapping/ConstructorEntry.java b/src/cuchaz/enigma/mapping/ConstructorEntry.java
index e0fa7cf..0f7dab6 100644
--- a/src/cuchaz/enigma/mapping/ConstructorEntry.java
+++ b/src/cuchaz/enigma/mapping/ConstructorEntry.java
@@ -14,7 +14,7 @@ import java.io.Serializable;
 
 import cuchaz.enigma.Util;
 
-public class ConstructorEntry implements Entry, Serializable
+public class ConstructorEntry implements BehaviorEntry, Serializable
 {
         private static final long serialVersionUID = -868346075317366758L;
         
@@ -54,6 +54,7 @@ public class ConstructorEntry implements Entry, Serializable
                 return m_classEntry.getName();
         }
         
+        @Override
         public String getSignature( )
         {
                 return m_signature;
diff --git a/src/cuchaz/enigma/mapping/MethodEntry.java b/src/cuchaz/enigma/mapping/MethodEntry.java
index b4b9c9b..a311e63 100644
--- a/src/cuchaz/enigma/mapping/MethodEntry.java
+++ b/src/cuchaz/enigma/mapping/MethodEntry.java
@@ -14,7 +14,7 @@ import java.io.Serializable;
 
 import cuchaz.enigma.Util;
 
-public class MethodEntry implements Entry, Serializable
+public class MethodEntry implements BehaviorEntry, Serializable
 {
         private static final long serialVersionUID = 4770915224467247458L;
         
@@ -68,6 +68,7 @@ public class MethodEntry implements Entry, Serializable
                 return m_name;
         }
         
+        @Override
         public String getSignature( )
         {
                 return m_signature;
diff --git a/src/cuchaz/enigma/mapping/Translator.java b/src/cuchaz/enigma/mapping/Translator.java
index fc41f94..a1230db 100644
--- a/src/cuchaz/enigma/mapping/Translator.java
+++ b/src/cuchaz/enigma/mapping/Translator.java
@@ -172,6 +172,19 @@ public class Translator
                 );
         }
         
+        public BehaviorEntry translateEntry( BehaviorEntry in )
+        {
+                if( in instanceof MethodEntry )
+                {
+                        return translateEntry( (MethodEntry)in );
+                }
+                else if( in instanceof ConstructorEntry )
+                {
+                        return translateEntry( (ConstructorEntry)in );
+                }
+                throw new Error( "Wrong entry type!" );
+        }
+        
         public String translate( ArgumentEntry in )
         {
                 for( String className : getSelfAndAncestors( in.getClassName() ) )