org.nlp2rdf.nif21.impl.NIF21AnnotationUnit Maven / Gradle / Ivy

Go to download
package org.nlp2rdf.nif21.impl;


import com.hp.hpl.jena.rdf.model.Model;
import com.hp.hpl.jena.rdf.model.Resource;
import com.hp.hpl.jena.rdf.model.ResourceFactory;
import com.hp.hpl.jena.vocabulary.RDF;
import org.nlp2rdf.NIFAnnotationUnit;
import org.nlp2rdf.NIFVisitor;
import org.nlp2rdf.bean.NIFBean;
import org.nlp2rdf.nif21.NIF21Format;

public class NIF21AnnotationUnit implements NIFAnnotationUnit, NIF21Format {

    public void accept(NIFVisitor visitor) {
        visitor.visit(this);
    }


    public void add(Model model, NIFBean entity) {

        if (model != null && entity != null && entity.isMention()) {
            Resource contextRes = model.getResource(entity.getContext().getNIF21());

           /* Resource unitRes = model.createResource().addProperty(RDF.type,
                    ResourceFactory.createResource(NIF_PROPERTY_ENTITY_OCCURRENCE));*/

            contextRes.addProperty(model.createProperty(RDF_PROPERTY_IDENTREF),
                    model.createResource(entity.getTaIdentRef()));

            if (entity.hasTypes()) {

                entity.getTypes().stream().forEach(type -> contextRes.addProperty(
                        model.createProperty(RDF_PROPERTY_CLASS_REF),
                        model.createResource(type)));

            }

            if (entity.hasTaClassRef()) {

                entity.getTaClassRef().stream().forEach(type -> contextRes.addProperty(
                        model.createProperty(NIF_CLASS_REF),
                        model.createResource(type)));

            }


            if (entity.getScore() != null) {

                contextRes.addProperty(model.createProperty(RDF_PROPERTY_CONFIDENCE),
                        model.createTypedLiteral(entity.getScore()));

            }
            contextRes.addProperty(model.createProperty(RDF_PROPERTY_ANNOTATOR),
                    model.createProperty(entity.getAnnotator()));

//            contextRes.addProperty(model.createProperty(NIF_PROPERTY_ANNOTATION_UNIT), unitRes);


        }
    }
}