aimplanning

from planning import HLA, RealWorldPlanningProblem

def criarJogada():
    library_2 = {
        'HLA': [
            'Jogada(Goleiro,GolAdversario)',
            'Jogada(Goleiro,GolAdversario)',

            'Passe(Goleiro, Jogador)',

            'Finalizacao(Jogador, GolAdversario)',
            'Finalizacao(Jogador, GolAdversario)',

            'Chute_Direita(Jogador, GolAdversario)',
            'Chute_Esquerda(Jogador, GolAdversario)',
            'TiroDeMeta(Goleiro, GolAdversario)'],
        'steps': [
            ['Passe(Goleiro, Jogador)', 'Finalizacao(Jogador, GolAdversario)'],
            ['TiroDeMeta(Goleiro, GolAdversario)'],
            [],
            ['Chute_Direita(Jogador, GolAdversario)'],
            ['Chute_Esquerda(Jogador, GolAdversario)'],
            [],
            [],
            []
        ],
        'precond': [
            ['BolaEm(Goleiro)'],
            ['BolaEm(Goleiro)'],
            ['BolaEm(Goleiro)'],
            ['BolaEm(Jogador)'],
            ['BolaEm(Jogador)'],
            ['BolaEm(Jogador)'],
            ['BolaEm(Jogador)'],
            ['BolaEm(Goleiro) & Fadiga(Baixa)']
        ],
        'effect': [
            ['BolaEm(GolAdversario) & ~BolaEm(Goleiro)'],
            ['BolaEm(GolAdversario) & ~BolaEm(Goleiro) & ~Fadiga(Baixa)'],
            ['BolaEm(Jogador) & BolaEm(Goleiro)'],
            ['BolaEm(GolAdversario) & ~BolaEm(Jogador)'],
            ['BolaEm(GolAdversario) & ~BolaEm(Jogador)'],
            ['BolaEm(GolAdversario) & ~BolaEm(Jogador)'],
            ['BolaEm(GolAdversario) & ~BolaEm(Jogador)'],
            ['BolaEm(GolAdversario) & ~BolaEm(Goleiro) & ~Fadiga(Baixa)']
        ]
    }

    jogada_go_ga = HLA('Jogada(Goleiro,GolAdversario)', precond='BolaEm(Goleiro)', effect='BolaEm(GolAdversario) & ~BolaEm(Goleiro)')
    tiro_go_ga = HLA('TiroDeMeta(Goleiro,GolAdversario)', precond='BolaEm(Goleiro)', effect='BolaEm(GolAdversario) & ~BolaEm(Goleiro) & ~Fadiga(Baixa)')
    lancamento_go_so = HLA('Lancamento(Goleiro, SetorOfensivo)', 'BolaEm(Goleiro) & Saida(Rapida)','BolaEm(SetorOfensivo) & ~BolaEm(Goleiro)')
    assistencia_so_ga = HLA('Assistencia(SetorOfensivo, GolAdversario)', 'BolaEm(SetorOfensivo)', 'BolaEm(GolAdversario) & ~BolaEm(SetorOfensivo)')

    actions = [jogada_go_ga, tiro_go_ga, lancamento_go_so, assistencia_so_ga]

    problem = RealWorldPlanningProblem(initial='BolaEm(Goleiro)', goals='BolaEm(GolAdversario)', actions=actions)

    plan = problem.hierarchical_search(library_2)
    print('---Possivel Jogada---')
    print(plan, '\n')
    print([x.__dict__ for x in plan])
    print()

if __name__ == '__main__':
    criarJogada()