X-Git-Url: http://developer.intra2net.com/git/?p=libt2n;a=blobdiff_plain;f=codegen%2Fmain.cpp;h=85c56ebee340afb096f15333c482081a8cd4fa4f;hp=e2f7bc0dc5cb37dacc7a94d3ed5793f7a893d969;hb=85106017e48b4dfdec213dc2ff83486edfa7ea10;hpb=e50cc1d69c0f69e431af2b4eac683ee3c5e35e68 diff --git a/codegen/main.cpp b/codegen/main.cpp index e2f7bc0..85c56eb 100644 --- a/codegen/main.cpp +++ b/codegen/main.cpp @@ -1,14 +1,52 @@ #include #include #include +#include +#include +#include +#include +#include -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; is.length() ) || ( std::string(s,0,prefix.length())!=prefix ) ) return error; + return std::string(s, prefix.length(), s.length()-prefix.length()); +} +//! extract group from attributes +std::string +extract_group(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); +} //! get child element by id /*! @@ -33,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 @@ -45,188 +92,475 @@ std::string get_namespace(const xmlpp::Element* root, const std::string &id) return element->get_attribute("name")->get_value(); } +struct type_info +{ + std::string name; + 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;} +}; + +std::ostream &operator<<(std::ostream &o, const type_info &t) { + o << t.name; + 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(line)+": error: "+msg) + {} +}; + //! get type by id /*! \return type name or empty string on error */ -std::string get_type(const xmlpp::Element* root, const std::string &id) +type_info get_type(const xmlpp::Element* root, const std::string &id) { - std::string error; + type_info error; const xmlpp::Element* element(get_element_by_id(root, id)); if (!element) return error; // TODO: not yet complete // if we recurse - when do we stop? // if it is a typedef? yes? (hmm if the typedef is in the file parsed this will not work) + + // TODO: const and reference types handling is a ugly hack + std::string tag(element->get_name()); if (tag=="ReferenceType") { assert(element->get_attribute("type")); - std::string ret(get_type(root, element->get_attribute("type")->get_value())); + type_info ret(get_type(root, element->get_attribute("type")->get_value())); if (ret==error) return error; - return ret+"&"; + // 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; }else if (tag=="CvQualifiedType") { assert(element->get_attribute("type")); - std::string ret(get_type(root, element->get_attribute("type")->get_value())); + type_info ret(get_type(root, element->get_attribute("type")->get_value())); if (ret==error) return error; - return std::string("const ")+ret; + ret.name=std::string("const ")+ret.name; + return ret; + }else if (tag=="PointerType") { + // not yet supported + return error; } assert(element->get_attribute("name")); - assert(element->get_attribute("context")); - return get_namespace(root, element->get_attribute("context")->get_value())+"::"+element->get_attribute("name")->get_value(); + type_info ret; + 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; } -void parse_function(const xmlpp::Element* root, const xmlpp::Node* node, unsigned int indentation = 0) { - const xmlpp::Element* element = dynamic_cast(node); - if (!element) return; +struct t2n_procedure +{ + typedef std::list > Args; - const xmlpp::Attribute* attributes = element->get_attribute("attributes"); - const xmlpp::Attribute* name = element->get_attribute("name"); - const xmlpp::Attribute* returns = element->get_attribute("returns"); - if ((!attributes)||(!name)||(!returns)) return; + std::string group; + type_info ret_type; + std::string name; + std::string mangled; + Args args; - // check wether the function is marked (TODO: improve) - if (attributes->get_value().find("gccxml(libt2n")==std::string::npos) return; + std::string ret_classname() const { + return name+mangled+"_res"; + } + std::string cmd_classname() const { + return name+mangled+"_cmd"; + } +}; - // TODO: extract command group - // something like: sed 's,gccxml(libt2n-\([a-z]*}),\1,' +std::ostream &operator<<(std::ostream &o, const t2n_procedure::Args &args) { + for (t2n_procedure::Args::const_iterator it=args.begin();it!=args.end();++it) { + if (it!=args.begin()) o << ", "; + o << it->second << " " << it->first; + } + return o; +} - // we need the return type - std::string ret_type(get_type(root, returns->get_value())); +std::ostream &operator<<(std::ostream &o, const t2n_procedure &f) { + o << f.ret_type << " " << f.name << "(" << f.args << ")"; + return o; +} - // and the argument types - typedef std::map Args; - Args args; +class Parser +{ +public: + Parser(const std::string &fname, const std::string &default_group) : m_fname(fname), m_default_group(default_group) {} - xmlpp::Node::NodeList list = node->get_children("Argument"); - for(xmlpp::Node::NodeList::iterator iter = list.begin(); iter != list.end(); ++iter) - { - const xmlpp::Element* arg = dynamic_cast(*iter); - if ( arg ) { - assert(arg->get_name() == "Argument"); - assert(arg->get_attribute("name")); - assert(arg->get_attribute("type")); - assert(args.find(arg->get_attribute("name")->get_value())==args.end()); - args[arg->get_attribute("name")->get_value()]=get_type(root, arg->get_attribute("type")->get_value()); + std::list get_procedures() { + xmlpp::DomParser parser; + // parser.set_validate(); + parser.set_substitute_entities(); //We just want the text to be resolved/unescaped automatically. + parser.parse_file(m_fname); + if(parser) + { + //Walk the tree: + const xmlpp::Node* pNode = parser.get_document()->get_root_node(); //deleted by DomParser. + const xmlpp::Element* root = dynamic_cast(pNode); + assert(root); + visit_node(root); } + return m_procedures; } +protected: + std::string m_fname; + std::string m_default_group; + std::list m_procedures; + + void parse_function(const xmlpp::Element* root, const xmlpp::Node* node) { + const xmlpp::Element* element = dynamic_cast(node); + if (!element) return; + + 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)||(!mangled)||(!returns)) return; + + // check wether the procedure is marked (TODO: improve) + // attributes are speparated by spaces? - std::cerr << "Found marked Function: " << ret_type << " " << name->get_value() << "("; - for (Args::const_iterator it=args.begin();it!=args.end();++it) { - if (it!=args.begin()) std::cerr << ", "; - std::cerr << it->second << " " << it->first; + t2n_procedure f; + f.group=extract_group(attributes->get_value()); + if (f.group.empty()) return; + if (f.group=="default") f.group=m_default_group; + + // 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) + { + const xmlpp::Element* arg = dynamic_cast(*iter); + if ( arg ) { + assert(arg->get_name() == "Argument"); + assert(arg->get_attribute("name")); + assert(arg->get_attribute("type")); + f.args.push_back(std::pair(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(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; + m_procedures.push_back(f); } - std::cerr << ");" << std::endl; -} -void print_node(const xmlpp::Element* root, const xmlpp::Node* node = NULL, unsigned int indentation = 0) -{ - if (!node) node=root; - std::cout << std::endl; //Separate nodes by an empty line. - - const xmlpp::ContentNode* nodeContent = dynamic_cast(node); - const xmlpp::TextNode* nodeText = dynamic_cast(node); - const xmlpp::CommentNode* nodeComment = dynamic_cast(node); - - if(nodeText && nodeText->is_white_space()) //Let's ignore the indenting - you don't always want to do this. - return; + void visit_node(const xmlpp::Element* root, const xmlpp::Node* node = NULL, unsigned int indentation = 0) + { + if (!node) node=root; + + const xmlpp::ContentNode* nodeContent = dynamic_cast(node); + const xmlpp::TextNode* nodeText = dynamic_cast(node); + const xmlpp::CommentNode* nodeComment = dynamic_cast(node); + + if(nodeText && nodeText->is_white_space()) //Let's ignore the indenting - you don't always want to do this. + return; - std::string nodename = node->get_name(); + std::string nodename = node->get_name(); - 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, indentation); + if(!nodeText && !nodeComment && !nodename.empty()) //Let's not say "name: text". + { + if (node->get_name() == "Function") parse_function(root, node); + } + if(!nodeContent) + { + //Recurse through child nodes: + xmlpp::Node::NodeList list = node->get_children(); + for(xmlpp::Node::NodeList::iterator iter = list.begin(); iter != list.end(); ++iter) + { + visit_node(root, *iter, indentation + 2); //recursive + } + } } +}; + +//! find used groups +std::set +used_groups(const std::list &funcs) { + typedef std::set Groups; + Groups groups; + for (std::list::const_iterator it=funcs.begin();it!=funcs.end();++it) + // since we use std::set each group is inserted only once + groups.insert(it->group); + return groups; +} + +void output_common_hpp(std::ostream &o, const std::list &procs) { + std::set groups(used_groups(procs)); + + for (std::set::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\n" + << " void serialize(Archive & ar, const unsigned int /* version */)\n" + << " {ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(libt2n::command);}\n" + << "};\n"; } - 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; + + for (std::list::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" + << " template\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" + << "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"; } + for (std::list::const_iterator it=procs.begin();it!=procs.end();++it) { + o << "class " << it->cmd_classname() << " : public " << "cmd_group_" << it->group << "\n" + << "{\n" + << "private:\n"; + for (t2n_procedure::Args::const_iterator ait=it->args.begin();ait!=it->args.end();++ait) { + o << " " << ait->second.noref() << " " << ait->first << ";\n"; + } + o << " friend class boost::serialization::access;\n" + << " template\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) { + o << " ar & BOOST_SERIALIZATION_NVP(" << ait->first << ");\n"; + } - //Treat the various node types differently: - if(nodeText) - { - print_indentation(indentation); - std::cout << "text = \"" << nodeText->get_content() << "\"" << std::endl; + // default constructor + o << " }\n" + << "\n" + << "public:\n" + << " " << 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 (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) { + if (ait!=it->args.begin()) o << ", "; + o << ait->first << "(_" << ait->first << ")"; + } + o << " {}\n" + << " libt2n::result* operator()();\n" + << "};\n"; } - else if(nodeComment) - { - print_indentation(indentation); - std::cout << "comment = " << nodeComment->get_content() << std::endl; +} + +void output_common_cpp(std::ostream &o, const std::list &procs, const std::string &common_hpp) { + std::set groups(used_groups(procs)); + + o << "#include \"" << common_hpp << "\"\n" + << "#include \n" + << "\n" + << "/* register types with boost serialization */\n"; + for (std::set::iterator it=groups.begin();it!=groups.end();++it) { + o << "BOOST_CLASS_EXPORT(cmd_group_" << *it << ")\n"; } - else if(nodeContent) - { - print_indentation(indentation); - std::cout << "content = " << nodeContent->get_content() << std::endl; + for (std::list::const_iterator it=procs.begin();it!=procs.end();++it) { + o << "BOOST_CLASS_EXPORT("<ret_classname()<<")\n" + << "BOOST_CLASS_EXPORT("<cmd_classname()<<")\n"; } - else if(const xmlpp::Element* nodeElement = dynamic_cast(node)) - { - //A normal Element node: +} - //line() works only for ElementNodes. - print_indentation(indentation); - std::cout << " line = " << node->get_line() << std::endl; +void output_client_hpp(std::ostream &o, const std::list &procs) { + std::set groups(used_groups(procs)); - //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; + o << "#include \n"; + + for (std::set::iterator it=groups.begin();it!=groups.end();++it) { + o << "class cmd_group_" << *it << "_client : public libt2n::command_client\n" + << "{\n" + << "public:\n" + << "cmd_group_" << *it << "_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" + << " {}\n"; + for (std::list::const_iterator pit=procs.begin();pit!=procs.end();++pit) { + if (pit->group==*it) { + o << " " << *pit << ";\n"; + } } + o << "};\n"; + } +} - const xmlpp::Attribute* attribute = nodeElement->get_attribute("title"); - if(attribute) - { - std::cout << "title found: =" << attribute->get_value() << std::endl; - +void output_client_cpp(std::ostream &o, const std::list &procs, const std::string &common_hpp, const std::string &common_cpp, const std::string &client_hpp) { + std::set groups(used_groups(procs)); + + o << "#include \"" << client_hpp << "\"\n" + << "#include \"" << common_hpp << "\"\n" + << "// fake\n"; + for (std::list::const_iterator it=procs.begin();it!=procs.end();++it) { + o << "libt2n::result* " << it->cmd_classname() << "::operator()() { return NULL; }\n"; + } + + for (std::set::iterator it=groups.begin();it!=groups.end();++it) { + for (std::list::const_iterator pit=procs.begin();pit!=procs.end();++pit) { + if (pit->group==*it) { + o << pit->ret_type << " cmd_group_" << *it << "_client::" << pit->name << "(" << pit->args << ")\n" + << "{\n" + << " libt2n::result_container rc;\n" + << " send_command(new " << pit->cmd_classname() << "("; + for (t2n_procedure::Args::const_iterator ait=pit->args.begin();ait!=pit->args.end();++ait) { + if (ait!=pit->args.begin()) o << ", "; + o << ait->first; + } + 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(!nodeContent) - { - //Recurse through child nodes: - xmlpp::Node::NodeList list = node->get_children(); - for(xmlpp::Node::NodeList::iterator iter = list.begin(); iter != list.end(); ++iter) - { - print_node(root, *iter, indentation + 2); //recursive + + // include in this compilation unit to ensure the compilation unit is used + // see also: + // http://www.google.de/search?q=g%2B%2B+static+initializer+in+static+library + o << "#include \"" << common_cpp << "\"\n"; +} + +void output_server_hpp(std::ostream &o, const std::list &procs, const std::string &common_hpp) { + o << "#include \"" << common_hpp << "\"\n"; + + // output function declarations + for (std::list::const_iterator it=procs.begin();it!=procs.end();++it) + o << *it << ";\n"; +} + +void output_server_cpp(std::ostream &o, const std::list &procs, const std::string &common_hpp, const std::string &common_cpp) { + o << "#include \"" << common_hpp << "\"\n"; + + for (std::list::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 << ")); }\n"; } + o << "#include \"" << common_cpp << "\"\n"; } -int main(int argc, char* argv[]) +struct header_file : public std::ofstream { - std::string filepath; - if(argc > 1 ) - filepath = argv[1]; //Allow the user to specify a different XML file to parse. - else - filepath = "example.xml"; - - try - { - xmlpp::DomParser parser; - // parser.set_validate(); - parser.set_substitute_entities(); //We just want the text to be resolved/unescaped automatically. - parser.parse_file(filepath); - if(parser) - { - //Walk the tree: - const xmlpp::Node* pNode = parser.get_document()->get_root_node(); //deleted by DomParser. - const xmlpp::Element* root = dynamic_cast(pNode); - assert(root); - print_node(root); - } + 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 << "#ifndef " << macro << "\n" + << "#define " << macro << "\n"; } - catch(const std::exception& ex) - { - std::cout << "Exception caught: " << ex.what() << std::endl; + ~header_file() { + *this << "#endif" << std::endl; } +}; - return 0; +struct cpp_file : public std::ofstream +{ + cpp_file(const char* fname) : std::ofstream(fname) { + std::cerr << "create cpp: '" << fname << "'" << std::endl; + *this << "// automatically generated code - do not edit\n" << std::endl; + } +}; + +int +main(int argc, char* argv[]) +{ + if (argc < 3) { + std::cerr << "Usage: " << argv[0] << "default-group gccxml-file1 gccxml-file2 ... " << std::endl; + return 1; + } + + try{ + std::string group(argv[1]); + std::list xmlfiles; + for (int i=2;i procedures; + for (std::list::iterator it=xmlfiles.begin();it!=xmlfiles.end();++it) { + Parser parser(*it, group); + const std::list &p(parser.get_procedures()); + std::copy(p.begin(), p.end(), std::back_inserter(procedures)); + } + + std::cerr << "Procedures:" << std::endl; + for (std::list::const_iterator it=procedures.begin();it!=procedures.end();++it) + std::cerr << *it << ";" << std::endl; + + std::set groups(used_groups(procedures)); + std::cerr << "Used groups:" << std::endl; + for (std::set::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_hpp_fname(prefix+"server.hxx"); + 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" + << "#include \"" << group << ".hxx\"\n"; + + 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" + << "#include \"" << group << ".hxx\"\n"; + 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); + + header_file server_hpp(server_hpp_fname.c_str()); + output_server_hpp(server_hpp, procedures, common_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; }