From c3480bb733b8bca976718a3dc2f09a21cb4b1b45 Mon Sep 17 00:00:00 2001
From: RncLsn <rnc.lsn@gmail.com>
Date: Tue, 4 Aug 2015 18:30:46 +0100
Subject: Rules parsing.

---
 src/uk/ac/ox/cs/pagoda/hermit/RuleHelper.java | 83 +++++++++++++++++++++++++--
 1 file changed, 79 insertions(+), 4 deletions(-)

(limited to 'src/uk/ac/ox/cs/pagoda/hermit')

diff --git a/src/uk/ac/ox/cs/pagoda/hermit/RuleHelper.java b/src/uk/ac/ox/cs/pagoda/hermit/RuleHelper.java
index 81c99a4..43c5849 100644
--- a/src/uk/ac/ox/cs/pagoda/hermit/RuleHelper.java
+++ b/src/uk/ac/ox/cs/pagoda/hermit/RuleHelper.java
@@ -2,9 +2,16 @@ package uk.ac.ox.cs.pagoda.hermit;
 
 import org.semanticweb.HermiT.model.*;
 import uk.ac.ox.cs.pagoda.MyPrefixes;
+import uk.ac.ox.cs.pagoda.owl.OWLHelper;
 import uk.ac.ox.cs.pagoda.util.Namespace;
 
+import java.security.InvalidParameterException;
+import java.util.ArrayList;
+
 public class RuleHelper {
+    private static final String OR = "|";
+    private static final String IF = ":-";
+    private static final String AND = ",";
 
 //	public static String abbreviateIRI(String text) {
 //		String prefixName, prefixIRI; 
@@ -37,16 +44,16 @@ public class RuleHelper {
 		boolean lastSpace = true;
 		for (Atom headAtom: clause.getHeadAtoms()) {
 			if ((atomText = getText(headAtom)) == null) continue; 
-			if (!lastSpace)	buf.append(" v ");
+			if (!lastSpace)	buf.append(" ").append(OR).append(" ");
 			buf.append(atomText);
 			lastSpace = false;
 		}
-		buf.append(" :- ");
+        buf.append(" ").append(IF).append(" ");
 		lastSpace = true;
 		for (Atom bodyAtom: clause.getBodyAtoms()) {
 //		for (String str: strs[1].split(", ")) {
 			if ((atomText = getText(bodyAtom)) == null) continue; 
-			if (!lastSpace) buf.append(", ");
+			if (!lastSpace) buf.append(AND).append(" ");
 			buf.append(atomText);
 			lastSpace = false;
 		}
@@ -74,7 +81,7 @@ public class RuleHelper {
 			else builder.append(getText(p));
 			builder.append("("); 
 			builder.append(getText(atom.getArgument(0))); 
-			builder.append(","); 
+			builder.append(",");
 			builder.append(getText(atom.getArgument(1))); 
 			builder.append(")"); 
 		}
@@ -95,4 +102,72 @@ public class RuleHelper {
 		return MyPrefixes.PAGOdAPrefixes.abbreviateIRI(t.toString());
 	}
 
+	public static Term parseTerm(String s) {
+		s = s.trim();
+		if(s.startsWith("?")) return Variable.create(s.substring(1));
+		return Individual.create(MyPrefixes.PAGOdAPrefixes.expandIRI(s));
+	}
+
+    public static Atom parseAtom(String s) {
+        s = s.trim();
+
+        String[] split = s.split("\\(");
+        String predicateIri = OWLHelper.removeAngles(MyPrefixes.PAGOdAPrefixes.expandText(split[0]));
+        String[] predicateArgs = split[1].substring(0, split[1].length() - 1).split(",");
+        int numOfargs = predicateArgs.length;
+        Term terms[] = new Term[predicateArgs.length];
+        for (int i = 0; i < terms.length; i++)
+            terms[i] = parseTerm(predicateArgs[i]);
+        if(numOfargs == 1) {
+            AtomicConcept atomicConcept = AtomicConcept.create(predicateIri);
+            return Atom.create(atomicConcept, terms);
+        }
+        else if(numOfargs == 2) {
+            AtomicRole atomicRole = AtomicRole.create(predicateIri);
+            return Atom.create(atomicRole, terms);
+        }
+        else
+            throw new InvalidParameterException();
+        // TODO? add equality (owl:sameAs)?
+    }
+
+    public static DLClause parseClause(String s) {
+        s = s.trim();
+        if(s.endsWith(".")) s = s.substring(0, s.length()-1).trim();
+
+        String[] headAndBody = s.split(IF);
+        String[] headAtomsStr = splitAtoms(headAndBody[0], OR);
+        String[] bodyAtomsStr = splitAtoms(headAndBody[1], AND);
+
+        Atom[] headAtoms = new Atom[headAtomsStr.length];
+        Atom[] bodyAtoms = new Atom[bodyAtomsStr.length];
+        for (int i = 0; i < headAtoms.length; i++)
+            headAtoms[i] = parseAtom(headAtomsStr[i]);
+        for (int i = 0; i < bodyAtoms.length; i++)
+            bodyAtoms[i] = parseAtom(bodyAtomsStr[i]);
+
+        return DLClause.create(headAtoms, bodyAtoms);
+    }
+
+    private static String[] splitAtoms(String s, String operator) {
+        char op = operator.charAt(0);
+        ArrayList<String> result = new ArrayList<>();
+
+        int b = 0;
+        boolean betweenParenthesis = false;
+        for (int i = 0; i < s.length(); i++) {
+            if(s.charAt(i) == '(')
+                betweenParenthesis = true;
+            else if(s.charAt(i) == ')')
+                betweenParenthesis = false;
+            else if(s.charAt(i) == op && !betweenParenthesis) {
+                result.add(s.substring(b, i));
+                b = i + 1;
+            }
+        }
+        if(b < s.length()) result.add(s.substring(b));
+
+        return result.toArray(new String[0]);
+    }
+
 }
-- 
cgit v1.2.3