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)
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))