sz: update to version 2.1.8.1 (#15154)

The newer versions of SZ now uses CMake to build.

1 files changed, 81 insertions, 12 deletions

diff --git a/var/spack/repos/builtin/packages/sz/package.py b/var/spack/repos/builtin/packages/sz/package.py
index 3453a5675e..774e123bf3 100644
--- a/var/spack/repos/builtin/packages/sz/package.py
+++ b/var/spack/repos/builtin/packages/sz/package.py
@@ -3,21 +3,19 @@
 #
 # SPDX-License-Identifier: (Apache-2.0 OR MIT)
 
-
 from spack import *
 
 
-class Sz(AutotoolsPackage):
-    """Error-bounded Lossy Compressor for HPC Data."""
+class Sz(CMakePackage):
+    """Error-bounded Lossy Compressor for HPC Data"""
 
     homepage = "https://collab.cels.anl.gov/display/ESR/SZ"
     url      = "https://github.com/disheng222/SZ/archive/v2.1.8.0.tar.gz"
+    git      = "https://github.com/disheng222/sz"
     maintainers = ['disheng222']
-    parallel = False
-
-    git      = "https://github.com/disheng222/SZ.git"
 
     version('develop', branch='master')
+    version('2.1.8.1', sha256='a27c9c9da16c9c4232c54813ba79178945f47609043f11501d49a171e47d3f46')
     version('2.1.8.0', sha256='8d6bceb59a03d52e601e29d9b35c21b146c248abae352f9a4828e91d8d26aa24')
     version('2.0.2.0',  sha256='176c65b421bdec8e91010ffbc9c7bf7852c799972101d6b66d2a30d9702e59b0')
     version('1.4.13.5', sha256='b5e37bf3c377833eed0a7ca0471333c96cd2a82863abfc73893561aaba5f18b9')
@@ -32,17 +30,88 @@ class Sz(AutotoolsPackage):
     version('1.4.10.0', sha256='cf23cf1ffd7c69c3d3128ae9c356b6acdc03a38f92c02db5d9bfc04f3fabc506')
     version('1.4.9.2',  sha256='9dc785274d068d04c2836955fc93518a9797bfd409b46fea5733294b7c7c18f8')
 
-    variant('fortran', default=False,
-            description='Enable fortran compilation')
+    variant('python', default=False, description="builds the python wrapper")
+    variant('netcdf', default=False, description="build the netcdf reader")
+    variant('hdf5', default=False, description="build the hdf5 filter")
+    variant('pastri', default=False, description="build the pastri mode")
+    variant('time_compression', default=False, description="build the time based compression mode")
+    variant('random_access', default=False, description="build the random access compression mode")
+    variant('fortran', default=False, description='Enable fortran compilation')
+    variant('shared', default=True, description="build shared versions of the libraries")
 
     # Part of latest sources don't support -O3 optimization
     # with Fujitsu compiler.
     patch('fix_optimization.patch', when='@2.0.2.0:%fj')
 
-    def configure_args(self):
+    depends_on('zlib')
+    depends_on('zstd')
+
+    extends('python', when="+python")
+    depends_on('python@3:', when="+python", type=('build', 'link', 'run'))
+    depends_on('swig@3.12:', when="+python", type='build')
+    depends_on('py-numpy', when="+python", type=('build', 'link', 'run'))
+    depends_on('hdf5', when="+hdf5")
+    depends_on('netcdf-c', when="+netcdf")
+
+    @property
+    def build_directory(self):
+        """autotools needs a different build directory to work"""
+        if self.version >= Version("2.1.8.1"):
+            return "spack-build"
+        else:
+            return "."
+
+    @when("@:2.1.8.0")
+    def cmake(self, spec, prefix):
+        """use autotools before 2.1.8.1"""
+        configure_args = ["--prefix=" + prefix]
+        if "+fortran" in spec:
+            configure_args.append("--enable-fortran")
+        else:
+            configure_args.append("--disable-fortran")
+        configure(*configure_args)
+
+    def cmake_args(self):
+        """configure the package with CMake for version 2.1.8.1 and later"""
         args = []
-        if '+fortran' in self.spec:
-            args += ['--enable-fortran']
+
+        if "+python" in self.spec:
+            args.append("-DBUILD_PYTHON_WRAPPER=ON")
+        else:
+            args.append("-DBUILD_PYTHON_WRAPPER=OFF")
+
+        if "+netcdf" in self.spec:
+            args.append("-DBUILD_NETCDF_READER=ON")
+        else:
+            args.append("-DBUILD_NETCDF_READER=OFF")
+
+        if "+hdf5" in self.spec:
+            args.append("-DBUILD_HDF5_FILTER=ON")
+        else:
+            args.append("-DBUILD_HDF5_FILTER=OFF")
+
+        if "+pastri" in self.spec:
+            args.append("-DBUILD_PASTRI=ON")
+        else:
+            args.append("-DBUILD_PASTRI=OFF")
+
+        if "+time_compression" in self.spec:
+            args.append("-DBUILD_TIMECMPR=ON")
+        else:
+            args.append("-DBUILD_TIMECMPR=OFF")
+
+        if "+random_access" in self.spec:
+            args.append("-DBUILD_RANDOMACCESS=ON")
+        else:
+            args.append("-DBUILD_RANDOMACCESS=OFF")
+
+        if "+fortran" in self.spec:
+            args.append("-DBUILD_FORTRAN=ON")
+        else:
+            args.append("-DBUILD_FORTRAN=OFF")
+
+        if "+shared" in self.spec:
+            args.append("-DBUILD_SHARED_LIBS=ON")
         else:
-            args += ['--disable-fortran']
+            args.append("-DBUILD_SHARED_LIBS=OFF")
         return args