#!/usr/bin/env python3

import pyactr as actr


addition = actr.ACTRModel()
actr.chunktype("countOrder", ("first", "second"))
actr.chunktype("add", ("arg1", "arg2", "sum", "count"))

dm = addition.decmem

for i in range(0, 11):
    dm.add(actr.makechunk("chunk" + str(i), "countOrder", first=i, second=i + 1))

addition.goal.add(actr.makechunk("", "add", arg1=5, arg2=2))

addition.productionstring(
    name="init_addition",
    string="""
    =g>
    isa     add
    arg1    =num1
    arg2    =num2
    sum     None
    ==>
    =g>
    isa     add
    sum     =num1
    count   0
    +retrieval>
    isa     countOrder
    first   =num1
""",
)