oss.c
上传用户:kjfoods
上传日期:2020-07-06
资源大小:29949k
文件大小:13k
源码类别:

midi

开发平台:

Unix_Linux

  1. /*****************************************************************************
  2.  * oss.c : OSS input module for vlc
  3.  *****************************************************************************
  4.  * Copyright (C) 2002-2009 the VideoLAN team
  5.  * $Id: eb0d6246d296a97e01508e02f073579ee2244e02 $
  6.  *
  7.  * Authors: Benjamin Pracht <bigben at videolan dot org>
  8.  *          Richard Hosking <richard at hovis dot net>
  9.  *          Antoine Cellerier <dionoea at videolan d.t org>
  10.  *          Dennis Lou <dlou99 at yahoo dot com>
  11.  *
  12.  * This program is free software; you can redistribute it and/or modify
  13.  * it under the terms of the GNU General Public License as published by
  14.  * the Free Software Foundation; either version 2 of the License, or
  15.  * (at your option) any later version.
  16.  *
  17.  * This program is distributed in the hope that it will be useful,
  18.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  19.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  20.  * GNU General Public License for more details.
  21.  *
  22.  * You should have received a copy of the GNU General Public License
  23.  * along with this program; if not, write to the Free Software
  24.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  25.  *****************************************************************************/
  26. /*****************************************************************************
  27.  * Preamble
  28.  *****************************************************************************/
  29. #ifdef HAVE_CONFIG_H
  30. # include "config.h"
  31. #endif
  32. #include <vlc_common.h>
  33. #include <vlc_plugin.h>
  34. #include <vlc_access.h>
  35. #include <vlc_demux.h>
  36. #include <vlc_input.h>
  37. #include <vlc_vout.h>
  38. #include <ctype.h>
  39. #include <fcntl.h>
  40. #include <unistd.h>
  41. #include <sys/ioctl.h>
  42. #include <sys/mman.h>
  43. #ifdef HAVE_SYS_SOUNDCARD_H
  44. #  include <sys/soundcard.h>
  45. #endif
  46. #ifdef HAVE_SOUNDCARD_H
  47. #  include <soundcard.h>
  48. #endif
  49. #include <poll.h>
  50. /*****************************************************************************
  51.  * Module descriptior
  52.  *****************************************************************************/
  53. static int  DemuxOpen ( vlc_object_t * );
  54. static void DemuxClose( vlc_object_t * );
  55. #define STEREO_TEXT N_( "Stereo" )
  56. #define STEREO_LONGTEXT N_( 
  57.     "Capture the audio stream in stereo." )
  58. #define SAMPLERATE_TEXT N_( "Samplerate" )
  59. #define SAMPLERATE_LONGTEXT N_( 
  60.     "Samplerate of the captured audio stream, in Hz (eg: 11025, 22050, 44100, 48000)" )
  61. #define CACHING_TEXT N_("Caching value in ms")
  62. #define CACHING_LONGTEXT N_( 
  63.     "Caching value for OSS captures. This " 
  64.     "value should be set in milliseconds." )
  65. #define OSS_DEFAULT "/dev/dsp"
  66. #define CFG_PREFIX "oss-"
  67. vlc_module_begin ()
  68.     set_shortname( N_("OSS") )
  69.     set_description( N_("OSS input") )
  70.     set_category( CAT_INPUT )
  71.     set_subcategory( SUBCAT_INPUT_ACCESS )
  72.     add_shortcut( "oss" )
  73.     set_capability( "access_demux", 10 )
  74.     set_callbacks( DemuxOpen, DemuxClose )
  75.     add_bool( CFG_PREFIX "stereo", true, NULL, STEREO_TEXT, STEREO_LONGTEXT,
  76.                 true )
  77.     add_integer( CFG_PREFIX "samplerate", 48000, NULL, SAMPLERATE_TEXT,
  78.                 SAMPLERATE_LONGTEXT, true )
  79.     add_integer( CFG_PREFIX "caching", DEFAULT_PTS_DELAY / 1000, NULL,
  80.                 CACHING_TEXT, CACHING_LONGTEXT, true )
  81. vlc_module_end ()
  82. /*****************************************************************************
  83.  * Access: local prototypes
  84.  *****************************************************************************/
  85. static int DemuxControl( demux_t *, int, va_list );
  86. static int Demux( demux_t * );
  87. static block_t* GrabAudio( demux_t *p_demux );
  88. static int OpenAudioDev( demux_t * );
  89. static int OpenAudioDevOss( demux_t * );
  90. static bool ProbeAudioDevOss( demux_t *, const char *psz_device );
  91. struct buffer_t
  92. {
  93.     void *  start;
  94.     size_t  length;
  95. };
  96. struct demux_sys_t
  97. {
  98.     const char *psz_device;  /* OSS device from MRL */
  99.     int  i_fd;
  100.     /* Audio */
  101.     int i_cache;
  102.     unsigned int i_sample_rate;
  103.     bool b_stereo;
  104.     size_t i_max_frame_size;
  105.     block_t *p_block;
  106.     es_out_id_t *p_es;
  107.     int64_t i_next_demux_date; /* Used to handle oss:// as input-slave properly */
  108. };
  109. static int FindMainDevice( demux_t *p_demux )
  110. {
  111.     msg_Dbg( p_demux, "opening device '%s'", p_demux->p_sys->psz_device );
  112.     if( ProbeAudioDevOss( p_demux, p_demux->p_sys->psz_device ) )
  113.     {
  114.         msg_Dbg( p_demux, "'%s' is an audio device", p_demux->p_sys->psz_device );
  115.         p_demux->p_sys->i_fd = OpenAudioDev( p_demux );
  116.     }
  117.     if( p_demux->p_sys->i_fd < 0 )
  118.         return VLC_EGENERIC;
  119.     return VLC_SUCCESS;
  120. }
  121. /*****************************************************************************
  122.  * DemuxOpen: opens oss device, access_demux callback
  123.  *****************************************************************************
  124.  *
  125.  * url: <oss device>::::
  126.  *
  127.  *****************************************************************************/
  128. static int DemuxOpen( vlc_object_t *p_this )
  129. {
  130.     demux_t     *p_demux = (demux_t*)p_this;
  131.     demux_sys_t *p_sys;
  132.     /* Only when selected */
  133.     if( *p_demux->psz_access == '' ) return VLC_EGENERIC;
  134.     /* Set up p_demux */
  135.     p_demux->pf_control = DemuxControl;
  136.     p_demux->pf_demux = Demux;
  137.     p_demux->info.i_update = 0;
  138.     p_demux->info.i_title = 0;
  139.     p_demux->info.i_seekpoint = 0;
  140.     p_demux->p_sys = p_sys = calloc( 1, sizeof( demux_sys_t ) );
  141.     if( p_sys == NULL ) return VLC_ENOMEM;
  142.     p_sys->i_sample_rate = var_CreateGetInteger( p_demux, CFG_PREFIX "samplerate" );
  143.     p_sys->b_stereo = var_CreateGetBool( p_demux, CFG_PREFIX "stereo" );
  144.     p_sys->i_cache = var_CreateGetInteger( p_demux, CFG_PREFIX "caching" );
  145.     p_sys->i_fd = -1;
  146.     p_sys->p_es = NULL;
  147.     p_sys->p_block = NULL;
  148.     p_sys->i_next_demux_date = -1;
  149.     if( p_demux->psz_path && *p_demux->psz_path )
  150.         p_sys->psz_device = p_demux->psz_path;
  151.     else
  152.         p_sys->psz_device = OSS_DEFAULT;
  153.     if( FindMainDevice( p_demux ) != VLC_SUCCESS )
  154.     {
  155.         DemuxClose( p_this );
  156.         return VLC_EGENERIC;
  157.     }
  158.     return VLC_SUCCESS;
  159. }
  160. /*****************************************************************************
  161.  * Close: close device, free resources
  162.  *****************************************************************************/
  163. static void DemuxClose( vlc_object_t *p_this )
  164. {
  165.     demux_t     *p_demux = (demux_t *)p_this;
  166.     demux_sys_t *p_sys   = p_demux->p_sys;
  167.     if( p_sys->i_fd >= 0 ) close( p_sys->i_fd );
  168.     if( p_sys->p_block ) block_Release( p_sys->p_block );
  169.     free( p_sys );
  170. }
  171. /*****************************************************************************
  172.  * DemuxControl:
  173.  *****************************************************************************/
  174. static int DemuxControl( demux_t *p_demux, int i_query, va_list args )
  175. {
  176.     demux_sys_t *p_sys = p_demux->p_sys;
  177.     bool *pb;
  178.     int64_t    *pi64;
  179.     switch( i_query )
  180.     {
  181.         /* Special for access_demux */
  182.         case DEMUX_CAN_PAUSE:
  183.         case DEMUX_CAN_SEEK:
  184.         case DEMUX_SET_PAUSE_STATE:
  185.         case DEMUX_CAN_CONTROL_PACE:
  186.             pb = (bool*)va_arg( args, bool * );
  187.             *pb = false;
  188.             return VLC_SUCCESS;
  189.         case DEMUX_GET_PTS_DELAY:
  190.             pi64 = (int64_t*)va_arg( args, int64_t * );
  191.             *pi64 = (int64_t)p_sys->i_cache * 1000;
  192.             return VLC_SUCCESS;
  193.         case DEMUX_GET_TIME:
  194.             pi64 = (int64_t*)va_arg( args, int64_t * );
  195.             *pi64 = mdate();
  196.             return VLC_SUCCESS;
  197.         case DEMUX_SET_NEXT_DEMUX_TIME:
  198.             p_sys->i_next_demux_date = (int64_t)va_arg( args, int64_t );
  199.             return VLC_SUCCESS;
  200.         /* TODO implement others */
  201.         default:
  202.             return VLC_EGENERIC;
  203.     }
  204.     return VLC_EGENERIC;
  205. }
  206. /*****************************************************************************
  207.  * Demux: Processes the audio frame
  208.  *****************************************************************************/
  209. static int Demux( demux_t *p_demux )
  210. {
  211.     demux_sys_t *p_sys = p_demux->p_sys;
  212.     struct pollfd fd;
  213.     fd.fd = p_sys->i_fd;
  214.     fd.events = POLLIN|POLLPRI;
  215.     fd.revents = 0;
  216.     block_t *p_block = NULL;
  217.     do
  218.     {
  219.         if( p_block )
  220.         {
  221.             es_out_Send( p_demux->out, p_sys->p_es, p_block );
  222.             p_block = NULL;
  223.         }
  224.         /* Wait for data */
  225.         if( poll( &fd, 1, 10 ) ) /* Timeout after 0.01 seconds. Bigger delays are an issue when used with/as an input-slave since all the inputs run in the same thread. */
  226.         {
  227.             if( fd.revents & (POLLIN|POLLPRI) )
  228.             {
  229.                 p_block = GrabAudio( p_demux );
  230.                 if( p_block )
  231.                     es_out_Control( p_demux->out, ES_OUT_SET_PCR, p_block->i_pts );
  232.             }
  233.         }
  234.     } while( p_block && p_sys->i_next_demux_date > 0 &&
  235.              p_block->i_pts < p_sys->i_next_demux_date );
  236.     if( p_block )
  237.         es_out_Send( p_demux->out, p_sys->p_es, p_block );
  238.     return 1;
  239. }
  240. /*****************************************************************************
  241.  * GrabAudio: Grab an audio frame
  242.  *****************************************************************************/
  243. static block_t* GrabAudio( demux_t *p_demux )
  244. {
  245.     demux_sys_t *p_sys = p_demux->p_sys;
  246.     struct audio_buf_info buf_info;
  247.     int i_read = 0, i_correct;
  248.     block_t *p_block;
  249.     if( p_sys->p_block ) p_block = p_sys->p_block;
  250.     else p_block = block_New( p_demux, p_sys->i_max_frame_size );
  251.     if( !p_block )
  252.     {
  253.         msg_Warn( p_demux, "cannot get block" );
  254.         return 0;
  255.     }
  256.     p_sys->p_block = p_block;
  257.     i_read = read( p_sys->i_fd, p_block->p_buffer,
  258.                 p_sys->i_max_frame_size );
  259.     if( i_read <= 0 ) return 0;
  260.     p_block->i_buffer = i_read;
  261.     p_sys->p_block = 0;
  262.     /* Correct the date because of kernel buffering */
  263.     i_correct = i_read;
  264.     if( ioctl( p_sys->i_fd, SNDCTL_DSP_GETISPACE, &buf_info ) == 0 )
  265.     {
  266.         i_correct += buf_info.bytes;
  267.     }
  268.     /* Timestamp */
  269.     p_block->i_pts = p_block->i_dts =
  270.         mdate() - INT64_C(1000000) * (mtime_t)i_correct /
  271.         2 / ( p_sys->b_stereo ? 2 : 1) / p_sys->i_sample_rate;
  272.     return p_block;
  273. }
  274. /*****************************************************************************
  275.  * OpenAudioDev: open and set up the audio device and probe for capabilities
  276.  *****************************************************************************/
  277. static int OpenAudioDevOss( demux_t *p_demux )
  278. {
  279.     int i_fd;
  280.     int i_format;
  281.     i_fd = open( p_demux->p_sys->psz_device, O_RDONLY | O_NONBLOCK );
  282.     if( i_fd < 0 )
  283.     {
  284.         msg_Err( p_demux, "cannot open OSS audio device (%m)" );
  285.         goto adev_fail;
  286.     }
  287.     i_format = AFMT_S16_LE;
  288.     if( ioctl( i_fd, SNDCTL_DSP_SETFMT, &i_format ) < 0
  289.         || i_format != AFMT_S16_LE )
  290.     {
  291.         msg_Err( p_demux,
  292.                  "cannot set audio format (16b little endian) (%m)" );
  293.         goto adev_fail;
  294.     }
  295.     if( ioctl( i_fd, SNDCTL_DSP_STEREO,
  296.                &p_demux->p_sys->b_stereo ) < 0 )
  297.     {
  298.         msg_Err( p_demux, "cannot set audio channels count (%m)" );
  299.         goto adev_fail;
  300.     }
  301.     if( ioctl( i_fd, SNDCTL_DSP_SPEED,
  302.                &p_demux->p_sys->i_sample_rate ) < 0 )
  303.     {
  304.         msg_Err( p_demux, "cannot set audio sample rate (%m)" );
  305.         goto adev_fail;
  306.     }
  307.     p_demux->p_sys->i_max_frame_size = 6 * 1024;
  308.     return i_fd;
  309.  adev_fail:
  310.     if( i_fd >= 0 ) close( i_fd );
  311.     return -1;
  312. }
  313. static int OpenAudioDev( demux_t *p_demux )
  314. {
  315.     demux_sys_t *p_sys = p_demux->p_sys;
  316.     int i_fd = OpenAudioDevOss( p_demux );
  317.     if( i_fd < 0 )
  318.         return i_fd;
  319.     msg_Dbg( p_demux, "opened adev=`%s' %s %dHz",
  320.              p_sys->psz_device, p_sys->b_stereo ? "stereo" : "mono",
  321.              p_sys->i_sample_rate );
  322.     es_format_t fmt;
  323.     es_format_Init( &fmt, AUDIO_ES, VLC_FOURCC('a','r','a','w') );
  324.     fmt.audio.i_channels = p_sys->b_stereo ? 2 : 1;
  325.     fmt.audio.i_rate = p_sys->i_sample_rate;
  326.     fmt.audio.i_bitspersample = 16;
  327.     fmt.audio.i_blockalign = fmt.audio.i_channels * fmt.audio.i_bitspersample / 8;
  328.     fmt.i_bitrate = fmt.audio.i_channels * fmt.audio.i_rate * fmt.audio.i_bitspersample;
  329.     msg_Dbg( p_demux, "new audio es %d channels %dHz",
  330.              fmt.audio.i_channels, fmt.audio.i_rate );
  331.     p_sys->p_es = es_out_Add( p_demux->out, &fmt );
  332.     return i_fd;
  333. }
  334. /*****************************************************************************
  335.  * ProbeAudioDev: probe audio for capabilities
  336.  *****************************************************************************/
  337. static bool ProbeAudioDevOss( demux_t *p_demux, const char *psz_device )
  338. {
  339.     int i_caps;
  340.     int i_fd = open( psz_device, O_RDONLY | O_NONBLOCK );
  341.     if( i_fd < 0 )
  342.     {
  343.         msg_Err( p_demux, "cannot open device %s for OSS audio (%m)", psz_device );
  344.         goto open_failed;
  345.     }
  346.     /* this will fail if the device is video */
  347.     if( ioctl( i_fd, SNDCTL_DSP_GETCAPS, &i_caps ) < 0 )
  348.     {
  349.         msg_Err( p_demux, "cannot get audio caps (%m)" );
  350.         goto open_failed;
  351.     }
  352.     if( i_fd >= 0 ) close( i_fd );
  353.     return true;
  354. open_failed:
  355.     if( i_fd >= 0 ) close( i_fd );
  356.     return false;
  357. }