Exemple #1
0
def test_plot_Nieuwstadt(sim_data):
    """
    plot Nieuwstadt timeseries
    """
    # make directory
    localpath = Path.cwd()
    (localpath / "plots/output/Nieuwstadt/all_variables/").mkdir(parents=True, exist_ok=True)
    les_data_path = localpath / "les_data/Nieuwstadt.nc"
    if not les_data_path.is_file():
        url_ = r"https://drive.google.com/uc?export=download&id=1YLk80OuZSI0o-jeCxXgSBAF2PfSK6SRY"
        os.system(f"curl -sLo {les_data_path} '{url_}'")
    les_data = Dataset(les_data_path, 'r')

    f1 = "plots/output/Nieuwstadt/"
    f2 = f1 + "all_variables/"
    cn = "Nieuwstadt_"
    t0 = 6
    t1 = 8
    zmin = 0.0
    zmax = 2.5
    cb_min = [0., 0.]
    cb_max = [0.01, 2.5]
    fixed_cbar = True
    cb_min_t = [300, 300, 300, 0, 0, 0, 0, 0, 0, 0, 0, 2, 2, 4.5,\
                -0.4, 0, -0.7, -0.45,\
                0, -0.03, 0,\
                -0.045, -0.06, 0,\
                0, 0, -0.04]
    cb_max_t = [305, 305, 302, 1, 1, 1, 1, 1, 100, 1, 100, 5, 5, 5,\
                0, 2.4, 0.7, 0.45,\
                0.28, 0.01, 1.6,\
                0.06, 0.045, 0.1,\
                0.01, 0.1, 0.08]

    scm_dict = cmn.read_scm_data(sim_data)
    les_dict = cmn.read_les_data(les_data)

    scm_dict_t = cmn.read_scm_data_timeseries(sim_data)
    les_dict_t = cmn.read_les_data_timeseries(les_data)

    pls.plot_closures(scm_dict, les_dict, t0, t1, zmin, zmax, cn+"closures.pdf", folder=f1)
    pls.plot_spec_hum(scm_dict, les_dict, t0, t1, zmin, zmax,  cn+"humidities.pdf", folder=f1)
    pls.plot_upd_prop(scm_dict, les_dict, t0, t1, zmin, zmax,  cn+"updraft_properties.pdf", folder=f1)
    pls.plot_fluxes(scm_dict, les_dict, t0, t1, zmin, zmax,  cn+"mean_fluxes.pdf", folder=f1)
    pls.plot_tke_comp(scm_dict, les_dict, t0, t1, zmin, zmax,  cn+"tke_components.pdf", folder=f1)

    pls.plot_cvar_mean(scm_dict, les_dict, t0, t1, zmin, zmax,  cn+"var_covar_mean.pdf", folder=f2)
    pls.plot_cvar_comp(scm_dict, t0, t1, zmin, zmax,  cn+"var_covar_components.pdf", folder=f2)
    pls.plot_tke_break(scm_dict, les_dict, t0, t1, zmin, zmax, cn+"tke_breakdown.pdf",folder=f2)

    pls.plot_contour_t(scm_dict, les_dict, fixed_cbar, cb_min_t, cb_max_t, zmin, zmax, folder=f2)
    pls.plot_mean_prof(scm_dict, les_dict, t0, t1,  zmin, zmax, folder=f2)

    pls.plot_main(scm_dict_t, les_dict_t, scm_dict, les_dict,
                  cn+"main_timeseries.pdf", cb_min, cb_max, zmin, zmax, folder=f1)

    pls.plot_1D(scm_dict_t, les_dict_t, cn, folder=f2)
def test_plot_Nieuwstadt(sim_data):
    """
    plot Nieuwstadt timeseries
    """
    # make directory
    localpath = os.getcwd()
    try:
        os.mkdir(localpath + "/plots/output/Nieuwstadt/")
    except:
        print('Nieuwstadt folder exists')
    try:
        os.mkdir(localpath + "/plots/output/Nieuwstadt/all_variables/")
    except:
        print('Nieuwstadt/all_variables folder exists')

    if (os.path.exists(localpath + "/les_data/Nieuwstadt.nc")):
        les_data = Dataset(localpath + "/les_data/Nieuwstadt.nc", 'r')
    else:
        url_ = "https://www.dropbox.com/s/wkfy1mcbbo9iyx7/Nieuwstadt.nc?dl=0"
        os.system("wget -O " + localpath + "/les_data/Nieuwstadt.nc " + url_)
        les_data = Dataset(localpath + "/les_data/Nieuwstadt.nc", 'r')

    f1 = "plots/output/Nieuwstadt/"
    f2 = f1 + "all_variables/"
    cn = "Nieuwstadt_"
    t0 = 6
    t1 = 8
    zmin = 0.0
    zmax = 2.5
    cb_min = [0., 0.]
    cb_max = [0.01, 2.5]
    fixed_cbar = True
    cb_min_t = [300, 300, 300, 0, 0, 0, 0, 0, 0, 0, 0, 2, 2, 4.5,\
                -0.4, 0, -0.7, -0.45,\
                0, -0.03, 0,\
                -0.045, -0.06, 0,\
                0, 0, -0.04]
    cb_max_t = [305, 305, 302, 1, 1, 1, 1, 1, 100, 1, 100, 5, 5, 5,\
                0, 2.4, 0.7, 0.45,\
                0.28, 0.01, 1.6,\
                0.06, 0.045, 0.1,\
                0.01, 0.1, 0.08]

    scm_dict = cmn.read_scm_data(sim_data)
    les_dict = cmn.read_les_data(les_data)

    scm_dict_t = cmn.read_scm_data_timeseries(sim_data)
    les_dict_t = cmn.read_les_data_timeseries(les_data)

    pls.plot_closures(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "closures.pdf",
                      folder=f1)
    pls.plot_spec_hum(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "humidities.pdf",
                      folder=f1)
    pls.plot_upd_prop(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "updraft_properties.pdf",
                      folder=f1)
    pls.plot_fluxes(scm_dict,
                    les_dict,
                    t0,
                    t1,
                    zmin,
                    zmax,
                    cn + "mean_fluxes.pdf",
                    folder=f1)
    pls.plot_tke_comp(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "tke_components.pdf",
                      folder=f1)

    pls.plot_cvar_mean(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_mean.pdf",
                       folder=f2)
    pls.plot_cvar_comp(scm_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_components.pdf",
                       folder=f2)
    pls.plot_tke_break(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "tke_breakdown.pdf",
                       folder=f2)

    pls.plot_contour_t(scm_dict,
                       les_dict,
                       fixed_cbar,
                       cb_min_t,
                       cb_max_t,
                       zmin,
                       zmax,
                       folder=f2)
    pls.plot_mean_prof(scm_dict, les_dict, t0, t1, zmin, zmax, folder=f2)

    pls.plot_main(scm_dict_t,
                  les_dict_t,
                  scm_dict,
                  les_dict,
                  cn + "main_timeseries.pdf",
                  cb_min,
                  cb_max,
                  zmin,
                  zmax,
                  folder=f1)

    pls.plot_1D(scm_dict_t, les_dict_t, cn, folder=f2)
