from ABI.Structs import *


class Translator():

    @staticmethod
    def entry_flag_translator(s):
        translation = {
            "MMIO": DEVTREE_ENTRY_FLAGS_MMIO,
        }
        return translation.get(s, 0)

    @staticmethod
    def irq_flag_translator(s):
        translation = {
            "EDGE_TRIGGER": DEVTREE_IRQ_FLAGS_EDGE_TRIGGER,
        }
        return translation.get(s, 0)

    @staticmethod
    def entry_type(s):
        translation = {
            "IO": DEVTREE_ENTRY_TYPE_IO,
            "FB": DEVTREE_ENTRY_TYPE_FB,
            "UART": DEVTREE_ENTRY_TYPE_UART,
            "RAM": DEVTREE_ENTRY_TYPE_RAM,
            "STORAGE": DEVTREE_ENTRY_TYPE_STORAGE,
            "BUS_CONTROLLER": DEVTREE_ENTRY_TYPE_BUS_CONTROLLER,
            "RTC": DEVTREE_ENTRY_TYPE_RTC
        }
        return translation.get(s, DEVTREE_ENTRY_TYPE_IO)

    @staticmethod
    def number(s):
        return int(s, base=0)

    @staticmethod
    def flags(s, flagcb):
        flags = 0x0
        ents = s.split(" ")

        for ent in ents:
            t = flagcb(s)
            if t != None:
                flags |= t

        return flags