def load_pddl(domain_file, problem_dir):
"""
Parse domain and problem PDDL files.
Parameters
----------
domain_file : str
Path to a PDDL domain file.
problem_dir : str
Path to a directory of PDDL problem files.
Returns
-------
domain : PDDLDomainParser
problems : [ PDDLProblemParser ]
"""
domain = PDDLDomainParser(domain_file)
problems = []
problem_files = [
f for f in glob.glob(os.path.join(problem_dir, "*.pddl"))
]
for problem_file in sorted(problem_files):
problem = PDDLProblemParser(problem_file, domain.domain_name,
domain.types, domain.predicates)
problems.append(problem)
return domain, problems
def load_pddl(domain_file, problem_dir, operators_as_actions=False):
"""
Parse domain and problem PDDL files.
Parameters
----------
domain_file : str
Path to a PDDL domain file.
problem_dir : str
Path to a directory of PDDL problem files.
operators_as_actions : bool
See class docstirng.
Returns
-------
domain : PDDLDomainParser
problems : [ PDDLProblemParser ]
"""
domain = PDDLDomainParser(
domain_file,
expect_action_preds=(not operators_as_actions),
operators_as_actions=operators_as_actions)
problems = []
problem_files = [
f for f in glob.glob(os.path.join(problem_dir, "*.pddl"))
]
for problem_file in sorted(problem_files):
problem = PDDLProblemParser(problem_file, domain.domain_name,
domain.types, domain.predicates,
domain.actions, domain.constants)
problems.append(problem)
return domain, problems
def test_hierarchical_spaces():
dir_path = os.path.dirname(os.path.realpath(__file__))
domain_file = os.path.join(dir_path, 'pddl',
'hierarchical_type_test_domain.pddl')
problem_file = os.path.join(dir_path, 'pddl',
'hierarchical_type_test_domain',
'hierarchical_type_test_problem.pddl')
domain = PDDLDomainParser(domain_file)
problem = PDDLProblemParser(problem_file, domain.domain_name, domain.types,
domain.predicates, domain.actions)
actions = list(domain.actions)
action_predicates = [domain.predicates[a] for a in actions]
space = LiteralSpace(set(domain.predicates.values()) -
set(action_predicates),
type_to_parent_types=domain.type_to_parent_types)
all_ground_literals = space.all_ground_literals(
State(problem.initial_state, problem.objects, problem.goal))
ispresent = Predicate("ispresent", 1, [Type("entity")])
islight = Predicate("islight", 1, [Type("object")])
isfurry = Predicate("isfurry", 1, [Type("animal")])
ishappy = Predicate("ishappy", 1, [Type("animal")])
attending = Predicate("attending", 2, [Type("animal"), Type("object")])
nomsy = Type("jindo")("nomsy")
rover = Type("corgi")("rover")
rene = Type("cat")("rene")
block1 = Type("block")("block1")
block2 = Type("block")("block2")
cylinder1 = Type("cylinder")("cylinder1")
assert all_ground_literals == {
ispresent(nomsy),
ispresent(rover),
ispresent(rene),
ispresent(block1),
ispresent(block2),
ispresent(cylinder1),
islight(block1),
islight(block2),
islight(cylinder1),
isfurry(nomsy),
isfurry(rover),
isfurry(rene),
ishappy(nomsy),
ishappy(rover),
ishappy(rene),
attending(nomsy, block1),
attending(nomsy, block2),
attending(nomsy, cylinder1),
attending(rover, block1),
attending(rover, block2),
attending(rover, cylinder1),
attending(rene, block1),
attending(rene, block2),
attending(rene, cylinder1),
}
print("Test passed.")
def test_hierarchical_types():
dir_path = os.path.dirname(os.path.realpath(__file__))
domain_file = os.path.join(dir_path, 'pddl',
'hierarchical_type_test_domain.pddl')
problem_file = os.path.join(dir_path, 'pddl',
'hierarchical_type_test_domain',
'hierarchical_type_test_problem.pddl')
domain = PDDLDomainParser(domain_file)
problem = PDDLProblemParser(problem_file, domain.domain_name, domain.types,
domain.predicates)
assert set(domain.types.keys()) == {
Type("dog"),
Type("cat"),
Type("animal"),
Type("block"),
Type("cylinder"),
Type("jindo"),
Type("corgi"),
Type("object"),
Type("entity")
}
assert domain.type_hierarchy == {
Type("animal"): {Type("dog"), Type("cat")},
Type("dog"): {Type("jindo"), Type("corgi")},
Type("object"): {Type("block"), Type("cylinder")},
Type("entity"): {Type("object"), Type("animal")},
}
assert domain.type_to_parent_types == {
Type("entity"): {Type("entity")},
Type("object"): {Type("object"), Type("entity")},
Type("animal"): {Type("animal"), Type("entity")},
Type("dog"): {Type("dog"), Type("animal"),
Type("entity")},
Type("cat"): {Type("cat"), Type("animal"),
Type("entity")},
Type("corgi"):
{Type("corgi"),
Type("dog"),
Type("animal"),
Type("entity")},
Type("jindo"):
{Type("jindo"),
Type("dog"),
Type("animal"),
Type("entity")},
Type("block"): {Type("block"),
Type("object"),
Type("entity")},
Type("cylinder"): {Type("cylinder"),
Type("object"),
Type("entity")},
}
print("Test passed.")
def _create_problem_file(self, raw_problem_fname, use_cache=True):
if (not use_cache) or (raw_problem_fname not in self._problem_files):
problem_fname = os.path.split(raw_problem_fname)[-1]
problem_fname = problem_fname.split('.pddl')[0]
problem_fname += '_{}_with_diffs_{}.pddl'.format(
self.domain_name, random.randint(0, 9999999))
problem_fname = os.path.join('/tmp', problem_fname)
# Parse raw problem
problem_parser = PDDLProblemParser(raw_problem_fname,
self.domain_name.lower(),
self._types, self._predicates,
self._actions)
new_initial_state = set(problem_parser.initial_state)
# Add actions
action_lits = set(
self._action_space.all_ground_literals(State(
new_initial_state, problem_parser.objects,
problem_parser.goal),
valid_only=False))
new_initial_state |= action_lits
# Add 'Different' pairs for each pair of objects
Different = Predicate('Different', 2)
for obj1 in problem_parser.objects:
for obj2 in problem_parser.objects:
if obj1 == obj2:
continue
# if obj1.var_type != obj2.var_type:
# continue
diff_lit = Different(obj1, obj2)
new_initial_state.add(diff_lit)
# Write out new temporary problem file
problem_parser.initial_state = frozenset(new_initial_state)
problem_parser.write(problem_fname)
# Add to cache
self._problem_files[raw_problem_fname] = problem_fname
return self._problem_files[raw_problem_fname]
def integration_test():
dir_path = os.path.dirname(os.path.realpath(__file__))
domain_file = os.path.join(dir_path, 'pddl', 'test_domain.pddl')
problem_file = os.path.join(dir_path, 'pddl', 'test_domain', 'test_problem.pddl')
domain = PDDLDomainParser(domain_file)
problem = PDDLProblemParser(problem_file, domain.domain_name, domain.types,
domain.predicates)
## Check domain
type1 = Type('type1')
type2 = Type('type2')
# Action predicates
action_pred = Predicate('actionpred', 1, [type1])
# Predicates
pred1 = Predicate('pred1', 1, [type1])
pred2 = Predicate('pred2', 1, [type2])
pred3 = Predicate('pred3', 1, [type1, type2, type2])
assert set(domain.predicates.values()) == { pred1, pred2, pred3, action_pred }
assert domain.actions == { action_pred.name }
# Operators
assert len(domain.operators) == 1
operator1 = Predicate('action1', 4, [type1, type1, type2, type2])
assert operator1 in domain.operators
operator = domain.operators[operator1]
# Operator parameters
assert len(operator.params) == 4
assert operator.params[0] == type1('?a')
assert operator.params[1] == type1('?b')
assert operator.params[2] == type2('?c')
assert operator.params[3] == type2('?d')
# Operator preconditions (set of Literals)
assert len(operator.preconds.literals) == 4
assert set(operator.preconds.literals) == { action_pred('?b'), pred1('?b'),
pred3('?a', '?c', '?d'), Not(pred2('?c')) }
# Operator effects (set of Literals)
assert len(operator.effects.literals) == 3
assert set(operator.effects.literals) == { Anti(pred2('?d')), pred2('?c'),
pred3('?b', '?d', '?c')}
## Check problem
# Objects
assert set(problem.objects) == {type1('a1'), type1('a2'), type1('b1'),
type1('b2'), type1('b3'), type2('c1'), type2('c2'), type2('d1'),
type2('d2'), type2('d3')}
# Goal
assert isinstance(problem.goal, LiteralConjunction)
assert set(problem.goal.literals) == {pred2('c2'), pred3('b1', 'c1', 'd1')}
# Init
assert problem.initial_state == { pred1('b2'), pred2('c1'),
pred3('a1', 'c1', 'd1'), pred3('a2', 'c2', 'd2'), action_pred('a1'),
action_pred('a2'), action_pred('b1'), action_pred('b2')}
print("Test passed.")
from transform.parse_func import parse_objects, parse_initial_state, parse_goal
import os
TAR_DIR = "transform/puzzles/hard"
if __name__ == "__main__":
cnt = 0
for file in sorted(os.listdir(TAR_DIR)):
sok = SokobanGame()
boards = sok.new_board(os.path.join(TAR_DIR, file))
for b in boards:
domian_file = "/Users/yangchen/PycharmProjects/pddlgym/pddlgym/pddl/sokoban.pddl"
problem_file = "/Users/yangchen/PycharmProjects/pddlgym/pddlgym/pddl/sokoban_test/task02.pddl"
domain = PDDLDomainParser(domian_file)
problem = PDDLProblemParser(problem_file, domain.domain_name,
domain.types, domain.predicates,
domain.actions, domain.constants)
problem.objects = parse_objects(b, domain.types)
problem.goal = parse_goal(b, domain.types, domain.predicates)
problem.initial_state = parse_initial_state(
b, domain.types, domain.predicates)
with open(
'/Users/yangchen/PycharmProjects/pddlgym/pddlgym/pddl/sokoban/{}.pddl'
.format(cnt), 'w') as f:
PDDLProblemParser.create_pddl_file(f, problem.objects,
problem.initial_state,
problem.problem_name,
problem.domain_name,
problem.goal)
def _create_problem_file(self, raw_problem_fname, use_cache=True):
if (not use_cache) or (raw_problem_fname not in self._problem_files):
problem_fname = os.path.split(raw_problem_fname)[-1]
problem_fname = problem_fname.split('.pddl')[0]
problem_fname += '_{}_with_diffs_{}.pddl'.format(
self.domain_name, random.randint(0, 9999999))
problem_fname = os.path.join('/tmp', problem_fname)
# Parse raw problem
action_names = [
] # purposely empty b/c we WANT action literals in there
problem_parser = PDDLProblemParser(raw_problem_fname,
self.domain_name.lower(),
self._types, self._predicates,
action_names)
# Add action literals (in case they're not already present in the initial state)
# which will be true when the original domain uses operators_as_actions
init_state = State(problem_parser.initial_state,
problem_parser.objects, None)
act_lits = self._action_space.all_ground_literals(init_state,
valid_only=False)
problem_parser.initial_state = frozenset(
act_lits | problem_parser.initial_state)
# Add 'Different' pairs for each pair of objects
Different = Predicate('Different', 2)
init_state = set(problem_parser.initial_state)
for obj1 in problem_parser.objects:
for obj2 in problem_parser.objects:
if obj1 == obj2:
continue
# if obj1.var_type != obj2.var_type:
# continue
diff_lit = Different(obj1, obj2)
init_state.add(diff_lit)
problem_parser.initial_state = frozenset(init_state)
# Also add 'different' pairs for goal if it's existential
if isinstance(problem_parser.goal, Exists):
diffs = []
for var1 in problem_parser.goal.variables:
for var2 in problem_parser.goal.variables:
if var1 == var2:
continue
diffs.append(Different(var1, var2))
problem_parser.goal = Exists(
problem_parser.goal.variables,
type(problem_parser.goal.body)(
problem_parser.goal.body.literals + diffs))
# If no objects, write a dummy one to make FF not crash.
if not problem_parser.objects:
problem_parser.objects.append("DUMMY")
# Write out new temporary problem file
problem_parser.write(problem_fname)
# Add to cache
self._problem_files[raw_problem_fname] = (problem_fname,
problem_parser.objects)
return self._problem_files[raw_problem_fname]