def test_plot_DYCOMS_RF01(sim_data):
    """
    plot DYCOMS_RF01 timeseries
    """
    # make directory
    localpath = os.getcwd()
    try:
        os.mkdir(localpath + "/plots/output/DYCOMS_RF01/")
    except:
        print('DYCOMS_RF01 folder exists')
    try:
        os.mkdir(localpath + "/plots/output/DYCOMS_RF01/all_variables/")
    except:
        print('DYCOMS_RF01/all_variables folder exists')

    if (os.path.exists(localpath + "/les_data/DYCOMS_RF01.nc")):
        les_data = Dataset(localpath + "/les_data/DYCOMS_RF01.nc", 'r')
    else:
        url_ = "https://www.dropbox.com/s/dh636h4owlt6a79/DYCOMS_RF01.nc?dl=0"
        os.system("wget -O " + localpath + "/les_data/DYCOMS_RF01.nc " + url_)
        les_data = Dataset(localpath + "/les_data/DYCOMS_RF01.nc", 'r')

    f1 = "plots/output/DYCOMS_RF01/"
    f2 = f1 + "all_variables/"
    cn = "DYCOMS_RF01_"
    t0 = 2
    t1 = 4
    zmin = 0.0
    zmax = 1.2
    cb_min = [0., 0.]
    cb_max = [0.9, 1.4]
    fixed_cbar = True
    cb_min_t = [287.5, 287.5, 288.5, 0, 0, 0, -1, -1, 0, -1, 0, 0, 0, 9,\
                -0.16, 0, 4.2, -5.5,\
                 0, -0.25, 0,\
                -0.06, -0.1, -0.1,\
                 0., -0.05, -1e-5]
    cb_max_t = [307.5, 307.5, 289.5, 0.9, 0.9, 1.5, 1, 1, 100, 1, 100,  12, 12, 11,\
                0, 1.4, 7.2, -3.5,\
                0.24, 0.05, 1.5,\
                0.02, 0.15, 0.15,\
                0.05, 0.008, 0.0001]

    scm_dict = cmn.read_scm_data(sim_data)
    les_dict = cmn.read_les_data(les_data)

    scm_dict_t = cmn.read_scm_data_timeseries(sim_data)
    les_dict_t = cmn.read_les_data_timeseries(les_data)

    pls.plot_closures(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "closures.pdf",
                      folder=f1)
    pls.plot_spec_hum(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "humidities.pdf",
                      folder=f1)
    pls.plot_upd_prop(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "updraft_properties.pdf",
                      folder=f1)
    pls.plot_fluxes(scm_dict,
                    les_dict,
                    t0,
                    t1,
                    zmin,
                    zmax,
                    cn + "mean_fluxes.pdf",
                    folder=f1)
    pls.plot_tke_comp(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "tke_components.pdf",
                      folder=f1)

    pls.plot_cvar_mean(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_mean.pdf",
                       folder=f2)
    pls.plot_cvar_comp(scm_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_components.pdf",
                       folder=f2)
    pls.plot_tke_break(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "tke_breakdown.pdf",
                       folder=f2)

    pls.plot_contour_t(scm_dict,
                       les_dict,
                       fixed_cbar,
                       cb_min_t,
                       cb_max_t,
                       zmin,
                       zmax,
                       folder=f2)
    pls.plot_mean_prof(scm_dict, les_dict, t0, t1, zmin, zmax, folder=f2)

    pls.plot_main(scm_dict_t,
                  les_dict_t,
                  scm_dict,
                  les_dict,
                  cn + "main_timeseries.pdf",
                  cb_min,
                  cb_max,
                  zmin,
                  zmax,
                  folder=f1)

    pls.plot_1D(scm_dict_t, les_dict_t, cn, folder=f2)
def test_plot_DYCOMS_RF01(sim_data):
    """
    plot DYCOMS_RF01 timeseries
    """
    # make directory
    localpath = Path.cwd()
    (localpath / "plots/output/DYCOMS_RF01/all_variables/").mkdir(
        parents=True, exist_ok=True)
    les_data_path = localpath / "les_data/DYCOMS_RF01.nc"
    if not les_data_path.is_file():
        url_ = r"https://drive.google.com/uc?export=download&id=1qBpJRVZsaQeJLuBEUwzLDv0aNlZsBaEq"
        os.system(f"curl -sLo {les_data_path} '{url_}'")
    les_data = Dataset(les_data_path, 'r')

    f1 = "plots/output/DYCOMS_RF01/"
    f2 = f1 + "all_variables/"
    cn = "DYCOMS_RF01_"
    t0 = 2
    t1 = 4
    zmin = 0.0
    zmax = 1.2
    cb_min = [0., 0.]
    cb_max = [0.9, 1.4]
    fixed_cbar = True
    cb_min_t = [287.5, 287.5, 288.5, 0, 0, 0, -1, -1, 0, -1, 0, 0, 0, 9,\
                -0.16, 0, 4.2, -5.5,\
                 0, -0.25, 0,\
                -0.06, -0.1, -0.1,\
                 0., -0.05, -1e-5]
    cb_max_t = [307.5, 307.5, 289.5, 0.9, 0.9, 1.5, 1, 1, 100, 1, 100,  12, 12, 11,\
                0, 1.4, 7.2, -3.5,\
                0.24, 0.05, 1.5,\
                0.02, 0.15, 0.15,\
                0.05, 0.008, 0.0001]

    scm_dict = cmn.read_scm_data(sim_data)
    les_dict = cmn.read_les_data(les_data)

    scm_dict_t = cmn.read_scm_data_timeseries(sim_data)
    les_dict_t = cmn.read_les_data_timeseries(les_data)

    pls.plot_closures(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "closures.pdf",
                      folder=f1)
    pls.plot_spec_hum(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "humidities.pdf",
                      folder=f1)
    pls.plot_upd_prop(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "updraft_properties.pdf",
                      folder=f1)
    pls.plot_fluxes(scm_dict,
                    les_dict,
                    t0,
                    t1,
                    zmin,
                    zmax,
                    cn + "mean_fluxes.pdf",
                    folder=f1)
    pls.plot_tke_comp(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "tke_components.pdf",
                      folder=f1)

    pls.plot_cvar_mean(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_mean.pdf",
                       folder=f2)
    pls.plot_cvar_comp(scm_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_components.pdf",
                       folder=f2)
    pls.plot_tke_break(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "tke_breakdown.pdf",
                       folder=f2)

    pls.plot_contour_t(scm_dict,
                       les_dict,
                       fixed_cbar,
                       cb_min_t,
                       cb_max_t,
                       zmin,
                       zmax,
                       folder=f2)
    pls.plot_mean_prof(scm_dict, les_dict, t0, t1, zmin, zmax, folder=f2)

    pls.plot_main(scm_dict_t,
                  les_dict_t,
                  scm_dict,
                  les_dict,
                  cn + "main_timeseries.pdf",
                  cb_min,
                  cb_max,
                  zmin,
                  zmax,
                  folder=f1)

    pls.plot_1D(scm_dict_t, les_dict_t, cn, folder=f2)
