# coding=utf-8
#
# Copyright © 2015, 2017 Intel Corporation
#
# Permission is hereby granted, free of charge, to any person obtaining a
# copy of this software and associated documentation files (the "Software"),
# to deal in the Software without restriction, including without limitation
# the rights to use, copy, modify, merge, publish, distribute, sublicense,
# and/or sell copies of the Software, and to permit persons to whom the
# Software is furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice (including the next
# paragraph) shall be included in all copies or substantial portions of the
# Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
# IN THE SOFTWARE.
#

import argparse
import functools
import os
import textwrap
import xml.etree.cElementTree as et

from mako.template import Template

MAX_API_VERSION = 1.0

SUPPORTED_EXTENSIONS = [
    'VK_AMD_draw_indirect_count',
    'VK_NV_dedicated_allocation',
    'VK_KHR_descriptor_update_template',
    'VK_KHR_get_physical_device_properties2',
    'VK_KHR_incremental_present',
    'VK_KHR_maintenance1',
    'VK_KHR_push_descriptor',
    'VK_KHR_sampler_mirror_clamp_to_edge',
    'VK_KHR_shader_draw_parameters',
    'VK_KHR_surface',
    'VK_KHR_swapchain',
    'VK_KHR_wayland_surface',
    'VK_KHR_xcb_surface',
    'VK_KHR_xlib_surface',
    'VK_KHR_get_memory_requirements2',
    'VK_KHR_dedicated_allocation',
    'VK_KHR_external_memory_capabilities',
    'VK_KHR_external_memory',
    'VK_KHR_external_memory_fd',
    'VK_KHR_storage_buffer_storage_class',
    'VK_KHR_variable_pointers',
    'VK_KHR_external_semaphore_capabilities',
    'VK_KHR_external_semaphore',
    'VK_KHR_external_semaphore_fd',
    'VK_KHR_bind_memory2',
]

# We generate a static hash table for entry point lookup
# (vkGetProcAddress). We use a linear congruential generator for our hash
# function and a power-of-two size table. The prime numbers are determined
# experimentally.

TEMPLATE_H = Template(textwrap.dedent("""\
    /* This file generated from ${filename}, don't edit directly. */

    struct radv_dispatch_table {
       union {
          void *entrypoints[${len(entrypoints)}];
          struct {
          % for _, name, _, _, _, guard in entrypoints:
            % if guard is not None:
    #ifdef ${guard}
              PFN_vk${name} ${name};
    #else
              void *${name};
    # endif
            % else:
              PFN_vk${name} ${name};
            % endif
          % endfor
          };
       };
    };

    % for type_, name, args, num, h, guard in entrypoints:
      % if guard is not None:
    #ifdef ${guard}
      % endif
      ${type_} radv_${name}(${args});
      % if guard is not None:
    #endif // ${guard}
      % endif
    % endfor
    """), output_encoding='utf-8')

TEMPLATE_C = Template(textwrap.dedent(u"""\
    /*
     * Copyright © 2015 Intel Corporation
     *
     * Permission is hereby granted, free of charge, to any person obtaining a
     * copy of this software and associated documentation files (the "Software"),
     * to deal in the Software without restriction, including without limitation
     * the rights to use, copy, modify, merge, publish, distribute, sublicense,
     * and/or sell copies of the Software, and to permit persons to whom the
     * Software is furnished to do so, subject to the following conditions:
     *
     * The above copyright notice and this permission notice (including the next
     * paragraph) shall be included in all copies or substantial portions of the
     * Software.
     *
     * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
     * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
     * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
     * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
     * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
     * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
     * IN THE SOFTWARE.
     */

    /* This file generated from ${filename}, don't edit directly. */

    #include "radv_private.h"

    struct radv_entrypoint {
       uint32_t name;
       uint32_t hash;
    };

    /* We use a big string constant to avoid lots of reloctions from the entry
     * point table to lots of little strings. The entries in the entry point table
     * store the index into this big string.
     */

    static const char strings[] =
    % for _, name, _, _, _, _ in entrypoints:
        "vk${name}\\0"
    % endfor
    ;

    static const struct radv_entrypoint entrypoints[] = {
    % for _, _, _, num, h, _ in entrypoints:
        { ${offsets[num]}, ${'{:0=#8x}'.format(h)} },
    % endfor
    };

    /* Weak aliases for all potential implementations. These will resolve to
     * NULL if they're not defined, which lets the resolve_entrypoint() function
     * either pick the correct entry point.
     */

    % for layer in ['radv']:
      % for type_, name, args, _, _, guard in entrypoints:
        % if guard is not None:
    #ifdef ${guard}
        % endif
        ${type_} ${layer}_${name}(${args}) __attribute__ ((weak));
        % if guard is not None:
    #endif // ${guard}
        % endif
      % endfor

      const struct radv_dispatch_table ${layer}_layer = {
      % for _, name, args, _, _, guard in entrypoints:
        % if guard is not None:
    #ifdef ${guard}
        % endif
        .${name} = ${layer}_${name},
        % if guard is not None:
    #endif // ${guard}
        % endif
      % endfor
      };
    % endfor

    static void * __attribute__ ((noinline))
    radv_resolve_entrypoint(uint32_t index)
    {
       return radv_layer.entrypoints[index];
    }

    /* Hash table stats:
     * size ${hash_size} entries
     * collisions entries:
    % for i in xrange(10):
     *     ${i}${'+' if i == 9 else ''}     ${collisions[i]}
    % endfor
     */

    #define none ${'{:#x}'.format(none)}
    static const uint16_t map[] = {
    % for i in xrange(0, hash_size, 8):
      % for j in xrange(i, i + 8):
        ## This is 6 because the 0x is counted in the length
        % if mapping[j] & 0xffff == 0xffff:
          none,
        % else:
          ${'{:0=#6x}'.format(mapping[j] & 0xffff)},
        % endif
      % endfor
    % endfor
    };

    void *
    radv_lookup_entrypoint(const char *name)
    {
       static const uint32_t prime_factor = ${prime_factor};
       static const uint32_t prime_step = ${prime_step};
       const struct radv_entrypoint *e;
       uint32_t hash, h, i;
       const char *p;

       hash = 0;
       for (p = name; *p; p++)
          hash = hash * prime_factor + *p;

       h = hash;
       do {
          i = map[h & ${hash_mask}];
          if (i == none)
             return NULL;
          e = &entrypoints[i];
          h += prime_step;
       } while (e->hash != hash);

       if (strcmp(name, strings + e->name) != 0)
          return NULL;

       return radv_resolve_entrypoint(i);
    }"""), output_encoding='utf-8')

