Skip to content

brahmap.core.separate_map_vectors

Source code in brahmap/core/GLS.py 
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
def separate_map_vectors(
    map_vector: np.ndarray, processed_samples: ProcessTimeSamples
) -> np.ndarray:
    try:
        map_vector = np.reshape(
            map_vector,
            shape=(processed_samples.solver_type, processed_samples.new_npix),
            order="F",
        )
    except TypeError:
        # `newshape` parameter has been deprecated since numpy 2.1.0. This part should be removed once the support is dropped for lower version
        map_vector = np.reshape(
            map_vector,
            newshape=(processed_samples.solver_type, processed_samples.new_npix),
            order="F",
        )

    output_maps = np.ma.MaskedArray(
        data=np.empty(processed_samples.npix, dtype=processed_samples.dtype_float),
        mask=~processed_samples.pixel_flag,
        fill_value=-1.6375e30,
    )

    output_maps = np.tile(A=output_maps, reps=(processed_samples.solver_type, 1))

    for idx in range(processed_samples.solver_type):
        output_maps[idx][~output_maps[idx].mask] = map_vector[idx]

    return output_maps