display progress in materialization

Author: LorenzBuehmann

components-core/src/main/java/org/dllearner/reasoning/ClosedWorldReasoner.java

@@ -26,6 +26,7 @@
 import org.dllearner.core.annotations.NoConfigOption;
 import org.dllearner.core.config.ConfigOption;
 import org.dllearner.kb.OWLAPIOntology;
+import org.dllearner.utilities.Helper;
 import org.dllearner.utilities.MapUtils;
 import org.dllearner.utilities.OWLAPIUtils;
 import org.joda.time.DateTime;
@@ -45,6 +46,7 @@
 import java.io.*;
 import java.util.*;
 import java.util.Map.Entry;
+import java.util.concurrent.atomic.AtomicInteger;
 import java.util.concurrent.atomic.AtomicReference;
 import java.util.function.Function;
 import java.util.stream.Collectors;
@@ -292,8 +294,15 @@ private void materialize() {
 
         individuals = (TreeSet<OWLIndividual>) baseReasoner.getIndividuals();
 
-        logger.debug("materialising concepts");
+        int totalEntities = baseReasoner.getClasses().size() +
+                            baseReasoner.getObjectProperties().size() +
+                            baseReasoner.getDatatypeProperties().size();
+
+        AtomicInteger i = new AtomicInteger();
+
+        logger.info("materialising concepts");
         baseReasoner.getClasses().stream().filter(cls -> !cls.getIRI().isReservedVocabulary()).forEach(cls -> {
+            Helper.displayProgressPercentage(i.getAndIncrement(), totalEntities);
             TreeSet<OWLIndividual> pos = (TreeSet<OWLIndividual>) baseReasoner.getIndividuals(cls);
             classInstancesPos.put(cls, pos);
 
@@ -337,13 +346,18 @@ private void materialize() {
 //        }
 
         // materialize the object property facts
-        logger.debug("materialising object properties ...");
-        baseReasoner.getObjectProperties().forEach(p -> opPos.put(p, baseReasoner.getPropertyMembers(p)));
-        logger.debug("finished materialising object properties.");
+        logger.info("materialising object properties ...");
+        baseReasoner.getObjectProperties().forEach(p -> {
+            Helper.displayProgressPercentage(i.getAndIncrement(), totalEntities);
+            opPos.put(p, baseReasoner.getPropertyMembers(p));
+        });
 
         // materialize the data property facts
-        logger.debug("materialising datatype properties");
-        baseReasoner.getDatatypeProperties().forEach(p -> dpPos.put(p, baseReasoner.getDatatypeMembers(p)));
+        logger.info("materialising datatype properties");
+        baseReasoner.getDatatypeProperties().forEach(p -> {
+            Helper.displayProgressPercentage(i.getAndIncrement(), totalEntities);
+            dpPos.put(p, baseReasoner.getDatatypeMembers(p));
+        });
 
         for (OWLDataProperty dp : baseReasoner.getBooleanDatatypeProperties()) {
             bdPos.put(dp, (TreeSet<OWLIndividual>) baseReasoner.getTrueDatatypeMembers(dp));

components-core/src/main/java/org/dllearner/utilities/Helper.java

@@ -236,4 +236,34 @@ public static OWLClass checkConcepts(AbstractReasonerComponent rs, Set<OWLClass>
 		return null;
 	}
 
+	public static void displayProgressPercentage(int done, int total) {
+		int size = 5;
+		String iconLeftBoundary = "[";
+		String iconDone = "=";
+		String iconRemain = ".";
+		String iconRightBoundary = "]";
+
+		if (done > total) {
+			throw new IllegalArgumentException();
+		}
+		int donePercents = (100 * done) / total;
+		int doneLength = size * donePercents / 100;
+
+		StringBuilder bar = new StringBuilder(iconLeftBoundary);
+		for (int i = 0; i < size; i++) {
+			if (i < doneLength) {
+				bar.append(iconDone);
+			} else {
+				bar.append(iconRemain);
+			}
+		}
+		bar.append(iconRightBoundary);
+
+		System.out.print("\r" + bar + " " + donePercents + "%");
+
+		if (done == total) {
+			System.out.print("\n");
+		}
+	}
+
 }