Skip to content
Snippets Groups Projects
Commit 73101651 authored by ferrari's avatar ferrari
Browse files

Auto compile c_corr

parent ac1857d8
No related branches found
No related tags found
No related merge requests found
from distutils.core import setup from distutils.core import setup
from Cython.Build import cythonize from Cython.Build import cythonize
setup( setup(ext_modules=cythonize("c_corr.pyx", annotate=True, compiler_directives={'language_level': "3"}))
ext_modules = cythonize("c_corr.pyx", annotate=True, compiler_directives={'language_level' : "3"})
)
\ No newline at end of file
import argparse
import itertools
import os import os
import sys 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 scipy.signal as sg
import soundfile as sf import soundfile as sf
import c_corr from numpy.fft import rfft, irfft
from gsrp_smart_util import *
from math import ceil
from scipy.signal.windows import tukey from scipy.signal.windows import tukey
from sklearn.linear_model import LinearRegression
from sklearn.pipeline import Pipeline
from sklearn.preprocessing import PolynomialFeatures
try: from gsrp_smart_util import *
from tqdm import trange
except ImportError:
trange = range
class BColors: class BColors:
...@@ -32,6 +26,20 @@ class BColors: ...@@ -32,6 +26,20 @@ class BColors:
UNDERLINE = '\033[4m' 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): def intlist(s):
return list(map(int, s.split(','))) return list(map(int, s.split(',')))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment