import sys
import textwrap
import toml

def group_blocks(registers):
    blocks = []
    current_block = {}

    for reg in registers:
        reg["addr"] = int(reg["addr"], 16)

    registers = list(sorted(registers, key=lambda r: r["addr"]))
    base = registers[0]["addr"]

    for i, reg in enumerate(registers):
        # Check if this is the same block as before
        if registers[i-1]["addr"] + 1 != reg["addr"]:
            if i != 0:
                blocks.append(current_block)
            current_block = {"addr": reg["addr"], "regs": []}

        current_block["regs"].append(reg)

    blocks.append(current_block)

    return (base, blocks)

def peripheral(filename):
    data = toml.load(open(filename))

    base, blocks = group_blocks(data["reg"])

    info = data["info"]
    print(f"""\
    <peripheral>
      <name>{info["name"]}</name>
      <description>{info["desc"]}</description>
      <baseAddress>{hex(base)}</baseAddress>
""")

    if "int" in data:
        for interrupt in data["int"]:
            print(f"""\
      <interrupt>
        <name>{interrupt["name"]}</name>
        <description>{interrupt["desc"]}</description>
        <value>{interrupt["vect"]}</value>
      </interrupt>""")

        print("")


    for block in blocks:
        print(f"""\
      <addressBlock>
        <offset>{hex(block["addr"]-base)}</offset>
        <size>{hex(len(block["regs"]))}</size>
        <usage>registers</usage>
      </addressBlock>""")

    print(f"""
      <registers>""")

    for block in blocks:
        for reg in block["regs"]:
            print(f"""\
        <register>
          <name>{reg["name"]}</name>
          <description>{reg["desc"]}</description>
          <addressOffset>{hex(reg["addr"]-base)}</addressOffset>""")

            if "cstm" in reg:
                print(textwrap.indent(reg["cstm"].strip(), "          "))

            if "safe" in reg and "fields" in reg:
                print(f'Warning: {info["name"]}[{reg["name"]}] has both "safe" and "fields[]"!', file=sys.stderr)

            if "safe" in reg:
                # The whole register is safe to write
                print("""\
          <writeConstraint>
            <range>
              <minimum>0</minimum>
              <maximum>255</maximum>
            </range>
          </writeConstraint>""")

            if "fields" in reg:
                print("""
          <fields>""")

                for field in reg["fields"]:
                    print(f"""\
            <field>
              <name>{field["name"]}</name>
              <description>{field["desc"]}</description>
              <bitRange>[{field["rnge"]}]</bitRange>""")

                    if "cstm" in field:
                        print(textwrap.indent(field["cstm"].strip(), "              "))

                    # Write constraint
                    if "safe" in field:
                        rnge = [int(x) for x in field["rnge"].split(":")]
                        bits = rnge[0] - rnge[1] + 1
                        maxi = 2**bits-1
                        print(f"""\
              <writeConstraint>
                <range>
                  <minimum>0</minimum>
                  <maximum>{maxi}</maximum>
                </range>
              </writeConstraint>""")
                    elif "valu" in field:
                        print("""\
              <writeConstraint>
                <useEnumeratedValues>true</useEnumeratedValues>
              </writeConstraint>
                        """)

                    if "valu" in field:
                        print("""
              <enumeratedValues>""")
                        for value in data["val"][field["valu"]]["values"]:
                            print(f"""\
                <enumeratedValue>
                  <name>{value["name"]}</name>
                  <description>{value["desc"]}</description>
                  <value>{value["valu"]}</value>
                </enumeratedValue>""")
                        print("""\
              </enumeratedValues>""")

                    print(f"""\
            </field>""")

                print("""\
          </fields>""")

            print("""\
        </register>""")

    print(f"""\
      </registers>
    </peripheral>""")


if __name__ == "__main__":
    import sys
    peripheral(sys.argv[1])