org.protege.owl.codegeneration.impl.CodeGenerationHelper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of code-generation Show documentation
Generate Java code from an OWL ontology in the Protege Desktop editing environment.
The newest version!
package org.protege.owl.codegeneration.impl;

import java.lang.reflect.Constructor;
import java.util.Collection;
import java.util.HashSet;
import java.util.Set;

import org.protege.owl.codegeneration.CodeGenerationRuntimeException;
import org.protege.owl.codegeneration.HandledDatatypes;
import org.protege.owl.codegeneration.WrappedIndividual;
import org.protege.owl.codegeneration.inference.CodeGenerationInference;
import org.semanticweb.owlapi.model.IRI;
import org.semanticweb.owlapi.model.OWLAxiom;
import org.semanticweb.owlapi.model.OWLDataFactory;
import org.semanticweb.owlapi.model.OWLDataProperty;
import org.semanticweb.owlapi.model.OWLIndividual;
import org.semanticweb.owlapi.model.OWLLiteral;
import org.semanticweb.owlapi.model.OWLNamedIndividual;
import org.semanticweb.owlapi.model.OWLObjectProperty;
import org.semanticweb.owlapi.model.OWLOntology;
import org.semanticweb.owlapi.model.OWLOntologyManager;

public class CodeGenerationHelper {
    private OWLOntology owlOntology;    
    private OWLDataFactory owlDataFactory;
    private OWLOntologyManager manager;
    private CodeGenerationInference inference;
    
    
    public CodeGenerationHelper(CodeGenerationInference inference) {
        this.inference = inference;
        this.owlOntology = inference.getOWLOntology();        
        manager = owlOntology.getOWLOntologyManager();
        owlDataFactory = manager.getOWLDataFactory();
    }
    
    public OWLOntology getOwlOntology() {
        return owlOntology;
    }
    
    public  Collection getPropertyValues(OWLNamedIndividual i, OWLObjectProperty p, Class c) {
        try {
            Constructor constructor = c.getConstructor(CodeGenerationInference.class, IRI.class);
            Set results = new HashSet();
            for (OWLNamedIndividual j : inference.getPropertyValues(i, p)) {
                results.add(constructor.newInstance(inference, j.getIRI()));
            }
            return results;
        }
        catch (Exception e) {
            throw new CodeGenerationRuntimeException(e);
        }
    }
 
    public void addPropertyValue(OWLNamedIndividual i, OWLObjectProperty p, WrappedIndividual j) {
    	OWLAxiom axiom = owlDataFactory.getOWLObjectPropertyAssertionAxiom(p, i, j.getOwlIndividual());
    	manager.addAxiom(owlOntology, axiom);
    }
    
    public void removePropertyValue(OWLNamedIndividual i, OWLObjectProperty p, WrappedIndividual j) {
    	OWLAxiom axiom = owlDataFactory.getOWLObjectPropertyAssertionAxiom(p, i, j.getOwlIndividual());
    	for (OWLOntology imported : owlOntology.getImportsClosure()) {
        	manager.removeAxiom(imported, axiom);
    	}
    }
    
    public  Collection getPropertyValues(OWLNamedIndividual i, OWLDataProperty p, Class c) {
        Set results = new HashSet();
        for (OWLLiteral l : inference.getPropertyValues(i, p)) {
            results.add(c.cast(getObjectFromLiteral(l)));
        }
        return results;
    }
    
    public void addPropertyValue(OWLNamedIndividual i, OWLDataProperty p, Object o) {
    	OWLLiteral literal = getLiteralFromObject(owlDataFactory, o);
    	if (literal != null) {
    		OWLAxiom axiom = owlDataFactory.getOWLDataPropertyAssertionAxiom(p, i, literal);
    		manager.addAxiom(owlOntology, axiom);
    	}
    	else {
    		throw new CodeGenerationRuntimeException("Invalid type for property value object " + o);
    	}
    }

    public void removePropertyValue(OWLNamedIndividual i, OWLDataProperty p, Object o) {
    	OWLLiteral literal = getLiteralFromObject(owlDataFactory, o);
    	if (literal != null) {
    		OWLAxiom axiom = owlDataFactory.getOWLDataPropertyAssertionAxiom(p, i, literal);
    		manager.removeAxiom(owlOntology, axiom);
    	}
    	else {
    		throw new CodeGenerationRuntimeException("Invalid type for property value object " + o);
    	}
    }
    
    public static Object getObjectFromLiteral(OWLLiteral literal) {
    	Object o = null;
    	for (HandledDatatypes handled : HandledDatatypes.values()) {
    		if (handled.isMatch(literal.getDatatype())) {
    			o = handled.getObject(literal);
    			break;
    		}
    	}
    	if (o == null) {
    		o = literal;
    	}
    	return o;
    }
    
    public static OWLLiteral getLiteralFromObject(OWLDataFactory owlDataFactory, Object o) {
    	OWLLiteral literal = null;
    	if (o instanceof OWLLiteral) {
    		literal = (OWLLiteral) o;
    	}
    	else {
    		for (HandledDatatypes handled : HandledDatatypes.values()) {
    			literal = handled.getLiteral(owlDataFactory, o);
    			if (literal != null) {
    				break;
    			}
    		}
    	}
    	return literal;
    }
    
}