1 /*
2 * Copyright (C) 2014 by Kyle Keen <keenerd@gmail.com>
3 *
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 2 of the License, or
7 * (at your option) any later version.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
16 */
17
18 /* a collection of user friendly tools
19 * todo: use strtol for more flexible int parsing
20 * */
21
22 #include <string.h>
23 #include <stdio.h>
24 #include <stdlib.h>
25
26 #ifndef _WIN32
27 #include <unistd.h>
28 #else
29 #include <windows.h>
30 #include <fcntl.h>
31 #include <io.h>
32 #define _USE_MATH_DEFINES
33 #endif
34
35 #include <math.h>
36
37 #include "rtl-sdr.h"
38
atofs(char * s)39 double atofs(char *s)
40 /* standard suffixes */
41 {
42 char last;
43 int len;
44 double suff = 1.0;
45 len = strlen(s);
46 last = s[len-1];
47 s[len-1] = '\0';
48 switch (last) {
49 case 'g':
50 case 'G':
51 suff *= 1e3;
52 /* fall-through */
53 case 'm':
54 case 'M':
55 suff *= 1e3;
56 /* fall-through */
57 case 'k':
58 case 'K':
59 suff *= 1e3;
60 suff *= atof(s);
61 s[len-1] = last;
62 return suff;
63 }
64 s[len-1] = last;
65 return atof(s);
66 }
67
atoft(char * s)68 double atoft(char *s)
69 /* time suffixes, returns seconds */
70 {
71 char last;
72 int len;
73 double suff = 1.0;
74 len = strlen(s);
75 last = s[len-1];
76 s[len-1] = '\0';
77 switch (last) {
78 case 'h':
79 case 'H':
80 suff *= 60;
81 /* fall-through */
82 case 'm':
83 case 'M':
84 suff *= 60;
85 /* fall-through */
86 case 's':
87 case 'S':
88 suff *= atof(s);
89 s[len-1] = last;
90 return suff;
91 }
92 s[len-1] = last;
93 return atof(s);
94 }
95
atofp(char * s)96 double atofp(char *s)
97 /* percent suffixes */
98 {
99 char last;
100 int len;
101 double suff = 1.0;
102 len = strlen(s);
103 last = s[len-1];
104 s[len-1] = '\0';
105 switch (last) {
106 case '%':
107 suff *= 0.01;
108 suff *= atof(s);
109 s[len-1] = last;
110 return suff;
111 }
112 s[len-1] = last;
113 return atof(s);
114 }
115
nearest_gain(rtlsdr_dev_t * dev,int target_gain)116 int nearest_gain(rtlsdr_dev_t *dev, int target_gain)
117 {
118 int i, r, err1, err2, count, nearest;
119 int* gains;
120 r = rtlsdr_set_tuner_gain_mode(dev, 1);
121 if (r < 0) {
122 fprintf(stderr, "WARNING: Failed to enable manual gain.\n");
123 return r;
124 }
125 count = rtlsdr_get_tuner_gains(dev, NULL);
126 if (count <= 0) {
127 return 0;
128 }
129 gains = malloc(sizeof(int) * count);
130 count = rtlsdr_get_tuner_gains(dev, gains);
131 nearest = gains[0];
132 for (i=0; i<count; i++) {
133 err1 = abs(target_gain - nearest);
134 err2 = abs(target_gain - gains[i]);
135 if (err2 < err1) {
136 nearest = gains[i];
137 }
138 }
139 free(gains);
140 return nearest;
141 }
142
verbose_set_frequency(rtlsdr_dev_t * dev,uint32_t frequency)143 int verbose_set_frequency(rtlsdr_dev_t *dev, uint32_t frequency)
144 {
145 int r;
146 r = rtlsdr_set_center_freq(dev, frequency);
147 if (r < 0) {
148 fprintf(stderr, "WARNING: Failed to set center freq.\n");
149 } else {
150 fprintf(stderr, "Tuned to %u Hz.\n", frequency);
151 }
152 return r;
153 }
154
verbose_set_sample_rate(rtlsdr_dev_t * dev,uint32_t samp_rate)155 int verbose_set_sample_rate(rtlsdr_dev_t *dev, uint32_t samp_rate)
156 {
157 int r;
158 r = rtlsdr_set_sample_rate(dev, samp_rate);
159 if (r < 0) {
160 fprintf(stderr, "WARNING: Failed to set sample rate.\n");
161 } else {
162 fprintf(stderr, "Sampling at %u S/s.\n", samp_rate);
163 }
164 return r;
165 }
166
verbose_direct_sampling(rtlsdr_dev_t * dev,int on)167 int verbose_direct_sampling(rtlsdr_dev_t *dev, int on)
168 {
169 int r;
170 r = rtlsdr_set_direct_sampling(dev, on);
171 if (r != 0) {
172 fprintf(stderr, "WARNING: Failed to set direct sampling mode.\n");
173 return r;
174 }
175 if (on == 0) {
176 fprintf(stderr, "Direct sampling mode disabled.\n");}
177 if (on == 1) {
178 fprintf(stderr, "Enabled direct sampling mode, input 1/I.\n");}
179 if (on == 2) {
180 fprintf(stderr, "Enabled direct sampling mode, input 2/Q.\n");}
181 return r;
182 }
183
verbose_offset_tuning(rtlsdr_dev_t * dev)184 int verbose_offset_tuning(rtlsdr_dev_t *dev)
185 {
186 int r;
187 r = rtlsdr_set_offset_tuning(dev, 1);
188 if (r != 0) {
189 fprintf(stderr, "WARNING: Failed to set offset tuning.\n");
190 } else {
191 fprintf(stderr, "Offset tuning mode enabled.\n");
192 }
193 return r;
194 }
195
verbose_auto_gain(rtlsdr_dev_t * dev)196 int verbose_auto_gain(rtlsdr_dev_t *dev)
197 {
198 int r;
199 r = rtlsdr_set_tuner_gain_mode(dev, 0);
200 if (r != 0) {
201 fprintf(stderr, "WARNING: Failed to set tuner gain.\n");
202 } else {
203 fprintf(stderr, "Tuner gain set to automatic.\n");
204 }
205 return r;
206 }
207
verbose_gain_set(rtlsdr_dev_t * dev,int gain)208 int verbose_gain_set(rtlsdr_dev_t *dev, int gain)
209 {
210 int r;
211 r = rtlsdr_set_tuner_gain_mode(dev, 1);
212 if (r < 0) {
213 fprintf(stderr, "WARNING: Failed to enable manual gain.\n");
214 return r;
215 }
216 r = rtlsdr_set_tuner_gain(dev, gain);
217 if (r != 0) {
218 fprintf(stderr, "WARNING: Failed to set tuner gain.\n");
219 } else {
220 fprintf(stderr, "Tuner gain set to %0.2f dB.\n", gain/10.0);
221 }
222 return r;
223 }
224
verbose_ppm_set(rtlsdr_dev_t * dev,int ppm_error)225 int verbose_ppm_set(rtlsdr_dev_t *dev, int ppm_error)
226 {
227 int r;
228 if (ppm_error == 0) {
229 return 0;}
230 r = rtlsdr_set_freq_correction(dev, ppm_error);
231 if (r < 0) {
232 fprintf(stderr, "WARNING: Failed to set ppm error.\n");
233 } else {
234 fprintf(stderr, "Tuner error set to %i ppm.\n", ppm_error);
235 }
236 return r;
237 }
238
verbose_reset_buffer(rtlsdr_dev_t * dev)239 int verbose_reset_buffer(rtlsdr_dev_t *dev)
240 {
241 int r;
242 r = rtlsdr_reset_buffer(dev);
243 if (r < 0) {
244 fprintf(stderr, "WARNING: Failed to reset buffers.\n");}
245 return r;
246 }
247
verbose_device_search(char * s)248 int verbose_device_search(char *s)
249 {
250 int i, device_count, device, offset;
251 char *s2;
252 char vendor[256], product[256], serial[256];
253 device_count = rtlsdr_get_device_count();
254 if (!device_count) {
255 fprintf(stderr, "No supported devices found.\n");
256 return -1;
257 }
258 fprintf(stderr, "Found %d device(s):\n", device_count);
259 for (i = 0; i < device_count; i++) {
260 rtlsdr_get_device_usb_strings(i, vendor, product, serial);
261 fprintf(stderr, " %d: %s, %s, SN: %s\n", i, vendor, product, serial);
262 }
263 fprintf(stderr, "\n");
264 /* does string look like raw id number */
265 device = (int)strtol(s, &s2, 0);
266 if (s2[0] == '\0' && device >= 0 && device < device_count) {
267 fprintf(stderr, "Using device %d: %s\n",
268 device, rtlsdr_get_device_name((uint32_t)device));
269 return device;
270 }
271 /* does string exact match a serial */
272 for (i = 0; i < device_count; i++) {
273 rtlsdr_get_device_usb_strings(i, vendor, product, serial);
274 if (strcmp(s, serial) != 0) {
275 continue;}
276 device = i;
277 fprintf(stderr, "Using device %d: %s\n",
278 device, rtlsdr_get_device_name((uint32_t)device));
279 return device;
280 }
281 /* does string prefix match a serial */
282 for (i = 0; i < device_count; i++) {
283 rtlsdr_get_device_usb_strings(i, vendor, product, serial);
284 if (strncmp(s, serial, strlen(s)) != 0) {
285 continue;}
286 device = i;
287 fprintf(stderr, "Using device %d: %s\n",
288 device, rtlsdr_get_device_name((uint32_t)device));
289 return device;
290 }
291 /* does string suffix match a serial */
292 for (i = 0; i < device_count; i++) {
293 rtlsdr_get_device_usb_strings(i, vendor, product, serial);
294 offset = strlen(serial) - strlen(s);
295 if (offset < 0) {
296 continue;}
297 if (strncmp(s, serial+offset, strlen(s)) != 0) {
298 continue;}
299 device = i;
300 fprintf(stderr, "Using device %d: %s\n",
301 device, rtlsdr_get_device_name((uint32_t)device));
302 return device;
303 }
304 fprintf(stderr, "No matching devices found.\n");
305 return -1;
306 }
307
308 // vim: tabstop=8:softtabstop=8:shiftwidth=8:noexpandtab
309