4 files changed, 200 insertions, 0 deletions
diff --git a/sample/vector/Sample.java b/sample/vector/Sample.java
new file mode 100644
index 00000000..7a47aadf
--- /dev/null
+++ b/sample/vector/Sample.java
@@ -0,0 +1,14 @@
+package sample.vector;
+
+public class Sample extends java.util.Vector {
+    public void add(X e) {
+	super.addElement(e);
+    }
+
+    public X at(int i) {
+	return (X)super.elementAt(i);
+    }
+}
+
+class X {
+}
diff --git a/sample/vector/Sample2.java b/sample/vector/Sample2.java
new file mode 100644
index 00000000..dd5c965e
--- /dev/null
+++ b/sample/vector/Sample2.java
@@ -0,0 +1,13 @@
+package sample.vector;
+
+public class Sample2 extends java.util.Vector {
+    public Object add(Object[] args) {
+	super.addElement(args[0]);
+	return null;
+    }
+
+    public Object at(Object[] args) {
+	int i = ((Integer)args[0]).intValue();
+	return super.elementAt(i);
+    }
+}
diff --git a/sample/vector/Test.j b/sample/vector/Test.j
new file mode 100644
index 00000000..6f524c93
--- /dev/null
+++ b/sample/vector/Test.j
@@ -0,0 +1,38 @@
+/*
+  A sample program using sample.vector.VectorAssistant
+  and the javassist.preproc package.
+
+  This automatically produces the classes representing vectors of integer
+  and vectors of java.lang.String.
+
+  To compile and run this program, do as follows:
+
+    % java javassist.tool.Compiler sample/vector/Test.j
+    % javac sample/vector/Test.java
+    % java sample.vector.Test
+
+  The first line produces one source file (sample/Test.java) and
+  two class files (sample/vector/intVector.class and
+  sample/vector/StringVector.class).
+*/
+
+package sample.vector;
+
+import java.util.Vector by sample.vector.VectorAssistant(java.lang.String);
+import java.util.Vector by sample.vector.VectorAssistant(int);
+
+public class Test {
+    public static void main(String[] args) {
+	intVector iv = new intVector();
+	iv.add(3);
+	iv.add(4);
+	for (int i = 0; i < iv.size(); ++i)
+	    System.out.println(iv.at(i));
+
+	StringVector sv = new StringVector();
+	sv.add("foo");
+	sv.add("bar");
+	for (int i = 0; i < sv.size(); ++i)
+	    System.out.println(sv.at(i));
+    }
+}
diff --git a/sample/vector/VectorAssistant.java b/sample/vector/VectorAssistant.java
new file mode 100644
index 00000000..44fdd41c
--- /dev/null
+++ b/sample/vector/VectorAssistant.java
@@ -0,0 +1,135 @@
+package sample.vector;
+
+import java.io.IOException;
+import javassist.*;
+import javassist.preproc.Assistant;
+
+/**
+ * This is a Javassist program which produce a new class representing
+ * vectors of a given type.  For example,
+ *
+ * <ul>import java.util.Vector by sample.vector.VectorAssistant(int)</ul>
+ *
+ * <p>requests the Javassist preprocessor to substitute the following
+ * lines for the original import declaration:
+ *
+ * <ul><pre>
+ * import java.util.Vector;
+ * import sample.vector.intVector;
+ * </pre></ul>
+ *
+ * <p>The Javassist preprocessor calls <code>VectorAssistant.assist()</code>
+ * and produces class <code>intVector</code> equivalent to:
+ *
+ * <ul><pre>
+ * package sample.vector;
+ *
+ * public class intVector extends Vector {
+ *   pubilc void add(int value) {
+ *     addElement(new Integer(value));
+ *   }
+ *
+ *   public int at(int index) {
+ *     return elementAt(index).intValue();
+ *   }
+ * }
+ * </pre></ul>
+ *
+ * <p><code>VectorAssistant.assist()</code> uses
+ * <code>sample.vector.Sample</code> and <code>sample.vector.Sample2</code>
+ * as a template to produce the methods <code>add()</code> and
+ * <code>at()</code>.
+ */
+public class VectorAssistant implements Assistant {
+    public final String packageName = "sample.vector.";
+
+    /**
+     * Calls <code>makeSubclass()</code> and produces a new vector class.
+     * This method is called by a <code>javassist.preproc.Compiler</code>.
+     *
+     * @see javassist.preproc.Compiler
+     */
+    public CtClass[] assist(ClassPool pool, String vec, String[] args)
+	throws CannotCompileException
+    {
+	if (args.length != 1)
+	    throw new CannotCompileException(
+			"VectorAssistant receives a single argument.");
+
+	try {
+	    CtClass subclass;
+	    CtClass elementType = pool.get(args[0]);
+	    if (elementType.isPrimitive())
+		subclass = makeSubclass2(pool, elementType);
+	    else
+		subclass = makeSubclass(pool, elementType);
+
+	    CtClass[] results = { subclass, pool.get(vec) };
+	    return results;
+	}
+	catch (NotFoundException e) {
+	    throw new CannotCompileException(e);
+	}
+	catch (IOException e) {
+	    throw new CannotCompileException(e);
+	}
+    }
+
+    /**
+     * Produces a new vector class.  This method does not work if
+     * the element type is a primitive type.
+     *
+     * @param type	the type of elements
+     */
+    public CtClass makeSubclass(ClassPool pool, CtClass type)
+	throws CannotCompileException, NotFoundException, IOException
+    {
+	CtClass vec = pool.makeClass(makeClassName(type));
+	vec.setSuperclass(pool.get("java.util.Vector"));
+
+	CtClass c = pool.get("sample.vector.Sample");
+	CtMethod addmethod = c.getDeclaredMethod("add");
+	CtMethod atmethod = c.getDeclaredMethod("at");
+
+	ClassMap map = new ClassMap();
+	map.put("sample.vector.X", type.getName());
+
+	vec.addMethod(CtNewMethod.copy(addmethod, "add", vec, map));
+	vec.addMethod(CtNewMethod.copy(atmethod, "at", vec, map));
+	pool.writeFile(vec.getName());
+	return vec;
+    }
+
+    /**
+     * Produces a new vector class.  This uses wrapped methods so that
+     * the element type can be a primitive type.
+     *
+     * @param type	the type of elements
+     */
+    public CtClass makeSubclass2(ClassPool pool, CtClass type)
+	throws CannotCompileException, NotFoundException, IOException
+    {
+	CtClass vec = pool.makeClass(makeClassName(type));
+	vec.setSuperclass(pool.get("java.util.Vector"));
+
+	CtClass c = pool.get("sample.vector.Sample2");
+	CtMethod addmethod = c.getDeclaredMethod("add");
+	CtMethod atmethod = c.getDeclaredMethod("at");
+
+	CtClass[] args1 = { type };
+	CtClass[] args2 = { CtClass.intType };
+	CtMethod m
+	    = CtNewMethod.wrapped(CtClass.voidType, "add", args1,
+				  null, addmethod, null, vec);
+	vec.addMethod(m);
+	m = CtNewMethod.wrapped(type, "at", args2,
+				null, atmethod, null, vec);
+	vec.addMethod(m);
+	pool.writeFile(vec.getName());
+	return vec;
+    }
+
+    private String makeClassName(CtClass type) {
+	return packageName + type.getSimpleName() + "Vector";
+    }
+}