Fix a bug when there are no data in middle years

Author: MrJGao

R/base.R

@@ -1,7 +1,7 @@
-#************************************************************************************
+#*******************************************************************************
 # Description: Needed library and pre-defined functions
 # Date: 2020-11-22
-#************************************************************************************
+#*******************************************************************************
 
 .datatable.aware <- TRUE
 
@@ -62,9 +62,9 @@ FormatAvgData <- function(date_vec, vi_vec) {
         evi2 = vi_vec,
         avg_date = ""
     )
-    vi_dt$avg_date <- as.Date(paste0("1970", substr(vi_dt$date, 5, 10)))
+    vi_dt[, avg_date := as.Date(paste0("1970", substr(vi_dt$date, 5, 10)))]
     vi_dt <- stats::na.omit(vi_dt)
-    vi_dt <- data.table::setorder(vi_dt, date)
+    data.table::setorder(vi_dt, date)
 
     # Find unique dates in the averaged year
     unique_dates <- unique(vi_dt$avg_date)

R/mod_fit.R

@@ -115,12 +115,13 @@ FitBLSP <- function(date_vec, vi_vec,
     # year id vector
     if (is.null(start_yr) || is.null(end_yr)) {
         yr <- lubridate::year(date_vec) - lubridate::year(date_vec)[1] + 1
-        years <- sort(unique(year(date_vec)))
-        numYears <- length(unique(yr))
+        tmp <- sort(unique(year(date_vec)))
+        years <- tmp[1]:tmp[length(tmp)]
+        numYears <- length(1:yr[length(yr)])
     } else {
-        yr <- start_yr:end_yr - end_yr + 1
+        yr <- lubridate::year(date_vec) - lubridate::year(date_vec)[1] + 1
         years <- start_yr:end_yr
-        numYears <- length(yr)
+        numYears <- length(years)
     }
     # If user specified weights
     if (is.null(weights_vec)) {