Update GetConceptRecordCount.R

OHDSI · May 10, 2024 · 6f5de62 · 6f5de62
1 parent f3de058
commit 6f5de62
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/R/GetConceptRecordCount.R b/R/GetConceptRecordCount.R
@@ -63,6 +63,7 @@ getConceptRecordCount <- function(conceptIds = NULL,
                                   stratifyByYearMonth = FALSE,
                                   stratifyByAgeGroup = FALSE,
                                   stratifyByIncidence = FALSE,
+                                  getOverallCounts = FALSE,
                                   domainTableName = c(
                                     "drug_exposure",
                                     "condition_occurrence",
@@ -274,6 +275,11 @@ getConceptRecordCount <- function(conceptIds = NULL,
       dplyr::filter(incidence != 'Y')
   }
 
+  if (!getOverallCounts) {
+    iterations <- iterations |> 
+      dplyr::filter(includeConceptId == 'Y')
+  }
+
   existingOutput <- c()
 
   for (i in (1:nrow(iterations))) {