From 731016511501a7cbb66faed21f3ebb27a7df3e17 Mon Sep 17 00:00:00 2001
From: ferrari <maxence.ferrari@gmail.com>
Date: Wed, 8 Feb 2023 11:01:10 +0100
Subject: [PATCH] Auto compile c_corr
---
cython_setup.py | 4 +---
gsrp_tdoa_hyperres.py | 36 ++++++++++++++++++++++--------------
2 files changed, 23 insertions(+), 17 deletions(-)
diff --git a/cython_setup.py b/cython_setup.py
index 73f0fe2..05b8e58 100755
--- a/cython_setup.py
+++ b/cython_setup.py
@@ -1,6 +1,4 @@
from distutils.core import setup
from Cython.Build import cythonize
-setup(
- ext_modules = cythonize("c_corr.pyx", annotate=True, compiler_directives={'language_level' : "3"})
-)
\ No newline at end of file
+setup(ext_modules=cythonize("c_corr.pyx", annotate=True, compiler_directives={'language_level': "3"}))
diff --git a/gsrp_tdoa_hyperres.py b/gsrp_tdoa_hyperres.py
index ce507ab..c960369 100755
--- a/gsrp_tdoa_hyperres.py
+++ b/gsrp_tdoa_hyperres.py
@@ -1,23 +1,17 @@
+import argparse
+import itertools
import os
import sys
-import itertools
-import argparse
-from sklearn.pipeline import Pipeline
-from sklearn.preprocessing import PolynomialFeatures
-from sklearn.linear_model import LinearRegression
-import numpy as np
-from numpy.fft import rfft, irfft
+
import scipy.signal as sg
import soundfile as sf
-import c_corr
-from gsrp_smart_util import *
-from math import ceil
+from numpy.fft import rfft, irfft
from scipy.signal.windows import tukey
+from sklearn.linear_model import LinearRegression
+from sklearn.pipeline import Pipeline
+from sklearn.preprocessing import PolynomialFeatures
-try:
- from tqdm import trange
-except ImportError:
- trange = range
+from gsrp_smart_util import *
class BColors:
@@ -32,6 +26,20 @@ class BColors:
UNDERLINE = '\033[4m'
+try:
+ import c_corr
+except ModuleNotFoundError:
+ print(f'{BColors.WARNING}Unable to load c_corr. Attempting to recompile...{BColors.ENDC}')
+ from subprocess import run
+ run('python cython_setup.py build_ext --inplace'.split())
+ import c_corr
+ print(f'{BColors.OKGREEN}Done{BColors.ENDC}')
+try:
+ from tqdm import trange
+except ImportError:
+ trange = range
+
+
def intlist(s):
return list(map(int, s.split(',')))
--
GitLab