Module flarefly.pdf_configs

Helper functions to get the PDF configurations for signal and background PDFs.

Functions

def get_bkg_pdf_config(pdf_kind: PDFKind)

Expand source code

def get_bkg_pdf_config(pdf_kind: PDFKind):
    """Get the configuration for a specific PDF."""
    if pdf_kind not in BACKGROUND_PDF_CONFIGS:
        Logger(f"Unknown background PDF name: {pdf_kind.value}.", "FATAL")

    # Get the specific config
    config = BACKGROUND_PDF_CONFIGS[pdf_kind].copy()

    return config

Get the configuration for a specific PDF.

def get_kde_pdf(kde_kind: PDFKind)

Expand source code

def get_kde_pdf(kde_kind: PDFKind):
    """Get the KDE PDF class for a specific KDE type."""
    if kde_kind not in KDE_MAP:
        Logger(
            f"Unknown KDE type: {kde_kind.value}. "
            f"Available types: {list(KDE_MAP.keys())}",
            "FATAL"
        )

    return KDE_MAP[kde_kind]

Get the KDE PDF class for a specific KDE type.

def get_signal_pdf_config(pdf_kind: PDFKind)

Expand source code

def get_signal_pdf_config(pdf_kind: PDFKind):
    """Get the configuration for a specific PDF."""
    if pdf_kind not in SIGNAL_PDF_CONFIGS:
        Logger(f"Unknown signal PDF name: {pdf_kind.value}.", "FATAL")

    # Get the specific config
    config = SIGNAL_PDF_CONFIGS[pdf_kind].copy()

    return config

Get the configuration for a specific PDF.