plot_orf_density.py 文件源码

def plot_main(cds_start,cds_end,psites_array,orf_tstart,orf_tstop,outname):
    """
    the main plot function
    """
    plt.figure(figsize=(8,4))
    if cds_start is not None:
        gs = gridspec.GridSpec(3,1,height_ratios=[10,1,1],hspace=0.6,left=0.2,right=0.95)
    else:
        gs = gridspec.GridSpec(2,1,height_ratios=[11,1],hspace=0.6,left=0.2,right=0.95)

    ax1 = plt.subplot(gs[0])
    ax2 = plt.subplot(gs[1])
    plot_ORF(ax1,psites_array,orf_tstart)
    plot_annotation(ax2,psites_array.size,orf_tstart,orf_tstop,"Predicted","#3994FF")

    if cds_start is not None:
        ax3 = plt.subplot(gs[2])
        plot_annotation(ax3,psites_array.size,cds_start,cds_end,"Annotated","#006DD5")
    # plt.tight_layout()
    plt.savefig(outname + ".pdf")