org.opencds.cqf.tooling.modelinfo.fhir.FHIRClassInfoBuilder Maven / Gradle / Ivy

Go to download
package org.opencds.cqf.tooling.modelinfo.fhir;

import java.util.Collection;
import java.util.Map;

import org.hl7.elm_modelinfo.r1.ClassInfo;
import org.hl7.elm_modelinfo.r1.TypeInfo;
import org.hl7.fhir.r4.model.StructureDefinition;
import org.hl7.fhir.r4.model.StructureDefinition.StructureDefinitionKind;
import org.hl7.fhir.r4.model.StructureDefinition.TypeDerivationRule;
import org.opencds.cqf.tooling.modelinfo.ClassInfoBuilder;

public class FHIRClassInfoBuilder extends ClassInfoBuilder {

    public FHIRClassInfoBuilder(Map structureDefinitions) {
        super(new FHIRClassInfoSettings(), structureDefinitions);
    }

    @Override
    protected void innerBuild() {
        if (!this.settings.useCQLPrimitives) {
            System.out.println("Building Primitives");
            this.buildFor("FHIR", (x -> x.getKind() == StructureDefinitionKind.PRIMITIVETYPE));
        }

        System.out.println("Building ComplexTypes");
        this.buildFor("FHIR", (x -> (x.getKind() == StructureDefinitionKind.COMPLEXTYPE && (x.getBaseDefinition() == null
                || !x.getBaseDefinition().equals("http://hl7.org/fhir/StructureDefinition/Extension")))
                && !x.getUrl().equals("http://hl7.org/fhir/StructureDefinition/elementdefinition-de")));

        System.out.println("Building Resources");
        this.buildFor("FHIR", (x -> x.getKind() == StructureDefinitionKind.RESOURCE
                && (!x.hasDerivation() || x.getDerivation() == TypeDerivationRule.SPECIALIZATION)));
    }

    @Override
    public void afterBuild() {
        //Clean up Content Reference Specifiers
        Collection typeInfoValues = this.getTypeInfos().values();
        typeInfoValues.stream().map(x -> (ClassInfo)x).forEach(
                x -> x.getElement().stream()
                        .filter(y -> this.hasContentReferenceTypeSpecifier(y))
                        .forEach(y -> this.fixupContentReferenceSpecifier("FHIR", y))
        );
    }
}