cinelerra-5.1/cinelerra/devicedvbinput.C

   1 // jam job dvb file testing
   2 //  1) define TEST_DATA as mpeg ts file "name.ts"
   3 //  2) mod record.C Record::resync(), comment out: record_channel->drain_audio()
   4 //  3) enable audio delay in audioidevice.C  AudioDevice::run_input, #if 0 <-> #if 1
   5 //#define TEST_DATA "/tmp/xx.ts"
   6
   7 /*
   8  * CINELERRA
   9  * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
  10  *
  11  * This program is free software; you can redistribute it and/or modify
  12  * it under the terms of the GNU General Public License as published by
  13  * the Free Software Foundation; either version 2 of the License, or
  14  * (at your option) any later version.
  15  *
  16  * This program is distributed in the hope that it will be useful,
  17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19  * GNU General Public License for more details.
  20  *
  21  * You should have received a copy of the GNU General Public License
  22  * along with this program; if not, write to the Free Software
  23  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  24  *
  25  */
  26
  27 #ifdef HAVE_DVB
  28 #include "chantables.h"
  29 #include "devicedvbinput.h"
  30 #include "devicempeginput.h"
  31 #include "recordconfig.h"
  32 #include "signalstatus.h"
  33
  34 #include <stdio.h>
  35 #include <stdlib.h>
  36 #include <stdint.h>
  37 #include <unistd.h>
  38 #include <string.h>
  39 #include <errno.h>
  40 #include <fcntl.h>
  41 #include <linux/dvb/dmx.h>
  42 #include <linux/dvb/frontend.h>
  43 #include <sys/ioctl.h>
  44
  45
  46 DVBInputStatus::DVBInputStatus(DeviceDVBInput *dvb_input)
  47  : Thread(1, 0, 0)
  48 {
  49         this->dvb_input = dvb_input;
  50         signal_status = 0;
  51         done = 1;
  52 }
  53
  54 DVBInputStatus::~DVBInputStatus()
  55 {
  56         stop();
  57         if( signal_status ) signal_status->disconnect();
  58 }
  59
  60 void DVBInputStatus::stop()
  61 {
  62         if( done ) return;
  63         done = 1;
  64         Thread::cancel();
  65         Thread::join();
  66 }
  67
  68 void DVBInputStatus::start()
  69 {
  70         if( !done ) return;
  71         done = 0;
  72         Thread::start();
  73 }
  74
  75 void DVBInputStatus::lock_dvb()
  76 {
  77         Thread::disable_cancel();
  78         dvb_input->dvb_lock->lock("DVBInputStatus::lock_dvb");
  79 }
  80
  81 void DVBInputStatus::unlock_dvb()
  82 {
  83         dvb_input->dvb_lock->unlock();
  84         Thread::enable_cancel();
  85 }
  86
  87 void DVBInputStatus::run()
  88 {
  89         lock_dvb();
  90
  91         while( !done ) {
  92                 dvb_input->status_dev();
  93                 update();
  94                 unlock_dvb();
  95                 usleep(250000);
  96                 lock_dvb();
  97         }
  98
  99         unlock_dvb();
 100 }
 101
 102 void DVBInputStatus::update()
 103 {
 104         if( !signal_status ) return;
 105         signal_status->lock_window("DVBInputStatus::update");
 106         signal_status->update();
 107         signal_status->unlock_window();
 108 }
 109
 110
 111 DeviceDVBInput::DeviceDVBInput(const char *name, int no)
 112  : DeviceMPEGInput(name, no)
 113 {
 114         frontend_fd = -1;
 115         demux_fd = -1;
 116         dvb_fd = -1;
 117         reset_signal();
 118         dvb_locked = 0;
 119         dvb_lock = new Mutex("DeviceDVBInput::dvb_lock", 0);
 120         sprintf(frontend_path, "%s/frontend%d", name, no);
 121         sprintf(demux_path, "%s/demux%d", name, no);
 122         sprintf(dvb_path, "%s/dvr%d", name, no);
 123         dvb_input_status = new DVBInputStatus(this);
 124 }
 125
 126 DeviceMPEGInput *DeviceDVBInput::
 127 NewDVBInput(const char *name, int no)
 128 {
 129         return (DeviceMPEGInput *) new DeviceDVBInput(name, no);
 130 }
 131
 132 DeviceDVBInput::~DeviceDVBInput()
 133 {
 134         close_dev();
 135         delete dvb_input_status;
 136         delete dvb_lock;
 137         dvb_close(1);
 138 }
 139
 140
 141 DeviceMPEGInput* DeviceDVBInput::get_mpeg_input(VideoDevice *device)
 142 {
 143         DeviceMPEGInput* mpeg_device = get_mpeg_video(device, NewDVBInput,
 144                 device->in_config->dvb_in_adapter, device->in_config->dvb_in_device);
 145         if( !mpeg_device ) return 0;
 146         device->channel->has_subchan = 1;
 147         device->channel->has_scanning = 1;
 148         device->channel->use_frequency = 1;
 149         mpeg_device->set_device_number(0);
 150         Channel *input_src = device->new_input_source((char*)"input0");
 151         input_src->device_index = 0;
 152         input_src->tuner = 0;
 153         return mpeg_device;
 154 }
 155
 156 DeviceMPEGInput* DeviceDVBInput::get_mpeg_input(AudioDevice *device)
 157 {
 158         return get_mpeg_audio(device, NewDVBInput,
 159                 device->in_config->dvb_in_adapter, device->in_config->dvb_in_device);
 160 }
 161
 162
 163 int DeviceDVBInput::wait_signal(int ms, int trys)
 164 {
 165         Timer timer;
 166         for( int retry=trys; --retry>=0; ) {
 167                 timer.update();
 168                 if( !dvb_status() && has_signal() ) break;
 169                 int dt = ms - timer.get_difference();
 170                 if( dt > 0 ) Timer::delay(dt);
 171         }
 172         return has_signal() ? 0 : 1;
 173 }
 174
 175
 176 int DeviceDVBBuffer::read(int retries, int usec, int bsz)
 177 {
 178         while( --retries >= 0 ) {
 179                 struct timeval tv;  tv.tv_sec = 0;  tv.tv_usec = usec;
 180                 fd_set rd_fd;  FD_ZERO(&rd_fd);  FD_SET(fd, &rd_fd);
 181                 fd_set er_fd;  FD_ZERO(&er_fd);  FD_SET(fd, &er_fd);
 182                 int ret = select(fd+1, &rd_fd, 0, &er_fd, &tv);
 183                 if( ret < 0 ) break;
 184                 if( !ret ) continue;
 185                 if( !FD_ISSET(fd, &er_fd) ) {
 186                         select(0, 0, 0, 0, &tv);
 187                         continue;
 188                 }
 189                 if( !FD_ISSET(fd, &rd_fd) ) continue;
 190                 if( (ret=::read(fd, bfr, bsz)) > 0 )
 191                         return sz = ret;
 192         }
 193         return -1;
 194 }
 195
 196 int DeviceDVBInput::dvb_sync()
 197 {
 198         const int SYNC = 0x47, TS_BLKSZ = 188;
 199         const int BSZ = 100*TS_BLKSZ, XSZ = BSZ*5;
 200         DeviceDVBBuffer data(dvb_fd, BSZ);
 201
 202         int nsync = 0, xfr = 0;
 203         while( xfr < XSZ ) {
 204                 if( data.read(5, 200000, BSZ) < 0 ) break;
 205                 int ofs = 0;
 206                 while( ofs < data.size() ) {
 207                         nsync = 0;
 208                         int i = ofs;
 209                         while( i<data.size() && data[i]!=SYNC ) ++i;
 210                         if( i >= data.size() ) break;
 211                         ofs = i;
 212                         while( (ofs+=TS_BLKSZ)<data.size() && data[ofs]==SYNC ) {
 213                                 ++nsync;  i = ofs;
 214                         }
 215                 }
 216                 if( nsync >= 32 && (ofs-=data.size()) >= 0 ) {
 217                         while( ofs > 0 ) {
 218                                 if( data.read(5, 200000, ofs) < 0 ) return 1;
 219                                 ofs -= data.size();
 220                         }
 221                         return 0;
 222                 }
 223                 xfr += data.size();
 224         }
 225         return 1;
 226 }
 227
 228 int DeviceDVBInput::dvb_open()
 229 {
 230         int ret = 0;
 231 #ifndef TEST_DATA
 232 #ifdef HAVE_DVB
 233         if( frontend_fd < 0 &&
 234            (frontend_fd = ::open(frontend_path, O_RDWR)) < 0 ) {
 235                 fprintf(stderr, "DeviceDVBInput::dvb_open %s: %s\n",
 236                         frontend_path, strerror(errno));
 237                 ret = 1;
 238         }
 239         if( !ret && ioctl(frontend_fd, FE_GET_INFO, &fe_info) < 0 ) {
 240                 fprintf(stderr,
 241                         "DeviceDVBInput::dvb_open FE_GET_INFO: %s\n",
 242                         strerror(errno));
 243                 ret = 1;
 244         }
 245
 246         pwr_min = snr_min = 0;
 247         pwr_max = snr_max = 65535;
 248
 249 // Set frequency
 250         int index = -1, table = -1;
 251         if( !ret && (index = get_channel()) < 0 ) ret = 1;
 252         if( !ret && (table = get_channel_table()) < 0 ) ret = 1;
 253         if( !ret ) {
 254                 uint32_t frequency = 0;
 255                 struct dvb_frontend_parameters frontend_param;
 256                 bzero(&frontend_param, sizeof(frontend_param));
 257                 ret = 1;
 258                 switch(table) {
 259                 case NETTUNE_AIR:
 260                         if( index >= chanlists[NTSC_DVB].count ) break;
 261                         frequency = chanlists[NTSC_DVB].list[index].freq * 1000000;
 262                         if( frequency < fe_info.frequency_min ||
 263                                 frequency > fe_info.frequency_max ) break;
 264                         frontend_param.frequency = frequency;
 265                         frontend_param.u.vsb.modulation = VSB_8;
 266                         ret = 0;
 267                         break;
 268                 case NETTUNE_CABLE:
 269                         if( index >= chanlists[CATV_DVB].count ) break;
 270                         frequency = chanlists[CATV_DVB].list[index].freq * 1000000;
 271                         if( frequency < fe_info.frequency_min ||
 272                                 frequency > fe_info.frequency_max ) break;
 273                         frontend_param.frequency = frequency;
 274                         frontend_param.u.vsb.modulation = QAM_AUTO;
 275                         ret = 0;
 276                         break;
 277                 }
 278
 279                 if( !ret && ioctl(frontend_fd, FE_SET_FRONTEND, &frontend_param) < 0 ) {
 280                         fprintf(stderr,
 281                                 "DeviceDVBInput::dvb_open FE_SET_FRONTEND frequency=%d: %s\n",
 282                                 frontend_param.frequency, strerror(errno));
 283                         ret = 1;
 284                 }
 285
 286                 if( !ret && wait_signal(333,3) ) {
 287                         fprintf(stderr,
 288                                  "DeviceDVBInput::dvb_open: no signal, index=%d frequency=%d\n",
 289                                 index, frontend_param.frequency);
 290                         ret = 1;
 291                 }
 292
 293                 if( !ret && ioctl(frontend_fd, FE_GET_FRONTEND, &frontend_param) ) {
 294                         fprintf(stderr,
 295                                 "DeviceDVBInput::dvb_open FE_GET_FRONTEND: %s\n",
 296                                 strerror(errno));
 297                         ret = 1;
 298                 }
 299                 if( !ret ) { // goofy quirks
 300                         if( !strcmp("Samsung S5H1409 QAM/8VSB Frontend", fe_info.name) ) {
 301                                 switch(frontend_param.u.vsb.modulation) {
 302                                 case QAM_64:  snr_min = 200;  snr_max = 300;  break;
 303                                 case QAM_256: snr_min = 260;  snr_max = 400;  break;
 304                                 case VSB_8:   snr_min = 111;  snr_max = 300;  break;
 305                                 default: break;
 306                                 }
 307                                 pwr_min = snr_min;  pwr_max = snr_max;
 308                         }
 309                         else if( !strcmp("Auvitek AU8522 QAM/8VSB Frontend", fe_info.name) ) {
 310                                 switch(frontend_param.u.vsb.modulation) {
 311                                 case QAM_64:  snr_min = 190;  snr_max = 290;  break;
 312                                 case QAM_256: snr_min = 280;  snr_max = 400;  break;
 313                                 case VSB_8:   snr_min = 115;  snr_max = 270;  break;
 314                                 default: break;
 315                                 }
 316                                 pwr_min = snr_min;  pwr_max = snr_max;
 317                         }
 318                 }
 319         }
 320
 321         if( !ret && (demux_fd = ::open(demux_path, O_RDWR)) < 0 ) {
 322                 fprintf(stderr,
 323                         "DeviceDVBInput::dvb_open %s for video: %s\n",
 324                         demux_path,
 325                         strerror(errno));
 326                 ret = 1;
 327         }
 328
 329 // Setting exactly one PES filter to 0x2000 dumps the entire
 330 // transport stream.
 331         if( !ret ) {
 332                 struct dmx_pes_filter_params pesfilter;
 333                 memset(&pesfilter,0,sizeof(pesfilter));
 334                 pesfilter.pid = 0x2000;
 335                 pesfilter.input = DMX_IN_FRONTEND;
 336                 pesfilter.output = DMX_OUT_TS_TAP;
 337                 pesfilter.pes_type = DMX_PES_OTHER;
 338                 pesfilter.flags = DMX_IMMEDIATE_START;
 339                 if( ioctl(demux_fd, DMX_SET_PES_FILTER, &pesfilter) < 0 ) {
 340                         fprintf(stderr,
 341                 "DeviceDVBInput::dvb_open DMX_SET_PES_FILTER for raw: %s\n",
 342                                 strerror(errno));
 343                         ret = 1;
 344                 }
 345         }
 346 // Open transport stream for reading
 347         if( !ret && (dvb_fd = ::open(dvb_path, O_RDONLY+O_NONBLOCK)) < 0 ) {
 348                 fprintf(stderr, "DeviceDVBInput::dvb_open %s: %s\n",
 349                         dvb_path, strerror(errno));
 350                 ret = 1;
 351         }
 352 #else
 353         dvb_fd = -1;
 354         ret = 1;
 355 #endif
 356 #else
 357         dvb_fd = open(TEST_DATA,O_RDONLY);
 358         if( dvb_fd < 0 ) ret = 1;
 359 #endif
 360         reset_signal();
 361         dvb_locked = 0;
 362         if( !ret ) ret = dvb_sync();
 363         if( !ret ) ret = dvb_status();
 364         if( ret ) dvb_close();
 365         return ret;
 366 }
 367
 368
 369 void DeviceDVBInput::dvb_close(int fe)
 370 {
 371         if( dvb_fd >= 0 ) { ::close(dvb_fd); dvb_fd = -1; }
 372         if( demux_fd >= 0 ) { ::close(demux_fd); demux_fd = -1; }
 373         if( fe && frontend_fd >= 0 ) { ::close(frontend_fd); frontend_fd = -1; }
 374 }
 375
 376
 377 int DeviceDVBInput::dvb_status()
 378 {
 379 //note: this function can take > 500ms to execute  (slowww)
 380 #ifndef TEST_DATA
 381         int locked = 0;
 382         reset_signal();
 383 #ifdef HAVE_DVB
 384         int &fe = frontend_fd;
 385         if( fe < 0 ) return -1;
 386         fe_status_t st;  memset(&st, 0, sizeof(st));
 387         if( ioctl(fe, FE_READ_STATUS, &st) ) return 1;
 388         if( (st & FE_TIMEDOUT) != 0 ) return 1;
 389         signal_lck = (st & FE_HAS_LOCK) != 0 ? 1 : 0;
 390         signal_crr = (st & FE_HAS_CARRIER) != 0 ? 1 : 0;
 391         uint16_t power = 0;
 392         signal_pwr = ioctl(fe,FE_READ_SIGNAL_STRENGTH,&power) ? -1 :
 393                 drange(power, pwr_min, pwr_max);
 394         uint16_t ratio = 0;
 395         signal_snr = ioctl(fe, FE_READ_SNR, &ratio) ? -1 :
 396                 drange(ratio, snr_min, snr_max);
 397         uint32_t rate = 0;
 398         signal_ber = ioctl(fe, FE_READ_BER, &rate) ? -1 : rate;
 399         uint32_t errs = 0;
 400         signal_unc = ioctl(fe, FE_READ_UNCORRECTED_BLOCKS, &errs) ? -1 : errs;
 401         if( signal_lck && signal_ber >= 0 && signal_ber < 255 ) locked = 1;
 402         if( dvb_locked != locked ) {
 403                 printf(_("** %scarrier, dvb_locked %s\n"),
 404                          signal_crr ? "" : _("no "), locked ? _("lock") : _("lost") );
 405         }
 406 #endif
 407         dvb_locked = locked;
 408 #endif
 409         return 0;
 410 }
 411
 412 int DeviceDVBInput::drange(int v, int min, int max)
 413 {
 414         if( (v-=min) < 0 || (max-=min) <= 0 ) return 0;
 415         return v>max ? 65535U : (65535U * v) / max;
 416 }
 417
 418 void DeviceDVBInput::reset_signal()
 419 {
 420         signal_pwr = signal_crr = signal_lck =
 421         signal_snr = signal_ber = signal_unc = -1;
 422 }
 423
 424 int DeviceDVBInput::open_dev(int color_model)
 425 {
 426         int ret = 0;
 427         dvb_lock->lock("DeviceDVBInput::open_dev");
 428         if( dvb_fd < 0 )
 429         {
 430                 ret = dvb_open();
 431                 if( ret ) {
 432                         printf("DeviceDVBInput::open_dev: adaptor %s open failed\n",
 433                                 dev_name);
 434                 }
 435         }
 436         if( !ret )
 437                 dvb_input_status->start();
 438         else
 439                 dvb_close();
 440         dvb_lock->unlock();
 441         return ret;
 442 }
 443
 444 void DeviceDVBInput::close_dev()
 445 {
 446         dvb_lock->lock("DeviceDVBInput::close_dev");
 447         dvb_input_status->stop();
 448         dvb_close();
 449         dvb_lock->unlock();
 450 }
 451
 452 int DeviceDVBInput::status_dev()
 453 {
 454         int result = 0;
 455 #ifndef TEST_DATA
 456         result = frontend_fd >= 0 ? dvb_status() : 1;
 457 #endif
 458         return result;
 459 }
 460
 461 void DeviceDVBInput::set_signal_status(SignalStatus *stat)
 462 {
 463         dvb_lock->lock("DeviceDVBInput::set_signal_status");
 464         if( dvb_input_status->signal_status )
 465                 dvb_input_status->signal_status->disconnect();
 466         dvb_input_status->signal_status = stat;
 467         if( stat ) stat->dvb_input = this;
 468         dvb_lock->unlock();
 469 }
 470
 471 #endif