KnowledgeBase.java

Go to the documentation of this file.

package edu.tum.cs.logic;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.StringReader;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Vector;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import edu.tum.cs.logic.parser.FormulaParser;
import edu.tum.cs.logic.parser.ParseException;
import edu.tum.cs.srl.Database;
import edu.tum.cs.util.FileUtil;

public class KnowledgeBase implements Iterable<Formula> {
        protected Vector<Formula> formulas;
        protected HashMap<Formula, Integer> templateIDs;
        
        public KnowledgeBase() {
                formulas = new Vector<Formula>();
                templateIDs = new HashMap<Formula, Integer>();
        }
        
        public KnowledgeBase(String filename) throws IOException, ParseException {
                this();
                // read KB file
                String fileContent = FileUtil.readTextFile(filename);
                // remove comments
                Pattern comments = Pattern.compile("//.*?$|/\\*.*?\\*/", Pattern.MULTILINE | Pattern.DOTALL);
                Matcher matcher = comments.matcher(fileContent);
                fileContent = matcher.replaceAll("");
                // read lines
                BufferedReader br = new BufferedReader(new StringReader(fileContent));
                String line;
                for(;;) {
                        line = br.readLine();
                        if(line == null)
                                break;
                        line = line.trim();
                        if(line.length() == 0)
                                continue;
                        if(line.endsWith("."))
                                formulas.add(FormulaParser.parse(line.substring(0, line.length()-1)));
                        else
                                System.err.println("Warning: Line without terminating period ignored: " + line);
                }
        }
        
        public Vector<Formula> getFormulas() {
                return formulas;
        }

        public KnowledgeBase ground(Database db, WorldVariables worldVars, boolean simplify) throws Exception {
                KnowledgeBase ret = new KnowledgeBase();
                Integer formulaID = 0;
                for(Formula f : formulas) {
                        int i = ret.formulas.size();
                        f.addAllGroundingsTo(ret.formulas, db, worldVars, simplify);
                        for(; i < ret.formulas.size(); i++) {
                                templateIDs.put(ret.formulas.get(i), formulaID);
                        }
                        formulaID++;
                }
                return ret;
        }

        public Iterator<Formula> iterator() {
                return formulas.iterator();
        }
        
        public int size() {
                return formulas.size();
        }
        
        public Integer getTemplateID(Formula f) {
                return templateIDs.get(f);
        }
}

---

srldb
Author(s): Dominik Jain, Stefan Waldherr, Moritz Tenorth
autogenerated on Fri Jan 11 09:58:29 2013