diff --git a/pom.xml b/pom.xml
index 997337b92..999505946 100644
--- a/pom.xml
+++ b/pom.xml
@@ -4,7 +4,7 @@
4.0.0
com.mysema.codegen
codegen
- 0.2.5-SNAPSHOT
+ 0.2.6
Codegen
Code generation and compilation for Java
diff --git a/src/main/java/com/mysema/codegen/ScalaWriter.java b/src/main/java/com/mysema/codegen/ScalaWriter.java
index 321cfeaa5..58e10de64 100644
--- a/src/main/java/com/mysema/codegen/ScalaWriter.java
+++ b/src/main/java/com/mysema/codegen/ScalaWriter.java
@@ -50,6 +50,8 @@ public class ScalaWriter extends AbstractCodeWriter{
private static final String PUBLIC = "public ";
private static final String PUBLIC_CLASS = "class ";
+
+ private static final String PUBLIC_OBJECT = "object ";
private static final String VAR = "var ";
@@ -155,8 +157,14 @@ public class ScalaWriter extends AbstractCodeWriter{
return this;
}
- public ScalaWriter beginClass(String header) throws IOException{
- line(PUBLIC_CLASS, header);
+ public ScalaWriter beginObject(String header) throws IOException {
+ line(PUBLIC_OBJECT, header, " {");
+ goIn();
+ return this;
+ }
+
+ public ScalaWriter beginClass(String header) throws IOException {
+ line(PUBLIC_CLASS, header, " {");
goIn();
return this;
}
@@ -308,7 +316,11 @@ public class ScalaWriter extends AbstractCodeWriter{
}
}
if (rv.endsWith("[]")){
- return "Array[" + rv.substring(0, rv.length()-2) + "]";
+ rv = rv.substring(0, rv.length()-2);
+ if (classes.contains(rv)){
+ rv = rv.substring(packageName.length()+1);
+ }
+ return "Array[" + rv + "]";
}else{
return rv;
}