xref: /qemu/scripts/qapi/visit.py (revision 19bf7c87081835449d5683ecb0858255bf5a0546) 
1#
2# QAPI visitor generator
3#
4# Copyright IBM, Corp. 2011
5#
6# Authors:
7#  Anthony Liguori <aliguori@us.ibm.com>
8#  Michael Roth    <mdroth@linux.vnet.ibm.com>
9#
10# This work is licensed under the terms of the GNU GPLv2.
11# See the COPYING.LIB file in the top-level directory.
12
13from ordereddict import OrderedDict
14from qapi import *
15import sys
16import os
17import getopt
18import errno
19
20def generate_visit_struct_body(field_prefix, members):
21    ret = ""
22    if len(field_prefix):
23        field_prefix = field_prefix + "."
24    for argname, argentry, optional, structured in parse_args(members):
25        if optional:
26            ret += mcgen('''
27visit_start_optional(m, (obj && *obj) ? &(*obj)->%(c_prefix)shas_%(c_name)s : NULL, "%(name)s", errp);
28if ((*obj)->%(prefix)shas_%(c_name)s) {
29''',
30                         c_prefix=c_var(field_prefix), prefix=field_prefix,
31                         c_name=c_var(argname), name=argname)
32            push_indent()
33
34        if structured:
35            ret += mcgen('''
36visit_start_struct(m, NULL, "", "%(name)s", 0, errp);
37''',
38                         name=argname)
39            ret += generate_visit_struct_body(field_prefix + argname, argentry)
40            ret += mcgen('''
41visit_end_struct(m, errp);
42''')
43        else:
44            ret += mcgen('''
45visit_type_%(type)s(m, (obj && *obj) ? &(*obj)->%(c_prefix)s%(c_name)s : NULL, "%(name)s", errp);
46''',
47                         c_prefix=c_var(field_prefix), prefix=field_prefix,
48                         type=type_name(argentry), c_name=c_var(argname),
49                         name=argname)
50
51        if optional:
52            pop_indent()
53            ret += mcgen('''
54}
55visit_end_optional(m, errp);
56''')
57    return ret
58
59def generate_visit_struct(name, members):
60    ret = mcgen('''
61
62void visit_type_%(name)s(Visitor *m, %(name)s ** obj, const char *name, Error **errp)
63{
64    visit_start_struct(m, (void **)obj, "%(name)s", name, sizeof(%(name)s), errp);
65''',
66                name=name)
67    push_indent()
68    ret += generate_visit_struct_body("", members)
69    pop_indent()
70
71    ret += mcgen('''
72    visit_end_struct(m, errp);
73}
74''')
75    return ret
76
77def generate_visit_list(name, members):
78    return mcgen('''
79
80void visit_type_%(name)sList(Visitor *m, %(name)sList ** obj, const char *name, Error **errp)
81{
82    GenericList *i, **head = (GenericList **)obj;
83
84    visit_start_list(m, name, errp);
85
86    for (*head = i = visit_next_list(m, head, errp); i; i = visit_next_list(m, &i, errp)) {
87        %(name)sList *native_i = (%(name)sList *)i;
88        visit_type_%(name)s(m, &native_i->value, NULL, errp);
89    }
90
91    visit_end_list(m, errp);
92}
93''',
94                name=name)
95
96def generate_visit_enum(name, members):
97    return mcgen('''
98
99void visit_type_%(name)s(Visitor *m, %(name)s * obj, const char *name, Error **errp)
100{
101    visit_type_enum(m, (int *)obj, %(name)s_lookup, "%(name)s", name, errp);
102}
103''',
104                 name=name)
105
106def generate_visit_union(name, members):
107    ret = generate_visit_enum('%sKind' % name, members.keys())
108
109    ret += mcgen('''
110
111void visit_type_%(name)s(Visitor *m, %(name)s ** obj, const char *name, Error **errp)
112{
113}
114''',
115                 name=name)
116
117    return ret
118
119def generate_declaration(name, members, genlist=True):
120    ret = mcgen('''
121
122void visit_type_%(name)s(Visitor *m, %(name)s ** obj, const char *name, Error **errp);
123''',
124                name=name)
125
126    if genlist:
127        ret += mcgen('''
128void visit_type_%(name)sList(Visitor *m, %(name)sList ** obj, const char *name, Error **errp);
129''',
130                 name=name)
131
132    return ret
133
134def generate_decl_enum(name, members, genlist=True):
135    return mcgen('''
136
137void visit_type_%(name)s(Visitor *m, %(name)s * obj, const char *name, Error **errp);
138''',
139                name=name)
140
141try:
142    opts, args = getopt.gnu_getopt(sys.argv[1:], "chp:o:",
143                                   ["source", "header", "prefix=", "output-dir="])
144except getopt.GetoptError, err:
145    print str(err)
146    sys.exit(1)
147
148output_dir = ""
149prefix = ""
150c_file = 'qapi-visit.c'
151h_file = 'qapi-visit.h'
152
153do_c = False
154do_h = False
155
156for o, a in opts:
157    if o in ("-p", "--prefix"):
158        prefix = a
159    elif o in ("-o", "--output-dir"):
160        output_dir = a + "/"
161    elif o in ("-c", "--source"):
162        do_c = True
163    elif o in ("-h", "--header"):
164        do_h = True
165
166if not do_c and not do_h:
167    do_c = True
168    do_h = True
169
170c_file = output_dir + prefix + c_file
171h_file = output_dir + prefix + h_file
172
173try:
174    os.makedirs(output_dir)
175except os.error, e:
176    if e.errno != errno.EEXIST:
177        raise
178
179def maybe_open(really, name, opt):
180    if really:
181        return open(name, opt)
182    else:
183        import StringIO
184        return StringIO.StringIO()
185
186fdef = maybe_open(do_c, c_file, 'w')
187fdecl = maybe_open(do_h, h_file, 'w')
188
189fdef.write(mcgen('''
190/* THIS FILE IS AUTOMATICALLY GENERATED, DO NOT MODIFY */
191
192/*
193 * schema-defined QAPI visitor functions
194 *
195 * Copyright IBM, Corp. 2011
196 *
197 * Authors:
198 *  Anthony Liguori   <aliguori@us.ibm.com>
199 *
200 * This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
201 * See the COPYING.LIB file in the top-level directory.
202 *
203 */
204
205#include "%(header)s"
206''',
207                 header=basename(h_file)))
208
209fdecl.write(mcgen('''
210/* THIS FILE IS AUTOMATICALLY GENERATED, DO NOT MODIFY */
211
212/*
213 * schema-defined QAPI visitor function
214 *
215 * Copyright IBM, Corp. 2011
216 *
217 * Authors:
218 *  Anthony Liguori   <aliguori@us.ibm.com>
219 *
220 * This work is licensed under the terms of the GNU LGPL, version 2.1 or later.
221 * See the COPYING.LIB file in the top-level directory.
222 *
223 */
224
225#ifndef %(guard)s
226#define %(guard)s
227
228#include "qapi/qapi-visit-core.h"
229#include "%(prefix)sqapi-types.h"
230''',
231                  prefix=prefix, guard=guardname(h_file)))
232
233exprs = parse_schema(sys.stdin)
234
235for expr in exprs:
236    if expr.has_key('type'):
237        ret = generate_visit_struct(expr['type'], expr['data'])
238        ret += generate_visit_list(expr['type'], expr['data'])
239        fdef.write(ret)
240
241        ret = generate_declaration(expr['type'], expr['data'])
242        fdecl.write(ret)
243    elif expr.has_key('union'):
244        ret = generate_visit_union(expr['union'], expr['data'])
245        fdef.write(ret)
246
247        ret = generate_decl_enum('%sKind' % expr['union'], expr['data'].keys())
248        ret += generate_declaration(expr['union'], expr['data'])
249        fdecl.write(ret)
250    elif expr.has_key('enum'):
251        ret = generate_visit_enum(expr['enum'], expr['data'])
252        fdef.write(ret)
253
254        ret = generate_decl_enum(expr['enum'], expr['data'])
255        fdecl.write(ret)
256
257fdecl.write('''
258#endif
259''')
260
261fdecl.flush()
262fdecl.close()
263
264fdef.flush()
265fdef.close()
266