sjero.net Git - linphone/blob - coreapi/ec-calibrator.c

   1 /*
   2 linphone
   3 Copyright (C) 2011 Belledonne Communications SARL
   4 Author: Simon MORLAT (simon.morlat@linphone.org)
   5
   6 This program is free software; you can redistribute it and/or
   7 modify it under the terms of the GNU General Public License
   8 as published by the Free Software Foundation; either version 2
   9 of the License, or (at your option) any later version.
  10
  11 This program is distributed in the hope that it will be useful,
  12 but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 GNU General Public License for more details.
  15
  16 You should have received a copy of the GNU General Public License
  17 along with this program; if not, write to the Free Software
  18 Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  19 */
  20
  21 #include "private.h"
  22
  23 #include "mediastreamer2/mstonedetector.h"
  24 #include "mediastreamer2/dtmfgen.h"
  25
  26 #include "lpconfig.h"
  27
  28
  29
  30 static void ecc_init_filters(EcCalibrator *ecc){
  31         unsigned int rate;
  32         MSTickerParams params={0};
  33         params.name="Echo calibrator";
  34         params.prio=MS_TICKER_PRIO_HIGH;
  35         ecc->ticker=ms_ticker_new_with_params(&params);
  36
  37         ecc->sndread=ms_snd_card_create_reader(ecc->play_card);
  38         ms_filter_call_method(ecc->sndread,MS_FILTER_SET_SAMPLE_RATE,&ecc->rate);
  39         ecc->det=ms_filter_new(MS_TONE_DETECTOR_ID);
  40         ms_filter_call_method(ecc->det,MS_FILTER_SET_SAMPLE_RATE,&ecc->rate);
  41         ecc->rec=ms_filter_new(MS_FILE_REC_ID);
  42
  43         ms_filter_link(ecc->sndread,0,ecc->det,0);
  44         ms_filter_link(ecc->det,0,ecc->rec,0);
  45
  46         ecc->play=ms_filter_new(MS_FILE_PLAYER_ID);
  47         ecc->gen=ms_filter_new(MS_DTMF_GEN_ID);
  48         ms_filter_call_method(ecc->gen,MS_FILTER_SET_SAMPLE_RATE,&ecc->rate);
  49         ecc->resampler=ms_filter_new(MS_RESAMPLE_ID);
  50         ecc->sndwrite=ms_snd_card_create_writer(ecc->capt_card);
  51
  52         ms_filter_link(ecc->play,0,ecc->gen,0);
  53         ms_filter_link(ecc->gen,0,ecc->resampler,0);
  54         ms_filter_link(ecc->resampler,0,ecc->sndwrite,0);
  55
  56         ms_filter_call_method(ecc->sndwrite,MS_FILTER_SET_SAMPLE_RATE,&ecc->rate);
  57         ms_filter_call_method(ecc->sndwrite,MS_FILTER_GET_SAMPLE_RATE,&rate);
  58         ms_filter_call_method(ecc->resampler,MS_FILTER_SET_SAMPLE_RATE,&ecc->rate);
  59         ms_filter_call_method(ecc->resampler,MS_FILTER_SET_OUTPUT_SAMPLE_RATE,&rate);
  60
  61         ms_ticker_attach(ecc->ticker,ecc->play);
  62         ms_ticker_attach(ecc->ticker,ecc->sndread);
  63 }
  64
  65 static void ecc_deinit_filters(EcCalibrator *ecc){
  66         ms_ticker_detach(ecc->ticker,ecc->play);
  67         ms_ticker_detach(ecc->ticker,ecc->sndread);
  68
  69         ms_filter_unlink(ecc->play,0,ecc->gen,0);
  70         ms_filter_unlink(ecc->gen,0,ecc->resampler,0);
  71         ms_filter_unlink(ecc->resampler,0,ecc->sndwrite,0);
  72
  73         ms_filter_unlink(ecc->sndread,0,ecc->det,0);
  74         ms_filter_unlink(ecc->det,0,ecc->rec,0);
  75
  76         ms_filter_destroy(ecc->sndread);
  77         ms_filter_destroy(ecc->det);
  78         ms_filter_destroy(ecc->rec);
  79         ms_filter_destroy(ecc->play);
  80         ms_filter_destroy(ecc->gen);
  81         ms_filter_destroy(ecc->resampler);
  82         ms_filter_destroy(ecc->sndwrite);
  83
  84         ms_ticker_destroy(ecc->ticker);
  85 }
  86
  87 static void on_tone_sent(void *data, MSFilter *f, unsigned int event_id, void *arg){
  88         MSDtmfGenEvent *ev=(MSDtmfGenEvent*)arg;
  89         EcCalibrator *ecc=(EcCalibrator*)data;
  90         ecc->sent_count++;
  91         ecc->acc-=ev->tone_start_time;
  92         ms_message("Sent tone at %u",(unsigned int)ev->tone_start_time);
  93 }
  94
  95 static void on_tone_received(void *data, MSFilter *f, unsigned int event_id, void *arg){
  96         MSToneDetectorEvent *ev=(MSToneDetectorEvent*)arg;
  97         EcCalibrator *ecc=(EcCalibrator*)data;
  98         ecc->recv_count++;
  99         ecc->acc+=ev->tone_start_time;
 100         ms_message("Received tone at %u",(unsigned int)ev->tone_start_time);
 101 }
 102
 103 static void ecc_play_tones(EcCalibrator *ecc){
 104         MSDtmfGenCustomTone tone;
 105         MSToneDetectorDef expected_tone;
 106
 107
 108         ms_filter_set_notify_callback(ecc->det,on_tone_received,ecc);
 109
 110         expected_tone.frequency=2000;
 111         expected_tone.min_duration=40;
 112         expected_tone.min_amplitude=0.02;
 113
 114         ms_filter_call_method (ecc->det,MS_TONE_DETECTOR_ADD_SCAN,&expected_tone);
 115
 116         tone.frequency=1000;
 117         tone.duration=1000;
 118         tone.amplitude=1.0;
 119
 120         /*play an initial tone to startup the audio playback/capture*/
 121         ms_filter_call_method(ecc->gen,MS_DTMF_GEN_PLAY_CUSTOM,&tone);
 122         ms_sleep(2);
 123
 124         ms_filter_set_notify_callback(ecc->gen,on_tone_sent,ecc);
 125         tone.frequency=2000;
 126         tone.duration=100;
 127
 128         ms_filter_call_method(ecc->gen,MS_DTMF_GEN_PLAY_CUSTOM,&tone);
 129         ms_sleep(1);
 130         ms_filter_call_method(ecc->gen,MS_DTMF_GEN_PLAY_CUSTOM,&tone);
 131         ms_sleep(1);
 132         ms_filter_call_method(ecc->gen,MS_DTMF_GEN_PLAY_CUSTOM,&tone);
 133         ms_sleep(1);
 134
 135         if (ecc->sent_count==3 && ecc->recv_count==3){
 136                 int delay=ecc->acc/3;
 137                 if (delay<0){
 138                         ms_error("Quite surprising calibration result, delay=%i",delay);
 139                         ecc->status=LinphoneEcCalibratorFailed;
 140                 }else{ms_message("Echo calibration estimated delay to be %i ms",delay);
 141                         ecc->delay=delay;
 142                         ecc->status=LinphoneEcCalibratorDone;
 143                 }
 144         }else{
 145                 ms_error("Echo calibration failed, tones received = %i",ecc->recv_count);
 146                 ecc->status=LinphoneEcCalibratorFailed;
 147         }
 148
 149 }
 150
 151 static void  * ecc_thread(void *p){
 152         EcCalibrator *ecc=(EcCalibrator*)p;
 153
 154         ecc_init_filters(ecc);
 155         ecc_play_tones(ecc);
 156         ecc_deinit_filters(ecc);
 157         ms_thread_exit(NULL);
 158         return NULL;
 159 }
 160
 161 EcCalibrator * ec_calibrator_new(MSSndCard *play_card, MSSndCard *capt_card, unsigned int rate, LinphoneEcCalibrationCallback cb, void *cb_data ){
 162         EcCalibrator *ecc=ms_new0(EcCalibrator,1);
 163
 164         ecc->rate=rate;
 165         ecc->cb=cb;
 166         ecc->cb_data=cb_data;
 167         ecc->capt_card=capt_card;
 168         ecc->play_card=play_card;
 169         ms_thread_create(&ecc->thread,NULL,ecc_thread,ecc);
 170         return ecc;
 171 }
 172
 173 LinphoneEcCalibratorStatus ec_calibrator_get_status(EcCalibrator *ecc){
 174         return ecc->status;
 175 }
 176
 177 void ec_calibrator_destroy(EcCalibrator *ecc){
 178         ms_thread_join(ecc->thread,NULL);
 179         ms_free(ecc);
 180 }
 181
 182 int linphone_core_start_echo_calibration(LinphoneCore *lc, LinphoneEcCalibrationCallback cb, void *cb_data){
 183         if (lc->ecc!=NULL){
 184                 ms_error("Echo calibration is still on going !");
 185                 return -1;
 186         }
 187         unsigned int rate = lp_config_get_int(lc->config,"sound","echo_cancellation_rate",8000);
 188         lc->ecc=ec_calibrator_new(lc->sound_conf.play_sndcard,lc->sound_conf.capt_sndcard,rate,cb,cb_data);
 189         return 0;
 190 }
 191