RF: Do device setup for BBTK Button Box in setupDevices

psychopy · Nov 22, 2023 · 4e78268 · 4e78268
1 parent e16447b
commit 4e78268
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/psychopy_bbtk/components/tpad.py b/psychopy_bbtk/components/tpad.py
@@ -61,14 +61,16 @@ def addRequirements(self):
             importFrom="psychopy_bbtk"
         )
 
-    def writeInitCode(self, buff):
+    def writeDeviceCode(self, buff):
         # get inits
         inits = getInitVals(self.params)
-        # make Keyboard object
+        # make ButtonGroup object
         code = (
-            "%(name)s = tpad.TPadButtonGroup(\n"
-            "    %(serialPort)s,\n"
-            "    channels=%(nButtons)s\n"
+            "deviceManager.addDevice(\n"
+            "    deviceClass='psychopy_bbtk.tpad.TPadButtonGroup',\n"
+            "    deviceName=%(deviceName)s,\n"
+            "    pad=%(serialPort)s,\n"
+            "    channels=%(nButtons)s,\n"
             ")\n"
         )
-        buff.writeIndentedLines(code % inits)
+        buff.writeOnceIndentedLines(code % inits)