fr_generator.py

"""
 * Copyright © 2020 drewg3r
 * https://github.com/drewg3r/DM-2

fr_generator.py: generates finitary relations S and R from A and B sets.
                 aSb if a is the father of b.
                 aRb if a is the father-in-law of b.
                 Algorithm can be optimized!
"""

import random
import core


def generate(A, B, s_possibility, r_possibility):
    A = list(A)
    B = list(B)
    S = set()
    R = set()
    used_children = []
    fathers_wd = set()
    used_boys = set()
    used_fathers = set()
    for a in A:
        for b in B:
            if a in core.men_names:
                if b not in used_children:
                    if random.randint(1, s_possibility) == 1:
                        S.add((a, b))
                        used_children.append(b)
                        if b in core.women_names:
                            fathers_wd.add(a)

    for f in fathers_wd:
        for b in B:
            if b in core.men_names:
                if (f, b) not in S:
                    if b not in used_boys:
                        if f not in used_fathers:
                            if random.randint(1, r_possibility) == 1:
                                R.add((f, b))
                                used_fathers.add(f)
                                used_boys.add(b)

    return S, R