Merge lp:~maddevelopers/mg5amcnlo/heft_fix into lp:~maddevelopers/mg5amcnlo/combine_v0_6_0

[Johan Alwall]

Included special calls for HEFT VVVS interaction in the FortranHelasCallWriter. Note that the new version of HEFT must be used (v. 1.5.0). g g > h + Ng, g g > h3 + Ng with N=1,2,3 have been tested agains regular V4, new V4, new V5, and UFO models.

[lp:~maddevelopers/mg5amcnlo/heft_fix]

59. By Johan Alwall on 2010-12-07

Added acceptance test for v4 HEFT

[Olivier Mattelaer (olivier-mattelaer)]

It's look perfect but:

======================================================================
FAIL: test_v4_heft (tests.acceptance_tests.test_cmd.TestCmdShell2)
Test the import of models and the export of Helas Routine
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Users/omatt/Documents/Eclipse2/heft/tests/acceptance_tests/test_cmd.py", line 367, in test_v4_heft
    'check')))
AssertionError: False is not True

review: Needs Fixing

[Johan Alwall (johan-alwall)]

I can't reproduce this. Did you run with the latest CVS version of
MG/ME (or, equivalently, by running bin/create_release.py)?

> ======================================================================
> FAIL: test_v4_heft (tests.acceptance_tests.test_cmd.TestCmdShell2)
> Test the import of models and the export of Helas Routine
> ----------------------------------------------------------------------
> Traceback (most recent call last):
>  File "/Users/omatt/Documents/Eclipse2/heft/tests/acceptance_tests/test_cmd.py", line 367, in test_v4_heft
>    'check')))
> AssertionError: False is not True

[Olivier Mattelaer (olivier-mattelaer)]

Sorry for that indeed, my cvs was not up-to-date for this.

review: Approve

=== modified file 'madgraph/iolibs/helas_call_writers.py'
--- madgraph/iolibs/helas_call_writers.py	2010-10-28 00:17:50 +0000
+++ madgraph/iolibs/helas_call_writers.py	2010-12-07 22:48:10 +0000
@@ -211,7 +211,7 @@
     self_dict = {1: 'H', 2: 'F', -2: 'F', 3: 'J', 5: 'U'}
     # Dictionaries used for sorting the letters in the Helas call
     sort_wf = {'O': 0, 'I': 1, 'S': 2, 'T': 3, 'V': 4}
