Merge pull request #449 from SaikiranGudla/master

rx_tx: Add tx_data_type
analogdevicesinc · Jul 5, 2023 · 0778bab · 0778bab
2 parents 03aa90a + 7ad8b30
commit 0778bab
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/adi/rx_tx.py b/adi/rx_tx.py
@@ -307,6 +307,7 @@ class tx(dds, rx_tx_common):
     _txdac: iio.Device = []
     _tx_channel_names: List[str] = []
     _complex_data = False
+    _tx_data_type = np.int16
     __txbuf = None
     _output_byte_filename = "out.bin"
     _push_to_file = False
@@ -459,7 +460,7 @@ def tx(self, data_np=None):
 
             indx = 0
             stride = self._num_tx_channels_enabled * 2
-            data = np.empty(stride * len(data_np[0]), dtype=np.int16)
+            data = np.empty(stride * len(data_np[0]), dtype=self._tx_data_type)
             for chan in data_np:
                 i = np.real(chan)
                 q = np.imag(chan)
@@ -475,7 +476,7 @@ def tx(self, data_np=None):
 
             indx = 0
             stride = self._num_tx_channels_enabled
-            data = np.empty(stride * len(data_np[0]), dtype=np.int16)
+            data = np.empty(stride * len(data_np[0]), dtype=self._tx_data_type)
             for chan in data_np:
                 data[indx::stride] = chan.astype(int)
                 indx = indx + 1