sharded-array-for-python/scripts/code_gen.py at main · IntelPython/sharded-array-for-python

This repository was archived by the owner on Jan 26, 2026. It is now read-only.

History

52 lines (44 loc) · 1.33 KB

Raw

import os

import sys

sys.path.append(os.path.abspath(os.path.dirname(sys.argv[1])))

exec(f"import {os.path.basename(sys.argv[1]).split('.')[0]} as api")

sys.stdout = open(sys.argv[2], "w")

print(

"""// Auto-generated file

// #######################################################

// !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

// !! DO NOT EDIT, USE ../scripts/code_gen.py TO UPDATE !!

// !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

// #######################################################

// SPDX-License-Identifier: BSD-3-Clause

#pragma once

#ifdef DEF_PY11_ENUMS

#include <pybind11/pybind11.h>

#include <pybind11/stl.h>

namespace py = pybind11;

#endif

namespace SHARPY {

"""

)

prev = 0

for cat, lst in api.api_categories.items():

print(f"enum {cat}Id : int {{")

for x in lst:

x = x + f" = {prev}" if x == lst[0] else x

print(f" {x.upper()},")

prev = f"{cat.upper()}_LAST"

print(f" {prev}")

print("};\n")

print(

"""#ifdef DEF_PY11_ENUMS

static void def_enums(py::module_ & m)

{"""

)

for cat, lst in api.api_categories.items():

print(f' py::enum_<{cat}Id>(m, "{cat}Id")')

for x in lst:

print(f' .value("{x.upper()}", {x.upper()})')

print(" .export_values();\n")

print("}\n#endif\n} // namespace SHARPY")

# Close the file

sys.stdout.close()

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Expand file tree

Search code, repositories, users, issues, pull requests...

FilesExpand file tree

code_gen.py

Latest commit

History

code_gen.py

File metadata and controls

Expand file tree