Exemple #5
0
def test_plot_Rico(sim_data):
    """
    plot Rico timeseries
    """
    # make directory
    localpath = Path.cwd()
    (localpath / "plots/output/Rico/all_variables/").mkdir(parents=True,
                                                           exist_ok=True)
    les_data_path = localpath / "les_data/Rico.nc"
    if not les_data_path.is_file():
        url_ = r"https://drive.google.com/uc?export=download&id=1Wyt8S1I7cNYYP3e-edSgXDTRpBrkwo1t"
        os.system(f"curl -sLo {les_data_path} '{url_}'")
    les_data = Dataset(les_data_path, 'r')

    f1 = "plots/output/Rico/"
    f2 = f1 + "all_variables/"
    cn = "Rico_"
    t0 = 22
    t1 = 24
    zmin = 0.0
    zmax = 4.0
    cb_min = [0., 0.]
    cb_max = [0.05, 5]
    fixed_cbar = True
    cb_min_t = [296, 296, 297, 0, 0, 0, 0, 0, 0, 0, 9, 0, 0, 9,\
                -0.2, 0, -10, -7,\
                 0, -0.05, 0,\
                -0.2, -0.05, -0.2,\
                 0., -0.3, -1e-4]
    cb_max_t = [330, 330, 308, 0.04, 0.02, 1.5, 0.05, 0.03, 100, 3, 100, 18, 18, 18,\
                0, 4, 3, -3,\
                0.24, 0.03, 1.5,\
                0.04, 0.05, 0.04,\
                0.3, 0.05, 4e-4]

    scm_dict = cmn.read_scm_data(sim_data)
    les_dict = cmn.read_les_data(les_data)

    scm_dict_t = cmn.read_scm_data_timeseries(sim_data)
    les_dict_t = cmn.read_les_data_timeseries(les_data)

    pls.plot_closures(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "closures.pdf",
                      folder=f1)
    pls.plot_spec_hum(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "humidities.pdf",
                      folder=f1)
    pls.plot_upd_prop(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "updraft_properties.pdf",
                      folder=f1)
    pls.plot_fluxes(scm_dict,
                    les_dict,
                    t0,
                    t1,
                    zmin,
                    zmax,
                    cn + "mean_fluxes.pdf",
                    folder=f1)
    pls.plot_tke_comp(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "tke_components.pdf",
                      folder=f1)

    pls.plot_cvar_mean(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_mean.pdf",
                       folder=f2)
    pls.plot_cvar_comp(scm_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_components.pdf",
                       folder=f2)
    pls.plot_tke_break(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "tke_breakdown.pdf",
                       folder=f2)

    pls.plot_contour_t(scm_dict,
                       les_dict,
                       fixed_cbar,
                       cb_min_t,
                       cb_max_t,
                       zmin,
                       zmax,
                       folder=f2)
    pls.plot_mean_prof(scm_dict, les_dict, t0, t1, zmin, zmax, folder=f2)

    pls.plot_main(scm_dict_t,
                  les_dict_t,
                  scm_dict,
                  les_dict,
                  cn + "main_timeseries.pdf",
                  cb_min,
                  cb_max,
                  zmin,
                  zmax,
                  folder=f1)

    pls.plot_1D(scm_dict_t, les_dict_t, cn, folder=f2)
Exemple #6
0
def test_plot_Bomex(sim_data):
    """
    plot Bomex timeseries
    """
    # make directory
    localpath = os.getcwd()
    try:
        os.mkdir(localpath + "/plots/output/Bomex/")
    except:
        print('Bomex folder exists')
    try:
        os.mkdir(localpath + "/plots/output/Bomex/all_variables/")
    except:
        print('Bomex/all_variables folder exists')

    if (os.path.exists(localpath + "/les_data/Bomex.nc")):
        les_data = Dataset(localpath + "/les_data/Bomex.nc", 'r')
    else:
        url_ = "https://www.dropbox.com/s/zrhxou8i80bfdk2/Bomex.nc?dl=0"
        os.system("wget -O " + localpath + "/les_data/Bomex.nc " + url_)
        les_data = Dataset(localpath + "/les_data/Bomex.nc", 'r')

    f1 = "plots/output/Bomex/"
    f2 = f1 + "all_variables/"
    cn = "Bomex_"
    t0 = 4
    t1 = 6
    zmin = 0.0
    zmax = 2.2
    cb_min = [0., 0.]
    cb_max = [0.021, 4.8]
    fixed_cbar = True
    cb_min_t = [298, 298, 298.5, 0, 0, 0, -1, -1, 0, -1, 0, 2, 2, 12.5,\
                -0.15, 0, -9, -2,\
                 0, -0.075, 0,\
                -0.09, -0.015, -0.08,\
                 0., -0.16, -1e-5]
    cb_max_t = [312, 312, 301, 0.02, 0.008, 2, 1, 1, 100, 1, 100, 18, 18, 18,\
                0, 4, -4, 1,\
                0.24, 0.035, .5,\
                0.015, 0.075, 0.04,\
                0.16, 0.02, 0.0002]

    scm_dict = cmn.read_scm_data(sim_data)
    les_dict = cmn.read_les_data(les_data)

    scm_dict_t = cmn.read_scm_data_timeseries(sim_data)
    les_dict_t = cmn.read_les_data_timeseries(les_data)

    pls.plot_closures(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "closures.pdf",
                      folder=f1)
    pls.plot_spec_hum(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "humidities.pdf",
                      folder=f1)
    pls.plot_upd_prop(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "updraft_properties.pdf",
                      folder=f1)
    pls.plot_fluxes(scm_dict,
                    les_dict,
                    t0,
                    t1,
                    zmin,
                    zmax,
                    cn + "mean_fluxes.pdf",
                    folder=f1)
    pls.plot_tke_comp(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "tke_components.pdf",
                      folder=f1)

    pls.plot_cvar_mean(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_mean.pdf",
                       folder=f2)
    pls.plot_cvar_comp(scm_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_components.pdf",
                       folder=f2)
    pls.plot_tke_break(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "tke_breakdown.pdf",
                       folder=f2)

    pls.plot_contour_t(scm_dict,
                       les_dict,
                       fixed_cbar,
                       cb_min_t,
                       cb_max_t,
                       zmin,
                       zmax,
                       folder=f2)
    pls.plot_mean_prof(scm_dict, les_dict, t0, t1, zmin, zmax, folder=f2)

    pls.plot_main(scm_dict_t,
                  les_dict_t,
                  scm_dict,
                  les_dict,
                  cn + "main_timeseries.pdf",
                  cb_min,
                  cb_max,
                  zmin,
                  zmax,
                  folder=f1)

    pls.plot_1D(scm_dict_t, les_dict_t, cn, folder=f2)
