Correct names folders for graphs

inbo · May 13, 2024 · c7460b2 · c7460b2
1 parent c7092db
commit c7460b2
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 7 deletions.
diff --git a/src/stock_calculations/functions/get_stocks.R b/src/stock_calculations/functions/get_stocks.R
@@ -198,12 +198,18 @@ get_stocks <- function(survey_form,
 
   if (graph == TRUE) {
 
-    if (!dir.exists(paste0(dir, "splines_per_profile/"))) {
-      dir.create(paste0(dir, "splines_per_profile/"), recursive = TRUE)
+    for (survey_form_i in survey_forms) {
+
+      if (!dir.exists(paste0(dir, survey_form_i,
+                             "_splines_per_profile/"))) {
+        dir.create(paste0(dir, survey_form_i,
+                          "_splines_per_profile/"), recursive = TRUE)
+      }
     }
 
-    if (!dir.exists(paste0(dir, "splines_per_plot/"))) {
-      dir.create(paste0(dir, "splines_per_plot/"), recursive = TRUE)
+    if (!dir.exists(paste0(dir, code_survey, "_splines_per_plot/"))) {
+      dir.create(paste0(dir, code_survey, "_splines_per_plot/"),
+                 recursive = TRUE)
     }
   }
 
@@ -2422,7 +2428,7 @@ if (length(survey_forms) == 2) {
 
 
     ggsave(filename = paste0(plot_id_i, ".png"),
-           path = paste0(dir, "splines_per_plot/"),
+           path = paste0(dir, code_survey, "_splines_per_plot/"),
            plot = p,
            dpi = 500,
            height = height_i,

diff --git a/src/stock_calculations/functions/soilspline.R b/src/stock_calculations/functions/soilspline.R
@@ -51,8 +51,8 @@ soilspline <- function(id,
                  as.character(format(as.Date(
                    get_date_local(path = "./data/raw_data/")),
                    format = "%Y%m%d")),
-                 "_carbon_stocks/",
-                 "splines_per_profile/")
+                 "_carbon_stocks/", survey_form,
+                 "_splines_per_profile/")
 
   assertthat::assert_that(
     dir.exists(path))