trunk/activemq-openwire-generator/src/main/java/org/apache/activemq/openwire/tool/CppClassesGenerator.java

/**
 * 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;
    }

}