spack/var/spack/repos/builtin/packages/cpmd/package.py
Tom Scogland 18c2f1a57a
refactor: packages import spack.package explicitly (#30404)
Explicitly import package utilities in all packages, and corresponding fallout.

This includes:

* rename `spack.package` to `spack.package_base`
* rename `spack.pkgkit` to `spack.package`
* update all packages in builtin, builtin_mock and tutorials to include `from spack.package import *`
* update spack style
  * ensure packages include the import
  * automatically add the new import and remove any/all imports of `spack` and `spack.pkgkit`
    from packages when using `--fix`
  * add support for type-checking packages with mypy when SPACK_MYPY_CHECK_PACKAGES
    is set in the environment
* fix all type checking errors in packages in spack upstream
* update spack create to include the new imports
* update spack repo to inject the new import, injection persists to allow for a deprecation period

Original message below:
 
As requested @adamjstewart, update all packages to use pkgkit.  I ended up using isort to do this,
so repro is easy:

```console
$ isort -a 'from spack.pkgkit import *' --rm 'spack' ./var/spack/repos/builtin/packages/*/package.py
$ spack style --fix
```

There were several line spacing fixups caused either by space manipulation in isort or by packages
that haven't been touched since we added requirements, but there are no functional changes in here.

* [x] add config to isort to make sure this is maintained going forward
2022-05-28 12:55:44 -04:00

107 lines
4.0 KiB
Python

# Copyright 2013-2022 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
import os
from spack.package import *
class Cpmd(MakefilePackage):
"""The CPMD code is a parallelized plane wave / pseudopotential
implementation of Density Functional Theory, particularly
designed for ab-initio molecular dynamics.
Move to new directory, download CPMD main archive and patch.to.XXXXs
manually, and run Spack"""
homepage = "https://www.cpmd.org/wordpress/"
basedir = os.getcwd()
url = "file://{0}/cpmd-v4.3.tar.gz".format(basedir)
manual_download = True
version('4.3', sha256='4f31ddf045f1ae5d6f25559d85ddbdab4d7a6200362849df833632976d095df4')
variant('omp', description='Enables the use of OMP instructions',
default=False)
variant('mpi', description='Build with MPI support', default=False)
depends_on('lapack')
depends_on('mpi', when='+mpi')
conflicts('^openblas threads=none', when='+omp')
conflicts('^openblas threads=pthreads', when='+omp')
patch('file://{0}/patch.to.4612'.format(basedir), sha256='3b7d91e04c40418ad958069234ec7253fbf6c4be361a1d5cfd804774eeb44915', level=0, when='@4.3')
patch('file://{0}/patch.to.4615'.format(basedir), sha256='5ec5790fb6ca64632bcc1b0f5b8f3423c54455766a0979ff4136624bbe8d49eb', level=0, when='@4.3')
patch('file://{0}/patch.to.4616'.format(basedir), sha256='ac0bc215c4259f55da4dc59803fe636f797e241f8a01974e05730c9778ad44c4', level=0, when='@4.3')
patch('file://{0}/patch.to.4621'.format(basedir), sha256='2d2bc7e37246032fc354f51da7dbdb5a219dd228867399931b0e94da1265d5ca', level=0, when='@4.3')
patch('file://{0}/patch.to.4624'.format(basedir), sha256='0a19687528264bf91c9f50ffdc0b920a8511eecf5259b667c8c29350f9dabc53', level=0, when='@4.3')
def edit(self, spec, prefix):
# patch configure file
cbase = 'LINUX-GFORTRAN'
cp = FileFilter(join_path('configure', cbase))
# Compilers
if spec.satisfies('+mpi'):
fc = spec["mpi"].mpifc
cc = spec["mpi"].mpicc
else:
fc = spack_fc
cc = spack_cc
cp.filter('FC=.+', "FC='{0}'".format(fc))
cp.filter('CC=.+', "CC='{0}'".format(cc))
cp.filter('LD=.+', "LD='{0}'".format(fc))
# MPI flag
if spec.satisfies('+mpi'):
cp.filter('-D__Linux', '-D__Linux -D__PARALLEL')
# OMP flag
if spec.satisfies('+omp'):
cp.filter('-fopenmp', self.compiler.openmp_flag)
# lapack
cp.filter(
'LIBS=.+',
"LIBS='{0}'".format(spec['lapack'].libs.ld_flags)
)
# LFLAGS
cp.filter("'-static '", '')
# Compiler specific
if spec.satisfies('%fj'):
cp.filter('-ffixed-form', '-Fixed')
cp.filter('-ffree-line-length-none', '')
cp.filter('-falign-commons', '-Kalign_commons')
# create Makefile
bash = which('bash')
if spec.satisfies('+omp'):
bash('./configure.sh', '-omp', cbase)
else:
bash('./configure.sh', cbase)
def install(self, spec, prefix):
install_tree('.', prefix)
def test(self):
test_dir = self.test_suite.current_test_data_dir
test_file = join_path(test_dir, '1-h2o-pbc-geoopt.inp')
opts = []
if self.spec.satisfies('+mpi'):
exe_name = self.spec['mpi'].prefix.bin.mpirun
opts.extend(['-n', '2'])
opts.append(join_path(self.prefix.bin, 'cpmd.x'))
else:
exe_name = 'cpmd.x'
opts.append(test_file)
opts.append(test_dir)
expected = ['2 1 H O 1.84444 0.97604',
'3 1 H O 1.84444 0.97604',
'2 1 3 H O H 103.8663'
]
self.run_test(exe_name, options=opts, expected=expected)