add option: extract output from lsunit_wb file for SWAT+

Author: nguyenta

R/defaultValue.R

@@ -97,27 +97,39 @@ dataOutputExtractionSWATPlus <- data.frame(FileType = c('channel_sd_day.txt',
                                                         'channel_sdmorph_day.txt', 
                                                         'channel_sdmorph_mon.txt', 
                                                         'channel_sdmorph_yr.txt',
+                                                        'lsunit_wb_day.txt', 
+                                                        'lsunit_wb_mon.txt', 
+                                                        'lsunit_wb_yr.txt',
                                                         'userReadSwatOutput'),
                                    FileName = c('channel_sd_day.txt', 
                                                 'channel_sd_mon.txt', 
                                                 'channel_sd_yr.txt',
                                                 'channel_sdmorph_day.txt', 
                                                 'channel_sdmorph_mon.txt', 
                                                 'channel_sdmorph_yr.txt',
+                                                'lsunit_wb_day.txt', 
+                                                'lsunit_wb_mon.txt', 
+                                                'lsunit_wb_yr.txt',
                                                 NA),
                                    Column = c("48, 53",
                                               "48, 53",
                                               "48, 53",
                                               "48, 53",
                                               "48, 53",
                                               "48, 53",
+                                              "48, 53",
+                                              "48, 53",
+                                              "48, 53",
                                               "1"),
                                    Reach = c("1, 2 * 1, 2, 3", 
                                              "1, 2 * 1, 2, 3", 
                                              "1, 2 * 1, 2, 3", 
                                              "1, 2 * 1, 2, 3", 
                                              "1, 2 * 1, 2, 3", 
                                              "1, 2 * 1, 2, 3", 
+                                             "1, 2 * 1, 2, 3", 
+                                             "1, 2 * 1, 2, 3", 
+                                             "1, 2 * 1, 2, 3", 
                                              NA)
 )
 
@@ -129,6 +141,9 @@ columnsOutputExtractionSWATPlus <- data.frame(title= colnames(dataOutputExtracti
                                                                 'channel_sdmorph_day.txt', 
                                                                 'channel_sdmorph_mon.txt', 
                                                                 'channel_sdmorph_yr.txt',
+                                                                'lsunit_wb_day.txt', 
+                                                                'lsunit_wb_mon.txt', 
+                                                                'lsunit_wb_yr.txt',
                                                                 'userReadSwatOutput'),
                                                       NA,
                                                       NA, 

R/readSaveOutput.R

@@ -161,7 +161,10 @@ saveOutput <- function(workingDirectory,
                fileType[i] == "channel_sd_yr.txt"       |
                fileType[i] == "channel_sdmorph_day.txt" |
                fileType[i] == "channel_sdmorph_mon.txt" |
-               fileType[i] == "channel_sdmorph_yr.txt"){
+               fileType[i] == "channel_sdmorph_yr.txt"  |
+               fileType[i] == "lsunit_wb_day.txt"       |
+               fileType[i] == "lsunit_wb_mon.txt"       |
+               fileType[i] == "lsunit_wb_yr.txt"){
       output <- readChannelFile(workingDirectory,
                                   coreNumber, 
                                   fileName[i], 
@@ -337,8 +340,7 @@ appendListObject <- function(listA, listB){
 #++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++#
 
 #-------------------------------------------------------------------------------
-# Read channel_sd_xxx.txt file, data is always at daily timestep
-#                 xxx could be "day", "month", or "yr"  readChannel_sd_dayFile
+# Read channel and lsunit data at daily, monthly, yearly time step
 # ------------------------------------------------------------------------------
 readChannelFile <- function(workingDirectory, 
                                coreNumber, 
@@ -354,7 +356,10 @@ readChannelFile <- function(workingDirectory,
                 "channel_sd_yr.txt",
                 "channel_sdmorph_day.txt", 
                 "channel_sdmorph_mon.txt", 
-                "channel_sdmorph_yr.txt")
+                "channel_sdmorph_yr.txt",
+                "lsunit_wb_day.txt",
+                "lsunit_wb_mon.txt",
+                "lsunit_wb_yr.txt")
 
   filePath <- paste(workingDirectory, "/TxtInOut_", coreNumber, "/", 
                     fileName, sep = "")