nicer output
[libt2n] / codegen / main.cpp
index ec97403..fe400b2 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
@@ -98,6 +122,7 @@ struct type_info
      std::string noref_name;
      bool operator==(const type_info& o) {return (name==o.name) && (noref_name == o.noref_name);}
      std::string noref() const {return noref_name.empty() ? name : noref_name;}
+     bool isVoid() const {return name=="void";}
 };
 
 std::ostream &operator<<(std::ostream &o, const type_info &t) {
@@ -145,8 +170,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"));
@@ -178,6 +210,7 @@ struct t2n_procedure
      std::string cmd_classname() const {
          return name+mangled+"_cmd";
      }
+     bool hasReturn() const {return !ret_type.isVoid();}
 };
 
 std::ostream &operator<<(std::ostream &o, const t2n_procedure::Args &args) {
@@ -193,6 +226,20 @@ std::ostream &operator<<(std::ostream &o, const t2n_procedure &f) {
      return o;
 }
 
+std::pair<std::string, unsigned>
+get_file_and_line(const xmlpp::Element* root, const xmlpp::Element* element)
+{
+  return std::pair<std::string, unsigned>(get_file(root, element->get_attribute("file")->get_value()),
+                                         boost::lexical_cast<unsigned>(element->get_attribute("line")->get_value())-1);
+}
+
+std::string
+get_file_and_line_as_string(const xmlpp::Element* root, const xmlpp::Element* element)
+{
+  std::pair<std::string, unsigned> fl(get_file_and_line(root,element));
+  return std::string(fl.first)+":"+boost::lexical_cast<std::string>(fl.second);
+}
+
 class Parser
 {
 public:
@@ -231,7 +278,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());
@@ -251,13 +298,14 @@ protected:
                    if (f.args.back().second.name.empty()) {
                         assert(element->get_attribute("file"));
                         assert(element->get_attribute("line"));
-                        throw parse_error(get_file(root, element->get_attribute("file")->get_value()),
-                                          boost::lexical_cast<unsigned>(element->get_attribute("line")->get_value())-1,
+                        std::pair<std::string, unsigned> file_and_line(get_file_and_line(root, element));
+                        throw parse_error(file_and_line.first,
+                                          file_and_line.second,
                                           std::string("type of parameter `")+f.args.back().first+"' not (yet?) supported");
                    }
               }
          }
-         std::cerr << "Found function: " << f << std::endl;
+         std::cerr << get_file_and_line_as_string(root, element) << ": found procedure: " << f << std::endl;
          m_procedures.push_back(f);
      }
 
@@ -291,7 +339,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"
@@ -303,23 +351,27 @@ void output_common_hpp(std::ostream &o, const std::string &group, const std::lis
     for (std::list<t2n_procedure>::const_iterator it=procs.begin();it!=procs.end();++it) {
        o << "class " << it->ret_classname() << " : public libt2n::result\n"
          << "{\n"
-         << "private:\n"
-         << "  " << it->ret_type << " res;\n"
-         << "  friend class boost::serialization::access;\n"
+         << "private:\n";
+       if (it->hasReturn())
+         o << "        " << it->ret_type << " res;\n";
+       o << "  friend class boost::serialization::access;\n"
          << "  template<class Archive>\n"
          << "  void serialize(Archive & ar, const unsigned int /* version */)\n"
          << "  {\n"
-         << "          ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(libt2n::result);\n"
-         << "          ar & BOOST_SERIALIZATION_NVP(res);\n"
-         << "  }\n"
+         << "          ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(libt2n::result);\n";
+       if (it->hasReturn())
+         o << "                ar & BOOST_SERIALIZATION_NVP(res);\n";
+       o << "  }\n"
          << "public:\n"
-         << "  " << it->ret_classname() << "() {}\n"
-         << "  " << it->ret_classname() << "(const " << it->ret_type << " &_res) : res(_res) {}\n"
-         << "  " << it->ret_type << " get_data() { return res; }\n"
-         << "};\n";
+         << "  " << it->ret_classname() << "() {}\n";
+       if (it->hasReturn()) {
+         o << "        " << it->ret_classname() << "(const " << it->ret_type << " &_res) : res(_res) {}\n"
+           << "        " << it->ret_type << " get_data() { return res; }\n";
+       }
+       o << "};\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 +381,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";
        }
@@ -341,18 +393,21 @@ void output_common_hpp(std::ostream &o, const std::string &group, const std::lis
          << "  " << it->cmd_classname() << "() {}\n";
        
        // constructor taking all arguments
-       o << "  " << it->cmd_classname() << "(";
-       for (t2n_procedure::Args::const_iterator ait=it->args.begin();ait!=it->args.end();++ait) {
+       if (!it->args.empty()) {
+         o << "        " << it->cmd_classname() << "(";
+         for (t2n_procedure::Args::const_iterator ait=it->args.begin();ait!=it->args.end();++ait) {
            if (ait!=it->args.begin()) o << ", ";
            o << ait->second << " _" << ait->first;
-       }
-       o << ") : ";
-       for (t2n_procedure::Args::const_iterator ait=it->args.begin();ait!=it->args.end();++ait) {
+         }
+         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";
        }
-       o << " {}\n"
-         << "  libt2n::result* operator()();\n"
+       o << "  libt2n::result* operator()();\n"
          << "};\n";
     }
 }
@@ -362,7 +417,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 +427,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 +450,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() << "(";
@@ -405,9 +460,10 @@ void output_client_cpp(std::ostream &o, const std::string &group, const std::lis
         }
         o << "), rc);\n"
           << " " << pit->ret_classname() << "* res=dynamic_cast<" << pit->ret_classname() << "*>(rc.get_result());\n"
-          << " if (!res) throw libt2n::t2n_communication_error(\"result object of wrong type\");\n"
-          << " return res->get_data();\n"
-          << "}\n";
+          << " if (!res) throw libt2n::t2n_communication_error(\"result object of wrong type\");\n";
+        if (pit->hasReturn())
+          o << "       return res->get_data();\n";
+        o << "}\n";
      }
 
      // include in this compilation unit to ensure the compilation unit is used
@@ -429,12 +485,21 @@ void output_server_cpp(std::ostream &o, const std::string &group, const std::lis
 
      for (std::list<t2n_procedure>::const_iterator it=procs.begin();it!=procs.end();++it) {
          o << *it << ";\n";
-         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 << ", ";
-              o << ait->first;
+         o << "libt2n::result* " << it->cmd_classname() << "::operator()() { ";
+         if (it->hasReturn()) {
+           o << "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 << "));";
+         }else{
+           o << it->name << "(); return new " << it->ret_classname() << "();";
          }
-         o << ")); }\n";
+         o << " }\n";
      }
      o << "#include \"" << common_cpp << "\"\n";
 }
@@ -444,7 +509,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 +529,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)