From ae44b1d7b90cc1c49ddab1e9d913e3b90fc4b2f1 Mon Sep 17 00:00:00 2001
From: iarspider <iarspider@gmail.com>
Date: Thu, 16 Jul 2020 15:33:23 +0200
Subject: [PATCH] New package: openloops (#17520)
* New package: OpenLoops
* install() for openloops
* Working OpenLoops recipe
* Flake-8
* Only copy collection file if required; add clarification to num_jobs
* Add __future__ import just in case
* Fix missing space
* Remove __future__ import
* Changes from review, pt. 1
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
* Replace print() with write()
* Flake-8
Co-authored-by: iarspider <iarpsider@gmail.com>
Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
---
.../builtin/packages/openloops/package.py | 123 ++++++++++++++++++
.../builtin/packages/openloops/sft1.coll | 26 ++++
.../builtin/packages/openloops/sft2.coll | 76 +++++++++++
3 files changed, 225 insertions(+)
create mode 100644 var/spack/repos/builtin/packages/openloops/package.py
create mode 100644 var/spack/repos/builtin/packages/openloops/sft1.coll
create mode 100644 var/spack/repos/builtin/packages/openloops/sft2.coll
diff --git a/var/spack/repos/builtin/packages/openloops/package.py b/var/spack/repos/builtin/packages/openloops/package.py
new file mode 100644
index 0000000000..884910beb3
--- /dev/null
+++ b/var/spack/repos/builtin/packages/openloops/package.py
@@ -0,0 +1,123 @@
+# Copyright 2013-2020 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)
+
+from spack import *
+import os
+
+
+class Openloops(Package):
+ """The OpenLoops 2 program is a fully automated implementation of the
+ Open Loops algorithm combined with on-the-fly reduction methods,
+ which allows for the fast and stable numerical evaluation of tree
+ and one-loop matrix elements for any Standard Model process
+ at NLO QCD and NLO EW. """
+
+ homepage = "https://openloops.hepforge.org/"
+ url = "https://openloops.hepforge.org/downloads?f=OpenLoops-2.1.1.tar.gz"
+
+ version('2.1.1', sha256='f1c47ece812227eab584e2c695fef74423d2f212873f762b8658f728685bcb91')
+
+ all_processes = (
+ "tbln", "tbln_ew", "tbqq", "tbw", "pptttt", "pptttt_ew", "pptt",
+ "pptt_ew", "ppttbb", "ppttj", "ppttj_ew", "ppttjj",
+ "pptaj", "pptajj", "pptllj", "pptlljj", "pptln", "pptw", "pptwj",
+ "pptzj", "pptzjj", "ppthj", "ppthjj", "pptj",
+ "pptjj", "ppjj", "ppjj_ew", "ppjjj", "ppjjj_ew", "ppjjj_nf5", "ppjjjj",
+ "pplllvvv_ew", "ppatt", "ppatt_ew",
+ "ppattj", "pplltt", "pplltt_ew", "ppllttj", "ppllttj_ew", "pplntt",
+ "pplnttj", "ppwtt", "ppwtt_ew", "ppwttj",
+ "ppwttj_ew", "ppztt", "ppztt_ew", "ppzttj", "ppaatt", "ppwwtt",
+ "ppzatt", "ppzztt", "ppvvvv", "ppaaaj2", "ppllaa",
+ "ppllaaj", "pplllla", "ppvvv", "ppvvv2", "ppvvv_ew", "ppvvvj",
+ "ppaajj", "ppaajj2", "ppaajjj", "pplla", "pplla2",
+ "pplla_ew", "ppllaj", "ppllaj2", "ppllaj_ew", "ppllaj_nf5", "ppllajj",
+ "ppllll", "ppllll2", "ppllll2_nf5",
+ "ppllll2_onlyh", "ppllll_ew", "ppllllj", "ppllllj2", "ppllllj2_nf5",
+ "ppllllj2_nf5_notridr", "ppllllj2_nf5_sr",
+ "ppllllj2_onlyh", "ppllnnjj_ew", "ppllnnjj_vbs", "pplnajj", "ppvv",
+ "ppvv2", "ppvv_ew", "ppvvj", "ppvvj2",
+ "ppvvj_ew", "ppwajj", "ppwwjj", "ppzajj", "ppzwj_ew", "ppzwjj",
+ "ppzzjj", "ppajj", "ppajj2", "ppajj_ew", "ppajjj",
+ "ppllj", "ppllj2", "ppllj_ew", "ppllj_nf5", "pplljj", "pplljj_ew",
+ "pplljjj", "pplnj_ckm", "pplnjj", "pplnjj_ckm",
+ "pplnjj_ew", "pplnjjj", "ppnnjj_ew", "ppnnjjj", "ppvj", "ppvj2",
+ "ppvj_ew", "ppwj_ckm", "ppwjj", "ppwjj_ckm",
+ "ppwjj_ew", "ppwjjj", "ppzjj", "ppzjj_ew", "ppzjjj", "pphtt",
+ "pphtt_ew", "pphttj", "pphlltt", "pphll", "pphll2",
+ "pphll_ew", "pphllj", "pphllj2", "pphllj_ew", "pphlljj", "pphlljj_top",
+ "pphlnj_ckm", "pphlnjj", "pphv", "pphv_ew",
+ "pphwjj", "pphz2", "pphzj2", "pphzjj", "pphhtt", "pphhv", "pphhh2",
+ "heftpphh", "heftpphhj", "heftpphhjj", "pphh2",
+ "pphhj2", "pphhjj2", "pphhjj_vbf", "bbhj", "heftpphj", "heftpphjj",
+ "heftpphjjj", "pphbb", "pphbbj", "pphj2",
+ "pphjj2", "pphjj_vbf", "pphjj_vbf_ew", "pphjjj2", "eetttt", "eettttj",
+ "eellllbb", "eett", "eett_ew", "eettj",
+ "eettjj", "eevtt", "eevttj", "eevttjj", "eevvtt", "eevvttj",
+ "eellll_ew", "eevv_ew", "eevvjj", "eell_ew", "eevjj",
+ "eehtt", "eehttj", "eehll_ew", "eehvtt", "eehhtt", "heftppllj",
+ "heftpplljj", "heftpplljjj")
+
+ variant('compile_extra', default=False,
+ description='Compile real radiation tree amplitudes')
+ variant('processes', description='Processes to install. See https://' +
+ 'openloops.hepforge.org/process_' +
+ 'library.php?repo=public for details',
+ values=disjoint_sets(('all.coll',), ('lhc.coll',), ('lcg.coll',),
+ all_processes).with_default('lhc.coll'))
+
+ variant('num_jobs', description='Number of parallel jobs to run. ' +
+ 'Set to 1 if compiling a large number' +
+ 'of processes (e.g. lcg.coll)', default=0)
+ depends_on('python', type=("build", "run"))
+
+ phases = ['configure', 'build', 'build_processes', 'install']
+
+ def configure(self, spec, prefix):
+ spack_env = ('PATH LD_LIBRARY_PATH CPATH C_INCLUDE_PATH' +
+ 'CPLUS_INCLUDE_PATH INTEL_LICENSE_FILE').split()
+ for k in env.keys():
+ if k.startswith('SPACK_'):
+ spack_env.append(k)
+
+ spack_env = ' '.join(spack_env)
+ is_intel = self.spec.satisfies('%intel')
+ njobs = self.spec.variants['num_jobs'].value
+
+ with open('openloops.cfg', 'w') as f:
+ f.write('[OpenLoops]\n')
+ f.write('import_env={0}\n'.format(spack_env))
+ f.write('num_jobs = {0}\n'.format(njobs))
+ f.write('cc = {0}\n'.format(env['SPACK_CC']))
+ f.write('cxx = {0}\n'.format(env['SPACK_CXX']))
+ f.write('fortran_compiler = {0}\n'.format(env['SPACK_FC']))
+ if self.spec.satisfies('@1.3.1') and not is_intel:
+ f.write('gfortran_f_flags = -ffree-line-length-none\n')
+ if self.spec.satisfies('@2.1.1') and not is_intel:
+ f.write('gfortran_f_flags = -ffree-line-length-none ' +
+ '-fdollar-ok -mcmodel=medium\n')
+
+ if self.spec.satisfies('@:1.999.999 processes=lcg.coll'):
+ copy(join_path(os.path.dirname(__file__), 'sft1.coll'), 'lcg.coll')
+ elif self.spec.satisfies('@2:2.999.999 processes=lcg.coll'):
+ copy(join_path(os.path.dirname(__file__), 'sft2.coll'), 'lcg.coll')
+
+ def build(self, spec, prefix):
+ scons = Executable('./scons')
+ scons('generator=1', 'compile=2')
+
+ def build_processes(self, spec, prefix):
+ ol = Executable('./openloops')
+ processes = self.spec.variants['processes'].value
+ if 'compile_extra' in self.spec:
+ ce = 'compile_extra=1'
+ else:
+ ce = ''
+
+ ol('libinstall', ce, *processes)
+
+ def install(self, spec, prefix):
+ install_tree(join_path(self.stage.path, 'spack-src'),
+ self.prefix,
+ ignore=lambda x: x in ('process_obj', 'process_src'))
diff --git a/var/spack/repos/builtin/packages/openloops/sft1.coll b/var/spack/repos/builtin/packages/openloops/sft1.coll
new file mode 100644
index 0000000000..1a4bb5d404
--- /dev/null
+++ b/var/spack/repos/builtin/packages/openloops/sft1.coll
@@ -0,0 +1,26 @@
+ppll
+ppllj
+pplljj
+ppln
+pplnj
+pplnjj
+pptt
+ppttj
+ppttjj
+pptj
+pptjj
+ppjj
+ppjjj
+ppjjjj
+ppaa
+ppaaj
+ppaajj
+ppw
+ppwj
+ppwjj
+ppz
+ppzj
+ppzjj
+tbw
+eevvjj
+eett
diff --git a/var/spack/repos/builtin/packages/openloops/sft2.coll b/var/spack/repos/builtin/packages/openloops/sft2.coll
new file mode 100644
index 0000000000..b7d9b2423c
--- /dev/null
+++ b/var/spack/repos/builtin/packages/openloops/sft2.coll
@@ -0,0 +1,76 @@
+pptttt
+pptt
+pptt_ew
+ppttbb
+ppttj
+ppttj_ew
+pptw
+pptj
+pptjj
+ppjj
+ppjj_ew
+ppjjj
+ppjjj_ew
+ppatt
+ppattj
+pplltt
+ppllttj
+pplntt
+pplnttj
+ppwtt
+ppwttj
+ppllaa
+pplllla
+ppvvv
+ppvvv_ew
+ppvvvj
+ppaaj
+ppaaj2
+pplla
+pplla_ew
+ppllaj
+ppllll
+ppllll2
+ppllll2_onlyh
+ppllll_ew
+ppllllj
+ppllllj2
+ppllllj2_onlyh
+ppllnnjj_ew
+ppllnnjj_vbs
+pplnajj
+ppvv
+ppvv2
+ppvv_ew
+ppvvj
+ppvvj2
+ppvvj_ew
+ppajj
+ppajj_ew
+ppllj
+ppllj_ew
+pplljj
+pplljj_ew
+pplnjj
+pplnjj_ew
+ppnnjj_ew
+ppvj
+ppvj_ew
+ppwjj
+ppwjj_ew
+ppzjj
+ppzjj_ew
+pphtt
+pphtt_ew
+pphttj
+bbhj
+heftpphj
+heftpphjj
+pph2
+pphbb
+pphbbj
+pphj2
+pphjj_vbf
+pphjj_vbf_ew
+eett
+tbw
--
GitLab