[libt2n] / codegen / main.cpp

#include <libxml++/libxml++.h>
#include <cassert>
#include <iostream>
#include <set>
#include <fstream>

void print_indentation(unsigned int indentation)
{
     for(unsigned int i = 0; i < indentation; ++i)
	  std::cout << " ";
}

//! get child element by id
/*!
  \return pointer to element having id or null on error
  \todo find libxmlpp pendant
*/
const xmlpp::Element* get_element_by_id(const xmlpp::Element* element, const std::string &id)
{
     const xmlpp::Attribute* cid = element->get_attribute("id");
     if ( cid && ( cid->get_value() == id)) return element;

     //Recurse through child nodes:
     xmlpp::Node::NodeList list = element->get_children();
     for(xmlpp::Node::NodeList::iterator iter = list.begin(); iter != list.end(); ++iter)
     {
	  const xmlpp::Element* element = dynamic_cast<const xmlpp::Element*>(*iter);
	  if (element) {
	       const xmlpp::Element* match = get_element_by_id(element, id);
	       if (match) return match;
	  }
     }
     return NULL;
}

//! get namespace by id
/*!
  \return namespace name or empty string on error
*/
std::string get_namespace(const xmlpp::Element* root, const std::string &id)
{
     std::string error;
     const xmlpp::Element* element(get_element_by_id(root, id));
     if ((!element)||(!element->get_attribute("name"))) return error;
     return element->get_attribute("name")->get_value();
}

//! get type by id
/*!
  \return type name or empty string on error
*/
std::string get_type(const xmlpp::Element* root, const std::string &id)
{
     std::string 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 (perhaps it is good enough to remove the const and &?)
     //       non-const reference types/pointers (output error message? not yet supported)

     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()));
	  if (ret==error) return error;
	  return ret+"&";
     }else if (tag=="CvQualifiedType") {
	  assert(element->get_attribute("type"));
	  std::string ret(get_type(root, element->get_attribute("type")->get_value()));
	  if (ret==error) return error;
	  return std::string("const ")+ret;
     }

     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();
}

struct t2n_procedure
{
     typedef std::map<std::string, std::string> Args;

     std::string group;
     std::string ret_type;
     std::string name;
     Args  args;

     std::string ret_classname() const {
	  return name+"_res";
     }
     std::string cmd_classname() const {
	  return name+"_cmd";
     }
};

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

std::ostream &operator<<(std::ostream &o, const t2n_procedure &f) {
     o << f.ret_type << " " << f.name << "(" << f.args << ")";
     return o;
}

class Parser
{
public:
     Parser(const std::string &fname) : m_fname(fname) {}

     std::list<t2n_procedure> 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<const xmlpp::Element*>(pNode);
	       assert(root);
	       visit_node(root);
	  }
	  return m_procedures;
     }
protected:
     std::string m_fname;
     std::list<t2n_procedure> m_procedures;

     void parse_function(const xmlpp::Element* root, const xmlpp::Node* node) {
	  const xmlpp::Element* element = dynamic_cast<const xmlpp::Element*>(node);
	  if (!element) return;

	  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;

	  // check wether the procedure is marked (TODO: improve)
	  // attributes are speparated by spaces?
	  if (attributes->get_value().find("gccxml(libt2n")==std::string::npos) return;

	  // TODO: extract command group
	  // something like: sed 's,gccxml(libt2n-\([a-z]*}),\1,'
	  t2n_procedure f;
	  f.group=("default");

	  // we need the return type
	  f.ret_type=get_type(root, returns->get_value());
	  f.name=name->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<const xmlpp::Element*>(*iter);
	       if ( arg ) {
		    assert(arg->get_name() == "Argument");
		    assert(arg->get_attribute("name"));
		    assert(arg->get_attribute("type"));
		    assert(f.args.find(arg->get_attribute("name")->get_value())==f.args.end());
		    f.args[arg->get_attribute("name")->get_value()]=get_type(root, arg->get_attribute("type")->get_value());
	       }
	  }
	  m_procedures.push_back(f);
     }

     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<const xmlpp::ContentNode*>(node);
	       const xmlpp::TextNode* nodeText = dynamic_cast<const xmlpp::TextNode*>(node);
	       const xmlpp::CommentNode* nodeComment = dynamic_cast<const xmlpp::CommentNode*>(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();

	       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;
	       }
	       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:
		    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<std::string>
