You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
tdebindings/dcoppython/shell/gen_marshal_code.py

277 lines
9.0 KiB

#!/usr/bin/env python
# Julian Rockey 2003
# Generate marshall/demarshal functions from marshal_funcs.data file
import sys
import re
def cap_first(str):
"""Capitalise first letter of string."""
return str[0].upper() + str[1:]
def set_method(attr):
"""Return the name for a QT class setter method for an attribute."""
return "set" + cap_first(attr)
class DictMaker:
"""Generate code for marshalling/demarshalling types using Python dictionaries."""
supported_types = ['string']
re_dictmap = re.compile("%dict\-map(.*)")
re_dictmap_constructor = re.compile("%constructor (.+)")
def __init__(self):
self.attr_list = []
self.current_type = None
self.operation = None
self.constructor = None
self.type_handlers = {}
for type in self.supported_types:
self.type_handlers[type] = (eval('self.handle_%s_marsh' % type),
eval('self.handle_%s_demarsh' % type))
def handle_string_marsh(self, attribute):
"""Handle marshalling of string item from the dictionary."""
return ["if (%s && !PyString_Check(%s)) return false;" % (attribute, attribute),
"if (%s) { qobj.%s(TQString(PyString_AsString(%s)));" % (attribute, set_method(attribute), attribute),
"PyDict_DelItemString(dict,(char*)\"%s\"); } " % (attribute)]
def handle_string_demarsh(self, attribute):
"""Handle demarshalling of string items into the dictionary."""
return ["PyObject *%s = PyString_FromString(qobj.%s().utf8().data() );" % (attribute ,attribute),
"PyDict_SetItemString(dict, (char*)\"%s\", %s);" % (attribute, attribute)
]
def pre_code_for(self, operation, attribute):
if operation==MARSHAL:
return ["PyObject *%s = PyDict_GetItemString(dict,(char*)\"%s\");" % (attribute, attribute) ]
return []
def post_code_for(self, operation, attribute):
return []
def code_for(self, operation, type, attribute):
if operation!=None and (type in self.type_handlers):
return self.pre_code_for(operation, attribute) + \
self.type_handlers[type][not not operation](attribute) + \
self.post_code_for(operation, attribute)
return []
def set_current_type(self, current_type):
self.current_type = current_type
self.constructor = "";
def set_operation(self, operation):
if operation in [None, MARSHAL, DEMARSHAL]:
self.operation = operation
def check_dictmap(self, line):
if self.operation not in [MARSHAL,DEMARSHAL]: return []
m=self.re_dictmap_constructor.match(line)
if m:
self.constructor = m.groups()[0]
return ['']
m=self.re_dictmap.match(line)
if not m: return []
if self.operation==MARSHAL:
result = ["{",
"if (!PyDict_Check(obj)) return false;",
"%s qobj%s;" % (self.current_type,self.constructor),
"PyObject *dict = PyDict_Copy(obj);"
]
if self.operation==DEMARSHAL:
result = ["{",
"PyObject *dict = PyDict_New();",
"if (!dict) return NULL;",
"%s qobj%s;" % (self.current_type,self.constructor),
"(*str) >> qobj;"
]
if m.groups()[0].strip():
self.attr_list = [tuple(x.split(':')) for x in m.groups()[0].strip().split(',') ]
for attribute, type in self.attr_list:
result += self.code_for(self.operation, type, attribute)
if self.operation==MARSHAL:
result += ["if (str) (*str) << qobj;",
"Py_DECREF(dict);",
"return true;",
"}"
]
if self.operation==DEMARSHAL:
result += ["return dict;",
"}"
]
return result
class DocType:
"""A class to hold documentation information for each type."""
def __init__(self, type):
self.type = type
self.demarshal_asme = None
self.asme = []
self.info = []
def add_asme(self, asme):
if self.demarshal_asme == None: self.demarshal_asme = asme
self.asme += [asme]
def add_info(self,info):
self.info += [info]
def xml(self):
return ['<type dcoptype="%s">' % self.type,
' <demarshal-asme>%s</demarshal-asme>' % self.demarshal_asme] + \
[' <marshal-asme>%s</marshal-asme>' % asme for asme in self.asme ] + \
[' <info>%s</info>' % info for info in self.info ] + \
['</type>']
MARSHAL, DEMARSHAL, TOPYOBJ, FROMPYOBJ = 0,1,2,3
if len(sys.argv)!=4:
print "Use: gen_marshal_code.py <input file> <output file> <doc-xml-output file>"
raise RuntimeError
nowt, in_name, code_name, doc_xml_name = tuple(sys.argv)
##in_name, code_name, doc_xml_name = "marshal_funcs.data", "marshal_funcs.h", "marshal_funcs_doc.xml"
gen_code_comments = ['/*',
' * This code was generated by gen_marshal_code.py',
' * Please do not modify, or it\'ll be overwritten!',
' */',
' ',
]
re_type = re.compile(r"type\: *([^\s]+).*")
re_marshDemarsh = re.compile("%% *(de)?marshal *.*")
re_tofromPyobj = re.compile("%% *(to|from)_pyobj *.*")
re_defaultCode = re.compile("%defaultcode *.*")
re_docInfo = re.compile("%doc *([^ ]+) *(.*)")
in_file = open(in_name,"r")
code = []
types = {}
doc_types = {}
current_operation = None
dict_maker = DictMaker()
for l in in_file.readlines():
l=l[:-1]
# match a "type:" line
m=re_type.match(l)
if m:
current_type = m.groups()[0]
types[current_type]={}
doc_types[current_type] = DocType(current_type)
dict_maker.set_current_type(current_type)
continue
m=re_docInfo.match(l)
if m:
doc_cmd, rest = m.groups()
if doc_cmd=="as":
doc_types[current_type].add_asme(rest)
if doc_cmd=="info":
doc_types[current_type].add_info(rest)
continue
# match a "%% marshal" or "%% demarshal" line
m=re_marshDemarsh.match(l)
if m:
if m.groups()[0]:
current_operation = DEMARSHAL
code.append("PyObject *demarshal_" + current_type + \
"(TQDataStream *str)")
else:
current_operation = MARSHAL
code.append("bool marshal_" + current_type + \
"(PyObject *obj, TQDataStream *str)")
dict_maker.set_operation(current_operation)
continue
m=re_tofromPyobj.match(l)
if m:
if m.groups()[0]=='to':
current_operation = TOPYOBJ
code += ["PyObject *toPyObject_%s(%s val)" % (current_type,current_type)]
elif m.groups()[0]=='from':
current_operation = FROMPYOBJ
code += ["%s fromPyObject_%s(PyObject *obj, bool *ok)" % (current_type,current_type)]
continue
if l.strip()=='%%':
current_operation = None
dict_maker.set_operation(current_operation)
if current_operation!=None:
types[current_type][current_operation]=1
dict_code = dict_maker.check_dictmap(l)
if dict_code:
code += dict_code
continue
m=re_defaultCode.match(l)
if m:
if current_operation==MARSHAL:
code += [
"{",
" bool ok;",
" %s qobj=fromPyObject_%s(obj,&ok);" % (current_type,current_type),
" if (ok && str) (*str) << qobj;",
" return ok;",
"}"
]
continue
if current_operation==DEMARSHAL:
code += [
"{",
" %s qobj;" % current_type,
" (*str) >> qobj;",
" return toPyObject_%s(qobj);" % current_type,
"}"
]
continue
code.append(l)
in_file.close()
code.append("void Marshaller::initFuncs() {")
for t in types:
if MARSHAL in types[t]:
code.append("m_marsh_funcs[\"" + t + "\"]=marshal_" + t + ";")
if DEMARSHAL in types[t]:
code.append("m_demarsh_funcs[\"" + t + "\"]=demarshal_" + t + ";")
code.append("}")
out_file = open(code_name,"w")
out_file.writelines([x + '\n' for x in gen_code_comments])
out_file.writelines([x + '\n' for x in code])
out_file.close()
xml_file = file(doc_xml_name,"w")
print >>xml_file, '<?xml version="1.0" ?>'
print >>xml_file, '<!-- This file was auto-generated by gen_marshal_code.py. Changes will be lost! -->'
print >>xml_file, "<types>"
[ [xml_file.write(x+"\n") for x in doc.xml()] for doc in doc_types.values() ] # silly one-liner
print >>xml_file, "</types>"
xml_file.close()