Exemple #7
0
def test_plot_ARM_SGP(sim_data):
    """
    plot ARM_SGP timeseries
    """
    # make directory
    localpath = os.getcwd()
    try:
        os.mkdir(localpath + "/plots/output/ARM_SGP/")
    except:
        print('ARM_SGP folder exists')
    try:
        os.mkdir(localpath + "/plots/output/ARM_SGP/all_variables/")
    except:
        print('ARM_SGP/all_variables folder exists')

    if (os.path.exists(localpath + "/les_data/ARM_SGP.nc")):
        les_data = Dataset(localpath + "/les_data/ARM_SGP.nc", 'r')
    else:
        url_ = "https://www.dropbox.com/s/j7s0jmmkwn7av62/ARM_SGP.nc?dl=0"
        os.system("wget -O " + localpath + "/les_data/ARM_SGP.nc " + url_)
        les_data = Dataset(localpath + "/les_data/ARM_SGP.nc", 'r')

    f1 = "plots/output/ARM_SGP/"
    f2 = f1 + "all_variables/"
    cn = "ARM_SGP_"
    t0 = 8
    t1 = 11
    zmin = 0.0
    zmax = 3.5
    cb_min = [0., 0.]
    cb_max = [0.05, 7]
    fixed_cbar = True
    cb_min_t = [295, 295, 300, 0, 0, 0, -1, -1, 0, -1, 0, 2.5, 2.5, 9,\
                -0.32, 0, 9, -0.1,\
                 0, -0.1, 0,\
                -0.25, -0.12, -0.08,\
                -0.0, -0.44, -0.045]
    cb_max_t = [335, 335, 312, 0.05, 0.016, 4, 1, 1, 100, 1, 100, 20, 20, 20,\
                0, 7, 11, 0.1,\
                0.28, 0.03, 1.05,\
                0.12, 0.25, 0.18,\
                0.44, 0.0, 0.25]

    scm_dict = cmn.read_scm_data(sim_data)
    les_dict = cmn.read_les_data(les_data)

    scm_dict_t = cmn.read_scm_data_timeseries(sim_data)
    les_dict_t = cmn.read_les_data_timeseries(les_data)

    pls.plot_closures(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "closures.pdf",
                      folder=f1)
    pls.plot_spec_hum(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "humidities.pdf",
                      folder=f1)
    pls.plot_upd_prop(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "updraft_properties.pdf",
                      folder=f1)
    pls.plot_fluxes(scm_dict,
                    les_dict,
                    t0,
                    t1,
                    zmin,
                    zmax,
                    cn + "mean_fluxes.pdf",
                    folder=f1)
    pls.plot_tke_comp(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "tke_components.pdf",
                      folder=f1)

    pls.plot_cvar_mean(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_mean.pdf",
                       folder=f2)
    pls.plot_cvar_comp(scm_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_components.pdf",
                       folder=f2)
    pls.plot_tke_break(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "tke_breakdown.pdf",
                       folder=f2)

    pls.plot_contour_t(scm_dict,
                       les_dict,
                       fixed_cbar,
                       cb_min_t,
                       cb_max_t,
                       zmin,
                       zmax,
                       folder=f2)
    pls.plot_mean_prof(scm_dict, les_dict, t0, t1, zmin, zmax, folder=f2)

    pls.plot_main(scm_dict_t,
                  les_dict_t,
                  scm_dict,
                  les_dict,
                  cn + "main_timeseries.pdf",
                  cb_min,
                  cb_max,
                  zmin,
                  zmax,
                  folder=f1)

    pls.plot_1D(scm_dict_t, les_dict_t, cn, folder=f2)
