Allow portfolios to not contain the same years.

R/pp_measures.R

@@ -25,6 +25,26 @@ pp_measures <- function(D, id = NULL) {
     warning("You have given a data frame with more than one sector.\nTargets and instruments will be assumed to be equal for every sector.")
   }
 
+  # In case of portfolios not covering the same years
+  clean.years <- FALSE
+  D.years <- D %>%
+    select(Country, Year) %>%
+    unique()
+  D.years.n <- D.years %>%
+    group_by(Country) %>%
+    summarize(N = n())
+  if (length(unique(D.years.n$N)) > 1) {
+    message("At least one portfolio contains a different number of years.")
+    clean.years <- TRUE
+    # This is used later in the cleaning of the extra years
+    D.years.yn <- D.years %>%
+      group_by(Country, Year) %>%
+      summarize(N = n()) %>%
+      ungroup() %>%
+      spread(Year, N, fill = 0) %>%
+      gather(Year, N, -Country)
+  }
+
   # Agree in the portfolio(s) to pass and convert into matrix form
   # to make the calculations
   P.full <- pp_array(D, return_matrix = FALSE)
@@ -98,5 +118,16 @@ pp_measures <- function(D, id = NULL) {
     mutate(Year = as.integer(as.numeric(Year))) %>%
     mutate(Measure = factor(as.character(Measure))) %>%
     mutate(Measure.label = factor(as.character(Measure.label)))
+
+  # Clean years for which some countries do not have data
+  if (clean.years) {
+    clean.countries <- as.character(D.years.n$Country[D.years.n$N < max(D.years.n$N)])
+    for (c in 1:length(clean.countries)) {
+      message(paste0("Cleaning years from ", clean.countries[c]))
+      extra.years <- as.vector(filter(D.years.yn, Country == clean.countries[c] & N == 0)$Year)
+      O <- O %>%
+        filter(Country != clean.countries[c] | (Country == clean.countries[] & !(Year %in% extra.years)))
+    }
+  }
   return(O)
 }