refactored functionality out of FieldGen into Type.Class.Field

[org.ibex.classgen.git] / src / org / ibex / classgen / FieldGen.java

diff --git a/src/org/ibex/classgen/FieldGen.java b/src/org/ibex/classgen/FieldGen.java
index 805a727..7f5242c 100644 (file)
--- a/src/org/ibex/classgen/FieldGen.java
+++ b/src/org/ibex/classgen/FieldGen.java
@@ -2,8 +2,53 @@ package org.ibex.classgen;
 
 import java.io.*;
 
-public class FieldGen {
-    public void dump(DataOutput o) {
-        throw new Error("FIXME");
+/** Class representing a field in a generated classfile
+    @see ClassFile#addField */
+public class FieldGen extends Type.Class.Field.Body {
+    private final ClassFile.AttrGen attrs;
+
+    StringBuffer debugToString(StringBuffer sb) {
+        sb.append(ClassFile.flagsToString(flags, false));
+        sb.append(getField().getType().debugToString());
+        sb.append(" ");
+        sb.append(getField().getName());
+        if(attrs.contains("ConstantValue"))
+            sb.append(" = \"").append(attrs.get("ConstantValue")).append("\"");
+        sb.append(";");
+        return sb;
+    }
+    
+    FieldGen(ClassFile cf, DataInput in, ConstantPool cp) throws IOException {
+        this(in.readShort(),
+             cf.getType().field(cp.getUtf8KeyByIndex(in.readShort()),
+                                Type.fromDescriptor(cp.getUtf8KeyByIndex(in.readShort()))));
+    }
+
+    private FieldGen(int flags, Type.Class.Field field) { this(field, flags); }
+    FieldGen(Type.Class.Field field, int flags) { this(field, flags, new ClassFile.AttrGen()); }
+    FieldGen(Type.Class.Field field, int flags, ClassFile.AttrGen attrs) {
+        field.super(flags);
+        this.attrs = attrs;
+   }
+    
+    /** Sets the ContantValue attribute for this field. 
+        @param val The value to set this field to. Must be an Integer, Long, Float, Double, or String */
+    public void setConstantValue(Object val) {
+        if (!isStatic()) throw new IllegalStateException("field does not have the STATIC bit set");
+        attrs.put("ConstantValue",val);
+    }
+    
+    void finish(ConstantPool cp) {
+        cp.addUtf8(getField().getName());
+        cp.addUtf8(getField().getType().getDescriptor());
+        
+        attrs.finish(cp);
+    }
+    
+    void dump(DataOutput o, ConstantPool cp) throws IOException {
+        o.writeShort(getFlags());
+        o.writeShort(cp.getUtf8Index(getField().getName()));
+        o.writeShort(cp.getUtf8Index(getField().getType().getDescriptor()));
+        attrs.dump(o,cp);
     }
 }