start of nicer error handling
[libt2n] / codegen / main.cpp
index 0c60908..a8580ba 100644 (file)
@@ -4,25 +4,27 @@
 #include <set>
 #include <fstream>
 #include <list>
+#include <stdexcept>
+#include <boost/lexical_cast.hpp>
 
-void print_indentation(unsigned int indentation)
-{
-     for(unsigned int i = 0; i < indentation; ++i)
-         std::cout << " ";
-}
-
+//! convert string to upper case
 std::string
 toupper(std::string s) {
      for (unsigned i=0; i<s.length(); ++i) s[i]=toupper(s[i]);
      return s;
 }
 
+//! replace all characters f by r in string s
 std::string
 replace(std::string s, char f, char r) {
      for (unsigned i=0; i<s.length(); ++i) if (s[i]==f) s[i]=r;
      return s;
 }
 
+//! strip prefix from string s
+/*!
+  \return string s without prefix or an empty string on error
+ */
 std::string
 strip(std::string s, std::string prefix)
 {
@@ -31,6 +33,7 @@ strip(std::string s, std::string prefix)
      return std::string(s, prefix.length(), s.length()-prefix.length());
 }
 
+//! extract group from attributes
 std::string
 extract_group(const std::string &attrs)
 {
@@ -68,6 +71,15 @@ const xmlpp::Element* get_element_by_id(const xmlpp::Element* element, const std
      return NULL;
 }
 
+std::string
+get_file(const xmlpp::Element* root, const std::string &file_id)
+{
+     std::string error;
+     const xmlpp::Element* e=get_element_by_id(root, file_id);
+     if ((!e)||(!e->get_attribute("name"))) return error;
+     return e->get_attribute("name")->get_value();
+}
+
 //! get namespace by id
 /*!
   \return namespace name or empty string on error
@@ -80,6 +92,27 @@ std::string get_namespace(const xmlpp::Element* root, const std::string &id)
      return element->get_attribute("name")->get_value();
 }
 
+//! get default group
+std::string get_default_group(const xmlpp::Element* root) {
+     std::string error;
+
+     if (   (root->get_attribute("name"))
+        && (root->get_attribute("attributes"))
+        && (root->get_name() == "Typedef")
+        && (root->get_attribute("name")->get_value()=="libt2n_default_group"))
+         return extract_group(root->get_attribute("attributes")->get_value());
+
+     //Recurse through child nodes:
+     xmlpp::Node::NodeList list = root->get_children();
+     for(xmlpp::Node::NodeList::iterator iter = list.begin(); iter != list.end(); ++iter)
+     {
+         if (const xmlpp::Element* elem=dynamic_cast<const xmlpp::Element*>(*iter)) {
+              std::string ret(get_default_group(elem));
+              if (!ret.empty()) return ret;
+         }
+     }
+     return error;
+}
 
 struct type_info
 {
@@ -94,6 +127,13 @@ std::ostream &operator<<(std::ostream &o, const type_info &t) {
      return o;
 }
 
+struct parse_error : public std::runtime_error
+{
+     parse_error(const std::string &file, unsigned line, const std::string &msg)
+         : std::runtime_error(file+":"+boost::lexical_cast<std::string>(line)+": error: "+msg)
+         {}
+};
+
 //! get type by id
 /*!
   \return type name or empty string on error
@@ -126,12 +166,21 @@ type_info get_type(const xmlpp::Element* root, const std::string &id)
          if (ret==error) return error;
          ret.name=std::string("const ")+ret.name;
          return ret;
+     }else if (tag=="PointerType") {
+         // not yet supported
+         return error;
      }
 
      assert(element->get_attribute("name"));
      type_info ret;
-     if (element->get_attribute("context"))
-         ret.name=get_namespace(root, element->get_attribute("context")->get_value())+"::";
+     if (element->get_attribute("context")) {
+         ret.name=get_namespace(root, element->get_attribute("context")->get_value());
+         if (ret.name!="::")
+              ret.name+="::";
+         else
+              // do not explicitely add ::
+              ret.name="";
+     }
      ret.name+=element->get_attribute("name")->get_value();
      return ret;
 }
@@ -143,13 +192,14 @@ struct t2n_procedure
      std::string group;
      type_info ret_type;
      std::string name;
+     std::string mangled;
      Args  args;
 
      std::string ret_classname() const {
-         return name+"_res";
+         return name+mangled+"_res";
      }
      std::string cmd_classname() const {
-         return name+"_cmd";
+         return name+mangled+"_cmd";
      }
 };
 
@@ -196,8 +246,9 @@ protected:
 
          const xmlpp::Attribute* attributes = element->get_attribute("attributes");
          const xmlpp::Attribute* name = element->get_attribute("name");
+         const xmlpp::Attribute* mangled = element->get_attribute("mangled");
          const xmlpp::Attribute* returns = element->get_attribute("returns");
-         if ((!attributes)||(!name)||(!returns)) return;
+         if ((!attributes)||(!name)||(!mangled)||(!returns)) return;
 
          // check wether the procedure is marked (TODO: improve)
          // attributes are speparated by spaces?
@@ -206,10 +257,12 @@ protected:
          f.group=extract_group(attributes->get_value());
          if (f.group.empty()) return;
          // todo: handle default group
+         if (f.group=="default") f.group=get_default_group(root);
 
          // we need the return type
          f.ret_type=get_type(root, returns->get_value());
          f.name=name->get_value();
+         f.mangled=mangled->get_value();
 
          xmlpp::Node::NodeList list = node->get_children("Argument");
          for(xmlpp::Node::NodeList::iterator iter = list.begin(); iter != list.end(); ++iter)
@@ -220,6 +273,14 @@ protected:
                    assert(arg->get_attribute("name"));
                    assert(arg->get_attribute("type"));
                    f.args.push_back(std::pair<std::string, type_info>(arg->get_attribute("name")->get_value(), get_type(root, arg->get_attribute("type")->get_value())));
+                   // todo: ugly - could be any other error
+                   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::string("type of parameter `")+f.args.back().first+"' not (yet?) supported");
+                   }
               }
          }
          std::cerr << "Found function: " << f << std::endl;
@@ -241,61 +302,8 @@ protected:
 
               if(!nodeText && !nodeComment && !nodename.empty()) //Let's not say "name: text".
               {
-                    //   print_indentation(indentation);
-                   //    std::cout << "Node name = " << node->get_name() << std::endl;
-                   //    std::cout << "Node name = " << nodename << std::endl;
-                   if (node->get_name() == "Function") {
-                        parse_function(root, node);
-                   }
-              }
-#if 0
-              else if(nodeText) //Let's say when it's text. - e.g. let's say what that white space is.
-              {
-                   print_indentation(indentation);
-                   std::cout << "Text Node" << std::endl;
-              }
-
-              //Treat the various node types differently: 
-              if(nodeText)
-              {
-                   print_indentation(indentation);
-                   std::cout << "text = \"" << nodeText->get_content() << "\"" << std::endl;
-              }
-              else if(nodeComment)
-              {
-                   print_indentation(indentation);
-                   std::cout << "comment = " << nodeComment->get_content() << std::endl;
+                   if (node->get_name() == "Function") parse_function(root, node);
               }
-              else if(nodeContent)
-              {
-                   print_indentation(indentation);
-                   std::cout << "content = " << nodeContent->get_content() << std::endl;
-              }
-              else if(const xmlpp::Element* nodeElement = dynamic_cast<const xmlpp::Element*>(node))
-              {
-                   //A normal Element node:
-
-                   //line() works only for ElementNodes.
-                   print_indentation(indentation);
-                   std::cout << "     line = " << node->get_line() << std::endl;
-
-                   //Print attributes:
-                   const xmlpp::Element::AttributeList& attributes = nodeElement->get_attributes();
-                   for(xmlpp::Element::AttributeList::const_iterator iter = attributes.begin(); iter != attributes.end(); ++iter)
-                   {
-                        const xmlpp::Attribute* attribute = *iter;
-                        print_indentation(indentation);
-                        std::cout << "  Attribute " << attribute->get_name() << " = " << attribute->get_value() << std::endl;
-                   }
-
-                   const xmlpp::Attribute* attribute = nodeElement->get_attribute("title");
-                   if(attribute)
-                   {
-                        std::cout << "title found: =" << attribute->get_value() << std::endl;
-      
-                   }
-              }
-#endif
               if(!nodeContent)
               {
                    //Recurse through child nodes:
@@ -322,14 +330,13 @@ used_groups(const std::list<t2n_procedure> &funcs) {
 void output_common_hpp(std::ostream &o, const std::list<t2n_procedure> &procs) {
      std::set<std::string> groups(used_groups(procs));
 
-     o << "#include \"codegen-stubhead.hxx\"\n";
      for (std::set<std::string>::iterator it=groups.begin();it!=groups.end();++it) {
          o << "class cmd_group_" << *it << " : public libt2n::command\n"
            << "{\n"
            << "private:\n"
            << "        friend class boost::serialization::access;\n"
            << "        template<class Archive>\n"
-           << "        void serialize(Archive & ar, const unsigned int version)\n"
+           << "        void serialize(Archive & ar, const unsigned int /* version */)\n"
            << "        {ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(libt2n::command);}\n"
            << "};\n";
      }
