]> git.rkrishnan.org Git - dttsp.git/blob - jDttSP/sdr.c
ead39f0e61391978bd1f065ad70de6c3497f1be3
[dttsp.git] / jDttSP / sdr.c
1 /* sdr.c
2
3 This file is part of a program that implements a Software-Defined Radio.
4
5 Copyright (C) 2004 by Frank Brickle, AB2KT and Bob McGwier, N4HY
6
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 2 of the License, or
10 (at your option) any later version.
11
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 GNU General Public License for more details.
16
17 You should have received a copy of the GNU General Public License
18 along with this program; if not, write to the Free Software
19 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
20
21 The authors can be reached by email at
22
23 ab2kt@arrl.net
24 or
25 rwmcgwier@comcast.net
26
27 or by paper mail at
28
29 The DTTS Microwave Society
30 6 Kathleen Place
31 Bridgewater, NJ 08807
32 */
33
34 #include <common.h>
35
36 //========================================================================
37 /* initialization and termination */
38
39 /* global and general info,
40    not specifically attached to
41    tx, rx, or scheduling */
42
43 PRIVATE void
44 setup_all(void) {
45
46   uni.samplerate = loc.def.rate;
47   uni.buflen = loc.def.size;
48   uni.mode.sdr = loc.def.mode;
49   uni.mode.trx = RX;
50
51   if (uni.meter.flag) {
52     uni.meter.chan.path = loc.path.meter;
53     uni.meter.chan.size = loc.mult.ring * sizeof(REAL);
54     uni.meter.val = -200.0;
55     uni.meter.chan.c = openChan(uni.meter.chan.path, uni.meter.chan.size);
56   }
57
58   uni.wisdom.path = loc.path.wisdom;
59   uni.wisdom.bits = FFTW_OUT_OF_PLACE | FFTW_ESTIMATE;
60   {
61     FILE *f = fopen(uni.wisdom.path, "r");
62     if (f) {
63 #define WBUFLEN 2048
64 #define WSTRLEN 64      
65       char *line = alloca(WBUFLEN);
66       fgets(line, WBUFLEN, f);
67       if ((strlen(line) > WSTRLEN) &&
68           (fftw_import_wisdom_from_string(line) != FFTW_FAILURE))
69         uni.wisdom.bits = FFTW_OUT_OF_PLACE | FFTW_MEASURE | FFTW_USE_WISDOM;
70 #undef WSTRLEN
71 #undef WBUFLEN      
72       fclose(f);
73     }
74   }
75
76   uni.multirx.lis = 0;
77   uni.multirx.nrx = loc.def.nrx;
78
79   uni.mix.rx.flag = uni.mix.tx.flag = FALSE;
80   uni.mix.rx.gain = uni.mix.tx.gain = 1.0;
81
82   uni.tick = 0;
83 }
84
85 /* purely rx */
86
87 PRIVATE void
88 setup_rx(int k) {
89
90   /* conditioning */
91   rx[k].iqfix = newCorrectIQ(0.0, 1.0);
92   rx[k].filt.coef = newFIR_Bandpass_COMPLEX(-4800.0,
93                                             4800.0,
94                                             uni.samplerate,
95                                             uni.buflen + 1);
96   rx[k].filt.ovsv = newFiltOvSv(FIRcoef(rx[k].filt.coef),
97                                 FIRsize(rx[k].filt.coef),
98                                 uni.wisdom.bits);
99   normalize_vec_COMPLEX(rx[k].filt.ovsv->zfvec,
100                         rx[k].filt.ovsv->fftlen);
101
102   // hack for EQ
103   rx[k].filt.save = newvec_COMPLEX(rx[k].filt.ovsv->fftlen, "RX filter cache");
104   memcpy((char *) rx[k].filt.save,
105          (char *) rx[k].filt.ovsv->zfvec,
106          rx[k].filt.ovsv->fftlen * sizeof(COMPLEX));
107
108   /* buffers */
109   /* note we overload the internal filter buffers
110      we just created */
111   rx[k].buf.i = newCXB(FiltOvSv_fetchsize(rx[k].filt.ovsv),
112                        FiltOvSv_fetchpoint(rx[k].filt.ovsv),
113                        "init rx.buf.i");
114   rx[k].buf.o = newCXB(FiltOvSv_storesize(rx[k].filt.ovsv),
115                        FiltOvSv_storepoint(rx[k].filt.ovsv),
116                        "init rx[k].buf.o");
117   
118   /* conversion */
119   rx[k].osc.freq = -11025.0;
120   rx[k].osc.phase = 0.0;
121   rx[k].osc.gen = newOSC(uni.buflen,
122                          ComplexTone,
123                          rx[k].osc.freq,
124                          rx[k].osc.phase,
125                          uni.samplerate,
126                          "SDR RX Oscillator");
127
128   rx[k].agc.gen = newDigitalAgc(agcMED, // Mode
129                              7,         // Hang
130                              7,         // Size
131                              48,        // Ramp
132                              3,         // Over
133                              3,         // Rcov
134                              CXBsize(rx[k].buf.o),      // BufSize
135                              100.0,     // MaxGain
136                              0.707,     // Limit
137                              1.0,       // CurGain
138                              CXBbase(rx[k].buf.o));
139   rx[k].agc.flag = TRUE;
140
141   /* demods */
142   rx[k].am.gen = newAMD(48000.0,        // REAL samprate
143                         0.0,    // REAL f_initial
144                         -500.0, // REAL f_lobound,
145                         500.0,  // REAL f_hibound,
146                         400.0,  // REAL f_bandwid,
147                         CXBsize(rx[k].buf.o),   // int size,
148                         CXBbase(rx[k].buf.o),   // COMPLEX *ivec,
149                         CXBbase(rx[k].buf.o),   // COMPLEX *ovec,
150                         AMdet,  // AM Mode AMdet == rectifier,
151                                 //         SAMdet == synchronous detector
152                         "AM detector blew");    // char *tag
153   rx[k].fm.gen = newFMD(48000,  // REAL samprate
154                         0.0,    // REAL f_initial
155                         -6000.0,        // REAL f_lobound
156                         6000.0, // REAL f_hibound
157                         10000.0,        // REAL f_bandwid
158                         CXBsize(rx[k].buf.o),   // int size
159                         CXBbase(rx[k].buf.o),   // COMPLEX *ivec
160                         CXBbase(rx[k].buf.o),   // COMPLEX *ovec
161                         "New FM Demod structure");      // char *error message;
162
163   /* noise reduction */
164   rx[k].anf.gen = new_lmsr(rx[k].buf.o, // CXB signal,
165                            64,          // int delay,
166                            0.01,                // REAL adaptation_rate,
167                            0.00001,     // REAL leakage,
168                            45,          // int adaptive_filter_size,
169                            LMADF_INTERFERENCE);
170   rx[k].anf.flag = FALSE;
171   rx[k].anr.gen = new_lmsr(rx[k].buf.o, // CXB signal,
172                            64,          // int delay,
173                            0.01,                // REAL adaptation_rate,
174                            0.00001,     // REAL leakage,
175                            45,          // int adaptive_filter_size,
176                            LMADF_NOISE);
177   rx[k].anr.flag = FALSE;
178
179   rx[k].nb.thresh = 3.3;
180   rx[k].nb.gen = new_noiseblanker(rx[k].buf.i, rx[k].nb.thresh);
181   rx[k].nb.flag = FALSE;
182
183   rx[k].nb_sdrom.thresh = 2.5;
184   rx[k].nb_sdrom.gen = new_noiseblanker(rx[k].buf.i, rx[k].nb_sdrom.thresh);
185   rx[k].nb_sdrom.flag = FALSE;
186
187   rx[k].spot.gen = newSpotToneGen(-12.0,        // gain
188                                   700.0,        // freq
189                                   5.0,  // ms rise
190                                   5.0,  // ms fall
191                                   uni.buflen,
192                                   uni.samplerate);
193
194   rx[k].scl.pre.val = 1.0;
195   rx[k].scl.pre.flag = FALSE;
196   rx[k].scl.post.val = 1.0;
197   rx[k].scl.post.flag = FALSE;
198
199   memset((char *) &rx[k].squelch, 0, sizeof(rx[k].squelch));
200   rx[k].squelch.thresh = -30.0;
201   rx[k].squelch.power = 0.0;
202   rx[k].squelch.flag = rx[k].squelch.running = rx[k].squelch.set = FALSE;
203   rx[k].squelch.num = (int) (0.0395 * uni.samplerate + 0.5);
204
205   rx[k].mode = uni.mode.sdr;
206   rx[k].bin.flag = FALSE;
207
208   {
209     REAL pos = 0.5, // 0 <= pos <= 1, left->right
210          theta = (1.0 - pos) * M_PI / 2.0;
211     rx[k].azim = Cmplx(cos(theta), sin(theta));
212   }
213
214   rx[k].tick = 0;
215 }
216
217 /* purely tx */
218
219 PRIVATE void
220 setup_tx(void) {
221
222   /* conditioning */
223   tx.iqfix = newCorrectIQ(0.0, 1.0);
224   tx.filt.coef = newFIR_Bandpass_COMPLEX(300.0,
225                                          3000.0,
226                                          uni.samplerate,
227                                          uni.buflen + 1);
228   tx.filt.ovsv = newFiltOvSv(FIRcoef(tx.filt.coef),
229                              FIRsize(tx.filt.coef),
230                              uni.wisdom.bits);
231   normalize_vec_COMPLEX(tx.filt.ovsv->zfvec,
232                         tx.filt.ovsv->fftlen);
233
234   // hack for EQ
235   tx.filt.save = newvec_COMPLEX(tx.filt.ovsv->fftlen, "TX filter cache");
236   memcpy((char *) tx.filt.save,
237          (char *) tx.filt.ovsv->zfvec,
238          tx.filt.ovsv->fftlen * sizeof(COMPLEX));
239
240   /* buffers */
241   tx.buf.i = newCXB(FiltOvSv_fetchsize(tx.filt.ovsv),
242                     FiltOvSv_fetchpoint(tx.filt.ovsv),
243                     "init tx.buf.i");
244   tx.buf.o = newCXB(FiltOvSv_storesize(tx.filt.ovsv),
245                     FiltOvSv_storepoint(tx.filt.ovsv),
246                     "init tx.buf.o");
247   
248   /* conversion */
249   tx.osc.freq = 0.0;
250   tx.osc.phase = 0.0;
251   tx.osc.gen = newOSC(uni.buflen,
252                       ComplexTone,
253                       tx.osc.freq,
254                       tx.osc.phase,
255                       uni.samplerate,
256                       "SDR TX Oscillator");
257
258   tx.agc.gen = newDigitalAgc(agcFAST,   // Mode
259                              3,         // Hang
260                              3,         // Size
261                              3,         // Over
262                              3,         // Rcov
263                              48,        // Ramp
264                              CXBsize(tx.buf.o), // BufSize
265                              1.0,       // MaxGain
266                              0.900,     // Limit
267                              1.0,       // CurGain
268                              CXBbase(tx.buf.o));
269   tx.agc.flag = TRUE;
270
271   tx.spr.gen = newSpeechProc(0.4, 10.0, CXBbase(tx.buf.i), CXBsize(tx.buf.i));
272   tx.spr.flag = FALSE;
273
274   tx.scl.dc = cxzero;
275   tx.scl.pre.val = 1.0;
276   tx.scl.pre.flag = FALSE;
277   tx.scl.post.val = 1.0;
278   tx.scl.post.flag = FALSE;
279
280   tx.mode = uni.mode.sdr;
281
282   tx.tick = 0;
283   /* not much else to do for TX */
284 }
285
286 /* how the outside world sees it */
287
288 void
289 setup_workspace(void) {
290   int k;
291
292   setup_all();
293
294   for (k = 0; k < uni.multirx.nrx; k++) {
295     setup_rx(k);
296     uni.multirx.act[k] = FALSE;
297   }
298   uni.multirx.act[0] = TRUE;
299   uni.multirx.nac = 1;
300
301   setup_tx();
302 }
303
304 void
305 destroy_workspace(void) {
306   int k;
307
308   /* TX */
309   delSpeechProc(tx.spr.gen);
310   delDigitalAgc(tx.agc.gen);
311   delOSC(tx.osc.gen);
312   delvec_COMPLEX(tx.filt.save);
313   delFiltOvSv(tx.filt.ovsv);
314   delFIR_Bandpass_COMPLEX(tx.filt.coef);
315   delCorrectIQ(tx.iqfix);
316   delCXB(tx.buf.o);
317   delCXB(tx.buf.i);
318
319   /* RX */
320   for (k = 0; k < uni.multirx.nrx; k++) {
321     delSpotToneGen(rx[k].spot.gen);
322     delDigitalAgc(rx[k].agc.gen);
323     del_nb(rx[k].nb_sdrom.gen);
324     del_nb(rx[k].nb.gen);
325     del_lmsr(rx[k].anf.gen);
326     del_lmsr(rx[k].anr.gen);
327     delAMD(rx[k].am.gen);
328     delFMD(rx[k].fm.gen);
329     delOSC(rx[k].osc.gen);
330     delvec_COMPLEX(rx[k].filt.save);
331     delFiltOvSv(rx[k].filt.ovsv);
332     delFIR_Bandpass_COMPLEX(rx[k].filt.coef);
333     delCorrectIQ(rx[k].iqfix);
334     delCXB(rx[k].buf.o);
335     delCXB(rx[k].buf.i);
336   }
337   
338   /* all */
339   if (uni.meter.flag)
340     closeChan(uni.meter.chan.c);
341 }
342
343 //////////////////////////////////////////////////////////////////////////
344 // execution
345 //////////////////////////////////////////////////////////////////////////
346
347 //========================================================================
348 // util
349
350 PRIVATE REAL
351 CXBnorm(CXB buff) {
352   int i;
353   double sum = 0.0;
354   for (i = 0; i < CXBhave(buff); i++)
355     sum += Csqrmag(CXBdata(buff, i));
356   return sqrt(sum);
357 }
358
359 //========================================================================
360 /* all */
361
362 /* tap off S-meter from some buf */
363
364 PRIVATE void
365 do_meter(COMPLEX *vec, int len) {
366   int i;
367   
368   uni.meter.val = 0;
369
370   switch (uni.meter.type) {
371   case AVG_SIGNAL_STRENGTH:
372     for (i = 0; i < len; i++)
373       uni.meter.val += Csqrmag(vec[i]);
374     uni.meter.val =
375       uni.meter.avgval = 0.9 * uni.meter.avgval + log10(uni.meter.val + 1e-20);
376     break;
377   case SIGNAL_STRENGTH:
378     for (i = 0; i < len; i++)
379       uni.meter.val += Csqrmag(vec[i]);
380     uni.meter.avgval = uni.meter.val = 10.0 * log10(uni.meter.val + 1e-20);
381     break;
382   case ADC_REAL:
383     for(i = 0; i < len; i++)
384       uni.meter.val = max(fabs(vec[i].re), uni.meter.val);
385     uni.meter.val = 20.0 * log10(uni.meter.val + 1e-10);
386     break;
387   case ADC_IMAG:
388     for(i = 0; i < len; i++)
389       uni.meter.val = max(fabs(vec[i].im), uni.meter.val);
390     uni.meter.val = 20.0 * log10(uni.meter.val + 1e-10);
391     break;
392   default:
393     break;
394   }
395
396   putChan_nowait(uni.meter.chan.c,
397                  (char *) &uni.meter.val,
398                  sizeof(uni.meter.val));
399 }
400
401 //========================================================================
402 /* RX processing */ 
403
404 PRIVATE BOOLEAN
405 should_do_rx_squelch(int k) {
406   if (rx[k].squelch.flag) {
407     int i, n = CXBhave(rx[k].buf.o);
408     rx[k].squelch.power = 0.0;
409     for (i = 0; i < n; i++)
410       rx[k].squelch.power += Csqrmag(CXBdata(rx[k].buf.o, i));
411     return rx[k].squelch.thresh > 10.0 * log10(rx[k].squelch.power);
412   } else
413     return rx[k].squelch.set = FALSE;
414 }
415
416 // apply squelch
417 // slew into silence first time
418
419 PRIVATE void
420 do_squelch(int k) {
421   rx[k].squelch.set = TRUE;
422   if (!rx[k].squelch.running) {
423     int i, m = rx[k].squelch.num, n = CXBhave(rx[k].buf.o) - m;
424     for (i = 0; i < m; i++)
425       CXBdata(rx[k].buf.o, i) = Cscl(CXBdata(rx[k].buf.o, i), 1.0 - (REAL) i / m);
426     memset((void *) (CXBbase(rx[k].buf.o) + m), 0, n * sizeof(COMPLEX));
427     rx[k].squelch.running = TRUE;
428   } else
429     memset((void *) CXBbase(rx[k].buf.o), 0, CXBhave(rx[k].buf.o) * sizeof(COMPLEX));
430 }
431
432 // lift squelch
433 // slew out from silence to full scale
434
435 PRIVATE void
436 no_squelch(int k) {
437   if (rx[k].squelch.running) {
438     int i, m = rx[k].squelch.num;
439     for (i = 0; i < m; i++)
440       CXBdata(rx[k].buf.o, i) = Cscl(CXBdata(rx[k].buf.o, i), (REAL) i / m);
441     rx[k].squelch.running = FALSE;
442   }
443 }
444
445 /* pre-condition for (nearly) all RX modes */
446
447 PRIVATE void
448 do_rx_pre(int k) {
449   int i, n = min(CXBhave(rx[k].buf.i), uni.buflen);
450
451   //
452   // do shrinkage here?
453   //
454
455   if (rx[k].scl.pre.flag)
456     for (i = 0; i < n; i++)
457       CXBdata(rx[k].buf.i, i) = Cscl(CXBdata(rx[k].buf.i, i),
458                                      rx[k].scl.pre.val); 
459
460   if (rx[k].nb.flag) noiseblanker(rx[k].nb.gen);
461   if (rx[k].nb_sdrom.flag) SDROMnoiseblanker(rx[k].nb_sdrom.gen);
462
463   correctIQ(rx[k].buf.i, rx[k].iqfix);
464
465   /* 2nd IF conversion happens here */
466
467   if (rx[k].osc.gen->Frequency != 0.0) {
468     ComplexOSC(rx[k].osc.gen);
469     for (i = 0; i < n; i++)
470       CXBdata(rx[k].buf.i, i) = Cmul(CXBdata(rx[k].buf.i, i),
471                                      OSCCdata(rx[k].osc.gen, i));
472   } 
473
474   /* filtering, metering, squelch, & AGC */
475
476   if (rx[k].mode != SPEC) {
477
478     if (rx[k].tick == 0)
479       reset_OvSv(rx[k].filt.ovsv);
480     
481     filter_OvSv(rx[k].filt.ovsv);
482     CXBhave(rx[k].buf.o) = CXBhave(rx[k].buf.i);
483
484     if (uni.meter.flag)
485       do_meter(CXBbase(rx[k].buf.o), uni.buflen);
486
487     if (should_do_rx_squelch(k))
488       do_squelch(k);
489
490     else if (rx[k].agc.flag)
491       DigitalAgc(rx[k].agc.gen, rx[k].tick);
492
493   } else if (uni.meter.flag)
494     do_meter(CXBbase(rx[k].buf.o), uni.buflen);
495 }
496
497 PRIVATE void
498 do_rx_post(int k) {
499   int i, n = CXBhave(rx[k].buf.o);
500
501   if (!rx[k].squelch.set)  {
502     no_squelch(k);
503     // spotting tone
504     if (rx[k].spot.flag) {
505       // remember whether it's turned itself off during this pass
506       rx[k].spot.flag = SpotTone(rx[k].spot.gen);
507       for (i = 0; i < n; i++)
508         CXBdata(rx[k].buf.o, i) = Cadd(CXBdata(rx[k].buf.o, i),
509                                        CXBdata(rx[k].spot.gen->buf, i));
510     }
511   }
512
513   // final scaling
514
515   if (rx[k].scl.post.flag)
516     for (i = 0; i < n; i++)
517       CXBdata(rx[k].buf.o, i) = Cscl(CXBdata(rx[k].buf.o, i),
518                                      rx[k].scl.post.val);
519
520   // not binaural?
521   // position in stereo field
522
523   if (!rx[k].bin.flag)
524     for (i = 0; i < n; i++)
525       CXBdata(rx[k].buf.o, i) = Cscl(rx[k].azim, CXBreal(rx[k].buf.o, i));
526 }
527
528 /* demod processing */
529
530 PRIVATE void
531 do_rx_SBCW(int k) {
532   if (rx[k].anr.flag) lmsr_adapt(rx[k].anr.gen);
533   if (rx[k].anf.flag) lmsr_adapt(rx[k].anf.gen);
534 }
535
536 PRIVATE void
537 do_rx_AM(int k) { AMDemod(rx[k].am.gen); }
538
539 PRIVATE void
540 do_rx_FM(int k) { FMDemod(rx[k].fm.gen); }
541
542 PRIVATE void
543 do_rx_DRM(int k) {}
544
545 PRIVATE void
546 do_rx_SPEC(int k) {
547   memcpy(CXBbase(rx[k].buf.o),
548          CXBbase(rx[k].buf.i),
549          sizeof(COMPLEX) * CXBhave(rx[k].buf.i));
550   if (rx[k].agc.flag) DigitalAgc(rx[k].agc.gen, rx[k].tick);
551 }
552
553 PRIVATE void
554 do_rx_NIL(int k) {
555   int i, n = min(CXBhave(rx[k].buf.i), uni.buflen);
556   for (i = 0; i < n; i++) CXBdata(rx[k].buf.o, i) = cxzero;
557 }
558
559 /* overall dispatch for RX processing */
560
561 PRIVATE void
562 do_rx(int k) {
563   do_rx_pre(k);
564   switch (rx[k].mode) {
565   case USB:
566   case LSB:
567   case CWU:
568   case CWL:
569   case DSB:  do_rx_SBCW(k); break;
570   case AM:
571   case SAM:  do_rx_AM(k); break;
572   case FMN:  do_rx_FM(k);   break;
573   case DRM:  do_rx_DRM(k);  break;
574   case SPEC:
575     default: do_rx_SPEC(k); break;
576   }
577   do_rx_post(k);
578 }  
579
580 //==============================================================
581 /* TX processing */
582
583 /* pre-condition for (nearly) all TX modes */
584
585 PRIVATE void
586 do_tx_pre(void) {
587
588   if (tx.scl.pre.flag) {
589     int i, n = CXBhave(tx.buf.i);
590     for (i = 0; i < n; i++)
591       CXBdata(tx.buf.i, i) = Cmplx(CXBreal(tx.buf.i, i) * tx.scl.pre.val, 0.0);
592   }
593
594   correctIQ(tx.buf.i, tx.iqfix);
595
596   if (tx.spr.flag) SpeechProcessor(tx.spr.gen);
597
598   if (tx.tick == 0) reset_OvSv(tx.filt.ovsv);
599   filter_OvSv(tx.filt.ovsv);
600 }
601
602 PRIVATE void
603 do_tx_post(void) {
604   CXBhave(tx.buf.o) = CXBhave(tx.buf.i);
605
606   if (tx.agc.flag) DigitalAgc(tx.agc.gen, tx.tick);
607   if (tx.scl.post.flag) {
608     int i, n = CXBhave(tx.buf.o);
609     for (i = 0; i < n; i++)
610       CXBdata(tx.buf.o, i) = Cscl(CXBdata(tx.buf.o, i), tx.scl.post.val);
611   }
612   if (uni.meter.flag) do_meter(CXBbase(tx.buf.o), uni.buflen);
613   if (tx.osc.gen->Frequency != 0.0) {
614     int i;
615     ComplexOSC(tx.osc.gen);
616     for (i = 0; i < CXBhave(tx.buf.o); i++)
617       CXBdata(tx.buf.o, i) = Cmul(CXBdata(tx.buf.o, i), OSCCdata(tx.osc.gen, i));
618   }
619 }
620
621 /* modulator processing */
622
623 PRIVATE void
624 do_tx_SBCW(void) {
625   int i, n = min(CXBhave(tx.buf.o), uni.buflen); 
626
627   if ((tx.norm = CXBnorm(tx.buf.o)) > 0.0)
628     for (i = 0; i < n; i++) {
629       tx.scl.dc = Cadd(Cscl(tx.scl.dc, 0.99),
630                        Cscl(CXBdata(tx.buf.o, i), -0.01));
631       CXBdata(tx.buf.o, i) = Cadd(CXBdata(tx.buf.o, i), tx.scl.dc);
632     }
633 }
634
635 PRIVATE void
636 do_tx_AM(void) {
637   int i, n = min(CXBhave(tx.buf.o), uni.buflen); 
638
639   if ((tx.norm = CXBnorm(tx.buf.o)) > 0.0)
640     for (i = 0; i < n; i++) { 
641       tx.scl.dc = Cadd(Cscl(tx.scl.dc, 0.999),
642                        Cscl(CXBdata(tx.buf.o, i), -0.001));
643       CXBreal(tx.buf.o, i) =
644         0.49995 + 0.49995 * (CXBreal(tx.buf.o, i) - tx.scl.dc.re);
645       CXBimag(tx.buf.o, i) = 0.0;
646     }
647 }
648
649 PRIVATE void
650 do_tx_FM(void) {
651   int i, n = min(CXBhave(tx.buf.o), uni.buflen);
652   if ((tx.norm = CXBnorm(tx.buf.o)) > 0.0)
653     for (i = 0; i < n; i++) {
654       tx.scl.dc = Cadd(Cscl(tx.scl.dc, 0.999),
655                        Cscl(CXBdata(tx.buf.o, i), 0.001));
656       tx.osc.phase += (CXBreal(tx.buf.o, i) - tx.scl.dc.re) * CvtMod2Freq;
657       if (tx.osc.phase >= TWOPI) tx.osc.phase -= TWOPI;
658       if (tx.osc.phase < 0.0) tx.osc.phase += TWOPI;
659       CXBdata(tx.buf.o, i) =
660         Cscl(Cmplx(cos(tx.osc.phase), sin(tx.osc.phase)), 0.99999);
661     }
662 }
663
664 PRIVATE void
665 do_tx_NIL(void) {
666   int i, n = min(CXBhave(tx.buf.i), uni.buflen);
667   for (i = 0; i < n; i++) CXBdata(tx.buf.o, i) = cxzero;
668 }
669
670 /* general TX processing dispatch */
671
672 PRIVATE void
673 do_tx(void) {
674   do_tx_pre();
675   switch (tx.mode) {
676   case USB:
677   case LSB:
678   case CWU:
679   case CWL:
680   case DSB:  do_tx_SBCW(); break;
681   case AM:
682   case SAM:  do_tx_AM();   break;
683   case FMN:  do_tx_FM();   break;
684   case DRM:
685   case SPEC:
686     default: do_tx_NIL(); break;
687   }
688   do_tx_post();
689 }
690
691 //========================================================================
692 /* overall buffer processing;
693    come here when there are buffers to work on */
694
695 void
696 process_samples(float *bufl, float *bufr,
697                 float *auxl, float *auxr,
698                 int n) {
699   int i, k;
700
701   switch (uni.mode.trx) {
702
703   case RX:
704
705     // make copies of the input for all receivers
706     for (k = 0; k < uni.multirx.nrx; k++)
707       if (uni.multirx.act[k]) {
708         for (i = 0; i < n; i++)
709           CXBimag(rx[k].buf.i, i) = bufl[i], CXBreal(rx[k].buf.i, i) = bufr[i];
710         CXBhave(rx[k].buf.i) = n;
711       }
712
713     // prepare buffers for mixing
714     memset((char *) bufl, 0, n * sizeof(float));
715     memset((char *) bufr, 0, n * sizeof(float));
716
717     // run all receivers
718     for (k = 0; k < uni.multirx.nrx; k++)
719       if (uni.multirx.act[k]) {
720         do_rx(k), rx[k].tick++;
721         // mix
722         for (i = 0; i < n; i++)
723           bufl[i] += CXBimag(rx[k].buf.o, i),
724           bufr[i] += CXBreal(rx[k].buf.o, i);
725         CXBhave(rx[k].buf.o) = n;
726       }
727
728     if (uni.mix.rx.flag)
729       for (i = 0; i < n; i++)
730         bufl[i] += auxl[i] * uni.mix.rx.gain,
731         bufr[i] += auxr[i] * uni.mix.rx.gain;
732
733     break;
734
735   case TX:
736
737     if (uni.mix.tx.flag)
738       for (i = 0; i < n; i++)
739         bufl[i] += auxl[i] * uni.mix.tx.gain,
740         bufr[i] += auxr[i] * uni.mix.tx.gain;
741
742     for (i = 0; i < n; i++)
743       CXBimag(tx.buf.i, i) = bufl[i], CXBreal(tx.buf.i, i) = bufr[i];
744     CXBhave(tx.buf.i) = n;
745
746     do_tx(), tx.tick++;
747
748     for (i = 0; i < n; i++)
749       bufl[i] = (float) CXBimag(tx.buf.o, i), bufr[i] = (float) CXBreal(tx.buf.o, i);
750     CXBhave(tx.buf.o) = n;
751
752     break;
753   }
754
755   uni.tick++;
756 }