lightsout.py

import re
import sys
import subprocess
import glob

PLANEJADOR = "<alterar_planejador>".strip().split(' ')

def inverte(mapa, x, y):
    if (x >= len(mapa) or x < 0) or (y >= len(mapa) or y < 0):
        return

    local = mapa[x][y]
    alteracao = 'd' if local.lower() == 'l' else 'l'
    mapa[x][y] = alteracao.upper() if local.isupper() else alteracao


def apertar(mapa, x, y):
    inverte(mapa, x - 1, y)
    inverte(mapa, x + 1, y)
    inverte(mapa, x, y - 1)
    inverte(mapa, x, y + 1)

    if mapa[x][y].isupper():
        inverte(mapa, x, y)

def main():
    entradas = []
    mapa = []
    linha = coluna = 0

    while True:
        try:
            valores = input()
            mapa.append([*valores])
            entradas += [
                ' '.join([
                    (f"(ta-quebrada x{linha} y{coluna})" if valor.islower() else ""),
                    (f"(ta-ligada x{linha} y{coluna})" if valor.lower() == 'l' else "")
                ]).strip()
                for coluna, valor in enumerate(valores)
                if valor in ['L', 'l', 'd']
            ]

            linha += 1
            coluna = len(valores) - 1
        except (EOFError, KeyboardInterrupt):
            break

    incrementos = []
    for l in range(0, linha-1):
        incrementos.append(f'(inc x{l} x{l+1})')
        incrementos.append(f'(inc x{l+1} x{l})')

    for c in range(0, coluna):
        incrementos.append(f'(inc y{c} y{c+1})')
        incrementos.append(f'(inc y{c+1} y{c})')

    init = '\n'.join(entradas + incrementos)
    objects = ' '.join([f'x{l}' for l in range(0, linha)]) + ' - linha ' + ' '.join([f'y{c}' for c in range(0, coluna+1)]) + ' - coluna'

    goals = []
    for l in range(0, linha):
        for c in range(0, coluna+1):
            goals.append(f'(not (ta-ligada x{l} y{c}))')
    goals = '\n'.join(goals)

    with open('<alterar_local_problema>', 'w') as f:
        f.write(f"""
            (define (problem lightsoutproblem)
                (:domain lightsout)
                (:objects {objects})
                (:init
                    (verdadeiro)
                    {init}
                )
                (:goal
                    (and
                        {goals}
                    )
                )
            )
            """)

    with open('<alterar_local_dominio>', 'w') as f:
        f.write("""
            <alterar_dominio>
        """)

    subprocess.run(PLANEJADOR + ['<alterar_local_dominio>', '<alterar_local_problema>'], stdout=subprocess.DEVNULL)

    files = glob.glob('<alterar_local_sas_plan>' + '*')
    files.sort()

    if len(files) == 0:
        sys.exit(120)

    resposta = []
    with open(files[-1], 'r') as f:
        for line in f.readlines():
            if 'apertar' not in line.lower():
                continue

            x = re.findall(r'x\d{1,2}?', line)
            y = re.findall(r'y\d{1,2}?', line)

            for valor_x, valor_y in zip(x, y):
                clique = f"({valor_x[1:]}, {valor_y[1:]})"
                apertar(mapa, int(valor_x[1:]), int(valor_y[1:]))
                resposta.append(clique)

    for linha in range(len(mapa)):
        for coluna in range(len(mapa)):
            if mapa[linha][coluna] in ['l', 'L']:
                sys.exit(120)

    print(';'.join(resposta))


if __name__ == "__main__":
    main()