Exemple #8
0
def test_plot_Gabls(sim_data):
    """
    plot Gabls timeseries
    """
    # make directory
    localpath = Path.cwd()
    (localpath / "plots/output/Gabls/all_variables/").mkdir(parents=True,
                                                            exist_ok=True)
    les_data_path = localpath / "les_data/Gabls.nc"
    if not les_data_path.is_file():
        url_ = r"https://drive.google.com/uc?export=download&id=1gSyyz4Jx0JqaeO7gHaD9--uOD_tamvYA"
        os.system(f"curl -sLo {les_data_path} '{url_}'")
    les_data = Dataset(les_data_path, 'r')

    f1 = "plots/output/Gabls/"
    f2 = f1 + "all_variables/"
    cn = "Gabls_"
    t0 = 8
    t1 = 9
    zmin = 0.0
    zmax = 0.25
    cb_min = [0., 0.]
    cb_max = [0.01, 0.4]
    fixed_cbar = True
    cb_min_t = [261, 261, 262, 0, 0, 0, 0, 0, 0,0, 0, 0, 0, 0,\
                -0.1, 0, 0, -1,\
                 0, -0.02, 0,\
                -0.01, -0.003, -0.02,\
                 0., 0, -0.01]
    cb_max_t = [268, 268, 265, 1, 1, 1, 1, 1, 100, 1, 100, 2, 2, 2,\
                0, 0.8, 10, 4,\
                0.4, 0.003, 1,\
                0.003, 0.01, 0.012,\
                0.009, 0.1, 0.1]

    scm_dict = cmn.read_scm_data(sim_data)
    les_dict = cmn.read_les_data(les_data)

    scm_dict_t = cmn.read_scm_data_timeseries(sim_data)
    les_dict_t = cmn.read_les_data_timeseries(les_data)

    pls.plot_closures(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "closures.pdf",
                      folder=f1)
    pls.plot_spec_hum(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "humidities.pdf",
                      folder=f1)
    pls.plot_upd_prop(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "updraft_properties.pdf",
                      folder=f1)
    pls.plot_fluxes(scm_dict,
                    les_dict,
                    t0,
                    t1,
                    zmin,
                    zmax,
                    cn + "mean_fluxes.pdf",
                    folder=f1)
    pls.plot_tke_comp(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "tke_components.pdf",
                      folder=f1)

    pls.plot_cvar_mean(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_mean.pdf",
                       folder=f2)
    pls.plot_cvar_comp(scm_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_components.pdf",
                       folder=f2)
    pls.plot_tke_break(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "tke_breakdown.pdf",
                       folder=f2)

    pls.plot_contour_t(scm_dict,
                       les_dict,
                       fixed_cbar,
                       cb_min_t,
                       cb_max_t,
                       zmin,
                       zmax,
                       folder=f2)
    pls.plot_mean_prof(scm_dict, les_dict, t0, t1, zmin, zmax, folder=f2)

    pls.plot_main(scm_dict_t,
                  les_dict_t,
                  scm_dict,
                  les_dict,
                  cn + "main_timeseries.pdf",
                  cb_min,
                  cb_max,
                  zmin,
                  zmax,
                  folder=f1)

    pls.plot_1D(scm_dict_t, les_dict_t, cn, folder=f2)
Exemple #9
0
def test_plot_Gabls(sim_data):
    """
    plot Gabls timeseries
    """
    # make directory
    localpath = os.getcwd()
    try:
        os.mkdir(localpath + "/plots/output/Gabls/")
    except:
        print('Gabls folder exists')
    try:
        os.mkdir(localpath + "/plots/output/Gabls/all_variables/")
    except:
        print('Gabls/all_variables folder exists')

    if (os.path.exists(localpath + "/les_data/Gabls.nc")):
        les_data = Dataset(localpath + "/les_data/Gabls.nc", 'r')
    else:
        url_ = "https://www.dropbox.com/s/241bj5yucslpb53/Gabls.nc?dl=0"
        os.system("wget -O " + localpath + "/les_data/Gabls.nc " + url_)
        les_data = Dataset(localpath + "/les_data/Gabls.nc", 'r')

    f1 = "plots/output/Gabls/"
    f2 = f1 + "all_variables/"
    cn = "Gabls_"
    t0 = 8
    t1 = 9
    zmin = 0.0
    zmax = 0.25
    cb_min = [0., 0.]
    cb_max = [0.01, 0.4]
    fixed_cbar = True
    cb_min_t = [261, 261, 262, 0, 0, 0, 0, 0, 0,0, 0, 0, 0, 0,\
                -0.1, 0, 0, -1,\
                 0, -0.02, 0,\
                -0.01, -0.003, -0.02,\
                 0., 0, -0.01]
    cb_max_t = [268, 268, 265, 1, 1, 1, 1, 1, 100, 1, 100, 2, 2, 2,\
                0, 0.8, 10, 4,\
                0.4, 0.003, 1,\
                0.003, 0.01, 0.012,\
                0.009, 0.1, 0.1]

    scm_dict = cmn.read_scm_data(sim_data)
    les_dict = cmn.read_les_data(les_data)

    scm_dict_t = cmn.read_scm_data_timeseries(sim_data)
    les_dict_t = cmn.read_les_data_timeseries(les_data)

    pls.plot_closures(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "closures.pdf",
                      folder=f1)
    pls.plot_spec_hum(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "humidities.pdf",
                      folder=f1)
    pls.plot_upd_prop(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "updraft_properties.pdf",
                      folder=f1)
    pls.plot_fluxes(scm_dict,
                    les_dict,
                    t0,
                    t1,
                    zmin,
                    zmax,
                    cn + "mean_fluxes.pdf",
                    folder=f1)
    pls.plot_tke_comp(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "tke_components.pdf",
                      folder=f1)

    pls.plot_cvar_mean(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_mean.pdf",
                       folder=f2)
    pls.plot_cvar_comp(scm_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_components.pdf",
                       folder=f2)
    pls.plot_tke_break(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "tke_breakdown.pdf",
                       folder=f2)

    pls.plot_contour_t(scm_dict,
                       les_dict,
                       fixed_cbar,
                       cb_min_t,
                       cb_max_t,
                       zmin,
                       zmax,
                       folder=f2)
    pls.plot_mean_prof(scm_dict, les_dict, t0, t1, zmin, zmax, folder=f2)

    pls.plot_main(scm_dict_t,
                  les_dict_t,
                  scm_dict,
                  les_dict,
                  cn + "main_timeseries.pdf",
                  cb_min,
                  cb_max,
                  zmin,
                  zmax,
                  folder=f1)

    pls.plot_1D(scm_dict_t, les_dict_t, cn, folder=f2)
