def main():
    args = parse_args()

    tags = ['staplestates{}'.format(i) for i in range(1, args.stapletypes + 1)]
    filebase = '{}/{}-{}'.format(args.input_dir, args.system, args.vari)
    out_filebase = '{}/{}-{}_staplestates-means'.format(args.output_dir,
            args.system, args.vari)
    figsize = (plot.cm_to_inches(18), plot.cm_to_inches(12))

    plot.set_default_appearance()
    f = plt.figure(figsize=figsize, dpi=300)
    gs = gridspec.GridSpec(1, 2, width_ratios=[10, 1], height_ratios=[1])
    ax = create_axis(f, gs, args.system)
    aves, stds = plot.read_expectations(filebase)
    if args.rtag:
        aves = aves[aves[args.rtag] == args.rvalue]
        stds = stds[stds[args.rtag] == args.rvalue]

    for j, tag in enumerate(tags):
        xvars = aves[args.xtag]
        ax.errorbar(xvars, aves[tag], yerr=stds[tag], marker='o')

        # Write to file
        plt.tight_layout(pad=0.5, h_pad=0, w_pad=0)
        f.savefig(out_filebase + '.png', transparent=True)
        f.savefig(out_filebase + '.pdf', transparent=True)
def main():
    args = parse_args()
    inp_filebase = '{}/{}-{}'.format(args.input_dir, args.system, args.vari)
    out_filebase = '{}/{}-{}_staplestates-melting'.format(args.output_dir,
            args.system, args.vari)
    index_to_stapletype = np.loadtxt(args.mapfile, dtype=int)
    figsize = (plot.cm_to_inches(18), plot.cm_to_inches(12))

    plot.set_default_appearance()
    f = plt.figure(figsize=figsize, dpi=300)
    gs = gridspec.GridSpec(1, 1, width_ratios=[1], height_ratios=[1])
    gs_main = gridspec.GridSpecFromSubplotSpec(1, 1, subplot_spec=gs[:, :])
    ax = f.add_subplot(gs_main[0])
    ax.axis('off')

    aves, stds = plot.read_expectations(inp_filebase)
    if args.rtag:
        aves = aves[aves[args.rtag] == args.rvalue]
        stds = stds[stds[args.rtag] == args.rvalue]

    temps = aves.temp

    melting_points = estimate_melting_points(args.stapletypes, aves, temps)
    melting_array = fill_assembled_shape_array(melting_points, index_to_stapletype)

    # Plot simulation melting points
    min_t = np.min(melting_array)
    max_t = np.max(melting_array)
    im = ax.imshow(melting_array, vmin=min_t, vmax=max_t)
    plt.colorbar(im, orientation='horizontal')
    plt.tight_layout(pad=0.5, h_pad=0, w_pad=0)
    f.savefig(out_filebase + '.png', transparent=True)
    f.savefig(out_filebase + '.pdf', transparent=True)
示例#3
0
def main():
    args = parse_args()

    tags = ['numstaples', 'numfulldomains', 'nummisdomains', 'numstackedpairs']
    yaxis_labels = [
        '(Mis)bound staples', 'Bound domain pairs', 'Misbound domain pairs',
        'Stacked pairs'
    ]
    figsize = (plot.cm_to_inches(14), plot.cm_to_inches(11))

    plot.set_default_appearance()
    f = plt.figure(figsize=figsize, dpi=300)
    gs = gridspec.GridSpec(3,
                           2,
                           width_ratios=[1, 1],
                           height_ratios=[10, 10, 1])
    gs_main = gridspec.GridSpecFromSubplotSpec(2,
                                               2,
                                               subplot_spec=gs[:2, :],
                                               wspace=0.3,
                                               hspace=0.3)
    axes = create_axes(f, gs_main, yaxis_labels)

    all_assembled_values = []
    for assembled_values in args.all_assembled_values:
        parsed_values = []
        for assembled_value in assembled_values.split(','):
            parsed_values.append(int(assembled_value))

        all_assembled_values.append(parsed_values)

    for assembled_values in all_assembled_values:
        for i, assembled_value in enumerate(assembled_values):
            ax = axes[i]
            ax.axhline(assembled_value, linestyle='--', color='0.8')

    for system, vari in zip(args.systems, args.varis):
        sim_filebases = '{}/{}-{}'.format(args.input_dir, system, vari)
        all_aves, all_stds = plot.read_expectations(sim_filebases)
        xvars = all_aves[args.xtag]
        for i, tag in enumerate(tags):
            means = all_aves[tag]
            stds = all_stds[tag]
            ax = axes[i]
            ax.errorbar(xvars, means, yerr=stds, marker='o', label=vari)

    # Plot legend
    gs_lgd = gridspec.GridSpecFromSubplotSpec(1, 1, subplot_spec=gs[2, :])
    ax = f.add_subplot(gs_lgd[0])
    ax.set_axis_off()
    handles, labels = axes[0].get_legend_handles_labels()
    ax.legend(handles, labels, loc='center', frameon=False, ncol=1)

    gs.tight_layout(f, pad=1.0, h_pad=0, w_pad=0)
    f.savefig(args.output_filebase + '.png', transparent=True)
    f.savefig(args.output_filebase + '.pdf', transparent=True)
