homebrew-core/Formula/fftw.rb
Mike McQuaid 7a4dabfc1a Use hash rockets again. ()
This restores 1.8 hash rockets because they look nicer with e.g. `depends_on :foo => :bar`
2016-09-24 16:45:48 +01:00

85 lines
3.1 KiB
Ruby

class Fftw < Formula
desc "C routines to compute the Discrete Fourier Transform"
homepage "http://www.fftw.org"
url "http://www.fftw.org/fftw-3.3.5.tar.gz"
sha256 "8ecfe1b04732ec3f5b7d279fdb8efcad536d555f9d1e8fabd027037d45ea8bcf"
bottle do
cellar :any
sha256 "6732254af0d37777ffd9ae7a4dc16d672def6656cdd85c4d8f1e96a53571e123" => :sierra
sha256 "8961b3780e24d67c520deccf222aa12ce95f41eba1b6d7a7181e83bda9e4a99e" => :el_capitan
sha256 "d7358e5c3e9426c245425ab0d4fe29659d7bdb41158bedb43ab0c2433b165c69" => :yosemite
sha256 "f31992664fb2c54ae917c6e31a44e4a798b4ff062fbc66e50fcf7f56c418c791" => :mavericks
end
option "with-fortran", "Enable Fortran bindings"
option :universal
option "with-mpi", "Enable MPI parallel transforms"
option "with-openmp", "Enable OpenMP parallel transforms"
depends_on :fortran => :optional
depends_on :mpi => [:cc, :optional]
needs :openmp if build.with? "openmp"
def install
args = ["--enable-shared",
"--disable-debug",
"--prefix=#{prefix}",
"--enable-threads",
"--disable-dependency-tracking"]
simd_args = ["--enable-sse2"]
simd_args << "--enable-avx" if ENV.compiler == :clang && Hardware::CPU.avx? && !build.bottle?
simd_args << "--enable-avx2" if ENV.compiler == :clang && Hardware::CPU.avx2? && !build.bottle?
args << "--disable-fortran" if build.without? "fortran"
args << "--enable-mpi" if build.with? "mpi"
args << "--enable-openmp" if build.with? "openmp"
ENV.universal_binary if build.universal?
# single precision
# enable-sse2, enable-avx and enable-avx2 work for both single and double precision
system "./configure", "--enable-single", *(args + simd_args)
system "make", "install"
# clean up so we can compile the double precision variant
system "make", "clean"
# double precision
# enable-sse2, enable-avx and enable-avx2 work for both single and double precision
system "./configure", *(args + simd_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.to_s.split
system "./fftw"
end
end