demo.py

#
# Copyright 2015 Hernán M. Foffani
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
#

import sys
import pddlpy
import argparse

parser = argparse.ArgumentParser(description='Demo PDDLPY.')
parser.add_argument('demonumber', metavar='N', type=int, nargs='+',
                    help='the number(s) of the demo')

def run_demo(demonumber):
    print('===')
    print(f'Run DEMO {demonumber}')
    domainfile = "./examples-pddl/domain-0%d.pddl" % demonumber
    problemfile = "./examples-pddl/problem-0%d.pddl" % demonumber
    domprob = pddlpy.DomainProblem(domainfile, problemfile)
    print()
    print("DOMAIN PROBLEM")
    print("objects")
    print("\t", domprob.worldobjects())
    print("operators")
    print("\t", list( domprob.operators() ))
    print("init",)
    print("\t", domprob.initialstate())
    print("goal",)
    print("\t", domprob.goals())

    print()
    ops_to_test = { 1:"op2", 2:"move", 3:"move", 4:"move", 5:"A1", 6:"op2" }
    op = ops_to_test[demonumber]
    print("ground for operator", op, "applicable if (adjacent loc1 loc2)")
    for o in domprob.ground_operator(op):
        if ("adjacent","loc1","loc2") in o.precondition_pos:
            print()
            print( "\tvars", o.variable_list )
            print( "\tpre+", o.precondition_pos )
            print( "\tpre-", o.precondition_neg )
            print( "\teff+", o.effect_pos )
            print( "\teff-", o.effect_neg )
    print()


if __name__ == '__main__':
    args = parser.parse_args()
    for n in args.demonumber:
        run_demo(n)