used_groups(const std::list<t2n_procedure> &funcs) {
     typedef std::set<std::string> Groups;
     Groups groups;
     for (std::list<t2n_procedure>::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<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"
	    << "	{ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(libt2n::command);}\n"
	    << "};\n";
     }
     
     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"
	    << "	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"
	    << "public:\n"
	    << "	" << it->ret_classname() << "(const " << it->ret_type << " &_res) : res(_res) {}\n"
	    << "	" << it->ret_type << " get_data() { return res; }\n"
	    << "};\n";
     }
     for (std::list<t2n_procedure>::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 << " " << ait->first << ";\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(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";
	  }

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

void output_common_cpp(std::ostream &o, const std::list<t2n_procedure> &procs, const std::string &common_hpp) {
     std::set<std::string> groups(used_groups(procs));

     o << "#include \"" << common_hpp << "\"\n"
       << "#include <boost/serialization/export.hxx>\n"
       << "\n"
       << "/* register types with boost serialization */\n";
     for (std::set<std::string>::iterator it=groups.begin();it!=groups.end();++it) {
	  o << "BOOST_CLASS_EXPORT(cmd_group_" << *it << ")\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";
     }
}

void output_client_hpp(std::ostream &o, const std::list<t2n_procedure> &procs) {
     std::set<std::string> groups(used_groups(procs));

     o << "#include <command_client.hxx>\n";

     for (std::set<std::string>::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<t2n_procedure>::const_iterator pit=procs.begin();pit!=procs.end();++pit) {
	       if (pit->group==*it) {
		    o << "	" << *pit << ";\n";
	       }
	  }
	  o << "};\n";
     }
}

void output_client_cpp(std::ostream &o, const std::list<t2n_procedure> &procs, const std::string &common_hpp, const std::string &common_cpp, const std::string &client_hpp) {
     std::set<std::string> groups(used_groups(procs));

     o << "#include \"" << client_hpp << "\"\n"
       << "#include \"" << common_hpp << "\"\n"
       << "// fake\n";
     for (std::list<t2n_procedure>::const_iterator it=procs.begin();it!=procs.end();++it) {
	  o << "libt2n::result* " << it->cmd_classname() << "::operator()() { return NULL; }\n";
     }

     for (std::set<std::string>::iterator it=groups.begin();it!=groups.end();++it) {
	  for (std::list<t2n_procedure>::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";
	       }
	  }
     }

     // 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_cpp(std::ostream &o, const std::list<t2n_procedure> &procs, const std::string &common_hpp, const std::string &common_cpp) {
     o << "#include \"" << common_hpp << "\"\n";

     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 << ")); }\n";
     }
     o << "#include \"" << common_cpp << "\"\n";
}


std::string
toupper(std::string s) {
     for (unsigned i=0; i<s.length(); ++i) s[i]=toupper(s[i]);
     return 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;
}

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 << "#ifndef " << macro << "\n"
		<< "#define " << macro << "\n";
     }
     ~header_file() {
	  *this << "#endif" << std::endl;
     }
};

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] << " gccxml-file outputprefix" << std::endl;
	  return 1;
     }
     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+"client.cpp");

     header_file common_hpp(common_hpp_fname.c_str());
     output_common_hpp(common_hpp, procedures);

     cpp_file common_cpp(common_cpp_fname.c_str());
     output_common_cpp(common_cpp, procedures, "common-stub.hxx");

     header_file client_hpp(client_hpp_fname.c_str());
     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);
     return 0;
}
Commit	Line	Data
e50cc1d6 JT	1	#include <libxml++/libxml++.h>
	2	#include <cassert>
	3	#include <iostream>
