/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.activemq.openwire.tool;
import java.io.File;
import java.io.PrintWriter;
import java.util.Iterator;
import java.util.List;
import org.codehaus.jam.JClass;
import org.codehaus.jam.JProperty;
/**
*
*/
public class CppClassesGenerator extends MultiSourceGenerator {
protected String targetDir = "./src/main/cpp";
public Object run() {
filePostFix = getFilePostFix();
if (destDir == null) {
destDir = new File(targetDir + "/activemq/command");
}
return super.run();
}
protected String getFilePostFix() {
return ".cpp";
}
/**
* Converts the Java type to a C++ type name
*/
public String toCppType(JClass type) {
String name = type.getSimpleName();
if (name.equals("String")) {
return "p<string>";
} else if (type.isArrayType()) {
if (name.equals("byte[]")) {
name = "char[]";
} else if (name.equals("DataStructure[]")) {
name = "IDataStructure[]";
}
return "array<" + name.substring(0, name.length() - 2) + ">";
} else if (name.equals("Throwable") || name.equals("Exception")) {
return "p<BrokerError>";
} else if (name.equals("ByteSequence")) {
return "array<char>";
} else if (name.equals("boolean")) {
return "bool";
} else if (name.equals("long")) {
return "long long";
} else if (name.equals("byte")) {
return "char";
} else if (name.equals("Command") || name.equals("DataStructure")) {
return "p<I" + name + ">";
} else if (!type.isPrimitiveType()) {
return "p<" + name + ">";
} else {
return name;
}
}
/**
* Converts the Java type to a C++ default value
*/
public String toCppDefaultValue(JClass type) {
String name = type.getSimpleName();
if (name.equals("boolean")) {
return "false";
} else if (!type.isPrimitiveType()) {
return "NULL";
} else {
return "0";
}
}
/**
* Converts the Java type to the name of the C++ marshal method to be used
*/
public String toMarshalMethodName(JClass type) {
String name = type.getSimpleName();
if (name.equals("String")) {
return "marshalString";
} else if (type.isArrayType()) {
if (type.getArrayComponentType().isPrimitiveType() && name.equals("byte[]")) {
return "marshalByteArray";
} else {
return "marshalObjectArray";
}
} else if (name.equals("ByteSequence")) {
return "marshalByteArray";
} else if (name.equals("short")) {
return "marshalShort";
} else if (name.equals("int")) {
return "marshalInt";
} else if (name.equals("long")) {
return "marshalLong";
} else if (name.equals("byte")) {
return "marshalByte";
} else if (name.equals("double")) {
return "marshalDouble";
} else if (name.equals("float")) {
return "marshalFloat";
} else if (name.equals("boolean")) {
return "marshalBoolean";
} else if (!type.isPrimitiveType()) {
return "marshalObject";
} else {
return name;
}
}
/**
* Converts the Java type to the name of the C++ unmarshal method to be used
*/
public String toUnmarshalMethodName(JClass type) {
String name = type.getSimpleName();
if (name.equals("String")) {
return "unmarshalString";
} else if (type.isArrayType()) {
if (type.getArrayComponentType().isPrimitiveType() && name.equals("byte[]")) {
return "unmarshalByteArray";
} else {
return "unmarshalObjectArray";
}
} else if (name.equals("ByteSequence")) {
return "unmarshalByteArray";
} else if (name.equals("short")) {
return "unmarshalShort";
} else if (name.equals("int")) {
return "unmarshalInt";
} else if (name.equals("long")) {
return "unmarshalLong";
} else if (name.equals("byte")) {
return "unmarshalByte";
} else if (name.equals("double")) {
return "unmarshalDouble";
} else if (name.equals("float")) {
return "unmarshalFloat";
} else if (name.equals("boolean")) {
return "unmarshalBoolean";
} else if (!type.isPrimitiveType()) {
return "unmarshalObject";
} else {
return name;
}
}
/**
* Converts the Java type to a C++ pointer cast
*/
public String toUnmarshalCast(JClass type) {
String name = toCppType(type);
if (name.startsWith("p<")) {
return "p_cast<" + name.substring(2);
} else if (name.startsWith("array<") && (type.isArrayType() && !type.getArrayComponentType().isPrimitiveType()) && !type.getSimpleName().equals("ByteSequence")) {
return "array_cast<" + name.substring(6);
} else {
return "";
}
}
protected void generateLicence(PrintWriter out) {
out.println("/**");
out.println(" * Licensed to the Apache Software Foundation (ASF) under one or more");
out.println(" * contributor license agreements. See the NOTICE file distributed with");
out.println(" * this work for additional information regarding copyright ownership.");
out.println(" * The ASF licenses this file to You under the Apache License, Version 2.0");
out.println(" * (the \"License\"); you may not use this file except in compliance with");
out.println(" * the License. You may obtain a copy of the License at");
out.println(" *");
out.println(" * http://www.apache.org/licenses/LICENSE-2.0");
out.println(" *");
out.println(" * Unless required by applicable law or agreed to in writing, software");
out.println(" * distributed under the License is distributed on an \"AS IS\" BASIS,");
out.println(" * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.");
out.println(" * See the License for the specific language governing permissions and");
out.println(" * limitations under the License.");
out.println(" */");
}
protected void generateFile(PrintWriter out) throws Exception {
generateLicence(out);
out.println("#include \"activemq/command/" + className + ".hpp\"");
out.println("");
out.println("using namespace apache::activemq::command;");
out.println("");
out.println("/*");
out.println(" *");
out.println(" * Command and marshalling code for OpenWire format for " + className + "");
out.println(" *");
out.println(" *");
out.println(" * NOTE!: This file is autogenerated - do not modify!");
out.println(" * if you need to make a change, please see the Groovy scripts in the");
out.println(" * activemq-core module");
out.println(" *");
out.println(" */");
out.println("" + className + "::" + className + "()");
out.println("{");
List properties = getProperties();
for (Iterator iter = properties.iterator(); iter.hasNext();) {
JProperty property = (JProperty)iter.next();
String value = toCppDefaultValue(property.getType());
String propertyName = property.getSimpleName();
String parameterName = decapitalize(propertyName);
out.println(" this->" + parameterName + " = " + value + " ;");
}
out.println("}");
out.println("");
out.println("" + className + "::~" + className + "()");
out.println("{");
out.println("}");
out.println("");
out.println("unsigned char " + className + "::getDataStructureType()");
out.println("{");
out.println(" return " + className + "::TYPE ; ");
out.println("}");
for (Iterator iter = properties.iterator(); iter.hasNext();) {
JProperty property = (JProperty)iter.next();
String type = toCppType(property.getType());
String propertyName = property.getSimpleName();
String parameterName = decapitalize(propertyName);
out.println("");
out.println(" ");
out.println("" + type + " " + className + "::get" + propertyName + "()");
out.println("{");
out.println(" return " + parameterName + " ;");
out.println("}");
out.println("");
out.println("void " + className + "::set" + propertyName + "(" + type + " " + parameterName + ")");
out.println("{");
out.println(" this->" + parameterName + " = " + parameterName + " ;");
out.println("}");
}
out.println("");
out.println("int " + className + "::marshal(p<IMarshaller> marshaller, int mode, p<IOutputStream> ostream) throw (IOException)");
out.println("{");
out.println(" int size = 0 ;");
out.println("");
out.println(" size += " + baseClass + "::marshal(marshaller, mode, ostream) ; ");
for (Iterator iter = properties.iterator(); iter.hasNext();) {
JProperty property = (JProperty)iter.next();
String marshalMethod = toMarshalMethodName(property.getType());
String propertyName = decapitalize(property.getSimpleName());
out.println(" size += marshaller->" + marshalMethod + "(" + propertyName + ", mode, ostream) ; ");
}
out.println(" return size ;");
out.println("}");
out.println("");
out.println("void " + className + "::unmarshal(p<IMarshaller> marshaller, int mode, p<IInputStream> istream) throw (IOException)");
out.println("{");
out.println(" " + baseClass + "::unmarshal(marshaller, mode, istream) ; ");
for (Iterator iter = properties.iterator(); iter.hasNext();) {
JProperty property = (JProperty)iter.next();
String cast = toUnmarshalCast(property.getType());
String unmarshalMethod = toUnmarshalMethodName(property.getType());
String propertyName = decapitalize(property.getSimpleName());
out.println(" " + propertyName + " = " + cast + "(marshaller->" + unmarshalMethod + "(mode, istream)) ; ");
}
out.println("}");
}
public String getTargetDir() {
return targetDir;
}
public void setTargetDir(String targetDir) {
this.targetDir = targetDir;
}
}