pulled {Method,Member,Field}Ref into Type.Class; made them inner classes; much cleaner
[org.ibex.classgen.git] / src / org / ibex / classgen / Type.java
index 33af24d..491cc12 100644 (file)
 package org.ibex.classgen;
 
 import java.util.StringTokenizer;
+import java.util.Hashtable;
 
 public class Type {
-    public static final Type VOID = new Type("V");
-    public static final Type INT = new Type("I");
-    public static final Type LONG = new Type("J");
-    public static final Type BOOLEAN = new Type("Z");
-    public static final Type DOUBLE = new Type("D");
-    public static final Type FLOAT = new Type("F");
+
+    private static Hashtable instances = new Hashtable();  // this has to appear at the top of the file
+
+    // Public API //////////////////////////////////////////////////////////////////////////////
+
+    public static final Type VOID = new Type("V", "void");
+    public static final Type INT = new Type("I", "int");
+    public static final Type LONG = new Type("J", "long");
+    public static final Type BOOLEAN = new Type("Z", "boolean");
+    public static final Type DOUBLE = new Type("D", "double");
+    public static final Type FLOAT = new Type("F", "float");
+    public static final Type BYTE = new Type("B", "byte");
+    public static final Type CHAR = new Type("C", "char");
+    public static final Type SHORT = new Type("S", "short");
     
-    public static final Type.Object OBJECT = new Type.Object("java.lang.Object");
-    public static final Type.Object STRING = new Type.Object("java.lang.String");
-    public static final Type.Object STRINGBUFFER = new Type.Object("java.lang.StringBuffer");
-    public static final Type.Object INTEGER_OBJECT = new Type.Object("java.lang.Integer");
-    public static final Type.Object DOUBLE_OBJECT = new Type.Object("java.lang.Double");
-    public static final Type.Object FLOAT_OBJECT = new Type.Object("java.lang.Float");
+    public static final Type.Class OBJECT = new Type.Class("java.lang.Object");
+    public static final Type.Class STRING = new Type.Class("java.lang.String");
+    public static final Type.Class STRINGBUFFER = new Type.Class("java.lang.StringBuffer");
+    public static final Type.Class INTEGER_OBJECT = new Type.Class("java.lang.Integer");
+    public static final Type.Class DOUBLE_OBJECT = new Type.Class("java.lang.Double");
+    public static final Type.Class FLOAT_OBJECT = new Type.Class("java.lang.Float");
     
+    /** A zero element Type[] array (can be passed as the "args" param when a method takes no arguments */
     public static final Type[] NO_ARGS = new Type[0];
     
-    final String descriptor;
-    
-    Type(String descriptor) { this.descriptor = descriptor; }
-    
-    public final String getDescriptor() { return descriptor; }
-    public int hashCode() { return descriptor.hashCode(); }
-    public boolean equals(java.lang.Object o) { return o instanceof Type && ((Type)o).descriptor.equals(descriptor); }
-    
-    public static Type arrayType(Type base) { return arrayType(base,1); }
-    public static Type arrayType(Type base, int dim) {
-        StringBuffer sb = new StringBuffer(base.descriptor.length() + dim);
-        for(int i=0;i<dim;i++) sb.append("[");
-        sb.append(base.descriptor);
-        return new Type(sb.toString());
+    /** guarantee: there will only be one instance of Type for a given descriptor ==> equals() and == are interchangeable */
+    public static Type instance(String d) {
+        Type ret = (Type)instances.get(d);
+        if (ret != null) return ret;
+        if (d.startsWith("[")) return new Type.Array(instance(d.substring(1)));
+        return new Type.Class(d);
     }
-    
-    public static class Object extends Type {
-        public Object(String s) { super(_initHelper(s)); }
-        
+
+    public       String  toString() { return toString; }
+    public final String  getDescriptor() { return descriptor; }
+    public       int     hashCode() { return descriptor.hashCode(); }
+    public       boolean equals(java.lang.Object o) { return this==o; }
+
+    public Type.Array  makeArray() { return (Type.Array)instance("["+descriptor); }
+
+    public Type.Ref    asRef()       { throw new RuntimeException("attempted to use "+this+" as a Type.Ref, which it is not"); }
+    public Type.Class  asClass()     { throw new RuntimeException("attempted to use "+this+" as a Type.Class, which it is not"); }
+    public Type.Array  asArray()     { throw new RuntimeException("attempted to use "+this+" as a Type.Array, which it is not"); }
+    public boolean     isPrimitive() { return !isRef(); }
+    public boolean     isRef()       { return false; }
+    public boolean     isClass()     { return false; }
+    public boolean     isArray()     { return false; }
+
+    // Protected/Private //////////////////////////////////////////////////////////////////////////////
+
+    protected final String descriptor;
+    protected final String toString;
+    protected Type(String descriptor) { this(descriptor, descriptor); }
+    protected Type(String descriptor, String humanReadable) {
+        this.toString = humanReadable;
+        instances.put(this.descriptor = descriptor, this);
+    }
+
+    public static class Ref extends Type {
+        protected Ref(String descriptor) { super(descriptor); }
+        protected Ref(String descriptor, String humanReadable) { super(descriptor, humanReadable); }
+        public    Type.Ref asRef() { return this; }
+        public    boolean  isRef() { return true; }
+    }
+
+    public static class Class extends Type.Ref {
+        protected Class(String s) { super(_initHelper(s), _initHelper2(s)); }
+        public Type.Class asClass() { return this; }
+        public boolean isClass() { return true; }
+        public String getShortName() { return toString.substring(toString.lastIndexOf('.')+1); }
+        String internalForm() { return descriptor.substring(1, descriptor.length()-1); }
         private static String _initHelper(String s) {
-            if(!s.startsWith("L") || !s.endsWith(";")) s = "L" + s.replace('.','/') + ";";
-            if(!validDescriptorString(s)) throw new IllegalArgumentException("invalid descriptor string");
+            if (!s.startsWith("L") || !s.endsWith(";")) s = "L" + s.replace('.', '/') + ";";
             return s;
         }
-
-        public String[] components() {
-            StringTokenizer st = new StringTokenizer(descriptor.substring(1,descriptor.length()-1),"/");
+        private static String _initHelper2(String s) {
+            if (s.startsWith("L") && s.endsWith(";")) s = s.substring(1, s.length()-1);
+            return s.replace('/', '.');
+        }
+        String[] components() {
+            StringTokenizer st = new StringTokenizer(descriptor.substring(1, descriptor.length()-1), "/");
             String[] a = new String[st.countTokens()];
             for(int i=0;st.hasMoreTokens();i++) a[i] = st.nextToken();
             return a;
         }
-        
-        public String internalForm() { return descriptor.substring(1,descriptor.length()-1); }
-        
-        // FEATURE: Do a proper check here (invalid chars, etc)
-        static boolean validDescriptorString(String s) {
-            return s.startsWith("L") && s.endsWith(";");
+
+        public Method method(String name, Type returnType, Type[] argTypes) { return new Method(name, returnType, argTypes); }
+        public Field  field(String name, Type type) { return new Field(name, type); }
+
+        public abstract class Member {
+            public final String name;
+            private Member(String name) { this.name = name; }
+            public Type.Class getDeclaringClass() { return Type.Class.this; }
+            public abstract String getDescriptor();
+            public boolean equals(Object o_) {
+                if(!(o_ instanceof Member)) return false;
+                Member o = (Member) o_;
+                return o.getDeclaringClass().equals(getDeclaringClass()) &&
+                    o.name.equals(name) &&
+                    o.getDescriptor().equals(getDescriptor());
+            }
+            public int hashCode() { return getDeclaringClass().hashCode() ^ name.hashCode() ^ getDescriptor().hashCode(); }
+        }
+    
+        public class Field extends Member {
+            /** Create a reference to field <i>name</i> of class <i>c</i> with the type <i>t</i>  */    
+            public final Type type;
+            private Field(String name, Type t) { super(name); this.type = t; }
+            public String getDescriptor() { return name; }
+        }
+
+        public class Method extends Member {
+            final Type[] argTypes;
+            final Type   returnType;
+            private Method(String name, Type returnType, Type[] argTypes) {
+                super(name);
+                this.argTypes = argTypes;
+                this.returnType = returnType;
+            }
+            public String getDescriptor() {
+                StringBuffer sb = new StringBuffer(argTypes.length*4);
+                sb.append("(");
+                for(int i=0;i<argTypes.length;i++) sb.append(argTypes[i].getDescriptor());
+                sb.append(")");
+                sb.append(returnType.getDescriptor());
+                return sb.toString();
+            }
         }
+
     }    
+
+    public static class Array extends Type.Ref {
+        protected Array(Type t) { super("[" + t.getDescriptor(), t.toString() + "[]"); }
+        public Type.Array asArray() { return this; }
+        public boolean isArray() { return true; }
+        public int dimension() { return getDescriptor().lastIndexOf('['); }
+    }
+
 }