added support for pointer to const
[libt2n] / codegen / main.cpp
index ec97403..b1a0ce1 100644 (file)
@@ -1,3 +1,22 @@
+/*
+    Copyright (C) 2006                                                    
+    intra2net.com                                                         
+
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 2 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program; if not, write to the Free Software
+    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+*/
+
 #include <libxml++/libxml++.h>
 #include <cassert>
 #include <iostream>
@@ -6,6 +25,16 @@
 #include <list>
 #include <stdexcept>
 #include <boost/lexical_cast.hpp>
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+
+//! map group to class name
+std::string
+groupClass(const std::string &group) {
+    return std::string("cmd_group_")+group;
+}
 
 //! convert string to upper case
 std::string
@@ -77,19 +106,14 @@ std::string get_namespace(const xmlpp::Element* root, const std::string &id)
      return element->get_attribute("name")->get_value();
 }
 
-//! extract group from attributes
-std::string
-extract_group(const std::string &attrs)
+//! procedure marked for export?
+bool
+is_marked(const std::string &attrs)
 {
      // todo: improve this
-     std::string error;
      std::string to_match("gccxml(libt2n-");
      std::string::size_type p(attrs.find(to_match));
-     if (p==std::string::npos) return error;
-     std::string group(attrs, p+to_match.length(), attrs.length());
-     p=group.find_first_of(')');
-     assert(p!=std::string::npos);
-     return std::string(group,0,p);
+     return (p!=std::string::npos);
 }
 
 struct type_info
@@ -145,8 +169,15 @@ type_info get_type(const xmlpp::Element* root, const std::string &id)
          ret.name=std::string("const ")+ret.name;
          return ret;
      }else if (tag=="PointerType") {
-         // not yet supported
-         return error;
+          // todo: nearly the same as reference type handling
+         assert(element->get_attribute("type"));
+         type_info ret(get_type(root, element->get_attribute("type")->get_value()));
+         if (ret==error) return error;
+         // at the moment we only support const &
+         // todo: nice error message!
+         if ((ret.noref_name=strip(ret.name,"const ")).empty()) return error;
+         ret.name=ret.name+"*";
+         return ret;
      }
 
      assert(element->get_attribute("name"));
@@ -231,7 +262,7 @@ protected:
          // attributes are speparated by spaces?
 
          t2n_procedure f;
-         if (extract_group(attributes->get_value()).empty()) return;
+         if (!is_marked(attributes->get_value())) return;
 
          // we need the return type
          f.ret_type=get_type(root, returns->get_value());
