src/mapi/glapi/gen/gl_marshal.py

   1
   2 # Copyright (C) 2012 Intel Corporation
   3 #
   4 # Permission is hereby granted, free of charge, to any person obtaining a
   5 # copy of this software and associated documentation files (the "Software"),
   6 # to deal in the Software without restriction, including without limitation
   7 # the rights to use, copy, modify, merge, publish, distribute, sublicense,
   8 # and/or sell copies of the Software, and to permit persons to whom the
   9 # Software is furnished to do so, subject to the following conditions:
  10 #
  11 # The above copyright notice and this permission notice (including the next
  12 # paragraph) shall be included in all copies or substantial portions of the
  13 # Software.
  14 #
  15 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  18 # THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20 # FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  21 # IN THE SOFTWARE.
  22
  23 from __future__ import print_function
  24
  25 import contextlib
  26 import getopt
  27 import gl_XML
  28 import license
  29 import marshal_XML
  30 import sys
  31
  32 header = """
  33 #include "api_exec.h"
  34 #include "glthread_marshal.h"
  35 #include "bufferobj.h"
  36 #include "dispatch.h"
  37
  38 #define COMPAT (ctx->API != API_OPENGL_CORE)
  39
  40 static inline int safe_mul(int a, int b)
  41 {
  42     if (a < 0 || b < 0) return -1;
  43     if (a == 0 || b == 0) return 0;
  44     if (a > INT_MAX / b) return -1;
  45     return a * b;
  46 }
  47 """
  48
  49
  50 file_index = 0
  51 file_count = 1
  52 current_indent = 0
  53
  54
  55 def out(str):
  56     if str:
  57         print(' '*current_indent + str)
  58     else:
  59         print('')
  60
  61
  62 @contextlib.contextmanager
  63 def indent(delta = 3):
  64     global current_indent
  65     current_indent += delta
  66     yield
  67     current_indent -= delta
  68
  69
  70 class PrintCode(gl_XML.gl_print_base):
  71     def __init__(self):
  72         super(PrintCode, self).__init__()
  73
  74         self.name = 'gl_marshal.py'
  75         self.license = license.bsd_license_template % (
  76             'Copyright (C) 2012 Intel Corporation', 'INTEL CORPORATION')
  77
  78     def printRealHeader(self):
  79         print(header)
  80
  81     def printRealFooter(self):
  82         pass
  83
  84     def print_sync_call(self, func, unmarshal = 0):
  85         call = 'CALL_{0}(ctx->CurrentServerDispatch, ({1}))'.format(
  86             func.name, func.get_called_parameter_string())
  87         if func.return_type == 'void':
  88             out('{0};'.format(call))
  89             if func.marshal_call_after and not unmarshal:
  90                 out(func.marshal_call_after);
  91         else:
  92             out('return {0};'.format(call))
  93             assert not func.marshal_call_after
  94
  95     def print_sync_dispatch(self, func):
  96         self.print_sync_call(func)
  97
  98     def print_sync_body(self, func):
  99         out('/* {0}: marshalled synchronously */'.format(func.name))
 100         out('{0} GLAPIENTRY'.format(func.return_type))
 101         out('_mesa_marshal_{0}({1})'.format(func.name, func.get_parameter_string()))
 102         out('{')
 103         with indent():
 104             out('GET_CURRENT_CONTEXT(ctx);')
 105             out('_mesa_glthread_finish_before(ctx, "{0}");'.format(func.name))
 106             self.print_sync_call(func)
 107         out('}')
 108         out('')
 109         out('')
 110
 111     def print_async_dispatch(self, func):
 112         out('cmd = _mesa_glthread_allocate_command(ctx, '
 113             'DISPATCH_CMD_{0}, cmd_size);'.format(func.name))
 114         for p in func.fixed_params:
 115             if p.count:
 116                 out('memcpy(cmd->{0}, {0}, {1});'.format(
 117                         p.name, p.size_string()))
 118             else:
 119                 out('cmd->{0} = {0};'.format(p.name))
 120         if func.variable_params:
 121             out('char *variable_data = (char *) (cmd + 1);')
 122             i = 1
 123             for p in func.variable_params:
 124                 if p.img_null_flag:
 125                     out('cmd->{0}_null = !{0};'.format(p.name))
 126                     out('if (!cmd->{0}_null) {{'.format(p.name))
 127                     with indent():
 128                         out(('memcpy(variable_data, {0}, {0}_size);').format(p.name))
 129                         if i < len(func.variable_params):
 130                             out('variable_data += {0}_size;'.format(p.name))
 131                     out('}')
 132                 else:
 133                     out(('memcpy(variable_data, {0}, {0}_size);').format(p.name))
 134                     if i < len(func.variable_params):
 135                         out('variable_data += {0}_size;'.format(p.name))
 136                 i += 1
 137
 138         if not func.fixed_params and not func.variable_params:
 139             out('(void) cmd;')
 140
 141         if func.marshal_call_after:
 142             out(func.marshal_call_after);
 143
 144         # Uncomment this if you want to call _mesa_glthread_finish for debugging
 145         #out('_mesa_glthread_finish(ctx);')
 146
 147     def get_type_size(self, str):
 148         if str.find('*') != -1:
 149             return 8;
 150
 151         mapping = {
 152             'GLboolean': 1,
 153             'GLbyte': 1,
 154             'GLubyte': 1,
 155             'GLenum': 2, # uses GLenum16
 156             'GLshort': 2,
 157             'GLushort': 2,
 158             'GLint': 4,
 159             'GLuint': 4,
 160             'GLbitfield': 4,
 161             'GLsizei': 4,
 162             'GLfloat': 4,
 163             'GLclampf': 4,
 164             'GLfixed': 4,
 165             'GLclampx': 4,
 166             'GLhandleARB': 4,
 167             'int': 4,
 168             'float': 4,
 169             'GLdouble': 8,
 170             'GLclampd': 8,
 171             'GLintptr': 8,
 172             'GLsizeiptr': 8,
 173             'GLint64': 8,
 174             'GLuint64': 8,
 175             'GLuint64EXT': 8,
 176             'GLsync': 8,
 177         }
 178         val = mapping.get(str, 9999)
 179         if val == 9999:
 180             print('Unhandled type in gl_marshal.py.get_type_size: ' + str, file=sys.stderr)
 181         return val
 182
 183     def print_async_struct(self, func):
 184         out('struct marshal_cmd_{0}'.format(func.name))
 185         out('{')
 186         with indent():
 187             out('struct marshal_cmd_base cmd_base;')
 188
 189             # Sort the parameters according to their size to pack the structure optimally
 190             for p in sorted(func.fixed_params, key=lambda p: self.get_type_size(p.type_string())):
 191                 if p.count:
 192                     out('{0} {1}[{2}];'.format(
 193                             p.get_base_type_string(), p.name, p.count))
 194                 else:
 195                     type = p.type_string()
 196                     if type == 'GLenum':
 197                         type = 'GLenum16'
 198                     out('{0} {1};'.format(type, p.name))
 199
 200             for p in func.variable_params:
 201                 if p.img_null_flag:
 202                     out('bool {0}_null; /* If set, no data follows '
 203                         'for "{0}" */'.format(p.name))
 204
 205             for p in func.variable_params:
 206                 if p.count_scale != 1:
 207                     out(('/* Next {0} bytes are '
 208                          '{1} {2}[{3}][{4}] */').format(
 209                             p.size_string(marshal = 1), p.get_base_type_string(),
 210                             p.name, p.counter, p.count_scale))
 211                 else:
 212                     out(('/* Next {0} bytes are '
 213                          '{1} {2}[{3}] */').format(
 214                             p.size_string(marshal = 1), p.get_base_type_string(),
 215                             p.name, p.counter))
 216         out('};')
 217
 218     def print_async_unmarshal(self, func):
 219         out('void')
 220         out(('_mesa_unmarshal_{0}(struct gl_context *ctx, '
 221              'const struct marshal_cmd_{0} *cmd)').format(func.name))
 222         out('{')
 223         with indent():
 224             for p in func.fixed_params:
 225                 if p.count:
 226                     p_decl = '{0} * {1} = cmd->{1};'.format(
 227                             p.get_base_type_string(), p.name)
 228                 else:
 229                     p_decl = '{0} {1} = cmd->{1};'.format(
 230                             p.type_string(), p.name)
 231
 232                 if not p_decl.startswith('const '):
 233                     # Declare all local function variables as const, even if
 234                     # the original parameter is not const.
 235                     p_decl = 'const ' + p_decl
 236
 237                 out(p_decl)
 238
 239             if func.variable_params:
 240                 for p in func.variable_params:
 241                     out('{0} * {1};'.format(
 242                             p.get_base_type_string(), p.name))
 243                 out('const char *variable_data = (const char *) (cmd + 1);')
 244                 i = 1
 245                 for p in func.variable_params:
 246                     out('{0} = ({1} *) variable_data;'.format(
 247                             p.name, p.get_base_type_string()))
 248
 249                     if p.img_null_flag:
 250                         out('if (cmd->{0}_null)'.format(p.name))
 251                         with indent():
 252                             out('{0} = NULL;'.format(p.name))
 253                         if i < len(func.variable_params):
 254                             out('else')
 255                             with indent():
 256                                 out('variable_data += {0};'.format(p.size_string(False, marshal = 1)))
 257                     elif i < len(func.variable_params):
 258                         out('variable_data += {0};'.format(p.size_string(False, marshal = 1)))
 259                     i += 1
 260
 261             self.print_sync_call(func, unmarshal = 1)
 262         out('}')
 263
 264     def validate_count_or_fallback(self, func):
 265         # Check that any counts for variable-length arguments might be < 0, in
 266         # which case the command alloc or the memcpy would blow up before we
 267         # get to the validation in Mesa core.
 268         list = []
 269         for p in func.parameters:
 270             if p.is_variable_length():
 271                 list.append('{0}_size < 0'.format(p.name))
 272                 list.append('({0}_size > 0 && !{0})'.format(p.name))
 273
 274         if len(list) == 0:
 275             return
 276
 277         list.append('(unsigned)cmd_size > MARSHAL_MAX_CMD_SIZE')
 278
 279         out('if (unlikely({0})) {{'.format(' || '.join(list)))
 280         with indent():
 281             out('_mesa_glthread_finish_before(ctx, "{0}");'.format(func.name))
 282             self.print_sync_dispatch(func)
 283             out('return;')
 284         out('}')
 285
 286     def print_async_marshal(self, func):
 287         out('void GLAPIENTRY')
 288         out('_mesa_marshal_{0}({1})'.format(
 289                 func.name, func.get_parameter_string()))
 290         out('{')
 291         with indent():
 292             out('GET_CURRENT_CONTEXT(ctx);')
 293             for p in func.variable_params:
 294                 out('int {0}_size = {1};'.format(p.name, p.size_string(marshal = 1)))
 295
 296             struct = 'struct marshal_cmd_{0}'.format(func.name)
 297             size_terms = ['sizeof({0})'.format(struct)]
 298             for p in func.variable_params:
 299                 if p.img_null_flag:
 300                     size_terms.append('({0} ? {0}_size : 0)'.format(p.name))
 301                 else:
 302                     size_terms.append('{0}_size'.format(p.name))
 303             out('int cmd_size = {0};'.format(' + '.join(size_terms)))
 304             out('{0} *cmd;'.format(struct))
 305
 306             self.validate_count_or_fallback(func)
 307
 308             if func.marshal_sync:
 309                 out('if ({0}) {{'.format(func.marshal_sync))
 310                 with indent():
 311                     out('_mesa_glthread_finish_before(ctx, "{0}");'.format(func.name))
 312                     self.print_sync_dispatch(func)
 313                     out('return;')
 314                 out('}')
 315
 316         with indent():
 317             self.print_async_dispatch(func)
 318         out('}')
 319
 320     def print_async_body(self, func):
 321         out('/* {0}: marshalled asynchronously */'.format(func.name))
 322         self.print_async_struct(func)
 323         self.print_async_unmarshal(func)
 324         self.print_async_marshal(func)
 325         out('')
 326         out('')
 327
 328     def print_unmarshal_dispatch_cmd(self, api):
 329         out('const _mesa_unmarshal_func _mesa_unmarshal_dispatch[NUM_DISPATCH_CMD] = {')
 330         with indent():
 331             for func in api.functionIterateAll():
 332                 flavor = func.marshal_flavor()
 333                 if flavor in ('skip', 'sync'):
 334                     continue
 335                 out('[DISPATCH_CMD_{0}] = (_mesa_unmarshal_func)_mesa_unmarshal_{0},'.format(func.name))
 336         out('};')
 337         out('')
 338         out('')
 339
 340     def print_create_marshal_table(self, api):
 341         out('/* _mesa_create_marshal_table takes a long time to compile with -O2 */')
 342         out('#ifdef __GNUC__')
 343         out('__attribute__((optimize("O1")))')
 344         out('#endif')
 345         out('struct _glapi_table *')
 346         out('_mesa_create_marshal_table(const struct gl_context *ctx)')
 347         out('{')
 348         with indent():
 349             out('struct _glapi_table *table;')
 350             out('')
 351             out('table = _mesa_alloc_dispatch_table();')
 352             out('if (table == NULL)')
 353             with indent():
 354                 out('return NULL;')
 355             out('')
 356             for func in api.functionIterateAll():
 357                 if func.marshal_flavor() == 'skip':
 358                     continue
 359                 # Don't use the SET_* functions, because they increase compile time
 360                 # by 20 seconds (on Ryzen 1700X).
 361                 out('if (_gloffset_{0} >= 0)'.format(func.name))
 362                 out('   ((_glapi_proc *)(table))[_gloffset_{0}] = (_glapi_proc)_mesa_marshal_{0};'
 363                     .format(func.name))
 364             out('')
 365             out('return table;')
 366         out('}')
 367         out('')
 368         out('')
 369
 370     def printBody(self, api):
 371         # The first file only contains the dispatch tables
 372         if file_index == 0:
 373             self.print_unmarshal_dispatch_cmd(api)
 374             self.print_create_marshal_table(api)
 375             return
 376
 377         # The remaining files contain the marshal and unmarshal functions
 378         func_per_file = (len(api.functionIterateAll()) // (file_count - 1)) + 1
 379         i = -1
 380         for func in api.functionIterateAll():
 381             i += 1
 382             if i // func_per_file != (file_index - 1):
 383                 continue
 384
 385             flavor = func.marshal_flavor()
 386             if flavor in ('skip', 'custom'):
 387                 continue
 388             elif flavor == 'async':
 389                 self.print_async_body(func)
 390             elif flavor == 'sync':
 391                 self.print_sync_body(func)
 392
 393
 394 def show_usage():
 395     print('Usage: %s [-f input_file_name]' % sys.argv[0])
 396     sys.exit(1)
 397
 398
 399 if __name__ == '__main__':
 400     file_name = 'gl_API.xml'
 401
 402     try:
 403         (args, trail) = getopt.getopt(sys.argv[1:], 'm:f:i:n:')
 404     except Exception:
 405         show_usage()
 406
 407     for (arg,val) in args:
 408         if arg == '-f':
 409             file_name = val
 410         elif arg == '-i':
 411             file_index = int(val)
 412         elif arg == '-n':
 413             file_count = int(val)
 414
 415     assert file_index < file_count
 416     printer = PrintCode()
 417
 418     api = gl_XML.parse_GL_API(file_name, marshal_XML.marshal_item_factory())
 419     printer.Print(api)