From 6be71526eec00c726ee10be0dc93023d53f1a27d Mon Sep 17 00:00:00 2001
From: Gowtham Rao <rao@ohdsi.org>
Date: Tue, 17 Sep 2024 16:50:31 -0400
Subject: [PATCH] Update ExtractConceptSetsInCohortDefinition.R

---
 R/ExtractConceptSetsInCohortDefinition.R | 21 ++++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

diff --git a/R/ExtractConceptSetsInCohortDefinition.R b/R/ExtractConceptSetsInCohortDefinition.R
index 82fd1e2..06d4358 100644
--- a/R/ExtractConceptSetsInCohortDefinition.R
+++ b/R/ExtractConceptSetsInCohortDefinition.R
@@ -365,6 +365,7 @@ extractConceptSetsInCohortDefinition <-
       dplyr::mutate(conceptSetUsedInEntryEventToQuerySource = 0)
 
     if (length(codeSetsIdsInPrimaryCriteria) > 0) {
+      
       conceptSetExpression <- conceptSetExpression |>
         dplyr::select(-dplyr::all_of(c("conceptSetUsedInEntryEvent",
                                        "conceptSetUsedInEntryEventToQuerySource"))) |>
@@ -373,13 +374,19 @@ extractConceptSetsInCohortDefinition <-
             dplyr::distinct() |>
             dplyr::mutate(conceptSetUsedInEntryEvent = 1),
           by = "conceptSetId"
-        ) |> 
-        dplyr::left_join(
-          dplyr::tibble(conceptSetId = codeSetsIdsUsedToQuerySourceConceptsInPrimaryCriteria) |> 
-            dplyr::distinct() |> 
-            dplyr::mutate(conceptSetUsedInEntryEventToQuerySource = 1),
-          by = ("conceptSetId")
-        )
+        ) 
+      
+        if (length(codeSetsIdsUsedToQuerySourceConceptsInPrimaryCriteria) > 0) {
+          conceptSetExpression <- conceptSetExpression |> 
+            dplyr::left_join(
+              dplyr::tibble(conceptSetId = codeSetsIdsUsedToQuerySourceConceptsInPrimaryCriteria) |> 
+                dplyr::distinct() |> 
+                dplyr::mutate(conceptSetUsedInEntryEventToQuerySource = 1),
+              by = ("conceptSetId")
+            )
+        } else {
+          conceptSetExpression$conceptSetUsedInEntryEventToQuerySource <- as.integer(0)
+        }
     }
 
     uniqueConceptSets <- conceptSetExpression |>