060fbd87 JT	4	#include <set>
060fbd87 JT	5	#include <fstream>
e50cc1d6 JT	6
	7	void print_indentation(unsigned int indentation)
	8	{
	9	for(unsigned int i = 0; i < indentation; ++i)
	10	std::cout << " ";
	11	}
	12
e50cc1d6 JT	13	//! get child element by id
	14	/*!
	15	\return pointer to element having id or null on error
	16	\todo find libxmlpp pendant
	17	*/
	18	const xmlpp::Element* get_element_by_id(const xmlpp::Element* element, const std::string &id)
	19	{
	20	const xmlpp::Attribute* cid = element->get_attribute("id");
	21	if ( cid && ( cid->get_value() == id)) return element;
	22
	23	//Recurse through child nodes:
	24	xmlpp::Node::NodeList list = element->get_children();
	25	for(xmlpp::Node::NodeList::iterator iter = list.begin(); iter != list.end(); ++iter)
	26	{
	27	const xmlpp::Element* element = dynamic_cast<const xmlpp::Element>(iter);
	28	if (element) {
	29	const xmlpp::Element* match = get_element_by_id(element, id);
	30	if (match) return match;
	31	}
	32	}
	33	return NULL;
	34	}
	35
	36	//! get namespace by id
	37	/*!
	38	\return namespace name or empty string on error
	39	*/
	40	std::string get_namespace(const xmlpp::Element* root, const std::string &id)
	41	{
	42	std::string error;
	43	const xmlpp::Element* element(get_element_by_id(root, id));
	44	if ((!element)\|\|(!element->get_attribute("name"))) return error;
	45	return element->get_attribute("name")->get_value();
	46	}
	47
	48	//! get type by id
	49	/*!
	50	\return type name or empty string on error
	51	*/
	52	std::string get_type(const xmlpp::Element* root, const std::string &id)
	53	{
	54	std::string error;
	55	const xmlpp::Element* element(get_element_by_id(root, id));
	56	if (!element) return error;
	57
	58	// TODO: not yet complete
	59	// if we recurse - when do we stop?
	60	// if it is a typedef? yes? (hmm if the typedef is in the file parsed this will not work)
060fbd87 JT	61
	62	// TODO: const and reference types (perhaps it is good enough to remove the const and &?)
	63	// non-const reference types/pointers (output error message? not yet supported)
	64
e50cc1d6 JT	65	std::string tag(element->get_name());
	66	if (tag=="ReferenceType") {
	67	assert(element->get_attribute("type"));
	68	std::string ret(get_type(root, element->get_attribute("type")->get_value()));
	69	if (ret==error) return error;
	70	return ret+"&";
	71	}else if (tag=="CvQualifiedType") {
	72	assert(element->get_attribute("type"));
	73	std::string ret(get_type(root, element->get_attribute("type")->get_value()));
	74	if (ret==error) return error;
	75	return std::string("const ")+ret;
	76	}
	77
	78	assert(element->get_attribute("name"));
	79	assert(element->get_attribute("context"));
	80	return get_namespace(root, element->get_attribute("context")->get_value())+"::"+element->get_attribute("name")->get_value();
	81	}
	82
060fbd87 JT	83	struct t2n_procedure
	84	{
	85	typedef std::map<std::string, std::string> Args;
e50cc1d6	86
060fbd87 JT	87	std::string group;
	88	std::string ret_type;
	89	std::string name;
	90	Args args;
e50cc1d6	91
060fbd87 JT	92	std::string ret_classname() const {
	93	return name+"_res";
	94	}
	95	std::string cmd_classname() const {
	96	return name+"_cmd";
	97	}
	98	};
e50cc1d6	99
060fbd87 JT	100	std::ostream &operator<<(std::ostream &o, const t2n_procedure::Args &args) {
	101	for (t2n_procedure::Args::const_iterator it=args.begin();it!=args.end();++it) {
	102	if (it!=args.begin()) o << ", ";
	103	o << it->second << " " << it->first;
	104	}
	105	return o;
	106	}
