data_preprocess/raster2graph/util/edges_utils.py

edges = {
    0: "0000",
    1: "0001",
    2: "0010",
    3: "0011",
    4: "0100",
    5: "0110",
    6: "0111",
    7: "1000",
    8: "1001",
    9: "1011",
    10: "1100",
    11: "1101",
    12: "1110",
    13: "1111",
    14: "0101",
    15: "1010",
}


def get_edges_alldirections(edges_class):
    return edges[edges_class]


edges_rev = {
    "0000": 0,
    "0001": 1,
    "0010": 2,
    "0011": 3,
    "0100": 4,
    "0110": 5,
    "0111": 6,
    "1000": 7,
    "1001": 8,
    "1011": 9,
    "1100": 10,
    "1101": 11,
    "1110": 12,
    "1111": 13,
    "0101": 14,
    "1010": 15,
}


def get_edges_alldirections_rev(edges_class_rev):
    return edges_rev[edges_class_rev]