Search in sources :

Example 11 with ModelInfo

use of org.hl7.elm_modelinfo.r1.ModelInfo in project clinical_quality_language by cqframework.

the class CqlTranslator method loadModelInfo.

public static void loadModelInfo(File modelInfoXML) {
    final ModelInfo modelInfo = JAXB.unmarshal(modelInfoXML, ModelInfo.class);
    final VersionedIdentifier modelId = new VersionedIdentifier().withId(modelInfo.getName()).withVersion(modelInfo.getVersion());
    final ModelInfoProvider modelProvider = (VersionedIdentifier modelIdentifier) -> modelInfo;
    final ModelInfoLoader modelInfoLoader = new ModelInfoLoader();
    modelInfoLoader.registerModelInfoProvider(modelProvider);
}
Also used : ModelInfo(org.hl7.elm_modelinfo.r1.ModelInfo) VersionedIdentifier(org.hl7.elm.r1.VersionedIdentifier)

Example 12 with ModelInfo

use of org.hl7.elm_modelinfo.r1.ModelInfo in project clinical_quality_language by cqframework.

the class ModelManager method buildModel.

private Model buildModel(VersionedIdentifier identifier) {
    Model model = null;
    if (identifier == null) {
        throw new IllegalArgumentException("Model identifier is required");
    }
    if (identifier.getId() == null || identifier.getId().equals("")) {
        throw new IllegalArgumentException("Model identifier Id is required");
    }
    String modelPath = NamespaceManager.getPath(identifier.getSystem(), identifier.getId());
    pushLoading(modelPath);
    try {
        ModelInfo modelInfo = modelInfoLoader.getModelInfo(identifier);
        if (identifier.getId().equals("System")) {
            model = new SystemModel(modelInfo);
        } else {
            model = new Model(modelInfo, this);
        }
    } catch (ClassNotFoundException e) {
        throw new IllegalArgumentException(String.format("Could not load model information for model %s, version %s.", identifier.getId(), identifier.getVersion()));
    } finally {
        popLoading(modelPath);
    }
    return model;
}
Also used : ModelInfo(org.hl7.elm_modelinfo.r1.ModelInfo) SystemModel(org.cqframework.cql.cql2elm.model.SystemModel) Model(org.cqframework.cql.cql2elm.model.Model) SystemModel(org.cqframework.cql.cql2elm.model.SystemModel)

Example 13 with ModelInfo

use of org.hl7.elm_modelinfo.r1.ModelInfo in project quality-measure-and-cohort-service by Alvearie.

the class SparkSchemaCreator method getDataTypeForContextKey.

