rr_extractor.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "28da862c",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2021-09-13T08:40:03.307851Z",
     "start_time": "2021-09-13T08:40:02.537237Z"
    }
   },
   "outputs": [],
   "source": [
    "import networkx as nx\n",
    "import json\n",
    "import gc\n",
    "\n",
    "from networkx.classes.function import nodes\n",
    "from pandas.core.indexes import base\n",
    "\n",
    "get_attr = lambda line, attr : line.split(\"%s=\\\"\" % attr, 1)[1].split('\"', 1)[0]\n",
    "id = 0\n",
    "u = 0\n",
    "\n",
    "\n",
    "def parse_switch(switch_dict, line):\n",
    "    \"\"\"Parses all the switch blocks.\n",
    "\n",
    "    Parameters\n",
    "    ----------\n",
    "    lines : List[str]\n",
    "        Read RR-graph file.\n",
    "\n",
    "    Returns\n",
    "    -------\n",
    "    Dict[int, Dict[int, str]]\n",
    "        A dictionary of dictionaries, indexed by block ids, block name and ptc nums,\n",
    "        holding the pin names.\n",
    "    \"\"\"\n",
    "\n",
    "    if line.startswith(\"<switch \"):\n",
    "        id = int(get_attr(line, \"id\"))\n",
    "        name = get_attr(line, \"name\")\n",
    "        type = get_attr(line, \"type\")\n",
    "        switch_dict.update({id : {'name' : name, 'type' : type}})\n",
    "\n",
    "def parse_block_pins(pin_dict, line):\n",
    "    \"\"\"Parses all the block pins.\n",
    "\n",
    "    Parameters\n",
    "    ----------\n",
    "    lines : List[str]\n",
    "        Read RR-graph file.\n",
    "\n",
    "    Returns\n",
    "    -------\n",
    "    Dict[int, Dict[int, str]]\n",
    "        A dictionary of dictionaries, indexed by block ids, block name and ptc nums,\n",
    "        holding the pin names.\n",
    "    \"\"\"\n",
    "    global bid\n",
    "    if line.startswith(\"<block_type \"):\n",
    "        bid = int(get_attr(line, \"id\"))\n",
    "        name = get_attr(line, \"name\")\n",
    "        pin_dict.update({bid : {'name' : name}})\n",
    "\n",
    "    if \"<pin_class \" in line:\n",
    "        ptc = int(get_attr(line, \"ptc\"))\n",
    "        name = line.split(\"</pin>\")[0].split('>')[-1]\n",
    "        pin_dict[bid].update({ptc : name})\n",
    "\n",
    "def parse_segments(seg_dict, line):\n",
    "    \"\"\"Parses all segment declarations.\n",
    "\n",
    "    Parameters\n",
    "    ----------\n",
    "    lines : List[str]\n",
    "        Read RR-graph file.\n",
    "\n",
    "    Returns\n",
    "    -------\n",
    "    Dict[int, str]\n",
    "        A mapping from segment ids to segment names.\n",
    "    \"\"\"\n",
    "\n",
    "    if line.startswith(\"<segment id\"):\n",
    "        seg_dict.update({int(get_attr(line, \"id\")) : get_attr(line, \"name\")})\n",
    "\n",
    "def parse_grid(grid, line):\n",
    "    \"\"\"Parses the grid layout.\n",
    "\n",
    "    Parameters\n",
    "    ----------\n",
    "    lines : List[str]\n",
    "        Read RR-graph file.\n",
    "\n",
    "    Returns\n",
    "    -------\n",
    "    Dict[Tuple[int], int]\n",
    "        A dictionary holding block id for each tile.\n",
    "    \"\"\"\n",
    "\n",
    "    if line.startswith(\"<grid_loc \"):\n",
    "        x = int(get_attr(line, 'x'))\n",
    "        y = int(get_attr(line, 'y'))\n",
    "        bid = int(get_attr(line, \"block_type_id\"))\n",
    "        grid.update({'%d,%d'%(x, y) : bid})\n",
    "\n",
    "def parse_edges(line, G):\n",
    "    \"\"\"Parses all RR-graph edges.\n",
    "\n",
    "    Parameters\n",
    "    ----------\n",
    "    lines : List[str]\n",
    "        Read RR-graph file.\n",
    "    G : nx.DiGraph\n",
    "        RR-graph with nodes added.\n",
    "\n",
    "    Returns\n",
    "    -------\n",
    "    None\n",
    "    \"\"\"\n",
    "\n",
    "#     if line.startswith(\"<edge \"):\n",
    "#         u = int(get_attr(line, \"src_node\"))\n",
    "#         v = int(get_attr(line, \"sink_node\"))\n",
    "#         switch = int(get_attr(line, \"switch_id\"))\n",
    "#         if G.has_node(u) and G.has_node(v):\n",
    "#             G.add_edge(u, v, switch_id = switch)\n",
    "    if line.startswith(\"<edge \"):\n",
    "        u = int(get_attr(line, \"src_node\"))\n",
    "        v = int(get_attr(line, \"sink_node\"))\n",
    "        switch = int(get_attr(line, \"switch_id\"))   \n",
    "        if G.has_node(u) and G.has_node(v):\n",
    "            G.add_edge(u, v, switch_id = switch)\n",
    "#             G.add_edge(u, v, src_xl = G.nodes[u][\"xlow\"], src_yl = G.nodes[u][\"ylow\"], src_xh = G.nodes[u][\"xhigh\"], src_yh = G.nodes[u][\"yhigh\"], src_seg = G.nodes[u][\"seg\"], \n",
    "#                       destin_xl = G.nodes[v][\"xlow\"], destin_yl = G.nodes[v][\"ylow\"], destin_xh = G.nodes[v][\"xhigh\"], destin_yh = G.nodes[v][\"yhigh\"],destin_seg = G.nodes[v][\"seg\"], switch_id = switch)\n",
    "        \n",
    "\n",
    "def edge_direction(type, direction):\n",
    "    # 0-->Bi_Dir 1-->INC_DIR(?), SINK and IPIN -1-->DEC_DIR(?), SOURCE and OPIN\n",
    "    if \"CHANY\" or \"CHANX\" in type:\n",
    "        if direction == \"BI_DIR\":\n",
    "            return 0\n",
    "        # else:\n",
    "        #     return 1 if direction == \"INC_DIR\" else -1\n",
    "\n",
    "    if \"SINK\" or \"IPIN\" in type:\n",
    "        return 1\n",
    "    else:\n",
    "        return -1\n",
    "\n",
    "\n",
    "def find_edge(G, connection_graph:dict, switch_dict):\n",
    "    \"\"\"\n",
    "    1.Extract ID from connection_graph\n",
    "    2.Search all edge information of this ID, only as the source. Record Source, Sink, and Switch ID\n",
    "    \"\"\"\n",
    "    edge_graph = {}\n",
    "    history =[]\n",
    "    for key in connection_graph:\n",
    "        if key == \"counter\":\n",
    "            break\n",
    "        i = 1\n",
    "        \n",
    "        if key not in edge_graph:\n",
    "            edge_graph.update({key:{'source&destin' : connection_graph[key]}})\n",
    "\n",
    "        \n",
    "        for predecessor in G.predecessors(key):\n",
    "            \n",
    "            if '(%d, %d)'%(predecessor, key) in history:\n",
    "                continue\n",
    "            else:\n",
    "                history.append('(%d, %d)'%(predecessor, key))\n",
    "                \n",
    "            edge_graph[key].update({'switch_id_4source_%d' %(i) : G.edges[predecessor,key]['switch_id'], \n",
    "                            'switch_id_4source_%d' %(i) : switch_dict[G.edges[predecessor,key]['switch_id']]})\n",
    "            edge_graph[key].update({'source_%d' %(i): G.nodes[predecessor]})\n",
    "            \n",
    "            i += 1\n",
    "            \n",
    "            txt = \"<edge source_id=%s source_seg=%s source_coord=(%s, %s)-(%s, %s) destin_id=%s destin_seg=%s destin_coord=(%s, %s)-(%s, %s) switch_name=%s\\n>\"\\\n",
    "                %(predecessor, G.nodes[predecessor][\"seg\"],G.nodes[predecessor][\"xlow\"], G.nodes[predecessor][\"ylow\"], G.nodes[predecessor][\"xhigh\"], G.nodes[predecessor][\"yhigh\"],\n",
    "                key, connection_graph[key][\"seg\"], connection_graph[key][\"xlow\"], connection_graph[key][\"ylow\"], connection_graph[key][\"xhigh\"], connection_graph[key][\"yhigh\"], switch_dict[G.edges[predecessor,key]['switch_id']])\n",
    "            with open(r'/home/ymen/Desktop/node.txt', 'a') as f:\n",
    "                print(txt, file=f)\n",
    "                \n",
    "        i = 1\n",
    "        \n",
    "        for successor in G.successors(key):\n",
    "            \n",
    "            if '(%d, %d)'%(key, successor) in history:\n",
    "                continue\n",
    "            else:\n",
    "                history.append('(%d, %d)'%(key, successor))\n",
    "                \n",
    "            edge_graph[key].update({'switch_id_4destin_%d' %(i) : G.edges[key,successor]['switch_id'], \n",
    "                                    'switch_id_4destin_%d' %(i) : switch_dict[G.edges[key,successor]['switch_id']]})\n",
    "            edge_graph[key].update({'destin_%d' %(i) : G.nodes[successor]})\n",
    "            \n",
    "            i += 1\n",
    "            txt = \"<edge source_id=%s source_seg=%s source_coord=(%s, %s)-(%s, %s) destin_id=%s destin_seg=%s destin_coord=(%s, %s)-(%s, %s) switch_name=%s\\n>\"\\\n",
    "                    %(key, connection_graph[key][\"seg\"], connection_graph[key][\"xlow\"], connection_graph[key][\"ylow\"], connection_graph[key][\"xhigh\"], connection_graph[key][\"yhigh\"],\n",
    "                    successor, G.nodes[successor][\"seg\"],G.nodes[successor][\"xlow\"], G.nodes[successor][\"ylow\"], G.nodes[successor][\"xhigh\"], G.nodes[successor][\"yhigh\"],switch_dict[G.edges[key,successor]['switch_id']])\n",
    "                \n",
    "            with open(r'/home/ymen/Desktop/node.txt', 'a') as f:\n",
    "                print(txt, file=f)\n",
    "\n",
    "            \n",
    "        \n",
    "        \n",
    "#     for key in connection_graph:\n",
    "#         if key == 'counter': \n",
    "#             break\n",
    "\n",
    "#         i = 1\n",
    "        \n",
    "#         if not key in edge_graph:\n",
    "#             edge_graph.update({key:{}})\n",
    "        \n",
    "#         direct = edge_direction(connection_graph[key][\"base_type\"], connection_graph[key][\"direction\"]) \n",
    "\n",
    "#         #No need to find predecessors\n",
    "#         if direct == 1:\n",
    "#             edge_graph[key].update({'destin' : connection_graph[key]})\n",
    "\n",
    "#             for predecessor in G.predecessors(key):\n",
    "#                 if '(%d, %d)'%(predecessor, key) in history:\n",
    "#                     continue\n",
    "#                 else:\n",
    "#                     history.append('(%d, %d)'%(predecessor, key))\n",
    "#                     edge_graph[key].update({'switch_id_%d' %(i) : G.edges[predecessor,key]['switch_id'], \n",
    "#                                     'switch_name_%d' %(i) : switch_dict[G.edges[predecessor,key]['switch_id']]})\n",
    "#                     edge_graph[key].update({'source_%d' %(i): G.nodes[predecessor]})\n",
    "#                     i += 1\n",
    "\n",
    "#             edge_graph[key].update({'counter' : \"%d\" %(i-1), 'direction' : \"1\"})\n",
    "#             continue\n",
    "        \n",
    "#         if direct == -1:\n",
    "#             edge_graph.update({key:{'source' : connection_graph[key]}})\n",
    "\n",
    "#             for successor in G.successors(key):\n",
    "#                 if '(%d, %d)'%(key, successor) in history:\n",
    "#                     continue\n",
    "#                 else:\n",
    "#                     history.append('(%d, %d)'%(key, successor))\n",
    "#                     edge_graph[key].update({'switch_id_%d' %(i) : G.edges[key,successor]['switch_id'], \n",
    "#                                     'switch_name_%d' %(i) : switch_dict[G.edges[key,successor]['switch_id']]})\n",
    "#                     edge_graph[key].update({'destin_%d' %(i) : G.nodes[successor]})\n",
    "#                     i += 1\n",
    "\n",
    "#             edge_graph[key].update({'counter' : \"%d\" %(i-1), 'direction' : \"-1\"})\n",
    "#             continue\n",
    "        \n",
    "#         if direct == 0:\n",
    "#             edge_graph.update({key:{'source&destin' : connection_graph[key]}})\n",
    "            \n",
    "#             for predecessor in G.predecessors(key):\n",
    "#                 if '(%d, %d)'%(predecessor, key) in history:\n",
    "#                     continue\n",
    "#                 else:\n",
    "#                     history.append('(%d, %d)'%(predecessor, key))\n",
    "#                     edge_graph[key].update({'switch_id_4source_%d' %(i) : G.edges[predecessor,key]['switch_id'], \n",
    "#                                     'switch_name_4source_%d' %(i) : switch_dict[G.edges[predecessor,key]['switch_id']]})\n",
    "#                     edge_graph[key].update({'source_%d' %(i): G.nodes[predecessor]})\n",
    "#                     i += 1\n",
    "#                     txt = \"<edge source_id=%s source_seg=%s source_coord=(%s, %s)-(%s, %s) destin_id=%s destin_seg=%s destin_coord=(%s, %s)-(%s, %s) switch_name=%s\\n>\"\\\n",
    "#                         %(predecessor, G.nodes[predecessor][\"seg\"],G.nodes[predecessor][\"xlow\"], G.nodes[predecessor][\"ylow\"], G.nodes[predecessor][\"xhigh\"], G.nodes[predecessor][\"yhigh\"],\n",
    "#                         key, connection_graph[key][\"seg\"], connection_graph[key][\"xlow\"], connection_graph[key][\"ylow\"], connection_graph[key][\"xhigh\"], connection_graph[key][\"yhigh\"], switch_dict[G.edges[predecessor,key]['switch_id']])\n",
    "#                     with open(r'/home/ymen/Desktop/node.txt', 'a') as f:\n",
    "#                         print(txt, file=f)\n",
    "#             edge_graph[key].update({'counter_4source' : \"%d\" %(i-1)})\n",
    "\n",
    "#             i = 1\n",
    "\n",
    "#             for successor in G.successors(key):\n",
    "#                 if '(%d, %d)'%(key, successor) in history:\n",
    "#                     continue\n",
    "#                 else:\n",
    "#                     history.append('(%d, %d)'%(key, successor))\n",
    "#                 edge_graph[key].update({'switch_id_4destin_%d' %(i) : G.edges[key,successor]['switch_id'], \n",
    "#                                 'switch_name_4destin_%d' %(i) : switch_dict[G.edges[key,successor]['switch_id']]})\n",
    "#                 edge_graph[key].update({'destin_%d' %(i) : G.nodes[successor]})\n",
    "#                 i += 1\n",
    "#                 txt = \"<edge source_id=%s source_seg=%s source_coord=(%s, %s)-(%s, %s) destin_id=%s destin_seg=%s destin_coord=(%s, %s)-(%s, %s) switch_name=%s\\n>\"\\\n",
    "#                     %(key, connection_graph[key][\"seg\"], connection_graph[key][\"xlow\"], connection_graph[key][\"ylow\"], connection_graph[key][\"xhigh\"], connection_graph[key][\"yhigh\"],\n",
    "#                     successor, G.nodes[successor][\"seg\"],G.nodes[successor][\"xlow\"], G.nodes[successor][\"ylow\"], G.nodes[successor][\"xhigh\"], G.nodes[successor][\"yhigh\"],switch_dict[G.edges[key,successor]['switch_id']])\n",
    "                \n",
    "#                 with open(r'/home/ymen/Desktop/node.txt', 'a') as f:\n",
    "#                         print(txt, file=f)\n",
    "\n",
    "#             edge_graph[key].update({'counter_4destin' : \"%d\" %(i-1), 'direction' : \"0\"})\n",
    "                \n",
    "    return edge_graph\n",
    "\n",
    "def long_straight(x_target, y_target, xl_input, yl_input, xh_input, yh_input, tile_type):\n",
    "    \"\"\"\n",
    "    NN6BEG&NN2BEG\n",
    "    SS6BEG&SS2BEG\n",
    "    WW6BEG&WW2BEG\n",
    "    EE6BEG&EE2BEG\n",
    "    \"\"\"\n",
    "    nn_dict = [\"NN6BEG\", \"NN2BEG\"]\n",
    "    ee_dict = [\"EE2BEG\", \"EE6BEG\"]\n",
    "    ww_dict = [\"WW6BEG\", \"WW2BEG\"]\n",
    "    ss_dict = [\"SS6BEG\", \"SS2BEG\"]\n",
    "    if xl_input == x_target and yl_input == y_target:\n",
    "        if tile_type in nn_dict or tile_type in ee_dict:\n",
    "            return 1\n",
    "    if xh_input == x_target and yh_input == y_target:\n",
    "        if tile_type in ss_dict or tile_type in ww_dict:\n",
    "            return 1\n",
    "    return 0\n",
    "\n",
    "def short_straight():\n",
    "    pass\n",
    "def tilt_straight():\n",
    "    pass\n",
    "\n",
    "def get_connection_graph(G, x, y):\n",
    "    \"\"\"Get all connections of (x, y) coordinate.\n",
    "\n",
    "    Parameters\n",
    "    ----------\n",
    "    G : nx.DiGraph\n",
    "        Parsed RR-graph.\n",
    "    x : int\n",
    "        x-coordinate of the requested tile.\n",
    "    y : int\n",
    "        y-coordinate of the requested tile.\n",
    "\n",
    "    Returns\n",
    "    -------\n",
    "    connection\n",
    "    \"\"\"\n",
    "    nn_dict = [\"NN6BEG\", \"NN4BEG\", \"NN2BEG\"]\n",
    "    ee_dict = [\"EE2BEG\", \"EE4BEG\", \"EE6BEG\"]\n",
    "    ww_dict = [\"WW6BEG\", \"WW4BEG\", \"WW2BEG\"]\n",
    "    ss_dict = [\"SS6BEG\", \"SS4BEG\", \"SS2BEG\"]\n",
    "    \n",
    "    index = 0\n",
    "    connection = {}\n",
    "    tmp = {'counter' : {}}\n",
    "    for u, attrs in G.nodes(data = True):\n",
    "        #Get all nodes starting, ending at this tile, or pin of this tile\n",
    "#         if (attrs[\"xlow\"] == x and attrs[\"ylow\"] == y) or (attrs[\"xhigh\"] == x and attrs[\"yhigh\"] == y):\n",
    "#             connection.update({u : {'base_type' : attrs[\"base_type\"], 'direction' : attrs[\"direction\"], 'ptc' : attrs[\"ptc\"], \n",
    "#                 'xlow' : attrs[\"xlow\"],'xhigh' : attrs[\"xhigh\"],'ylow' : attrs[\"ylow\"],'yhigh' : attrs[\"yhigh\"],'seg' : attrs[\"seg\"]}})    \n",
    "    \n",
    "        index = 0\n",
    "    \n",
    "        if attrs[\"xlow\"] == x and attrs[\"ylow\"] == y and (attrs[\"seg\"] in nn_dict or attrs[\"seg\"] in ee_dict):\n",
    "            connection.update({u : {'base_type' : attrs[\"base_type\"], 'direction' : attrs[\"direction\"], 'ptc' : attrs[\"ptc\"], \n",
    "                'xlow' : attrs[\"xlow\"],'xhigh' : attrs[\"xhigh\"],'ylow' : attrs[\"ylow\"],'yhigh' : attrs[\"yhigh\"],'seg' : attrs[\"seg\"]}})                \n",
    "            index = 1\n",
    "            \n",
    "        if attrs[\"xhigh\"] == x and attrs[\"yhigh\"] == y and (attrs[\"seg\"] in ww_dict or attrs[\"seg\"] in ss_dict):\n",
    "            connection.update({u : {'base_type' : attrs[\"base_type\"], 'direction' : attrs[\"direction\"], 'ptc' : attrs[\"ptc\"], \n",
    "                'xlow' : attrs[\"xlow\"],'xhigh' : attrs[\"xhigh\"],'ylow' : attrs[\"ylow\"],'yhigh' : attrs[\"yhigh\"],'seg' : attrs[\"seg\"]}})                \n",
    "            index = 1    \n",
    "        \n",
    "        if index :\n",
    "            if attrs['seg'] != \"None\":\n",
    "                if attrs[\"seg\"] not in tmp['counter']:\n",
    "                    tmp['counter'].update({attrs[\"seg\"] : 1})\n",
    "                else:\n",
    "                    tmp['counter'][attrs[\"seg\"]] += 1\n",
    "            txt = \"<node direction=%s id=%s type=%s coordinate=(%s, %s)-(%s, %s) segment_name=%s>\\n\"\\\n",
    "                    %(attrs[\"direction\"], u, attrs[\"base_type\"], attrs[\"xlow\"], attrs[\"ylow\"], attrs[\"xhigh\"], attrs[\"yhigh\"], attrs[\"seg\"])\n",
    "            with open(r'/home/ymen/Desktop/node.txt', 'a') as f:\n",
    "                print(txt, file=f)\n",
    "                    \n",
    "    connection.update(tmp)\n",
    "\n",
    "    return connection\n",
    "\n",
    "\n",
    "def generate_graph_node(G, seg_dict, line): \n",
    "    global u\n",
    "    if line.startswith(\"<node \"):\n",
    "        u = int(get_attr(line, \"id\"))\n",
    "        base_type = get_attr(line, \"type\")\n",
    "        ptc = int(get_attr(line, \"ptc\"))\n",
    "        xlow = int(get_attr(line, \"xlow\"))\n",
    "        xhigh = int(get_attr(line, \"xhigh\"))\n",
    "        ylow = int(get_attr(line, \"ylow\"))\n",
    "        yhigh = int(get_attr(line, \"yhigh\"))\n",
    "\n",
    "        if base_type == \"CHANY\" or base_type == \"CHANX\":\n",
    "            direction = get_attr(line, \"direction\")\n",
    "        else:\n",
    "            direction = '0'\n",
    "\n",
    "        G.add_node(u, base_type = base_type, direction = direction, ptc = ptc, xlow = xlow, xhigh = xhigh, ylow = ylow, yhigh = yhigh, seg = \"None\")\n",
    "\n",
    "    try:\n",
    "        G.nodes[u][\"seg\"] = seg_dict[int(get_attr(line, \"segment_id\"))]\n",
    "    except:\n",
    "        pass\n",
    "\n",
    "def crossing(x1, y1, x2, y2, x3, y3, x4, y4, target_x, target_y):\n",
    "    if x1 == x2 and y3 == y4:\n",
    "        return True if x1 == target_x and y3 == target_y else False\n",
    "    if x3 == x4 and y1 == y2:\n",
    "        return True if x3 == target_x and y1 == target_y else False\n",
    "\n",
    "       "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "b47ff09f",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2021-09-13T08:40:09.297729Z",
     "start_time": "2021-09-13T08:40:05.358310Z"
    }
   },
   "outputs": [],
   "source": [
    "rr = r\"/home/ymen/Desktop/rr_graph/test_full_size.xml\"\n",
    "with open(rr, \"r\") as inf:\n",
    "    lines = inf.readlines()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "2b898e67",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2021-09-13T08:41:33.197261Z",
     "start_time": "2021-09-13T08:40:09.299225Z"
    }
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "\n",
    "\n",
    "\n",
    "switch_dict = {}\n",
    "seg_dict = {}\n",
    "pin_dict = {}\n",
    "grid_dict = {}\n",
    "edge = {}\n",
    "G = nx.DiGraph()\n",
    "\n",
    "with open(r'/home/ymen/Desktop/rr_graph/part_rr_graph.xml', 'a') as f:\n",
    "    for line in lines:\n",
    "        parse_switch(switch_dict, line)\n",
    "        parse_segments(seg_dict, line)\n",
    "        parse_block_pins(pin_dict, line)\n",
    "        parse_grid(grid_dict, line)\n",
    "        generate_graph_node(G, seg_dict, line)\n",
    "        if line.startswith(\"<edge \"):\n",
    "            print(line, file=f)\n",
    "del lines\n",
    "del switch_dict\n",
    "del pin_dict\n",
    "del grid_dict\n",
    "gc.collect()\n",
    "\n",
    "# with open(r'/home/myz1237/Desktop/output.txt', 'w') as f:\n",
    "#     print(json.dumps(switch_dict, indent=4), file=f)\n",
    "#     print(json.dumps(seg_dict, indent=4), file=f)\n",
    "#     print(json.dumps(pin_dict, indent=4), file=f)\n",
    "#     print(json.dumps(grid_dict, indent=4), file=f)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "fb7c0711",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2021-09-13T08:58:17.223731Z",
     "start_time": "2021-09-13T08:41:33.198925Z"
    }
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "with open(r'/home/ymen/Desktop/rr_graph/part_rr_graph.xml', 'r') as f:\n",
    "    for line in f:\n",
    "        parse_edges(line, G)\n",
    "del f\n",
    "gc.collect()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "id": "46f0400d",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2021-09-13T07:36:56.114608Z",
     "start_time": "2021-09-13T07:36:56.112411Z"
    }
   },
   "outputs": [],
   "source": [
    "x1 = 40\n",
    "y1 = 15\n",
    "\n",
    "x2 = 45\n",
    "y2 = 13"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "id": "206420a3",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2021-09-13T08:17:31.472439Z",
     "start_time": "2021-09-13T07:58:45.835776Z"
    }
   },
   "outputs": [],
   "source": [
    "open(r'/home/ymen/Desktop/newnode_%d%d.txt'%(x2,y2), 'w').close()\n",
    "\n",
    "target_coord = \"(%d, %d)\"%(x2, y2)\n",
    "history = []\n",
    "\n",
    "for u, v, attrs in G.edges(data=True):\n",
    "    \n",
    "    src_lcoord = \"(%d, %d)\"%(G.nodes[u][\"xlow\"], G.nodes[u][\"ylow\"])\n",
    "    src_hcoord = \"(%d, %d)\"%(G.nodes[u][\"xhigh\"], G.nodes[u][\"yhigh\"])\n",
    "    src_seg = G.nodes[u][\"seg\"]\n",
    "    \n",
    "    destin_lcoord = \"(%d, %d)\"%(G.nodes[v][\"xlow\"], G.nodes[v][\"ylow\"])\n",
    "    destin_hcoord = \"(%d, %d)\"%(G.nodes[v][\"xhigh\"], G.nodes[v][\"yhigh\"])\n",
    "    destin_seg = G.nodes[v][\"seg\"]\n",
    "    \n",
    "\n",
    "    \n",
    "    txt = \"\"\n",
    "    if attrs[\"switch_id\"] == 5 and crossing(int(G.nodes[u][\"xlow\"]), int(G.nodes[u][\"ylow\"]), int(G.nodes[u][\"xhigh\"]), int(G.nodes[u][\"yhigh\"]),\n",
    "                                           int(G.nodes[v][\"xlow\"]), int(G.nodes[v][\"ylow\"]), int(G.nodes[v][\"xhigh\"]), int(G.nodes[v][\"yhigh\"]), x2, y2):\n",
    "        if \"%d %d\"%(u, v) in history:\n",
    "            continue\n",
    "        else:\n",
    "            history.append(\"%d %d\"%(u, v))\n",
    "            history.append(\"%d %d\"%(v, u))\n",
    "            history.append(\"%d\"%(u))\n",
    "            history.append(\"%d\"%(v))\n",
    "            \n",
    "        txt = \"<short source_id=\\\"%s\\\" source_seg=\\\"%s\\\" source_coord=\\\"%s-%s\\\" destin_id=\\\"%s\\\" destin_seg=\\\"%s\\\" destin_coord=\\\"%s-%s\\\">\"\\\n",
    "            %(u, G.nodes[u][\"seg\"], src_lcoord, src_hcoord, v, G.nodes[v][\"seg\"], destin_lcoord, destin_hcoord)\n",
    "        with open(r'/home/ymen/Desktop/newnode_%d%d.txt'%(x2,y2), 'a') as f:\n",
    "            print(txt, file=f)\n",
    "        continue\n",
    "        \n",
    "for u, v, attrs in G.edges(data=True):\n",
    "    \n",
    "    src_lcoord = \"(%d, %d)\"%(G.nodes[u][\"xlow\"], G.nodes[u][\"ylow\"])\n",
    "    src_hcoord = \"(%d, %d)\"%(G.nodes[u][\"xhigh\"], G.nodes[u][\"yhigh\"])\n",
    "    src_seg = G.nodes[u][\"seg\"]\n",
    "    \n",
    "    destin_lcoord = \"(%d, %d)\"%(G.nodes[v][\"xlow\"], G.nodes[v][\"ylow\"])\n",
    "    destin_hcoord = \"(%d, %d)\"%(G.nodes[v][\"xhigh\"], G.nodes[v][\"yhigh\"])\n",
    "    destin_seg = G.nodes[v][\"seg\"]\n",
    "        \n",
    "    txt = \"\"\n",
    "    \n",
    "    if \"%d\"%(u) in history or \"%d\"%(v) in history:\n",
    "        continue\n",
    "        \n",
    "    if src_lcoord == destin_lcoord and src_lcoord == target_coord:\n",
    "        txt = \"<edge source_id=\\\"%s\\\" source_seg=\\\"%s\\\" source_coord=\\\"%s-%s\\\" destin_id=\\\"%s\\\" destin_seg=\\\"%s\\\" destin_coord=\\\"%s-%s\\\">\"\\\n",
    "            %(u, G.nodes[u][\"seg\"], src_lcoord, src_hcoord, v, G.nodes[v][\"seg\"], destin_lcoord, destin_hcoord)\n",
    "        with open(r'/home/ymen/Desktop/newnode_%d%d.txt'%(x2,y2), 'a') as f:\n",
    "            print(txt, file=f)\n",
    "        continue\n",
    "    if src_lcoord == destin_hcoord and src_lcoord == target_coord:\n",
    "        txt = \"<edge source_id=\\\"%s\\\" source_seg=\\\"%s\\\" source_coord=\\\"%s-%s\\\" destin_id=\\\"%s\\\" destin_seg=\\\"%s\\\" destin_coord=\\\"%s-%s\\\">\"\\\n",
    "            %(u, G.nodes[u][\"seg\"], src_lcoord, src_hcoord, v, G.nodes[v][\"seg\"], destin_lcoord, destin_hcoord)\n",
    "        with open(r'/home/ymen/Desktop/newnode_%d%d.txt'%(x2,y2), 'a') as f:\n",
    "            print(txt, file=f)\n",
    "        continue\n",
    "    if src_hcoord == destin_lcoord and src_hcoord == target_coord:\n",
    "        txt = \"<edge source_id=\\\"%s\\\" source_seg=\\\"%s\\\" source_coord=\\\"%s-%s\\\" destin_id=\\\"%s\\\" destin_seg=\\\"%s\\\" destin_coord=\\\"%s-%s\\\">\"\\\n",
    "            %(u, G.nodes[u][\"seg\"], src_lcoord, src_hcoord, v, G.nodes[v][\"seg\"], destin_lcoord, destin_hcoord)\n",
    "        with open(r'/home/ymen/Desktop/newnode_%d%d.txt'%(x2,y2), 'a') as f:\n",
    "            print(txt, file=f)\n",
    "        continue\n",
    "    if src_hcoord == destin_hcoord and src_hcoord == target_coord:\n",
    "        txt = \"<edge source_id=\\\"%s\\\" source_seg=\\\"%s\\\" source_coord=\\\"%s-%s\\\" destin_id=\\\"%s\\\" destin_seg=\\\"%s\\\" destin_coord=\\\"%s-%s\\\">\"\\\n",
    "            %(u, G.nodes[u][\"seg\"], src_lcoord, src_hcoord, v, G.nodes[v][\"seg\"], destin_lcoord, destin_hcoord)\n",
    "        with open(r'/home/ymen/Desktop/newnode_%d%d.txt'%(x2,y2), 'a') as f:\n",
    "            print(txt, file=f)\n",
    "        continue\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "995b4f46",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2021-09-13T07:42:48.669938Z",
     "start_time": "2021-09-13T07:42:44.502656Z"
    },
    "code_folding": []
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "46"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "sorted_seg_enum_source_column = [\"SS6BEG\", \"WW4BEG\", \"NN6BEG\", \"EE4BEG\", \"SW6BEG\", \"NW6BEG\", \"NE6BEG\", \"SE6BEG\", \"SS2BEG\", \"WW2BEG\", \"NN2BEG\", \"EE2BEG\", \"SW2BEG\", \"NW2BEG\", \"NE2BEG\", \"EL1BEG\", \"ER1BEG\", \"SL1BEG\", \"SR1BEG\", \"WL1BEG\", \"WR1BEG\", \"NL1BEG\", \"NR1BEG\"]\n",
    "oblique_seg = [\"SW6BEG\", \"NW6BEG\", \"NE6BEG\", \"SE6BEG\",\"SW2BEG\", \"NW2BEG\", \"NE2BEG\", \"SE2BEG\"]\n",
    "sorted_seg_enum_source_column_special = [\"SS6BEG\", \"WW4BEG\", \"NN6BEG\", \"EE4BEG\", \"SW6BEG\", \"NW6BEG\", \"NE6BEG\", \"SE6BEG\", \"SS2BEG\", \"WW2BEG\", \"NN2BEG\", \"EE2BEG\", \"SW2BEG\", \"NW2BEG\", \"SE2BEG\", \"NE2BEG\", \"EL1BEG\", \"ER1BEG\", \"SL1BEG\", \"SR1BEG\", \"WL1BEG\", \"WR1BEG\", \"NL1BEG\", \"NR1BEG\",\\\n",
    "                                        \"SR1BEG_S\", \"ER1BEG_S\", \"WR1BEG_S\", \"EL1BEG_N\",\"WL1BEG_N\", \"NL1BEG_N\"]\n",
    "short_collect = []\n",
    "counter = {}\n",
    "with open(r'/home/ymen/Desktop/newnode_%d%d.txt'%(x2,y2), 'r') as f:\n",
    "    counter.update({\"crossing\" : {}})\n",
    "    for key in oblique_seg:\n",
    "        counter[\"crossing\"].update({\"%s\"%(key) : {}})\n",
    "    \n",
    "    for line in f:\n",
    "        if line.startswith(\"<short\"):\n",
    "            source_id = get_attr(line, \"source_id\")\n",
    "            source_seg = get_attr(line, \"source_seg\")\n",
    "            source_coord = get_attr(line, \"source_coord\")\n",
    "            destin_id = get_attr(line, \"destin_id\")\n",
    "            destin_seg = get_attr(line, \"destin_seg\")\n",
    "            destin_coord = get_attr(line, \"destin_coord\")\n",
    "            txt = \"%s %s\"%(source_coord, destin_coord)\n",
    "            counter[\"crossing\"][source_seg].update({\"%s&%s\"%(source_id, destin_id) : txt})\n",
    "with open(r'/home/ymen/Desktop/newnode_%d%d.txt'%(x2,y2), 'r') as f:\n",
    "    for line in f:\n",
    "        if line.startswith(\"<edge\"):\n",
    "            source_id = get_attr(line, \"source_id\")\n",
    "            source_seg = get_attr(line, \"source_seg\")\n",
    "            destin_id = get_attr(line, \"destin_id\")\n",
    "            destin_seg = get_attr(line, \"destin_seg\")\n",
    "            destin_coord = get_attr(line, \"destin_coord\")\n",
    "            if destin_seg not in sorted_seg_enum_source_column_special:\n",
    "                continue\n",
    "            if destin_seg not in counter:\n",
    "                counter.update({destin_seg : {}})\n",
    "                counter[destin_seg].update({destin_id : destin_coord})\n",
    "            else:\n",
    "                counter[destin_seg].update({destin_id : destin_coord})\n",
    "            if destin_seg in oblique_seg:\n",
    "                i = 1\n",
    "                for successor in G.successors(int(destin_id)):\n",
    "                    if G.edges[int(destin_id), successor][\"switch_id\"] == 5:\n",
    "                        txt = \"(%s, %s)-(%s, %s)  %s\"%(G.nodes[successor][\"xlow\"],G.nodes[successor][\"ylow\"], G.nodes[successor][\"xhigh\"], G.nodes[successor][\"yhigh\"], destin_coord)\n",
    "                        counter[destin_seg].update({destin_id : txt})\n",
    "             \n",
    "for key in counter:\n",
    "    if key == \"crossing\":\n",
    "        for key1 in counter[\"crossing\"]:\n",
    "            counter[\"crossing\"][key1].update({\"counter\" : len(counter[\"crossing\"][key1])})\n",
    "        continue\n",
    "        \n",
    "    counter[key].update({\"counter\" : len(counter[key])})\n",
    "with open(r'/home/ymen/Desktop/extractnewnode_%d%d.txt'%(x2,y2), 'w') as f:\n",
    "    print(json.dumps(counter, indent=4), file=f)\n",
    "    \n",
    "del counter\n",
    "gc.collect()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "38a0058d",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2021-09-07T10:07:02.373336Z",
     "start_time": "2021-09-07T10:07:01.986834Z"
    }
   },
   "outputs": [],
   "source": [
    "open(r'/home/ymen/Desktop/node.txt', 'w').close()\n",
    "connection_graph = get_connection_graph(G, x2, y2)\n",
    "with open(r'/home/ymen/Desktop/output2.json', 'w') as f:\n",
    "     print(json.dumps(connection_graph, indent=4), file=f)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "e5ca6857",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2021-09-07T10:07:02.545341Z",
     "start_time": "2021-09-07T10:07:02.375068Z"
    },
    "scrolled": true
   },
   "outputs": [
    {
     "ename": "NameError",
     "evalue": "name 'switch_dict' is not defined",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mNameError\u001b[0m                                 Traceback (most recent call last)",
      "\u001b[0;32m<ipython-input-8-e875df79cb72>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m\u001b[0m\n\u001b[0;32m----> 1\u001b[0;31m \u001b[0medge\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mfind_edge\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mG\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mconnection_graph\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mswitch_dict\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m      2\u001b[0m \u001b[0;32mwith\u001b[0m \u001b[0mopen\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34mr'/home/ymen/Desktop/edge2.json'\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;34m'w'\u001b[0m\u001b[0;34m)\u001b[0m \u001b[0;32mas\u001b[0m \u001b[0mf\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m      3\u001b[0m      \u001b[0mprint\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mjson\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mdumps\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0medge\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mindent\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;36m4\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mfile\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0mf\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;31mNameError\u001b[0m: name 'switch_dict' is not defined"
     ]
    }
   ],
   "source": [
    "edge = find_edge(G, connection_graph, switch_dict)\n",
    "with open(r'/home/ymen/Desktop/edge2.json', 'w') as f:\n",
    "     print(json.dumps(edge, indent=4), file=f)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "08e3595c",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d7acbcab",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3983b333",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.8"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}