RtspFileRaw.cxx
上传用户:sy_wanhua
上传日期:2013-07-25
资源大小:3048k
文件大小:8k
源码类别:

流媒体/Mpeg4/MP4

开发平台:

C/C++

  1. /* ====================================================================
  2.  * The Vovida Software License, Version 1.0 
  3.  * 
  4.  * Copyright (c) 2000 Vovida Networks, Inc.  All rights reserved.
  5.  * 
  6.  * Redistribution and use in source and binary forms, with or without
  7.  * modification, are permitted provided that the following conditions
  8.  * are met:
  9.  * 
  10.  * 1. Redistributions of source code must retain the above copyright
  11.  *    notice, this list of conditions and the following disclaimer.
  12.  * 
  13.  * 2. Redistributions in binary form must reproduce the above copyright
  14.  *    notice, this list of conditions and the following disclaimer in
  15.  *    the documentation and/or other materials provided with the
  16.  *    distribution.
  17.  * 
  18.  * 3. The names "VOCAL", "Vovida Open Communication Application Library",
  19.  *    and "Vovida Open Communication Application Library (VOCAL)" must
  20.  *    not be used to endorse or promote products derived from this
  21.  *    software without prior written permission. For written
  22.  *    permission, please contact vocal@vovida.org.
  23.  *
  24.  * 4. Products derived from this software may not be called "VOCAL", nor
  25.  *    may "VOCAL" appear in their name, without prior written
  26.  *    permission of Vovida Networks, Inc.
  27.  * 
  28.  * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESSED OR IMPLIED
  29.  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  30.  * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, TITLE AND
  31.  * NON-INFRINGEMENT ARE DISCLAIMED.  IN NO EVENT SHALL VOVIDA
  32.  * NETWORKS, INC. OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT DAMAGES
  33.  * IN EXCESS OF 281421,000, NOR FOR ANY INDIRECT, INCIDENTAL, SPECIAL,
  34.  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  35.  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  36.  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  37.  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  38.  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
  39.  * USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
  40.  * DAMAGE.
  41.  * 
  42.  * ====================================================================
  43.  * 
  44.  * This software consists of voluntary contributions made by Vovida
  45.  * Networks, Inc. and many individuals on behalf of Vovida Networks,
  46.  * Inc.  For more information on Vovida Networks, Inc., please see
  47.  * <http://www.vovida.org/>.
  48.  *
  49.  */
  50. static const char* const RtspFileRaw_cxx_version =
  51.     "$Id: RtspFileRaw.cxx,v 1.12 2001/05/15 20:26:08 bko Exp $";
  52. #include "RtspFileRaw.hxx"
  53. #include "RtpFileTypes.hxx"
  54. #include <fcntl.h>
  55. #include <unistd.h>
  56. #include <sys/stat.h>
  57. RtspFileRaw::RtspFileRaw( const string& shortFilename )
  58.     : RtspFile( shortFilename ),
  59.       m_xAudioFile( 0 )
  60. {
  61.     myFileMode = RTSP_FILE_MODE_UNKNOWN;
  62.     // read codec from filename - has default value if none found
  63.     string name = shortFilename;
  64.     string::size_type pos = name.find_last_of( '-' );
  65.     if( pos != string::npos )
  66.     {
  67.         // some type of codec string was found
  68.         bool success = false;
  69.         string codecString2 = name.substr( pos+1, string::npos );
  70.         cpLog( LOG_DEBUG, "Codec string %s", codecString2.c_str());
  71.         for( int i = 0; i< myNumberOfCodecs; i++ )
  72.         {
  73.             if( rtpFileTypeInfo[i].name == codecString2 )
  74.             {
  75.                 myFtIndex = i;
  76.                 name = name.substr( 0, pos );
  77.                 success = true;
  78.                 break;
  79.             }
  80.         }
  81.         if( !success )
  82.         {
  83.             // non codec string was found
  84.             cpLog( LOG_DEBUG, "Non-codec string found, using default value" );
  85.             myFtIndex = RtspConfiguration::instance().defaultCodec;
  86.         }
  87.     }
  88.     else
  89.     {
  90.         // no codec string was found
  91.         cpLog( LOG_DEBUG, "No codec string found, using default value" );
  92.         myFtIndex = RtspConfiguration::instance().defaultCodec;
  93.     }
  94.     // save stripped name
  95.     myShortFilename = name;
  96. }
  97. RtspFileRaw::~RtspFileRaw()
  98. {
  99.     close();
  100. }
  101. bool
  102. RtspFileRaw::saveHeader( const int ftIndex, RtspSdp& rtspSdp )
  103. {
  104.     //TODO
  105.     if( myFtIndex != ftIndex )
  106.     {
  107.         cpLog( LOG_ERR, "%d %d", myFtIndex, ftIndex );
  108.     }
  109. // RtspFileHandler will call this exact code if SDP_FILE defined
  110. #ifndef SDP_FILE
  111.     if( !writeSdpFile( rtspSdp ) )
  112.     {
  113.         cpLog( LOG_ERR, "Error in writeSdpFile()" );
  114.         return false;
  115.     }
  116. #ifndef REAL_AUDIO
  117.     Data controlData = "a=control:rtsp://";
  118.     //TODO need to get this data into this class
  119.     //controlData += filenameUrl();
  120.     controlData += "rn";
  121.     deleteLineSdpFile( "a=control:" );
  122.     appendSdpFile( controlData.getData(), controlData.length() );
  123. #endif
  124. #endif
  125.     FILE* xAudioFile = fopen( localFilename().getData(), "wb" );
  126.     if( !xAudioFile )
  127.     {
  128.         cpLog( LOG_DEBUG, "fopen %s - failed while saving header",
  129.                localFilename().getData() );
  130.         return false;
  131.     }
  132.     dummyWrite( xAudioFile, rtpFileTypeInfo[ftIndex].packetSize );
  133.     // close audio file
  134.     fclose( xAudioFile );
  135.     return true;
  136. }
  137. bool
  138. RtspFileRaw::loadHeader( int* ftIndex, int* lengthInMs )
  139. {
  140.     // begin kim
  141.     *ftIndex = myFtIndex;
  142.     *lengthInMs = 0;
  143.     // end kim
  144. /* // begin mera
  145.         if(myFtIndex)
  146.         *ftIndex=myFtIndex;
  147.     else
  148.         myFtIndex=*ftIndex;
  149.     FILE* audioFile = fopen(filename().getData(),"rb");
  150.     if( !audioFile )
  151.     {
  152.         cpLog( LOG_ERR, "fopen %s - failed ", filename().getData());
  153.         return false;
  154.     }
  155.     fclose( audioFile );
  156.     // end mera
  157. */
  158.     // read total file length
  159.     struct stat ifstat;
  160.     if( stat( localFilename().getData(), &ifstat ) )
  161.     {
  162.         cpLog( LOG_ERR, "Error in stat()" );
  163.         return false;
  164.     }
  165.     // calculate file length
  166.     *lengthInMs = ( (int)( (float)ifstat.st_size /
  167.                     (float)rtpFileTypeInfo[*ftIndex].packetSize ) ) *
  168.                   rtpFileTypeInfo[*ftIndex].intervalMs;
  169.     cpLog( LOG_DEBUG, "File length is %d ms", *lengthInMs );
  170.     return true;
  171. }
  172. bool
  173. RtspFileRaw::open( bool bReadWrite, int ftIndex )
  174. {
  175.     if( !bReadWrite ) // read mode
  176.     {
  177.         m_xAudioFile = fopen( localFilename().getData(), "rb" );
  178.         if( !m_xAudioFile )
  179.         {
  180.             cpLog( LOG_ERR, "fopen %s - failed",
  181.                    localFilename().getData() );
  182.             return false;
  183.         }
  184.        myFileMode = RTSP_FILE_MODE_PLAY;
  185.     }
  186.     else // write mode
  187.     {
  188.         m_xAudioFile = fopen( localFilename().getData(), "wb" );
  189.         if( !m_xAudioFile )
  190.         {
  191.             cpLog( LOG_ERR, "fopen %s - failed",
  192.                    localFilename().getData() );
  193.             return false;
  194.         }
  195.         myFileMode = RTSP_FILE_MODE_REC;
  196.     }
  197.     return true;
  198. }
  199. bool
  200. RtspFileRaw::close()
  201. {
  202.     if( m_xAudioFile )
  203.     {
  204.         if( fclose( m_xAudioFile ) != 0 )
  205.         {
  206.             cpLog( LOG_ERR, "Error in fclose()" );
  207.         }
  208.         m_xAudioFile = 0;
  209.         updateSdpFileRange();
  210.         return true;
  211.     }
  212.     return false;
  213. }
  214. int
  215. RtspFileRaw::read( void* data, int max, 
  216.                    unsigned short* pSeqNum, unsigned int* pTS )
  217. {
  218.     int cc = fread( data, max, 1, m_xAudioFile );
  219.     if( cc > 0 )
  220.     {
  221.         cc = max;
  222.     }
  223.     return cc;
  224. }
  225. int
  226. RtspFileRaw::write( void* data, int max, 
  227.                     unsigned short uSeqNum, unsigned int uTS )
  228. {
  229.     int cc = fwrite( data, max, 1, m_xAudioFile );
  230.     if( cc > 0 )
  231.     {
  232.         cc = max;
  233.     }
  234.     return cc;
  235. }
  236. long
  237. RtspFileRaw::seek( long samples, int whence )
  238. {
  239.     long seek_size = ( (long)( (float)samples /
  240.                        (float)rtpFileTypeInfo[myFtIndex].intervalMs ) ) *
  241.                      rtpFileTypeInfo[myFtIndex].packetSize;
  242.     return fseek( m_xAudioFile, seek_size, whence );
  243. }
  244. void
  245. RtspFileRaw::dummyWrite( FILE* xAudioFile, const int dummyLen )
  246. {
  247.     // write some dummy data
  248.     cpLog( LOG_DEBUG, "Writing %d bytes of dummy data", dummyLen );
  249.     char dummyBuffer[dummyLen];
  250.     memset( dummyBuffer, 0, dummyLen );
  251.     if( !fwrite( dummyBuffer, dummyLen, 1, xAudioFile ) )
  252.     {
  253.         cpLog( LOG_DEBUG, "fwrite result error %d", dummyLen );
  254.     }
  255. }
  256. /* Local Variables: */
  257. /* c-file-style: "stroustrup" */
  258. /* indent-tabs-mode: nil */
  259. /* c-file-offsets: ((access-label . -) (inclass . ++)) */
  260. /* c-basic-offset: 4 */
  261. /* End: */