private Tuple2<String, DataType> getDataTypeForContextKey(String contextName, Set<Tuple2<String, String>> usingInfos) {
    ContextDefinition contextDefinition = contextDefinitions.getContextDefinitionByName(contextName);
    String primaryDataType = contextDefinition.getPrimaryDataType();
    String primaryKeyColumn = contextDefinition.getPrimaryKeyColumn();
    DataType keyType = null;
    ModelManager modelManager = translator.newModelManager();
    // Try to find the key column's type information from a single model info.
    for (Tuple2<String, String> usingInfo : usingInfos) {
        VersionedIdentifier modelInfoIdentifier = new VersionedIdentifier().withId(usingInfo._1()).withVersion(usingInfo._2());
        ModelInfo modelInfo = modelManager.getModelInfoLoader().getModelInfo(modelInfoIdentifier);
        // Look for a ClassInfo element matching primaryDataType for the context
        List<ClassInfo> classInfos = getClassInfos(primaryDataType, modelInfo);
        if (!classInfos.isEmpty()) {
            if (classInfos.size() == 1) {
                ClassInfo classInfo = classInfos.get(0);
                List<ClassInfoElement> elements = classInfo.getElement().stream().filter(x -> x.getName().equals(primaryKeyColumn)).collect(Collectors.toList());
                // check base type
                String baseType = classInfo.getBaseType();
                if (classInfo.getBaseType() != null) {
                    List<ClassInfo> baseClassInfos = getClassInfos(baseType, modelInfo);
                    baseClassInfos.stream().map(ClassInfo::getElement).flatMap(List::stream).filter(element -> element.getName().equals(primaryKeyColumn)).forEach(elements::add);
                }
                // check choice types
                Collection<String> choiceTypes = ModelUtils.getChoiceTypeNames(classInfo);
                choiceTypes.stream().map(type -> getClassInfos(type, modelInfo)).flatMap(List::stream).map(ClassInfo::getElement).flatMap(List::stream).filter(element -> element.getName().equals(primaryKeyColumn)).findFirst().ifPresent(elements::add);
                // A future ModelInfo file may contain the information
                if (elements.isEmpty()) {
                    continue;
                } else if (elements.size() == 1) {
                    String elementType = elements.get(0).getElementType();
                    // store it
                    if (keyType == null) {
                        keyType = getSparkTypeForSystemValue(elementType);
                    } else {
                        throw new IllegalArgumentException("Multiple definitions found for " + primaryDataType + "." + primaryKeyColumn + " in the provided ModelInfo files. Cannot infer key type for context: " + contextName);
                    }
                } else if (elements.size() > 1) {
                    throw new IllegalArgumentException("ModelInfo " + modelInfoIdentifier + " contains multiple element definitions for " + primaryKeyColumn + " for type " + primaryDataType);
                }
            } else {
                throw new IllegalArgumentException("ModelInfo " + modelInfoIdentifier + " contains multiple definitions for type " + primaryDataType);
            }
        }
    }
    if (keyType == null) {
        throw new IllegalArgumentException("Could not locate type information for " + primaryDataType + "." + primaryKeyColumn + " in the provided ModelInfo files. Cannot infer key type for context: " + contextName);
    }
    return new Tuple2<>(contextDefinition.getPrimaryKeyColumn(), keyType);
}
Also used : DataType(org.apache.spark.sql.types.DataType) ModelInfo(org.hl7.elm_modelinfo.r1.ModelInfo) CqlToElmTranslator(com.ibm.cohort.cql.translation.CqlToElmTranslator) HashMap(java.util.HashMap) Format(com.ibm.cohort.cql.library.Format) ExpressionDef(org.cqframework.cql.elm.execution.ExpressionDef) ClassInfo(org.hl7.elm_modelinfo.r1.ClassInfo) HashSet(java.util.HashSet) ContextDefinition(com.ibm.cohort.cql.spark.aggregation.ContextDefinition) Map(java.util.Map) CqlLibraryReader(org.opencds.cqf.cql.engine.execution.CqlLibraryReader) CqlEvaluationRequest(com.ibm.cohort.cql.evaluation.CqlEvaluationRequest) ModelUtils(com.ibm.cohort.cql.spark.optimizer.ModelUtils) DataTypes(org.apache.spark.sql.types.DataTypes) StructField(org.apache.spark.sql.types.StructField) StructType(org.apache.spark.sql.types.StructType) ModelManager(org.cqframework.cql.cql2elm.ModelManager) Collection(java.util.Collection) VersionedIdentifier(org.hl7.elm.r1.VersionedIdentifier) Set(java.util.Set) CqlLibrary(com.ibm.cohort.cql.library.CqlLibrary) CqlLibraryProvider(com.ibm.cohort.cql.library.CqlLibraryProvider) Tuple2(scala.Tuple2) Collectors(java.util.stream.Collectors) CqlLibraryDescriptor(com.ibm.cohort.cql.library.CqlLibraryDescriptor) List(java.util.List) CqlEvaluationRequests(com.ibm.cohort.cql.evaluation.CqlEvaluationRequests) Library(org.cqframework.cql.elm.execution.Library) ClassInfoElement(org.hl7.elm_modelinfo.r1.ClassInfoElement) QName(javax.xml.namespace.QName) ContextDefinitions(com.ibm.cohort.cql.spark.aggregation.ContextDefinitions) ModelInfo(org.hl7.elm_modelinfo.r1.ModelInfo) ContextDefinition(com.ibm.cohort.cql.spark.aggregation.ContextDefinition) ModelManager(org.cqframework.cql.cql2elm.ModelManager) ClassInfoElement(org.hl7.elm_modelinfo.r1.ClassInfoElement) VersionedIdentifier(org.hl7.elm.r1.VersionedIdentifier) Tuple2(scala.Tuple2) DataType(org.apache.spark.sql.types.DataType) List(java.util.List) ClassInfo(org.hl7.elm_modelinfo.r1.ClassInfo)