def main():
    args = parse_args()

    op_tag = 'numfulldomains'
    tagbase = 'domainstate'
    tags = ['{}{}'.format(tagbase, i) for i in range(args.scaffolddomains)]
    inp_filebase = '{}/{}-{}-{}'.format(args.input_dir, args.system, args.vari,
                                        op_tag)
    index_to_domaintype = np.loadtxt(args.mapfile, dtype=int)
    figsize = (plot.cm_to_inches(18), plot.cm_to_inches(12))
    for op_value in range(1, args.scaffolddomains + 1):
        out_filebase = '{}/{}-{}_{}-{}-{}-freqs'.format(
            args.output_dir, args.system, args.vari, op_tag, op_value, tagbase)

        plot.set_default_appearance()
        f = plt.figure(figsize=figsize, dpi=300)
        gs = gridspec.GridSpec(1, 1, width_ratios=[1], height_ratios=[1])
        gs_main = gridspec.GridSpecFromSubplotSpec(1, 1, subplot_spec=gs[:, :])
        ax = f.add_subplot(gs_main[0])
        ax.axis('off')

        aves, stds = plot.read_expectations(inp_filebase)
        if not op_value in aves[op_tag].values:
            continue

        reduced_aves = aves[aves[op_tag] == op_value]
        freqs = [reduced_aves[t] for t in tags]
        freq_array = fill_assembled_shape_array(freqs, index_to_domaintype)

        # Plot simulation melting points
        min_t = np.min(0)
        max_t = np.max(1)
        im = ax.imshow(freq_array, vmin=min_t, vmax=max_t)
        plt.colorbar(im, orientation='horizontal')
        plt.tight_layout(pad=0.5, h_pad=0, w_pad=0)
        f.savefig(out_filebase + '.png', transparent=True)
        f.savefig(out_filebase + '.pdf', transparent=True)
示例#5
0
def main():
    args = parse_args()

    tags = ['numstaples', 'numfulldomains', 'nummisdomains', 'numstackedpairs']
    yaxis_labels = [
        '(Mis)bound staples', 'Bound domain pairs', 'Misbound domain pairs',
        'Stacked pairs'
    ]
    figsize = (plot.cm_to_inches(14), plot.cm_to_inches(11))

    plot.set_default_appearance()
    f = plt.figure(figsize=figsize, dpi=300)
    gs = gridspec.GridSpec(3,
                           2,
                           width_ratios=[1, 1],
                           height_ratios=[10, 10, 1])
    gs_main = gridspec.GridSpecFromSubplotSpec(2,
                                               2,
                                               subplot_spec=gs[:2, :],
                                               wspace=0.3,
                                               hspace=0.3)
    gs_cb = gridspec.GridSpecFromSubplotSpec(1, 1, subplot_spec=gs[2, :])
    axes = create_axes(f, gs_main, yaxis_labels)
    norm = mpl.colors.Normalize(vmin=args.smults[0], vmax=args.smults[-1])
    cmap = mpl.cm.viridis
    scalarmap = mpl.cm.ScalarMappable(norm=norm, cmap=cmap)

    all_assembled_values = []
    for assembled_values in args.all_assembled_values:
        parsed_values = []
        for assembled_value in assembled_values.split(','):
            parsed_values.append(int(assembled_value))

        all_assembled_values.append(parsed_values)

    for assembled_values in all_assembled_values:
        for i, assembled_value in enumerate(assembled_values):
            ax = axes[i]
            ax.axhline(assembled_value, linestyle='--', color='0.8')

    for system, vari in zip(args.systems, args.varis):
        sim_filebases = '{}/{}-{}'.format(args.input_dir, system, vari)
        all_aves, all_stds = plot.read_expectations(sim_filebases)
        for smult in args.smults:
            reduced_aves = all_aves[all_aves['bias'] == smult]
            reduced_stds = all_stds[all_stds['bias'] == smult]
            xvars = reduced_aves['temp']
            for i, tag in enumerate(tags):
                means = reduced_aves[tag]
                stds = reduced_stds[tag]
                ax = axes[i]
                ax.errorbar(xvars,
                            means,
                            yerr=stds,
                            marker='o',
                            label=vari,
                            color=scalarmap.to_rgba(smult))

    # Plot legend
    ax = f.add_subplot(gs_cb[0])
    cbar = mpl.colorbar.ColorbarBase(ax,
                                     orientation='horizontal',
                                     cmap=cmap,
                                     norm=norm)

    gs.tight_layout(f, pad=1.0, h_pad=0, w_pad=0)
    f.savefig(args.output_filebase + '.png', transparent=True)
    f.savefig(args.output_filebase + '.pdf', transparent=True)