-    sort_amp = {'S': 1, 'V': 2, 'T': 0, 'O': 3, 'I': 4}
+    sort_amp = {'S': 0, 'V': 2, 'T': 1, 'O': 3, 'I': 4}
 
     def default_setup(self):
         """Set up special Helas calls (wavefunctions and amplitudes)
@@ -225,35 +225,29 @@
         # Gluon 4-vertex division tensor calls ggT for the FR sm and mssm
 
         key = ((3, 3, 5, 3), 'A')
-
         call = lambda wf: \
                "CALL UVVAXX(W(1,%d),W(1,%d),%s,zero,zero,zero,W(1,%d))" % \
                (wf.get('mothers')[0].get('number'),
                 wf.get('mothers')[1].get('number'),
-
                 wf.get('coupling'),
                 wf.get('number'))
         self.add_wavefunction(key, call)
 
         key = ((3, 5, 3, 1), 'A')
-
         call = lambda wf: \
                "CALL JVTAXX(W(1,%d),W(1,%d),%s,zero,zero,W(1,%d))" % \
                (wf.get('mothers')[0].get('number'),
                 wf.get('mothers')[1].get('number'),
-
                 wf.get('coupling'),
                 wf.get('number'))
         self.add_wavefunction(key, call)
 
         key = ((3, 3, 5), 'A')
-
         call = lambda amp: \
                "CALL VVTAXX(W(1,%d),W(1,%d),W(1,%d),%s,zero,AMP(%d))" % \
                (amp.get('mothers')[0].get('number'),
                 amp.get('mothers')[1].get('number'),
                 amp.get('mothers')[2].get('number'),
-
                 amp.get('coupling'),
                 amp.get('number'))
         self.add_amplitude(key, call)
@@ -318,6 +312,43 @@
                 amp.get('number'))
         self.add_amplitude(key, call)
 
+        # HEFT VVVS calls
+
+        key = ((1, 3, 3, 3, 3), '')
+        call = lambda wf: \
+               "CALL JVVSXX(W(1,%d),W(1,%d),W(1,%d),DUM1,%s,%s,%s,W(1,%d))" % \
+               (wf.get('mothers')[0].get('number'),
+                wf.get('mothers')[1].get('number'),
+                wf.get('mothers')[2].get('number'),
+                wf.get('coupling'),
+                wf.get('mass'),
+                wf.get('width'),
+                wf.get('number'))
+        self.add_wavefunction(key, call)
+
+        key = ((3, 3, 3, 1, 4), '')
+        call = lambda wf: \
+               "CALL HVVVXX(W(1,%d),W(1,%d),W(1,%d),DUM1,%s,%s,%s,W(1,%d))" % \
+               (wf.get('mothers')[0].get('number'),
+                wf.get('mothers')[1].get('number'),
+                wf.get('mothers')[2].get('number'),
+                wf.get('coupling'),
+                wf.get('mass'),
+                wf.get('width'),
+                wf.get('number'))
+        self.add_wavefunction(key, call)
+
+        key = ((1, 3, 3, 3), '')
+        call = lambda amp: \
+               "CALL VVVSXX(W(1,%d),W(1,%d),W(1,%d),W(1,%d),DUM1,%s,AMP(%d))" % \
+               (amp.get('mothers')[0].get('number'),
+                amp.get('mothers')[1].get('number'),
+                amp.get('mothers')[2].get('number'),
+                amp.get('mothers')[3].get('number'),
+                amp.get('coupling'),
+                amp.get('number'))
+        self.add_amplitude(key, call)
+
     def get_wavefunction_call(self, wavefunction):
         """Return the function for writing the wavefunction
         corresponding to the key. If the function doesn't exist,
 
=== modified file 'madgraph/iolibs/import_v4.py'
--- madgraph/iolibs/import_v4.py	2010-12-07 00:44:02 +0000
+++ madgraph/iolibs/import_v4.py	2010-12-07 22:48:10 +0000
@@ -334,6 +334,10 @@
                         # Should only use one Helas amplitude for electroweak
                         # 4-vector vertices with FR. I choose W3W3NX.
                         myinter.set('lorentz', ['WWVVN'])
