Commit 42530eef authored by Andreas Schmidt's avatar Andreas Schmidt
Browse files

Exporting is now part of multi_correlation.

parent 3819c1bd

xlap/analyse/correlation.py

+4 −5
Original line number Diff line number Diff line
@@ -27,7 +27,7 @@ def corr_multi(dfs, duration, **kwargs): 
    kwargs["ax"].set_xlabel("{} [us]".format(duration))





def multi_correlation(dfs, config):

def multi_correlation(dfs, config, export=False, file_name="MultiCorrelation.pdf"):

    durations = [x + "_D" for x in extract_durations(config)]

    durations.remove("EndToEnd_D")
@@ -46,11 +46,10 @@ def multi_correlation(dfs, config): 
        corr_multi(dfs, duration, grid=True, ax=ax)



    plt.subplots_adjust(wspace=0.3,hspace=0.3)



def correlation(df, config, export=False, file_name="Correlation.pdf"):

    res = multi_correlation([df], config)

    if export and file_name is not None:

        fig.savefig(file_name)

    plt.tight_layout()

    plt.show()

    return res
 


def correlation(df, config, export=False, file_name="SingleCorrelation.pdf"):

    return multi_correlation([df], config, export, file_name)