From 6f5de621ab452cbe3b74a07f789f38af2bb283c1 Mon Sep 17 00:00:00 2001
From: Gowtham Rao <rao@ohdsi.org>
Date: Fri, 10 May 2024 10:21:25 -0400
Subject: [PATCH] Update GetConceptRecordCount.R

---
 R/GetConceptRecordCount.R | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/R/GetConceptRecordCount.R b/R/GetConceptRecordCount.R
index 9d801ea..f2d9d37 100644
--- 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))) {