compute_figure.py 文件源码

def plot_figure_video_structure(structures, structure_combined, structure_optimized, rigidity_refined):
    # Figure 92
    PTH='./figure_structure/'
    if not os.path.isdir(PTH):
        os.makedirs(PTH)


    structure_min = np.percentile(structure_optimized[rigidity_refined==1].ravel(), 2)
    structure_max = np.percentile(structure_optimized[rigidity_refined==1].ravel(), 98)

    Is_fwd = structure2image(structures[1], rigidity_refined,
                             structure_min=structure_min,
                             structure_max=structure_max)
    Is_comb = structure2image(structure_combined, rigidity_refined,
                             structure_min=structure_min,
                             structure_max=structure_max)
    Is_opt = structure2image(structure_optimized, rigidity_refined,
                             structure_min=structure_min,
                             structure_max=structure_max)

    io.imsave(PTH+'structure_fwd.png', Is_fwd)
    io.imsave(PTH+'structure_comb.png', Is_comb)
    io.imsave(PTH+'structure_opt.png', Is_opt)