Update ExtractConceptSetsInCohortDefinition.R

OHDSI · Sep 17, 2024 · ae4d053 · ae4d053
1 parent ed403a7
commit ae4d053
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/R/ExtractConceptSetsInCohortDefinition.R b/R/ExtractConceptSetsInCohortDefinition.R
@@ -88,13 +88,21 @@ extractConceptSetsInCohortDefinition <-
         )
 
       } else {
-        if (names(codesets) == "CodesetId") {
-          codeSetsIdsInPrimaryCriteria <- c(
-            codeSetsIdsInPrimaryCriteria,
-            as.double(codesets)
-          ) |>
+        if (any(
+          names(codesets) == "CodesetId",
+          stringr::str_detect(string = names(codesets), pattern = 'SourceConcept')
+        )) {
+          #is substring of name 'SourceConcept'
+          codeSetsIdsInPrimaryCriteria <- c(codeSetsIdsInPrimaryCriteria, as.double(codesets)) |>
             unique() |>
             sort()
+
+          if (!names(codesets) == 'CodesetId') {
+            codeSetsIdsUsedToQuerySourceConceptsInPrimaryCriteria <- c(
+              codeSetsIdsUsedToQuerySourceConceptsInPrimaryCriteria,
+              codeSetsIdsInPrimaryCriteria
+            )
+          }
         }
       }
     }