#!/usr/bin/env -S python3 -m dg
import "/nmigen_dg/*"
import "/nmigen/cli/main"

# iCEBreaker 7-Segment Pmod

Segment7x2 = subclass Elaboratable where
    __init__ = decimal: True ~> None where
        @number = Signal 8 # i
        @segs   = Signal 7 # o
        @select = Signal 1 # o
        @decimal = decimal

    elaborate = platform ~> m where with m = Module! =>
        select = Signal!
        Sync$ Drive select ~select

        seg7 = Submodule$ Segment7!

        if not @decimal =>
            Comb$ Drive seg7.number $ Mux select
                @number >> 4
                @number & 0x0f
        if @decimal =>
            Comb$ Drive seg7.number $ Mux select
                @number // 10
                @number % 10

        Comb$ Drive @segs   seg7.segs
        Comb$ Drive @select select


Segment7 = subclass Elaboratable where
    __init__ = self -> None where
        @number = Signal 4 # i
        @segs   = Signal 7 # o

    elaborate = platform ~> m where with m = Module! =>
        Switch @number
            0x0 ,-> Comb$ Drive @segs 0b0111111
            0x1 ,-> Comb$ Drive @segs 0b0000110
            0x2 ,-> Comb$ Drive @segs 0b1011011
            0x3 ,-> Comb$ Drive @segs 0b1001111
            0x4 ,-> Comb$ Drive @segs 0b1100110
            0x5 ,-> Comb$ Drive @segs 0b1101101
            0x6 ,-> Comb$ Drive @segs 0b1111101
            0x7 ,-> Comb$ Drive @segs 0b0000111
            0x8 ,-> Comb$ Drive @segs 0b1111111
            0x9 ,-> Comb$ Drive @segs 0b1101111
            0xa ,-> Comb$ Drive @segs 0b1110111
            0xb ,-> Comb$ Drive @segs 0b1111100
            0xc ,-> Comb$ Drive @segs 0b0111001
            0xd ,-> Comb$ Drive @segs 0b1011110
            0xe ,-> Comb$ Drive @segs 0b1111001
            0xf ,-> Comb$ Drive @segs 0b1110001


if __name__ == "__main__" =>
    design = Segment7x2!
    main design ports: [design.number, design.segs, design.select]