1 files changed, 79 insertions, 4 deletions

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]);
+    }
+
 }