001package org.dllearner.kb.dataset;
002
003import org.semanticweb.owlapi.apibinding.OWLManager;
004import org.semanticweb.owlapi.model.OWLOntology;
005import org.semanticweb.owlapi.model.OWLOntologyCreationException;
006import org.semanticweb.owlapi.model.OWLOntologyManager;
007
008import java.io.File;
009import java.util.ArrayList;
010import java.util.Collection;
011
012/**
013 * @author Lorenz Buehmann
014 */
015public class SimpleFileOWLOntologyDataset implements OWLOntologyDataset{
016
017        private final File directory;
018        private final OWLOntologyManager man = OWLManager.createOWLOntologyManager();
019
020        public SimpleFileOWLOntologyDataset(File directory) {
021                this.directory = directory;
022        }
023
024        @Override
025        public Collection<OWLOntology> loadOntologies() {
026                Collection<OWLOntology> ontologies = new ArrayList<>();
027
028                int i = 0;
029                for (File file : directory.listFiles()) {
030                        try {
031                                OWLOntology ontology = man.loadOntologyFromOntologyDocument(file);
032                                ontologies.add(ontology);
033                                if(i == 10) break;
034                        } catch (OWLOntologyCreationException e) {
035                                e.printStackTrace();
036                        }
037                }
038                return ontologies;
039        }
040}