Ejemplo n.º 1
0
        mass_spectrum, choose_molecular_formula=False)

    mass_spectrum_by_classes.plot_ms_assigned_unassigned()

    plt.show()
    all_classes = 0
    for classe in mass_spectrum_by_classes.get_classes(threshold_perc=0,
                                                       isotopologue=True):

        mass_spectrum_by_classes.plot_dbe_vs_carbon_number(classe)
        #plt.show()

        mass_spectrum_by_classes.plot_ms_class(classe)
        plt.show()

        mass_spectrum_by_classes.plot_mz_error_class(classe)

        mz_calc_l = mass_spectrum_by_classes.mz_calc(classe)

        mf_l = mass_spectrum_by_classes.molecular_formula(classe)

        print(
            "# Name; m/z value; charge; ion formula; collision cross section [A^2]"
        )

        for index, mz_calc in enumerate(mz_calc_l):

            #print("%s : %s : %s : %.6f : 1+ : %s" %(classe, mf_l[index].class_label, mf_l[index].string, mz_calc, mf_l[index].string))

            print("%s %.6f 1+ %s" % (mf_l[index].string.replace(
                ' ', ''), mz_calc, mf_l[index].string.replace(' ', '')))
Ejemplo n.º 2
0
def create_plots(mass_spectrum, workflow_params, dirloc):

    ms_by_classes = HeteroatomsClassification(mass_spectrum,
                                              choose_molecular_formula=False)

    if workflow_params.plot_ms_assigned_unassigned:
        print("Plotting assigned vs. unassigned mass spectrum")
        ax_ms = ms_by_classes.plot_ms_assigned_unassigned()
        plt.savefig(dirloc / "assigned_unassigned.png", bbox_inches='tight')
        plt.clf()

    if workflow_params.plot_mz_error:
        print("Plotting mz_error")
        ax_ms = ms_by_classes.plot_mz_error()
        plt.savefig(dirloc / "mz_error.png", bbox_inches='tight')
        plt.clf()

    if workflow_params.plot_van_krevelen:
        van_krevelen_dirloc = dirloc / "van_krevelen"
        van_krevelen_dirloc.mkdir(exist_ok=True, parents=True)

    if workflow_params.plot_c_dbe:
        c_dbe_dirloc = dirloc / "dbe_vs_c"
        c_dbe_dirloc.mkdir(exist_ok=True, parents=True)

    if workflow_params.plot_ms_classes:
        ms_class_dirloc = dirloc / "ms_class"
        ms_class_dirloc.mkdir(exist_ok=True, parents=True)

    if workflow_params.plot_mz_error_classes:
        mz_error_class_dirloc = dirloc / "mz_error_class"
        mz_error_class_dirloc.mkdir(exist_ok=True, parents=True)

    pbar = tqdm(ms_by_classes.get_classes())

    for classe in pbar:

        pbar.set_description_str(
            desc="Plotting results for class {}".format(classe), refresh=True)

        if workflow_params.plot_van_krevelen:
            ax_c = ms_by_classes.plot_van_krevelen(classe)
            plt.savefig(van_krevelen_dirloc / "{}.png".format(classe),
                        bbox_inches='tight')
            plt.clf()

        if workflow_params.plot_mz_error_classes:
            ax_c = ms_by_classes.plot_mz_error_class(classe)
            plt.savefig(mz_error_class_dirloc / "{}.png".format(classe),
                        bbox_inches='tight')
            plt.clf()

        if workflow_params.plot_ms_classes:
            ax_c = ms_by_classes.plot_ms_class(classe)
            plt.savefig(ms_class_dirloc / "{}.png".format(classe),
                        bbox_inches='tight')
            plt.clf()

        if workflow_params.plot_c_dbe:
            ax_c = ms_by_classes.plot_dbe_vs_carbon_number(classe)
            plt.savefig(c_dbe_dirloc / "{}.png".format(classe),
                        bbox_inches='tight')
            plt.clf()