framework/pddl/main.py

import argparse
from helper.fs import FS
from src.model.asm4_structure import Asm4Structure

from src.usecases.assembly_to_pddl_use_case import AssemblyToPddlUseCase
 


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument('--stepStructurePath', help='json step by FreeCad')
    parser.add_argument('--outPath', help='save pddl path')
    args = parser.parse_args()
    
    if args.stepStructurePath == None or args.outPath == None:
        parser.print_help()
    
    data = FS.readJSON(args.stepStructurePath)

    assemblyToPddlUseCase = AssemblyToPddlUseCase.call(assembly=data,rootLabel=data[0])
    FS.writeFile(assemblyToPddlUseCase['problem'] ,args.outPath, 'problem.pddl')
    FS.writeFile(assemblyToPddlUseCase['domain'] ,args.outPath, 'domain.pddl')