#!/usr/bin/env python3

import pyactr as actr

Model = actr.ACTRModel()

DM = Model.decmem
g = Model.goal
imaginal = Model.set_goal(name="imaginal", delay=0.2)

actr.chunktype("number", ("number", "next", "ones", "tens", "hundreds"))
actr.chunktype("math_goal", ("op", "task"))
actr.chunktype(
    "math_op",
    (
        "op",
        "arg1",
        "arg2",
        "result",
        "hundreds1",
        "tens1",
        "ones1",
        "hundreds2",
        "tens2",
        "ones2",
        "hundreds_ans",
        "tens_ans",
        "ones_ans",
        "carry",
    ),
)

g.add(actr.makechunk("", "math_goal", op="greater", task="calc"))
imaginal.add(
    actr.makechunk(
        "",
        "math_op",
        op="greater",
        arg1=5,
        arg2=2,
        ones1=5,
        tens1=0,
        hundreds1=0,
        ones2=2,
        tens2=0,
        hundreds2=0,
    )
)


# https://stackoverflow.com/a/39644726
def get_digit(number, n):
    return number // 10**n % 10


# Add numbers 0-999 to decmem
for i in range(0, 1000):
    DM.add(
        actr.makechunk(
            f"number{str(i)}",
            "number",
            number=i,
            next=i + 1,
            ones=get_digit(i, 0),
            tens=get_digit(i, 1),
            hundreds=get_digit(i, 2),
        )
    )

# Add comparison relations to single digit numbers
for i in range(0, 10):
    for j in range(0, 10):
        DM.add(
            actr.makechunk(
                f"greater{i}{j}",
                "math_op",
                op="greater",
                arg1=i,
                arg2=j,
                result=max(i, j),
            )
        )
        DM.add(
            actr.makechunk(
                f"lesser{i}{j}",
                "math_op",
                op="lesser",
                arg1=i,
                arg2=j,
                result=min(i, j),
            )
        )
        DM.add(
            actr.makechunk(
                f"plus{i}{j}", "math_op", op="add", arg1=i, arg2=j, result=i + j
            )
        )


def unwind_prod():
    # Model.productionstring(
    #     name="init_math_op",
    #     string="""
    #     =g>
    #     isa     math_op
    #     task    None
    #     arg1    =num1
    #     arg2    =num2
    #     ==>
    #     +retrieval>
    #     isa     number
    #     number    =num1
    #     =g>
    #     task      unwind_left
    #     """,
    # )

    Model.productionstring(
        name="terminate_op",
        string="""
            =g>
            isa     math_op
            task    finished_op
            =imaginal>
            result  =answer
            ==>
            ~g>""",
    )


def greater_than_prod():
    # TODO: if two numbers are equal, extra effort to determine greater one?
    prod_start = Model.productionstring(
        name="greater_start",
        string="""
            =g>
            isa     math_goal
            task    calc
            =imaginal>
            isa     math_op
            op      greater
            hundreds1   =hun1
            hundreds2   =hun1
            tens1       =tens1
            tens2       =tens1
            ones1       =ones1
            ones2       ~=ones1
            ones2       =ones2
            ==>
            +retrieval>
            isa     math_op
            op      greater
            arg1    =ones1
            arg2    =ones2
        """,
    )
    print(prod_start)

    Model.productionstring(
        name="greater_end",
        string="""
        =g>
        isa     math_goal
        =imaginal>
        op      greater
        =retrieval>
        isa     mat_op
        op      greater
        result  =ans
        ==>
        =imaginal>
        isa     math_op
        result  =ans
        ~g>
        """,
    )


def addition():
    Model.productionstring(
        name="start_add",
        string="""
    =goal>
        isa         math_goal
    =imaginal>
        op           add
        ones1        =num1
        ones2      =num2
        one_ans      None
    ==>
    =imaginal>
        isa         math_op
        one-ans     busy
    +retrieval>
        isa         math_op
        op          add
        arg1        =num1
        arg2        =num2
    """,
    )

    Model.productionstring(
        name="add_ones",
        string="""
    =goal>
        isa         math_goal
    =imaginal>
        isa         math_op
        op          add
        one-ans     busy
        ones1        =num1
        ones2        =num2
    =retrieval>
        isa         math_op
        arg1      =num1
        arg2      =num2
        result    =result
    ==>
    =goal>
        isa         math_goal
    =imaginal>
        ones_ans    =result
        carry       busy
    +retrieval>
        isa         math_op
        op          add
        arg1        10
        result      =result
    """,
    )

    Model.productionstring(
        name="process_carry",
        string="""
    =goal>
        isa         math_goal
    =imaginal>
        isa         math_op
        op          add
        tens1       =num1
        tens2       =num2
        carry       busy
        ones_ans    =ones
    =retrieval>
        isa         math_op
        op          add
        arg1        10
        result      =ones
        arg2        =remainder
    ==>
    =goal>
        isa         math_goal
    =imaginal>
        isa         math_op
        op          add
        carry       1
        tens_ans    busy
        one_ans     =remainder
    +retrieval>
        isa         math_op
        op          add
        arg1        =num1
        arg2        =num2
    """,
    )

    Model.productionstring(
        name="increment-sum",
        string="""
    =goal>
        ISA         add
        sum         =sum
        count       =count
        - arg2        =count
    =retrieval>
        ISA         number
        number      =sum
        next        =newsum
    ==>
    =goal>
        ISA         add
        sum         =newsum
    +retrieval>
        ISA         number
        number      =count
    """,
    )


# print(DM)

# unwind_prod()
greater_than_prod()

# Model.goal.add(actr.makechunk("goal", "math_op", op="greater", arg1=5, arg2=9))
print("g: ", g)
print("imaginal: ", imaginal)
x = Model.simulation()
x.run()
# print(list(DM))
numbers = [x for x in list(DM) if x.typename != "number" and x.typename != "math_op"]
# print(numbers)