Faster parse_transform_serialize_r_block()

Author: krlmlr

R/transform-block.R

@@ -28,7 +28,9 @@ parse_transform_serialize_r_block <- function(pd_nested,
                                               base_indention) {
   if (!all(pd_nested$is_cached, na.rm = TRUE) || !cache_is_activated()) {
     transformed_pd <- apply_transformers(pd_nested, transformers)
-    flattened_pd <- post_visit_one(transformed_pd, extract_terminals) %>%
+    flattened_pd <-
+      # Special transformer: returns a list of pd
+      vec_rbind(!!!post_visit_one(transformed_pd, extract_terminals)) %>%
       enrich_terminals(transformers$use_raw_indention) %>%
       apply_ref_indention() %>%
       set_regex_indention(

R/visit.R

@@ -188,12 +188,19 @@ context_towards_terminals <- function(pd_nested,
 #' @param pd_nested A nested parse table.
 #' @keywords internal
 extract_terminals <- function(pd_nested) {
-  vec_rbind(
-    !!!ifelse(pd_nested$terminal | pd_nested$is_cached,
-      split(pd_nested, seq_len(nrow(pd_nested))),
-      pd_nested$child
-    )
-  )
+  terminal <- pd_nested$terminal
+  is_cached <- pd_nested$is_cached
+
+  child <- pd_nested$child
+
+  for (i in seq_len(nrow(pd_nested))) {
+    if (terminal[[i]] || is_cached[[i]]) {
+      child[[i]] <- list(pd_nested[i, ])
+    }
+  }
+
+  # child is a list of data frame lists here
+  unlist(unname(child), recursive = FALSE)
 }
 
 #' Enrich flattened parse table