bladerf: make source less picky wrt rf gain

1 files changed, 7 insertions, 10 deletions

diff --git a/lib/bladerf/bladerf_source_c.cc b/lib/bladerf/bladerf_source_c.cc
index aba6964..5196044 100644
--- a/lib/bladerf/bladerf_source_c.cc
+++ b/lib/bladerf/bladerf_source_c.cc
@@ -496,17 +496,14 @@ double bladerf_source_c::set_gain( double gain, const std::string & name, size_t
 
   if( name == "LNA" ) {
     bladerf_lna_gain g;
-    if( gain == 0.0 ) {
-      g = BLADERF_LNA_GAIN_BYPASS;
-    } else if( gain == 3.0 ) {
-      g = BLADERF_LNA_GAIN_MID;
-    } else if( gain == 6.0 ) {
-      g = BLADERF_LNA_GAIN_MAX;
-    } else {
-      std::cerr << "Invalid LNA gain requested: " << gain << ", "
-                << "setting to LNA_MAX (6dB)" << std::endl;
+
+    if ( gain >= 6.0f )
       g = BLADERF_LNA_GAIN_MAX;
-    }
+    else if ( gain >= 3.0f )
+      g = BLADERF_LNA_GAIN_MID;
+    else /* gain < 3.0f */
+      g = BLADERF_LNA_GAIN_BYPASS;
+
     ret = bladerf_set_lna_gain( _dev, g );
   } else if( name == "VGA1" ) {
     ret = bladerf_set_rxvga1( _dev, (int)gain );