From 921f03fc4461c98a792a3efe8ad338fe9872b28c Mon Sep 17 00:00:00 2001
From: HaoYizhi <haoyizhi@agentlab.cn>
Date: Thu, 27 Jul 2023 12:56:22 +0800
Subject: [PATCH] change the way to find connected firm

---
 __pycache__/firm.cpython-38.pyc | Bin 8137 -> 7385 bytes
 firm.py                         |  40 ++++++++++----------------------
 2 files changed, 12 insertions(+), 28 deletions(-)

diff --git a/__pycache__/firm.cpython-38.pyc b/__pycache__/firm.cpython-38.pyc
index 14d7b0929dd9bcfbae416527bd4cb08b63fb22a7..ea365f5668f9e780fa5cdc90bc9db7bbcbbadf13 100644
GIT binary patch
delta 1960
zcma)7U1*zC7(Va$ev>q5(xzS0U+21N+_1`AWwWE3>!7k9JDqjR)!6ZUO`EjoauR=#
zFJ#zZvx{lrpa`3gc;^o~83;wt33}m$f_kI561*D-f*^vj`M&3Cvs9f}IL|rfJ?FfC
z-}9tjj{UGF@<ccsWcc^R@1HqejeZ#U*@EQNKdr+656FqYi=kb3CUzjyhvd27ko+}p
zt!(4gz@gzNd<%?uEW;{1K{I4{7FI!gy<u}8%f*B`(W?3ip9P0I+IiczjI(8^0j`Y-
zAEoy=Cm6j?vl7R>Dc;+#j<e~9OPa^ZRjbBIx~Hkxz+=eRnLst@>7G&Jg=&+<@bnrN
za~`jTJgo-6@^i%&k1qqIg<-nzZWw3o=$KR^0ZQuvtN<o!P_h*uOUYE8%aKegE>xzT
z>6jOJ$;!pmQKZ^9sR3F~HiGFG<m`H|GmfQ#aTSKGFbrd>FsvyIMd1zqk%oUhN3K1q
z8m;gewz}|2G3Hs9m=mb*OYBW;8D4@4_pD4?wY|o&CI-GjJ8CR;fl(K7EgnmQZ&hu_
z&u6d%bAk`zR~=;zdz>Xhvfh${`{mwn5A2px;W0Vg5tCP1?~>OeaX2Px{q4qXq`=E^
zOY3pGH{W_kJMAwJEp!AE_)Re)-?zHtr=5eLNi>P$1h?M4CjV%iFdrl13HeBLKskIv
zE=DIIE<cIxKUMEYyT#P(yjV=-+~V^~BAd!BIj#uNlRXG^dv2zXa-Fm*`pI^Xpp)QE
zf^mYQ1X~F@2<}Bl+VXI$uj5f%)&t}*Rd5$({iQ6&?hb3>B;B1NNXoTXKfEr#k8SI;
z!G<7k#&7D}pZN{0%WZ9yG6~7Vp>*-qx-P0v!>_su)2pX>2^6Qf2iR)J5+}|TwGBFo
zk$i9!@|V{w9Cb70v8wKI2k(1O<A8pe&guPS)Cp=LV~CH+9uf*AqT?eI6@uJXGqKiM
zSBT-6sE~J>EGB722vk!DwL%IJm;kb9T_N3gw?*7=G)G5$G*i|<i(T<4YvHZ3l?i7e
z$XQEi!p;$A(N;wpQWh1Dwb*_gIj0dsRY5_p8V8j@VYE%k*c`@>qWv?<ep2}6F!`YB
zNoOhG1yD}d#%k1805<Y`KB#yeI41C0F%h^OQ)9&HZ*cgmW}M04jxjmexnrl#TzdT^
z!94`@Ybof-cR2p95Pl>fjQyv(pPJnzV~WdlW4`Dv7DOAxJSZ=2Nw}wJNmRV05iw0$
zC6B);BaR1kQ@D#u&$-E<*iQG72&e3Yf-gJqG}%ZeB1P~F!Hn$g8a(0Gto}Jf=4Nuv
zf}0X<{#mzBT*FI+7Z4baLlj!T#1V%G*KtJ(`Zo==;JJDAgRYMOMr5`-v0dD@lwL~0
z*SEaZ9Y69A?ZX7B5lZDQZDRzacfa~Wv}z}y!jqO?@GKdqYE_{e>KQg2-2KMnQ$6qM
zZ-dw_?cO2SDu;Wo!y(z-Hx7d`)t7FYCnwzRvq~YOeiSL~`{XZupBwbUvrDe^JA+4X
mCC(6}3CO?55-bv^LXXM=13R@zd!IZzkTCa={eXORVDDcKj=TE+

delta 2736
zcmbtWU2IfE6rMBpXYbzq>9)JAT_`^)5SBlMAQfzE!3rdX3Q~{|%36B2Keu%2&aRq}
zyIByDUgKYJUQCFxoA^NE6Y|2Fn(!cRJ{S}Fgfa1fL}SDUO$_y%+3vR6R3W;_eD}<m
zJLfxhX3jTzXZP*y=&?w|X85~&?`!8m&-Lif0hqb*SKuXpXXL@)xzJ7=6PpocH^}q$
z26;dDRVj#F1Dl2|pWux7n#+8iq$jw19HzO5)qD<&b8%3OXk~qh4}-%U?UGgX^s<Xk
z0kjRDchP)&CK=5iVMUI8Q~Xl%_OgLXMa|brND^=@H;4md*`AhUj_w%m^P)cQv)s@!
zqXq^wJ#$LKBuq-e(<y<@1#dukjhYu@ggtZI6r&z5M?9?pz;dP0Le%3Ifxc`S#7TUq
z37@@=g_G;wKq20NDL|?q1@(bE#ehmH<~HL~eXAF6Y)stBCaNRl`bY)Pd!o*zBgYrS
zRYK!4Cq!f(S2S#jAx&7(utpk>*9oFE0*_N@&nmb0XiY=@+d|y4t}rL!^DFEV?IOGl
zKKCp)QEsiUaTED|qaJNdIpS|t%a9HG+6YwLYUl6-7z{Te+KC6Fi#hBlOEt-l!|7E!
z==kWvV`3r&B845XiGWU(*d-H@?o$5@h;7&%0)*NR4;Bkxi(|`gL>OTn8(3R}0xN^h
zT<(E`gcV-aD$Ic-bM?zSrypfAT+k5}6l}&2&y(FvKqp=F5JU*JBOvj5R>IDrl+b+v
zTi!x2fzKtd{g2Er$9iU<k`hNTvZ5FLWUXl=zqFKGLgHEZZX^LO%j+xS#x5LxPu`3i
zfgRG0u8mXwLz<%_P{N{9K8SaPj$>E6O0Y2RiX2EBG*6K8q`Vhhqk?@ZqfLjQO{SZA
z4v7$r?IC!Mpo?HHfm*ZGWc3oX5j;WgB0(R);{>e)w414*{IhA*V~44)Vo{jc@Y#GJ
zJ?OY{b8KUaE>6?vDFSse*2~votKpg~#h%<~LzwY7lh>X)aIo?7^f_HyuxyYwV}6O!
zHnAyD`Uy~!z{PV?!Frg6f$vW7BA}q4Jb8Fhuym=MZWXjzA&Xr(dm3_g=RF)n>$neP
z-Q#q!Jg9KMIE%(<{$11)4PzSOj}nz4>XVLLb6%dRmhI_Cs5v@Txltbt&0C8{O$-<C
z0!lLs$3%Im2~o`(P+pTt97QXDF~VK|<w<y&R*uNI=GM|_bVf0^eJDrf5`_rTbib+)
zC`g!SlN&>Yu#^M^v(c)rj6PI|p!)t`_1m)!t_&)}{L0;(XSs$tL@ZiOhe}F_mZQ;@
zp+O~Sq{-Q89ovewXS>a`RVY$;L#W07CTA;EiS$*gh>E;l96?;DPAp~ipM^ubg0q(v
zXJc)Ck8d;x!?`W{QP}p5WU_^Aqk~!bR{ZI)M^~*y+8~ZSl>E+8aukN$%a-{MC=Um=
zifWxx+yR2dPx+CeJ%VKXN|I6f*HHHthf2kj1k4&^h0H`=v=B2X#%y$OXeje~AuTew
zH!}HxNYaQ+@NmsJv{Y{r6wwr!iODxJsSt`dh9LzxX_3CrWh2gzuTg!(5cwQ=rgdE@
z#jE0@rqX3J8^sND&Sq)}eu=*P2MCNOAO>MDu_YkNb+jy!CEySAId0S~{huT%jI%5r
zp5t0oz_@NpyDfa<_tpxm=tKswm4N<L3EC1tGL}lnt4-~LjVm;3jSlA18D}ITI_Sl^
z0^{;JjvC0PhsESXRk@XDt1VEnJW5lV2}q!YQuF<?xxF<pMt!w_!R(}yN#`?Xhvm-p
zr%;sWQjufrpL5YK+mjn$yX;QhgZ*;TieA_$Cs&NLWbqwbPPL8d!su5Q)i}C)K3aLp
x=tJWLd3&X^?hslcL*Nona4|`cBcKVQPaf&mq8+ym$nzaZQ{CSC<kgNne*@wGKotN0

diff --git a/firm.py b/firm.py
index db10d0c..692bea7 100644
--- a/firm.py
+++ b/firm.py
@@ -120,18 +120,12 @@ class FirmAgent(ap.Agent):
                 if self.is_prf_conn:
                     for firm in \
                             self.dct_cand_alt_supp_up_prod_disrupted[product]:
-                        out_edges = self.model.firm_network.graph.out_edges(
-                            self.model.firm_network.positions[firm], keys=True)
-                        in_edges = self.model.firm_network.graph.in_edges(
-                            self.model.firm_network.positions[firm], keys=True)
-                        lst_adj_firm = []
-                        lst_adj_firm += \
-                            [ap.AgentIter(self.model, edge[1]).to_list()[
-                                0].code for edge in out_edges]
-                        lst_adj_firm += \
-                            [ap.AgentIter(self.model, edge[0]).to_list()[
-                                0].code for edge in in_edges]
-                        if self.code in lst_adj_firm:
+                        node_self = self.get_firm_network_node()
+                        node_firm = firm.get_firm_network_node()
+                        if self.model.firm_network.graph.\
+                                has_edge(node_self, node_firm) or \
+                                self.model.firm_network.graph.\
+                                has_edge(node_firm, node_self):
                             lst_firm_connect.append(firm)
                 if len(lst_firm_connect) == 0:
                     # select based on size or not
@@ -201,22 +195,12 @@ class FirmAgent(ap.Agent):
                     lst_firm_connect = []
                     if self.is_prf_conn:
                         for firm in lst_firm:
-                            out_edges = \
-                                self.model.firm_network.graph.out_edges(
-                                    self.model.firm_network.positions[firm],
-                                    keys=True)
-                            in_edges = \
-                                self.model.firm_network.graph.in_edges(
-                                    self.model.firm_network.positions[firm],
-                                    keys=True)
-                            lst_adj_firm = []
-                            lst_adj_firm += \
-                                [ap.AgentIter(self.model, edge[1]).to_list()[
-                                    0].code for edge in out_edges]
-                            lst_adj_firm += \
-                                [ap.AgentIter(self.model, edge[0]).to_list()[
-                                    0].code for edge in in_edges]
-                            if self.code in lst_adj_firm:
+                            node_self = self.get_firm_network_node()
+                            node_firm = firm.get_firm_network_node()
+                            if self.model.firm_network.graph.\
+                                    has_edge(node_self, node_firm) or \
+                                    self.model.firm_network.graph.\
+                                    has_edge(node_firm, node_self):
                                 lst_firm_connect.append(firm)
                     if len(lst_firm_connect) == 0:
                         # handling based on size or not