visualize_composite.py

from matplotlib.pyplot import imshow, show
from numpy import round

from get_data import get_features
from utils import *
from utils import typical_input
from visualize import get_bbox


def get_snow_composite(latitudes, longitudes, dfb_beginning, dfb_ending, slot_step):
    visible_features = get_features(
        "visible", latitudes, longitudes, dfb_beginning, dfb_ending, False, slot_step
    )
    vis = visible_features[:, :, :, 1]
    sir = visible_features[:, :, :, 0]
    mir = get_features(
        "infrared", latitudes, longitudes, dfb_beginning, dfb_ending, False, slot_step
    )[:, :, :, 1]
    mir -= 250
    mir /= 100
    (nb_slots, nb_latitudes, nb_longitudes) = np.shape(vis)
    vis[vis < 0] = 0
    sir[sir < 0] = 0
    mir[mir < 0] = 0
    composite = np.empty((nb_slots, nb_latitudes, nb_longitudes, 3))
    composite[:, :, :, 0] = vis
    composite[:, :, :, 1] = sir
    composite[:, :, :, 2] = mir
    return 255 * composite


def infrared_low_cloud_composite(
    latitudes, longitudes, dfb_beginning, dfb_ending, slot_step
):
    infrared_features = get_features(
        "infrared",
        latitudes,
        longitudes,
        dfb_beginning,
        dfb_ending,
        "channel",
        slot_step,
    )
    (nb_slots, nb_latitudes, nb_longitudes) = np.shape(infrared_features)[0:3]
    composite = np.zeros((nb_slots, nb_latitudes, nb_longitudes, 3))
    try:
        fir = infrared_features[:, :, :, 2]
        fir = round(2 * (fir - 200))
        composite[:, :, :, 0] = fir
    except IndexError:
        pass
    lir = infrared_features[:, :, :, 1]
    mir = infrared_features[:, :, :, 0]
    mir = round(2 * (mir - 200))
    lir = round(2 * (lir - 200))
    composite[:, :, :, 1] = lir
    composite[:, :, :, 2] = mir
    composite[(lir <= 0) | (lir > 250)] = 0
    return 255 * composite


if __name__ == "__main__":
    slot_step = 1
    (
        beginning,
        ending,
        latitude_beginning,
        latitude_end,
        longitude_beginning,
        longitude_end,
    ) = typical_input()
    latitudes, longitudes = get_latitudes_longitudes(
        latitude_beginning, latitude_end, longitude_beginning, longitude_end
    )
    date_begin, date_end = print_date_from_dfb(beginning, ending)
    composite = infrared_low_cloud_composite(
        latitudes, longitudes, beginning, ending, slot_step
    )

    bbox = get_bbox(
        latitude_beginning, latitude_end, longitude_beginning, longitude_end
    )

    start = 0
    for slot in range(start, np.shape(composite)[0]):
        if not np.all(composite[slot, :, :, 0] == 0):
            imshow(composite[slot])
            show()