Esempio n. 1
0
output_dir = '/home/philipp/final/plots/surf_diff'
basedir = '/home/philipp/final/'
file_extension = 'pdf'

figsize = (4.5, 3)

for case in [test_cases.Giluwe, test_cases.Borden]:
    filepaths = glob.glob(
        os.path.join(basedir, '{:s}/*/data_logger.pkl'.format(case.name)))
    filepaths = sorted(filepaths)
    for path in filepaths:
        idir, temp = os.path.split(path)
        gdir, exp = os.path.split(idir)
        dl = load_pickle(path)
        exp_name = experiment_naming_engine.get_experiment_name2(exp)
        if exp_name is not None and len(dl.step_indices) > 0:
            print(exp_name + ' ' + case.name)
            ice_mask = np.load(os.path.join(gdir, 'ref_ice_mask.npy'))

            diff_optimized_surf = dl.surfs[-1] - dl.ref_surf
            cbar_min = diff_optimized_surf.min()
            cbar_max = diff_optimized_surf.max()

            surf_noise = None
            if os.path.exists(os.path.join(idir, 'dem_noise.npy')):
                surf_noise = np.load(os.path.join(idir, 'dem_noise.npy'))
                cbar_min = min(surf_noise.min(), cbar_min)
                cbar_max = max(surf_noise.max(), cbar_max)

            cbar_min_max = max(abs(cbar_min), abs(cbar_max))
Esempio n. 2
0
                            os.path.join(basedir, case.name, exp,
                                         'dem_noise.npy'))[ice_mask])
                    surf_differences.append(surf_diff)
                    all_widths[0].append(0.5 * box_width)
                    all_widths[0].append(0.5 * box_width)
                    all_positions[0].append(position + 0.25 * box_width)
                    all_positions[0].append(position + 0.75 * box_width)
                    fill_colors[0].append(known_reg_parameters[reg_parameters])
                    fill_colors[0].append(known_reg_parameters[reg_parameters])
                else:
                    surf_differences.append(surf_diff)
                    all_widths[0].append(box_width)
                    all_positions[0].append(position + 0.5 * box_width)
                    fill_colors[0].append(known_reg_parameters[reg_parameters])

                labels.append(get_experiment_name2(exp))
                # all_widths[0].append(box_width)
                all_positions[1].append(position + 0.5 * box_width)
                position += small_spacer + box_width
            position += medium_spacer
        hline_positions.append(position + 0.5 *
                               (large_spacer - medium_spacer - small_spacer))
        position += large_spacer

    diff_collections = [surf_differences, bed_differences]

    flierprops = dict(marker='.',
                      markerfacecolor='blue',
                      markersize=5,
                      linestyle='none')
    fig, ax_arr = plt.subplots(nrows=len(diff_collections), figsize=(9, 13))