python: Fixes --samples argument by multiplying num_samples by sample…

…_size Fixes #714 Signed-off-by: Cormier, Jonathan <[email protected]>
analogdevicesinc · Jul 14, 2021 · 84a7fd4 · 84a7fd4
1 parent 48e5852
commit 84a7fd4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bindings/python/examples/iio_readdev.py b/bindings/python/examples/iio_readdev.py
@@ -214,6 +214,7 @@ def __init__(self, ctx, arguments):
         """
         buffer_builder = BufferBuilder(ctx, arguments)
         self.buffer = buffer_builder.create()
+        self.device = buffer_builder.dev
         self.arguments = arguments
 
     def read(self):
@@ -224,7 +225,7 @@ def read(self):
 
             if self.arguments.num_samples > 0:
                 sys.stdout.buffer.write(
-                    samples[: min(self.arguments.num_samples, len(samples))]
+                    samples[: min(self.arguments.num_samples * self.device.sample_size, len(samples))]
                 )
                 self.arguments.num_samples -= min(
                     self.arguments.num_samples, len(samples)