Exemple #10
0
def test_plot_Rico(sim_data):
    """
    plot Rico timeseries
    """
    # make directory
    localpath = os.getcwd()
    try:
        os.mkdir(localpath + "/plots/output/Rico/")
    except:
        print('Rico folder exists')
    try:
        os.mkdir(localpath + "/plots/output/Rico/all_variables/")
    except:
        print('Rico/all_variables folder exists')

    if (os.path.exists(localpath + "/les_data/Rico.nc")):
        les_data = Dataset(localpath + "/les_data/Rico.nc", 'r')
    else:
        url_ = "https://www.dropbox.com/s/c2bvey47y8xryuc/Rico.nc?dl=0"
        os.system("wget -O " + localpath + "/les_data/Rico.nc " + url_)
        les_data = Dataset(localpath + "/les_data/Rico.nc", 'r')

    f1 = "plots/output/Rico/"
    f2 = f1 + "all_variables/"
    cn = "Rico_"
    t0 = 22
    t1 = 24
    zmin = 0.0
    zmax = 4.0
    cb_min = [0., 0.]
    cb_max = [0.05, 5]
    fixed_cbar = True
    cb_min_t = [296, 296, 297, 0, 0, 0, 0, 0, 0, 0, 9, 0, 0, 9,\
                -0.2, 0, -10, -7,\
                 0, -0.05, 0,\
                -0.2, -0.05, -0.2,\
                 0., -0.3, -1e-4]
    cb_max_t = [330, 330, 308, 0.04, 0.02, 1.5, 0.05, 0.03, 100, 3, 100, 18, 18, 18,\
                0, 4, 3, -3,\
                0.24, 0.03, 1.5,\
                0.04, 0.05, 0.04,\
                0.3, 0.05, 4e-4]

    scm_dict = cmn.read_scm_data(sim_data)
    les_dict = cmn.read_les_data(les_data)

    scm_dict_t = cmn.read_scm_data_timeseries(sim_data)
    les_dict_t = cmn.read_les_data_timeseries(les_data)

    pls.plot_closures(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "closures.pdf",
                      folder=f1)
    pls.plot_spec_hum(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "humidities.pdf",
                      folder=f1)
    pls.plot_upd_prop(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "updraft_properties.pdf",
                      folder=f1)
    pls.plot_fluxes(scm_dict,
                    les_dict,
                    t0,
                    t1,
                    zmin,
                    zmax,
                    cn + "mean_fluxes.pdf",
                    folder=f1)
    pls.plot_tke_comp(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "tke_components.pdf",
                      folder=f1)

    pls.plot_cvar_mean(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_mean.pdf",
                       folder=f2)
    pls.plot_cvar_comp(scm_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_components.pdf",
                       folder=f2)
    pls.plot_tke_break(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "tke_breakdown.pdf",
                       folder=f2)

    pls.plot_contour_t(scm_dict,
                       les_dict,
                       fixed_cbar,
                       cb_min_t,
                       cb_max_t,
                       zmin,
                       zmax,
                       folder=f2)
    pls.plot_mean_prof(scm_dict, les_dict, t0, t1, zmin, zmax, folder=f2)

    pls.plot_main(scm_dict_t,
                  les_dict_t,
                  scm_dict,
                  les_dict,
                  cn + "main_timeseries.pdf",
                  cb_min,
                  cb_max,
                  zmin,
                  zmax,
                  folder=f1)

    pls.plot_1D(scm_dict_t, les_dict_t, cn, folder=f2)
Exemple #11
0
def test_plot_Tan2018(sim_data):
    """
    plot Tan2018 timeseries
    """
    # make directory
    localpath = os.getcwd()
    try:
        os.mkdir(localpath + "/plots/output/Tan2018/")
    except:
        print('Tan2018 folder exists')
    try:
        os.mkdir(localpath + "/plots/output/Tan2018/all_variables/")
    except:
        print('Tan2018/all_variables folder exists')

    if (os.path.exists(localpath + "/les_data/Bomex.nc")):
        les_data = Dataset(localpath + "/les_data/Bomex.nc", 'r')
    else:
        url_ = "https://www.dropbox.com/s/zrhxou8i80bfdk2/Bomex.nc?dl=0"
        os.system("wget -O " + localpath + "/les_data/Bomex.nc " + url_)
        les_data = Dataset(localpath + "/les_data/Bomex.nc", 'r')

    f1 = "plots/output/Tan2018/"
    f2 = f1 + "all_variables/"
    cn = "Tan2018_"
    t0 = 5
    t1 = 6
    zmin = 0.0
    zmax = 2.5
    cb_max = [1, 1]  #TODO
    fixed_cbar = True
    cb_min_t = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,\
                0, 0, 0, 0,\
                0, 0, 0,\
                0, 0, 0,\
                0, 0, 0]#TODO
    cb_max_t = [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,\
                1, 1, 1, 1,\
                1, 1, 1,\
                1, 1, 1,\
                1, 1, 1]#TODO

    scm_dict = cmn.read_scm_data(sim_data)
    les_dict = cmn.read_les_data(les_data)

    scm_dict_t = cmn.read_scm_data_timeseries(sim_data)
    les_dict_t = cmn.read_les_data_timeseries(les_data)

    pls.plot_closures(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "closures.pdf",
                      folder=f1)
    pls.plot_spec_hum(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "humidities.pdf",
                      folder=f1)
    pls.plot_upd_prop(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "updraft_properties.pdf",
                      folder=f1)
    pls.plot_fluxes(scm_dict,
                    les_dict,
                    t0,
                    t1,
                    zmin,
                    zmax,
                    cn + "mean_fluxes.pdf",
                    folder=f1)
    pls.plot_tke_comp(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "tke_components.pdf",
                      folder=f1)

    pls.plot_cvar_mean(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_mean.pdf",
                       folder=f2)
    pls.plot_cvar_comp(scm_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_components.pdf",
                       folder=f2)
    pls.plot_tke_break(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "tke_breakdown.pdf",
                       folder=f2)

    pls.plot_contour_t(scm_dict,
                       les_dict,
                       fixed_cbar,
                       cb_min_t,
                       cb_max_t,
                       zmin,
                       zmax,
                       folder=f2)
    pls.plot_mean_prof(scm_dict, les_dict, t0, t1, zmin, zmax, folder=f2)

    pls.plot_main(scm_dict_t,
                  les_dict_t,
                  scm_dict,
                  les_dict,
                  cn + "main_timeseries.pdf",
                  cb_min,
                  cb_max,
                  zmin,
                  zmax,
                  folder=f1)

    pls.plot_1D(scm_dict_t, les_dict_t, cn, folder=f2)