+                elif values[len(part_list)] in ['DUM', 'DUM0', 'DUM1']:
+                    # We can just use the second coupling, since the first
+                    # is a dummy
+                    myinter.set('couplings', {(0, 0):values[len(part_list)+1]})
                 elif pdg_codes == [21, 21, 21, 21]:
                     # gggg
                     myinter.set('couplings', {(0, 0):values[len(part_list)],
 
=== modified file 'tests/acceptance_tests/test_cmd.py'
--- tests/acceptance_tests/test_cmd.py	2010-11-24 20:48:25 +0000
+++ tests/acceptance_tests/test_cmd.py	2010-12-07 22:48:10 +0000
@@ -338,6 +338,47 @@
         self.assertTrue(me_groups)
         self.assertAlmostEqual(float(me_groups.group('value')), 1.953735e-2)
         
+    def test_v4_heft(self):
+        """Test the import of models and the export of Helas Routine """
+
+        if os.path.isdir(self.out_dir):
+            shutil.rmdir(self.out_dir)
+
+        self.do('import model_v4 heft')
+        self.do('generate g g > h g g')
+        self.do('output standalone %s ' % self.out_dir)
+
+        devnull = open(os.devnull,'w')
+        # Check that the Model and Aloha output compile
+        subprocess.call(['make'],
+                        stdout=devnull, stderr=devnull, 
+                        cwd=os.path.join(self.out_dir, 'Source'))
+        self.assertTrue(os.path.exists(os.path.join(self.out_dir,
+                                               'lib', 'libdhelas3.a')))
+        self.assertTrue(os.path.exists(os.path.join(self.out_dir,
+                                               'lib', 'libmodel.a')))
+        # Check that check_sa.f compiles
+        subprocess.call(['make', 'check'],
+                        stdout=devnull, stderr=devnull, 
+                        cwd=os.path.join(self.out_dir, 'SubProcesses',
+                                         'P0_gg_hgg'))
+        self.assertTrue(os.path.exists(os.path.join(self.out_dir,
+                                                    'SubProcesses', 'P0_gg_hgg',
+                                                    'check')))
+        # Check that the output of check is correct 
+        logfile = os.path.join(self.out_dir,'SubProcesses', 'P0_gg_hgg',
+                               'check.log')
+        subprocess.call('./check', 
+                        stdout=open(logfile, 'w'), stderr=devnull,
+                        cwd=os.path.join(self.out_dir, 'SubProcesses',
+                                         'P0_gg_hgg'), shell=True)
+        log_output = open(logfile, 'r').read()
+        me_re = re.compile('Matrix element\s*=\s*(?P<value>[\d\.eE\+-]+)\s*GeV',
+                           re.IGNORECASE)
+        me_groups = me_re.search(log_output)
+        self.assertTrue(me_groups)
+        self.assertAlmostEqual(float(me_groups.group('value')), 1.10908942e-06)
+        
     def test_madevent_ufo_aloha(self):
         """Test MadEvent output with UFO/ALOHA"""
 
 
=== modified file 'tests/parallel_tests/sample_script.py'
--- tests/parallel_tests/sample_script.py	2010-12-05 05:38:08 +0000
+++ tests/parallel_tests/sample_script.py	2010-12-07 22:48:10 +0000
@@ -53,12 +53,19 @@
                     'e+ x1+ > e+ h1 x1+','e+ x1+ > e+ h2 x1+','e+ x1+ > e+ h3 x1+']
     my_proc_list += ['el+ h2 > el+ w+ w-']
                    
-    my_proc_list += me_comparator.create_proc_list(['g', 'go'], initial=2,
-                                                  final=2)
-
-    # Create a MERunner object for MG4
-    #my_mg4 = me_comparator.MG4Runner()
-    #my_mg4.setup(mg4_path)
+    #my_proc_list += me_comparator.create_proc_list(['g', 'go'], initial=2,
+    #                                              final=2)
+    #my_proc_list = me_comparator.create_proc_list(['g', 'h', 'h3'], initial=2,
+    #                                               final=4)
+
+    my_proc_list = ['h > g g g g']
+    # Create a MERunner object for MG4
+    my_mg4 = me_comparator.MG4Runner()
+    my_mg4.setup(mg4_path)
+
+    # Create a MERunner object for MG4
+    my_mg4_2 = me_comparator.MG4Runner()
+    my_mg4_2.setup(mg4_path)
 
     # Create a MERunner object for MG5
     my_mg5 = me_comparator.MG5Runner()
@@ -69,21 +76,21 @@
     my_mg5_ufo.setup(mg5_path, mg4_path)
 
     # Create a MERunner object for C++
-    my_mg5_cpp = me_comparator.MG5_CPP_Runner()
-    my_mg5_cpp.setup(mg5_path, mg4_path)
+    #my_mg5_cpp = me_comparator.MG5_CPP_Runner()
+    #my_mg5_cpp.setup(mg5_path, mg4_path)
 
     # Create and setup a comparator
     my_comp = me_comparator.MEComparator()
-    my_comp.set_me_runners(my_mg5_cpp, my_mg5_ufo, my_mg5)
+    my_comp.set_me_runners(my_mg5, my_mg5_ufo, my_mg4, my_mg4_2)
 
     # Run the actual comparison
     my_comp.run_comparison(my_proc_list,
-                       model='mssm', orders={'QED':4, 'QCD':4, 'HIG':1, 'HIW':1}, energy=2000)
+                       model=['heft_fix', 'heft', 'heft', 'heft_fix'], orders={'QED':0, 'QCD':3, 'HIG':1, 'HIW':1}, energy=2000)
 
     # Do some cleanup
     #my_comp.cleanup()
 
-    filename = "mssm_results.log"
+    filename = "heft_results.log"
 
     # Print the output
     my_comp.output_result(filename=filename)