Add parameters to limit concept count

DESCRIPTION

@@ -33,6 +33,6 @@ Remotes:
   ohdsi/CirceR,
   ohdsi/SqlRender
 License: Apache License (>= 2)
-RoxygenNote: 7.2.3
+RoxygenNote: 7.3.1
 Encoding: UTF-8
 Language: en-US

R/GetConceptRecordCount.R

@@ -57,6 +57,12 @@ getConceptRecordCount <- function(conceptIds = NULL,
                                   cohortDatabaseSchema = NULL,
                                   cohortTableName = NULL,
                                   cohortDefinitionId = NULL,
+                                  stratifyByGender = FALSE,
+                                  stratifyByYear = FALSE,
+                                  stratifyByYearQuarter = FALSE,
+                                  stratifyByYearMonth = FALSE,
+                                  stratifyByAgeGroup = FALSE,
+                                  stratifyByIncidence = FALSE,
                                   domainTableName = c(
                                     "drug_exposure",
                                     "condition_occurrence",
@@ -238,6 +244,36 @@ getConceptRecordCount <- function(conceptIds = NULL,
     dplyr::arrange() |>
     dplyr::mutate(combination = dplyr::row_number())
 
+  if (!stratifyByGender) {
+    iterations <- iterations |> 
+      dplyr::filter(!genderConceptId %in% c(8507, 8532))
+  }
+
+  if (!stratifyByYear) {
+    iterations <- iterations |> 
+      dplyr::filter(calendarType != 'Y')
+  }
+
+  if (!stratifyByYearQuarter) {
+    iterations <- iterations |> 
+      dplyr::filter(calendarType != 'Q')
+  }
+
+  if (!stratifyByYearMonth) {
+    iterations <- iterations |> 
+      dplyr::filter(calendarType != 'M')
+  }
+
+  if (!stratifyByAgeGroup) {
+    iterations <- iterations |> 
+      dplyr::filter(useAgeGroup != 'Y')
+  }
+
+  if (!stratifyByIncidence) {
+    iterations <- iterations |> 
+      dplyr::filter(incidence != 'Y')
+  }
+
   existingOutput <- c()
 
   for (i in (1:nrow(iterations))) {