VariableLogicCoupling.java

Go to the documentation of this file.

/*
 * Created on Oct 27, 2009
 *
 * TODO To change the template for this generated file go to
 * Window - Preferences - Java - Code Style - Code Templates
 */
package edu.tum.cs.srl.bayesnets.bln.coupling;

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

import edu.ksu.cis.bnj.ver3.core.BeliefNode;
import edu.ksu.cis.bnj.ver3.core.Discrete;
import edu.tum.cs.logic.GroundAtom;
import edu.tum.cs.logic.GroundLiteral;
import edu.tum.cs.logic.PossibleWorld;
import edu.tum.cs.logic.WorldVariables;
import edu.tum.cs.logic.WorldVariables.Block;

public class VariableLogicCoupling {
        protected WorldVariables worldVars;
        protected HashMap<BeliefNode, IVariableLogicCoupler> variable2groundAtomLookup;
        protected HashMap<GroundAtom, BeliefNode> groundAtom2variable;
        
        public VariableLogicCoupling() {
                worldVars = new WorldVariables();
                variable2groundAtomLookup = new HashMap<BeliefNode, IVariableLogicCoupler>();
                groundAtom2variable = new HashMap<GroundAtom, BeliefNode>();
        }
        
        public void addBooleanVariable(BeliefNode var, String predicateName, String[] params) {
                GroundAtom ga = new GroundAtom(predicateName, params);
                worldVars.add(ga);
                variable2groundAtomLookup.put(var, new BooleanVariable(ga));
                groundAtom2variable.put(ga, var);
        }
        
        public void addBlockVariable(BeliefNode var, Discrete dom, String functionName, String[] params) {
                String[] atomParams = new String[params.length+1];
                for(int i = 0; i < params.length; i++)
                        atomParams[i] = params[i];
                Vector<GroundAtom> block = new Vector<GroundAtom>(dom.getOrder());
                for(int i = 0; i < dom.getOrder(); i++) {
                        atomParams[atomParams.length-1] = dom.getName(i);
                        GroundAtom ga = new GroundAtom(functionName, atomParams.clone());
                        block.add(ga);                          
                } 
                Block b = worldVars.addBlock(block); 
                for(GroundAtom ga : b)
                        groundAtom2variable.put(ga, var);
                variable2groundAtomLookup.put(var, new BlockVariable(b));
        }
        
        public GroundLiteral getGroundLiteral(BeliefNode var, int domIdx) {
                IVariableLogicCoupler vlc = variable2groundAtomLookup.get(var);
                if(vlc == null)
                        throw new RuntimeException("Variable " + var + " has no logical coupling!");
                return vlc.getGroundLiteral(domIdx, worldVars);
        }
        
        public WorldVariables getWorldVars() {
                return worldVars;
        }
        
        public BeliefNode getVariable(GroundAtom gndAtom) {
                return groundAtom2variable.get(gndAtom);
        }
        
        public int getVariableValue(BeliefNode var, PossibleWorld w) {
                return variable2groundAtomLookup.get(var).getValue(w);
        }
        
        public void setVariableValue(BeliefNode var, int domIdx, PossibleWorld w) {
                variable2groundAtomLookup.get(var).setValue(w, domIdx);
        }

        public boolean hasCoupling(BeliefNode var) {
                return variable2groundAtomLookup.containsKey(var);
        }
        
        public Set<BeliefNode> getCoupledVariables() {
                return variable2groundAtomLookup.keySet();
        }
        
        public Set<GroundAtom> getCoupledGroundAtoms() {
                return groundAtom2variable.keySet();
        }
        
        public Iterable<String> getOriginalParams(BeliefNode var) {
                return variable2groundAtomLookup.get(var).getOriginalParams();
        }
}

---

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