@@ -341,7 +348,7 @@ void output_common_hpp(std::ostream &o, const std::list<t2n_procedure> &procs) {
            << "        " << it->ret_type << " res;\n"
            << "        friend class boost::serialization::access;\n"
            << "        template<class Archive>\n"
-           << "        void serialize(Archive & ar, const unsigned int version)\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"
@@ -361,7 +368,7 @@ void output_common_hpp(std::ostream &o, const std::list<t2n_procedure> &procs) {
          }
          o << "        friend class boost::serialization::access;\n"
            << "        template<class Archive>\n"
-           << "        void serialize(Archive & ar, const unsigned int version)\n"
+           << "        void serialize(Archive & ar, const unsigned int /* version */)\n"
            << "        {\n"
            << "                ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(cmd_group_" << it->group << ");\n";
          for (t2n_procedure::Args::const_iterator ait=it->args.begin();ait!=it->args.end();++ait) {
@@ -503,47 +510,100 @@ struct cpp_file : public std::ofstream
      }
 };
 
-int
-main(int argc, char* argv[])
+std::list<std::string>
+get_includes(const std::string &fname)
 {
-     if (argc != 3) {
-         std::cerr << "Usage: " << argv[0] << " gccxml-file outputprefix" << std::endl;
-         return 1;
+     // grep "#include" fname
+     std::ifstream in(fname.c_str());
+     std::string line;
+     std::list<std::string> ret;
+     while (std::getline(in,line)) {
+         if (line.find("#include")!=std::string::npos)
+              ret.push_back(line);
      }
-     std::string filepath(argv[1]);
-     std::string prefix(argv[2]);
-  
-     Parser parser(filepath);
-     std::list<t2n_procedure> procedures(parser.get_procedures());
-
-     std::cerr << "Procedures:" << std::endl;
-     for (std::list<t2n_procedure>::const_iterator it=procedures.begin();it!=procedures.end();++it)
-         std::cerr << *it << ";" << std::endl;
-
-     std::set<std::string> groups(used_groups(procedures));
-     std::cerr << "Used groups:" << std::endl;
-     for (std::set<std::string>::const_iterator it=groups.begin();it!=groups.end();++it)
-         std::cerr << *it << std::endl;
-
-     std::string common_hpp_fname(prefix+"common.hxx");
-     std::string common_cpp_fname(prefix+"common.cpp");
-     std::string client_hpp_fname(prefix+"client.hxx");
-     std::string client_cpp_fname(prefix+"client.cpp");
-     std::string server_cpp_fname(prefix+"server.cpp");
+     return ret;
+}
 
-     header_file common_hpp(common_hpp_fname.c_str());
-     output_common_hpp(common_hpp, procedures);
+void
+paste_includes(std::ostream &o, std::list<std::string> &i)
+{
+     o << std::endl
+       << "// copied includes begin" << std::endl;
+     for (std::list<std::string>::const_iterator it=i.begin(); it!=i.end(); ++it)
+         o << *it << std::endl;
+     o << "// copied includes end" << std::endl
+       << std::endl;
+}
 
-     cpp_file common_cpp(common_cpp_fname.c_str());
-     output_common_cpp(common_cpp, procedures, common_hpp_fname);
+struct RemoveGenerated
+{
+     RemoveGenerated(const std::string &_prefix) : prefix(_prefix) {}
+     bool operator()(const std::string &s) const {
+         return (s.find(prefix+"common.hxx")!=std::string::npos);
+     }
+     const std::string &prefix;
+};
 
-     header_file client_hpp(client_hpp_fname.c_str());
-     output_client_hpp(client_hpp, procedures);
+int
+main(int argc, char* argv[])
+{
+     if (argc != 4) {
+         std::cerr << "Usage: " << argv[0] << " header-file gccxml-file outputprefix" << std::endl;
+         return 1;
+     }
 
-     cpp_file client_cpp(client_cpp_fname.c_str());
-     output_client_cpp(client_cpp, procedures, common_hpp_fname, common_cpp_fname, client_hpp_fname);
+     try{
+         std::string headerfile(argv[1]);
+         std::string xmlfile(argv[2]);
+         std::string prefix(argv[3]);
+         std::list<std::string> includes(get_includes(headerfile));
+         remove_if(includes.begin(), includes.end(), RemoveGenerated(prefix));
+         includes.erase(remove_if(includes.begin(), includes.end(), RemoveGenerated(prefix)), includes.end());
 
-     cpp_file server_cpp(server_cpp_fname.c_str());
-     output_server_cpp(server_cpp, procedures, common_hpp_fname, common_cpp_fname);
-     return 0;
+         //paste_includes(std::cerr, includes);
+  
+         Parser parser(xmlfile);
+         std::list<t2n_procedure> procedures(parser.get_procedures());
+
+         std::cerr << "Procedures:" << std::endl;
+         for (std::list<t2n_procedure>::const_iterator it=procedures.begin();it!=procedures.end();++it)
+              std::cerr << *it << ";" << std::endl;
+
+         std::set<std::string> groups(used_groups(procedures));
+         std::cerr << "Used groups:" << std::endl;
+         for (std::set<std::string>::const_iterator it=groups.begin();it!=groups.end();++it)
+              std::cerr << *it << std::endl;
+
+         std::string common_hpp_fname(prefix+"common.hxx");
+         std::string common_cpp_fname(prefix+"common.cpp");
+         std::string client_hpp_fname(prefix+"client.hxx");
+         std::string client_cpp_fname(prefix+"client.cpp");
+         std::string server_cpp_fname(prefix+"server.cpp");
+
+         header_file common_hpp(common_hpp_fname.c_str());
+         common_hpp << "// boost serialization is picky about order of include files => we have to include this one first\n"
+                    << "#include \"codegen-stubhead.hxx\"\n";
+              paste_includes(common_hpp, includes);
+
+              output_common_hpp(common_hpp, procedures);
+
+              cpp_file common_cpp(common_cpp_fname.c_str());
+              output_common_cpp(common_cpp, procedures, common_hpp_fname);
+
+              header_file client_hpp(client_hpp_fname.c_str());
+              client_hpp << "// boost serialization is picky about order of include files => we have to include this one first\n"
+                         << "#include \"codegen-stubhead.hxx\"\n";
+              paste_includes(client_hpp, includes);
+              output_client_hpp(client_hpp, procedures);
+
+              cpp_file client_cpp(client_cpp_fname.c_str());
+              output_client_cpp(client_cpp, procedures, common_hpp_fname, common_cpp_fname, client_hpp_fname);
+
+              cpp_file server_cpp(server_cpp_fname.c_str());
+              output_server_cpp(server_cpp, procedures, common_hpp_fname, common_cpp_fname);
+     }catch(const parse_error &e){
+         std::cerr << e.what() << std::endl;
+         return EXIT_FAILURE;
+     }
+     return EXIT_SUCCESS;
 }