RunModel_GR1A.R 4.45 KB
Newer Older
1
2
RunModel_GR1A <- function(InputsModel, RunOptions, Param) {
  
3
4
  
  ## Initialization of variables
5
6
7
  NParam <- 1
  FortranOutputs <- .FortranOutputs(GR = "GR1A")$GR
  
8
  
9
  ## Arguments check
10
  if (!inherits(InputsModel, "InputsModel")) {
11
12
    stop("'InputsModel' must be of class 'InputsModel'")
  }
13
  if (!inherits(InputsModel, "yearly")) {
14
15
    stop("'InputsModel' must be of class 'yearly'")
  }
16
  if (!inherits(InputsModel, "GR")) {
17
18
    stop("'InputsModel' must be of class 'GR'")
  }
19
  if (!inherits(RunOptions, "RunOptions")) {
20
21
    stop("'RunOptions' must be of class 'RunOptions'")
  }
22
  if (!inherits(RunOptions, "GR")) {
23
24
25
26
27
28
29
30
31
32
33
34
    stop("'RunOptions' must be of class 'GR'")
  }
  if (!is.vector(Param) | !is.numeric(Param)) {
    stop("'Param' must be a numeric vector")
  }
  if (sum(!is.na(Param)) != NParam) {
    stop(paste("'Param' must be a vector of length", NParam, "and contain no NA"))
  }
  Param <- as.double(Param)
  
  
  ## Input data preparation
35
  if (identical(RunOptions$IndPeriod_WarmUp, 0L)) {
36
37
38
39
40
41
42
43
44
45
    RunOptions$IndPeriod_WarmUp <- NULL
  }
  IndPeriod1 <- c(RunOptions$IndPeriod_WarmUp, RunOptions$IndPeriod_Run)
  LInputSeries <- as.integer(length(IndPeriod1))
  if ("all" %in% RunOptions$Outputs_Sim) {
    IndOutputs <- as.integer(1:length(FortranOutputs))
  } else {
    IndOutputs <- which(FortranOutputs %in% RunOptions$Outputs_Sim)
  }
  
46
  
47
48
49
50
51
  ## Output data preparation
  IndPeriod2     <- (length(RunOptions$IndPeriod_WarmUp) + 1):LInputSeries
  ExportDatesR   <- "DatesR"   %in% RunOptions$Outputs_Sim
  ExportStateEnd <- "StateEnd" %in% RunOptions$Outputs_Sim
  
52
  
53
  ## Call GR model Fortan
54
  RESULTS <- .Fortran("frun_gr1a", PACKAGE = "airGR", 
55
56
57
58
59
60
61
62
63
64
65
                      ## inputs
                      LInputs = LInputSeries,                             ### length of input and output series
                      InputsPrecip = InputsModel$Precip[IndPeriod1],      ### input series of total precipitation [mm/y]
                      InputsPE = InputsModel$PotEvap[IndPeriod1],         ### input series potential evapotranspiration [mm/y]
                      NParam = as.integer(length(Param)),                 ### number of model parameter
                      Param = Param,                                      ### parameter set
                      NStates = as.integer(length(RunOptions$IniStates)), ### number of state variables used for model initialising
                      StateStart = RunOptions$IniStates,                  ### state variables used when the model run starts
                      NOutputs = as.integer(length(IndOutputs)),          ### number of output series
                      IndOutputs = IndOutputs,                            ### indices of output series
                      ## outputs
66
67
                      Outputs = matrix(as.double(-99e9), nrow = LInputSeries, ncol = length(IndOutputs)), ### output series [mm/y]
                      StateEnd = rep(as.double(-99e9), length(RunOptions$IniStates))                      ### state variables at the end of the model run
68
  )
69
70
  RESULTS$Outputs[RESULTS$Outputs   <= -99e8] <- NA
  RESULTS$StateEnd[RESULTS$StateEnd <= -99e8] <- NA
71
72
73
74
  
  
  ## Output data preparation
  ## OutputsModel only
75
  if (!ExportDatesR & !ExportStateEnd) {
76
77
78
79
    OutputsModel <- lapply(seq_len(RESULTS$NOutputs), function(i) RESULTS$Outputs[IndPeriod2, i])
    names(OutputsModel) <- FortranOutputs[IndOutputs]
  }
  ## DatesR and OutputsModel only
80
  if (ExportDatesR & !ExportStateEnd) {
81
    OutputsModel <- c(list(InputsModel$DatesR[RunOptions$IndPeriod_Run]), 
82
83
84
85
                      lapply(seq_len(RESULTS$NOutputs), function(i) RESULTS$Outputs[IndPeriod2, i]))
    names(OutputsModel) <- c("DatesR", FortranOutputs[IndOutputs])
  }
  ## OutputsModel and SateEnd only
86
  if (!ExportDatesR & ExportStateEnd) {
87
    OutputsModel <- c(lapply(seq_len(RESULTS$NOutputs), function(i) RESULTS$Outputs[IndPeriod2, i]), 
88
89
90
91
                      list(RESULTS$StateEnd))
    names(OutputsModel) <- c(FortranOutputs[IndOutputs], "StateEnd")
  }
  ## DatesR and OutputsModel and SateEnd
92
  if ((ExportDatesR & ExportStateEnd) | "all" %in% RunOptions$Outputs_Sim) {
93
94
    OutputsModel <- c(list(InputsModel$DatesR[RunOptions$IndPeriod_Run]), 
                      lapply(seq_len(RESULTS$NOutputs), function(i) RESULTS$Outputs[IndPeriod2, i]), 
95
96
97
98
99
100
101
102
103
                      list(RESULTS$StateEnd))
    names(OutputsModel) <- c("DatesR", FortranOutputs[IndOutputs], "StateEnd")
  }
  
  
  ## End
  class(OutputsModel) <- c("OutputsModel", "yearly", "GR")
  return(OutputsModel)
  
Delaigue Olivier's avatar
Delaigue Olivier committed
104
105
}