NONE = 0xffff
HASH_SIZE = 256
U32_MASK = 2**32 - 1
HASH_MASK = HASH_SIZE - 1

PRIME_FACTOR = 5024183
PRIME_STEP = 19


def cal_hash(name):
    """Calculate the same hash value that Mesa will calculate in C."""
    return functools.reduce(
        lambda h, c: (h * PRIME_FACTOR + ord(c)) & U32_MASK, name, 0)


def get_entrypoints(doc, entrypoints_to_defines):
    """Extract the entry points from the registry."""
    entrypoints = []

    enabled_commands = set()
    for feature in doc.findall('./feature'):
        assert feature.attrib['api'] == 'vulkan'
        if float(feature.attrib['number']) > MAX_API_VERSION:
            continue

        for command in feature.findall('./require/command'):
            enabled_commands.add(command.attrib['name'])

    for extension in doc.findall('.extensions/extension'):
        if extension.attrib['name'] not in SUPPORTED_EXTENSIONS:
            continue

        assert extension.attrib['supported'] == 'vulkan'
        for command in extension.findall('./require/command'):
            enabled_commands.add(command.attrib['name'])

    index = 0
    for command in doc.findall('./commands/command'):
        type = command.find('./proto/type').text
        fullname = command.find('./proto/name').text

        if fullname not in enabled_commands:
            continue

        shortname = fullname[2:]
        params = (''.join(p.itertext()) for p in command.findall('./param'))
        params = ', '.join(params)
        guard = entrypoints_to_defines.get(fullname)
        entrypoints.append((type, shortname, params, index, cal_hash(fullname), guard))
        index += 1

    return entrypoints


def get_entrypoints_defines(doc):
    """Maps entry points to extension defines."""
    entrypoints_to_defines = {}

    for extension in doc.findall('./extensions/extension[@protect]'):
        define = extension.attrib['protect']

        for entrypoint in extension.findall('./require/command'):
            fullname = entrypoint.attrib['name']
            entrypoints_to_defines[fullname] = define

    return entrypoints_to_defines


def gen_code(entrypoints):
    """Generate the C code."""
    i = 0
    offsets = []
    for _, name, _, _, _, _ in entrypoints:
        offsets.append(i)
        i += 2 + len(name) + 1

    mapping = [NONE] * HASH_SIZE
    collisions = [0] * 10
    for _, name, _, num, h, _ in entrypoints:
        level = 0
        while mapping[h & HASH_MASK] != NONE:
            h = h + PRIME_STEP
            level = level + 1
        if level > 9:
            collisions[9] += 1
        else:
            collisions[level] += 1
        mapping[h & HASH_MASK] = num

    return TEMPLATE_C.render(entrypoints=entrypoints,
                             offsets=offsets,
                             collisions=collisions,
                             mapping=mapping,
                             hash_mask=HASH_MASK,
                             prime_step=PRIME_STEP,
                             prime_factor=PRIME_FACTOR,
                             none=NONE,
                             hash_size=HASH_SIZE,
                             filename=os.path.basename(__file__))


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--outdir', help='Where to write the files.',
                        required=True)
    parser.add_argument('--xml', help='Vulkan API XML file.', required=True)
    args = parser.parse_args()

    doc = et.parse(args.xml)
    entrypoints = get_entrypoints(doc, get_entrypoints_defines(doc))

    with open(os.path.join(args.outdir, 'radv_entrypoints.h'), 'wb') as f:
        f.write(TEMPLATE_H.render(entrypoints=entrypoints,
                                  filename=os.path.basename(__file__)))
    with open(os.path.join(args.outdir, 'radv_entrypoints.c'), 'wb') as f:
        f.write(gen_code(entrypoints))


if __name__ == '__main__':
    main()