1 files changed, 20 insertions, 14 deletions

diff --git a/src/cuchaz/enigma/ClassFile.java b/src/cuchaz/enigma/ClassFile.java
index 221a119..c3c72a4 100644
--- a/src/cuchaz/enigma/ClassFile.java
+++ b/src/cuchaz/enigma/ClassFile.java
@@ -10,36 +10,42 @@
  ******************************************************************************/
 package cuchaz.enigma;
 
-import java.util.regex.Pattern;
 
 public class ClassFile
 {
-        private static Pattern m_obfuscatedClassPattern;
+        private String m_obfName;
+        private String m_deobfName;
         
-        static
+        public ClassFile( String obfName )
         {
-                m_obfuscatedClassPattern = Pattern.compile( "^[a-z]+$" );
+                m_obfName = obfName;
         }
         
-        private String m_name;
-        
-        public ClassFile( String name )
+        public String getName( )
         {
-                m_name = name;
+                if( m_deobfName != null )
+                {
+                        return m_deobfName;
+                }
+                return m_obfName;
         }
         
-        public String getName( )
+        public String getObfName( )
         {
-                return m_name;
+                return m_obfName;
         }
         
-        public boolean isObfuscated( )
+        public String getDeobfName( )
         {
-                return m_obfuscatedClassPattern.matcher( m_name ).matches();
+                return m_deobfName;
         }
-
+        public void setDeobfName( String val )
+        {
+                m_deobfName = val;
+        }
+        
         public String getPath( )
         {
-                return m_name.replace( ".", "/" ) + ".class";
+                return m_deobfName.replace( ".", "/" ) + ".class";
         }
 }