Flavor Sensitivity to new Physics

FlavorSensitivity.py

import matplotlib.pyplot as plt
import numpy
from wg1template.histogram_plots import create_solo_figure, add_descriptions_to_plot
from wg1template.plot_style import TangoColors
from wg1template.plot_utilities import export
from wg1template.point_plots import DataVariable, DataPointsPlot


def cpv_and_mixing(k, h_np=1.):
    # deltaF = 2
    return numpy.sqrt(k**2 / h_np)


def rare_decays(k, h_np=1.):
    # deltaF = 1
    return numpy.sqrt(k / h_np)


def direct_search(k, h_np=1.):
    return h_np * numpy.ones(len(k))


if __name__ == '__main__':
    form_factors = DataVariable(r"$\kappa$", r"", r'$\Lambda_\mathrm{NP}$', "")

    with plt.xkcd():
        dp = DataPointsPlot(form_factors)
        fig, ax = create_solo_figure(figsize=(5, 3), dpi=100)

        kappa = numpy.linspace(0, 1.5, num=100)
        h_np = 0.7
        ax.plot(kappa, direct_search(kappa, h_np), color=TangoColors.scarlet_red, ls='-',
                label=r'Direct Searches')
        ax.plot(kappa, rare_decays(kappa, h_np), color=TangoColors.sky_blue, ls=':',
                label=r'Rare Decays ($\Delta F = 1$)')
        ax.plot(kappa, cpv_and_mixing(kappa, h_np), color=TangoColors.orange, ls='--',
                label=r'CPV & mixing ($\Delta F = 2$)')

        dp.plot_on(ax, draw_legend=True, legend_kwargs={'ncol': 1, }, legend_inside=True)

        add_descriptions_to_plot(
            ax,
            experiment=r'',
            luminosity=r'',
            additional_info=''
        )
        ax.set_xticklabels([])
        ax.set_yticklabels([])
        ax.set_xticks([])
        ax.set_yticks([])

        plt.show()
        export(fig, 'flavour-sensitivity', '.')
        plt.close()