utils.py

Go to the documentation of this file.

#  Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
#  See https://llvm.org/LICENSE.txt for license information.
#  SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 
import platform
from pathlib import Path
import subprocess
import sys
 
_thisdir = Path(__file__).absolute().parent
 
 
def run_esiquery():
  """Run the esiquery executable with the same arguments as this script."""
  if platform.system() == "Windows":
    esiquery = _thisdir / "esiquery.exe"
  else:
    esiquery = _thisdir / "bin" / "esiquery"
  return subprocess.call([esiquery] + sys.argv[1:])
 
 
def run_esi_cosim():
  """Run the esi-cosim.py script with the same arguments as this script."""
  import importlib.util
  if platform.system() == "Windows":
    esi_cosim = _thisdir / "esi-cosim.py"
  else:
    esi_cosim = _thisdir / "bin" / "esi-cosim.py"
  spec = importlib.util.spec_from_file_location("esi_cosim", esi_cosim)
  assert spec is not None
  assert spec.loader is not None
  cosim_import = importlib.util.module_from_spec(spec)
  spec.loader.exec_module(cosim_import)
  return cosim_import.__main__(sys.argv)
 
 
def run_cppgen():
  from . import codegen
  return codegen.run()
 
 
def get_cmake_dir() -> Path:
  return _thisdir / "cmake"
 
 
def get_dll_dir() -> Path:
  """Return the directory where the ESI dll's are located"""
  import sys
  import os
  if sys.platform == "win32":
    dll_name = "ESICppRuntime.dll"
    for dir in [_thisdir, _thisdir.parent, _thisdir.parent.parent]:
      if (dir / dll_name).exists():
        return dir
    raise FileNotFoundError("ESICppRuntime.dll not found")
  else:
    so_name = "libESICppRuntime.so"
    for dir in [_thisdir, _thisdir.parent, _thisdir.parent.parent]:
      if (dir / "lib" / so_name).exists():
        return dir / "lib"
    raise FileNotFoundError("libESICppRuntime.so not found")