Source code for src.functions.SOLWEIGpython.Tgmaps_v1_cupy

import numpy as np
import cupy as cp


[docs]
def Tgmaps_v1(lc_grid, lc_class):
    "Function transferred to CuPy for populating grids with coefficients for Tg"
    #Tgmaps_v1 Populates grids with cooeficients for Tg wave
    #   Detailed explanation goes here

    id = cp.unique(lc_grid)
    TgK = cp.copy(lc_grid)
    Tstart = cp.copy(lc_grid)
    alb_grid = cp.copy(lc_grid)
    emis_grid = cp.copy(lc_grid)
    TmaxLST = cp.copy(lc_grid)

    for i in np.arange(0, id.__len__()):
        row = (lc_class[:, 0] == id[i])
        Tstart[Tstart == id[i]] = lc_class[row, 4]
        alb_grid[alb_grid == id[i]] = lc_class[row, 1]
        emis_grid[emis_grid == id[i]] = lc_class[row, 2]
        TmaxLST[TmaxLST == id[i]] = lc_class[row, 5]
        TgK[TgK == id[i]] = lc_class[row, 3]

    wall_pos = cp.where(lc_class[:, 0] == 99)
    TgK_wall = lc_class[wall_pos, 3]
    Tstart_wall = lc_class[wall_pos, 4]
    TmaxLST_wall = lc_class[wall_pos, 5]

    return TgK, Tstart, alb_grid, emis_grid, TgK_wall, Tstart_wall, TmaxLST, TmaxLST_wall