from sys import exit
from colour import *

from rs.reaction_system import ReactionSystem


class ReactionSystemWithConcentrations(ReactionSystem):
    def __init__(self):
        self.reactions = []
        self.meta_reactions = dict()
        self.permanent_entities = dict()
        self.background_set = []
        self.context_entities = []  # legacy. to be removed
        self.reactions_by_prod = None
        self.max_concentration = 0
        self.max_conc_per_ent = dict()

    def add_bg_set_entity(self, e):
        name = ""
        def_max_conc = -1
        if type(e) is tuple and len(e) == 2:
            name, def_max_conc = e
        elif type(e) is str:
            name = e
            print("\nWARNING: no maximal concentration level specified for:", e, "\n")
        else:
            raise RuntimeError("Bad entity type when adding background set element")

        self.assume_not_in_bgset(name)
        self.background_set.append(name)

        if def_max_conc != -1:
            ent_id = self.get_entity_id(name)
            self.max_conc_per_ent.setdefault(ent_id, 0)
            if self.max_conc_per_ent[ent_id] < def_max_conc:
                self.max_conc_per_ent[ent_id] = def_max_conc
            if self.max_concentration < def_max_conc:
                self.max_concentration = def_max_conc

    def get_max_concentration_level(self, e):
        if e in self.max_conc_per_ent:
            return self.max_conc_per_ent[e]
        else:
            return self.max_concentration

    def is_valid_entity_with_concentration(self, e):
        """Sanity check for entities with concentration"""

        if type(e) is tuple:
            if len(e) == 2 and type(e[1]) is int:
                return True

        if type(e) is list:
            if len(e) == 2 and type(e[1]) is int:
                return True

        print("FATAL. Invalid entity+concentration: {:s}".format(e))
        exit(1)

        return False

    def get_state_ids(self, state):
        """Returns entities of the given state without levels"""
        return [e for e, c in state]

    def has_non_zero_concentration(self, elem):
        if elem[1] < 1:
            raise RuntimeError("Unexpected concentration level in state: " + str(elem))

    def process_rip(self, R, I, P, ignore_empty_R=False):
        """Chcecks concentration levels and converts entities names into their ids"""

        if R == [] and not ignore_empty_R:
            raise RuntimeError("No reactants defined")

        reactants = []
        for r in R:
            self.is_valid_entity_with_concentration(r)
            self.has_non_zero_concentration(r)
            entity, level = r
            reactants.append((self.get_entity_id(entity), level))
            if self.max_concentration < level:
                self.max_concentration = level
        inhibitors = []
        for i in I:
            self.is_valid_entity_with_concentration(i)
            self.has_non_zero_concentration(i)
            entity, level = i
            inhibitors.append((self.get_entity_id(entity), level))
            if self.max_concentration < level:
                self.max_concentration = level
        products = []
        for p in P:
            self.is_valid_entity_with_concentration(p)
            self.has_non_zero_concentration(p)
            entity, level = p
            products.append((self.get_entity_id(entity), level))

        return reactants, inhibitors, products

    def add_reaction(self, R, I, P):
        """Adds a reaction"""

        if P == []:
            raise RuntimeError("No products defined")
        reaction = self.process_rip(R, I, P)
        self.reactions.append(reaction)

    def add_reaction_without_reactants(self, R, I, P):
        """Adds a reaction"""

        if P == []:
            raise RuntimeError("No products defined")
        reaction = self.process_rip(R, I, P, ignore_empty_R=True)
        self.reactions.append(reaction)

    def add_reaction_inc(self, incr_entity, incrementer, R, I):
        """Adds a macro/meta reaction for increasing the value of incr_entity"""

        reactants, inhibitors, products = self.process_rip(
            R, I, [], ignore_empty_R=True
        )
        incr_entity_id = self.get_entity_id(incr_entity)
        self.meta_reactions.setdefault(incr_entity_id, [])
        self.meta_reactions[incr_entity_id].append(
            ("inc", self.get_entity_id(incrementer), reactants, inhibitors)
        )

    def add_reaction_dec(self, decr_entity, decrementer, R, I):
        """Adds a macro/meta reaction for decreasing the value of incr_entity"""

        reactants, inhibitors, products = self.process_rip(
            R, I, [], ignore_empty_R=True
        )
        decr_entity_id = self.get_entity_id(decr_entity)
        self.meta_reactions.setdefault(decr_entity_id, [])
        self.meta_reactions[decr_entity_id].append(
            ("dec", self.get_entity_id(decrementer), reactants, inhibitors)
        )

    def add_permanency(self, ent, I):
        """Sets entity to be permanent unless it is inhibited"""

        ent_id = self.get_entity_id(ent)

        if ent_id in self.permanent_entities:
            raise RuntimeError("Permanency for {0} already defined.".format(ent))

        inhibitors = self.process_rip([], I, [], ignore_empty_R=True)[1]
        self.permanent_entities[ent_id] = inhibitors

    def set_context_entities(self, entities):
        raise NotImplementedError

    def entities_names_set_to_str(self, entities):
        s = ""
        for entity in entities:
            s += entity + ", "
        s = s[:-2]
        return s

    def entities_ids_set_to_str(self, entities):
        s = ""
        for entity in entities:
            s += self.get_entity_name(entity) + ", "
        s = s[:-2]
        return s

    def state_to_str(self, state):
        s = ""
        for ent, level in state:
            s += self.get_entity_name(ent) + "=" + str(level) + ", "
        s = s[:-2]
        return s

    def show_background_set(self):
        print(
            C_MARK_INFO
            + " Background set: {"
            + self.entities_names_set_to_str(self.background_set)
            + "}"
        )

    def show_meta_reactions(self):
        print(C_MARK_INFO + " Meta reactions:")
        for param_ent, reactions in self.meta_reactions.items():
            for r_type, command, reactants, inhibitors in reactions:
                if r_type == "inc" or r_type == "dec":
                    print(
                        " - [ Type="
                        + repr(r_type)
                        + " Operand=( "
                        + self.get_entity_name(param_ent)
                        + " ) Command=( "
                        + self.get_entity_name(command)
                        + " ) ] -- ( R={"
                        + self.state_to_str(reactants)
                        + "}, I={"
                        + self.state_to_str(inhibitors)
                        + "} )"
                    )
                else:
                    raise RuntimeError("Unknown meta-reaction type: " + repr(r_type))

    def show_max_concentrations(self):
        print(
            C_MARK_INFO
            + " Maximal allowed concentration levels (for optimized translation to RS):"
        )
        for e, max_conc in self.max_conc_per_ent.items():
            print(" - {0:^20} = {1:<6}".format(self.get_entity_name(e), max_conc))

    def show_permanent_entities(self):
        print(C_MARK_INFO + " Permanent entities:")
        for e, inhibitors in self.permanent_entities.items():
            print(
                " - {0:^20}{1:<6}".format(
                    self.get_entity_name(e) + ": ",
                    "I={" + self.state_to_str(inhibitors) + "}",
                )
            )

    def show(self, soft=False):
        self.show_background_set()
        self.show_reactions(soft)
        self.show_permanent_entities()
        self.show_meta_reactions()
        self.show_max_concentrations()

    def get_reactions_by_product(self):
        """Sorts reactions by their products and returns a dictionary of products"""

        if self.reactions_by_prod != None:
            return self.reactions_by_prod

        producible_entities = set()

        for reaction in self.reactions:
            product_entities = [e for e, c in reaction[2]]
            producible_entities = producible_entities.union(set(product_entities))

        reactions_by_prod = {}

        for p_e in producible_entities:
            reactions_by_prod[p_e] = []
            rcts_for_p_e = reactions_by_prod[p_e]

            for r in self.reactions:
                product_entities = [e for e, c in r[2]]

                if p_e in product_entities:
                    reactants = r[0]
                    inhibitors = r[1]
                    products = [(e, c) for e, c in r[2] if e == p_e]

                    prod_conc = products[0][1]
                    insert_place = None

                    # we need to order the reactions w.r.t. the concentration levels produced (increasing order)
                    for i in range(0, len(rcts_for_p_e)):
                        checked_conc = rcts_for_p_e[i][2][0][1]
                        if prod_conc <= checked_conc:
                            insert_place = i
                            break

                    if (
                        insert_place == None
                    ):  # empty or the is only one element which is smaller than the element being added
                        # we append (to the end)
                        rcts_for_p_e.append((reactants, inhibitors, products))
                    else:
                        rcts_for_p_e.insert(
                            insert_place, (reactants, inhibitors, products)
                        )

        # save in cache
        self.reactions_by_prod = reactions_by_prod

        return reactions_by_prod

    def get_reaction_system(self):
        rs = ReactionSystem()

        for reactants, inhibitors, products in self.reactions:
            new_reactants = []
            new_inhibitors = []
            new_products = []

            for ent, conc in reactants:
                n = self.get_entity_name(ent) + "#" + str(conc)
                rs.ensure_bg_set_entity(n)
                new_reactants.append(n)

            for ent, conc in inhibitors:
                n = self.get_entity_name(ent) + "#" + str(conc)
                rs.ensure_bg_set_entity(n)
                new_inhibitors.append(n)

            for ent, conc in products:
                for i in range(1, conc + 1):
                    n = self.get_entity_name(ent) + "#" + str(i)
                    rs.ensure_bg_set_entity(n)
                    new_products.append(n)

            rs.add_reaction(new_reactants, new_inhibitors, new_products)

        for param_ent, reactions in self.meta_reactions.items():
            for r_type, command, reactants, inhibitors in reactions:
                param_ent_name = self.get_entity_name(param_ent)

                new_reactants = []
                new_inhibitors = []

                for ent, conc in reactants:
                    n = self.get_entity_name(ent) + "#" + str(conc)
                    rs.ensure_bg_set_entity(n)
                    new_reactants.append(n)

                for ent, conc in inhibitors:
                    n = self.get_entity_name(ent) + "#" + str(conc)
                    rs.ensure_bg_set_entity(n)
                    new_inhibitors.append(n)

                max_cmd_c = self.max_concentration
                if command in self.max_conc_per_ent:
                    max_cmd_c = self.max_conc_per_ent[command]
                else:
                    print(
                        "WARNING:\n\tThere is no maximal concentration level defined for "
                        + self.get_entity_name(command)
                    )
                    print("\tThis is a very bad idea -- expect degraded performance\n")

                for l in range(1, max_cmd_c + 1):
                    cmd_ent = self.get_entity_name(command) + "#" + str(l)
                    rs.ensure_bg_set_entity(cmd_ent)

                    if r_type == "inc":
                        # pre_conc  -- predecessor concentration
                        # succ_conc -- successor concentration concentration

                        for i in range(1, self.max_concentration):
                            pre_conc = param_ent_name + "#" + str(i)
                            rs.ensure_bg_set_entity(pre_conc)
                            new_products = []
                            succ_value = i + l
                            for j in range(1, succ_value + 1):
                                if j > self.max_concentration:
                                    break
                                new_p = param_ent_name + "#" + str(j)
                                rs.ensure_bg_set_entity(new_p)
                                new_products.append(new_p)
                            if new_products != []:
                                rs.add_reaction(
                                    set(new_reactants + [pre_conc, cmd_ent]),
                                    set(new_inhibitors),
                                    set(new_products),
                                )

                    elif r_type == "dec":
                        for i in range(1, self.max_concentration + 1):
                            pre_conc = param_ent_name + "#" + str(i)
                            rs.ensure_bg_set_entity(pre_conc)
                            new_products = []
                            succ_value = i - l
                            for j in range(1, succ_value + 1):
                                if j > self.max_concentration:
                                    break
                                new_p = param_ent_name + "#" + str(j)
                                rs.ensure_bg_set_entity(new_p)
                                new_products.append(new_p)
                            if new_products != []:
                                rs.add_reaction(
                                    set(new_reactants + [pre_conc, cmd_ent]),
                                    set(new_inhibitors),
                                    set(new_products),
                                )

                    else:
                        raise RuntimeError(
                            "Unknown meta-reaction type: " + repr(r_type)
                        )

        for ent, inhibitors in self.permanent_entities.items():
            max_c = self.max_concentration
            if ent in self.max_conc_per_ent:
                max_c = self.max_conc_per_ent[ent]
            else:
                print(
                    "WARNING:\n\tThere is no maximal concentration level defined for "
                    + self.get_entity_name(ent)
                )
                print("\tThis is a very bad idea -- expect degraded performance\n")

            def e_value(i):
                return self.get_entity_name(ent) + "#" + str(i)

            for value in range(1, max_c + 1):
                new_reactants = []
                new_inhibitors = []
                new_products = []

                new_reactants = [e_value(value)]

                for e_inh, conc in inhibitors:
                    n = self.get_entity_name(e_inh) + "#" + str(conc)
                    rs.ensure_bg_set_entity(n)
                    new_inhibitors.append(n)

                for i in range(1, value + 1):
                    new_products.append(e_value(i))

                rs.add_reaction(new_reactants, new_inhibitors, new_products)

        return rs


class ReactionSystemWithAutomaton(object):
    def __init__(self, reaction_system, context_automaton):
        self.rs = reaction_system
        self.ca = context_automaton

    def show(self, soft=False):
        self.rs.show(soft)
        self.ca.show()

    def is_with_concentrations(self):
        if not isinstance(self.rs, ReactionSystemWithConcentrations):
            return False
        if not isinstance(self.ca, ContextAutomatonWithConcentrations):
            return False
        return True

    def sanity_check(self):
        pass

    def get_ordinary_reaction_system_with_automaton(self):
        if not self.is_with_concentrations():
            raise RuntimeError("Not RS/CA with concentrations")

        ors = self.rs.get_reaction_system()
        oca = self.ca.get_automaton_with_flat_contexts(ors)

        return ReactionSystemWithAutomaton(ors, oca)


# EOF