e50cc1d6	107
060fbd87 JT	108	std::ostream &operator<<(std::ostream &o, const t2n_procedure &f) {
	109	o << f.ret_type << " " << f.name << "(" << f.args << ")";
	110	return o;
	111	}
e50cc1d6	112
060fbd87 JT	113	class Parser
	114	{
	115	public:
	116	Parser(const std::string &fname) : m_fname(fname) {}
e50cc1d6	117
060fbd87 JT	118	std::list<t2n_procedure> get_procedures() {
	119	xmlpp::DomParser parser;
	120	// parser.set_validate();
	121	parser.set_substitute_entities(); //We just want the text to be resolved/unescaped automatically.
	122	parser.parse_file(m_fname);
	123	if(parser)
	124	{
	125	//Walk the tree:
	126	const xmlpp::Node* pNode = parser.get_document()->get_root_node(); //deleted by DomParser.
	127	const xmlpp::Element* root = dynamic_cast<const xmlpp::Element*>(pNode);
	128	assert(root);
	129	visit_node(root);
e50cc1d6	130	}
060fbd87	131	return m_procedures;
e50cc1d6	132	}
060fbd87 JT	133	protected:
	134	std::string m_fname;
	135	std::list<t2n_procedure> m_procedures;
	136
	137	void parse_function(const xmlpp::Element* root, const xmlpp::Node* node) {
	138	const xmlpp::Element* element = dynamic_cast<const xmlpp::Element*>(node);
	139	if (!element) return;
	140
	141	const xmlpp::Attribute* attributes = element->get_attribute("attributes");
	142	const xmlpp::Attribute* name = element->get_attribute("name");
	143	const xmlpp::Attribute* returns = element->get_attribute("returns");
	144	if ((!attributes)\|\|(!name)\|\|(!returns)) return;
	145
	146	// check wether the procedure is marked (TODO: improve)
	147	// attributes are speparated by spaces?
	148	if (attributes->get_value().find("gccxml(libt2n")==std::string::npos) return;
	149
	150	// TODO: extract command group
	151	// something like: sed 's,gccxml(libt2n-\([a-z]*}),\1,'
	152	t2n_procedure f;
	153	f.group=("default");
e50cc1d6	154
060fbd87 JT	155	// we need the return type
	156	f.ret_type=get_type(root, returns->get_value());
	157	f.name=name->get_value();
	158
	159	xmlpp::Node::NodeList list = node->get_children("Argument");
	160	for(xmlpp::Node::NodeList::iterator iter = list.begin(); iter != list.end(); ++iter)
	161	{
	162	const xmlpp::Element* arg = dynamic_cast<const xmlpp::Element>(iter);
	163	if ( arg ) {
	164	assert(arg->get_name() == "Argument");
	165	assert(arg->get_attribute("name"));
	166	assert(arg->get_attribute("type"));
	167	assert(f.args.find(arg->get_attribute("name")->get_value())==f.args.end());
	168	f.args[arg->get_attribute("name")->get_value()]=get_type(root, arg->get_attribute("type")->get_value());
	169	}
	170	}
	171	m_procedures.push_back(f);
e50cc1d6	172	}
e50cc1d6	173
060fbd87 JT	174	void visit_node(const xmlpp::Element* root, const xmlpp::Node* node = NULL, unsigned int indentation = 0)
	175	{
	176	if (!node) node=root;
060fbd87 JT	177
	178	const xmlpp::ContentNode* nodeContent = dynamic_cast<const xmlpp::ContentNode*>(node);
	179	const xmlpp::TextNode* nodeText = dynamic_cast<const xmlpp::TextNode*>(node);
	180	const xmlpp::CommentNode* nodeComment = dynamic_cast<const xmlpp::CommentNode*>(node);
e50cc1d6	181
060fbd87 JT	182	if(nodeText && nodeText->is_white_space()) //Let's ignore the indenting - you don't always want to do this.
060fbd87 JT	183	return;
e50cc1d6	184
060fbd87	185	std::string nodename = node->get_name();
e50cc1d6	186
060fbd87 JT	187	if(!nodeText && !nodeComment && !nodename.empty()) //Let's not say "name: text".
060fbd87 JT	188	{
f8b78aa8	189	// print_indentation(indentation);
060fbd87 JT	190	// std::cout << "Node name = " << node->get_name() << std::endl;
	191	// std::cout << "Node name = " << nodename << std::endl;
	192	if (node->get_name() == "Function") {
	193	parse_function(root, node);
	194	}
	195	}
f8b78aa8	196	#if 0
060fbd87 JT	197	else if(nodeText) //Let's say when it's text. - e.g. let's say what that white space is.
	198	{
	199	print_indentation(indentation);
	200	std::cout << "Text Node" << std::endl;
	201	}
	202
	203	//Treat the various node types differently:
	204	if(nodeText)
	205	{
	206	print_indentation(indentation);
	207	std::cout << "text = \"" << nodeText->get_content() << "\"" << std::endl;
	208	}
	209	else if(nodeComment)
	210	{
	211	print_indentation(indentation);
	212	std::cout << "comment = " << nodeComment->get_content() << std::endl;
	213	}
	214	else if(nodeContent)
	215	{
	216	print_indentation(indentation);
	217	std::cout << "content = " << nodeContent->get_content() << std::endl;
	218	}
	219	else if(const xmlpp::Element* nodeElement = dynamic_cast<const xmlpp::Element*>(node))
	220	{
	221	//A normal Element node:
	222
	223	//line() works only for ElementNodes.
	224	print_indentation(indentation);
	225	std::cout << " line = " << node->get_line() << std::endl;
	226
	227	//Print attributes:
	228	const xmlpp::Element::AttributeList& attributes = nodeElement->get_attributes();
	229	for(xmlpp::Element::AttributeList::const_iterator iter = attributes.begin(); iter != attributes.end(); ++iter)
	230	{
	231	const xmlpp::Attribute* attribute = *iter;
	232	print_indentation(indentation);
	233	std::cout << " Attribute " << attribute->get_name() << " = " << attribute->get_value() << std::endl;
	234	}
	235
	236	const xmlpp::Attribute* attribute = nodeElement->get_attribute("title");
	237	if(attribute)
	238	{
	239	std::cout << "title found: =" << attribute->get_value() << std::endl;
	240
	241	}
	242	}
f8b78aa8	243	#endif
060fbd87 JT	244	if(!nodeContent)
	245	{
	246	//Recurse through child nodes:
	247	xmlpp::Node::NodeList list = node->get_children();
	248	for(xmlpp::Node::NodeList::iterator iter = list.begin(); iter != list.end(); ++iter)
	249	{
	250	visit_node(root, *iter, indentation + 2); //recursive
	251	}
	252	}
e50cc1d6	253	}
060fbd87 JT	254	};
	255
	256	//! find used groups
	257	std::set<std::string>
	258	used_groups(const std::list<t2n_procedure> &funcs) {
	259	typedef std::set<std::string> Groups;
	260	Groups groups;
	261	for (std::list<t2n_procedure>::const_iterator it=funcs.begin();it!=funcs.end();++it)
	262	// since we use std::set each group is inserted only once
	263	groups.insert(it->group);
	264	return groups;
	265	}
	266
	267	void output_common_hpp(std::ostream &o, const std::list<t2n_procedure> &procs) {
	268	std::set<std::string> groups(used_groups(procs));
	269
	270	o << "#include \"codegen-stubhead.hxx\"\n";
	271	for (std::set<std::string>::iterator it=groups.begin();it!=groups.end();++it) {
	272	o << "class cmd_group_" << *it << " : public libt2n::command\n"
	273	<< "{\n"
	274	<< "private:\n"
	275	<< " friend class boost::serialization::access;\n"
	276	<< " template<class Archive>\n"
	277	<< " void serialize(Archive & ar, const unsigned int version)\n"
	278	<< " {ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(libt2n::command);}\n"
	279	<< "};\n";
e50cc1d6	280	}
060fbd87 JT	281
	282	for (std::list<t2n_procedure>::const_iterator it=procs.begin();it!=procs.end();++it) {
	283	o << "class " << it->ret_classname() << " : public libt2n::result\n"
	284	<< "{\n"
	285	<< "private:\n"
	286	<< " " << it->ret_type << " res;\n"
	287	<< " friend class boost::serialization::access;\n"
	288	<< " template<class Archive>\n"
	289	<< " void serialize(Archive & ar, const unsigned int version)\n"
	290	<< " {\n"
	291	<< " ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(libt2n::result);\n"
	292	<< " ar & BOOST_SERIALIZATION_NVP(res);\n"
	293	<< " }\n"
	294	<< "public:\n"
	295	<< " " << it->ret_classname() << "(const " << it->ret_type << " &_res) : res(_res) {}\n"
	296	<< " " << it->ret_type << " get_data() { return res; }\n"
	297	<< "};\n";
e50cc1d6	298	}
060fbd87 JT	299	for (std::list<t2n_procedure>::const_iterator it=procs.begin();it!=procs.end();++it) {
	300	o << "class " << it->cmd_classname() << " : public " << "cmd_group_" << it->group << "\n"
	301	<< "{\n"
	302	<< "private:\n";
	303	for (t2n_procedure::Args::const_iterator ait=it->args.begin();ait!=it->args.end();++ait) {
	304	o << " " << ait->second << " " << ait->first << ";\n";
	305	}
	306	o << " friend class boost::serialization::access;\n"
	307	<< " template<class Archive>\n"
	308	<< " void serialize(Archive & ar, const unsigned int version)\n"
	309	<< " {\n"
	310	<< " ar & BOOST_SERIALIZATION_BASE_OBJECT_NVP(cmd_group_" << it->group << ");\n";
	311	for (t2n_procedure::Args::const_iterator ait=it->args.begin();ait!=it->args.end();++ait) {
	312	o << " ar & BOOST_SERIALIZATION_NVP(" << ait->first << ");\n";
	313	}
e50cc1d6	314
060fbd87 JT	315	// default constructor
	316	o << " }\n"
	317	<< "\n"
	318	<< "public:\n"
	319	<< " " << it->cmd_classname() << "() {}\n";
	320
	321	// constructor taking all arguments
	322	o << " " << it->cmd_classname() << "(";
	323	for (t2n_procedure::Args::const_iterator ait=it->args.begin();ait!=it->args.end();++ait) {
	324	if (ait!=it->args.begin()) o << ", ";
	325	o << ait->second << " _" << ait->first;
	326	}
	327	o << ") : ";
	328	for (t2n_procedure::Args::const_iterator ait=it->args.begin();ait!=it->args.end();++ait) {
	329	if (ait!=it->args.begin()) o << ", ";
	330	o << ait->first << "(_" << ait->first << ")";
	331	}
	332	o << " {}\n"
	333	<< " libt2n::result* operator()();\n"
	334	<< "};\n";
e50cc1d6	335	}
060fbd87 JT	336	}
	337
	338	void output_common_cpp(std::ostream &o, const std::list<t2n_procedure> &procs, const std::string &common_hpp) {
	339	std::set<std::string> groups(used_groups(procs));
	340
	341	o << "#include \"" << common_hpp << "\"\n"
	342	<< "#include <boost/serialization/export.hxx>\n"
	343	<< "\n"
	344	<< "/* register types with boost serialization */\n";
	345	for (std::set<std::string>::iterator it=groups.begin();it!=groups.end();++it) {
	346	o << "BOOST_CLASS_EXPORT(cmd_group_" << *it << ")\n";
e50cc1d6	347	}
060fbd87 JT	348	for (std::list<t2n_procedure>::const_iterator it=procs.begin();it!=procs.end();++it) {
	349	o << "BOOST_CLASS_EXPORT("<<it->ret_classname()<<")\n"
	350	<< "BOOST_CLASS_EXPORT("<<it->cmd_classname()<<")\n";
e50cc1d6	351	}
060fbd87	352	}
e50cc1d6	353
060fbd87 JT	354	void output_client_hpp(std::ostream &o, const std::list<t2n_procedure> &procs) {
060fbd87 JT	355	std::set<std::string> groups(used_groups(procs));
e50cc1d6	356
060fbd87 JT	357	o << "#include <command_client.hxx>\n";
	358
	359	for (std::set<std::string>::iterator it=groups.begin();it!=groups.end();++it) {
	360	o << "class cmd_group_" << *it << "_client : public libt2n::command_client\n"
	361	<< "{\n"
	362	<< "public:\n"
	363	<< "cmd_group_" << *it << "_client(libt2n::client_connection &_c,\n"
	364	<< " long long _command_timeout_usec=command_timeout_usec_default,\n"
	365	<< " long long _hello_timeout_usec=hello_timeout_usec_default)\n"
	366	<< " : libt2n::command_client(_c,_command_timeout_usec,_hello_timeout_usec)\n"
	367	<< " {}\n";
	368	for (std::list<t2n_procedure>::const_iterator pit=procs.begin();pit!=procs.end();++pit) {
	369	if (pit->group==*it) {
	370	o << " " << *pit << ";\n";
	371	}
e50cc1d6	372	}
060fbd87 JT	373	o << "};\n";
	374	}
	375	}
