From 9ae29c65e57b2adbcef742b8879f69912367b41d Mon Sep 17 00:00:00 2001
From: Thierry Zamofing <thierry.zamofing@psi.ch>
Date: Wed, 30 Aug 2023 12:53:01 +0200
Subject: [PATCH] enhance generate.py

---
 generate.py | 49 ++++++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 44 insertions(+), 5 deletions(-)

diff --git a/generate.py b/generate.py
index c678541..218a0ae 100755
--- a/generate.py
+++ b/generate.py
@@ -204,7 +204,7 @@ powerPmacCreateController("{port}", 200, 1000)
 powerPmacCreateAxis({PORT}, {ADDR})
 '''
 
-  tplEVR='''
+  tplEXPMX1='''
 ###### initialize EVR ############
 # https://git.psi.ch/epics_driver_modules/mrfioc2
 require mrfioc2
@@ -214,6 +214,41 @@ runScript $(mrfioc2_DIR)/mrfioc2_evr-PCIe.cmd, "DEVICE=EVR0,EVR_SUBS=$(ESB_MX_DI
 # https://git.psi.ch/epics_driver_modules/mrfioc2_regDev
 require mrfioc2_regDev
 runScript $(mrfioc2_regDev_DIR)/mrfioc2_regDev_pulseID_RX.cmd, "DEVICE=EVR0,SYS=SAR-EXPMX"
+'''
+
+  tplEXPMX2='''
+#powerPmacCreateVirtualAxis(<port_name>, <axis_index>, <coord_sys_index>, <axis_name>,<commaSeparatedMotorIndices)
+# !!! COMMENTS AFTER 'powerPmacCreateVirtualAxis("$(PORT)",  9, 1, "X",1)' IS NOT ALLOWED !!!
+powerPmacCreateVirtualAxis(PPMAC, 9 , 1, "X",1,2,3,4)
+powerPmacCreateVirtualAxis(PPMAC, 10, 1, "Y",1,2,3,4)
+powerPmacCreateVirtualAxis(PPMAC, 11, 1, "A",1,2,3,4)
+powerPmacCreateVirtualAxis(PPMAC, 12, 1, "B",1,2,3,4)
+'''
+
+  tplEXPMX3='''
+#powerPmacCreateVirtualAxis(<port_name>, <axis_index>, <coord_sys_index>, <axis_name>,<commaSeparatedMotorIndices)
+# !!! COMMENTS AFTER 'powerPmacCreateVirtualAxis("$(PORT)",  9, 1, "X",1)' IS NOT ALLOWED !!!
+powerPmacCreateVirtualAxis(PPMAC, 9 , 1, "X",1,2,3,4)
+powerPmacCreateVirtualAxis(PPMAC, 10, 1, "Y",1,2,3,4)
+powerPmacCreateVirtualAxis(PPMAC, 11, 1, "U",1,2,3,4)
+powerPmacCreateVirtualAxis(PPMAC, 12, 1, "V",1,2,3,4)
+powerPmacCreateVirtualAxis(PPMAC, 13, 1, "W",1,2,3,4)
+
+# Load databse
+#dbLoadTemplate("$(ESB_MX_TEMPLATES)/$(P)3.subs", "PORT_M=$(PORT),P_M=$(P)")
+
+require PB_EXTRA
+#- Brakes - need to load motor status PVs as well
+dbLoadRecords("$(PB_EXTRA_TEMPLATES)motorStatus.db" "SYS=SAR-EXPMX, PORT=PPMAC, M=MOT_GIR_1, M_NO=1")
+dbLoadRecords("$(PB_EXTRA_TEMPLATES)motorStatus.db" "SYS=SAR-EXPMX, PORT=PPMAC, M=MOT_GIR_2, M_NO=2")
+dbLoadRecords("$(PB_EXTRA_TEMPLATES)motorStatus.db" "SYS=SAR-EXPMX, PORT=PPMAC, M=MOT_GIR_3, M_NO=3")
+dbLoadRecords("$(PB_EXTRA_TEMPLATES)motorStatus.db" "SYS=SAR-EXPMX, PORT=PPMAC, M=MOT_GIR_4, M_NO=4")
+dbLoadRecords("$(PB_EXTRA_TEMPLATES)motorStatus.db" "SYS=SAR-EXPMX, PORT=PPMAC, M=MOT_GIR_5, M_NO=5")
+dbLoadRecords("${PB_EXTRA_TEMPLATES}motorBrake.db", "SYS=SAR-EXPMX, PORT=PPMAC, M=MOT_GIR_1, M_NO=1")
+dbLoadRecords("${PB_EXTRA_TEMPLATES}motorBrake.db", "SYS=SAR-EXPMX, PORT=PPMAC, M=MOT_GIR_2, M_NO=2")
+dbLoadRecords("${PB_EXTRA_TEMPLATES}motorBrake.db", "SYS=SAR-EXPMX, PORT=PPMAC, M=MOT_GIR_3, M_NO=3")
+dbLoadRecords("${PB_EXTRA_TEMPLATES}motorBrake.db", "SYS=SAR-EXPMX, PORT=PPMAC, M=MOT_GIR_4, M_NO=4")
+dbLoadRecords("${PB_EXTRA_TEMPLATES}motorBrake.db", "SYS=SAR-EXPMX, PORT=PPMAC, M=MOT_GIR_5, M_NO=5")
 '''
 
   pwrBrkSubsHdr='''\
@@ -267,7 +302,6 @@ file PPMACMotor.template
 
   def genPowerBrickIOC(self):
     yamlParam=Generate.yamlParam
-    tplEVR=Generate.tplEVR
     tplCtrl=Generate.tplPwrBrkCtrl
     tplAxis=Generate.tplPwrBrkAxis
     #tplAxSR=Generate.tplMcsAxisSaveRestore
@@ -317,12 +351,17 @@ file PPMACMotor.template
         motPar['PORT']=v['port']
         motPar['P']=v['prefix']
         fh_sb.write(fmt.format(**motPar))
-        fh_ss.write(tplAxis.format(**motPar))
-        #fh_sr.write(tplAxSR.format(prefix=prefix,m=m))
+        if motPar['ADDR']<=8:
+          fh_ss.write(tplAxis.format(**motPar))
       fh_sb.write('}\n\n')
 
       if ioc=='SAR-CPPM-EXPMX1':
-        fh_ss.write(tplEVR)
+        fh_ss.write(Generate.tplEXPMX1)
+      elif ioc=='SAR-CPPM-EXPMX2':
+        fh_ss.write(Generate.tplEXPMX2)
+      elif ioc=='SAR-CPPM-EXPMX3':
+        fh_ss.write(Generate.tplEXPMX3)
+
       fh_ss.close()
       fh_sb.close()