powheg_V2.py

Go to the documentation of this file.

# Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
 
from .powheg_base import PowhegBase
import glob
 
class PowhegV2(PowhegBase):
    """! Base class for PowhegBox V2 processes.
 
    All V2 processes inherit from this class.
 
    @author James Robinson  <james.robinson@cern.ch>
    """
 
    def __init__(self, base_directory, executable_name, warning_output = [], info_output = [], error_output = [], **kwargs):
        """! Constructor.
 
        @param base_directory  path to PowhegBox code.
        @param executable_name folder containing appropriate PowhegBox executable.
        @param is_LO           True if this is a leading-order process.
        @param warning_output list of patterns which if found in the output will be treated as warning in the log.
        @param error_output list of patterns which if found in the output will be treated as error in the log.
        @param info_output list of patterns which if found in the output will be treated as info in the log.
        """
        super(PowhegV2, self).__init__(base_directory, "POWHEG-BOX-V2", executable_name=executable_name, warning_output=warning_output, info_output=info_output, error_output=error_output, **kwargs)
 
    @property
    def default_PDFs(self):
        """! Default PDFs for this process."""
        __PDF_list = list(range(260000, 260101)) # NNPDF30_nlo_as_0118 central with eigensets
        __PDF_list += [266000, 265000]      # NNPDF30_nlo_as_0119 and NNPDF30_nlo_as_0117
        __PDF_list += [303200]              # NNPDF30_nnlo_as_0118_hessian
        __PDF_list += [27400, 27100]        # MSHT20nnlo_as118, MSHT20nlo_as118
        __PDF_list += [14000, 14400]        # CT18NNLO, CT18NLO
        __PDF_list += [304400, 304200]      # NNPDF31_nnlo_as_0118_hessian, NNPDF31_nlo_as_0118_hessian
        __PDF_list += [331500, 331100]      # NNPDF40_nnlo_as_01180_hessian, NNPDF40_nlo_as_01180
        __PDF_list += [14200, 14300, 14100] # CT18ANNLO, CT18XNNLO and CT18ZNNLO
        __PDF_list += list(range(93300, 93343))  # PDF4LHC21_40_pdfas with eigensets
        return __PDF_list
 
    @property
    def default_scales(self):
        """! Default scale variations for this process."""
        return [[1.0, 1.0, 1.0, 0.5, 0.5, 2.0, 2.0],\
                [1.0, 0.5, 2.0, 0.5, 1.0, 1.0, 2.0]]
 
    @property
    def files_for_cleanup(self):
        """! Wildcarded list of files created by this process that can be deleted."""
        return [
            "bornequiv",
            "FlavRegList",
            "fort.18",
            "mint_upb_btildeupb.top",
            "mint_upb_remnupb.top",
            "pwg*.top",
            "pwgboundviolations*.dat",
            "pwgcounters*.dat",
            "pwgseeds.dat",
            "pwgubsigma.dat",
            "pwhg_checklimits",
            "realequiv*",
            "reweighting_input.xml",
            "virtequiv"
        ]
 
    @property
    def integration_file_names(self):
        """! Wildcarded list of integration files that might be created by this process."""
        """! All files matching these patterns will be included in the gridpack."""
        return [
            "pwg*xg*.dat",
            "pwg*upb*.dat",
            "pwggrid*.dat",
            "pwgfullgrid*.dat",
            "pwgubound*.dat",
            "pwg*stat.dat",
        ]
 
    @property
    def mandatory_integration_file_names(self):
        """! Wildcarded list of integration files that are needed for this process."""
        """! If some of the patterns don't match any files before running, a warning will be made to inform that no pre-made integration grid will be used."""
        return [
            "pwg*xg*.dat",
            #"pwg*upb*.dat", # not needed by ttj MiNNLO
            "pwggrid*.dat",
            #"pwgfullgrid*.dat", # not needed by ttj MiNNLO
            "pwgubound*.dat",
        ]
 
    @property
    def powheg_version(self):
        """! Version of PowhegBox process."""
        return "V2"
 
    def stage_is_completed(self, stage):
        """! Set whether the specified POWHEG-BOX generation stage is complete."""
        if stage == 1:
            required_files = ["pwg*xg*.dat"]
        elif stage == 2:
            required_files = ["pwg*upb*.dat", "pwggrid*.dat"]
        elif stage == 3:
            required_files = ["pwgfullgrid*.dat", "pwgubound*.dat"]
        else:
            return False
 
        # Check that required files have been found
        for required_file in required_files:
            if not glob.glob(required_file):
                return False
        return True