X-Git-Url: http://git.megacz.com/?p=org.ibex.classgen.git;a=blobdiff_plain;f=src%2Forg%2Fibex%2Fclassgen%2FType.java;h=29301d38b5dddea89e09e6927b8b3279e555c3a1;hp=33af24d0d2d7b1922839a0da124ad2d10b2f3f87;hb=57f8032e5579f575ef526b02957fc937f6efbb62;hpb=eaa26b83ad7f83df63910284014f84657f0e798f

diff --git a/src/org/ibex/classgen/Type.java b/src/org/ibex/classgen/Type.java
index 33af24d..29301d3 100644
--- a/src/org/ibex/classgen/Type.java
+++ b/src/org/ibex/classgen/Type.java
@@ -9,6 +9,9 @@ public class Type {
     public static final Type BOOLEAN = new Type("Z");
     public static final Type DOUBLE = new Type("D");
     public static final Type FLOAT = new Type("F");
+    public static final Type BYTE = new Type("B");
+    public static final Type CHAR = new Type("C");
+    public static final Type SHORT = new Type("S");
     
     public static final Type.Object OBJECT = new Type.Object("java.lang.Object");
     public static final Type.Object STRING = new Type.Object("java.lang.String");
@@ -17,45 +20,92 @@ public class Type {
     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");
     
+    /** 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; }
+    protected Type(String descriptor) { this.descriptor = descriptor; }
     
+    public static Type fromDescriptor(String descriptor) {
+        if(descriptor.equals("V")) return VOID;
+        if(descriptor.equals("I")) return INT;
+        if(descriptor.equals("J")) return LONG;
+        if(descriptor.equals("Z")) return BOOLEAN;
+        if(descriptor.equals("D")) return DOUBLE;
+        if(descriptor.equals("F")) return FLOAT;
+        if(descriptor.equals("B")) return BYTE;
+        if(descriptor.equals("C")) return CHAR;
+        if(descriptor.equals("S")) return SHORT;
+        if(descriptor.endsWith("[")) return new Type.Array(fromDescriptor(descriptor.substring(0, descriptor.indexOf('['))),
+                                                           descriptor.length() - descriptor.indexOf('['));
+        if(Type.Object.validDescriptorString(descriptor)) return new Type.Object(descriptor);
+        return null;
+    }
+        
+    /** Returns the Java descriptor string for this object ("I", or "Ljava/lang/String", "[[J", etc */
     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());
-    }
-    
+    /** Returns a one dimensional array type for the base type <i>base</i>
+        @param base The base type
+        @return A one dimensional array of the base type
+    */
+    public static Type arrayType(Type base) { return arrayType(base, 1); }
+    /** Returns a <i>dim</i> dimensional array type for the base type <i>base</i>
+        @param base The base type
+        @param dim Number if dimensions
+        @return A one dimensional array of the base type
+    */
+    public static Type arrayType(Type base, int dim) { return new Type.Array(base, dim); }
+
+    public Type.Object asObject() { throw new RuntimeException("attempted to use "+this+" as a Type.Object, 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 isObject() { return false; }
+    public boolean isArray() { return false; }
+
+    /** Class representing Object types (any non-primitive type) */
     public static class Object extends Type {
-        public Object(String s) { super(_initHelper(s)); }
-        
+        /** Create an Type.Object instance for the specified string. <i>s</i> can be a string in the form
+            "java.lang.String", "java/lang/String", or "Ljava/lang/String;".
+            @param s The type */
+        protected Object(String s) { super(_initHelper(s)); }
+        public Type.Object asObject() { return this; }
+        public boolean isObject() { return true; }
+
         private static String _initHelper(String s) {
-            if(!s.startsWith("L") || !s.endsWith(";")) s = "L" + s.replace('.','/') + ";";
+            if(!s.startsWith("L") || !s.endsWith(";")) s = "L" + s.replace('.', '/') + ";";
             if(!validDescriptorString(s)) throw new IllegalArgumentException("invalid descriptor string");
             return s;
         }
 
-        public String[] components() {
-            StringTokenizer st = new StringTokenizer(descriptor.substring(1,descriptor.length()-1),"/");
+        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); }
+        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 static class Array extends Object {
+        protected Array(Type t, int dim) {  super(_initHelper(t, dim)); }
+        public Type.Array asArray() { return this; }
+        public boolean isArray() { return true; }
+        String internalForm() { throw new Error("Type.Array does not have an internalForm()"); }
+        String[] components() { throw new Error("Type.Array does not have components()"); }
+        private static String _initHelper(Type t, int dim) {
+            StringBuffer sb = new StringBuffer(t.descriptor.length() + dim);
+            for(int i=0;i<dim;i++) sb.append("[");
+            sb.append(t.descriptor);
+            return sb.toString();
+        }
+    }
+
 }