@@ -291,7 +322,7 @@ protected:
 };
 
 void output_common_hpp(std::ostream &o, const std::string &group, const std::list<t2n_procedure> &procs) {
-    o << "class cmd_group_" << group << " : public libt2n::command\n"
+    o << "class " << groupClass(group) << " : public libt2n::command\n"
       << "{\n"
       << "private:\n"
       << "     friend class boost::serialization::access;\n"
@@ -319,7 +350,7 @@ void output_common_hpp(std::ostream &o, const std::string &group, const std::lis
          << "};\n";
     }
     for (std::list<t2n_procedure>::const_iterator it=procs.begin();it!=procs.end();++it) {
-       o << "class " << it->cmd_classname() << " : public " << "cmd_group_" << group << "\n"
+       o << "class " << it->cmd_classname() << " : public " << groupClass(group) << "\n"
          << "{\n"
          << "private:\n";
        for (t2n_procedure::Args::const_iterator ait=it->args.begin();ait!=it->args.end();++ait) {
@@ -329,7 +360,7 @@ void output_common_hpp(std::ostream &o, const std::string &group, const std::lis
          << "  template<class Archive>\n"
          << "  void serialize(Archive & ar, const unsigned int /* version */)\n"
          << "  {\n"
-         << "          ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(cmd_group_" << group << ");\n";
+         << "          ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(" << groupClass(group) << ");\n";
        for (t2n_procedure::Args::const_iterator ait=it->args.begin();ait!=it->args.end();++ait) {
            o << "              ar & BOOST_SERIALIZATION_NVP(" << ait->first << ");\n";
        }
@@ -349,7 +380,8 @@ void output_common_hpp(std::ostream &o, const std::string &group, const std::lis
        o << ") : ";
        for (t2n_procedure::Args::const_iterator ait=it->args.begin();ait!=it->args.end();++ait) {
            if (ait!=it->args.begin()) o << ", ";
-           o << ait->first << "(_" << ait->first << ")";
+           // pointers are const pointers and must be dereferenced
+           o << ait->first << "(" << ((ait->second.name.find_first_of('*')!=std::string::npos) ? "*" : "" ) << "_" << ait->first << ")";
        }
        o << " {}\n"
          << "  libt2n::result* operator()();\n"
@@ -362,7 +394,7 @@ void output_common_cpp(std::ostream &o, const std::string &group, const std::lis
        << "#include <boost/serialization/export.hpp>\n"
        << "\n"
        << "/* register types with boost serialization */\n";
-     o << "BOOST_CLASS_EXPORT(cmd_group_" << group << ")\n";
+     o << "BOOST_CLASS_EXPORT(" << groupClass(group) << ")\n";
      for (std::list<t2n_procedure>::const_iterator it=procs.begin();it!=procs.end();++it) {
          o << "BOOST_CLASS_EXPORT("<<it->ret_classname()<<")\n"
            << "BOOST_CLASS_EXPORT("<<it->cmd_classname()<<")\n";
@@ -372,10 +404,10 @@ void output_common_cpp(std::ostream &o, const std::string &group, const std::lis
 void output_client_hpp(std::ostream &o, const std::string &group, const std::list<t2n_procedure> &procs) {
      o << "#include <command_client.hxx>\n";
 
-     o << "class cmd_group_" << group << "_client : public libt2n::command_client\n"
+     o << "class " << groupClass(group) << "_client : public libt2n::command_client\n"
        << "{\n"
        << "public:\n"
-       << "cmd_group_" << group << "_client(libt2n::client_connection &_c,\n"
+       << groupClass(group) << "_client(libt2n::client_connection &_c,\n"
        << "    long long _command_timeout_usec=command_timeout_usec_default,\n"
        << "    long long _hello_timeout_usec=hello_timeout_usec_default)\n"
        << "    : libt2n::command_client(_c,_command_timeout_usec,_hello_timeout_usec)\n"
@@ -395,7 +427,7 @@ void output_client_cpp(std::ostream &o, const std::string &group, const std::lis
      }
 
      for (std::list<t2n_procedure>::const_iterator pit=procs.begin();pit!=procs.end();++pit) {
-        o << pit->ret_type << " cmd_group_" << group << "_client::" << pit->name << "(" << pit->args << ")\n"
+        o << pit->ret_type << " " << groupClass(group) << "_client::" << pit->name << "(" << pit->args << ")\n"
           << "{\n"
           << " libt2n::result_container rc;\n"
           << " send_command(new " << pit->cmd_classname() << "(";
@@ -432,6 +464,9 @@ void output_server_cpp(std::ostream &o, const std::string &group, const std::lis
          o << "libt2n::result* " << it->cmd_classname() << "::operator()() { return new " << it->ret_classname() << "(" << it->name << "(";
          for (t2n_procedure::Args::const_iterator ait=it->args.begin();ait!=it->args.end();++ait) {
               if (ait!=it->args.begin()) o << ", ";
+              // get pointer
+              if (ait->second.name.find_first_of('*')!=std::string::npos)
+                o << '&';
               o << ait->first;
          }
          o << ")); }\n";
@@ -444,7 +479,7 @@ struct header_file : public std::ofstream
      header_file(const char* fname) : std::ofstream(fname) {
          std::cerr << "create header: '" << fname << "'" << std::endl;
          std::string macro(replace(toupper(fname),'.','_'));
-         *this << "// automatically generated code - do not edit\n" << std::endl;
+         *this << "// automatically generated code (generated by libt2n-codegen " << VERSION << ") - do not edit\n" << std::endl;
          *this << "#ifndef " << macro << "\n"
                << "#define " << macro << "\n";
      }
@@ -464,12 +499,18 @@ struct cpp_file : public std::ofstream
 int
 main(int argc, char* argv[])
 {
-     if (argc < 3) {
-         std::cerr << "Usage: " << argv[0] << "default-group gccxml-file1 gccxml-file2 ... " << std::endl;
-         return 1;
-     }
+    // todo: maybe use getopt
+    if ((argc>1)&&(std::string(argv[1])=="--version")) {
+       std::cerr << VERSION << std::endl;
+       return 0;
+    }
+    if (argc < 3)
+    {
+       std::cerr << "Usage: " << argv[0] << "default-group gccxml-file1 gccxml-file2 ... " << std::endl;
+       return 1;
+    }
 
-     try{
+    try{
          std::string group(argv[1]);
          std::list<std::string> xmlfiles;
          for (int i=2;i<argc;++i)