e50cc1d6	376
060fbd87 JT	377	void output_client_cpp(std::ostream &o, const std::list<t2n_procedure> &procs, const std::string &common_hpp, const std::string &common_cpp, const std::string &client_hpp) {
	378	std::set<std::string> groups(used_groups(procs));
	379
	380	o << "#include \"" << client_hpp << "\"\n"
	381	<< "#include \"" << common_hpp << "\"\n"
	382	<< "// fake\n";
	383	for (std::list<t2n_procedure>::const_iterator it=procs.begin();it!=procs.end();++it) {
	384	o << "libt2n::result* " << it->cmd_classname() << "::operator()() { return NULL; }\n";
	385	}
	386
	387	for (std::set<std::string>::iterator it=groups.begin();it!=groups.end();++it) {
	388	for (std::list<t2n_procedure>::const_iterator pit=procs.begin();pit!=procs.end();++pit) {
	389	if (pit->group==*it) {
	390	o << pit->ret_type << " cmd_group_" << *it << "_client::" << pit->name << "(" << pit->args << ")\n"
	391	<< "{\n"
	392	<< " libt2n::result_container rc;\n"
	393	<< " send_command(new " << pit->cmd_classname() << "(";
	394	for (t2n_procedure::Args::const_iterator ait=pit->args.begin();ait!=pit->args.end();++ait) {
	395	if (ait!=pit->args.begin()) o << ", ";
	396	o << ait->first;
	397	}
	398	o << "), rc);\n"
	399	<< " " << pit->ret_classname() << "* res=dynamic_cast<" << pit->ret_classname() << "*>(rc.get_result());\n"
	400	<< " if (!res) throw libt2n::t2n_communication_error(\"result object of wrong type\");\n"
	401	<< " return res->get_data();\n"
	402	<< "}\n";
	403	}
e50cc1d6 JT	404	}
e50cc1d6 JT	405	}
060fbd87 JT	406
	407	// include in this compilation unit to ensure the compilation unit is used
	408	// see also:
	409	// http://www.google.de/search?q=g%2B%2B+static+initializer+in+static+library
	410	o << "#include \"" << common_cpp << "\"\n";
	411	}
	412
	413	void output_server_cpp(std::ostream &o, const std::list<t2n_procedure> &procs, const std::string &common_hpp, const std::string &common_cpp) {
	414	o << "#include \"" << common_hpp << "\"\n";
	415
	416	for (std::list<t2n_procedure>::const_iterator it=procs.begin();it!=procs.end();++it) {
	417	o << *it << ";\n";
	418	o << "libt2n::result* " << it->cmd_classname() << "::operator()() { return new " << it->ret_classname() << "(" << it->name << "(";
	419	for (t2n_procedure::Args::const_iterator ait=it->args.begin();ait!=it->args.end();++ait) {
	420	if (ait!=it->args.begin()) o << ", ";
	421	o << ait->first;
e50cc1d6	422	}
060fbd87	423	o << ")); }\n";
e50cc1d6	424	}
060fbd87 JT	425	o << "#include \"" << common_cpp << "\"\n";
	426	}
	427
	428
	429	std::string
	430	toupper(std::string s) {
	431	for (unsigned i=0; i<s.length(); ++i) s[i]=toupper(s[i]);
	432	return s;
e50cc1d6 JT	433	}
e50cc1d6 JT	434
060fbd87 JT	435	std::string
	436	replace(std::string s, char f, char r) {
	437	for (unsigned i=0; i<s.length(); ++i) if (s[i]==f) s[i]=r;
	438	return s;
	439	}
	440
	441	struct header_file : public std::ofstream
	442	{
	443	header_file(const char* fname) : std::ofstream(fname) {
	444	std::cerr << "create header: '" << fname << "'" << std::endl;
	445	std::string macro(replace(toupper(fname),'.','_'));
	446	*this << "// automatically generated code - do not edit\n" << std::endl;
	447	*this << "#ifndef " << macro << "\n"
	448	<< "#define " << macro << "\n";
	449	}
	450	~header_file() {
	451	*this << "#endif" << std::endl;
	452	}
	453	};
	454
	455	struct cpp_file : public std::ofstream
	456	{
	457	cpp_file(const char* fname) : std::ofstream(fname) {
	458	std::cerr << "create cpp: '" << fname << "'" << std::endl;
	459	*this << "// automatically generated code - do not edit\n" << std::endl;
	460	}
	461	};
	462
	463	int
	464	main(int argc, char* argv[])
