output_dir = '/home/philipp/HR_02/plots/bed_diff'
basedir = '/home/philipp/HR_02/'
file_extension = 'pdf'

figsize = (4.5, 3)

for case in [test_cases.Trikora]:  #, 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:
            ice_mask = np.load(os.path.join(gdir, 'ref_ice_mask.npy'))
            bed_measurements = None
            if exp_name.startswith('bed measurements'):
                bed_measurements = np.load(
                    os.path.join(idir, 'bed_measurements.pkl'))

            diff_first_guess = dl.first_guessed_bed - dl.true_bed
            diff_optimized = dl.beds[-1] - dl.true_bed
            cbar_min = min(diff_first_guess.min(), diff_optimized.min())
            cbar_max = max(diff_first_guess.max(), diff_optimized.max())
            cbar_min_max = max(abs(cbar_min), abs(cbar_max))
            norm = MidpointNormalize(midpoint=0.,
                                     vmin=-cbar_min_max,
                                     vmax=cbar_min_max)
Beispiel #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))