{ "cells": [ { "cell_type": "code", "execution_count": 1, "metadata": {}, "outputs": [], "source": [ "import pandas as pd\n", "import numpy as np\n", "import os\n", "import sumolib\n", "import copy\n", "from tqdm import tqdm\n", "from datetime import datetime" ] }, { "cell_type": "code", "execution_count": 2, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "2024-01-05 08:25:00\n", "2024-01-05 08:30:00\n", "2024-01-05 08:35:00\n", "2024-01-05 08:45:00\n" ] } ], "source": [ "midnight = int(datetime(2024, 1, 5, 0, 0, 0).timestamp())\n", "next_day = int(datetime(2024, 1, 6, 0, 0, 0).timestamp())\n", "fmins = range(midnight, next_day, 300)\n", "m = 105 # m ranges from 26 to 287\n", "# sigtable 시작시각 : 현재시각 - 1200 = 08:25\n", "# 시뮬레이션 시작시각 : 현재시각 - 900 = 08:30\n", "# 시뮬레이션 종료시각 : 현재시각 - 600 = 08:35\n", "# 현재시각 : 08:45\n", "present_time = fmins[m]\n", "sigtable_start = fmins[m] - 1200\n", "sim_start = fmins[m] - 900\n", "sim_end = fmins[m] - 600\n", "print(datetime.fromtimestamp(sigtable_start))\n", "print(datetime.fromtimestamp(sim_start))\n", "print(datetime.fromtimestamp(sim_end))\n", "print(datetime.fromtimestamp(present_time))\n", "net = sumolib.net.readNet('../../Data/networks/sn.net.xml')\n", "inter_node = pd.read_csv('../../data/tables/inter_node.csv', index_col=0)" ] }, { "cell_type": "code", "execution_count": 3, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "['i0', 'i1', 'i2', 'i3', 'i6', 'i7', 'i8', 'i9']\n" ] } ], "source": [ "histid = pd.read_csv(f'../../Data/tables/histids/histids_{fmins[50]}.csv', index_col=0)\n", "print(sorted(histid.node_id.unique()))" ] }, { "cell_type": "code", "execution_count": 4, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "{175: 'u00', 176: 'i1', 177: 'u20', 178: 'c30', 210: 'u60', 206: 'i7', 201: 'i8', 202: 'i9'}\n", "{'i0': 175, 'u00': 175, 'i1': 176, 'i2': 177, 'u20': 177, 'i3': 178, 'u30': 178, 'u31': 178, 'u32': 178, 'c30': 178, 'i6': 210, 'u60': 210, 'i7': 206, 'i8': 201, 'i9': 202}\n", "{'u00': 'i0', 'u20': 'i2', 'u30': 'i3', 'u31': 'i3', 'u32': 'i3', 'c30': 'i3'}\n", "{'i0': ['u00'], 'i1': [], 'i2': ['u20'], 'i3': ['c30', 'u30', 'u31', 'u32'], 'i6': [], 'i7': [], 'i8': [], 'i9': []}\n", "['c30', 'i0', 'i1', 'i2', 'i3', 'i6', 'i7', 'i8', 'i9', 'u00', 'u20', 'u30', 'u31', 'u32', 'u60']\n", "['i0', 'i1', 'i2', 'i3', 'i6', 'i7', 'i8', 'i9']\n", "['c30', 'u00', 'u20', 'u30', 'u31', 'u32', 'u60']\n" ] } ], "source": [ "inter2node = dict(zip(inter_node['inter_no'], inter_node['node_id']))\n", "node2inter = dict(zip(inter_node['node_id'], inter_node['inter_no']))\n", "ch2pa = {'u00': 'i0', 'u20': 'i2', 'u30': 'i3', 'u31': 'i3', 'u32': 'i3', 'c30': 'i3'}\n", "pa2ch = {'i0':['u00'], 'i1':[], 'i2':['u20'], 'i3':['c30', 'u30', 'u31', 'u32'], 'i6':[], 'i7':[], 'i8':[], 'i9':[]}\n", "node_ids = sorted(inter_node.node_id.unique())\n", "parent_ids = sorted(inter_node[inter_node.inter_type=='parent'].node_id.unique())\n", "child_ids = sorted(inter_node[inter_node.inter_type=='child'].node_id.unique())\n", "\n", "print(inter2node)\n", "print(node2inter)\n", "print(ch2pa)\n", "print(pa2ch)\n", "print(node_ids)\n", "print(parent_ids)\n", "print(child_ids)" ] }, { "cell_type": "code", "execution_count": 5, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "1704411900 2024-01-05 08:45:00\n" ] }, { "data": { "text/html": [ "
\n", "\n", "\n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", "
node_idstart_unixphas_Aphas_Bdurationinc_edge_Aout_edge_Ainc_edge_Bout_edge_B
0i617044104201124-571542115_01571500535_01NaNNaN
1i617044104201219-571542115_01571500535_01571500535_02.18571542115_01
2i617044104202239571500535_02.18571511538_01571500535_02.18571542115_01
3i617044104203365571511538_02.121571542115_01571511538_02.121571500585_01
4i617044104204423571500585_02571511538_01571500585_02571500535_01
..............................
215i017044116303426571545870_02571510153_01571510153_02571545870_01
216i017044116304433571510153_02571500487_01571510153_02571545870_01
217i117044116701137-571542810_01-571542797_02.99571542797_02.99571542810_01
218i117044116702293-571542810_01-571542797_02.99-571542810_01571543469_01
219i117044116703340571543469_02-571542797_02.99NaNNaN
\n", "

220 rows × 9 columns

\n", "
" ], "text/plain": [ " node_id start_unix phas_A phas_B duration inc_edge_A \\\n", "0 i6 1704410420 1 1 24 -571542115_01 \n", "1 i6 1704410420 1 2 19 -571542115_01 \n", "2 i6 1704410420 2 2 39 571500535_02.18 \n", "3 i6 1704410420 3 3 65 571511538_02.121 \n", "4 i6 1704410420 4 4 23 571500585_02 \n", ".. ... ... ... ... ... ... \n", "215 i0 1704411630 3 4 26 571545870_02 \n", "216 i0 1704411630 4 4 33 571510153_02 \n", "217 i1 1704411670 1 1 37 -571542810_01 \n", "218 i1 1704411670 2 2 93 -571542810_01 \n", "219 i1 1704411670 3 3 40 571543469_02 \n", "\n", " out_edge_A inc_edge_B out_edge_B \n", "0 571500535_01 NaN NaN \n", "1 571500535_01 571500535_02.18 571542115_01 \n", "2 571511538_01 571500535_02.18 571542115_01 \n", "3 571542115_01 571511538_02.121 571500585_01 \n", "4 571511538_01 571500585_02 571500535_01 \n", ".. ... ... ... \n", "215 571510153_01 571510153_02 571545870_01 \n", "216 571500487_01 571510153_02 571545870_01 \n", "217 -571542797_02.99 571542797_02.99 571542810_01 \n", "218 -571542797_02.99 -571542810_01 571543469_01 \n", "219 -571542797_02.99 NaN NaN \n", "\n", "[220 rows x 9 columns]" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "text/html": [ "
\n", "\n", "\n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", "
node_idphase_noring_typeinc_edgeout_edge
0i01A-571542797_02571500487_01
1i01B-571500487_01571542797_02
2i02A-571500487_01571545870_01
3i02B-571542797_02571510153_01
4i03A571545870_02571510153_01
..................
111u602B571500535_02-571500535_02
112u603ANaNNaN
113u603BNaNNaN
114u604ANaNNaN
115u604BNaNNaN
\n", "

116 rows × 5 columns

\n", "
" ], "text/plain": [ " node_id phase_no ring_type inc_edge out_edge\n", "0 i0 1 A -571542797_02 571500487_01\n", "1 i0 1 B -571500487_01 571542797_02\n", "2 i0 2 A -571500487_01 571545870_01\n", "3 i0 2 B -571542797_02 571510153_01\n", "4 i0 3 A 571545870_02 571510153_01\n", ".. ... ... ... ... ...\n", "111 u60 2 B 571500535_02 -571500535_02\n", "112 u60 3 A NaN NaN\n", "113 u60 3 B NaN NaN\n", "114 u60 4 A NaN NaN\n", "115 u60 4 B NaN NaN\n", "\n", "[116 rows x 5 columns]" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "print(fmins[m], datetime.fromtimestamp(fmins[m]))\n", "histid = pd.read_csv(f'../../Data/tables/histids/histids_{fmins[m]}.csv', index_col=0)\n", "match6 = pd.read_csv('../../Data/tables/matching/match6.csv', index_col=0)\n", "histid = histid.reset_index(drop=True)\n", "histid = histid.drop(columns=['inter_no'])\n", "match6 = match6[['node_id', 'phase_no', 'ring_type', 'inc_edge', 'out_edge']].reset_index(drop=True)\n", "display(histid)\n", "display(match6)" ] }, { "cell_type": "code", "execution_count": 6, "metadata": {}, "outputs": [ { "data": { "text/plain": [ "['i0', 'i1', 'i2', 'i3', 'i6', 'i8', 'i9']" ] }, "execution_count": 6, "metadata": {}, "output_type": "execute_result" } ], "source": [ "sorted(histid.node_id.unique())" ] }, { "cell_type": "code", "execution_count": 7, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "\n", "\n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", "
node_idstart_unixphas_Aphas_Bdurationinc_edge_Aout_edge_Ainc_edge_Bout_edge_B
0i617044104201124-571542115_01571500535_01NaNNaN
1i617044104201219-571542115_01571500535_01571500535_02.18571542115_01
2i617044104202239571500535_02.18571511538_01571500535_02.18571542115_01
3i617044104203365571511538_02.121571542115_01571511538_02.121571500585_01
4i617044104204423571500585_02571511538_01571500585_02571500535_01
..............................
411u0017044116303426NaNNaNNaNNaN
412u0017044116304433NaNNaNNaNNaN
413i117044116701137-571542810_01-571542797_02.99571542797_02.99571542810_01
414i117044116702293-571542810_01-571542797_02.99-571542810_01571543469_01
415i117044116703340571543469_02-571542797_02.99NaNNaN
\n", "

416 rows × 9 columns

\n", "
" ], "text/plain": [ " node_id start_unix phas_A phas_B duration inc_edge_A \\\n", "0 i6 1704410420 1 1 24 -571542115_01 \n", "1 i6 1704410420 1 2 19 -571542115_01 \n", "2 i6 1704410420 2 2 39 571500535_02.18 \n", "3 i6 1704410420 3 3 65 571511538_02.121 \n", "4 i6 1704410420 4 4 23 571500585_02 \n", ".. ... ... ... ... ... ... \n", "411 u00 1704411630 3 4 26 NaN \n", "412 u00 1704411630 4 4 33 NaN \n", "413 i1 1704411670 1 1 37 -571542810_01 \n", "414 i1 1704411670 2 2 93 -571542810_01 \n", "415 i1 1704411670 3 3 40 571543469_02 \n", "\n", " out_edge_A inc_edge_B out_edge_B \n", "0 571500535_01 NaN NaN \n", "1 571500535_01 571500535_02.18 571542115_01 \n", "2 571511538_01 571500535_02.18 571542115_01 \n", "3 571542115_01 571511538_02.121 571500585_01 \n", "4 571511538_01 571500585_02 571500535_01 \n", ".. ... ... ... \n", "411 NaN NaN NaN \n", "412 NaN NaN NaN \n", "413 -571542797_02.99 571542797_02.99 571542810_01 \n", "414 -571542797_02.99 -571542810_01 571543469_01 \n", "415 -571542797_02.99 NaN NaN \n", "\n", "[416 rows x 9 columns]" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "new_histids = []\n", "for parent_id in parent_ids:\n", " for child_id in pa2ch[parent_id]:\n", " new_histid = histid.copy()[histid.node_id==parent_id]\n", " new_histid[['inc_edge_A', 'out_edge_A', 'inc_edge_B', 'out_edge_B']] = np.nan\n", " for i, row in new_histid.iterrows():\n", " phas_A = row.phas_A\n", " phas_B = row.phas_B\n", " new_match = match6[match6.node_id==child_id]\n", " Arow = new_match[(new_match.phase_no==phas_A) & (new_match.ring_type=='A')]\n", " if ~ Arow[['inc_edge', 'out_edge']].isna().all().all():\n", " inc_edge = Arow.iloc[0].inc_edge\n", " out_edge = Arow.iloc[0].out_edge\n", " new_histid.loc[i, ['inc_edge_A', 'out_edge_A']] = [inc_edge, out_edge]\n", " Brow = new_match[(new_match.phase_no==phas_B) & (new_match.ring_type=='B')]\n", " if ~ Brow[['inc_edge', 'out_edge']].isna().all().all():\n", " inc_edge = Brow.iloc[0].inc_edge\n", " out_edge = Brow.iloc[0].out_edge\n", " new_histid.loc[i, ['inc_edge_B', 'out_edge_B']] = [inc_edge, out_edge]\n", " new_histid.loc[i, 'node_id'] = child_id\n", " new_histids.append(new_histid)\n", "histid_child = pd.concat(new_histids)\n", "histids = pd.concat([histid.copy(), histid_child])\n", "histids = histids.sort_values(by=['start_unix', 'node_id', 'phas_A', 'phas_B']).reset_index(drop=True)\n", "display(histids)" ] }, { "cell_type": "code", "execution_count": 8, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "['c30', 'i0', 'i1', 'i2', 'i3', 'i6', 'i8', 'i9', 'u00', 'u20', 'u30', 'u31', 'u32']\n", "[, , , , , , , , , , , , ]\n" ] } ], "source": [ "node_ids = sorted(histids.node_id.unique())\n", "nodes = [net.getNode(node_id) for node_id in node_ids]\n", "print(node_ids)\n", "print(nodes)" ] }, { "cell_type": "code", "execution_count": 9, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "c30 rrrrrr\n", "i0 grrrgrrrgrrrrrgrr\n", "i1 grrrrrrgrr\n", "i2 rrggrrr\n", "i3 grrrrgrrrrgrrrrgrrrr\n", "i6 grrrgrrrrgrrgrrr\n", "i8 grrrrrrrgrrrgrrr\n", "i9 rrrr\n", "u00 ggggrgggg\n", "u20 ggrggg\n", "u30 ggggrggg\n", "u31 ggggrggg\n", "u32 gggggggr\n" ] } ], "source": [ "node2init = {}\n", "for node in nodes:\n", " node_id = node.getID()\n", " conns = [(c.getJunctionIndex(), c) for c in node.getConnections()]\n", " conns = [c for c in conns if c[0] >= 0]\n", " conns = sorted(conns, key=lambda x: x[0])\n", " state = []\n", " for i, ci in conns:\n", " if ci.getTLLinkIndex() < 0:\n", " continue\n", " are_foes = False\n", " for j, cj in conns:\n", " if ci.getTo() == cj.getTo():\n", " continue\n", " if node.areFoes(i, j):\n", " are_foes = True\n", " break\n", " state.append('r' if are_foes else 'g')\n", " node2init[node_id] = state\n", "\n", "# 어떤 연결과도 상충이 일어나지는 않지만, 신호가 부여되어 있는 경우에는 r을 부여\n", "for _, row in histids.iterrows():\n", " node_id = row['node_id']\n", " inc_edge_A = row.inc_edge_A\n", " inc_edge_B = row.inc_edge_B\n", " out_edge_A = row.out_edge_A\n", " out_edge_B = row.out_edge_B\n", "\n", " if pd.isna(inc_edge_A) or pd.isna(out_edge_A):\n", " pass\n", " else:\n", " inc_edge_A = net.getEdge(inc_edge_A)\n", " out_edge_A = net.getEdge(out_edge_A)\n", " for conn in inc_edge_A.getConnections(out_edge_A):\n", " index = conn.getTLLinkIndex()\n", " if index >= 0:\n", " node2init[node_id][index] = 'r'\n", "\n", " if pd.isna(inc_edge_B) or pd.isna(out_edge_B):\n", " pass\n", " else:\n", " inc_edge_B = net.getEdge(inc_edge_B)\n", " out_edge_B = net.getEdge(out_edge_B)\n", " for conn in inc_edge_B.getConnections(out_edge_B):\n", " index = conn.getTLLinkIndex()\n", " if index >= 0:\n", " node2init[node_id][index] = 'r'\n", "for node_id in node_ids:\n", " print(node_id, \"\".join(node2init[node_id]))" ] }, { "cell_type": "code", "execution_count": 10, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "1704411900\n", "2024-01-05 08:45:00\n" ] }, { "data": { "text/html": [ "
\n", "\n", "\n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", "
node_idstart_unixphase_sumodurationstatestart_dt
98i21704410739040GGggGGG2024-01-05 08:25:39
99i21704410739125rrggrrr2024-01-05 08:25:39
100i21704410739271rrggrrr2024-01-05 08:25:39
101i21704410739334rrggGGG2024-01-05 08:25:39
102u201704410739040ggrggg2024-01-05 08:25:39
.....................
411u001704411630326ggggrgggg2024-01-05 08:40:30
412u001704411630433ggggrgggg2024-01-05 08:40:30
413i11704411670037gGGGGGrgrr2024-01-05 08:41:10
414i11704411670193grrGGGGgrr2024-01-05 08:41:10
415i11704411670240grrrrrrgGG2024-01-05 08:41:10
\n", "

318 rows × 6 columns

\n", "
" ], "text/plain": [ " node_id start_unix phase_sumo duration state start_dt\n", "98 i2 1704410739 0 40 GGggGGG 2024-01-05 08:25:39\n", "99 i2 1704410739 1 25 rrggrrr 2024-01-05 08:25:39\n", "100 i2 1704410739 2 71 rrggrrr 2024-01-05 08:25:39\n", "101 i2 1704410739 3 34 rrggGGG 2024-01-05 08:25:39\n", "102 u20 1704410739 0 40 ggrggg 2024-01-05 08:25:39\n", ".. ... ... ... ... ... ...\n", "411 u00 1704411630 3 26 ggggrgggg 2024-01-05 08:40:30\n", "412 u00 1704411630 4 33 ggggrgggg 2024-01-05 08:40:30\n", "413 i1 1704411670 0 37 gGGGGGrgrr 2024-01-05 08:41:10\n", "414 i1 1704411670 1 93 grrGGGGgrr 2024-01-05 08:41:10\n", "415 i1 1704411670 2 40 grrrrrrgGG 2024-01-05 08:41:10\n", "\n", "[318 rows x 6 columns]" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "print(present_time)\n", "print(datetime.fromtimestamp(present_time))\n", "sigtable = histids.copy()\n", "sigtable['init_state'] = sigtable['node_id'].map(node2init)\n", "sigtable['state'] = sigtable['init_state'].map(lambda x:''.join(x))\n", "for i, row in sigtable.iterrows():\n", " node_id = row.node_id\n", " inc_edge_A = row.inc_edge_A\n", " inc_edge_B = row.inc_edge_B\n", " out_edge_A = row.out_edge_A\n", " out_edge_B = row.out_edge_B\n", " state = copy.deepcopy(node2init)[node_id]\n", "\n", " if pd.isna(inc_edge_A) or pd.isna(out_edge_A):\n", " continue\n", " else:\n", " inc_edge_A = net.getEdge(inc_edge_A)\n", " out_edge_A = net.getEdge(out_edge_A)\n", " for conn in inc_edge_A.getConnections(out_edge_A):\n", " index = conn.getTLLinkIndex()\n", " if index >= 0:\n", " state[index] = 'G'\n", " sigtable.at[i, 'state'] = ''.join(state)\n", "\n", " if pd.isna(inc_edge_B) or pd.isna(out_edge_B):\n", " continue\n", " else:\n", " inc_edge_B = net.getEdge(inc_edge_B)\n", " out_edge_B = net.getEdge(out_edge_B)\n", " for conn in inc_edge_B.getConnections(out_edge_B):\n", " index = conn.getTLLinkIndex()\n", " if index >= 0:\n", " state[index] = 'G'\n", " sigtable.at[i, 'state'] = ''.join(state)\n", "sigtable = sigtable.dropna(subset='state')\n", "sigtable = sigtable.reset_index(drop=True)\n", "sigtable['phase_sumo'] = sigtable.groupby(['node_id', 'start_unix']).cumcount()\n", "sigtable = sigtable[sigtable.start_unix >= sigtable_start]\n", "sigtable = sigtable[['node_id', 'start_unix', 'phase_sumo', 'duration', 'state']]\n", "sigtable = sigtable.sort_values(by=['start_unix', 'node_id'])\n", "sigtable['start_dt'] = sigtable['start_unix'].apply(lambda x:datetime.fromtimestamp(x))\n", "display(sigtable)" ] }, { "cell_type": "code", "execution_count": 11, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
\n", "\n", "\n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", "
node_idstart_unixphase_sumodurationstatestart_dt
106i61704410760024grrrgGGGrgrrgrrr2024-01-05 08:26:00
107i61704410760119grrrgGGGrgrrgGGr2024-01-05 08:26:00
108i61704410760239grrrgrrrrgrrgGGG2024-01-05 08:26:00
109i61704410760365gGGGgrrrrgrrgrrr2024-01-05 08:26:00
110i61704410760423grrrgrrrrgGGgrrr2024-01-05 08:26:00
159i61704410930024grrrgGGGrgrrgrrr2024-01-05 08:28:50
160i61704410930119grrrgGGGrgrrgGGr2024-01-05 08:28:50
161i61704410930239grrrgrrrrgrrgGGG2024-01-05 08:28:50
162i61704410930365gGGGgrrrrgrrgrrr2024-01-05 08:28:50
163i61704410930423grrrgrrrrgGGgrrr2024-01-05 08:28:50
212i61704411100024grrrgGGGrgrrgrrr2024-01-05 08:31:40
213i61704411100119grrrgGGGrgrrgGGr2024-01-05 08:31:40
214i61704411100239grrrgrrrrgrrgGGG2024-01-05 08:31:40
215i61704411100365gGGGgrrrrgrrgrrr2024-01-05 08:31:40
216i61704411100423grrrgrrrrgGGgrrr2024-01-05 08:31:40
273i61704411270024grrrgGGGrgrrgrrr2024-01-05 08:34:30
274i61704411270119grrrgGGGrgrrgGGr2024-01-05 08:34:30
275i61704411270239grrrgrrrrgrrgGGG2024-01-05 08:34:30
276i61704411270365gGGGgrrrrgrrgrrr2024-01-05 08:34:30
277i61704411270423grrrgrrrrgGGgrrr2024-01-05 08:34:30
338i61704411440024grrrgGGGrgrrgrrr2024-01-05 08:37:20
339i61704411440119grrrgGGGrgrrgGGr2024-01-05 08:37:20
340i61704411440239grrrgrrrrgrrgGGG2024-01-05 08:37:20
341i61704411440365gGGGgrrrrgrrgrrr2024-01-05 08:37:20
342i61704411440423grrrgrrrrgGGgrrr2024-01-05 08:37:20
393i61704411609024grrrgGGGrgrrgrrr2024-01-05 08:40:09
394i61704411609119grrrgGGGrgrrgGGr2024-01-05 08:40:09
395i61704411609239grrrgrrrrgrrgGGG2024-01-05 08:40:09
396i61704411609365gGGGgrrrrgrrgrrr2024-01-05 08:40:09
397i61704411609423grrrgrrrrgGGgrrr2024-01-05 08:40:09
\n", "
" ], "text/plain": [ " node_id start_unix phase_sumo duration state \\\n", "106 i6 1704410760 0 24 grrrgGGGrgrrgrrr \n", "107 i6 1704410760 1 19 grrrgGGGrgrrgGGr \n", "108 i6 1704410760 2 39 grrrgrrrrgrrgGGG \n", "109 i6 1704410760 3 65 gGGGgrrrrgrrgrrr \n", "110 i6 1704410760 4 23 grrrgrrrrgGGgrrr \n", "159 i6 1704410930 0 24 grrrgGGGrgrrgrrr \n", "160 i6 1704410930 1 19 grrrgGGGrgrrgGGr \n", "161 i6 1704410930 2 39 grrrgrrrrgrrgGGG \n", "162 i6 1704410930 3 65 gGGGgrrrrgrrgrrr \n", "163 i6 1704410930 4 23 grrrgrrrrgGGgrrr \n", "212 i6 1704411100 0 24 grrrgGGGrgrrgrrr \n", "213 i6 1704411100 1 19 grrrgGGGrgrrgGGr \n", "214 i6 1704411100 2 39 grrrgrrrrgrrgGGG \n", "215 i6 1704411100 3 65 gGGGgrrrrgrrgrrr \n", "216 i6 1704411100 4 23 grrrgrrrrgGGgrrr \n", "273 i6 1704411270 0 24 grrrgGGGrgrrgrrr \n", "274 i6 1704411270 1 19 grrrgGGGrgrrgGGr \n", "275 i6 1704411270 2 39 grrrgrrrrgrrgGGG \n", "276 i6 1704411270 3 65 gGGGgrrrrgrrgrrr \n", "277 i6 1704411270 4 23 grrrgrrrrgGGgrrr \n", "338 i6 1704411440 0 24 grrrgGGGrgrrgrrr \n", "339 i6 1704411440 1 19 grrrgGGGrgrrgGGr \n", "340 i6 1704411440 2 39 grrrgrrrrgrrgGGG \n", "341 i6 1704411440 3 65 gGGGgrrrrgrrgrrr \n", "342 i6 1704411440 4 23 grrrgrrrrgGGgrrr \n", "393 i6 1704411609 0 24 grrrgGGGrgrrgrrr \n", "394 i6 1704411609 1 19 grrrgGGGrgrrgGGr \n", "395 i6 1704411609 2 39 grrrgrrrrgrrgGGG \n", "396 i6 1704411609 3 65 gGGGgrrrrgrrgrrr \n", "397 i6 1704411609 4 23 grrrgrrrrgGGgrrr \n", "\n", " start_dt \n", "106 2024-01-05 08:26:00 \n", "107 2024-01-05 08:26:00 \n", "108 2024-01-05 08:26:00 \n", "109 2024-01-05 08:26:00 \n", "110 2024-01-05 08:26:00 \n", "159 2024-01-05 08:28:50 \n", "160 2024-01-05 08:28:50 \n", "161 2024-01-05 08:28:50 \n", "162 2024-01-05 08:28:50 \n", "163 2024-01-05 08:28:50 \n", "212 2024-01-05 08:31:40 \n", "213 2024-01-05 08:31:40 \n", "214 2024-01-05 08:31:40 \n", "215 2024-01-05 08:31:40 \n", "216 2024-01-05 08:31:40 \n", "273 2024-01-05 08:34:30 \n", "274 2024-01-05 08:34:30 \n", "275 2024-01-05 08:34:30 \n", "276 2024-01-05 08:34:30 \n", "277 2024-01-05 08:34:30 \n", "338 2024-01-05 08:37:20 \n", "339 2024-01-05 08:37:20 \n", "340 2024-01-05 08:37:20 \n", "341 2024-01-05 08:37:20 \n", "342 2024-01-05 08:37:20 \n", "393 2024-01-05 08:40:09 \n", "394 2024-01-05 08:40:09 \n", "395 2024-01-05 08:40:09 \n", "396 2024-01-05 08:40:09 \n", "397 2024-01-05 08:40:09 " ] }, "execution_count": 11, "metadata": {}, "output_type": "execute_result" } ], "source": [ "sigtable[sigtable.node_id=='i6']" ] }, { "cell_type": "code", "execution_count": 12, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "2024-01-05 08:25:00\n", "2024-01-05 08:30:00\n", "2024-01-05 08:35:00\n", "2024-01-05 08:45:00\n" ] } ], "source": [ "# sigtable 시작시각 : 현재시각 - 1200 = 08:25\n", "# 시뮬레이션 시작시각 : 현재시각 - 900 = 08:30\n", "# 시뮬레이션 종료시각 : 현재시각 - 600 = 08:35\n", "# 현재시각 : 08:45\n", "print(datetime.fromtimestamp(sigtable_start))\n", "print(datetime.fromtimestamp(sim_start))\n", "print(datetime.fromtimestamp(sim_end))\n", "print(datetime.fromtimestamp(present_time))\n" ] }, { "cell_type": "code", "execution_count": 15, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "\n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", " \n", "\n" ] } ], "source": [ "offsets = {}\n", "asigtable_list = []\n", "for node_id in sorted(sigtable.node_id.unique()):\n", " asigtable = sigtable.copy()[sigtable.node_id==node_id]\n", " df = asigtable.copy()\n", " lsbs = asigtable[asigtable['start_unix'] < sim_start]['start_unix'].max() # the last start_unix before sim_start\n", " offsets[node_id] = lsbs - sim_start\n", " asigtable = asigtable[(asigtable['start_unix'] < sim_end) & (asigtable['start_unix'] >= lsbs)]\n", " asigtable_list.append(asigtable)\n", "sigtable = pd.concat(asigtable_list)\n", "\n", "strings = ['\\n']\n", "for node_id, group in sigtable.groupby('node_id'):\n", " strings.append(f' \\n')\n", " for i, row in group.iterrows():\n", " duration = row.duration\n", " state = row.state\n", " strings.append(f' \\n')\n", " strings.append(' \\n')\n", "strings.append('')\n", "strings = ''.join(strings)\n", "# 저장\n", "path_output = '../../Data/networks/sn.add.xml'\n", "with open(path_output, 'w') as f:\n", " f.write(strings)\n", "\n", "print(strings)" ] } ], "metadata": { "kernelspec": { "display_name": "rts", "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.10" } }, "nbformat": 4, "nbformat_minor": 2 }