Exemple #12
0
def test_plot_Bomex(sim_data):
    """
    plot Bomex timeseries
    """
    # make directory
    localpath = Path.cwd()
    (localpath / "plots/output/Bomex/all_variables/").mkdir(parents=True,
                                                            exist_ok=True)
    les_data_path = localpath / "les_data/Bomex.nc"
    if not les_data_path.is_file():
        url_ = r"https://drive.google.com/uc?export=download&id=1h8LcxaoBVHqtxwoaynux_3PrUpi_8GfY"
        os.system(f"curl -sLo {les_data_path} '{url_}'")
    les_data = Dataset(les_data_path, 'r')

    f1 = "plots/output/Bomex/"
    f2 = f1 + "all_variables/"
    cn = "Bomex_"
    t0 = 4
    t1 = 6
    zmin = 0.0
    zmax = 2.2
    cb_min = [0., 0.]
    cb_max = [0.021, 4.8]
    fixed_cbar = True
    cb_min_t = [298, 298, 298.5, 0, 0, 0, -1, -1, 0, -1, 0, 2, 2, 12.5,\
                -0.15, 0, -9, -2,\
                 0, -0.075, 0,\
                -0.09, -0.015, -0.08,\
                 0., -0.16, -1e-5]
    cb_max_t = [312, 312, 301, 0.02, 0.008, 2, 1, 1, 100, 1, 100, 18, 18, 18,\
                0, 4, -4, 1,\
                0.24, 0.035, .5,\
                0.015, 0.075, 0.04,\
                0.16, 0.02, 0.0002]

    scm_dict = cmn.read_scm_data(sim_data)
    les_dict = cmn.read_les_data(les_data)

    scm_dict_t = cmn.read_scm_data_timeseries(sim_data)
    les_dict_t = cmn.read_les_data_timeseries(les_data)

    pls.plot_closures(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "closures.pdf",
                      folder=f1)
    pls.plot_spec_hum(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "humidities.pdf",
                      folder=f1)
    pls.plot_upd_prop(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "updraft_properties.pdf",
                      folder=f1)
    pls.plot_fluxes(scm_dict,
                    les_dict,
                    t0,
                    t1,
                    zmin,
                    zmax,
                    cn + "mean_fluxes.pdf",
                    folder=f1)
    pls.plot_tke_comp(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "tke_components.pdf",
                      folder=f1)

    pls.plot_cvar_mean(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_mean.pdf",
                       folder=f2)
    pls.plot_cvar_comp(scm_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_components.pdf",
                       folder=f2)
    pls.plot_tke_break(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "tke_breakdown.pdf",
                       folder=f2)

    pls.plot_contour_t(scm_dict,
                       les_dict,
                       fixed_cbar,
                       cb_min_t,
                       cb_max_t,
                       zmin,
                       zmax,
                       folder=f2)
    pls.plot_mean_prof(scm_dict, les_dict, t0, t1, zmin, zmax, folder=f2)

    pls.plot_main(scm_dict_t,
                  les_dict_t,
                  scm_dict,
                  les_dict,
                  cn + "main_timeseries.pdf",
                  cb_min,
                  cb_max,
                  zmin,
                  zmax,
                  folder=f1)

    pls.plot_1D(scm_dict_t, les_dict_t, cn, folder=f2)
Exemple #13
0
def test_plot_GATE_III(sim_data):
    """
    plot GATE_III timeseries
    """
    # make directory
    localpath = os.getcwd()
    try:
        os.mkdir(localpath + "/plots/output/GATE_III/")
    except:
        print('GATE_III folder exists')
    try:
        os.mkdir(localpath + "/plots/output/GATE_III/all_variables/")
    except:
        print('GATE_III/all_variables folder exists')

    if (os.path.exists(localpath + "/les_data/GATE_III.nc")):
        les_data = Dataset(localpath + "/les_data/GATE_III.nc", 'r')
    else:
        url_ = "https://www.dropbox.com/s/snhxbzxt4btgiis/TRMM_LBA.nc?dl=0"
        os.system("wget -O "+localpath+"/les_data/TRMM_LBA.nc "+url_)
        les_data = Dataset(localpath + "/les_data/GATE_III.nc", 'r')

    f1 = "plots/output/GATE_III/"
    f2 = f1 + "all_variables/"
    cn = "GATE_III_"
    t0 = 22
    t1 = 24
    zmin = 0.0
    zmax = 15.0
    cb_min = [0, 0] #TODO
    cb_max = [1, 1] #TODO
    fixed_cbar = True
    cb_min_t = [290, 290, 294, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,\
                -0.4, 0, -10, -6,\
                0, -0.06, 0,\
                -0.12, -0.12, -0.15,\
                -.05, -0.35, -0.15]
    cb_max_t = [370, 370, 348, 0.04, 0.004, 0.28, 0.06, 0.06, 100, 0.06, 100, 20, 20, 20,\
                0, 9, 6, 10,\
                0.24, 0.04, 1.75,\
                0.12, 0.12, 0.3,\
                0.35, 0.05, 0.5]

    scm_dict = cmn.read_scm_data(sim_data)
    les_dict = cmn.read_les_data(les_data)

    scm_dict_t = cmn.read_scm_data_timeseries(sim_data)
    les_dict_t = cmn.read_les_data_timeseries(les_data)

    pls.plot_closures(scm_dict, les_dict, t0, t1, zmin, zmax, cn+"closures.pdf", folder=f1)
    pls.plot_spec_hum(scm_dict, les_dict, t0, t1, zmin, zmax,  cn+"humidities.pdf", folder=f1)
    pls.plot_upd_prop(scm_dict, les_dict, t0, t1, zmin, zmax,  cn+"updraft_properties.pdf", folder=f1)
    pls.plot_fluxes(scm_dict, les_dict, t0, t1, zmin, zmax,  cn+"mean_fluxes.pdf", folder=f1)
    pls.plot_tke_comp(scm_dict, les_dict, t0, t1, zmin, zmax,  cn+"tke_components.pdf", folder=f1)

    pls.plot_cvar_mean(scm_dict, les_dict, t0, t1, zmin, zmax,  cn+"var_covar_mean.pdf", folder=f2)
    pls.plot_cvar_comp(scm_dict, t0, t1, zmin, zmax,  cn+"var_covar_components.pdf", folder=f2)
    pls.plot_tke_break(scm_dict, les_dict, t0, t1, zmin, zmax, cn+"tke_breakdown.pdf",folder=f2)

    pls.plot_contour_t(scm_dict, les_dict, fixed_cbar, cb_min_t, cb_max_t, zmin, zmax, folder=f2)
    pls.plot_mean_prof(scm_dict, les_dict, t0, t1,  zmin, zmax, folder=f2)

    pls.plot_main(scm_dict_t, les_dict_t, scm_dict, les_dict,
                  cn+"main_timeseries.pdf", cb_min, cb_max, zmin, zmax, folder=f1)

    pls.plot_1D(scm_dict_t, les_dict_t, cn, folder=f2)
