From 6ea029d92c73be2ffde6429430c7ea971a4c390d Mon Sep 17 00:00:00 2001
From: Dimitri Stolnikov <horiz0n@gmx.net>
Date: Thu, 5 Jul 2012 00:28:52 +0200
Subject: [PATCH] add api function to control the IF gain for E4000 tuners

---
 include/rtl-sdr.h | 10 ++++++++++
 src/librtlsdr.c   | 38 +++++++++++++++++++++++++++-----------
 src/tuner_e4k.c   |  4 ++--
 3 files changed, 39 insertions(+), 13 deletions(-)

diff --git a/include/rtl-sdr.h b/include/rtl-sdr.h
index 1b0d82f..a7a3ee9 100644
--- a/include/rtl-sdr.h
+++ b/include/rtl-sdr.h
@@ -165,6 +165,16 @@ RTLSDR_API int rtlsdr_set_tuner_gain(rtlsdr_dev_t *dev, int gain);
  */
 RTLSDR_API int rtlsdr_get_tuner_gain(rtlsdr_dev_t *dev);
 
+/*!
+ * Set the intermediate frequency gain for the device.
+ *
+ * \param dev the device handle given by rtlsdr_open()
+ * \param stage intermediate frequency gain stage number (1 to 6 for E4000)
+ * \param gain in tenths of a dB, -30 means -3.0 dB.
+ * \return 0 on success
+ */
+RTLSDR_API int rtlsdr_set_tuner_if_gain(rtlsdr_dev_t *dev, int stage, int gain);
+
 /*!
  * Set the gain mode (automatic/manual) for the device.
  * Manual gain mode must be enabled for the gain setter function to work.
diff --git a/src/librtlsdr.c b/src/librtlsdr.c
index ec5a70d..1069314 100644
--- a/src/librtlsdr.c
+++ b/src/librtlsdr.c
@@ -51,7 +51,8 @@ typedef struct rtlsdr_tuner_iface {
 	int (*exit)(void *);
 	int (*set_freq)(void *, uint32_t freq /* Hz */);
 	int (*set_bw)(void *, int bw /* Hz */);
-	int (*set_gain)(void *, int gain /* dB */);
+	int (*set_gain)(void *, int gain /* tenth dB */);
+	int (*set_if_gain)(void *, int stage, int gain /* tenth dB */);
 	int (*set_gain_mode)(void *, int manual);
 } rtlsdr_tuner_iface_t;
 
@@ -80,7 +81,7 @@ struct rtlsdr_dev {
 	uint32_t tun_xtal; /* Hz */
 	uint32_t freq; /* Hz */
 	int corr; /* ppm */
-	int gain; /* dB */
+	int gain; /* tenth dB */
 	struct e4k_state e4k_s;
 };
 
@@ -113,14 +114,15 @@ int e4000_set_gain(void *dev, int gain) {
 	if(enhgain >= 0)
 		if(e4k_set_enh_gain(&devt->e4k_s, enhgain) == -EINVAL)
 			return -1;
-
 	return 0;
 }
-
+int e4000_set_if_gain(void *dev, int stage, int gain) {
+	rtlsdr_dev_t* devt = (rtlsdr_dev_t*)dev;
+	return e4k_if_gain_set(&devt->e4k_s, (uint8_t)stage, (int8_t)(gain / 10));
+}
 int e4000_set_gain_mode(void *dev, int manual) {
 	rtlsdr_dev_t* devt = (rtlsdr_dev_t*)dev;
-	e4k_enable_manual_gain(&devt->e4k_s, manual);
-	return 0;
+	return e4k_enable_manual_gain(&devt->e4k_s, manual);
 }
 
 int _fc0012_init(void *dev) { return fc0012_init(dev); }
@@ -157,26 +159,26 @@ int fc2580_set_gain_mode(void *dev, int manual) { return 0; }
 /* definition order must match enum rtlsdr_tuner */
 static rtlsdr_tuner_iface_t tuners[] = {
 	{
-		NULL, NULL, NULL, NULL, NULL, NULL /* dummy for unknown tuners */
+		NULL, NULL, NULL, NULL, NULL, NULL, NULL /* dummy for unknown tuners */
 	},
 	{
 		e4000_init, e4000_exit,
-		e4000_set_freq, e4000_set_bw, e4000_set_gain,
+		e4000_set_freq, e4000_set_bw, e4000_set_gain, e4000_set_if_gain,
 		e4000_set_gain_mode
 	},
 	{
 		_fc0012_init, fc0012_exit,
-		fc0012_set_freq, fc0012_set_bw, _fc0012_set_gain,
+		fc0012_set_freq, fc0012_set_bw, _fc0012_set_gain, NULL,
 		fc0012_set_gain_mode
 	},
 	{
 		_fc0013_init, fc0013_exit,
-		fc0013_set_freq, fc0013_set_bw, _fc0013_set_gain,
+		fc0013_set_freq, fc0013_set_bw, _fc0013_set_gain, NULL,
 		fc0013_set_gain_mode
 	},
 	{
 		fc2580_init, fc2580_exit,
-		_fc2580_set_freq, fc2580_set_bw, fc2580_set_gain,
+		_fc2580_set_freq, fc2580_set_bw, fc2580_set_gain, NULL,
 		fc2580_set_gain_mode
 	},
 };
@@ -749,6 +751,20 @@ int rtlsdr_get_tuner_gain(rtlsdr_dev_t *dev)
 	return dev->gain;
 }
 
+int rtlsdr_set_tuner_if_gain(rtlsdr_dev_t *dev, int stage, int gain)
+{
+	int r = 0;
+
+	if (!dev || !dev->tuner)
+		return -1;
+
+	if (dev->tuner->set_if_gain) {
+		r = dev->tuner->set_if_gain(dev, stage, gain);
+	}
+
+	return r;
+}
+
 int rtlsdr_set_tuner_gain_mode(rtlsdr_dev_t *dev, int mode)
 {
 	int r = 0;
diff --git a/src/tuner_e4k.c b/src/tuner_e4k.c
index 85a8a02..0232f88 100644
--- a/src/tuner_e4k.c
+++ b/src/tuner_e4k.c
@@ -723,8 +723,8 @@ static int find_stage_gain(uint8_t stage, int8_t val)
 
 /*! \brief Set the gain of one of the IF gain stages
  *  \param [e4k] handle to the tuner chip
- *  \param [stage] numbere of the stage (1..6)
- *  \param [value] gain value in dBm
+ *  \param [stage] number of the stage (1..6)
+ *  \param [value] gain value in dB
  *  \returns 0 on success, negative in case of error
  */
 int e4k_if_gain_set(struct e4k_state *e4k, uint8_t stage, int8_t value)