71 lines
2 KiB
Ruby
71 lines
2 KiB
Ruby
require 'formula'
|
|
|
|
class Fftw < Formula
|
|
homepage 'http://www.fftw.org'
|
|
url 'http://www.fftw.org/fftw-3.3.3.tar.gz'
|
|
sha1 '11487180928d05746d431ebe7a176b52fe205cf9'
|
|
|
|
option "with-fortran", "Enable Fortran bindings"
|
|
|
|
def install
|
|
args = ["--enable-shared",
|
|
"--disable-debug",
|
|
"--prefix=#{prefix}",
|
|
"--enable-threads",
|
|
"--disable-dependency-tracking"]
|
|
|
|
if build.include? "with-fortran"
|
|
ENV.fortran
|
|
else
|
|
args << "--disable-fortran" unless which 'gfortran'
|
|
end
|
|
|
|
# single precision
|
|
# enable-sse only works with single
|
|
system "./configure", "--enable-single",
|
|
"--enable-sse",
|
|
*args
|
|
system "make install"
|
|
|
|
# clean up so we can compile the double precision variant
|
|
system "make clean"
|
|
|
|
# double precision
|
|
# enable-sse2 only works with double precision (default)
|
|
system "./configure", "--enable-sse2", *args
|
|
system "make install"
|
|
|
|
# clean up so we can compile the long-double precision variant
|
|
system "make clean"
|
|
|
|
# long-double precision
|
|
# no SIMD optimization available
|
|
system "./configure", "--enable-long-double", *args
|
|
system "make install"
|
|
end
|
|
|
|
test do
|
|
# Adapted from the sample usage provided in the documentation:
|
|
# http://www.fftw.org/fftw3_doc/Complex-One_002dDimensional-DFTs.html
|
|
(testpath/'fftw.c').write <<-TEST_SCRIPT.undent
|
|
#include <fftw3.h>
|
|
|
|
int main(int argc, char* *argv)
|
|
{
|
|
fftw_complex *in, *out;
|
|
fftw_plan p;
|
|
long N = 1;
|
|
in = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * N);
|
|
out = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * N);
|
|
p = fftw_plan_dft_1d(N, in, out, FFTW_FORWARD, FFTW_ESTIMATE);
|
|
fftw_execute(p); /* repeat as needed */
|
|
fftw_destroy_plan(p);
|
|
fftw_free(in); fftw_free(out);
|
|
return 0;
|
|
}
|
|
TEST_SCRIPT
|
|
|
|
system ENV.cc, '-o', 'fftw', 'fftw.c', '-lfftw3', *ENV.cflags.split
|
|
system './fftw'
|
|
end
|
|
end
|