Exemple #14
0
def test_plot_TRMM_LBA(sim_data):
    """
    plot TRMM_LBA timeseries
    """
    # make directory
    localpath = Path.cwd()
    (localpath / "plots/output/TRMM_LBA/all_variables/").mkdir(parents=True,
                                                               exist_ok=True)
    les_data_path = localpath / "les_data/TRMM_LBA.nc"
    if not les_data_path.is_file():
        url_ = r"https://drive.google.com/uc?export=download&id=1wMQ_Q0kgXYyj4NEnz8cxAlIuZWrMMZRP"
        os.system(f"curl -sLo {les_data_path} '{url_}'")
    les_data = Dataset(les_data_path, 'r')

    f1 = "plots/output/TRMM_LBA/"
    f2 = f1 + "all_variables/"
    cn = "TRMM_LBA_"
    t0 = 4
    t1 = 6
    zmin = 0.0
    zmax = 14.0
    cb_min = [0., 0.]
    cb_max = [0.04, 9.]
    fixed_cbar = True
    cb_min_t = [290, 290, 294, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,\
                -0.4, 0, -10, -6,\
                0, -0.06, 0,\
                -0.12, -0.12, -0.15,\
                -.05, -0.35, -0.15]
    cb_max_t = [370, 370, 348, 0.04, 0.004, 0.28, 0.06, 0.06, 100, 0.06, 100, 20, 20, 20,\
                0, 9, 6, 10,\
                0.24, 0.04, 1.75,\
                0.12, 0.12, 0.3,\
                0.35, 0.05, 0.5]

    scm_dict = cmn.read_scm_data(sim_data)
    les_dict = cmn.read_les_data(les_data)

    scm_dict_t = cmn.read_scm_data_timeseries(sim_data)
    les_dict_t = cmn.read_les_data_timeseries(les_data)

    pls.plot_closures(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "closures.pdf",
                      folder=f1)
    pls.plot_spec_hum(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "humidities.pdf",
                      folder=f1)
    pls.plot_upd_prop(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "updraft_properties.pdf",
                      folder=f1)
    pls.plot_fluxes(scm_dict,
                    les_dict,
                    t0,
                    t1,
                    zmin,
                    zmax,
                    cn + "mean_fluxes.pdf",
                    folder=f1)
    pls.plot_tke_comp(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "tke_components.pdf",
                      folder=f1)

    pls.plot_cvar_mean(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_mean.pdf",
                       folder=f2)
    pls.plot_cvar_comp(scm_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_components.pdf",
                       folder=f2)
    pls.plot_tke_break(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "tke_breakdown.pdf",
                       folder=f2)

    pls.plot_contour_t(scm_dict,
                       les_dict,
                       fixed_cbar,
                       cb_min_t,
                       cb_max_t,
                       zmin,
                       zmax,
                       folder=f2)
    pls.plot_mean_prof(scm_dict, les_dict, t0, t1, zmin, zmax, folder=f2)

    pls.plot_main(scm_dict_t,
                  les_dict_t,
                  scm_dict,
                  les_dict,
                  cn + "main_timeseries.pdf",
                  cb_min,
                  cb_max,
                  zmin,
                  zmax,
                  folder=f1)

    pls.plot_1D(scm_dict_t, les_dict_t, cn, folder=f2)
Exemple #15
0
def test_plot_Tan2018(sim_data):
    """
    plot Tan2018 timeseries
    """
    # make directory
    localpath = Path.cwd()
    (localpath / "plots/output/Tan2018/all_variables/").mkdir(parents=True,
                                                              exist_ok=True)
    les_data_path = localpath / "les_data/Bomex.nc"
    if not les_data_path.is_file():
        url_ = r"https://drive.google.com/uc?export=download&id=1h8LcxaoBVHqtxwoaynux_3PrUpi_8GfY"
        os.system(f"curl -sLo {les_data_path} '{url_}'")
    les_data = Dataset(les_data_path, 'r')

    f1 = "plots/output/Tan2018/"
    f2 = f1 + "all_variables/"
    cn = "Tan2018_"
    t0 = 5
    t1 = 6
    zmin = 0.0
    zmax = 2.5
    cb_max = [1, 1]  #TODO
    fixed_cbar = True
    cb_min_t = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,\
                0, 0, 0, 0,\
                0, 0, 0,\
                0, 0, 0,\
                0, 0, 0]#TODO
    cb_max_t = [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,\
                1, 1, 1, 1,\
                1, 1, 1,\
                1, 1, 1,\
                1, 1, 1]#TODO

    scm_dict = cmn.read_scm_data(sim_data)
    les_dict = cmn.read_les_data(les_data)

    scm_dict_t = cmn.read_scm_data_timeseries(sim_data)
    les_dict_t = cmn.read_les_data_timeseries(les_data)

    pls.plot_closures(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "closures.pdf",
                      folder=f1)
    pls.plot_spec_hum(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "humidities.pdf",
                      folder=f1)
    pls.plot_upd_prop(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "updraft_properties.pdf",
                      folder=f1)
    pls.plot_fluxes(scm_dict,
                    les_dict,
                    t0,
                    t1,
                    zmin,
                    zmax,
                    cn + "mean_fluxes.pdf",
                    folder=f1)
    pls.plot_tke_comp(scm_dict,
                      les_dict,
                      t0,
                      t1,
                      zmin,
                      zmax,
                      cn + "tke_components.pdf",
                      folder=f1)

    pls.plot_cvar_mean(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_mean.pdf",
                       folder=f2)
    pls.plot_cvar_comp(scm_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "var_covar_components.pdf",
                       folder=f2)
    pls.plot_tke_break(scm_dict,
                       les_dict,
                       t0,
                       t1,
                       zmin,
                       zmax,
                       cn + "tke_breakdown.pdf",
                       folder=f2)

    pls.plot_contour_t(scm_dict,
                       les_dict,
                       fixed_cbar,
                       cb_min_t,
                       cb_max_t,
                       zmin,
                       zmax,
                       folder=f2)
    pls.plot_mean_prof(scm_dict, les_dict, t0, t1, zmin, zmax, folder=f2)

    pls.plot_main(scm_dict_t,
                  les_dict_t,
                  scm_dict,
                  les_dict,
                  cn + "main_timeseries.pdf",
                  cb_min,
                  cb_max,
                  zmin,
                  zmax,
                  folder=f1)

    pls.plot_1D(scm_dict_t, les_dict_t, cn, folder=f2)