e50cc1d6	465	{
f8b78aa8 JT	466	if (argc != 3) {
	467	std::cerr << "Usage: " << argv[0] << " gccxml-file outputprefix" << std::endl;
	468	return 1;
	469	}
	470	std::string filepath(argv[1]);
	471	std::string prefix(argv[2]);
e50cc1d6	472
060fbd87 JT	473	Parser parser(filepath);
	474	std::list<t2n_procedure> procedures(parser.get_procedures());
	475
	476	std::cerr << "Procedures:" << std::endl;
	477	for (std::list<t2n_procedure>::const_iterator it=procedures.begin();it!=procedures.end();++it)
	478	std::cerr << *it << ";" << std::endl;
	479
	480	std::set<std::string> groups(used_groups(procedures));
	481	std::cerr << "Used groups:" << std::endl;
	482	for (std::set<std::string>::const_iterator it=groups.begin();it!=groups.end();++it)
	483	std::cerr << *it << std::endl;
	484
060fbd87 JT	485	std::string common_hpp_fname(prefix+"common.hxx");
	486	std::string common_cpp_fname(prefix+"common.cpp");
	487	std::string client_hpp_fname(prefix+"client.hxx");
	488	std::string client_cpp_fname(prefix+"client.cpp");
	489	std::string server_cpp_fname(prefix+"client.cpp");
	490
	491	header_file common_hpp(common_hpp_fname.c_str());
	492	output_common_hpp(common_hpp, procedures);
	493
	494	cpp_file common_cpp(common_cpp_fname.c_str());
	495	output_common_cpp(common_cpp, procedures, "common-stub.hxx");
	496
	497	header_file client_hpp(client_hpp_fname.c_str());
	498	output_client_hpp(client_hpp, procedures);
	499
	500	cpp_file client_cpp(client_cpp_fname.c_str());
	501	output_client_cpp(client_cpp, procedures, common_hpp_fname, common_cpp_fname, client_hpp_fname);
e50cc1d6	502
060fbd87 JT	503	cpp_file server_cpp(server_cpp_fname.c_str());
060fbd87 JT	504	output_server_cpp(server_cpp, procedures, common_hpp_fname, common_cpp_fname);
e50cc1d6 JT	505	return 0;
e50cc1d6 JT	506	}