WorldVariables.java

Go to the documentation of this file.

package edu.tum.cs.logic;

import java.util.HashMap;
import java.util.Iterator;
import java.util.Set;
import java.util.Vector;

import edu.tum.cs.util.StringTool;

public class WorldVariables implements Iterable<GroundAtom> {
        protected HashMap<String, GroundAtom> vars;
        protected HashMap<Integer, Block> var2block;
        protected HashMap<Integer, GroundAtom> varsByIndex;

        public WorldVariables() {
                vars = new HashMap<String, GroundAtom>();
                var2block = new HashMap<Integer, Block>();
                varsByIndex = new HashMap<Integer, GroundAtom>();
        }

        public void add(GroundAtom gndAtom) {
                gndAtom.setIndex(vars.size());
                vars.put(gndAtom.toString(), gndAtom);
                varsByIndex.put(gndAtom.index, gndAtom);
        }

        public Block addBlock(Vector<GroundAtom> block) {
                Block b = new Block(block);
                for(GroundAtom ga : block) {
            if(!vars.containsKey(ga.toString()))
                add(ga);
            var2block.put(ga.index, b);
                }
                return b;
        }

        public GroundAtom get(String gndAtom) {
                return vars.get(gndAtom);
        }

        public GroundAtom get(Integer index) {
                return varsByIndex.get(index);
        }

        public Block getBlock(Integer idxGA) {
                return var2block.get(idxGA);
        }

        public int size() {
                return vars.size();
        }
        
        public Set<String> getVariableStrings() {
                return vars.keySet();
        }

        public String toString() {
                return "<" + StringTool.join(" \n", vars.keySet()) + ">";
        }

        public static class Block implements Iterable<GroundAtom> {
                protected Vector<GroundAtom> gndAtoms;
                protected GroundAtom trueOne;

                public Block(Vector<GroundAtom> list) {
                        gndAtoms = list;
                        trueOne = null;
                }

                public Iterator<GroundAtom> iterator() {
                        return gndAtoms.iterator();
                }

                public GroundAtom getTrueOne(IPossibleWorld w) {
                        //if(trueOne == null) {
                                for(GroundAtom ga : gndAtoms)
                                        if(ga.isTrue(w)) {
                                                trueOne = ga;
                                                break;
                                        }
                        //}
                        return trueOne;
                }

                /*
                public void setTrueOne(GroundAtom ga) {
                        trueOne = ga;
                }*/

                public int size() {
                        return gndAtoms.size();
                }

                public GroundAtom get(int index) {
                        return gndAtoms.get(index);
                }
                
                public int indexOf(GroundAtom gndAtom) {
                        return gndAtoms.indexOf(gndAtom);
                }
        }

        public Iterator<GroundAtom> iterator() {                
                return this.vars.values().iterator();
        }
}

---

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