snarled/snarl/utils.py

"""
Some utility code that gets reused
"""
from typing import Set, Tuple

from .types import connectivity_t, layer_t


def connectivity2layers(
        connectivity: connectivity_t,
        ) -> Tuple[Set[layer_t], Set[layer_t]]:
    """
    Extract the set of all metal layers and the set of all via layers
    from the connectivity description.
    """
    metal_layers = set()
    via_layers = set()
    for top, via, bot in connectivity:
        metal_layers.add(top)
        metal_layers.add(bot)
        if via is not None:
            via_layers.add(via)

    both = metal_layers.intersection(via_layers)
    if both:
        raise Exception(f'The following layers are both vias and metals!? {both}')

    return metal_layers, via_layers