general-recon/study_loader.py at development · CentralValleyModeling/general-recon · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
from collections import namedtuple

import pandas as pd
import yaml
from utils import list_files, load_data_mult, load_data
from pathlib import Path

# Scenario management
Study = namedtuple("Scenario", ["dv_path", "sv_path", "alias", "assumptions", "climate", "color"])


with open("constants/dvars.yaml", "r") as file:
    var_dict_dv = yaml.safe_load(file)
with open("constants/svars.yaml", "r") as file:
    var_dict_sv = yaml.safe_load(file)
with open("study_ledger.yaml", "r") as file:
    study_ledger = yaml.safe_load(file)

studies = [

    Study(r"C:\jobs\20230428_DCR23\models\9.3.1_danube_adj\DSS\output\DCR2023_DV_9.3.1_v2a_Danube_Adj_v1.8.dss",
          r"C:\jobs\20230428_DCR23\models\9.3.1_danube_adj\DSS\input\DCR2023_SV_Danube_Adj_v1.8.dss",
          "DCR23_Baseline", "DCR 2023", "Current", 1),

    Study(r"C:\jobs\20230428_DCR23\models\9.3.1_danube_cc50\DSS\output\DCR2023_DV_9.3.1_Danube_cc50_v1.8.dss",
          r"C:\jobs\20230428_DCR23\models\9.3.1_danube_cc50\DSS\input\DCR2023_SV_Danube_cc50_v1.8.dss",
          "DCR23_CC50", "DCR 2023", "2043_CC50", 1),

    Study(r"C:\jobs\20230428_DCR23\models\9.3.1_danube_cc75\DSS\output\DCR2023_DV_9.3.1_Danube_CC75_v1.8.dss",
          r"C:\jobs\20230428_DCR23\models\9.3.1_danube_cc75\DSS\input\DCR2023_SV_Danube_cc75_v1.8.dss",
          "DCR23_CC75", "DCR 2023", "2043_CC75", 1),

    Study(r"C:\jobs\20230428_DCR23\models\9.3.1_danube_cc95\DSS\output\DCR2023_DV_9.3.1_Danube_CC95_v1.8.dss",
          r"C:\jobs\20230428_DCR23\models\9.3.1_danube_cc95\DSS\input\DCR2023_SV_Danube_cc95_v1.8.dss",
          "DCR23_CC95", "DCR 2023", "2043_CC95", 1),

    Study(r"C:\jobs\2025_DCR25\models\9.7.0_danube_adj\DSS\output\DCR2025_DV_9.7.0_Danube_Adj_v3.0.dss",
          r"C:\jobs\2025_DCR25\models\9.7.0_danube_adj\DSS\input\DCR2023_SV_Danube_Adj_v1.8.dss",
          "DCR25_Baseline", "DCR 2025", "Current", 1),

    Study(r"C:\jobs\2025_DCR25\models\3.7.0_danube_cc50\DSS\output\CCA4_v3.7.0__2043_50CC_LU100_SLR15_20241220.dss",
          r"C:\jobs\2025_DCR25\models\3.7.0_danube_cc50\DSS\input\DCR2023_SV_Danube_2043_cc50_v1.9.dss",
          "DCR25_CC50", "DCR 2025", "2043_CC50", 1),

    Study(r"C:\jobs\2025_DCR25\models\3.7.0_danube_cc95\DSS\output\CCA5_v3.7.0__2043_95CC_LU100_SLR30_20241227.dss",
          r"C:\jobs\2025_DCR25\models\3.7.0_danube_cc95\DSS\input\DCR2023_SV_Danube_2043_cc95_v1.9.dss",
          "DCR25_CC95", "DCR 2025", "2043_CC95", 1)

]

date_map = pd.read_csv("constants/date_map.csv", index_col=0, parse_dates=True)

load_data(studies, var_dict_dv, date_map, "dv", "dv_data.csv")
load_data(studies, var_dict_sv, date_map, "sv", "sv_data.csv")