plot_DWave_results.py

""" plotter of results of computaiton on the DWave"""
import pickle
import matplotlib.pyplot as plt
from matplotlib import rc
from Qfile_solve import method_marker

rc('text',usetex=True)
rc('font', family='serif')


def grafitti(route):
    if route == 'enlarged':
        return '4 trains, 2 stations'
    elif route == '5trains':
        return '5 trains, 5 stations'
    elif route == None:
        return 'Default settings'
    elif route == 'rerouted':
        return 'Rerouted setting'

# annealing time = 250 μs
# number of runs = 3996

list_route_full = [["enlarged", "5trains"], [ None , "rerouted" ]]


for k, list_route in enumerate(list_route_full):
    fig, axes = plt.subplots(ncols=2)
    fig, ax = plt.subplots(1, 2, figsize=(5,3),sharey=True, sharex=True)

    for i, route in enumerate(list_route):
        non_feasible_css = [3, 3.5, 4, 4.5]
        non_feasible_def = { 'en':[], 'cs':[] }
        f = method_marker(route)
        for cs in non_feasible_css:
            with open(f"files/dwave_data/Qfile_samples_sol_real-anneal_numread3996_antime250_chainst{cs}" +f, 'rb') as fp:
                x = pickle.load(fp)

            non_feasible_def['en'].append(x[0][1])
            non_feasible_def['cs'].append(cs)

        feasible_ens = min(non_feasible_def['en'])
        feasible_index = non_feasible_def['en'].index(feasible_ens)
        feasible_css = non_feasible_def['cs'][feasible_index]
        non_feasible_def['en'].remove( feasible_ens )
        non_feasible_def['cs'].remove( feasible_css )

        with open('files/hybrid_data/Qfile_samples_sol_hybrid-anneal' +f, 'rb') as fp:
            hybrid = [pickle.load(fp)[0][1]]

        non_feasible_css_mod = non_feasible_def['cs']
        non_feasible_ens_mod = non_feasible_def['en']

        title = "annealing t = 250 μs, 3996 runs, default"

        if i ==0:
            ax[i].plot(non_feasible_css_mod, non_feasible_ens_mod, 'rs', label = 'D-Wave advantage non-feasible')
            if k == 0:
                ax[i].plot(feasible_css, feasible_ens, 'rs')
                ax[i].axhline(y =hybrid, color = 'g', label = 'D-Wave hybrid solver')
                if route == 'enlarged':
                    ax[i].axhline(y =-14.4, linestyle = '--', color = 'k',label = 'Ground energy')
                else:
                    ax[i].axhline(y =-21.49,  linestyle = '--',  color = 'k',label = 'Ground energy')
            elif k ==1:
                ax[i].plot(feasible_css, feasible_ens, 'go', label = 'D-Wave advantage feasible')
                ax[i].axhline(y =hybrid, color = 'g', label = 'Ground energy and D-Wave hybrid solver')
            ax[i].set_xticks(non_feasible_css)
            ax[i].set_title(f'{grafitti(route)}')

        else:
            ax[i].plot(non_feasible_css_mod, non_feasible_ens_mod, 'rs')
            if k == 0:
                ax[i].plot(feasible_css, feasible_ens, 'rs')
                ax[i].axhline(y =hybrid, color = 'g')
                if route == 'enlarged':
                    ax[i].axhline(y =-14.4, linestyle = '--', color = 'k')
                else:
                    ax[i].axhline(y =-21.49,  linestyle = '--',  color = 'k')
                ax[i].plot(feasible_css, feasible_ens, 'rs')
            elif k ==1:
                ax[i].plot(feasible_css, feasible_ens, 'go')

            ax[i].set_xticks(non_feasible_css)
            ax[i].set_title(f'{grafitti(route)}')
            ax[i].axhline(y =hybrid, color = 'g')
        ax[i].set_xlabel('Chain strength')


    fig.text(0.005, 0.42, 'Minimum energy', va='center', rotation='vertical')
    fig.legend(title = 'Annealing time = $250$ $\mu$$s$, $3996$ reads', loc='upper center',  borderaxespad=0.5, ncol=2, title_fontsize = 8, fontsize = 8)

    fig.tight_layout()
    fig.subplots_adjust(top=0.68)
    if k == 0:
        fig.savefig("plots/DW_trains_larger.pdf")
    elif k ==1:
        fig.savefig("plots/DW_trains.pdf")