counterGen.py

import cm2py as cm2
import math

def standard(bits, bounds=None):
    save = cm2.Save()

    ands = []
    flips = []
    clock = save.addBlock(cm2.OR, (bits,0,1))
    for i in range(bits):
        ands.append(save.addBlock(cm2.AND, (i,0,1)))
        flips.append(save.addBlock(cm2.FLIPFLOP, (i,0,0)))
        save.addConnection(ands[-1], flips[-1])
        save.addConnection(clock, ands[-1])
        for j in ands[:-1]:
            save.addConnection(flips[-1], j)
    
    if bounds:
        nand = save.addBlock(cm2.NAND, (-1,0,0), state=True)
        intermediate = save.addBlock(cm2.AND, (-1,0,1))
        resetAnd = save.addBlock(cm2.AND, (-1,0,2))
        save.addConnection(intermediate, resetAnd)
        save.addConnection(clock,resetAnd)
        maxBinary = format(bounds[1], f"0{bits}b")
        minBinary = format(bounds[0], f"0{bits}b")

        for i,bit in enumerate(maxBinary):
            if bit == "1":
                save.addConnection(flips[i], nand)
                save.addConnection(flips[i], intermediate)
            if bit != minBinary[i]:
                save.addConnection(resetAnd, flips[i])
            save.addConnection(nand, ands[i])
        
        for i,bit in enumerate(minBinary):
            if bit == "1":
                flips[i].properties = [2,0]

    return save

def reverse(bits, bounds=None):
    save = cm2.Save()

    ands = []
    flips = []
    clock = save.addBlock(cm2.NOR, (bits,0,1), state=True)
    for i in range(bits):
        ands.append(save.addBlock(cm2.NOR, (i,0,1)))
        flips.append(save.addBlock(cm2.FLIPFLOP, (i,0,0)))
        save.addConnection(ands[-1], flips[-1])
        save.addConnection(clock, ands[-1])
        for j in ands[:-1]:
            save.addConnection(flips[-1], j)
    
    if bounds:
        nor = save.addBlock(cm2.NOR, (-1,0,0), state=True)
        intermediate = save.addBlock(cm2.OR, (-1,0,1))
        resetNor = save.addBlock(cm2.NOR, (-1,0,2))
        save.addConnection(intermediate, resetNor)
        save.addConnection(clock,resetNor)
        maxBinary = format(bounds[1], f"0{bits}b")
        minBinary = format(bounds[0], f"0{bits}b")

        for i,bit in enumerate(minBinary):
            if bit == "0":
                save.addConnection(flips[i], nor)
                save.addConnection(flips[i], intermediate)
            if bit != maxBinary[i]:
                save.addConnection(resetNor, flips[i])
            save.addConnection(nor, ands[i])
        
        for i,bit in enumerate(maxBinary):
            if bit == "1":
                flips[i].properties = [2,0]

    return save

def both(bits):
    save = cm2.Save()

    ands = []
    nors = []
    flips = []
    clockUp = save.addBlock(cm2.OR, (bits,0,1))
    clockDown = save.addBlock(cm2.NOR, (bits,0,2), state=True)
    for i in range(bits):
        ands.append(save.addBlock(cm2.AND, (i,0,1)))
        nors.append(save.addBlock(cm2.NOR, (i,0,2), state=False))
        flips.append(save.addBlock(cm2.FLIPFLOP, (i,0,0)))
        save.addConnection(ands[-1], flips[-1])
        save.addConnection(clockUp, ands[-1])
        save.addConnection(nors[-1], flips[-1])
        save.addConnection(clockDown, nors[-1])
        for j in ands[:-1]:
            save.addConnection(flips[-1], j)
        for j in nors[:-1]:
            save.addConnection(flips[-1], j)

    return save