use of com.beust.jcommander.internal.DefaultConsole in project quality-measure-and-cohort-service by Alvearie.
the class CohortCLI method runWithArgs.
/**
* Simulate main method behavior in a non-static context for use in testing
* tools. This method is intended to be called only once. Multiple calls for the
* same library path will attempt duplicate library loading.
*
* @param args parameter values
* @param out location where contents that would normally go to stdout should
* be written
* @return CQLEvaluator
* @throws IOException IOException
*/
public CqlEvaluator runWithArgs(String[] args, PrintStream out) throws IOException {
Arguments arguments = new Arguments();
Console console = new DefaultConsole(out);
JCommander jc = JCommander.newBuilder().programName("cql-engine").console(console).addObject(arguments).build();
jc.parse(args);
CqlEvaluator wrapper = null;
if (arguments.isDisplayHelp) {
jc.usage();
} else {
FhirClientBuilderFactory factory = FhirClientBuilderFactory.newInstance();
FhirClientBuilder fhirClientBuilder = factory.newFhirClientBuilder();
readConnectionConfiguration(arguments);
MapCqlLibraryProviderFactory libraryProviderFactory = new MapCqlLibraryProviderFactory();
String[] filters = null;
if (arguments.filters != null) {
filters = arguments.filters.toArray(new String[arguments.filters.size()]);
}
CqlLibraryProvider backingLibraryProvider;
Path libraryFolder = Paths.get(arguments.libraryPath);
if (libraryFolder.toFile().isDirectory()) {
out.println(String.format("Loading libraries from folder '%s'", libraryFolder.toString()));
backingLibraryProvider = libraryProviderFactory.fromDirectory(libraryFolder, filters);
} else if (FileHelpers.isZip(libraryFolder.toFile())) {
out.println(String.format("Loading libraries from ZIP '%s'", libraryFolder.toString()));
backingLibraryProvider = libraryProviderFactory.fromZipFile(libraryFolder, filters);
} else {
out.println(String.format("Loading libraries from FHIR Library '%s'", libraryFolder.toString()));
IGenericClient measureClient = fhirClientBuilder.createFhirClient(measureServerConfig);
FhirResourceResolver<Library> libraryResolver = R4FhirServerResourceResolverFactory.createLibraryResolver(measureClient);
R4LibraryDependencyGatherer dependencyGatherer = new R4LibraryDependencyGatherer(libraryResolver);
List<Library> cqlLibraries = dependencyGatherer.gatherForLibraryId(arguments.libraryPath);
Map<CqlLibraryDescriptor, CqlLibrary> cqlLibraryMap = toCqlLibraryMap(cqlLibraries);
backingLibraryProvider = new MapCqlLibraryProvider(cqlLibraryMap);
}
CqlLibraryProvider fhirClasspathProvider = new ClasspathCqlLibraryProvider();
backingLibraryProvider = new PriorityCqlLibraryProvider(backingLibraryProvider, fhirClasspathProvider);
CqlToElmTranslator translator = new CqlToElmTranslator();
if (arguments.modelInfoFile != null && arguments.modelInfoFile.exists()) {
translator.registerModelInfo(arguments.modelInfoFile);
}
boolean isForceTranslation = arguments.sourceFormat == Format.CQL;
CqlLibraryProvider libraryProvider = new TranslatingCqlLibraryProvider(backingLibraryProvider, translator, isForceTranslation);
IGenericClient dataClient = fhirClientBuilder.createFhirClient(dataServerConfig);
IGenericClient termClient = fhirClientBuilder.createFhirClient(terminologyServerConfig);
CqlTerminologyProvider termProvider = new R4RestFhirTerminologyProvider(termClient);
Map<String, com.ibm.cohort.cql.evaluation.parameters.Parameter> parameters = null;
if (arguments.parameters != null) {
parameters = parseParameterArguments(arguments.parameters);
}
CqlVersionedIdentifier libraryIdentifier = new CqlVersionedIdentifier(arguments.libraryName, arguments.libraryVersion);
List<Pair<String, String>> contexts;
if (arguments.contextIds == null || arguments.contextIds.isEmpty()) {
// If no context ids are provided, perform one run using a null context
contexts = Collections.singletonList(null);
} else {
contexts = arguments.contextIds.stream().map(x -> new ImmutablePair<>(arguments.contextName, x)).collect(Collectors.toList());
}
try (RetrieveCacheContext cacheContext = new DefaultRetrieveCacheContext()) {
CqlDataProvider dataProvider = R4DataProviderFactory.createDataProvider(dataClient, termProvider, cacheContext, R4FhirModelResolverFactory.createCachingResolver(), !arguments.enableTerminologyOptimization, arguments.searchPageSize);
wrapper = new CqlEvaluator().setLibraryProvider(libraryProvider).setDataProvider(dataProvider).setTerminologyProvider(termProvider);
ZonedDateTime evaluationDateTime = ZonedDateTime.now();
for (Pair<String, String> context : contexts) {
String contextLabel = context == null ? "null" : context.getRight();
out.println("Context: " + contextLabel);
CqlEvaluationResult result = wrapper.evaluate(libraryIdentifier, parameters, context, arguments.expressions, arguments.loggingLevel, evaluationDateTime);
out.print(prettyPrintResult(result));
out.println("---");
}
}
}
return wrapper;
}
use of com.beust.jcommander.internal.DefaultConsole in project quality-measure-and-cohort-service by Alvearie.
the class MeasureCLI method runWithArgs.
public MeasureEvaluator runWithArgs(String[] args, PrintStream out) throws IOException {
MeasureEvaluator evaluator = null;
Arguments arguments = new Arguments();
Console console = new DefaultConsole(out);
JCommander jc = JCommander.newBuilder().programName("measure-engine").console(console).addObject(arguments).build();
jc.parse(args);
if (arguments.isDisplayHelp) {
jc.usage();
} else {
arguments.validate();
readDataServerConfiguration(arguments);
readTerminologyServerConfiguration(arguments);
FhirClientBuilderFactory fhirClientBuilderFactory = getFhirClientBuilderFactory();
IGenericClient dataServerClient = fhirClientBuilderFactory.newFhirClientBuilder().createFhirClient(dataServerConfig);
IGenericClient terminologyServerClient = fhirClientBuilderFactory.newFhirClientBuilder().createFhirClient(terminologyServerConfig);
String[] filters = (arguments.filters != null) ? arguments.filters.toArray(new String[arguments.filters.size()]) : null;
IParser parser = getFhirContext().newJsonParser().setPrettyPrint(true);
R4QualityMeasureResolverFactory resolverFactory = new R4QualityMeasureResolverFactory(parser);
FhirResourceResolver<Library> libraryResolver;
FhirResourceResolver<Measure> measureResolver;
if (arguments.measureServerConfigFile != null && FileHelpers.isZip(arguments.measureServerConfigFile)) {
R4QualityMeasureResolvers resolvers = resolverFactory.fromZipFile(arguments.measureServerConfigFile.toPath(), filters);
libraryResolver = resolvers.getLibraryResolver();
measureResolver = resolvers.getMeasureResolver();
} else if (arguments.measureServerConfigFile != null && arguments.measureServerConfigFile.isDirectory()) {
R4QualityMeasureResolvers resolvers = resolverFactory.fromDirectory(arguments.measureServerConfigFile.toPath(), filters);
libraryResolver = resolvers.getLibraryResolver();
measureResolver = resolvers.getMeasureResolver();
} else {
readMeasureServerConfiguration(arguments);
IGenericClient measureServerClient = fhirClientBuilderFactory.newFhirClientBuilder().createFhirClient(measureServerConfig);
libraryResolver = R4FhirServerResourceResolverFactory.createLibraryResolver(measureServerClient);
measureResolver = R4FhirServerResourceResolverFactory.createMeasureResolver(measureServerClient);
}
List<MeasureContext> measureContexts;
if (arguments.measureConfigurationFile != null) {
measureContexts = MeasureContextProvider.getMeasureContexts(arguments.measureConfigurationFile);
} else {
measureContexts = MeasureContextProvider.getMeasureContexts(arguments.resourceId, arguments.parameters);
}
validateMeasureContexts(measureContexts);
R4LibraryDependencyGatherer libraryDependencyGatherer = new R4LibraryDependencyGatherer(libraryResolver);
CqlTerminologyProvider terminologyProvider = new R4RestFhirTerminologyProvider(terminologyServerClient);
try (RetrieveCacheContext retrieveCacheContext = arguments.disableRetrieveCache ? null : new DefaultRetrieveCacheContext()) {
Map<String, CqlDataProvider> dataProviders = R4DataProviderFactory.createDataProviderMap(dataServerClient, terminologyProvider, retrieveCacheContext, R4FhirModelResolverFactory.createCachingResolver(), !arguments.enableTerminologyOptimization, arguments.searchPageSize);
evaluator = new MeasureEvaluator(measureResolver, libraryResolver, libraryDependencyGatherer, terminologyProvider, dataProviders);
for (String contextId : arguments.contextIds) {
out.println("Evaluating: " + contextId);
// Reports only returned for measures where patient is in initial population
List<MeasureReport> reports = evaluator.evaluatePatientMeasures(contextId, measureContexts, new MeasureEvidenceOptions(arguments.includeEvaluatedResources, arguments.defineReturnOption));
for (MeasureReport report : reports) {
if (arguments.reportFormat == ReportFormat.TEXT) {
out.println("Result for " + report.getMeasure());
for (MeasureReport.MeasureReportGroupComponent group : report.getGroup()) {
for (MeasureReport.MeasureReportGroupPopulationComponent pop : group.getPopulation()) {
String popCode = pop.getCode().getCodingFirstRep().getCode();
if (pop.getId() != null) {
popCode += "(" + pop.getId() + ")";
}
out.println(String.format("Population: %s = %d", popCode, pop.getCount()));
}
}
} else {
out.println(parser.encodeResourceToString(report));
}
out.println("---");
}
if (reports.isEmpty()) {
out.println("---");
}
}
}
}
return evaluator;
}
use of com.beust.jcommander.internal.DefaultConsole in project quality-measure-and-cohort-service by Alvearie.
the class MeasureImporter method runWithArgs.
/**
* Execute measure import process with given arguments. Console logging is
* redirected to the provided stream.
*
* @param args Program arguments
* @param out Sink for console output
* @return number of errors encountered during processing
* @throws Exception on any error.
*/
public static int runWithArgs(String[] args, PrintStream out) throws Exception {
int resultCode = 0;
Arguments arguments = new Arguments();
Console console = new DefaultConsole(out);
JCommander jc = JCommander.newBuilder().programName("measure-importer").console(console).addObject(arguments).build();
jc.parse(args);
if (arguments.isDisplayHelp) {
jc.usage();
resultCode = 1;
} else {
FhirContext fhirContext = FhirContext.forR4();
ObjectMapper om = new ObjectMapper();
FhirServerConfig config = om.readValue(arguments.measureServerConfigFile, FhirServerConfig.class);
IGenericClient client = FhirClientBuilderFactory.newInstance().newFhirClientBuilder(fhirContext).createFhirClient(config);
MeasureImporter importer = new MeasureImporter(client);
resultCode = importer.importFiles(arguments.artifactPaths, arguments.outputPath);
if (resultCode == 0) {
out.println("Process completed with no errors.");
} else {
out.println(String.format("Process completed with errors. Failed to process %d artifacts.", resultCode));
}
}
return resultCode;
}
use of com.beust.jcommander.internal.DefaultConsole in project quality-measure-and-cohort-service by Alvearie.
the class ValueSetImporter method runWithArgs.
void runWithArgs(String[] args, PrintStream out) throws IOException {
ValueSetImporterArguments arguments = new ValueSetImporterArguments();
Console console = new DefaultConsole(out);
JCommander jc = JCommander.newBuilder().programName("value-set-importer").console(console).addObject(arguments).build();
jc.parse(args);
if (arguments.isDisplayHelp) {
jc.usage();
} else {
arguments.validate();
FhirContext fhirContext = FhirContext.forR4();
// only connect to fhir server if we are not writing it to file system
IGenericClient client = null;
ObjectMapper om = new ObjectMapper();
if (arguments.fileOutputLocation == OutputLocations.NONE) {
FhirServerConfig config = om.readValue(arguments.measureServerConfigFile, FhirServerConfig.class);
client = FhirClientBuilderFactory.newInstance().newFhirClientBuilder(fhirContext).createFhirClient(config);
}
Map<String, String> codeSystemMappings = null;
if (arguments.filename != null) {
codeSystemMappings = ValueSetUtil.getMapFromInputStream(new FileInputStream(new File(arguments.filename)));
}
for (String arg : arguments.spreadsheets) {
try (InputStream is = new FileInputStream(arg)) {
ValueSetArtifact artifact = ValueSetUtil.createArtifact(is, codeSystemMappings);
// only import the value set to fhir server if we are not writing the value set to file system
if (arguments.fileOutputLocation == OutputLocations.NONE) {
String retVal = ValueSetUtil.importArtifact(client, artifact, arguments.overrideValueSets);
if (retVal == null) {
logger.error("Value set already exists! Please provide the override option if you would like to override this value set.");
}
} else {
// write value set to file system
ValueSet vs = artifact.getFhirResource();
// If the valueset id contains urn:oid, remove it to make a valid filename
String valueSetId = vs.getId().startsWith("urn:oid:") ? vs.getId().replace("urn:oid:", "") : vs.getId();
String vsFileName = valueSetId + "." + arguments.filesystemOutputFormat.toString().toLowerCase();
if (arguments.fileOutputLocation == OutputLocations.BOTH || arguments.fileOutputLocation == OutputLocations.S3) {
S3Configuration S3Config = om.readValue(arguments.S3JsonConfigs, S3Configuration.class);
AmazonS3 S3Client = createClient(S3Config.getAccess_key_id(), S3Config.getSecret_access_key(), S3Config.getEndpoint(), S3Config.getLocation());
putToS3(arguments, fhirContext, vs, vsFileName, S3Client);
}
if (arguments.fileOutputLocation == OutputLocations.BOTH || arguments.fileOutputLocation == OutputLocations.LOCAL) {
try (BufferedWriter writer = new BufferedWriter(new FileWriter(arguments.fileSystemOutputPath + System.getProperty("file.separator") + vsFileName))) {
// create the output dir if it doesn't exist
File outputDir = new File(arguments.fileSystemOutputPath);
if (!outputDir.exists()) {
outputDir.mkdir();
}
// write to xml or json format
if (arguments.filesystemOutputFormat == FileFormat.JSON) {
fhirContext.newJsonParser().encodeResourceToWriter(vs, writer);
} else if (arguments.filesystemOutputFormat == FileFormat.XML) {
fhirContext.newXmlParser().encodeResourceToWriter(vs, writer);
}
}
}
}
}
}
}
}
Aggregations