+# Parse the command line arguments for VAR=value assignments,
+# and apply them as environment variables.
+while len(sys.argv) > 1:
+ match = re.match(r'([A-Z]+)=(.*)', sys.argv[1])
+ if match is None:
+ break
+ os.environ[match.group(1)] = match.group(2)
+ del sys.argv[1]
+
+includes = ['../../include', '../cxx/include']
+includes += [os.path.normpath(os.path.join(srcdir, path)) for path in includes]
+includes += ['../..', np.get_include()]
+
+ldadd = shlex.split(os.environ.get('LDADD', ''))
+libdirs = ['../../.libs', '../cxx/.libs'] + \
+ [l[2:] for l in ldadd if l.startswith('-L')]
+libs = [l[2:] for l in ldadd if l.startswith('-l')] + ['sigrokcxx']
+
+def vpath(file):
+ vfile = os.path.join(srcdir, file)
+ return vfile if os.path.exists(vfile) else file
+
+def unvpath(file):
+ return os.path.relpath(file, srcdir) if file.startswith(srcdir) else file
+
+class build_py(_build_py):
+ def find_package_modules(self, package, pkg_dir):
+ mods = _build_py.find_package_modules(self, package, pkg_dir)
+ vmods = _build_py.find_package_modules(self, package, vpath(pkg_dir))
+ mods.extend([mod for mod in vmods if mod not in mods])
+ return mods
+ def check_package(self, package, package_dir):
+ return _build_py.check_package(self, package, vpath(package_dir))
+
+class build_ext(_build_ext):
+ def finalize_options(self):
+ _build_ext.finalize_options(self)
+ self.swig_opts = ['-c++', '-threads', '-Isigrok/core', '-I..',
+ '-I' + srcdir_parent] + ['-I%s' % i for i in includes] + self.swig_opts
+ def spawn (self, cmd):
+ cmd[1:-1] = [arg if arg.startswith('-') else unvpath(arg) for arg in
+ cmd[1:-1]]
+ _build_ext.spawn(self, cmd)
+ def swig_sources (self, sources, extension):
+ return [unvpath(src) for src in
+ _build_ext.swig_sources(self, sources, extension)]