tools/spec: pass files rather than paths to write_foo functions
This commit is contained in:
parent
336e9632d3
commit
deca2ded59
301
tools/spec.py
301
tools/spec.py
@ -2,7 +2,7 @@
|
||||
|
||||
import sys
|
||||
import logging
|
||||
from typing import List, Dict, Set
|
||||
from typing import List, Dict, Set, TextIO
|
||||
|
||||
import xml.etree.ElementTree
|
||||
|
||||
@ -699,167 +699,167 @@ def parse_extensions(reg: Registry, root):
|
||||
|
||||
|
||||
###############################################################################
|
||||
def write_header(path: str, q: List[Type], reg: Registry):
|
||||
with open(path, 'w') as dst:
|
||||
dst.write("#pragma once\n")
|
||||
def write_header(dst: TextIO, q: List[Type], reg: Registry):
|
||||
dst.write("#pragma once\n")
|
||||
|
||||
# Write the declarations and definitions for all types.
|
||||
for obj in q:
|
||||
dst.write(obj.declare())
|
||||
dst.write('\n')
|
||||
dst.write(obj.define(reg))
|
||||
dst.write('\n')
|
||||
# Write the declarations and definitions for all types.
|
||||
for obj in q:
|
||||
dst.write(obj.declare())
|
||||
dst.write('\n')
|
||||
dst.write(obj.define(reg))
|
||||
dst.write('\n')
|
||||
|
||||
# Define the default case for device and instance type traits.
|
||||
dst.write("""
|
||||
#include <type_traits>
|
||||
# Define the default case for device and instance type traits.
|
||||
dst.write("""
|
||||
#include <type_traits>
|
||||
|
||||
/// A type trait that tests if a Vulkan type is an instance type
|
||||
template <typename NativeT>
|
||||
struct is_instance:
|
||||
public std::false_type
|
||||
{};
|
||||
/// A type trait that tests if a Vulkan type is an instance type
|
||||
template <typename NativeT>
|
||||
struct is_instance:
|
||||
public std::false_type
|
||||
{};
|
||||
|
||||
|
||||
/// A type trait that tests if a Vulkan type is a device type
|
||||
template <typename NativeT>
|
||||
struct is_device:
|
||||
public std::false_type
|
||||
{};
|
||||
/// A type trait that tests if a Vulkan type is a device type
|
||||
template <typename NativeT>
|
||||
struct is_device:
|
||||
public std::false_type
|
||||
{};
|
||||
|
||||
|
||||
template <typename T>
|
||||
constexpr auto is_instance_v = is_instance<T>::value;
|
||||
template <typename T>
|
||||
constexpr auto is_instance_v = is_instance<T>::value;
|
||||
|
||||
template <typename T>
|
||||
constexpr auto is_device_v = is_device<T>::value;
|
||||
template <typename T>
|
||||
constexpr auto is_device_v = is_device<T>::value;
|
||||
""")
|
||||
|
||||
# Specialise traits for device and instance types.
|
||||
for obj in q:
|
||||
if not isinstance(obj, Handle):
|
||||
continue
|
||||
|
||||
device_value = "true_type" if obj.has_parent("VkDevice", reg) else "false_type"
|
||||
instance_value = "true_type" if obj.has_parent("VkInstance", reg) else "false_type"
|
||||
|
||||
dst.write(f"""
|
||||
template <> struct is_instance<{obj.name}>: public std::{instance_value} {{ }};
|
||||
template <> struct is_device<{obj.name}>: public std::{device_value} {{ }};
|
||||
""")
|
||||
|
||||
# Specialise traits for device and instance types.
|
||||
for obj in q:
|
||||
if not isinstance(obj, Handle):
|
||||
continue
|
||||
|
||||
device_value = "true_type" if obj.has_parent("VkDevice", reg) else "false_type"
|
||||
instance_value = "true_type" if obj.has_parent("VkInstance", reg) else "false_type"
|
||||
# -----------------------------------------------------------------------------
|
||||
def write_icd(dst: TextIO, q: List[Type], reg: Registry):
|
||||
commands = [i for i in q if isinstance(i, Command)]
|
||||
instance_commands = [i for i in commands if i.is_instance(reg)]
|
||||
device_commands = [i for i in commands if i.is_device(reg)]
|
||||
|
||||
assert len(instance_commands) + len(device_commands) == len(commands)
|
||||
|
||||
dst.write(f"""
|
||||
#include "vk.hpp"
|
||||
|
||||
#include <cruft/util/preprocessor.hpp>
|
||||
|
||||
#define MAP_COMMANDS(FUNC) MAP0(FUNC,{",".join(i.name for i in commands)})
|
||||
#define MAP_INSTANCE_COMMANDS(FUNC) MAP0(FUNC,{",".join(i.name for i in instance_commands)})
|
||||
#define MAP_DEVICE_COMMANDS(FUNC) MAP0(FUNC,{",".join(i.name for i in device_commands)})
|
||||
|
||||
namespace cruft::vk::icd {{
|
||||
class vendor;
|
||||
|
||||
struct func {{
|
||||
void *handle;
|
||||
void const *table;
|
||||
}};
|
||||
|
||||
struct instance_table {{
|
||||
instance_table (vendor &);
|
||||
""")
|
||||
|
||||
# Generate the vtable entries for instance methods
|
||||
dst.writelines((
|
||||
f"{obj.result} (*{obj.name}) ({','.join(p.param for p in obj.params)}) = nullptr;"
|
||||
for obj in instance_commands
|
||||
))
|
||||
|
||||
dst.write("""};
|
||||
struct device_table {
|
||||
""")
|
||||
|
||||
# Generate the vtable and entries for device_commands
|
||||
dst.writelines((
|
||||
f"{obj.result} (*{obj.name}) ({','.join(p.param for p in obj.params)}) = nullptr;"
|
||||
for obj in device_commands
|
||||
))
|
||||
|
||||
dst.write("""
|
||||
};
|
||||
}
|
||||
""")
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
def write_dispatch(dst: TextIO, q: List[Type], reg: Registry):
|
||||
dst.write("""
|
||||
#include "../vk.hpp"
|
||||
#include "vtable.hpp"
|
||||
|
||||
#include "icd/dispatch.hpp"
|
||||
|
||||
#include <cruft/util/debug.hpp>
|
||||
|
||||
#pragma GCC diagnostic ignored "-Wunused-parameter"
|
||||
|
||||
static cruft::vk::icd::instance_table const *i_table [[maybe_unused]] = nullptr;
|
||||
static cruft::vk::icd::device_table const *d_table [[maybe_unused]] = nullptr;
|
||||
|
||||
void (*cruft_vk_icdGetInstanceProcAddr) (
|
||||
VkInstance instance,
|
||||
const char* pName
|
||||
) = nullptr;
|
||||
|
||||
void cruft::vk::icd::init (vendor const &impl)
|
||||
{
|
||||
cruft_vk_icdGetInstanceProcAddr = impl.vtable.GetInstanceProc;
|
||||
}
|
||||
|
||||
""")
|
||||
|
||||
for obj in (i for i in q if isinstance(i, Command)):
|
||||
first_arg = reg.types[obj.params[0].type]
|
||||
|
||||
if not isinstance(first_arg, Handle):
|
||||
dst.write(f"""
|
||||
template <> struct is_instance<{obj.name}>: public std::{instance_value} {{ }};
|
||||
template <> struct is_device<{obj.name}>: public std::{device_value} {{ }};
|
||||
""")
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
def write_icd(path: str, q: List[Type], reg: Registry):
|
||||
with open(path, 'w') as icd:
|
||||
commands = [i for i in q if isinstance(i, Command)]
|
||||
instance_commands = [i for i in commands if i.is_instance(reg)]
|
||||
device_commands = [i for i in commands if i.is_device(reg)]
|
||||
|
||||
assert len(instance_commands) + len(device_commands) == len(commands)
|
||||
|
||||
icd.write(f"""
|
||||
#include "vk.hpp"
|
||||
|
||||
#include <cruft/util/preprocessor.hpp>
|
||||
|
||||
#define MAP_COMMANDS(FUNC) MAP0(FUNC,{",".join(i.name for i in commands)})
|
||||
#define MAP_INSTANCE_COMMANDS(FUNC) MAP0(FUNC,{",".join(i.name for i in instance_commands)})
|
||||
#define MAP_DEVICE_COMMANDS(FUNC) MAP0(FUNC,{",".join(i.name for i in device_commands)})
|
||||
|
||||
namespace cruft::vk::icd {{
|
||||
class vendor;
|
||||
|
||||
struct func {{
|
||||
void *handle;
|
||||
void const *table;
|
||||
}};
|
||||
|
||||
struct instance_table {{
|
||||
instance_table (vendor &);
|
||||
""")
|
||||
|
||||
for obj in instance_commands:
|
||||
icd.write(f"{obj.result} (*{obj.name}) ({','.join(p.param for p in obj.params)}) = nullptr;\n")
|
||||
|
||||
icd.write("""};
|
||||
|
||||
struct device_table {
|
||||
""")
|
||||
|
||||
for obj in device_commands:
|
||||
icd.write(f"{obj.result} (*{obj.name}) ({','.join(p.param for p in obj.params)}) = nullptr;\n")
|
||||
|
||||
icd.write("""
|
||||
};
|
||||
}
|
||||
""")
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
def write_dispatch(path: str, q: List[Type], reg: Registry):
|
||||
with open(path, 'w') as dispatch:
|
||||
dispatch.write("""
|
||||
#include "../vk.hpp"
|
||||
#include "vtable.hpp"
|
||||
|
||||
#include "icd/dispatch.hpp"
|
||||
|
||||
#include <cruft/util/debug.hpp>
|
||||
|
||||
#pragma GCC diagnostic ignored "-Wunused-parameter"
|
||||
|
||||
static cruft::vk::icd::instance_table const *i_table [[maybe_unused]] = nullptr;
|
||||
static cruft::vk::icd::device_table const *d_table [[maybe_unused]] = nullptr;
|
||||
|
||||
void (*cruft_vk_icdGetInstanceProcAddr) (
|
||||
VkInstance instance,
|
||||
const char* pName
|
||||
) = nullptr;
|
||||
|
||||
void cruft::vk::icd::init (vendor const &impl)
|
||||
{
|
||||
cruft_vk_icdGetInstanceProcAddr = impl.vtable.GetInstanceProc;
|
||||
}
|
||||
|
||||
""")
|
||||
|
||||
for obj in (i for i in q if isinstance(i, Command)):
|
||||
first_arg = reg.types[obj.params[0].type]
|
||||
|
||||
if not isinstance(first_arg, Handle):
|
||||
dispatch.write(f"""
|
||||
extern "C" {obj.result} {rename(obj.name)} ({", ".join(p.param for p in obj.params)}) noexcept {{
|
||||
unimplemented ();
|
||||
}}""")
|
||||
continue
|
||||
|
||||
if first_arg.has_parent('VkDevice', reg):
|
||||
table = "d_table"
|
||||
elif first_arg.has_parent('VkInstance', reg):
|
||||
table = 'i_table'
|
||||
else:
|
||||
raise Exception("Unknown param type")
|
||||
|
||||
dispatch.write(f"""
|
||||
extern "C" {obj.result} {rename(obj.name)} ({", ".join(p.param for p in obj.params)}) noexcept {{
|
||||
using first_arg_t = std::decay_t<decltype({obj.params[0].name})>;
|
||||
unimplemented ();
|
||||
}}""")
|
||||
continue
|
||||
|
||||
if constexpr (is_instance_v<first_arg_t>) {{
|
||||
auto const entry = reinterpret_cast<cruft::vk::icd::func const*> ({obj.params[0].name});
|
||||
auto const *table = reinterpret_cast<decltype({table})> (entry->table);
|
||||
if first_arg.has_parent('VkDevice', reg):
|
||||
table = "d_table"
|
||||
elif first_arg.has_parent('VkInstance', reg):
|
||||
table = 'i_table'
|
||||
else:
|
||||
raise Exception("Unknown param type")
|
||||
|
||||
return (table->{obj.name})(
|
||||
reinterpret_cast<decltype({obj.params[0].name})> (entry->handle)
|
||||
{", ".join([''] + [p.name for p in obj.params[1:]])}
|
||||
);
|
||||
}} else {{
|
||||
unimplemented ();
|
||||
}}
|
||||
dst.write(f"""
|
||||
extern "C" {obj.result} {rename(obj.name)} ({", ".join(p.param for p in obj.params)}) noexcept {{
|
||||
using first_arg_t = std::decay_t<decltype({obj.params[0].name})>;
|
||||
|
||||
if constexpr (is_instance_v<first_arg_t>) {{
|
||||
auto const entry = reinterpret_cast<cruft::vk::icd::func const*> ({obj.params[0].name});
|
||||
auto const *table = reinterpret_cast<decltype({table})> (entry->table);
|
||||
|
||||
return (table->{obj.name})(
|
||||
reinterpret_cast<decltype({obj.params[0].name})> (entry->handle)
|
||||
{", ".join([''] + [p.name for p in obj.params[1:]])}
|
||||
);
|
||||
}} else {{
|
||||
unimplemented ();
|
||||
}}
|
||||
""")
|
||||
|
||||
|
||||
}}
|
||||
""")
|
||||
|
||||
|
||||
###############################################################################
|
||||
@ -918,9 +918,14 @@ def main():
|
||||
q = reg.serialise(args.platform)
|
||||
|
||||
# Finally write out the header, icd, and dispatch code.
|
||||
write_header(args.dst, q, reg)
|
||||
write_icd(args.icd, q, reg)
|
||||
write_dispatch(args.dispatch, q, reg)
|
||||
with open(args.dst, 'w') as dst:
|
||||
write_header(dst, q, reg)
|
||||
|
||||
with open(args.icd, 'w') as dst:
|
||||
write_icd(dst, q, reg)
|
||||
|
||||
with open(args.dispatch, 'w') as dst:
|
||||
write_dispatch(dst, q, reg)
|
||||
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
|
||||
Loading…
Reference in New Issue
Block a user