automatic DC offset corrections

src/device.rs

@@ -214,6 +214,22 @@ pub trait DeviceTrait: Any + Send {
     ///
     /// Returns `Err(Error::NotSupported)` if unsupported in underlying driver.
     fn get_bandwidth_range(&self, direction: Direction, channel: usize) -> Result<Range, Error>;
+
+    //========================= AUTOMATIC DC OFFSET CORRECTIONS ===============================
+
+    /// Returns true if automatic corrections are supported
+    fn has_dc_offset_mode(&self, direction: Direction, channel: usize) -> Result<bool, Error>;
+
+    /// Enable or disable automatic DC offset corrections mode.
+    fn set_dc_offset_mode(
+        &self,
+        direction: Direction,
+        channel: usize,
+        automatic: bool,
+    ) -> Result<(), Error>;
+
+    /// Returns true if automatic DC offset mode is enabled
+    fn dc_offset_mode(&self, direction: Direction, channel: usize) -> Result<bool, Error>;
 }
 
 /// Wrapps a driver, implementing the [DeviceTrait].
@@ -619,6 +635,23 @@ impl<
     fn get_bandwidth_range(&self, direction: Direction, channel: usize) -> Result<Range, Error> {
         self.dev.get_bandwidth_range(direction, channel)
     }
+
+    fn has_dc_offset_mode(&self, direction: Direction, channel: usize) -> Result<bool, Error> {
+        self.dev.has_dc_offset_mode(direction, channel)
+    }
+
+    fn set_dc_offset_mode(
+        &self,
+        direction: Direction,
+        channel: usize,
+        automatic: bool,
+    ) -> Result<(), Error> {
+        self.dev.set_dc_offset_mode(direction, channel, automatic)
+    }
+
+    fn dc_offset_mode(&self, direction: Direction, channel: usize) -> Result<bool, Error> {
+        self.dev.dc_offset_mode(direction, channel)
+    }
 }
 
 #[doc(hidden)]
@@ -812,6 +845,24 @@ impl DeviceTrait for GenericDevice {
     fn get_bandwidth_range(&self, direction: Direction, channel: usize) -> Result<Range, Error> {
         self.as_ref().get_bandwidth_range(direction, channel)
     }
+
+    fn has_dc_offset_mode(&self, direction: Direction, channel: usize) -> Result<bool, Error> {
+        self.as_ref().has_dc_offset_mode(direction, channel)
+    }
+
+    fn set_dc_offset_mode(
+        &self,
+        direction: Direction,
+        channel: usize,
+        automatic: bool,
+    ) -> Result<(), Error> {
+        self.as_ref()
+            .set_dc_offset_mode(direction, channel, automatic)
+    }
+
+    fn dc_offset_mode(&self, direction: Direction, channel: usize) -> Result<bool, Error> {
+        self.as_ref().dc_offset_mode(direction, channel)
+    }
 }
 
 impl<

src/impls/aaronia_http.rs

@@ -552,6 +552,23 @@ impl DeviceTrait for AaroniaHttp {
     fn get_bandwidth_range(&self, _direction: Direction, _channel: usize) -> Result<Range, Error> {
         Err(Error::NotSupported)
     }
+
+    fn has_dc_offset_mode(&self, _direction: Direction, _channel: usize) -> Result<bool, Error> {
+        Err(Error::NotSupported)
+    }
+
+    fn set_dc_offset_mode(
+        &self,
+        _direction: Direction,
+        _channel: usize,
+        _automatic: bool,
+    ) -> Result<(), Error> {
+        Err(Error::NotSupported)
+    }
+
+    fn dc_offset_mode(&self, _direction: Direction, _channel: usize) -> Result<bool, Error> {
+        Err(Error::NotSupported)
+    }
 }
 
 impl RxStreamer {

src/impls/dummy.rs

@@ -427,6 +427,23 @@ impl DeviceTrait for Dummy {
             Err(Error::ValueError)
         }
     }
+
+    fn has_dc_offset_mode(&self, _direction: Direction, _channel: usize) -> Result<bool, Error> {
+        Err(Error::NotSupported)
+    }
+
+    fn set_dc_offset_mode(
+        &self,
+        _direction: Direction,
+        _channel: usize,
+        _automatic: bool,
+    ) -> Result<(), Error> {
+        Err(Error::NotSupported)
+    }
+
+    fn dc_offset_mode(&self, _direction: Direction, _channel: usize) -> Result<bool, Error> {
+        Err(Error::NotSupported)
+    }
 }
 
 impl crate::RxStreamer for RxStreamer {

src/impls/rtlsdr.rs

@@ -414,6 +414,23 @@ impl DeviceTrait for RtlSdr {
     fn get_bandwidth_range(&self, _direction: Direction, _channel: usize) -> Result<Range, Error> {
         Err(Error::NotSupported)
     }
+
+    fn has_dc_offset_mode(&self, _direction: Direction, _channel: usize) -> Result<bool, Error> {
+        Err(Error::NotSupported)
+    }
+
+    fn set_dc_offset_mode(
+        &self,
+        _direction: Direction,
+        _channel: usize,
+        _automatic: bool,
+    ) -> Result<(), Error> {
+        Err(Error::NotSupported)
+    }
+
+    fn dc_offset_mode(&self, _direction: Direction, _channel: usize) -> Result<bool, Error> {
+        Err(Error::NotSupported)
+    }
 }
 
 impl crate::RxStreamer for RxStreamer {

src/impls/soapy.rs

@@ -320,6 +320,25 @@ impl DeviceTrait for Soapy {
         let range = self.dev.bandwidth_range(direction.into(), channel)?;
         Ok(range.into())
     }
+
+    fn has_dc_offset_mode(&self, direction: Direction, channel: usize) -> Result<bool, Error> {
+        Ok(self.dev.has_dc_offset_mode(direction.into(), channel)?)
+    }
+
+    fn set_dc_offset_mode(
+        &self,
+        direction: Direction,
+        channel: usize,
+        automatic: bool,
+    ) -> Result<(), Error> {
+        Ok(self
+            .dev
+            .set_dc_offset_mode(direction.into(), channel, automatic)?)
+    }
+
+    fn dc_offset_mode(&self, direction: Direction, channel: usize) -> Result<bool, Error> {
+        Ok(self.dev.dc_offset_mode(direction.into(), channel)?)
+    }
 }
 
 impl crate::RxStreamer for RxStreamer {