Changeset 32:22525b9438b5 in xplra for src/client

Timestamp:

02/09/13 10:28:16 (11 years ago)

Author:

István Váradi <ivaradi@…>

Branch:

default

hg-Phase:

(<MercurialRepository 3 'hg:/home/ivaradi/xplane/hg/xplra' '/'>, 'public')

Message:

Implemented the setting of single datarefs

File:

• src/client/python/xplra.py (modified) (2 diffs)

src/client/python/xplra.py

@@ -155 +155 @@
         """Get the value of the byte array dataref with the given name as a
         string."""
-        return self.getByteArray(name, offset = offset).tostring()
+        s = ""
+        for b in self.getByteArray(name, offset = offset):
+            if b==0: break
+            s += chr(b)
+        return s
+
+    def setInt(self, name, value):
+        """Set the value of the integer dataref with the given name."""
+        self._setSingle(name, TYPE_INT, value)
+
+    def setFloat(self, name, value):
+        """Set the value of the float dataref with the given name."""
+        self._setSingle(name, TYPE_FLOAT, value)
+
+    def setDouble(self, name, value):
+        """Set the value of the double dataref with the given name."""
+        self._setSingle(name, TYPE_DOUBLE, value)
+
+    def setFloatArray(self, name, value, offset = 0):
+        """Set the value of the float array dataref with the given name."""
+        self._setSingle(name, TYPE_FLOAT_ARRAY, value, offset = offset)
+
+    def setIntArray(self, name, value, offset = 0):
+        """Set the value of the integer array dataref with the given name."""
+        self._setSingle(name, TYPE_INT_ARRAY, value, offset = offset)
+
+    def setByteArray(self, name, value, offset = 0):
+        """Set the value of the byte array dataref with the given name."""
+        self._setSingle(name, TYPE_BYTE_ARRAY, value, offset = offset)
+
+    def setString(self, name, value, length, offset = 0):
+        """Set the value of the byte array dataref with the given name from a
+        string.
+
+        The string will be padded with 0's so that its length is the given
+        one."""
+        value = [ord(c) for c in value[:length]]
+        value += [0] * (length - len(value))
+        self.setByteArray(name, value, offset)
 
     def disconnect(self):
@@ -211 +249 @@
             return arr
 
+    def _setSingle(self, name, type, value, offset = None):
+        """Set the single value of the given name and type."""
+        self._writeU8(COMMAND_SET_SINGLE)
+        self._writeString(name)
+        self._writeU8(type)
+        if offset is not None:
+            self._writeS32(len(value))
+            self._writeS32(offset)
+
+        self._writeValue(type, value)
+        self._flush()
+        self._checkResult()
+
+    def _writeValue(self, type, value):
+        """Write a value of the given type."""
+        if type==TYPE_INT:
+            self._writeS32(int(value))
+        elif type==TYPE_FLOAT:
+            self._writeFloat(float(value))
+        elif type==TYPE_DOUBLE:
+            self._writeDouble(float(value))
+        else:
+            arr = None
+            if type==TYPE_FLOAT_ARRAY:
+                arr = array.array("f")
+            elif type==TYPE_INT_ARRAY:
+                arr = array.array("i")
+            elif type==TYPE_BYTE_ARRAY:
+                arr = array.array("B")
+            arr.fromlist(value)
+            self._stream.write(arr.tostring())
+
     def _writeU8(self, x):
         """Write the given value as an unsigned, 8-bit value."""