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}