mirror of
https://github.com/Z3Prover/z3
synced 2025-04-06 17:44:08 +00:00
Add high level bindings for js (#6048)
* [Draft] Added unfinished code for high level bindings for js * * Rewrote structure of js api files * Added more high level apis * Minor fixes * Fixed wasm github action * Fix JS test * Removed ContextOptions type * * Added Ints to JS Api * Added tests to JS Api * Added run-time checks for contexts * Removed default contexts * Merged Context and createContext so that the api behaves the sames as in other constructors * Added a test for Solver * Added Reals * Added classes for IntVals and RealVals * Added abillity to specify logic for solver * Try to make CI tests not fail * Changed APIs after a round of review * Fix test * Added BitVectors * Made sort into getter * Added initial JS docs * Added more coercible types * Removed done TODOs
This commit is contained in:
Olaf Tomalka
GitHub
parent
3d00d1d56b
commit
7fdcbbaee9
10
.github/workflows/wasm-release.yml
vendored
10
.github/workflows/wasm-release.yml
vendored
|
|
@ -23,8 +23,8 @@ jobs:
|
||||||
- name: Setup node
|
- name: Setup node
|
||||||
uses: actions/setup-node@v3
|
uses: actions/setup-node@v3
|
||||||
with:
|
with:
|
||||||
node-version: 'lts/*'
|
node-version: "lts/*"
|
||||||
registry-url: 'https://registry.npmjs.org'
|
registry-url: "https://registry.npmjs.org"
|
||||||
|
|
||||||
- name: Prepare for publish
|
- name: Prepare for publish
|
||||||
run: |
|
run: |
|
||||||
|
|
@ -37,13 +37,13 @@ jobs:
|
||||||
with:
|
with:
|
||||||
no-install: true
|
no-install: true
|
||||||
version: ${{env.EM_VERSION}}
|
version: ${{env.EM_VERSION}}
|
||||||
actions-cache-folder: 'emsdk-cache'
|
actions-cache-folder: "emsdk-cache"
|
||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: npm ci
|
run: npm ci
|
||||||
|
|
||||||
- name: Build TypeScript
|
- name: Build TypeScript
|
||||||
run: npm run build-ts
|
run: npm run build:ts
|
||||||
|
|
||||||
- name: Build wasm
|
- name: Build wasm
|
||||||
run: |
|
run: |
|
||||||
|
|
@ -52,7 +52,7 @@ jobs:
|
||||||
source $(dirname $(which emsdk))/emsdk_env.sh
|
source $(dirname $(which emsdk))/emsdk_env.sh
|
||||||
which node
|
which node
|
||||||
which clang++
|
which clang++
|
||||||
npm run build-wasm
|
npm run build:wasm
|
||||||
|
|
||||||
- name: Test
|
- name: Test
|
||||||
run: npm test
|
run: npm test
|
||||||
|
|
|
||||||
10
.github/workflows/wasm.yml
vendored
10
.github/workflows/wasm.yml
vendored
|
|
@ -2,7 +2,7 @@ name: WebAssembly Build
|
||||||
|
|
||||||
on:
|
on:
|
||||||
push:
|
push:
|
||||||
branches: [ master ]
|
branches: [master]
|
||||||
pull_request:
|
pull_request:
|
||||||
|
|
||||||
defaults:
|
defaults:
|
||||||
|
|
@ -23,20 +23,20 @@ jobs:
|
||||||
- name: Setup node
|
- name: Setup node
|
||||||
uses: actions/setup-node@v3
|
uses: actions/setup-node@v3
|
||||||
with:
|
with:
|
||||||
node-version: 'lts/*'
|
node-version: "lts/*"
|
||||||
|
|
||||||
- name: Setup emscripten
|
- name: Setup emscripten
|
||||||
uses: mymindstorm/setup-emsdk@v11
|
uses: mymindstorm/setup-emsdk@v11
|
||||||
with:
|
with:
|
||||||
no-install: true
|
no-install: true
|
||||||
version: ${{env.EM_VERSION}}
|
version: ${{env.EM_VERSION}}
|
||||||
actions-cache-folder: 'emsdk-cache'
|
actions-cache-folder: "emsdk-cache"
|
||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: npm ci
|
run: npm ci
|
||||||
|
|
||||||
- name: Build TypeScript
|
- name: Build TypeScript
|
||||||
run: npm run build-ts
|
run: npm run build:ts
|
||||||
|
|
||||||
- name: Build wasm
|
- name: Build wasm
|
||||||
run: |
|
run: |
|
||||||
|
|
@ -45,7 +45,7 @@ jobs:
|
||||||
source $(dirname $(which emsdk))/emsdk_env.sh
|
source $(dirname $(which emsdk))/emsdk_env.sh
|
||||||
which node
|
which node
|
||||||
which clang++
|
which clang++
|
||||||
npm run build-wasm
|
npm run build:wasm
|
||||||
|
|
||||||
- name: Test
|
- name: Test
|
||||||
run: npm test
|
run: npm test
|
||||||
|
|
|
||||||
7
.gitignore
vendored
7
.gitignore
vendored
|
|
@ -66,6 +66,7 @@ src/api/python/z3/z3consts.py
|
||||||
src/api/python/z3/z3core.py
|
src/api/python/z3/z3core.py
|
||||||
src/ast/pattern/database.h
|
src/ast/pattern/database.h
|
||||||
src/util/version.h
|
src/util/version.h
|
||||||
|
src/util/z3_version.h
|
||||||
src/api/java/Native.cpp
|
src/api/java/Native.cpp
|
||||||
src/api/java/Native.java
|
src/api/java/Native.java
|
||||||
src/api/java/enumerations/*.java
|
src/api/java/enumerations/*.java
|
||||||
|
|
@ -76,11 +77,8 @@ src/api/ml/z3native.mli
|
||||||
src/api/ml/z3enums.mli
|
src/api/ml/z3enums.mli
|
||||||
src/api/ml/z3.mllib
|
src/api/ml/z3.mllib
|
||||||
src/api/js/node_modules/
|
src/api/js/node_modules/
|
||||||
src/api/js/*.js
|
|
||||||
src/api/js/build/
|
src/api/js/build/
|
||||||
src/api/js/**/*.d.ts
|
src/api/js/**/*.__GENERATED__.*
|
||||||
!src/api/js/scripts/*.js
|
|
||||||
!src/api/js/src/*.js
|
|
||||||
debug/*
|
debug/*
|
||||||
|
|
||||||
out/**
|
out/**
|
||||||
|
|
@ -92,3 +90,4 @@ examples/**/obj
|
||||||
CMakeSettings.json
|
CMakeSettings.json
|
||||||
# Editor temp files
|
# Editor temp files
|
||||||
*.swp
|
*.swp
|
||||||
|
.DS_Store
|
||||||
|
|
@ -14,11 +14,13 @@ import subprocess
|
||||||
|
|
||||||
ML_ENABLED=False
|
ML_ENABLED=False
|
||||||
MLD_ENABLED=False
|
MLD_ENABLED=False
|
||||||
|
JS_ENABLED=False
|
||||||
BUILD_DIR='../build'
|
BUILD_DIR='../build'
|
||||||
DOXYGEN_EXE='doxygen'
|
DOXYGEN_EXE='doxygen'
|
||||||
TEMP_DIR=os.path.join(os.getcwd(), 'tmp')
|
TEMP_DIR=os.path.join(os.getcwd(), 'tmp')
|
||||||
OUTPUT_DIRECTORY=os.path.join(os.getcwd(), 'api')
|
OUTPUT_DIRECTORY=os.path.join(os.getcwd(), 'api')
|
||||||
Z3PY_PACKAGE_PATH='../src/api/python/z3'
|
Z3PY_PACKAGE_PATH='../src/api/python/z3'
|
||||||
|
JS_API_PATH='../src/api/js'
|
||||||
Z3PY_ENABLED=True
|
Z3PY_ENABLED=True
|
||||||
DOTNET_ENABLED=True
|
DOTNET_ENABLED=True
|
||||||
JAVA_ENABLED=True
|
JAVA_ENABLED=True
|
||||||
|
|
@ -28,8 +30,8 @@ SCRIPT_DIR=os.path.abspath(os.path.dirname(__file__))
|
||||||
|
|
||||||
def parse_options():
|
def parse_options():
|
||||||
global ML_ENABLED, MLD_ENABLED, BUILD_DIR, DOXYGEN_EXE, TEMP_DIR, OUTPUT_DIRECTORY
|
global ML_ENABLED, MLD_ENABLED, BUILD_DIR, DOXYGEN_EXE, TEMP_DIR, OUTPUT_DIRECTORY
|
||||||
global Z3PY_PACKAGE_PATH, Z3PY_ENABLED, DOTNET_ENABLED, JAVA_ENABLED
|
global Z3PY_PACKAGE_PATH, Z3PY_ENABLED, DOTNET_ENABLED, JAVA_ENABLED, JS_ENABLED
|
||||||
global DOTNET_API_SEARCH_PATHS, JAVA_API_SEARCH_PATHS
|
global DOTNET_API_SEARCH_PATHS, JAVA_API_SEARCH_PATHS, JS_API_PATH
|
||||||
parser = argparse.ArgumentParser(description=__doc__)
|
parser = argparse.ArgumentParser(description=__doc__)
|
||||||
parser.add_argument('-b',
|
parser.add_argument('-b',
|
||||||
'--build',
|
'--build',
|
||||||
|
|
@ -46,6 +48,11 @@ def parse_options():
|
||||||
default=False,
|
default=False,
|
||||||
help='Include ML/OCaml API documentation'
|
help='Include ML/OCaml API documentation'
|
||||||
)
|
)
|
||||||
|
parser.add_argument('--js',
|
||||||
|
action='store_true',
|
||||||
|
default=False,
|
||||||
|
help='Include JS/TS API documentation'
|
||||||
|
)
|
||||||
parser.add_argument('--doxygen-executable',
|
parser.add_argument('--doxygen-executable',
|
||||||
dest='doxygen_executable',
|
dest='doxygen_executable',
|
||||||
default=DOXYGEN_EXE,
|
default=DOXYGEN_EXE,
|
||||||
|
|
@ -104,6 +111,7 @@ def parse_options():
|
||||||
pargs = parser.parse_args()
|
pargs = parser.parse_args()
|
||||||
ML_ENABLED = pargs.ml
|
ML_ENABLED = pargs.ml
|
||||||
MLD_ENABLED = pargs.mld
|
MLD_ENABLED = pargs.mld
|
||||||
|
JS_ENABLED = pargs.js
|
||||||
BUILD_DIR = pargs.build
|
BUILD_DIR = pargs.build
|
||||||
DOXYGEN_EXE = pargs.doxygen_executable
|
DOXYGEN_EXE = pargs.doxygen_executable
|
||||||
TEMP_DIR = pargs.temp_dir
|
TEMP_DIR = pargs.temp_dir
|
||||||
|
|
@ -224,6 +232,10 @@ try:
|
||||||
print("Java documentation disabled")
|
print("Java documentation disabled")
|
||||||
doxygen_config_substitutions['JAVA_API_FILES'] = ''
|
doxygen_config_substitutions['JAVA_API_FILES'] = ''
|
||||||
doxygen_config_substitutions['JAVA_API_SEARCH_PATHS'] = ''
|
doxygen_config_substitutions['JAVA_API_SEARCH_PATHS'] = ''
|
||||||
|
if JS_ENABLED:
|
||||||
|
print('Javascript documentation enabled')
|
||||||
|
else:
|
||||||
|
print('Javascript documentation disabled')
|
||||||
|
|
||||||
doxygen_config_file = temp_path('z3api.cfg')
|
doxygen_config_file = temp_path('z3api.cfg')
|
||||||
configure_file(
|
configure_file(
|
||||||
|
|
@ -241,7 +253,7 @@ try:
|
||||||
'{prefix}<a class="el" href="z3__api_8h.html">C API</a> '
|
'{prefix}<a class="el" href="z3__api_8h.html">C API</a> '
|
||||||
).format(
|
).format(
|
||||||
prefix=bullet_point_prefix)
|
prefix=bullet_point_prefix)
|
||||||
|
|
||||||
if Z3PY_ENABLED:
|
if Z3PY_ENABLED:
|
||||||
print("Python documentation enabled")
|
print("Python documentation enabled")
|
||||||
website_dox_substitutions['PYTHON_API'] = (
|
website_dox_substitutions['PYTHON_API'] = (
|
||||||
|
|
@ -274,6 +286,13 @@ try:
|
||||||
prefix=bullet_point_prefix)
|
prefix=bullet_point_prefix)
|
||||||
else:
|
else:
|
||||||
website_dox_substitutions['OCAML_API'] = ''
|
website_dox_substitutions['OCAML_API'] = ''
|
||||||
|
if JS_ENABLED:
|
||||||
|
website_dox_substitutions['JS_API'] = (
|
||||||
|
'{prefix}<a class="el" href="js/index.html">Javascript/Typescript API</a>'
|
||||||
|
).format(
|
||||||
|
prefix=bullet_point_prefix)
|
||||||
|
else:
|
||||||
|
website_dox_substitutions['JS_API'] = ''
|
||||||
configure_file(
|
configure_file(
|
||||||
doc_path('website.dox.in'),
|
doc_path('website.dox.in'),
|
||||||
temp_path('website.dox'),
|
temp_path('website.dox'),
|
||||||
|
|
@ -339,6 +358,18 @@ try:
|
||||||
exit(1)
|
exit(1)
|
||||||
print("Generated ML/OCaml documentation.")
|
print("Generated ML/OCaml documentation.")
|
||||||
|
|
||||||
|
if JS_ENABLED:
|
||||||
|
try:
|
||||||
|
subprocess.check_output(['npm', 'run', '--prefix=%s' % JS_API_PATH, 'check-engine'])
|
||||||
|
except subprocess.CalledProcessError as e:
|
||||||
|
print("ERROR: node version check failed.")
|
||||||
|
print(e.output)
|
||||||
|
exit(1)
|
||||||
|
if subprocess.call(['npm', 'run', '--prefix=%s' % JS_API_PATH, 'docs']) != 0:
|
||||||
|
print("ERROR: npm run docs failed.")
|
||||||
|
exit(1)
|
||||||
|
print("Generated Javascript documentation.")
|
||||||
|
|
||||||
print("Documentation was successfully generated at subdirectory '{}'.".format(OUTPUT_DIRECTORY))
|
print("Documentation was successfully generated at subdirectory '{}'.".format(OUTPUT_DIRECTORY))
|
||||||
except Exception:
|
except Exception:
|
||||||
exctype, value = sys.exc_info()[:2]
|
exctype, value = sys.exc_info()[:2]
|
||||||
|
|
|
||||||
|
|
@ -1,13 +1,13 @@
|
||||||
/**
|
/**
|
||||||
\mainpage An Efficient Theorem Prover
|
\mainpage An Efficient Theorem Prover
|
||||||
|
|
||||||
Z3 is a high-performance theorem prover being developed at <a class="el"
|
Z3 is a high-performance theorem prover being developed at <a class="el"
|
||||||
href="http://research.microsoft.com">Microsoft Research</a>.
|
href="http://research.microsoft.com">Microsoft Research</a>.
|
||||||
|
|
||||||
<b>The Z3 website is at <a class="el" href="http://github.com/z3prover">http://github.com/z3prover</a>.</b>
|
<b>The Z3 website is at <a class="el" href="http://github.com/z3prover">http://github.com/z3prover</a>.</b>
|
||||||
|
|
||||||
This website hosts the automatically generated documentation for the Z3 APIs.
|
This website hosts the automatically generated documentation for the Z3 APIs.
|
||||||
|
|
||||||
- \ref @C_API@
|
- \ref @C_API@
|
||||||
- \ref @CPP_API@ @DOTNET_API@ @JAVA_API@ @PYTHON_API@ @OCAML_API@
|
- \ref @CPP_API@ @DOTNET_API@ @JAVA_API@ @PYTHON_API@ @OCAML_API@ @JS_API@
|
||||||
*/
|
*/
|
||||||
|
|
|
||||||
|
|
@ -246,10 +246,13 @@ def rmf(fname):
|
||||||
|
|
||||||
def exec_compiler_cmd(cmd):
|
def exec_compiler_cmd(cmd):
|
||||||
r = exec_cmd(cmd)
|
r = exec_cmd(cmd)
|
||||||
if is_windows() or is_cygwin_mingw() or is_cygwin() or is_msys2():
|
# Windows
|
||||||
rmf('a.exe')
|
rmf('a.exe')
|
||||||
else:
|
# Unix
|
||||||
rmf('a.out')
|
rmf('a.out')
|
||||||
|
# Emscripten
|
||||||
|
rmf('a.wasm')
|
||||||
|
rmf('a.worker.js')
|
||||||
return r
|
return r
|
||||||
|
|
||||||
def test_cxx_compiler(cc):
|
def test_cxx_compiler(cc):
|
||||||
|
|
@ -293,6 +296,10 @@ def test_fpmath(cc):
|
||||||
t.commit()
|
t.commit()
|
||||||
# -Werror is needed because some versions of clang warn about unrecognized
|
# -Werror is needed because some versions of clang warn about unrecognized
|
||||||
# -m flags.
|
# -m flags.
|
||||||
|
# TODO(ritave): Safari doesn't allow SIMD WebAssembly extension, add a flag to build script
|
||||||
|
if exec_compiler_cmd([cc, CPPFLAGS, '-Werror', 'tstsse.cpp', LDFLAGS, '-msse -msse2 -msimd128']) == 0:
|
||||||
|
FPMATH_FLAGS='-msse -msse2 -msimd128'
|
||||||
|
return 'SSE2-EMSCRIPTEN'
|
||||||
if exec_compiler_cmd([cc, CPPFLAGS, '-Werror', 'tstsse.cpp', LDFLAGS, '-mfpmath=sse -msse -msse2']) == 0:
|
if exec_compiler_cmd([cc, CPPFLAGS, '-Werror', 'tstsse.cpp', LDFLAGS, '-mfpmath=sse -msse -msse2']) == 0:
|
||||||
FPMATH_FLAGS="-mfpmath=sse -msse -msse2"
|
FPMATH_FLAGS="-mfpmath=sse -msse -msse2"
|
||||||
return "SSE2-GCC"
|
return "SSE2-GCC"
|
||||||
|
|
@ -499,7 +506,7 @@ def find_ml_lib():
|
||||||
|
|
||||||
def is64():
|
def is64():
|
||||||
global LINUX_X64
|
global LINUX_X64
|
||||||
if is_sunos() and sys.version_info.major < 3:
|
if is_sunos() and sys.version_info.major < 3:
|
||||||
return LINUX_X64
|
return LINUX_X64
|
||||||
else:
|
else:
|
||||||
return LINUX_X64 and sys.maxsize >= 2**32
|
return LINUX_X64 and sys.maxsize >= 2**32
|
||||||
|
|
@ -625,11 +632,11 @@ elif os.name == 'posix':
|
||||||
else:
|
else:
|
||||||
LINUX_X64=False
|
LINUX_X64=False
|
||||||
|
|
||||||
|
|
||||||
if os.name == 'posix' and os.uname()[4] == 'arm64':
|
if os.name == 'posix' and os.uname()[4] == 'arm64':
|
||||||
IS_ARCH_ARM64 = True
|
IS_ARCH_ARM64 = True
|
||||||
|
|
||||||
|
|
||||||
def display_help(exit_code):
|
def display_help(exit_code):
|
||||||
print("mk_make.py: Z3 Makefile generator\n")
|
print("mk_make.py: Z3 Makefile generator\n")
|
||||||
print("This script generates the Makefile for the Z3 theorem prover.")
|
print("This script generates the Makefile for the Z3 theorem prover.")
|
||||||
|
|
@ -792,7 +799,7 @@ def extract_c_includes(fname):
|
||||||
linenum = 1
|
linenum = 1
|
||||||
for line in f:
|
for line in f:
|
||||||
m1 = std_inc_pat.match(line)
|
m1 = std_inc_pat.match(line)
|
||||||
if m1:
|
if m1:
|
||||||
root_file_name = m1.group(1)
|
root_file_name = m1.group(1)
|
||||||
slash_pos = root_file_name.rfind('/')
|
slash_pos = root_file_name.rfind('/')
|
||||||
if slash_pos >= 0 and root_file_name.find("..") < 0 : #it is a hack for lp include files that behave as continued from "src"
|
if slash_pos >= 0 and root_file_name.find("..") < 0 : #it is a hack for lp include files that behave as continued from "src"
|
||||||
|
|
@ -1650,7 +1657,7 @@ def set_key_file(self):
|
||||||
else:
|
else:
|
||||||
print("Keyfile '%s' could not be found; %s.dll will be unsigned." % (self.key_file, self.dll_name))
|
print("Keyfile '%s' could not be found; %s.dll will be unsigned." % (self.key_file, self.dll_name))
|
||||||
self.key_file = None
|
self.key_file = None
|
||||||
|
|
||||||
|
|
||||||
# build for dotnet core
|
# build for dotnet core
|
||||||
class DotNetDLLComponent(Component):
|
class DotNetDLLComponent(Component):
|
||||||
|
|
@ -1664,7 +1671,7 @@ class DotNetDLLComponent(Component):
|
||||||
self.assembly_info_dir = assembly_info_dir
|
self.assembly_info_dir = assembly_info_dir
|
||||||
self.key_file = default_key_file
|
self.key_file = default_key_file
|
||||||
|
|
||||||
|
|
||||||
def mk_makefile(self, out):
|
def mk_makefile(self, out):
|
||||||
if not is_dotnet_core_enabled():
|
if not is_dotnet_core_enabled():
|
||||||
return
|
return
|
||||||
|
|
@ -1680,7 +1687,7 @@ class DotNetDLLComponent(Component):
|
||||||
out.write(' ')
|
out.write(' ')
|
||||||
out.write(cs_file)
|
out.write(cs_file)
|
||||||
out.write('\n')
|
out.write('\n')
|
||||||
|
|
||||||
set_key_file(self)
|
set_key_file(self)
|
||||||
key = ""
|
key = ""
|
||||||
if not self.key_file is None:
|
if not self.key_file is None:
|
||||||
|
|
@ -1724,7 +1731,7 @@ class DotNetDLLComponent(Component):
|
||||||
ous.write(core_csproj_str)
|
ous.write(core_csproj_str)
|
||||||
|
|
||||||
dotnetCmdLine = [DOTNET, "build", csproj]
|
dotnetCmdLine = [DOTNET, "build", csproj]
|
||||||
|
|
||||||
dotnetCmdLine.extend(['-c'])
|
dotnetCmdLine.extend(['-c'])
|
||||||
if DEBUG_MODE:
|
if DEBUG_MODE:
|
||||||
dotnetCmdLine.extend(['Debug'])
|
dotnetCmdLine.extend(['Debug'])
|
||||||
|
|
@ -1733,19 +1740,19 @@ class DotNetDLLComponent(Component):
|
||||||
|
|
||||||
path = os.path.join(os.path.abspath(BUILD_DIR), ".")
|
path = os.path.join(os.path.abspath(BUILD_DIR), ".")
|
||||||
dotnetCmdLine.extend(['-o', "\"%s\"" % path])
|
dotnetCmdLine.extend(['-o', "\"%s\"" % path])
|
||||||
|
|
||||||
MakeRuleCmd.write_cmd(out, ' '.join(dotnetCmdLine))
|
MakeRuleCmd.write_cmd(out, ' '.join(dotnetCmdLine))
|
||||||
out.write('\n')
|
out.write('\n')
|
||||||
out.write('%s: %s\n\n' % (self.name, dllfile))
|
out.write('%s: %s\n\n' % (self.name, dllfile))
|
||||||
|
|
||||||
def main_component(self):
|
def main_component(self):
|
||||||
return is_dotnet_core_enabled()
|
return is_dotnet_core_enabled()
|
||||||
|
|
||||||
def has_assembly_info(self):
|
def has_assembly_info(self):
|
||||||
# TBD: is this required for dotnet core given that version numbers are in z3.csproj file?
|
# TBD: is this required for dotnet core given that version numbers are in z3.csproj file?
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
def mk_win_dist(self, build_path, dist_path):
|
def mk_win_dist(self, build_path, dist_path):
|
||||||
if is_dotnet_core_enabled():
|
if is_dotnet_core_enabled():
|
||||||
mk_dir(os.path.join(dist_path, INSTALL_BIN_DIR))
|
mk_dir(os.path.join(dist_path, INSTALL_BIN_DIR))
|
||||||
|
|
@ -2038,7 +2045,7 @@ class MLComponent(Component):
|
||||||
out.write('ml: %s.cma %s.cmxa %s.cmxs\n' % (z3mls, z3mls, z3mls))
|
out.write('ml: %s.cma %s.cmxa %s.cmxs\n' % (z3mls, z3mls, z3mls))
|
||||||
if IS_OSX:
|
if IS_OSX:
|
||||||
out.write('\tinstall_name_tool -id %s/libz3.dylib libz3.dylib\n' % (stubs_install_path))
|
out.write('\tinstall_name_tool -id %s/libz3.dylib libz3.dylib\n' % (stubs_install_path))
|
||||||
out.write('\tinstall_name_tool -change libz3.dylib %s/libz3.dylib api/ml/dllz3ml.so\n' % (stubs_install_path))
|
out.write('\tinstall_name_tool -change libz3.dylib %s/libz3.dylib api/ml/dllz3ml.so\n' % (stubs_install_path))
|
||||||
out.write('\n')
|
out.write('\n')
|
||||||
|
|
||||||
if IS_WINDOWS:
|
if IS_WINDOWS:
|
||||||
|
|
@ -3117,7 +3124,7 @@ def get_platform_toolset_str():
|
||||||
if len(tokens) < 2:
|
if len(tokens) < 2:
|
||||||
return default
|
return default
|
||||||
else:
|
else:
|
||||||
if tokens[0] == "15":
|
if tokens[0] == "15":
|
||||||
# Visual Studio 2017 reports 15.* but the PlatformToolsetVersion is 141
|
# Visual Studio 2017 reports 15.* but the PlatformToolsetVersion is 141
|
||||||
return "v141"
|
return "v141"
|
||||||
else:
|
else:
|
||||||
|
|
|
||||||
1
src/api/js/.nvmrc
Normal file
1
src/api/js/.nvmrc
Normal file
|
|
@ -0,0 +1 @@
|
||||||
|
v16.15.0
|
||||||
6
src/api/js/.prettierrc.json
Normal file
6
src/api/js/.prettierrc.json
Normal file
|
|
@ -0,0 +1,6 @@
|
||||||
|
{
|
||||||
|
"singleQuote": true,
|
||||||
|
"arrowParens": "avoid",
|
||||||
|
"printWidth": 120,
|
||||||
|
"trailingComma": "all"
|
||||||
|
}
|
||||||
|
|
@ -1,32 +1,86 @@
|
||||||
# Z3 TypeScript Bindings
|
# Z3 TypeScript Bindings
|
||||||
|
|
||||||
This project provides low-level TypeScript bindings for the [Z3 theorem prover](https://github.com/Z3Prover/z3). It is available on npm as [z3-solver](https://www.npmjs.com/package/z3-solver).
|
This project provides high-level and low-level TypeScript bindings for the [Z3 theorem prover](https://github.com/Z3Prover/z3). It is available on npm as [z3-solver](https://www.npmjs.com/package/z3-solver).
|
||||||
|
|
||||||
Z3 itself is distributed as a wasm artifact as part of this package. You can find the documentation for the Z3 API [here](https://z3prover.github.io/api/html/z3__api_8h.html), though note the differences below.
|
|
||||||
|
|
||||||
|
Z3 itself is distributed as a wasm artifact as part of this package.
|
||||||
|
|
||||||
## Using
|
## Using
|
||||||
|
|
||||||
This requires threads, which means you'll need to be running in an environment which supports `SharedArrayBuffer`. In browsers, in addition to ensuring the browser has implemented `SharedArrayBuffer`, you'll need to serve your page with [special headers](https://web.dev/coop-coep/). There's a [neat trick](https://github.com/gzuidhof/coi-serviceworker) for doing that client-side on e.g. Github Pages, though you shouldn't use that trick in more complex applications.
|
```javascript
|
||||||
|
const { init } = require('z3-solver');
|
||||||
|
const {
|
||||||
|
Z3, // Low-level C-like API
|
||||||
|
Context, // High-level Z3Py-like API
|
||||||
|
} = await init();
|
||||||
|
```
|
||||||
|
|
||||||
The Emscripten worker model will spawn multiple instances of `z3-built.js` for long-running operations. When building for the web, you should include that file as its own script on the page - using a bundler like webpack will prevent it from loading correctly. That script defines a global variable named `initZ3`. Your main script, which can be bundled, should do `let { init } = require('z3-solver/build/wrapper.js'); let { Z3 } = await init(initZ3);`.
|
This package has different initialization for browser and node. Your bundler and node should choose good version automatically, but you can import the one you need manually - `const { init } = require('z3-solver/node');` or `const { init } = require('z3-solver/browser');`.
|
||||||
|
|
||||||
Other than the differences below, the bindings can be used exactly as you'd use the C library. Because this is a wrapper around a C library, most of the values you'll use are just numbers representing pointers. For this reason you are strongly encouraged to make use of the TypeScript types to differentiate among the different kinds of value.
|
### Limitations
|
||||||
|
|
||||||
The module exports an `init` function, is an async function which initializes the library and returns `{ em, Z3 }` - `em` contains the underlying emscripten module, which you can use to e.g. kill stray threads, and `Z3` contains the actual bindings. The other module exports are the enums defined in the Z3 API.
|
The package requires threads, which means you'll need to be running in an environment which supports `SharedArrayBuffer`. In browsers, in addition to ensuring the browser has implemented `SharedArrayBuffer`, you'll need to serve your page with [special headers](https://web.dev/coop-coep/). There's a [neat trick](https://github.com/gzuidhof/coi-serviceworker) for doing that client-side on e.g. Github Pages, though you shouldn't use that trick in more complex applications.
|
||||||
|
|
||||||
[`test-ts-api.ts`](./test-ts-api.ts) contains a couple real cases translated very mechanically from [this file](https://github.com/Z3Prover/z3/blob/90fd3d82fce20d45ed2eececdf65545bab769503/examples/c/test_capi.c).
|
The Emscripten worker model will spawn multiple instances of `z3-built.js` for long-running operations. When building for the web, you should include that file as its own script on the page - using a bundler like webpack will prevent it from loading correctly.
|
||||||
|
|
||||||
|
## High-level
|
||||||
|
|
||||||
## Differences from the C API
|
You can find the documentation for the high-level Z3 API [here](https://z3prover.github.io/api/html/js/index.html). There are some usage examples in `src/high-level/high-level.test.ts`
|
||||||
|
|
||||||
### Integers
|
Most of the API requires a context to run, and so you need to initialize one to access most functionality.
|
||||||
|
|
||||||
|
```javascript
|
||||||
|
const { init } = require('z3-solver');
|
||||||
|
const { Context } = await init();
|
||||||
|
const { Solver, Int, And } = new Context('main');
|
||||||
|
|
||||||
|
const x = Int.const('x');
|
||||||
|
|
||||||
|
const solver = new Solver();
|
||||||
|
solver.add(And(x.ge(0), x.le(9)));
|
||||||
|
console.log(await solver.check());
|
||||||
|
// sat
|
||||||
|
```
|
||||||
|
|
||||||
|
Typescript types try to catch errors concerning mixing of Contexts during compile time. Objects returned from `new Context('name')` have a type narrowed by the provided Context name and will fail to compile if you try to mix them.
|
||||||
|
|
||||||
|
```typescript
|
||||||
|
const { Int: Int1 } = new Context('context1');
|
||||||
|
const { Int: Int2 } = new Context('context2');
|
||||||
|
|
||||||
|
const x = Int1.const('x');
|
||||||
|
const y = Int2.const('y');
|
||||||
|
|
||||||
|
// The below will fail to compile in Typescript
|
||||||
|
x.ge(y);
|
||||||
|
```
|
||||||
|
|
||||||
|
```typescript
|
||||||
|
// Use templated functions to propagate narrowed types
|
||||||
|
function add<Name extends string>(a: Arith<Name>, b: Arith<Name>): Arith<Name> {
|
||||||
|
return a.add(b).add(5);
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
Some long-running functions are promises and will run in a separate thread.
|
||||||
|
Currently Z3-solver is not thread safe, and so, high-level APIs ensures that only one long-running function can run at a time, and all other long-running requests will queue up and be run one after another.
|
||||||
|
|
||||||
|
## Low-level
|
||||||
|
|
||||||
|
You can find the documentation for the low-level Z3 API [here](https://z3prover.github.io/api/html/z3__api_8h.html), though note the differences below. `examples/low-level/` contains a couple real cases translated very mechanically from [this file](https://github.com/Z3Prover/z3/blob/90fd3d82fce20d45ed2eececdf65545bab769503/examples/c/test_capi.c).
|
||||||
|
|
||||||
|
The bindings can be used exactly as you'd use the C library. Because this is a wrapper around a C library, most of the values you'll use are just numbers representing pointers. For this reason you are strongly encouraged to make use of the TypeScript types to differentiate among the different kinds of value.
|
||||||
|
|
||||||
|
The module exports an `init` function, which is an async function which initializes the library and returns `{ em, Z3 }` - `em` contains the underlying emscripten module, which you can use to e.g. kill stray threads, and `Z3` contains the actual bindings. The other module exports are the enums defined in the Z3 API.
|
||||||
|
|
||||||
|
### Differences from the C API
|
||||||
|
|
||||||
|
#### Integers
|
||||||
|
|
||||||
JavaScript numbers are IEEE double-precisions floats. These can be used wherever the C API expects an `int`, `unsigned int`, `float`, or `double`.
|
JavaScript numbers are IEEE double-precisions floats. These can be used wherever the C API expects an `int`, `unsigned int`, `float`, or `double`.
|
||||||
|
|
||||||
`int64_t` and `uint64_t` cannot be precisely represented by JS numbers, so in the TS bindings they are represented by [BigInts](https://developer.mozilla.org/en-US/docs/Web/JavaScript/Data_structures#bigint_type).
|
`int64_t` and `uint64_t` cannot be precisely represented by JS numbers, so in the TS bindings they are represented by [BigInts](https://developer.mozilla.org/en-US/docs/Web/JavaScript/Data_structures#bigint_type).
|
||||||
|
|
||||||
### Out parameters
|
#### Out parameters
|
||||||
|
|
||||||
In C, to return multiple values a function will take an address to write to, conventionally referred to as an "out parameter". Sometimes the function returns a boolean to indicate success; other times it may return nothing or some other value.
|
In C, to return multiple values a function will take an address to write to, conventionally referred to as an "out parameter". Sometimes the function returns a boolean to indicate success; other times it may return nothing or some other value.
|
||||||
|
|
||||||
|
|
@ -56,7 +110,7 @@ is represented in the TS bindings as
|
||||||
|
|
||||||
```ts
|
```ts
|
||||||
function model_eval(c: Z3_context, m: Z3_model, t: Z3_ast, model_completion: boolean): Z3_ast | null {
|
function model_eval(c: Z3_context, m: Z3_model, t: Z3_ast, model_completion: boolean): Z3_ast | null {
|
||||||
// ...
|
// ...
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|
@ -64,12 +118,12 @@ Note that the boolean return type of the C function is translated into a nullabl
|
||||||
|
|
||||||
When the return value is of interest it is included in the returned record under the key `rv`.
|
When the return value is of interest it is included in the returned record under the key `rv`.
|
||||||
|
|
||||||
|
#### Arrays
|
||||||
### Arrays
|
|
||||||
|
|
||||||
The when the C API takes an array as an argument it will also require a parameter which specifies the length of the array (since arrays do not carry their own lengths in C). In the TS bindings this is automatically inferred.
|
The when the C API takes an array as an argument it will also require a parameter which specifies the length of the array (since arrays do not carry their own lengths in C). In the TS bindings this is automatically inferred.
|
||||||
|
|
||||||
For example, the C declaration
|
For example, the C declaration
|
||||||
|
|
||||||
```js
|
```js
|
||||||
unsigned Z3_rcf_mk_roots(Z3_context c, unsigned n, Z3_rcf_num const a[], Z3_rcf_num roots[]);
|
unsigned Z3_rcf_mk_roots(Z3_context c, unsigned n, Z3_rcf_num const a[], Z3_rcf_num roots[]);
|
||||||
```
|
```
|
||||||
|
|
@ -84,8 +138,7 @@ function rcf_mk_roots(c: Z3_context, a: Z3_rcf_num[]): { rv: number; roots: Z3_r
|
||||||
|
|
||||||
When there are multiple arrays which the C API expects to be of the same length, the TS bindings will enforce that this is the case.
|
When there are multiple arrays which the C API expects to be of the same length, the TS bindings will enforce that this is the case.
|
||||||
|
|
||||||
|
#### Null pointers
|
||||||
### Null pointers
|
|
||||||
|
|
||||||
Some of the C APIs accept or return null pointers (represented by types whose name end in `_opt`). These are represented in the TS bindings as `| null`. For example, the C declaration
|
Some of the C APIs accept or return null pointers (represented by types whose name end in `_opt`). These are represented in the TS bindings as `| null`. For example, the C declaration
|
||||||
|
|
||||||
|
|
@ -101,8 +154,7 @@ function model_get_const_interp(c: Z3_context, m: Z3_model, a: Z3_func_decl): Z3
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
|
#### Async functions
|
||||||
### Async functions
|
|
||||||
|
|
||||||
Certain long-running APIs are not appropriate to call on the main thread. In the TS bindings those APIs are marked as `async` and are automatically called on a different thread. This includes the following APIs:
|
Certain long-running APIs are not appropriate to call on the main thread. In the TS bindings those APIs are marked as `async` and are automatically called on a different thread. This includes the following APIs:
|
||||||
|
|
||||||
|
|
@ -122,5 +174,4 @@ Certain long-running APIs are not appropriate to call on the main thread. In the
|
||||||
- `Z3_fixedpoint_query_from_lvl`
|
- `Z3_fixedpoint_query_from_lvl`
|
||||||
- `Z3_polynomial_subresultants`
|
- `Z3_polynomial_subresultants`
|
||||||
|
|
||||||
Note that these are not thread-safe, and so only one call can be running at a time. Trying to call a second async API before the first completes will throw.
|
Note that these are not thread-safe, and so only one call can be running at a time. In contrast to high-level APIs, trying to call a second async API before the first completes will throw.
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,23 +0,0 @@
|
||||||
#!/bin/bash
|
|
||||||
|
|
||||||
set -euxo pipefail
|
|
||||||
|
|
||||||
|
|
||||||
export ROOT=$PWD
|
|
||||||
|
|
||||||
cd ../../..
|
|
||||||
export CXXFLAGS="-pthread -s USE_PTHREADS=1 -s DISABLE_EXCEPTION_CATCHING=0"
|
|
||||||
export LDFLAGS="-s WASM_BIGINT -s -pthread -s USE_PTHREADS=1"
|
|
||||||
if [ ! -f "build/Makefile" ]; then
|
|
||||||
emconfigure python scripts/mk_make.py --staticlib --single-threaded
|
|
||||||
fi
|
|
||||||
|
|
||||||
cd build
|
|
||||||
emmake make -j$(nproc) libz3.a
|
|
||||||
|
|
||||||
cd $ROOT
|
|
||||||
|
|
||||||
export EM_CACHE=$HOME/.emscripten/
|
|
||||||
export FNS=$(node scripts/list-exports.js)
|
|
||||||
export METHODS='["ccall","FS","allocate","UTF8ToString","intArrayFromString","ALLOC_NORMAL"]'
|
|
||||||
emcc build/async-fns.cc ../../../build/libz3.a --std=c++20 --pre-js src/async-wrapper.js -g2 -pthread -fexceptions -s WASM_BIGINT -s USE_PTHREADS=1 -s PTHREAD_POOL_SIZE=0 -s PTHREAD_POOL_SIZE_STRICT=0 -s MODULARIZE=1 -s 'EXPORT_NAME="initZ3"' -s EXPORTED_RUNTIME_METHODS=$METHODS -s EXPORTED_FUNCTIONS=$FNS -s DISABLE_EXCEPTION_CATCHING=0 -s SAFE_HEAP=0 -s DEMANGLE_SUPPORT=1 -s TOTAL_MEMORY=1GB -I z3/src/api/ -o build/z3-built.js
|
|
||||||
|
|
@ -1,4 +1,5 @@
|
||||||
import { init, Z3_error_code } from './build/node-wrapper';
|
import process from 'process';
|
||||||
|
import { init, Z3_error_code } from '../../build/node';
|
||||||
|
|
||||||
// demonstrates use of the raw API
|
// demonstrates use of the raw API
|
||||||
|
|
||||||
|
|
@ -3,24 +3,24 @@
|
||||||
// TypeScript can infer all of them.
|
// TypeScript can infer all of them.
|
||||||
// They're just here so readers can see what types things are.
|
// They're just here so readers can see what types things are.
|
||||||
|
|
||||||
|
// @ts-ignore we're not going to bother with types for this
|
||||||
|
import process from 'process';
|
||||||
|
import { sprintf } from 'sprintf-js';
|
||||||
import type {
|
import type {
|
||||||
|
Z3_app,
|
||||||
|
Z3_ast,
|
||||||
|
Z3_ast_vector,
|
||||||
Z3_config,
|
Z3_config,
|
||||||
Z3_context,
|
Z3_context,
|
||||||
|
Z3_func_decl,
|
||||||
|
Z3_func_entry,
|
||||||
|
Z3_func_interp,
|
||||||
|
Z3_model,
|
||||||
Z3_solver,
|
Z3_solver,
|
||||||
Z3_sort,
|
Z3_sort,
|
||||||
Z3_ast,
|
|
||||||
Z3_app,
|
|
||||||
Z3_model,
|
|
||||||
Z3_symbol,
|
Z3_symbol,
|
||||||
Z3_ast_vector,
|
} from '../../build/node';
|
||||||
Z3_func_decl,
|
import { init, Z3_ast_kind, Z3_lbool, Z3_sort_kind, Z3_symbol_kind } from '../../build/node';
|
||||||
Z3_func_interp,
|
|
||||||
Z3_func_entry,
|
|
||||||
} from './build/node-wrapper';
|
|
||||||
import { init, Z3_lbool, Z3_ast_kind, Z3_sort_kind, Z3_symbol_kind } from './build/node-wrapper';
|
|
||||||
|
|
||||||
// @ts-ignore we're not going to bother with types for this
|
|
||||||
import { sprintf } from 'sprintf-js';
|
|
||||||
|
|
||||||
let printf = (str: string, ...args: unknown[]) => console.log(sprintf(str.replace(/\n$/, ''), ...args));
|
let printf = (str: string, ...args: unknown[]) => console.log(sprintf(str.replace(/\n$/, ''), ...args));
|
||||||
|
|
||||||
7
src/api/js/jest.config.js
Normal file
7
src/api/js/jest.config.js
Normal file
|
|
@ -0,0 +1,7 @@
|
||||||
|
/** @type {import('ts-jest/dist/types').InitialOptionsTsJest} */
|
||||||
|
module.exports = {
|
||||||
|
preset: 'ts-jest',
|
||||||
|
testEnvironment: 'node',
|
||||||
|
roots: ['<rootDir>/src'],
|
||||||
|
verbose: true,
|
||||||
|
};
|
||||||
11342
src/api/js/package-lock.json
generated
11342
src/api/js/package-lock.json
generated
File diff suppressed because it is too large
Load diff
|
|
@ -1,27 +1,61 @@
|
||||||
{
|
{
|
||||||
"name": "z3-solver",
|
"name": "z3-solver",
|
||||||
"keywords": ["Z3", "theorem", "prover", "solver", "satisfiability", "smt", "satisfiability modulo theories"],
|
"keywords": [
|
||||||
|
"Z3",
|
||||||
|
"theorem",
|
||||||
|
"prover",
|
||||||
|
"solver",
|
||||||
|
"satisfiability",
|
||||||
|
"smt",
|
||||||
|
"satisfiability modulo theories"
|
||||||
|
],
|
||||||
"homepage": "https://github.com/Z3Prover/z3/tree/master/src/api/js",
|
"homepage": "https://github.com/Z3Prover/z3/tree/master/src/api/js",
|
||||||
"repository": "github:Z3Prover/z3",
|
"repository": "github:Z3Prover/z3",
|
||||||
"engines": {
|
"engines": {
|
||||||
"node": ">=16"
|
"node": ">=16 <18"
|
||||||
},
|
},
|
||||||
"main": "build/node-wrapper.js",
|
"browser": "build/browser.js",
|
||||||
"types": "build/node-wrapper.d.ts",
|
"main": "build/node.js",
|
||||||
|
"types": "build/node.d.ts",
|
||||||
"files": [
|
"files": [
|
||||||
"build/*.{js,d.ts,wasm}"
|
"build/*.{js,d.ts,wasm}"
|
||||||
],
|
],
|
||||||
"scripts": {
|
"scripts": {
|
||||||
"build-ts": "mkdir -p build && rm -rf build/*.ts && cp src/node-wrapper.ts build && node scripts/make-ts-wrapper.js > build/wrapper.ts && tsc",
|
"build:ts": "run-s -l build:ts:generate build:ts:tsc",
|
||||||
"build-wasm": "mkdir -p build && node scripts/make-cc-wrapper.js > build/async-fns.cc && ./build-wasm.sh",
|
"build:ts:tsc": "tsc --pretty --project tsconfig.build.json ",
|
||||||
"format": "prettier --write --single-quote --arrow-parens avoid --print-width 120 --trailing-comma all '{,src/,scripts/}*.{js,ts}'",
|
"build:ts:generate": "ts-node --transpileOnly scripts/make-ts-wrapper.ts src/low-level/wrapper.__GENERATED__.ts src/low-level/types.__GENERATED__.ts",
|
||||||
"test": "node test-ts-api.js"
|
"build:wasm": "ts-node --transpileOnly ./scripts/build-wasm.ts",
|
||||||
|
"clean": "rimraf build 'src/**/*.__GENERATED__.*'",
|
||||||
|
"lint": "prettier -c '{,src/,scripts/,examples}*.{js,ts}'",
|
||||||
|
"format": "prettier --write '{,src/,scripts/}*.{js,ts}'",
|
||||||
|
"test": "jest",
|
||||||
|
"docs": "typedoc",
|
||||||
|
"check-engine": "check-engine"
|
||||||
},
|
},
|
||||||
|
"contributors": [
|
||||||
|
"Kevin Gibbons <bakkot@gmail.com>",
|
||||||
|
"Nikolaj Bjorner",
|
||||||
|
"Olaf Tomalka <olaf@tomalka.me>"
|
||||||
|
],
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
|
"@types/jest": "^27.5.1",
|
||||||
"@types/node": "^17.0.8",
|
"@types/node": "^17.0.8",
|
||||||
|
"@types/prettier": "^2.6.1",
|
||||||
|
"@types/sprintf-js": "^1.1.2",
|
||||||
|
"check-engine": "^1.10.1",
|
||||||
|
"iter-tools": "^7.3.1",
|
||||||
|
"jest": "^28.1.0",
|
||||||
|
"npm-run-all": "^4.1.5",
|
||||||
"prettier": "^2.5.1",
|
"prettier": "^2.5.1",
|
||||||
|
"rimraf": "^3.0.2",
|
||||||
"sprintf-js": "^1.1.2",
|
"sprintf-js": "^1.1.2",
|
||||||
|
"ts-jest": "^28.0.3",
|
||||||
|
"ts-node": "^10.8.0",
|
||||||
|
"typedoc": "^0.22.17",
|
||||||
"typescript": "^4.5.4"
|
"typescript": "^4.5.4"
|
||||||
},
|
},
|
||||||
"license": "MIT"
|
"license": "MIT",
|
||||||
|
"dependencies": {
|
||||||
|
"async-mutex": "^0.3.2"
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -1,8 +1,6 @@
|
||||||
'use strict';
|
|
||||||
|
|
||||||
// things which you probably want to do off-thread
|
// things which you probably want to do off-thread
|
||||||
// from https://github.com/Z3Prover/z3/issues/5746#issuecomment-1006289146
|
// from https://github.com/Z3Prover/z3/issues/5746#issuecomment-1006289146
|
||||||
module.exports = [
|
export const asyncFuncs = [
|
||||||
'Z3_eval_smtlib2_string',
|
'Z3_eval_smtlib2_string',
|
||||||
'Z3_simplify',
|
'Z3_simplify',
|
||||||
'Z3_simplify_ex',
|
'Z3_simplify_ex',
|
||||||
77
src/api/js/scripts/build-wasm.ts
Normal file
77
src/api/js/scripts/build-wasm.ts
Normal file
|
|
@ -0,0 +1,77 @@
|
||||||
|
import assert from 'assert';
|
||||||
|
import { SpawnOptions, spawnSync as originalSpawnSync } from 'child_process';
|
||||||
|
import fs, { existsSync } from 'fs';
|
||||||
|
import os from 'os';
|
||||||
|
import path from 'path';
|
||||||
|
import process from 'process';
|
||||||
|
import { asyncFuncs } from './async-fns';
|
||||||
|
import { makeCCWrapper } from './make-cc-wrapper';
|
||||||
|
import { functions } from './parse-api';
|
||||||
|
|
||||||
|
console.log('--- Building WASM');
|
||||||
|
|
||||||
|
const SWAP_OPTS: SpawnOptions = {
|
||||||
|
shell: true,
|
||||||
|
stdio: 'inherit',
|
||||||
|
env: {
|
||||||
|
...process.env,
|
||||||
|
CXXFLAGS: '-pthread -s USE_PTHREADS=1 -s DISABLE_EXCEPTION_CATCHING=0',
|
||||||
|
LDFLAGS: '-s WASM_BIGINT -s -pthread -s USE_PTHREADS=1',
|
||||||
|
FPMATH_ENABLED: 'False', // Until Safari supports WASM SSE, we have to disable fast FP support
|
||||||
|
// TODO(ritave): Setting EM_CACHE breaks compiling on M1 MacBook
|
||||||
|
//EM_CACHE: path.join(os.homedir(), '.emscripten/'),
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
function spawnSync(command: string, opts: SpawnOptions = {}) {
|
||||||
|
console.log(`- ${command}`);
|
||||||
|
// TODO(ritave): Create a splitter that keeps track of quoted strings
|
||||||
|
const [cmd, ...args] = command.split(' ');
|
||||||
|
const { error, ...rest } = originalSpawnSync(cmd, args, { ...SWAP_OPTS, ...opts });
|
||||||
|
if (error !== undefined || rest.status !== 0) {
|
||||||
|
if (error !== undefined) {
|
||||||
|
console.error(error.message);
|
||||||
|
} else {
|
||||||
|
console.error(`Process exited with status ${rest.status}`);
|
||||||
|
}
|
||||||
|
process.exit(1);
|
||||||
|
}
|
||||||
|
return rest;
|
||||||
|
}
|
||||||
|
|
||||||
|
function exportedFuncs(): string[] {
|
||||||
|
const extras = ['_set_throwy_error_handler', '_set_noop_error_handler', ...asyncFuncs.map(f => '_async_' + f)];
|
||||||
|
|
||||||
|
// TODO(ritave): This variable is unused in original script, find out if it's important
|
||||||
|
const fns: any[] = (functions as any[]).filter(f => !asyncFuncs.includes(f.name));
|
||||||
|
|
||||||
|
return [...extras, ...(functions as any[]).map(f => '_' + f.name)];
|
||||||
|
}
|
||||||
|
|
||||||
|
assert(fs.existsSync('./package.json'), 'Not in the root directory of js api');
|
||||||
|
const z3RootDir = path.join(process.cwd(), '../../../');
|
||||||
|
|
||||||
|
// TODO(ritave): Detect if it's in the configuration we need
|
||||||
|
if (!existsSync(path.join(z3RootDir, 'build/Makefile'))) {
|
||||||
|
spawnSync('emconfigure python scripts/mk_make.py --staticlib --single-threaded --arm64=false', {
|
||||||
|
cwd: z3RootDir,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
spawnSync(`emmake make -j${os.cpus().length} libz3.a`, { cwd: path.join(z3RootDir, 'build') });
|
||||||
|
|
||||||
|
const ccWrapperPath = 'build/async-fns.cc';
|
||||||
|
console.log(`- Building ${ccWrapperPath}`);
|
||||||
|
fs.mkdirSync(path.dirname(ccWrapperPath), { recursive: true });
|
||||||
|
fs.writeFileSync(ccWrapperPath, makeCCWrapper());
|
||||||
|
|
||||||
|
const fns = JSON.stringify(exportedFuncs());
|
||||||
|
const methods = '["ccall","FS","allocate","UTF8ToString","intArrayFromString","ALLOC_NORMAL"]';
|
||||||
|
const libz3a = path.normalize('../../../build/libz3.a');
|
||||||
|
spawnSync(
|
||||||
|
`emcc build/async-fns.cc ${libz3a} --std=c++20 --pre-js src/low-level/async-wrapper.js -g2 -pthread -fexceptions -s WASM_BIGINT -s USE_PTHREADS=1 -s PTHREAD_POOL_SIZE=0 -s PTHREAD_POOL_SIZE_STRICT=0 -s MODULARIZE=1 -s 'EXPORT_NAME="initZ3"' -s EXPORTED_RUNTIME_METHODS=${methods} -s EXPORTED_FUNCTIONS=${fns} -s DISABLE_EXCEPTION_CATCHING=0 -s SAFE_HEAP=0 -s DEMANGLE_SUPPORT=1 -s TOTAL_MEMORY=1GB -I z3/src/api/ -o build/z3-built.js`,
|
||||||
|
);
|
||||||
|
|
||||||
|
fs.rmSync(ccWrapperPath);
|
||||||
|
|
||||||
|
console.log('--- WASM build finished');
|
||||||
|
|
@ -1,11 +0,0 @@
|
||||||
'use strict';
|
|
||||||
|
|
||||||
// this is called by build.sh to generate the names of the bindings to export
|
|
||||||
|
|
||||||
let { functions } = require('./parse-api.js');
|
|
||||||
let asyncFns = require('./async-fns.js');
|
|
||||||
|
|
||||||
let extras = ['_set_throwy_error_handler', '_set_noop_error_handler', ...asyncFns.map(f => '_async_' + f)];
|
|
||||||
let fns = functions.filter(f => !asyncFns.includes(f.name));
|
|
||||||
|
|
||||||
console.log(JSON.stringify([...extras, ...functions.map(f => '_' + f.name)]));
|
|
||||||
|
|
@ -1,40 +1,38 @@
|
||||||
'use strict';
|
|
||||||
|
|
||||||
// generates c wrappers with off-thread versions of specified functions
|
// generates c wrappers with off-thread versions of specified functions
|
||||||
|
|
||||||
let path = require('path');
|
import path from 'path';
|
||||||
|
import { asyncFuncs } from './async-fns';
|
||||||
|
import { functions } from './parse-api';
|
||||||
|
|
||||||
let { functions } = require('./parse-api.js');
|
export function makeCCWrapper() {
|
||||||
let asyncFns = require('./async-fns.js');
|
let wrappers = [];
|
||||||
|
|
||||||
let wrappers = [];
|
for (let fnName of asyncFuncs) {
|
||||||
|
let fn = functions.find(f => f.name === fnName);
|
||||||
|
if (fn == null) {
|
||||||
|
throw new Error(`could not find definition for ${fnName}`);
|
||||||
|
}
|
||||||
|
let wrapper;
|
||||||
|
if (fn.cRet === 'Z3_string') {
|
||||||
|
wrapper = `wrapper_str`;
|
||||||
|
} else if (['int', 'unsigned', 'void'].includes(fn.cRet) || fn.cRet.startsWith('Z3_')) {
|
||||||
|
wrapper = `wrapper`;
|
||||||
|
} else {
|
||||||
|
throw new Error(`async function with unknown return type ${fn.cRet}`);
|
||||||
|
}
|
||||||
|
|
||||||
for (let fnName of asyncFns) {
|
wrappers.push(
|
||||||
let fn = functions.find(f => f.name === fnName);
|
`
|
||||||
if (fn == null) {
|
|
||||||
throw new Error(`could not find definition for ${fnName}`);
|
|
||||||
}
|
|
||||||
let wrapper;
|
|
||||||
if (fn.cRet === 'Z3_string') {
|
|
||||||
wrapper = `wrapper_str`;
|
|
||||||
} else if (['int', 'unsigned', 'void'].includes(fn.cRet) || fn.cRet.startsWith('Z3_')) {
|
|
||||||
wrapper = `wrapper`;
|
|
||||||
} else {
|
|
||||||
throw new Error(`async function with unknown return type ${fn.cRet}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
wrappers.push(
|
|
||||||
`
|
|
||||||
extern "C" void async_${fn.name}(${fn.params
|
extern "C" void async_${fn.name}(${fn.params
|
||||||
.map(p => `${p.isConst ? 'const ' : ''}${p.cType}${p.isPtr ? '*' : ''} ${p.name}${p.isArray ? '[]' : ''}`)
|
.map(p => `${p.isConst ? 'const ' : ''}${p.cType}${p.isPtr ? '*' : ''} ${p.name}${p.isArray ? '[]' : ''}`)
|
||||||
.join(', ')}) {
|
.join(', ')}) {
|
||||||
${wrapper}<decltype(&${fn.name}), &${fn.name}>(${fn.params.map(p => `${p.name}`).join(', ')});
|
${wrapper}<decltype(&${fn.name}), &${fn.name}>(${fn.params.map(p => `${p.name}`).join(', ')});
|
||||||
}
|
}
|
||||||
`.trim(),
|
`.trim(),
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
console.log(`// THIS FILE IS AUTOMATICALLY GENERATED BY ${path.basename(__filename)}
|
return `// THIS FILE IS AUTOMATICALLY GENERATED BY ${path.basename(__filename)}
|
||||||
// DO NOT EDIT IT BY HAND
|
// DO NOT EDIT IT BY HAND
|
||||||
|
|
||||||
#include <thread>
|
#include <thread>
|
||||||
|
|
@ -112,4 +110,10 @@ extern "C" void set_noop_error_handler(Z3_context ctx) {
|
||||||
Z3_set_error_handler(ctx, noop_error_handler);
|
Z3_set_error_handler(ctx, noop_error_handler);
|
||||||
}
|
}
|
||||||
|
|
||||||
${wrappers.join('\n\n')}`);
|
${wrappers.join('\n\n')}
|
||||||
|
`;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (require.main === module) {
|
||||||
|
console.log(makeCCWrapper());
|
||||||
|
}
|
||||||
|
|
@ -1,434 +0,0 @@
|
||||||
'use strict';
|
|
||||||
|
|
||||||
let path = require('path');
|
|
||||||
let prettier = require('prettier');
|
|
||||||
|
|
||||||
let { primitiveTypes, types, enums, functions } = require('./parse-api.js');
|
|
||||||
let asyncFns = require('./async-fns.js');
|
|
||||||
|
|
||||||
let subtypes = {
|
|
||||||
__proto__: null,
|
|
||||||
Z3_sort: 'Z3_ast',
|
|
||||||
Z3_func_decl: 'Z3_ast',
|
|
||||||
};
|
|
||||||
|
|
||||||
let makePointerType = t =>
|
|
||||||
`export type ${t} = ` + (t in subtypes ? `Subpointer<'${t}', '${subtypes[t]}'>;` : `Pointer<'${t}'>;`);
|
|
||||||
|
|
||||||
// this supports a up to 6 out intergers/pointers
|
|
||||||
// or up to 3 out int64s
|
|
||||||
const BYTES_TO_ALLOCATE_FOR_OUT_PARAMS = 24;
|
|
||||||
|
|
||||||
const CUSTOM_IMPLEMENTATIONS = ['Z3_mk_context', 'Z3_mk_context_rc'];
|
|
||||||
|
|
||||||
function toEmType(type) {
|
|
||||||
if (type in primitiveTypes) {
|
|
||||||
type = primitiveTypes[type];
|
|
||||||
}
|
|
||||||
if (['boolean', 'number', 'string', 'bigint', 'void'].includes(type)) {
|
|
||||||
return type;
|
|
||||||
}
|
|
||||||
if (type.startsWith('Z3_')) {
|
|
||||||
return 'number';
|
|
||||||
}
|
|
||||||
throw new Error(`unknown parameter type ${type}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
function isZ3PointerType(type) {
|
|
||||||
return type.startsWith('Z3_');
|
|
||||||
}
|
|
||||||
|
|
||||||
function toEm(p) {
|
|
||||||
if (typeof p === 'string') {
|
|
||||||
// we've already set this, e.g. by replacing it with an expression
|
|
||||||
return p;
|
|
||||||
}
|
|
||||||
let { type } = p;
|
|
||||||
if (p.kind === 'out') {
|
|
||||||
throw new Error(`unknown out parameter type ${JSON.stringify(p)}`);
|
|
||||||
}
|
|
||||||
if (p.isArray) {
|
|
||||||
if (isZ3PointerType(type) || type === 'unsigned' || type === 'int') {
|
|
||||||
// this works for nullables also because null coerces to 0
|
|
||||||
return `intArrayToByteArr(${p.name} as unknown as number[])`;
|
|
||||||
} else if (type === 'boolean') {
|
|
||||||
return `boolArrayToByteArr(${p.name})`;
|
|
||||||
} else {
|
|
||||||
throw new Error(`only know how to deal with arrays of int/bool (got ${type})`);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (type in primitiveTypes) {
|
|
||||||
type = primitiveTypes[type];
|
|
||||||
}
|
|
||||||
|
|
||||||
if (['boolean', 'number', 'bigint', 'string'].includes(type)) {
|
|
||||||
return p.name;
|
|
||||||
}
|
|
||||||
if (type.startsWith('Z3_')) {
|
|
||||||
return p.name;
|
|
||||||
}
|
|
||||||
throw new Error(`unknown parameter type ${JSON.stringify(p)}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
let isInParam = p => ['in', 'in_array'].includes(p.kind);
|
|
||||||
function wrapFunction(fn) {
|
|
||||||
if (CUSTOM_IMPLEMENTATIONS.includes(fn.name)) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
let inParams = fn.params.filter(isInParam);
|
|
||||||
let outParams = fn.params.map((p, idx) => ({ ...p, idx })).filter(p => !isInParam(p));
|
|
||||||
|
|
||||||
// we'll figure out how to deal with these cases later
|
|
||||||
let unknownInParam = inParams.find(
|
|
||||||
p =>
|
|
||||||
p.isPtr ||
|
|
||||||
p.type === 'Z3_char_ptr' ||
|
|
||||||
(p.isArray && !(isZ3PointerType(p.type) || p.type === 'unsigned' || p.type === 'int' || p.type === 'boolean')),
|
|
||||||
);
|
|
||||||
if (unknownInParam) {
|
|
||||||
console.error(`skipping ${fn.name} - unknown in parameter ${JSON.stringify(unknownInParam)}`);
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (fn.ret === 'Z3_char_ptr') {
|
|
||||||
console.error(`skipping ${fn.name} - returns a string or char pointer`);
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
// console.error(fn.name);
|
|
||||||
|
|
||||||
let isAsync = asyncFns.includes(fn.name);
|
|
||||||
let trivial =
|
|
||||||
!['string', 'boolean'].includes(fn.ret) &&
|
|
||||||
!fn.nullableRet &&
|
|
||||||
outParams.length === 0 &&
|
|
||||||
!inParams.some(p => p.type === 'string' || p.isArray || p.nullable);
|
|
||||||
|
|
||||||
let name = fn.name.startsWith('Z3_') ? fn.name.substring(3) : fn.name;
|
|
||||||
|
|
||||||
let params = inParams.map(p => {
|
|
||||||
let type = p.type;
|
|
||||||
if (p.isArray && p.nullable) {
|
|
||||||
type = `(${type} | null)[]`;
|
|
||||||
} else if (p.isArray) {
|
|
||||||
type = `${type}[]`;
|
|
||||||
} else if (p.nullable) {
|
|
||||||
type = `${type} | null`;
|
|
||||||
}
|
|
||||||
return `${p.name}: ${type}`;
|
|
||||||
});
|
|
||||||
|
|
||||||
if (trivial && isAsync) {
|
|
||||||
// i.e. and async
|
|
||||||
return `${name}: function (${params.join(', ')}): Promise<${fn.ret}> {
|
|
||||||
return Mod.async_call(Mod._async_${fn.name}, ${fn.params.map(toEm).join(', ')});
|
|
||||||
}`;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (trivial) {
|
|
||||||
return `${name}: Mod._${fn.name} as ((${params.join(', ')}) => ${fn.ret})`;
|
|
||||||
}
|
|
||||||
|
|
||||||
// otherwise fall back to ccall
|
|
||||||
|
|
||||||
let ctypes = fn.params.map(p =>
|
|
||||||
p.kind === 'in_array' ? 'array' : p.kind === 'out_array' ? 'number' : p.isPtr ? 'number' : toEmType(p.type),
|
|
||||||
);
|
|
||||||
|
|
||||||
let prefix = '';
|
|
||||||
let infix = '';
|
|
||||||
let rv = 'ret';
|
|
||||||
let suffix = '';
|
|
||||||
|
|
||||||
let args = fn.params;
|
|
||||||
|
|
||||||
let arrayLengthParams = new Map();
|
|
||||||
for (let p of inParams) {
|
|
||||||
if (p.nullable && !p.isArray) {
|
|
||||||
// this would be easy to implement - just map null to 0 - but nothing actually uses nullable non-array input parameters, so we can't ensure we've done it right
|
|
||||||
console.error(`skipping ${fn.name} - nullable input parameter`);
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
if (!p.isArray) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
let { sizeIndex } = p;
|
|
||||||
if (arrayLengthParams.has(sizeIndex)) {
|
|
||||||
let otherParam = arrayLengthParams.get(sizeIndex);
|
|
||||||
prefix += `
|
|
||||||
if (${otherParam}.length !== ${p.name}.length) {
|
|
||||||
throw new TypeError(\`${otherParam} and ${p.name} must be the same length (got \${${otherParam}.length} and \{${p.name}.length})\`);
|
|
||||||
}
|
|
||||||
`.trim();
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
arrayLengthParams.set(sizeIndex, p.name);
|
|
||||||
|
|
||||||
let sizeParam = fn.params[sizeIndex];
|
|
||||||
if (!(sizeParam.kind === 'in' && sizeParam.type === 'unsigned' && !sizeParam.isPtr && !sizeParam.isArray)) {
|
|
||||||
throw new Error(
|
|
||||||
`size index is not unsigned int (for fn ${fn.name} parameter ${sizeIndex} got ${sizeParam.type})`,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
args[sizeIndex] = `${p.name}.length`;
|
|
||||||
params[sizeIndex] = null;
|
|
||||||
}
|
|
||||||
|
|
||||||
let returnType = fn.ret;
|
|
||||||
let cReturnType = toEmType(fn.ret);
|
|
||||||
if (outParams.length > 0) {
|
|
||||||
let mapped = [];
|
|
||||||
let memIdx = 0; // offset from `outAddress` where the data should get written, in units of 4 bytes
|
|
||||||
|
|
||||||
for (let outParam of outParams) {
|
|
||||||
if (outParam.isArray) {
|
|
||||||
if (isZ3PointerType(outParam.type) || outParam.type === 'unsigned') {
|
|
||||||
let { sizeIndex } = outParam;
|
|
||||||
|
|
||||||
let count;
|
|
||||||
if (arrayLengthParams.has(sizeIndex)) {
|
|
||||||
// i.e. this is also the length of an input array
|
|
||||||
count = args[sizeIndex];
|
|
||||||
} else {
|
|
||||||
let sizeParam = fn.params[sizeIndex];
|
|
||||||
if (!(sizeParam.kind === 'in' && sizeParam.type === 'unsigned' && !sizeParam.isPtr && !sizeParam.isArray)) {
|
|
||||||
throw new Error(
|
|
||||||
`size index is not unsigned int (for fn ${fn.name} parameter ${sizeIndex} got ${sizeParam.type})`,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
count = sizeParam.name;
|
|
||||||
}
|
|
||||||
let outArrayAddress = `outArray_${outParam.name}`;
|
|
||||||
prefix += `
|
|
||||||
let ${outArrayAddress} = Mod._malloc(4 * ${count});
|
|
||||||
try {
|
|
||||||
`.trim();
|
|
||||||
suffix =
|
|
||||||
`
|
|
||||||
} finally {
|
|
||||||
Mod._free(${outArrayAddress});
|
|
||||||
}
|
|
||||||
`.trim() + suffix;
|
|
||||||
args[outParam.idx] = outArrayAddress;
|
|
||||||
mapped.push({
|
|
||||||
name: outParam.name,
|
|
||||||
read:
|
|
||||||
`readUintArray(${outArrayAddress}, ${count})` +
|
|
||||||
(outParam.type === 'unsigned' ? '' : `as unknown as ${outParam.type}[]`),
|
|
||||||
type: `${outParam.type}[]`,
|
|
||||||
});
|
|
||||||
} else {
|
|
||||||
console.error(`skipping ${fn.name} - out array of ${outParam.type}`);
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
} else if (outParam.isPtr) {
|
|
||||||
function setArg() {
|
|
||||||
args[outParam.idx] = memIdx === 0 ? 'outAddress' : `outAddress + ${memIdx * 4}`;
|
|
||||||
}
|
|
||||||
let read, type;
|
|
||||||
if (outParam.type === 'string') {
|
|
||||||
read = `Mod.UTF8ToString(getOutUint(${memIdx}))`;
|
|
||||||
setArg();
|
|
||||||
++memIdx;
|
|
||||||
} else if (isZ3PointerType(outParam.type)) {
|
|
||||||
read = `getOutUint(${memIdx}) as unknown as ${outParam.type}`;
|
|
||||||
setArg();
|
|
||||||
++memIdx;
|
|
||||||
} else if (outParam.type === 'unsigned') {
|
|
||||||
read = `getOutUint(${memIdx})`;
|
|
||||||
setArg();
|
|
||||||
++memIdx;
|
|
||||||
} else if (outParam.type === 'int') {
|
|
||||||
read = `getOutInt(${memIdx})`;
|
|
||||||
setArg();
|
|
||||||
++memIdx;
|
|
||||||
} else if (outParam.type === 'uint64_t') {
|
|
||||||
if (memIdx % 2 === 1) {
|
|
||||||
++memIdx;
|
|
||||||
}
|
|
||||||
read = `getOutUint64(${memIdx / 2})`;
|
|
||||||
setArg();
|
|
||||||
memIdx += 2;
|
|
||||||
} else if (outParam.type === 'int64_t') {
|
|
||||||
if (memIdx % 2 === 1) {
|
|
||||||
++memIdx;
|
|
||||||
}
|
|
||||||
read = `getOutInt64(${memIdx / 2})`;
|
|
||||||
setArg();
|
|
||||||
memIdx += 2;
|
|
||||||
} else {
|
|
||||||
console.error(`skipping ${fn.name} - unknown out parameter type ${outParam.type}`);
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
if (memIdx > Math.floor(BYTES_TO_ALLOCATE_FOR_OUT_PARAMS / 4)) {
|
|
||||||
// prettier-ignore
|
|
||||||
console.error(`skipping ${fn.name} - out parameter sizes sum to ${memIdx * 4}, which is > ${BYTES_TO_ALLOCATE_FOR_OUT_PARAMS}`);
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
mapped.push({
|
|
||||||
name: outParam.name,
|
|
||||||
read,
|
|
||||||
type: outParam.type,
|
|
||||||
});
|
|
||||||
} else {
|
|
||||||
console.error(`skipping ${fn.name} - out param is neither pointer nor array`);
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
let ignoreReturn = fn.ret === 'boolean' || fn.ret === 'void';
|
|
||||||
if (outParams.length === 1) {
|
|
||||||
let outParam = mapped[0];
|
|
||||||
if (ignoreReturn) {
|
|
||||||
returnType = outParam.type;
|
|
||||||
rv = outParam.read;
|
|
||||||
} else {
|
|
||||||
returnType = `{ rv: ${fn.ret}, ${outParam.name} : ${outParam.type} }`;
|
|
||||||
rv = `{ rv: ret, ${outParam.name} : ${outParam.read} }`;
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
if (ignoreReturn) {
|
|
||||||
returnType = `{ ${mapped.map(p => `${p.name} : ${p.type}`).join(', ')} }`;
|
|
||||||
rv = `{ ${mapped.map(p => `${p.name}: ${p.read}`).join(', ')} }`;
|
|
||||||
} else {
|
|
||||||
returnType = `{ rv: ${fn.ret}, ${mapped.map(p => `${p.name} : ${p.type}`).join(', ')} }`;
|
|
||||||
rv = `{ rv: ret, ${mapped.map(p => `${p.name}: ${p.read}`).join(', ')} }`;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (fn.ret === 'boolean') {
|
|
||||||
// assume the boolean indicates success
|
|
||||||
infix += `
|
|
||||||
if (!ret) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
`.trim();
|
|
||||||
cReturnType = 'boolean';
|
|
||||||
returnType += ' | null';
|
|
||||||
} else if (fn.ret === 'void') {
|
|
||||||
cReturnType = 'void';
|
|
||||||
} else if (isZ3PointerType(fn.ret) || fn.ret === 'unsigned') {
|
|
||||||
cReturnType = 'number';
|
|
||||||
} else {
|
|
||||||
console.error(`skipping ${fn.name} - out parameter for function which returns non-boolean`);
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (fn.nullableRet) {
|
|
||||||
returnType += ' | null';
|
|
||||||
infix += `
|
|
||||||
if (ret === 0) {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
`.trim();
|
|
||||||
}
|
|
||||||
|
|
||||||
// prettier-ignore
|
|
||||||
let invocation = `Mod.ccall('${isAsync ? 'async_' : ''}${fn.name}', '${cReturnType}', ${JSON.stringify(ctypes)}, [${args.map(toEm).join(', ')}])`;
|
|
||||||
|
|
||||||
if (isAsync) {
|
|
||||||
invocation = `await Mod.async_call(() => ${invocation})`;
|
|
||||||
returnType = `Promise<${returnType}>`;
|
|
||||||
}
|
|
||||||
|
|
||||||
let out = `${name}: ${isAsync ? 'async' : ''} function(${params.filter(p => p != null).join(', ')}): ${returnType} {
|
|
||||||
${prefix}`;
|
|
||||||
if (infix === '' && suffix === '' && rv === 'ret') {
|
|
||||||
out += `return ${invocation};`;
|
|
||||||
} else {
|
|
||||||
out += `
|
|
||||||
let ret = ${invocation};
|
|
||||||
${infix}return ${rv};${suffix}
|
|
||||||
`.trim();
|
|
||||||
}
|
|
||||||
out += '}';
|
|
||||||
return out;
|
|
||||||
}
|
|
||||||
|
|
||||||
function wrapEnum(name, values) {
|
|
||||||
let enumEntries = Object.entries(values);
|
|
||||||
return `export enum ${name} {
|
|
||||||
${enumEntries.map(([k, v], i) => k + (v === (enumEntries[i - 1]?.[1] ?? -1) + 1 ? '' : ` = ${v}`) + ',').join('\n')}
|
|
||||||
};`;
|
|
||||||
}
|
|
||||||
|
|
||||||
function getValidOutArrayIndexes(size) {
|
|
||||||
return Array.from({ length: Math.floor(BYTES_TO_ALLOCATE_FOR_OUT_PARAMS / size) }, (_, i) => i).join(' | ');
|
|
||||||
}
|
|
||||||
|
|
||||||
let out = `
|
|
||||||
// THIS FILE IS AUTOMATICALLY GENERATED BY ${path.basename(__filename)}
|
|
||||||
// DO NOT EDIT IT BY HAND
|
|
||||||
|
|
||||||
interface Pointer<T extends string> extends Number {
|
|
||||||
readonly __typeName: T;
|
|
||||||
}
|
|
||||||
interface Subpointer<T extends string, S extends string> extends Pointer<S> {
|
|
||||||
readonly __typeName2: T;
|
|
||||||
}
|
|
||||||
|
|
||||||
${Object.entries(primitiveTypes)
|
|
||||||
.filter(e => e[0] !== 'void')
|
|
||||||
.map(e => `type ${e[0]} = ${e[1]};`)
|
|
||||||
.join('\n')}
|
|
||||||
|
|
||||||
${Object.keys(types)
|
|
||||||
.filter(k => k.startsWith('Z3'))
|
|
||||||
.map(makePointerType)
|
|
||||||
.join('\n')}
|
|
||||||
|
|
||||||
${Object.entries(enums)
|
|
||||||
.map(e => wrapEnum(e[0], e[1]))
|
|
||||||
.join('\n\n')}
|
|
||||||
|
|
||||||
export async function init(initModule: any) {
|
|
||||||
let Mod = await initModule();
|
|
||||||
|
|
||||||
// this works for both signed and unsigned, because JS will wrap for you when constructing the Uint32Array
|
|
||||||
function intArrayToByteArr(ints: number[]) {
|
|
||||||
return new Uint8Array((new Uint32Array(ints)).buffer);
|
|
||||||
}
|
|
||||||
|
|
||||||
function boolArrayToByteArr(bools: boolean[]) {
|
|
||||||
return bools.map(b => b ? 1 : 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
function readUintArray(address: number, count: number) {
|
|
||||||
return Array.from(new Uint32Array(Mod.HEAPU32.buffer, address, count));
|
|
||||||
}
|
|
||||||
|
|
||||||
let outAddress = Mod._malloc(${BYTES_TO_ALLOCATE_FOR_OUT_PARAMS});
|
|
||||||
let outUintArray = (new Uint32Array(Mod.HEAPU32.buffer, outAddress, 4));
|
|
||||||
let getOutUint = (i: ${getValidOutArrayIndexes(4)}) => outUintArray[i];
|
|
||||||
let outIntArray = (new Int32Array(Mod.HEAPU32.buffer, outAddress, 4));
|
|
||||||
let getOutInt = (i: ${getValidOutArrayIndexes(4)}) => outIntArray[i];
|
|
||||||
let outUint64Array = (new BigUint64Array(Mod.HEAPU32.buffer, outAddress, 2));
|
|
||||||
let getOutUint64 = (i: ${getValidOutArrayIndexes(8)}) => outUint64Array[i];
|
|
||||||
let outInt64Array = (new BigInt64Array(Mod.HEAPU32.buffer, outAddress, 2));
|
|
||||||
let getOutInt64 = (i: ${getValidOutArrayIndexes(8)}) => outInt64Array[i];
|
|
||||||
|
|
||||||
return {
|
|
||||||
em: Mod,
|
|
||||||
Z3: {
|
|
||||||
mk_context: function(c: Z3_config): Z3_context {
|
|
||||||
let ctx = Mod._Z3_mk_context(c);
|
|
||||||
Mod._set_noop_error_handler(ctx);
|
|
||||||
return ctx;
|
|
||||||
},
|
|
||||||
mk_context_rc: function(c: Z3_config): Z3_context {
|
|
||||||
let ctx = Mod._Z3_mk_context_rc(c);
|
|
||||||
Mod._set_noop_error_handler(ctx);
|
|
||||||
return ctx;
|
|
||||||
},
|
|
||||||
${functions
|
|
||||||
.map(wrapFunction)
|
|
||||||
.filter(f => f != null)
|
|
||||||
.join(',\n')}
|
|
||||||
|
|
||||||
}
|
|
||||||
};
|
|
||||||
}
|
|
||||||
`;
|
|
||||||
|
|
||||||
console.log(prettier.format(out, { singleQuote: true, parser: 'typescript' }));
|
|
||||||
468
src/api/js/scripts/make-ts-wrapper.ts
Normal file
468
src/api/js/scripts/make-ts-wrapper.ts
Normal file
|
|
@ -0,0 +1,468 @@
|
||||||
|
import assert from 'assert';
|
||||||
|
import fs from 'fs';
|
||||||
|
import path from 'path';
|
||||||
|
import prettier from 'prettier';
|
||||||
|
import { asyncFuncs } from './async-fns';
|
||||||
|
import { enums, Func, FuncParam, functions, primitiveTypes, types } from './parse-api';
|
||||||
|
|
||||||
|
assert(process.argv.length === 4, `Usage: ${process.argv[0]} ${process.argv[1]} wrapperFilePath typesFilePath`);
|
||||||
|
|
||||||
|
const wrapperFilePath = process.argv[2];
|
||||||
|
const typesFilePath = process.argv[3];
|
||||||
|
|
||||||
|
function makeTsWrapper() {
|
||||||
|
const subtypes = {
|
||||||
|
__proto__: null,
|
||||||
|
Z3_sort: 'Z3_ast',
|
||||||
|
Z3_func_decl: 'Z3_ast',
|
||||||
|
} as unknown as Record<string, string>;
|
||||||
|
|
||||||
|
const makePointerType = (t: string) =>
|
||||||
|
`export type ${t} = ` + (t in subtypes ? `Subpointer<'${t}', '${subtypes[t]}'>;` : `Pointer<'${t}'>;`);
|
||||||
|
|
||||||
|
// this supports a up to 6 out integers/pointers
|
||||||
|
// or up to 3 out int64s
|
||||||
|
const BYTES_TO_ALLOCATE_FOR_OUT_PARAMS = 24;
|
||||||
|
|
||||||
|
const CUSTOM_IMPLEMENTATIONS = ['Z3_mk_context', 'Z3_mk_context_rc'];
|
||||||
|
|
||||||
|
function toEmType(type: string) {
|
||||||
|
if (type in primitiveTypes) {
|
||||||
|
type = primitiveTypes[type];
|
||||||
|
}
|
||||||
|
if (['boolean', 'number', 'string', 'bigint', 'void'].includes(type)) {
|
||||||
|
return type;
|
||||||
|
}
|
||||||
|
if (type.startsWith('Z3_')) {
|
||||||
|
return 'number';
|
||||||
|
}
|
||||||
|
throw new Error(`unknown parameter type ${type}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
function isZ3PointerType(type: string) {
|
||||||
|
return type.startsWith('Z3_');
|
||||||
|
}
|
||||||
|
|
||||||
|
function toEm(p: string | FuncParam) {
|
||||||
|
if (typeof p === 'string') {
|
||||||
|
// we've already set this, e.g. by replacing it with an expression
|
||||||
|
return p;
|
||||||
|
}
|
||||||
|
let { type } = p;
|
||||||
|
if (p.kind === 'out') {
|
||||||
|
throw new Error(`unknown out parameter type ${JSON.stringify(p)}`);
|
||||||
|
}
|
||||||
|
if (p.isArray) {
|
||||||
|
if (isZ3PointerType(type) || type === 'unsigned' || type === 'int') {
|
||||||
|
// this works for nullables also because null coerces to 0
|
||||||
|
return `intArrayToByteArr(${p.name} as unknown as number[])`;
|
||||||
|
} else if (type === 'boolean') {
|
||||||
|
return `boolArrayToByteArr(${p.name})`;
|
||||||
|
} else {
|
||||||
|
throw new Error(`only know how to deal with arrays of int/bool (got ${type})`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (type in primitiveTypes) {
|
||||||
|
type = primitiveTypes[type];
|
||||||
|
}
|
||||||
|
|
||||||
|
if (['boolean', 'number', 'bigint', 'string'].includes(type)) {
|
||||||
|
return p.name;
|
||||||
|
}
|
||||||
|
if (type.startsWith('Z3_')) {
|
||||||
|
return p.name;
|
||||||
|
}
|
||||||
|
throw new Error(`unknown parameter type ${JSON.stringify(p)}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
const isInParam = (p: FuncParam) => p.kind !== undefined && ['in', 'in_array'].includes(p.kind);
|
||||||
|
function wrapFunction(fn: Func) {
|
||||||
|
if (CUSTOM_IMPLEMENTATIONS.includes(fn.name)) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
let inParams = fn.params.filter(isInParam);
|
||||||
|
let outParams = fn.params.map((p, idx) => ({ ...p, idx })).filter(p => !isInParam(p));
|
||||||
|
|
||||||
|
// we'll figure out how to deal with these cases later
|
||||||
|
let unknownInParam = inParams.find(
|
||||||
|
p =>
|
||||||
|
p.isPtr ||
|
||||||
|
p.type === 'Z3_char_ptr' ||
|
||||||
|
(p.isArray && !(isZ3PointerType(p.type) || p.type === 'unsigned' || p.type === 'int' || p.type === 'boolean')),
|
||||||
|
);
|
||||||
|
if (unknownInParam) {
|
||||||
|
console.error(`skipping ${fn.name} - unknown in parameter ${JSON.stringify(unknownInParam)}`);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (fn.ret === 'Z3_char_ptr') {
|
||||||
|
console.error(`skipping ${fn.name} - returns a string or char pointer`);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
// console.error(fn.name);
|
||||||
|
|
||||||
|
let isAsync = asyncFuncs.includes(fn.name);
|
||||||
|
let trivial =
|
||||||
|
!['string', 'boolean'].includes(fn.ret) &&
|
||||||
|
!fn.nullableRet &&
|
||||||
|
outParams.length === 0 &&
|
||||||
|
!inParams.some(p => p.type === 'string' || p.isArray || p.nullable);
|
||||||
|
|
||||||
|
let name = fn.name.startsWith('Z3_') ? fn.name.substring(3) : fn.name;
|
||||||
|
|
||||||
|
const params: (string | null)[] = inParams.map(p => {
|
||||||
|
let type = p.type;
|
||||||
|
if (p.isArray && p.nullable) {
|
||||||
|
type = `(${type} | null)[]`;
|
||||||
|
} else if (p.isArray) {
|
||||||
|
type = `${type}[]`;
|
||||||
|
} else if (p.nullable) {
|
||||||
|
type = `${type} | null`;
|
||||||
|
}
|
||||||
|
return `${p.name}: ${type}`;
|
||||||
|
});
|
||||||
|
|
||||||
|
if (trivial && isAsync) {
|
||||||
|
// i.e. and async
|
||||||
|
return `${name}: function (${params.join(', ')}): Promise<${fn.ret}> {
|
||||||
|
return Mod.async_call(Mod._async_${fn.name}, ${fn.params.map(toEm).join(', ')});
|
||||||
|
}`;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (trivial) {
|
||||||
|
return `${name}: Mod._${fn.name} as ((${params.join(', ')}) => ${fn.ret})`;
|
||||||
|
}
|
||||||
|
|
||||||
|
// otherwise fall back to ccall
|
||||||
|
|
||||||
|
const ctypes = fn.params.map(p =>
|
||||||
|
p.kind === 'in_array' ? 'array' : p.kind === 'out_array' ? 'number' : p.isPtr ? 'number' : toEmType(p.type),
|
||||||
|
);
|
||||||
|
|
||||||
|
let prefix = '';
|
||||||
|
let infix = '';
|
||||||
|
let rv = 'ret';
|
||||||
|
let suffix = '';
|
||||||
|
|
||||||
|
const args: (string | FuncParam)[] = fn.params;
|
||||||
|
|
||||||
|
let arrayLengthParams = new Map();
|
||||||
|
for (let p of inParams) {
|
||||||
|
if (p.nullable && !p.isArray) {
|
||||||
|
// this would be easy to implement - just map null to 0 - but nothing actually uses nullable non-array input parameters, so we can't ensure we've done it right
|
||||||
|
console.error(`skipping ${fn.name} - nullable input parameter`);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
if (!p.isArray) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
let { sizeIndex } = p;
|
||||||
|
assert(sizeIndex !== undefined);
|
||||||
|
if (arrayLengthParams.has(sizeIndex)) {
|
||||||
|
let otherParam = arrayLengthParams.get(sizeIndex);
|
||||||
|
prefix += `
|
||||||
|
if (${otherParam}.length !== ${p.name}.length) {
|
||||||
|
throw new TypeError(\`${otherParam} and ${p.name} must be the same length (got \${${otherParam}.length} and \{${p.name}.length})\`);
|
||||||
|
}
|
||||||
|
`.trim();
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
arrayLengthParams.set(sizeIndex, p.name);
|
||||||
|
|
||||||
|
const sizeParam = fn.params[sizeIndex];
|
||||||
|
if (!(sizeParam.kind === 'in' && sizeParam.type === 'unsigned' && !sizeParam.isPtr && !sizeParam.isArray)) {
|
||||||
|
throw new Error(
|
||||||
|
`size index is not unsigned int (for fn ${fn.name} parameter ${sizeIndex} got ${sizeParam.type})`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
args[sizeIndex] = `${p.name}.length`;
|
||||||
|
params[sizeIndex] = null;
|
||||||
|
}
|
||||||
|
|
||||||
|
let returnType = fn.ret;
|
||||||
|
let cReturnType = toEmType(fn.ret);
|
||||||
|
if (outParams.length > 0) {
|
||||||
|
let mapped = [];
|
||||||
|
let memIdx = 0; // offset from `outAddress` where the data should get written, in units of 4 bytes
|
||||||
|
|
||||||
|
for (let outParam of outParams) {
|
||||||
|
if (outParam.isArray) {
|
||||||
|
if (isZ3PointerType(outParam.type) || outParam.type === 'unsigned') {
|
||||||
|
let { sizeIndex } = outParam;
|
||||||
|
assert(sizeIndex !== undefined);
|
||||||
|
|
||||||
|
let count;
|
||||||
|
if (arrayLengthParams.has(sizeIndex)) {
|
||||||
|
// i.e. this is also the length of an input array
|
||||||
|
count = args[sizeIndex];
|
||||||
|
} else {
|
||||||
|
let sizeParam = fn.params[sizeIndex];
|
||||||
|
if (
|
||||||
|
!(sizeParam.kind === 'in' && sizeParam.type === 'unsigned' && !sizeParam.isPtr && !sizeParam.isArray)
|
||||||
|
) {
|
||||||
|
throw new Error(
|
||||||
|
`size index is not unsigned int (for fn ${fn.name} parameter ${sizeIndex} got ${sizeParam.type})`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
count = sizeParam.name;
|
||||||
|
}
|
||||||
|
let outArrayAddress = `outArray_${outParam.name}`;
|
||||||
|
prefix += `
|
||||||
|
let ${outArrayAddress} = Mod._malloc(4 * ${count});
|
||||||
|
try {
|
||||||
|
`.trim();
|
||||||
|
suffix =
|
||||||
|
`
|
||||||
|
} finally {
|
||||||
|
Mod._free(${outArrayAddress});
|
||||||
|
}
|
||||||
|
`.trim() + suffix;
|
||||||
|
args[outParam.idx] = outArrayAddress;
|
||||||
|
mapped.push({
|
||||||
|
name: outParam.name,
|
||||||
|
read:
|
||||||
|
`readUintArray(${outArrayAddress}, ${count})` +
|
||||||
|
(outParam.type === 'unsigned' ? '' : `as unknown as ${outParam.type}[]`),
|
||||||
|
type: `${outParam.type}[]`,
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
console.error(`skipping ${fn.name} - out array of ${outParam.type}`);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
} else if (outParam.isPtr) {
|
||||||
|
function setArg() {
|
||||||
|
args[outParam.idx] = memIdx === 0 ? 'outAddress' : `outAddress + ${memIdx * 4}`;
|
||||||
|
}
|
||||||
|
let read, type;
|
||||||
|
if (outParam.type === 'string') {
|
||||||
|
read = `Mod.UTF8ToString(getOutUint(${memIdx}))`;
|
||||||
|
setArg();
|
||||||
|
++memIdx;
|
||||||
|
} else if (isZ3PointerType(outParam.type)) {
|
||||||
|
read = `getOutUint(${memIdx}) as unknown as ${outParam.type}`;
|
||||||
|
setArg();
|
||||||
|
++memIdx;
|
||||||
|
} else if (outParam.type === 'unsigned') {
|
||||||
|
read = `getOutUint(${memIdx})`;
|
||||||
|
setArg();
|
||||||
|
++memIdx;
|
||||||
|
} else if (outParam.type === 'int') {
|
||||||
|
read = `getOutInt(${memIdx})`;
|
||||||
|
setArg();
|
||||||
|
++memIdx;
|
||||||
|
} else if (outParam.type === 'uint64_t') {
|
||||||
|
if (memIdx % 2 === 1) {
|
||||||
|
++memIdx;
|
||||||
|
}
|
||||||
|
read = `getOutUint64(${memIdx / 2})`;
|
||||||
|
setArg();
|
||||||
|
memIdx += 2;
|
||||||
|
} else if (outParam.type === 'int64_t') {
|
||||||
|
if (memIdx % 2 === 1) {
|
||||||
|
++memIdx;
|
||||||
|
}
|
||||||
|
read = `getOutInt64(${memIdx / 2})`;
|
||||||
|
setArg();
|
||||||
|
memIdx += 2;
|
||||||
|
} else {
|
||||||
|
console.error(`skipping ${fn.name} - unknown out parameter type ${outParam.type}`);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
if (memIdx > Math.floor(BYTES_TO_ALLOCATE_FOR_OUT_PARAMS / 4)) {
|
||||||
|
// prettier-ignore
|
||||||
|
console.error(`skipping ${fn.name} - out parameter sizes sum to ${memIdx * 4}, which is > ${BYTES_TO_ALLOCATE_FOR_OUT_PARAMS}`);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
mapped.push({
|
||||||
|
name: outParam.name,
|
||||||
|
read,
|
||||||
|
type: outParam.type,
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
console.error(`skipping ${fn.name} - out param is neither pointer nor array`);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let ignoreReturn = fn.ret === 'boolean' || fn.ret === 'void';
|
||||||
|
if (outParams.length === 1) {
|
||||||
|
let outParam = mapped[0];
|
||||||
|
if (ignoreReturn) {
|
||||||
|
returnType = outParam.type;
|
||||||
|
rv = outParam.read;
|
||||||
|
} else {
|
||||||
|
returnType = `{ rv: ${fn.ret}, ${outParam.name} : ${outParam.type} }`;
|
||||||
|
rv = `{ rv: ret, ${outParam.name} : ${outParam.read} }`;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
if (ignoreReturn) {
|
||||||
|
returnType = `{ ${mapped.map(p => `${p.name} : ${p.type}`).join(', ')} }`;
|
||||||
|
rv = `{ ${mapped.map(p => `${p.name}: ${p.read}`).join(', ')} }`;
|
||||||
|
} else {
|
||||||
|
returnType = `{ rv: ${fn.ret}, ${mapped.map(p => `${p.name} : ${p.type}`).join(', ')} }`;
|
||||||
|
rv = `{ rv: ret, ${mapped.map(p => `${p.name}: ${p.read}`).join(', ')} }`;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (fn.ret === 'boolean') {
|
||||||
|
// assume the boolean indicates success
|
||||||
|
infix += `
|
||||||
|
if (!ret) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
`.trim();
|
||||||
|
cReturnType = 'boolean';
|
||||||
|
returnType += ' | null';
|
||||||
|
} else if (fn.ret === 'void') {
|
||||||
|
cReturnType = 'void';
|
||||||
|
} else if (isZ3PointerType(fn.ret) || fn.ret === 'unsigned') {
|
||||||
|
cReturnType = 'number';
|
||||||
|
} else {
|
||||||
|
console.error(`skipping ${fn.name} - out parameter for function which returns non-boolean`);
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (fn.nullableRet) {
|
||||||
|
returnType += ' | null';
|
||||||
|
infix += `
|
||||||
|
if (ret === 0) {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
`.trim();
|
||||||
|
}
|
||||||
|
|
||||||
|
// prettier-ignore
|
||||||
|
let invocation = `Mod.ccall('${isAsync ? 'async_' : ''}${fn.name}', '${cReturnType}', ${JSON.stringify(ctypes)}, [${args.map(toEm).join(', ')}])`;
|
||||||
|
|
||||||
|
if (isAsync) {
|
||||||
|
invocation = `await Mod.async_call(() => ${invocation})`;
|
||||||
|
returnType = `Promise<${returnType}>`;
|
||||||
|
}
|
||||||
|
|
||||||
|
let out = `${name}: ${isAsync ? 'async' : ''} function(${params.filter(p => p != null).join(', ')}): ${returnType} {
|
||||||
|
${prefix}`;
|
||||||
|
if (infix === '' && suffix === '' && rv === 'ret') {
|
||||||
|
out += `return ${invocation};`;
|
||||||
|
} else {
|
||||||
|
out += `
|
||||||
|
let ret = ${invocation};
|
||||||
|
${infix}return ${rv};${suffix}
|
||||||
|
`.trim();
|
||||||
|
}
|
||||||
|
out += '}';
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
|
function wrapEnum(name: string, values: Record<string, number>) {
|
||||||
|
let enumEntries = Object.entries(values);
|
||||||
|
return `export enum ${name} {
|
||||||
|
${enumEntries.map(([k, v], i) => k + (v === (enumEntries[i - 1]?.[1] ?? -1) + 1 ? '' : ` = ${v}`) + ',').join('\n')}
|
||||||
|
};`;
|
||||||
|
}
|
||||||
|
|
||||||
|
function getValidOutArrayIndexes(size: number) {
|
||||||
|
return Array.from({ length: Math.floor(BYTES_TO_ALLOCATE_FOR_OUT_PARAMS / size) }, (_, i) => i).join(' | ');
|
||||||
|
}
|
||||||
|
|
||||||
|
const typesDocument = `// THIS FILE IS AUTOMATICALLY GENERATED BY ${path.basename(__filename)}
|
||||||
|
// DO NOT EDIT IT BY HAND
|
||||||
|
|
||||||
|
interface Pointer<T extends string> extends Number {
|
||||||
|
readonly __typeName: T;
|
||||||
|
}
|
||||||
|
interface Subpointer<T extends string, S extends string> extends Pointer<S> {
|
||||||
|
readonly __typeName2: T;
|
||||||
|
}
|
||||||
|
|
||||||
|
${Object.keys(types)
|
||||||
|
.filter(k => k.startsWith('Z3'))
|
||||||
|
.map(makePointerType)
|
||||||
|
.join('\n')}
|
||||||
|
|
||||||
|
${Object.entries(enums)
|
||||||
|
.map(e => wrapEnum(e[0], e[1]))
|
||||||
|
.join('\n\n')}
|
||||||
|
`;
|
||||||
|
|
||||||
|
const relativePath: string = path.relative(path.dirname(wrapperFilePath), path.dirname(typesFilePath)) || './';
|
||||||
|
const ext: string = path.extname(typesFilePath);
|
||||||
|
const basename: string = path.basename(typesFilePath);
|
||||||
|
const importPath = relativePath + basename.slice(0, -ext.length);
|
||||||
|
|
||||||
|
const wrapperDocument = `// THIS FILE IS AUTOMATICALLY GENERATED BY ${path.basename(__filename)}
|
||||||
|
// DO NOT EDIT IT BY HAND
|
||||||
|
|
||||||
|
import {
|
||||||
|
${Object.keys(types)
|
||||||
|
.filter(k => k.startsWith('Z3'))
|
||||||
|
.join(',\n')},
|
||||||
|
${Object.keys(enums).join(',\n')},
|
||||||
|
} from '${importPath}';
|
||||||
|
|
||||||
|
${Object.entries(primitiveTypes)
|
||||||
|
.filter(e => e[0] !== 'void')
|
||||||
|
.map(e => `type ${e[0]} = ${e[1]};`)
|
||||||
|
.join('\n')}
|
||||||
|
|
||||||
|
export async function init(initModule: any) {
|
||||||
|
let Mod = await initModule();
|
||||||
|
|
||||||
|
// this works for both signed and unsigned, because JS will wrap for you when constructing the Uint32Array
|
||||||
|
function intArrayToByteArr(ints: number[]) {
|
||||||
|
return new Uint8Array((new Uint32Array(ints)).buffer);
|
||||||
|
}
|
||||||
|
|
||||||
|
function boolArrayToByteArr(bools: boolean[]) {
|
||||||
|
return bools.map(b => b ? 1 : 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
function readUintArray(address: number, count: number) {
|
||||||
|
return Array.from(new Uint32Array(Mod.HEAPU32.buffer, address, count));
|
||||||
|
}
|
||||||
|
|
||||||
|
let outAddress = Mod._malloc(${BYTES_TO_ALLOCATE_FOR_OUT_PARAMS});
|
||||||
|
let outUintArray = (new Uint32Array(Mod.HEAPU32.buffer, outAddress, 4));
|
||||||
|
let getOutUint = (i: ${getValidOutArrayIndexes(4)}) => outUintArray[i];
|
||||||
|
let outIntArray = (new Int32Array(Mod.HEAPU32.buffer, outAddress, 4));
|
||||||
|
let getOutInt = (i: ${getValidOutArrayIndexes(4)}) => outIntArray[i];
|
||||||
|
let outUint64Array = (new BigUint64Array(Mod.HEAPU32.buffer, outAddress, 2));
|
||||||
|
let getOutUint64 = (i: ${getValidOutArrayIndexes(8)}) => outUint64Array[i];
|
||||||
|
let outInt64Array = (new BigInt64Array(Mod.HEAPU32.buffer, outAddress, 2));
|
||||||
|
let getOutInt64 = (i: ${getValidOutArrayIndexes(8)}) => outInt64Array[i];
|
||||||
|
|
||||||
|
return {
|
||||||
|
em: Mod,
|
||||||
|
Z3: {
|
||||||
|
mk_context: function(c: Z3_config): Z3_context {
|
||||||
|
let ctx = Mod._Z3_mk_context(c);
|
||||||
|
Mod._set_noop_error_handler(ctx);
|
||||||
|
return ctx;
|
||||||
|
},
|
||||||
|
mk_context_rc: function(c: Z3_config): Z3_context {
|
||||||
|
let ctx = Mod._Z3_mk_context_rc(c);
|
||||||
|
Mod._set_noop_error_handler(ctx);
|
||||||
|
return ctx;
|
||||||
|
},
|
||||||
|
${functions
|
||||||
|
.map(wrapFunction)
|
||||||
|
.filter(f => f != null)
|
||||||
|
.join(',\n')}
|
||||||
|
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
`;
|
||||||
|
|
||||||
|
return {
|
||||||
|
wrapperDocument: prettier.format(wrapperDocument, { singleQuote: true, parser: 'typescript' }),
|
||||||
|
typesDocument: prettier.format(typesDocument, { singleQuote: true, parser: 'typescript' }),
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
const { wrapperDocument, typesDocument } = makeTsWrapper();
|
||||||
|
fs.mkdirSync(path.dirname(wrapperFilePath), { recursive: true });
|
||||||
|
fs.writeFileSync(wrapperFilePath, wrapperDocument);
|
||||||
|
fs.mkdirSync(path.dirname(typesFilePath), { recursive: true });
|
||||||
|
fs.writeFileSync(typesFilePath, typesDocument);
|
||||||
|
|
@ -1,9 +1,8 @@
|
||||||
'use strict';
|
import assert from 'assert';
|
||||||
|
import fs from 'fs';
|
||||||
|
import path from 'path';
|
||||||
|
|
||||||
let fs = require('fs');
|
const files = [
|
||||||
let path = require('path');
|
|
||||||
|
|
||||||
let files = [
|
|
||||||
'z3_api.h',
|
'z3_api.h',
|
||||||
'z3_algebraic.h',
|
'z3_algebraic.h',
|
||||||
'z3_ast_containers.h',
|
'z3_ast_containers.h',
|
||||||
|
|
@ -15,15 +14,15 @@ let files = [
|
||||||
'z3_spacer.h',
|
'z3_spacer.h',
|
||||||
];
|
];
|
||||||
|
|
||||||
let aliases = {
|
const aliases = {
|
||||||
__proto__: null,
|
__proto__: null,
|
||||||
Z3_bool: 'boolean',
|
Z3_bool: 'boolean',
|
||||||
Z3_string: 'string',
|
Z3_string: 'string',
|
||||||
bool: 'boolean',
|
bool: 'boolean',
|
||||||
signed: 'int',
|
signed: 'int',
|
||||||
};
|
} as unknown as Record<string, string>;
|
||||||
|
|
||||||
let primitiveTypes = {
|
const primitiveTypes = {
|
||||||
__proto__: null,
|
__proto__: null,
|
||||||
Z3_char_ptr: 'string',
|
Z3_char_ptr: 'string',
|
||||||
unsigned: 'number',
|
unsigned: 'number',
|
||||||
|
|
@ -32,35 +31,49 @@ let primitiveTypes = {
|
||||||
int64_t: 'bigint',
|
int64_t: 'bigint',
|
||||||
double: 'number',
|
double: 'number',
|
||||||
float: 'number',
|
float: 'number',
|
||||||
};
|
} as unknown as Record<string, string>;
|
||||||
|
|
||||||
let optTypes = {
|
const optTypes = {
|
||||||
__proto__: null,
|
__proto__: null,
|
||||||
|
|
||||||
Z3_sort_opt: 'Z3_sort',
|
Z3_sort_opt: 'Z3_sort',
|
||||||
Z3_ast_opt: 'Z3_ast',
|
Z3_ast_opt: 'Z3_ast',
|
||||||
Z3_func_interp_opt: 'Z3_func_interp',
|
Z3_func_interp_opt: 'Z3_func_interp',
|
||||||
};
|
} as unknown as Record<string, string>;
|
||||||
|
|
||||||
// parse type declarations
|
// parse type declarations
|
||||||
let types = {
|
const types = {
|
||||||
__proto__: null,
|
__proto__: null,
|
||||||
|
|
||||||
// these are function types I can't be bothered to parse
|
// these are function types I can't be bothered to parse
|
||||||
Z3_error_handler: 'Z3_error_handler',
|
Z3_error_handler: 'Z3_error_handler',
|
||||||
Z3_push_eh: 'Z3_push_eh',
|
Z3_push_eh: 'Z3_push_eh',
|
||||||
Z3_pop_eh: 'Z3_pop_eh',
|
Z3_pop_eh: 'Z3_pop_eh',
|
||||||
Z3_fresh_eh: 'Z3_fresh_eh',
|
Z3_fresh_eh: 'Z3_fresh_eh',
|
||||||
Z3_fixed_eh: 'Z3_fixed_eh',
|
Z3_fixed_eh: 'Z3_fixed_eh',
|
||||||
Z3_eq_eh: 'Z3_eq_eh',
|
Z3_eq_eh: 'Z3_eq_eh',
|
||||||
Z3_final_eh: 'Z3_final_eh',
|
Z3_final_eh: 'Z3_final_eh',
|
||||||
Z3_created_eh: 'Z3_created_eh',
|
Z3_created_eh: 'Z3_created_eh',
|
||||||
Z3_decide_eh: 'Z3_decide_eh'
|
Z3_decide_eh: 'Z3_decide_eh',
|
||||||
|
} as unknown as Record<string, string>;
|
||||||
|
|
||||||
|
export type ApiParam = { kind: string; sizeIndex?: number; type: string };
|
||||||
|
export type Api = { params: ApiParam[]; ret: string; extra: boolean };
|
||||||
|
const defApis: Record<string, Api> = Object.create(null);
|
||||||
|
export type FuncParam = {
|
||||||
|
type: string;
|
||||||
|
cType: string;
|
||||||
|
name: string;
|
||||||
|
isConst: boolean;
|
||||||
|
isPtr: boolean;
|
||||||
|
isArray: boolean;
|
||||||
|
nullable: boolean;
|
||||||
|
kind?: string;
|
||||||
|
sizeIndex?: number;
|
||||||
};
|
};
|
||||||
|
export type Func = { ret: string; cRet: string; name: string; params: FuncParam[]; nullableRet: boolean };
|
||||||
let defApis = Object.create(null);
|
const functions: Func[] = [];
|
||||||
let functions = [];
|
let enums: Record<string, Record<string, number>> = Object.create(null);
|
||||||
let enums = Object.create(null);
|
|
||||||
for (let file of files) {
|
for (let file of files) {
|
||||||
let contents = fs.readFileSync(path.join(__dirname, '..', '..', file), 'utf8');
|
let contents = fs.readFileSync(path.join(__dirname, '..', '..', file), 'utf8');
|
||||||
|
|
||||||
|
|
@ -80,6 +93,7 @@ for (let file of files) {
|
||||||
/def_Type\(\s*'(?<name>[A-Za-z0-9_]+)',\s*'(?<cname>[A-Za-z0-9_]+)',\s*'(?<pname>[A-Za-z0-9_]+)'\)/g,
|
/def_Type\(\s*'(?<name>[A-Za-z0-9_]+)',\s*'(?<cname>[A-Za-z0-9_]+)',\s*'(?<pname>[A-Za-z0-9_]+)'\)/g,
|
||||||
);
|
);
|
||||||
for (let { groups } of typeMatches) {
|
for (let { groups } of typeMatches) {
|
||||||
|
assert(groups !== undefined);
|
||||||
pytypes[groups.name] = groups.cname;
|
pytypes[groups.name] = groups.cname;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -93,11 +107,12 @@ for (let file of files) {
|
||||||
let apiLines = contents.split('\n').filter(l => /def_API|extra_API/.test(l));
|
let apiLines = contents.split('\n').filter(l => /def_API|extra_API/.test(l));
|
||||||
for (let line of apiLines) {
|
for (let line of apiLines) {
|
||||||
let match = line.match(
|
let match = line.match(
|
||||||
/^\s*(?<def>def_API|extra_API) *\(\s*'(?<name>[A-Za-z0-9_]+)'\s*,\s*(?<ret>[A-Za-z0-9_]+)\s*,\s*\((?<params>((_in|_out|_in_array|_out_array|_fnptr|_inout_array)\([^)]+\)\s*,?\s*)*)\)\s*\)\s*$/,
|
/^\s*(?<def>def_API|extra_API) *\(\s*'(?<name>[A-Za-z0-9_]+)'\s*,\s*(?<ret>[A-Za-z0-9_]+)\s*,\s*\((?<params>((_in|_out|_in_array|_out_array|_fnptr|_inout_array)\([^)]+\)\s*,?\s*)*)\)\s*\)\s*$/,
|
||||||
);
|
);
|
||||||
if (match == null) {
|
if (match === null) {
|
||||||
throw new Error(`failed to match def_API call ${JSON.stringify(line)}`);
|
throw new Error(`failed to match def_API call ${JSON.stringify(line)}`);
|
||||||
}
|
}
|
||||||
|
assert(match.groups !== undefined);
|
||||||
let { name, ret, def } = match.groups;
|
let { name, ret, def } = match.groups;
|
||||||
let params = match.groups.params.trim();
|
let params = match.groups.params.trim();
|
||||||
let text = params;
|
let text = params;
|
||||||
|
|
@ -108,6 +123,7 @@ for (let file of files) {
|
||||||
if (match == null) {
|
if (match == null) {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
assert(match.groups !== undefined);
|
||||||
let kind = match.groups.kind;
|
let kind = match.groups.kind;
|
||||||
if (kind === 'inout_array') kind = 'in_array'; // https://github.com/Z3Prover/z3/discussions/5761
|
if (kind === 'inout_array') kind = 'in_array'; // https://github.com/Z3Prover/z3/discussions/5761
|
||||||
if (kind === 'in' || kind === 'out' || kind == 'fnptr') {
|
if (kind === 'in' || kind === 'out' || kind == 'fnptr') {
|
||||||
|
|
@ -135,10 +151,10 @@ for (let file of files) {
|
||||||
}
|
}
|
||||||
|
|
||||||
for (let match of contents.matchAll(/DEFINE_TYPE\((?<type>[A-Za-z0-9_]+)\)/g)) {
|
for (let match of contents.matchAll(/DEFINE_TYPE\((?<type>[A-Za-z0-9_]+)\)/g)) {
|
||||||
|
assert(match.groups !== undefined);
|
||||||
types[match.groups.type] = match.groups.type;
|
types[match.groups.type] = match.groups.type;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// parse enum declarations
|
// parse enum declarations
|
||||||
for (let idx = 0; idx < contents.length; ) {
|
for (let idx = 0; idx < contents.length; ) {
|
||||||
let nextIdx = contents.indexOf('typedef enum', idx);
|
let nextIdx = contents.indexOf('typedef enum', idx);
|
||||||
|
|
@ -156,12 +172,13 @@ for (let file of files) {
|
||||||
if (match === null) {
|
if (match === null) {
|
||||||
throw new Error(`could not parse enum ${JSON.stringify(slice)}`);
|
throw new Error(`could not parse enum ${JSON.stringify(slice)}`);
|
||||||
}
|
}
|
||||||
let vals = Object.create(null);
|
let vals: Record<string, number> = Object.create(null);
|
||||||
let next = 0;
|
let next = 0;
|
||||||
while (true) {
|
while (true) {
|
||||||
let blank = true;
|
let blank = true;
|
||||||
while (blank) {
|
while (blank) {
|
||||||
({ match, text } = eat(text, /^\s*(\/\/[^\n]*\n)?/));
|
({ match, text } = eat(text, /^\s*(\/\/[^\n]*\n)?/));
|
||||||
|
assert(match !== null);
|
||||||
blank = match[0].length > 0;
|
blank = match[0].length > 0;
|
||||||
}
|
}
|
||||||
({ match, text } = eat(text, /^[A-Za-z0-9_]+/));
|
({ match, text } = eat(text, /^[A-Za-z0-9_]+/));
|
||||||
|
|
@ -173,6 +190,7 @@ for (let file of files) {
|
||||||
|
|
||||||
({ match, text } = eat(text, /^= *(?<val>[^\n,\s]+)/));
|
({ match, text } = eat(text, /^= *(?<val>[^\n,\s]+)/));
|
||||||
if (match !== null) {
|
if (match !== null) {
|
||||||
|
assert(match.groups !== undefined);
|
||||||
let parsedVal = Number(match.groups.val);
|
let parsedVal = Number(match.groups.val);
|
||||||
if (Object.is(parsedVal, NaN)) {
|
if (Object.is(parsedVal, NaN)) {
|
||||||
throw new Error('unknown value ' + match.groups.val);
|
throw new Error('unknown value ' + match.groups.val);
|
||||||
|
|
@ -222,12 +240,14 @@ for (let file of files) {
|
||||||
if (match == null) {
|
if (match == null) {
|
||||||
throw new Error(`failed to match c definition: ${JSON.stringify(slice)}`);
|
throw new Error(`failed to match c definition: ${JSON.stringify(slice)}`);
|
||||||
}
|
}
|
||||||
|
assert(match.groups !== undefined);
|
||||||
|
|
||||||
let { ret, name, params } = match.groups;
|
let { ret, name, params } = match.groups;
|
||||||
let parsedParams = [];
|
let parsedParams = [];
|
||||||
|
|
||||||
if (params.trim() !== 'void') {
|
if (params.trim() !== 'void') {
|
||||||
for (let param of params.split(',')) {
|
for (let param of params.split(',')) {
|
||||||
let paramType, paramName, isConst, isPtr, isArray;
|
let paramType: string, paramName: string, isConst: boolean, isPtr: boolean, isArray: boolean;
|
||||||
|
|
||||||
let { match, text } = eat(param, /^\s*/);
|
let { match, text } = eat(param, /^\s*/);
|
||||||
({ match, text } = eat(text, /^[A-Za-z0-9_]+/));
|
({ match, text } = eat(text, /^[A-Za-z0-9_]+/));
|
||||||
|
|
@ -303,7 +323,7 @@ for (let file of files) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
function isKnownType(t) {
|
function isKnownType(t: string) {
|
||||||
return t in enums || t in types || t in primitiveTypes || ['string', 'boolean', 'void'].includes(t);
|
return t in enums || t in types || t in primitiveTypes || ['string', 'boolean', 'void'].includes(t);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -340,19 +360,19 @@ for (let fn of functions) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
function eat(str, regex) {
|
function eat(str: string, regex: string | RegExp) {
|
||||||
const match = str.match(regex);
|
const match: RegExpMatchArray | null = str.match(regex);
|
||||||
if (match == null) {
|
if (match === null) {
|
||||||
return { match, text: str };
|
return { match, text: str };
|
||||||
}
|
}
|
||||||
return { match, text: str.substring(match[0].length) };
|
return { match, text: str.substring(match[0].length) };
|
||||||
}
|
}
|
||||||
|
|
||||||
function eatWs(text) {
|
function eatWs(text: string) {
|
||||||
return eat(text, /^\s*/).text;
|
return eat(text, /^\s*/).text;
|
||||||
}
|
}
|
||||||
|
|
||||||
function expect(str, regex) {
|
function expect(str: string, regex: string | RegExp) {
|
||||||
let { text, match } = eat(str, regex);
|
let { text, match } = eat(str, regex);
|
||||||
if (match === null) {
|
if (match === null) {
|
||||||
throw new Error(`expected ${regex}, got ${JSON.stringify(text)}`);
|
throw new Error(`expected ${regex}, got ${JSON.stringify(text)}`);
|
||||||
|
|
@ -360,4 +380,4 @@ function expect(str, regex) {
|
||||||
return { text, match };
|
return { text, match };
|
||||||
}
|
}
|
||||||
|
|
||||||
module.exports = { primitiveTypes, types, enums, functions };
|
export { primitiveTypes, types, enums, functions };
|
||||||
16
src/api/js/src/browser.ts
Normal file
16
src/api/js/src/browser.ts
Normal file
|
|
@ -0,0 +1,16 @@
|
||||||
|
import { createApi, Z3HighLevel } from './high-level';
|
||||||
|
import { init as initWrapper, Z3LowLevel } from './low-level';
|
||||||
|
export * from './high-level/types';
|
||||||
|
export { Z3Core, Z3LowLevel } from './low-level';
|
||||||
|
export * from './low-level/types.__GENERATED__';
|
||||||
|
|
||||||
|
export async function init(): Promise<Z3LowLevel & Z3HighLevel> {
|
||||||
|
const initZ3 = (global as any).initZ3;
|
||||||
|
if (initZ3 === undefined) {
|
||||||
|
throw new Error('initZ3 was not imported correctly. Please consult documentation on how to load Z3 in browser');
|
||||||
|
}
|
||||||
|
|
||||||
|
const lowLevel = await initWrapper(initZ3);
|
||||||
|
const highLevel = createApi(lowLevel.Z3);
|
||||||
|
return { ...lowLevel, ...highLevel };
|
||||||
|
}
|
||||||
450
src/api/js/src/high-level/high-level.test.ts
Normal file
450
src/api/js/src/high-level/high-level.test.ts
Normal file
|
|
@ -0,0 +1,450 @@
|
||||||
|
import assert from 'assert';
|
||||||
|
import asyncToArray from 'iter-tools/methods/async-to-array';
|
||||||
|
import { init, killThreads } from '../jest';
|
||||||
|
import { Arith, Bool, Model, sat, unsat, Z3AssertionError, Z3HighLevel } from './types';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Generate all possible solutions from given assumptions.
|
||||||
|
*
|
||||||
|
* **NOTE**: The set of solutions might be infinite.
|
||||||
|
* Always ensure to limit amount generated, either by knowing that the
|
||||||
|
* solution space is constrainted, or by taking only a specified
|
||||||
|
* amount of solutions
|
||||||
|
* ```typescript
|
||||||
|
* import { sliceAsync } from 'iter-tools';
|
||||||
|
* // ...
|
||||||
|
* for await (const model of sliceAsync(10, solver.solutions())) {
|
||||||
|
* console.log(model.sexpr());
|
||||||
|
* }
|
||||||
|
* ```
|
||||||
|
* @see http://theory.stanford.edu/~nikolaj/programmingz3.html#sec-blocking-evaluations
|
||||||
|
* @returns Models with solutions. Nothing if no constants provided
|
||||||
|
*/
|
||||||
|
// TODO(ritave): Use faster solution https://stackoverflow.com/a/70656700
|
||||||
|
// TODO(ritave): Move to high-level.ts
|
||||||
|
async function* allSolutions<Name extends string>(...assertions: Bool<Name>[]): AsyncIterable<Model<Name>> {
|
||||||
|
if (assertions.length === 0) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const { Or } = assertions[0].ctx;
|
||||||
|
const solver = new assertions[0].ctx.Solver();
|
||||||
|
solver.add(...assertions);
|
||||||
|
|
||||||
|
while ((await solver.check()) === sat) {
|
||||||
|
const model = solver.model();
|
||||||
|
const decls = model.decls();
|
||||||
|
if (decls.length === 0) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
yield model;
|
||||||
|
|
||||||
|
solver.add(
|
||||||
|
Or(
|
||||||
|
...decls
|
||||||
|
// TODO(ritave): Assert on arity > 0
|
||||||
|
.filter(decl => decl.arity() === 0)
|
||||||
|
.map(decl => {
|
||||||
|
const term = decl.call();
|
||||||
|
// TODO(ritave): Assert not an array / uinterpeted sort
|
||||||
|
const value = model.eval(term, true);
|
||||||
|
return term.neq(value);
|
||||||
|
}),
|
||||||
|
),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async function prove(conjecture: Bool): Promise<void> {
|
||||||
|
const solver = new conjecture.ctx.Solver();
|
||||||
|
const { Not } = solver.ctx;
|
||||||
|
solver.add(Not(conjecture));
|
||||||
|
expect(await solver.check()).toStrictEqual(unsat);
|
||||||
|
}
|
||||||
|
|
||||||
|
async function solve(conjecture: Bool): Promise<Model> {
|
||||||
|
const solver = new conjecture.ctx.Solver();
|
||||||
|
solver.add(conjecture);
|
||||||
|
expect(await solver.check()).toStrictEqual(sat);
|
||||||
|
return solver.model();
|
||||||
|
}
|
||||||
|
|
||||||
|
describe('high-level', () => {
|
||||||
|
let api: { em: any } & Z3HighLevel;
|
||||||
|
|
||||||
|
beforeAll(async () => {
|
||||||
|
api = await init();
|
||||||
|
});
|
||||||
|
|
||||||
|
afterAll(async () => {
|
||||||
|
await killThreads(api.em);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('can set params', () => {
|
||||||
|
const { setParam, getParam, resetParams } = api;
|
||||||
|
|
||||||
|
expect(getParam('pp.decimal')).toStrictEqual('false');
|
||||||
|
setParam('pp.decimal', 'true');
|
||||||
|
expect(getParam('pp.decimal')).toStrictEqual('true');
|
||||||
|
setParam({ 'pp.decimal': 'false', timeout: 4 });
|
||||||
|
expect(getParam('pp.decimal')).toStrictEqual('false');
|
||||||
|
expect(getParam('timeout')).toStrictEqual('4');
|
||||||
|
|
||||||
|
resetParams();
|
||||||
|
expect(getParam('pp.decimal')).toStrictEqual('false');
|
||||||
|
expect(getParam('timeout')).toStrictEqual('4294967295');
|
||||||
|
});
|
||||||
|
|
||||||
|
it('proves x = y implies g(x) = g(y)', async () => {
|
||||||
|
const { Solver, Int, Function, Implies, Not } = new api.Context('main');
|
||||||
|
const solver = new Solver();
|
||||||
|
|
||||||
|
const sort = Int.sort();
|
||||||
|
const x = Int.const('x');
|
||||||
|
const y = Int.const('y');
|
||||||
|
const g = Function.declare('g', sort, sort);
|
||||||
|
|
||||||
|
const conjecture = Implies(x.eq(y), g.call(x).eq(g.call(y)));
|
||||||
|
solver.add(Not(conjecture));
|
||||||
|
expect(await solver.check()).toStrictEqual(unsat);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('disproves x = y implies g(g(x)) = g(y)', async () => {
|
||||||
|
const { Solver, Int, Function, Implies, Not } = new api.Context('main');
|
||||||
|
const solver = new Solver();
|
||||||
|
|
||||||
|
const sort = Int.sort();
|
||||||
|
const x = Int.const('x');
|
||||||
|
const y = Int.const('y');
|
||||||
|
const g = Function.declare('g', sort, sort);
|
||||||
|
const conjecture = Implies(x.eq(y), g.call(g.call(x)).eq(g.call(y)));
|
||||||
|
solver.add(Not(conjecture));
|
||||||
|
expect(await solver.check()).toStrictEqual(sat);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('checks that Context matches', () => {
|
||||||
|
const c1 = new api.Context('context');
|
||||||
|
const c2 = new api.Context('context');
|
||||||
|
const c3 = new api.Context('foo');
|
||||||
|
const c4 = new api.Context('bar');
|
||||||
|
|
||||||
|
// Contexts with the same name don't do type checking during compile time.
|
||||||
|
// We need to check for different context dynamically
|
||||||
|
expect(() => c1.Or(c2.Int.val(5).eq(2))).toThrowError(Z3AssertionError);
|
||||||
|
|
||||||
|
// On the other hand, this won't compile due to automatic generics
|
||||||
|
// @ts-expect-error
|
||||||
|
expect(() => c3.Or(c4.Int.val(5).eq(2))).toThrowError(Z3AssertionError);
|
||||||
|
|
||||||
|
const allUniqueContexes = new Set([c1, c2, c3, c4]).size === 4;
|
||||||
|
expect(allUniqueContexes).toStrictEqual(true);
|
||||||
|
|
||||||
|
expect(() => c1.Or(c1.Int.val(5).eq(2))).not.toThrowError();
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('booleans', () => {
|
||||||
|
it("proves De Morgan's Law", async () => {
|
||||||
|
const { Bool, Not, And, Eq, Or } = new api.Context('main');
|
||||||
|
const [x, y] = [Bool.const('x'), Bool.const('y')];
|
||||||
|
|
||||||
|
const conjecture = Eq(Not(And(x, y)), Or(Not(x), Not(y)));
|
||||||
|
|
||||||
|
await prove(conjecture);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('ints', () => {
|
||||||
|
it('finds a model', async () => {
|
||||||
|
const { Solver, Int, isIntVal } = new api.Context('main');
|
||||||
|
const solver = new Solver();
|
||||||
|
const x = Int.const('x');
|
||||||
|
const y = Int.const('y');
|
||||||
|
|
||||||
|
solver.add(x.ge(1)); // x >= 1
|
||||||
|
solver.add(y.lt(x.add(3))); // y < x + 3
|
||||||
|
|
||||||
|
expect(await solver.check()).toStrictEqual(sat);
|
||||||
|
|
||||||
|
const model = solver.model();
|
||||||
|
expect(model.length).toStrictEqual(2);
|
||||||
|
|
||||||
|
for (const decl of model) {
|
||||||
|
expect(decl.arity()).toStrictEqual(0);
|
||||||
|
}
|
||||||
|
const xValueExpr = model.get(x);
|
||||||
|
const yValueExpr = model.get(y);
|
||||||
|
assert(isIntVal(xValueExpr));
|
||||||
|
assert(isIntVal(yValueExpr));
|
||||||
|
const xValue = xValueExpr.value;
|
||||||
|
const yValue = yValueExpr.value;
|
||||||
|
assert(typeof xValue === 'bigint');
|
||||||
|
assert(typeof yValue === 'bigint');
|
||||||
|
expect(xValue).toBeGreaterThanOrEqual(1n);
|
||||||
|
expect(yValue).toBeLessThanOrEqual(xValue + 3n);
|
||||||
|
});
|
||||||
|
|
||||||
|
// TODO(ritave): After changes made since last commit (a332187c746c23450860deb210d94e6e042dd424),
|
||||||
|
// this test takes twice as long (from 5s to 10s). Figure out why
|
||||||
|
it('solves sudoku', async () => {
|
||||||
|
function toSudoku(data: string): (number | null)[][] {
|
||||||
|
const cells: (number | null)[][] = Array.from({ length: 9 }, () => Array.from({ length: 9 }, () => null));
|
||||||
|
|
||||||
|
const lines = data.trim().split('\n');
|
||||||
|
for (let row = 0; row < 9; row++) {
|
||||||
|
const line = lines[row].trim();
|
||||||
|
for (let col = 0; col < 9; col++) {
|
||||||
|
const char = line[col];
|
||||||
|
if (char !== '.') {
|
||||||
|
cells[row][col] = Number.parseInt(char);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return cells;
|
||||||
|
}
|
||||||
|
const INSTANCE = toSudoku(`
|
||||||
|
....94.3.
|
||||||
|
...51...7
|
||||||
|
.89....4.
|
||||||
|
......2.8
|
||||||
|
.6.2.1.5.
|
||||||
|
1.2......
|
||||||
|
.7....52.
|
||||||
|
9...65...
|
||||||
|
.4.97....
|
||||||
|
`);
|
||||||
|
|
||||||
|
const EXPECTED = toSudoku(`
|
||||||
|
715894632
|
||||||
|
234516897
|
||||||
|
689723145
|
||||||
|
493657218
|
||||||
|
867231954
|
||||||
|
152489763
|
||||||
|
376148529
|
||||||
|
928365471
|
||||||
|
541972386
|
||||||
|
`);
|
||||||
|
|
||||||
|
const { Solver, Int, Distinct, isIntVal } = new api.Context('main');
|
||||||
|
|
||||||
|
const cells: Arith[][] = [];
|
||||||
|
// 9x9 matrix of integer variables
|
||||||
|
for (let i = 0; i < 9; i++) {
|
||||||
|
const row = [];
|
||||||
|
for (let j = 0; j < 9; j++) {
|
||||||
|
row.push(Int.const(`x_${i}_${j}`));
|
||||||
|
}
|
||||||
|
cells.push(row);
|
||||||
|
}
|
||||||
|
|
||||||
|
const solver = new Solver();
|
||||||
|
|
||||||
|
// each cell contains a value 1<=x<=9
|
||||||
|
for (let i = 0; i < 9; i++) {
|
||||||
|
for (let j = 0; j < 9; j++) {
|
||||||
|
solver.add(cells[i][j].ge(1), cells[i][j].le(9));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// each row contains a digit only once
|
||||||
|
for (let i = 0; i < 9; i++) {
|
||||||
|
solver.add(Distinct(...cells[i]));
|
||||||
|
}
|
||||||
|
|
||||||
|
// each column contains a digit only once
|
||||||
|
for (let j = 0; j < 9; j++) {
|
||||||
|
const column = [];
|
||||||
|
for (let i = 0; i < 9; i++) {
|
||||||
|
column.push(cells[i][j]);
|
||||||
|
}
|
||||||
|
solver.add(Distinct(...column));
|
||||||
|
}
|
||||||
|
|
||||||
|
// each 3x3 contains a digit at most once
|
||||||
|
for (let iSquare = 0; iSquare < 3; iSquare++) {
|
||||||
|
for (let jSquare = 0; jSquare < 3; jSquare++) {
|
||||||
|
const square = [];
|
||||||
|
|
||||||
|
for (let i = iSquare * 3; i < iSquare * 3 + 3; i++) {
|
||||||
|
for (let j = jSquare * 3; j < jSquare * 3 + 3; j++) {
|
||||||
|
square.push(cells[i][j]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
solver.add(Distinct(...square));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (let i = 0; i < 9; i++) {
|
||||||
|
for (let j = 0; j < 9; j++) {
|
||||||
|
const digit = INSTANCE[i][j];
|
||||||
|
if (digit !== null) {
|
||||||
|
solver.add(cells[i][j].eq(digit));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
expect(await solver.check()).toStrictEqual(sat);
|
||||||
|
|
||||||
|
const model = solver.model();
|
||||||
|
const result = [];
|
||||||
|
for (let i = 0; i < 9; i++) {
|
||||||
|
let row = [];
|
||||||
|
for (let j = 0; j < 9; j++) {
|
||||||
|
const cell = model.eval(cells[i][j]);
|
||||||
|
assert(isIntVal(cell));
|
||||||
|
const value = cell.value;
|
||||||
|
assert(typeof value === 'bigint');
|
||||||
|
expect(value).toBeGreaterThanOrEqual(0n);
|
||||||
|
expect(value).toBeLessThanOrEqual(9n);
|
||||||
|
// JSON.stringify doesn't handle bigints
|
||||||
|
row.push(Number(value));
|
||||||
|
}
|
||||||
|
result.push(row);
|
||||||
|
}
|
||||||
|
expect(JSON.stringify(result)).toStrictEqual(JSON.stringify(EXPECTED));
|
||||||
|
}, 120_000);
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('reals', () => {
|
||||||
|
it('can work with numerals', async () => {
|
||||||
|
const { Real, And } = new api.Context('main');
|
||||||
|
const n1 = Real.val('1/2');
|
||||||
|
const n2 = Real.val('0.5');
|
||||||
|
const n3 = Real.val(0.5);
|
||||||
|
await prove(And(n1.eq(n2), n1.eq(n3)));
|
||||||
|
|
||||||
|
const n4 = Real.val('-1/3');
|
||||||
|
const n5 = Real.val('-0.3333333333333333333333333333333333');
|
||||||
|
await prove(n4.neq(n5));
|
||||||
|
});
|
||||||
|
|
||||||
|
it('can do non-linear arithmetic', async () => {
|
||||||
|
api.setParam('pp.decimal', true);
|
||||||
|
api.setParam('pp.decimal_precision', 20);
|
||||||
|
const { Real, Solver, isReal, isRealVal } = new api.Context('main');
|
||||||
|
const x = Real.const('x');
|
||||||
|
const y = Real.const('y');
|
||||||
|
const z = Real.const('z');
|
||||||
|
|
||||||
|
const solver = new Solver();
|
||||||
|
solver.add(x.mul(x).add(y.mul(y)).eq(1)); // x^2 + y^2 == 1
|
||||||
|
solver.add(x.mul(x).mul(x).add(z.mul(z).mul(z)).lt('1/2')); // x^3 + z^3 < 1/2
|
||||||
|
|
||||||
|
expect(await solver.check()).toStrictEqual(sat);
|
||||||
|
const model = solver.model();
|
||||||
|
|
||||||
|
expect(isRealVal(model.get(x))).toStrictEqual(true);
|
||||||
|
// solution of y is a polynomial
|
||||||
|
// https://stackoverflow.com/a/69740906
|
||||||
|
expect(isReal(model.get(y))).toStrictEqual(true);
|
||||||
|
expect(isRealVal(model.get(z))).toStrictEqual(true);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('bitvectors', () => {
|
||||||
|
it('can do simple proofs', async () => {
|
||||||
|
const { BitVec, Concat, Implies, isBitVecVal } = new api.Context('main');
|
||||||
|
|
||||||
|
const x = BitVec.const('x', 32);
|
||||||
|
|
||||||
|
const sSol = (await solve(x.sub(10).sle(0).eq(x.sle(10)))).get(x); // signed: (x - 10 <= 0) == (x <= 10)
|
||||||
|
const uSol = (await solve(x.sub(10).ule(0).eq(x.ule(10)))).get(x); // unsigned: (x - 10 <= 0) == (x <= 10)
|
||||||
|
|
||||||
|
assert(isBitVecVal(sSol) && isBitVecVal(uSol));
|
||||||
|
let v = sSol.asSignedValue();
|
||||||
|
expect(v - 10n <= 0n === v <= 10n).toStrictEqual(true);
|
||||||
|
v = uSol.value;
|
||||||
|
expect(v - 10n <= 0n === v <= 10n).toStrictEqual(true);
|
||||||
|
|
||||||
|
const y = BitVec.const('y', 32);
|
||||||
|
|
||||||
|
await prove(Implies(Concat(x, y).eq(Concat(y, x)), x.eq(y)));
|
||||||
|
});
|
||||||
|
|
||||||
|
it('finds x and y such that: x ^ y - 103 == x * y', async () => {
|
||||||
|
const { BitVec, isBitVecVal } = new api.Context('main');
|
||||||
|
|
||||||
|
const x = BitVec.const('x', 32);
|
||||||
|
const y = BitVec.const('y', 32);
|
||||||
|
|
||||||
|
const model = await solve(x.xor(y).sub(103).eq(x.mul(y)));
|
||||||
|
const xSol = model.get(x);
|
||||||
|
const ySol = model.get(y);
|
||||||
|
assert(isBitVecVal(xSol) && isBitVecVal(ySol));
|
||||||
|
const xv = xSol.asSignedValue();
|
||||||
|
const yv = ySol.asSignedValue();
|
||||||
|
|
||||||
|
// this solutions wraps around so we need to check using modulo
|
||||||
|
expect((xv ^ yv) - 103n === (xv * yv) % 2n ** 32n).toStrictEqual(true);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('Solver', () => {
|
||||||
|
it('can use push and pop', async () => {
|
||||||
|
const { Solver, Int } = new api.Context('main');
|
||||||
|
const solver = new Solver();
|
||||||
|
const x = Int.const('x');
|
||||||
|
|
||||||
|
solver.add(x.gt(0));
|
||||||
|
|
||||||
|
expect(await solver.check()).toStrictEqual(sat);
|
||||||
|
|
||||||
|
solver.push();
|
||||||
|
solver.add(x.lt(0));
|
||||||
|
|
||||||
|
expect(solver.numScopes()).toStrictEqual(1);
|
||||||
|
expect(await solver.check()).toStrictEqual(unsat);
|
||||||
|
|
||||||
|
solver.pop();
|
||||||
|
|
||||||
|
expect(solver.numScopes()).toStrictEqual(0);
|
||||||
|
expect(await solver.check()).toStrictEqual(sat);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('can find multiple solutions', async () => {
|
||||||
|
const { Int, isIntVal } = new api.Context('main');
|
||||||
|
|
||||||
|
const x = Int.const('x');
|
||||||
|
|
||||||
|
const solutions = await asyncToArray(allSolutions(x.ge(1), x.le(5)));
|
||||||
|
expect(solutions.length).toStrictEqual(5);
|
||||||
|
const results = solutions
|
||||||
|
.map(solution => {
|
||||||
|
const expr = solution.eval(x);
|
||||||
|
assert(isIntVal(expr));
|
||||||
|
return expr.value;
|
||||||
|
})
|
||||||
|
.sort((a, b) => {
|
||||||
|
assert(a !== null && b !== null && typeof a === 'bigint' && typeof b === 'bigint');
|
||||||
|
if (a < b) {
|
||||||
|
return -1;
|
||||||
|
} else if (a == b) {
|
||||||
|
return 0;
|
||||||
|
} else {
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
});
|
||||||
|
expect(results).toStrictEqual([1n, 2n, 3n, 4n, 5n]);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('AstVector', () => {
|
||||||
|
it('can use basic methods', async () => {
|
||||||
|
const { Solver, AstVector, Int } = new api.Context('main');
|
||||||
|
const solver = new Solver();
|
||||||
|
|
||||||
|
const vector = new AstVector<Arith>();
|
||||||
|
for (let i = 0; i < 5; i++) {
|
||||||
|
vector.push(Int.const(`int__${i}`));
|
||||||
|
}
|
||||||
|
|
||||||
|
const length = vector.length;
|
||||||
|
for (let i = 0; i < length; i++) {
|
||||||
|
solver.add(vector.get(i).gt(1));
|
||||||
|
}
|
||||||
|
|
||||||
|
expect(await solver.check()).toStrictEqual(sat);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
1888
src/api/js/src/high-level/high-level.ts
Normal file
1888
src/api/js/src/high-level/high-level.ts
Normal file
File diff suppressed because it is too large
Load diff
2
src/api/js/src/high-level/index.ts
Normal file
2
src/api/js/src/high-level/index.ts
Normal file
|
|
@ -0,0 +1,2 @@
|
||||||
|
export * from './high-level';
|
||||||
|
export * from './types';
|
||||||
1132
src/api/js/src/high-level/types.ts
Normal file
1132
src/api/js/src/high-level/types.ts
Normal file
File diff suppressed because it is too large
Load diff
90
src/api/js/src/high-level/utils.test.ts
Normal file
90
src/api/js/src/high-level/utils.test.ts
Normal file
|
|
@ -0,0 +1,90 @@
|
||||||
|
import { Z3AssertionError } from './types';
|
||||||
|
import { allSatisfy, assert, assertExhaustive, autoBind } from './utils';
|
||||||
|
|
||||||
|
describe('allSatisfy', () => {
|
||||||
|
it('returns null on empty array', () => {
|
||||||
|
expect(allSatisfy([], () => true)).toBeNull();
|
||||||
|
});
|
||||||
|
|
||||||
|
it('returns true if all satisfy', () => {
|
||||||
|
expect(allSatisfy([2, 4, 6, 8], arg => arg % 2 === 0)).toStrictEqual(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('returns false if any element fails', () => {
|
||||||
|
expect(allSatisfy([2, 4, 1, 8], arg => arg % 2 === 0)).toStrictEqual(false);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('assertExhaustive', () => {
|
||||||
|
enum MyEnum {
|
||||||
|
A,
|
||||||
|
B,
|
||||||
|
}
|
||||||
|
it('stops compilation', () => {
|
||||||
|
const result: MyEnum = MyEnum.A as any;
|
||||||
|
switch (result) {
|
||||||
|
case MyEnum.A:
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
// @ts-expect-error
|
||||||
|
assertExhaustive(result);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
it('allows compilation', () => {
|
||||||
|
const result: MyEnum = MyEnum.A as any;
|
||||||
|
switch (result) {
|
||||||
|
case MyEnum.A:
|
||||||
|
break;
|
||||||
|
case MyEnum.B:
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
assertExhaustive(result);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
it('throws', () => {
|
||||||
|
const result: MyEnum = undefined as any;
|
||||||
|
switch (result) {
|
||||||
|
case MyEnum.A:
|
||||||
|
throw new Error();
|
||||||
|
case MyEnum.B:
|
||||||
|
throw new Error();
|
||||||
|
default:
|
||||||
|
expect(() => assertExhaustive(result)).toThrowError();
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('autoBind', () => {
|
||||||
|
class Binded {
|
||||||
|
readonly name = 'Richard';
|
||||||
|
constructor(shouldBind: boolean) {
|
||||||
|
if (shouldBind === true) {
|
||||||
|
autoBind(this);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
test(): string {
|
||||||
|
return `Hello ${this.name}`;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
it('binds this', () => {
|
||||||
|
const { test: withoutBind } = new Binded(false);
|
||||||
|
const { test: withBind } = new Binded(true);
|
||||||
|
expect(() => withoutBind()).toThrowError(TypeError);
|
||||||
|
expect(withBind()).toStrictEqual('Hello Richard');
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('assert', () => {
|
||||||
|
it('throws on failure', () => {
|
||||||
|
expect(() => assert(false)).toThrowError(Z3AssertionError);
|
||||||
|
expect(() => assert(false, 'foobar')).toThrowError(new Z3AssertionError('foobar'));
|
||||||
|
});
|
||||||
|
|
||||||
|
it('does nothing on sucess', () => {
|
||||||
|
expect(() => assert(true, 'hello')).not.toThrow();
|
||||||
|
});
|
||||||
|
});
|
||||||
85
src/api/js/src/high-level/utils.ts
Normal file
85
src/api/js/src/high-level/utils.ts
Normal file
|
|
@ -0,0 +1,85 @@
|
||||||
|
import { Z3AssertionError } from './types';
|
||||||
|
|
||||||
|
function getAllProperties(obj: Record<string, any>) {
|
||||||
|
const properties = new Set<[any, string | symbol]>();
|
||||||
|
do {
|
||||||
|
for (const key of Reflect.ownKeys(obj)) {
|
||||||
|
properties.add([obj, key]);
|
||||||
|
}
|
||||||
|
} while ((obj = Reflect.getPrototypeOf(obj)!) && obj !== Object.prototype);
|
||||||
|
return properties;
|
||||||
|
}
|
||||||
|
|
||||||
|
// https://github.com/sindresorhus/auto-bind
|
||||||
|
// We modify it to use CommonJS instead of ESM
|
||||||
|
/*
|
||||||
|
MIT License
|
||||||
|
|
||||||
|
Copyright (c) Sindre Sorhus <sindresorhus@gmail.com> (https://sindresorhus.com)
|
||||||
|
|
||||||
|
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 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.
|
||||||
|
*/
|
||||||
|
export function autoBind<Self extends Record<string | symbol, any>>(self: Self): Self {
|
||||||
|
for (const [obj, key] of getAllProperties(self.constructor.prototype)) {
|
||||||
|
if (key === 'constructor') {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
const descriptor = Reflect.getOwnPropertyDescriptor(obj, key);
|
||||||
|
if (descriptor && typeof descriptor.value === 'function') {
|
||||||
|
(self[key] as any) = self[key].bind(self);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return self;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Use to ensure that switches are checked to be exhaustive at compile time
|
||||||
|
*
|
||||||
|
* @example Basic usage
|
||||||
|
* ```typescript
|
||||||
|
* enum Something {
|
||||||
|
* left,
|
||||||
|
* right,
|
||||||
|
* };
|
||||||
|
* const something = getSomething();
|
||||||
|
* switch (something) {
|
||||||
|
* case Something.left:
|
||||||
|
* ...
|
||||||
|
* case Something.right:
|
||||||
|
* ...
|
||||||
|
* default:
|
||||||
|
* assertExhaustive(something);
|
||||||
|
* }
|
||||||
|
* ```
|
||||||
|
*
|
||||||
|
* @param x - The param on which the switch operates
|
||||||
|
*/
|
||||||
|
export function assertExhaustive(x: never): never {
|
||||||
|
throw new Error('Unexpected code execution detected, should be caught at compile time');
|
||||||
|
}
|
||||||
|
|
||||||
|
export function assert(condition: boolean, reason?: string): asserts condition {
|
||||||
|
if (!condition) {
|
||||||
|
throw new Z3AssertionError(reason ?? 'Assertion failed');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Check the all elements of a `collection` satisfy the `premise`.
|
||||||
|
* If any of the items fail the `premise`, returns false;
|
||||||
|
* @returns null if the `collection` is empty, boolean otherwise
|
||||||
|
*/
|
||||||
|
export function allSatisfy<T>(collection: Iterable<T>, premise: (arg: T) => boolean): boolean | null {
|
||||||
|
let hasItems = false;
|
||||||
|
for (const arg of collection) {
|
||||||
|
hasItems = true;
|
||||||
|
if (!premise(arg)) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return hasItems === true ? true : null;
|
||||||
|
}
|
||||||
62
src/api/js/src/jest.ts
Normal file
62
src/api/js/src/jest.ts
Normal file
|
|
@ -0,0 +1,62 @@
|
||||||
|
// This file is not included in the build
|
||||||
|
|
||||||
|
// @ts-ignore no-implicit-any
|
||||||
|
import { createApi, Z3HighLevel } from './high-level';
|
||||||
|
import { init as initWrapper, Z3LowLevel } from './low-level';
|
||||||
|
import initModule = require('../build/z3-built');
|
||||||
|
|
||||||
|
export * from './high-level/types';
|
||||||
|
export { Z3Core, Z3LowLevel } from './low-level';
|
||||||
|
export * from './low-level/types.__GENERATED__';
|
||||||
|
|
||||||
|
export async function init(): Promise<Z3HighLevel & Z3LowLevel> {
|
||||||
|
const lowLevel = await initWrapper(initModule);
|
||||||
|
const highLevel = createApi(lowLevel.Z3);
|
||||||
|
return { ...lowLevel, ...highLevel };
|
||||||
|
}
|
||||||
|
|
||||||
|
function delay(ms: number): Promise<void> & { cancel(): void };
|
||||||
|
function delay(ms: number, result: Error): Promise<never> & { cancel(): void };
|
||||||
|
function delay<T>(ms: number, result: T): Promise<T> & { cancel(): void };
|
||||||
|
function delay<T>(ms: number, result?: T | Error): Promise<T | void> & { cancel(): void } {
|
||||||
|
let handle: any;
|
||||||
|
const promise = new Promise<void | T>(
|
||||||
|
(resolve, reject) =>
|
||||||
|
(handle = setTimeout(() => {
|
||||||
|
if (result instanceof Error) {
|
||||||
|
reject(result);
|
||||||
|
} else if (result !== undefined) {
|
||||||
|
resolve(result);
|
||||||
|
}
|
||||||
|
resolve();
|
||||||
|
}, ms)),
|
||||||
|
);
|
||||||
|
return { ...promise, cancel: () => clearTimeout(handle) };
|
||||||
|
}
|
||||||
|
|
||||||
|
function waitWhile(premise: () => boolean, pollMs: number = 100): Promise<void> & { cancel(): void } {
|
||||||
|
let handle: any;
|
||||||
|
const promise = new Promise<void>(resolve => {
|
||||||
|
handle = setInterval(() => {
|
||||||
|
if (premise()) {
|
||||||
|
clearTimeout(handle);
|
||||||
|
resolve();
|
||||||
|
}
|
||||||
|
}, pollMs);
|
||||||
|
});
|
||||||
|
return { ...promise, cancel: () => clearInterval(handle) };
|
||||||
|
}
|
||||||
|
|
||||||
|
export function killThreads(em: any): Promise<void> {
|
||||||
|
em.PThread.terminateAllThreads();
|
||||||
|
|
||||||
|
// Create a polling lock to wait for threads to return
|
||||||
|
// TODO(ritave): Threads should be killed automatically, or there should be a better way to wait for them
|
||||||
|
const lockPromise = waitWhile(() => !em.PThread.unusedWorkers.length && !em.PThread.runningWorkers.length);
|
||||||
|
const delayPromise = delay(5000, new Error('Waiting for threads to be killed timed out'));
|
||||||
|
|
||||||
|
return Promise.race([lockPromise, delayPromise]).finally(() => {
|
||||||
|
lockPromise.cancel();
|
||||||
|
delayPromise.cancel();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
@ -1,4 +1,5 @@
|
||||||
// this wrapper works with async-fns to provide promise-based off-thread versions of some functions
|
// this wrapper works with async-fns to provide promise-based off-thread versions of some functions
|
||||||
|
// It's prepended directly by emscripten to the resulting z3-built.js
|
||||||
|
|
||||||
let capability = null;
|
let capability = null;
|
||||||
function resolve_async(val) {
|
function resolve_async(val) {
|
||||||
4
src/api/js/src/low-level/index.ts
Normal file
4
src/api/js/src/low-level/index.ts
Normal file
|
|
@ -0,0 +1,4 @@
|
||||||
|
export * from './types.__GENERATED__';
|
||||||
|
export * from './wrapper.__GENERATED__';
|
||||||
|
export type Z3Core = Awaited<ReturnType<typeof import('./wrapper.__GENERATED__')['init']>>['Z3'];
|
||||||
|
export type Z3LowLevel = Awaited<ReturnType<typeof import('./wrapper.__GENERATED__')['init']>>;
|
||||||
|
|
@ -1,10 +0,0 @@
|
||||||
// @ts-ignore no-implicit-any
|
|
||||||
import initModule = require('./z3-built.js');
|
|
||||||
|
|
||||||
// @ts-ignore no-implicit-any
|
|
||||||
import { init as initWrapper } from './wrapper';
|
|
||||||
|
|
||||||
export * from './wrapper';
|
|
||||||
export function init() {
|
|
||||||
return initWrapper(initModule);
|
|
||||||
}
|
|
||||||
38
src/api/js/src/node.ts
Normal file
38
src/api/js/src/node.ts
Normal file
|
|
@ -0,0 +1,38 @@
|
||||||
|
// @ts-ignore no-implicit-any
|
||||||
|
import initModule = require('./z3-built');
|
||||||
|
|
||||||
|
import { createApi, Z3HighLevel } from './high-level';
|
||||||
|
import { init as initWrapper, Z3LowLevel } from './low-level';
|
||||||
|
export * from './high-level/types';
|
||||||
|
export { Z3Core, Z3LowLevel } from './low-level';
|
||||||
|
export * from './low-level/types.__GENERATED__';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The main entry point to the Z3 API
|
||||||
|
*
|
||||||
|
* ```typescript
|
||||||
|
* import { init, sat } from 'z3-solver';
|
||||||
|
*
|
||||||
|
* const { Context } = await init();
|
||||||
|
* const { Solver, Int } = new Context('main');
|
||||||
|
*
|
||||||
|
* const x = Int.const('x');
|
||||||
|
* const y = Int.const('y');
|
||||||
|
*
|
||||||
|
* const solver = new Solver();
|
||||||
|
* solver.add(x.add(2).le(y.sub(10))); // x + 2 <= y - 10
|
||||||
|
*
|
||||||
|
* if (await solver.check() !== sat) {
|
||||||
|
* throw new Error("couldn't find a solution")
|
||||||
|
* }
|
||||||
|
* const model = solver.model();
|
||||||
|
*
|
||||||
|
* console.log(`x=${model.get(x)}, y=${model.get(y)}`);
|
||||||
|
* // x=0, y=12
|
||||||
|
* ```
|
||||||
|
* @category Global */
|
||||||
|
export async function init(): Promise<Z3HighLevel & Z3LowLevel> {
|
||||||
|
const lowLevel = await initWrapper(initModule);
|
||||||
|
const highLevel = createApi(lowLevel.Z3);
|
||||||
|
return { ...lowLevel, ...highLevel };
|
||||||
|
}
|
||||||
4
src/api/js/tsconfig.build.json
Normal file
4
src/api/js/tsconfig.build.json
Normal file
|
|
@ -0,0 +1,4 @@
|
||||||
|
{
|
||||||
|
"extends": "./tsconfig.json",
|
||||||
|
"exclude": ["src/**/*.test.ts", "src/jest.ts"]
|
||||||
|
}
|
||||||
|
|
@ -6,9 +6,11 @@
|
||||||
"declaration": true,
|
"declaration": true,
|
||||||
"forceConsistentCasingInFileNames": true,
|
"forceConsistentCasingInFileNames": true,
|
||||||
"strict": true,
|
"strict": true,
|
||||||
"skipLibCheck": true
|
"skipLibCheck": true,
|
||||||
|
"outDir": "build/",
|
||||||
|
"allowJs": true,
|
||||||
|
"checkJs": false
|
||||||
},
|
},
|
||||||
"exclude": [
|
"include": ["src/**/*.ts"],
|
||||||
"src"
|
"exclude": ["node_modules"]
|
||||||
]
|
|
||||||
}
|
}
|
||||||
|
|
|
||||||
8
src/api/js/typedoc.json
Normal file
8
src/api/js/typedoc.json
Normal file
|
|
@ -0,0 +1,8 @@
|
||||||
|
{
|
||||||
|
"$schema": "https://typedoc.org/schema.json",
|
||||||
|
"entryPoints": ["./src/node.ts"],
|
||||||
|
"out": "../../../doc/api/html/js",
|
||||||
|
"exclude": ["./src/low-level/**/*"],
|
||||||
|
"readme": "./PUBLISHED_README.md",
|
||||||
|
"categorizeByGroup": false
|
||||||
|
}
|
||||||
Loading…
Reference in a new issue