Example 14 with ModelInfo

use of org.hl7.elm_modelinfo.r1.ModelInfo in project quality-measure-and-cohort-service by Alvearie.

the class ModelUtilsTest method testGetChoiceTypesValid.

@Test
public void testGetChoiceTypesValid() {
    ModelInfo modelInfo = JAXB.unmarshal(new File("src/test/resources/abstract-context/modelinfo/abstract-modelinfo-1.0.0.xml"), ModelInfo.class);
    TypeInfo typeInfo = modelInfo.getTypeInfo().stream().map(ClassInfo.class::cast).filter(classInfo -> classInfo.getName().equals("AlphaNumeric")).findFirst().orElse(null);
    Collection<String> choiceTypes = ModelUtils.getChoiceTypeNames(typeInfo);
    assertThat(choiceTypes, Matchers.containsInAnyOrder("Alpha", "Numeric"));
}
Also used : ModelInfo(org.hl7.elm_modelinfo.r1.ModelInfo) File(java.io.File) TypeInfo(org.hl7.elm_modelinfo.r1.TypeInfo) Test(org.junit.Test)

Example 15 with ModelInfo

use of org.hl7.elm_modelinfo.r1.ModelInfo in project quality-measure-and-cohort-service by Alvearie.

the class ModelUtilsTest method testGetChoiceTypesNone.

@Test
public void testGetChoiceTypesNone() {
    ModelInfo modelInfo = JAXB.unmarshal(new File("src/test/resources/abstract-context/modelinfo/abstract-modelinfo-1.0.0.xml"), ModelInfo.class);
    TypeInfo typeInfo = modelInfo.getTypeInfo().stream().map(ClassInfo.class::cast).filter(classInfo -> classInfo.getName().equals("Alpha")).findFirst().orElse(null);
    Collection<String> choiceTypes = ModelUtils.getChoiceTypeNames(typeInfo);
    assertThat(choiceTypes, Matchers.empty());
}
Also used : ModelInfo(org.hl7.elm_modelinfo.r1.ModelInfo) File(java.io.File) TypeInfo(org.hl7.elm_modelinfo.r1.TypeInfo) Test(org.junit.Test)

Aggregations

ModelInfo (org.hl7.elm_modelinfo.r1.ModelInfo)25 Test (org.junit.Test)13 ClassInfo (org.hl7.elm_modelinfo.r1.ClassInfo)12 QName (javax.xml.namespace.QName)11 File (java.io.File)7 Map (java.util.Map)5 TypeNode (com.ibm.cohort.cql.spark.optimizer.ModelUtils.TypeNode)4 IOException (java.io.IOException)4 Collection (java.util.Collection)4 Collectors (java.util.stream.Collectors)4 VersionedIdentifier (org.hl7.elm.r1.VersionedIdentifier)4 ModelSpecifier (org.hl7.elm_modelinfo.r1.ModelSpecifier)4 TypeInfo (org.hl7.elm_modelinfo.r1.TypeInfo)4 Arrays (java.util.Arrays)3 HashMap (java.util.HashMap)3 HashSet (java.util.HashSet)3 List (java.util.List)3 ObjectMapper (com.fasterxml.jackson.databind.ObjectMapper)2 CqlEvaluationRequest (com.ibm.cohort.cql.evaluation.CqlEvaluationRequest)2 CqlEvaluationRequests (com.ibm.cohort.cql.evaluation.CqlEvaluationRequests)2