From ef05b8a253bbb590ecfdfb253471a7147144c863 Mon Sep 17 00:00:00 2001
From: thesai <1021828630@qq.com>
Date: Sun, 18 Feb 2024 18:26:13 +0800
Subject: [PATCH] =?UTF-8?q?=E9=97=A8=E6=8E=A7=E7=AE=B1=E7=9B=B8=E5=85=B3?=
 =?UTF-8?q?=E7=BB=A7=E7=94=B5=E5=99=A8=E4=B8=8D=E5=86=8D=E8=83=BD=E7=9B=B4?=
 =?UTF-8?q?=E6=8E=A5=E6=8E=A7=E5=88=B6=E5=B1=8F=E8=94=BD=E9=97=A8=EF=BC=8C?=
 =?UTF-8?q?=E8=80=8C=E6=98=AF=E8=A2=AB=E8=81=94=E9=94=81=E9=87=87=E9=9B=86?=
 =?UTF-8?q?=E5=90=8E=E7=94=B1=E8=81=94=E9=94=81=E5=8F=91=E8=B5=B7=E6=8E=A7?=
 =?UTF-8?q?=E5=88=B6=EF=BC=9BMPJ=E5=92=8CJXTCPLJ=E6=94=B9=E4=B8=BA?=
 =?UTF-8?q?=E4=BB=85=E4=BE=9B=E8=81=94=E9=94=81=E9=87=87=E9=9B=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 message_server/psl_ms.go                      |  3 +-
 .../wayside/memory/wayside_simulation.go      | 28 +++++++++++++++++--
 2 files changed, 27 insertions(+), 4 deletions(-)

diff --git a/message_server/psl_ms.go b/message_server/psl_ms.go
index abfac53..0aa2390 100644
--- a/message_server/psl_ms.go
+++ b/message_server/psl_ms.go
@@ -45,7 +45,8 @@ func collectGateBoxPSLState(world ecs.World, mapId int32, box *data_proto.GatedB
 		for _, button := range pslStorage.PslButtons {
 			btnUidMap[boxUid+"_"+button.Code] = memory.GetMapElementId(button.Common)
 		}
-		btnArr := []*ecs.Entry{mkx.PCB, mkx.POB, mkx.PAB}
+		btnArr := []*ecs.Entry{mkx.PCB, mkx.PCBPL, mkx.POB, mkx.POBPL, mkx.PAB, mkx.PABPL, mkx.WRZF, mkx.WRZFPL,
+			mkx.QKQR, mkx.QKQRPL, mkx.MPL, mkx.JXTCPL}
 		for _, btn := range btnArr {
 			if btn == nil {
 				continue
diff --git a/ts/simulation/wayside/memory/wayside_simulation.go b/ts/simulation/wayside/memory/wayside_simulation.go
index 3a0e86e..3fa068d 100644
--- a/ts/simulation/wayside/memory/wayside_simulation.go
+++ b/ts/simulation/wayside/memory/wayside_simulation.go
@@ -814,6 +814,10 @@ func buildAndRelateElectronicComponent(repo *proto.Repository, relayGi *data_pro
 							mkx.PobjId = relay.Id
 						} else if strings.Contains(relay.GetCode(), "PAB") || strings.Contains(relay.GetCode(), "PDB") {
 							mkx.PabjId = relay.Id
+						} else if strings.Contains(relay.GetCode(), "WRZF") {
+							mkx.WrzfjId = relay.Id
+						} else if strings.Contains(relay.GetCode(), "QKQR") {
+							mkx.QkqrjId = relay.Id
 						}
 					}
 				}
@@ -1214,11 +1218,29 @@ func fillProtoRepository(repo *proto.Repository, storage *data_proto.RtssGraphic
 			repo.Buttons = append(repo.Buttons, repoButton)
 			switch button.Code {
 			case "PCB":
-				mkx.PcbButtonId = repoButton.Id
+				mkx.PcbaId = repoButton.Id
+			case "PCBPL":
+				mkx.PcbplaId = repoButton.Id
 			case "POB":
-				mkx.PobButtonId = repoButton.Id
+				mkx.PobaId = repoButton.Id
+			case "POBPL":
+				mkx.PobplaId = repoButton.Id
 			case "PAB":
-				mkx.PabButtonId = repoButton.Id
+				mkx.PabaId = repoButton.Id
+			case "PABPL":
+				mkx.PabplaId = repoButton.Id
+			case "WRZF":
+				mkx.WrzfaId = repoButton.Id
+			case "WRZFPL":
+				mkx.WrzfplaId = repoButton.Id
+			case "QKQR":
+				mkx.QkqraId = repoButton.Id
+			case "QKQRPL":
+				mkx.QkqrplaId = repoButton.Id
+			case "MPL":
+				mkx.MplaId = repoButton.Id
+			case "JXTCPL":
+				mkx.JxtcplaId = repoButton.Id
 			}
 		}
 	}