3b1b-manim/manimlib/utils/dict_ops.py

import itertools as it
import numpy as np


def merge_dicts_recursively(*dicts):
    """
    Creates a dict whose keyset is the union of all the
    input dictionaries.  The value for each key is based
    on the first dict in the list with that key.

    dicts later in the list have higher priority

    When values are dictionaries, it is applied recursively
    """
    result = dict()
    all_items = it.chain(*[d.items() for d in dicts])
    for key, value in all_items:
        if key in result and isinstance(result[key], dict) and isinstance(value, dict):
            result[key] = merge_dicts_recursively(result[key], value)
        else:
            result[key] = value
    return result


def soft_dict_update(d1, d2):
    """
    Adds key values pairs of d2 to d1 only when d1 doesn't
    already have that key
    """
    for key, value in list(d2.items()):
        if key not in d1:
            d1[key] = value


def dict_eq(d1, d2):
    if len(d1) != len(d2):
        return False
    for key in d1:
        value1 = d1[key]
        value2 = d2[key]
        if type(value1) != type(value2):
            return False
        if type(d1[key]) == np.ndarray:
            if any(d1[key] != d2[key]):
                return False
        elif d1[key] != d2[key]:
            return False
    return True