lib/libUPnP/Platinum/Source/Extras/PltFrameStream.cpp

   1 /*****************************************************************
   2 |
   3 |   Platinum - Frame Stream
   4 |
   5 | Copyright (c) 2004-2010, Plutinosoft, LLC.
   6 | All rights reserved.
   7 | http://www.plutinosoft.com
   8 |
   9 | This program is free software; you can redistribute it and/or
  10 | modify it under the terms of the GNU General Public License
  11 | as published by the Free Software Foundation; either version 2
  12 | of the License, or (at your option) any later version.
  13 |
  14 | OEMs, ISVs, VARs and other distributors that combine and
  15 | distribute commercially licensed software with Platinum software
  16 | and do not wish to distribute the source code for the commercially
  17 | licensed software under version 2, or (at your option) any later
  18 | version, of the GNU General Public License (the "GPL") must enter
  19 | into a commercial license agreement with Plutinosoft, LLC.
  20 | licensing@plutinosoft.com
  21 |
  22 | This program is distributed in the hope that it will be useful,
  23 | but WITHOUT ANY WARRANTY; without even the implied warranty of
  24 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  25 | GNU General Public License for more details.
  26 |
  27 | You should have received a copy of the GNU General Public License
  28 | along with this program; see the file LICENSE.txt. If not, write to
  29 | the Free Software Foundation, Inc.,
  30 | 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  31 | http://www.gnu.org/licenses/gpl-2.0.html
  32 |
  33 ****************************************************************/
  34
  35 /*----------------------------------------------------------------------
  36 |   includes
  37 +---------------------------------------------------------------------*/
  38 #include "PltFrameStream.h"
  39
  40 NPT_SET_LOCAL_LOGGER("platinum.core.framestream")
  41
  42 /*----------------------------------------------------------------------
  43 |   PLT_InputFrameStream::PLT_InputFrameStream
  44 +---------------------------------------------------------------------*/
  45 PLT_InputFrameStream::PLT_InputFrameStream(NPT_Reference<PLT_FrameBuffer>& frame_buffer,
  46                                            const char*                     boundary) :
  47     m_FrameBuffer(frame_buffer),
  48     m_LastFrameIndex(0),
  49     m_Boundary(boundary),
  50     m_Eos(false)
  51 {
  52     m_FrameBuffer->AddReader();
  53 }
  54
  55 /*----------------------------------------------------------------------
  56 |   PLT_InputFrameStream::~PLT_InputFrameStream
  57 +---------------------------------------------------------------------*/
  58 PLT_InputFrameStream::~PLT_InputFrameStream()
  59 {
  60     m_FrameBuffer->RemoveReader();
  61 }
  62
  63 /*----------------------------------------------------------------------
  64 |   PLT_InputFrameStream::GetAvailable
  65 +---------------------------------------------------------------------*/
  66 NPT_Result
  67 PLT_InputFrameStream::GetAvailable(NPT_LargeSize& available)
  68 {
  69     NPT_CHECK_WARNING(m_Part.GetAvailable(available));
  70
  71     if (available == 0 && !m_Eos) {
  72         NPT_CHECK_WARNING(FillBuffer());
  73         NPT_CHECK_WARNING(m_Part.GetAvailable(available));
  74     }
  75
  76     return NPT_SUCCESS;
  77 }
  78
  79 /*----------------------------------------------------------------------
  80 |   PLT_InputFrameStream::FillBuffer
  81 +---------------------------------------------------------------------*/
  82 NPT_Result
  83 PLT_InputFrameStream::FillBuffer()
  84 {
  85     // reset memorystream
  86     m_Part.SetDataSize(0);
  87
  88     // fetch next frame
  89     NPT_DataBuffer frame;
  90     NPT_Result result = m_FrameBuffer->GetNextFrame(m_LastFrameIndex, frame);
  91
  92     // error (EOS) or empty frame means we're done
  93     if (NPT_FAILED(result) || frame.GetDataSize() == 0) {
  94         m_Part.WriteLine("--" + m_Boundary + "--");
  95         m_Eos = true;
  96         return NPT_SUCCESS;
  97     }
  98
  99     m_Part.WriteLine("--" + m_Boundary);
 100     m_Part.WriteLine("Content-Type: " + NPT_String(m_FrameBuffer->GetMimeType()));
 101     m_Part.WriteLine("Content-Length: "+NPT_String::FromInteger(frame.GetDataSize()));
 102     m_Part.WriteLine("");
 103     m_Part.Write(frame.GetData(), frame.GetDataSize());
 104     m_Part.WriteLine("");
 105     return NPT_SUCCESS;
 106 }
 107
 108 /*----------------------------------------------------------------------
 109 |   PLT_InputFrameStream::Read
 110 +---------------------------------------------------------------------*/
 111 NPT_Result
 112 PLT_InputFrameStream::Read(void*     buffer,
 113                            NPT_Size  bytes_to_read,
 114                            NPT_Size* bytes_read /*= 0*/)
 115 {
 116
 117     if (bytes_read) *bytes_read = 0;
 118
 119     if (bytes_to_read == 0) {
 120         return NPT_SUCCESS;
 121     }
 122
 123     // make sure we have data
 124     NPT_LargeSize available;
 125     NPT_CHECK_WARNING(GetAvailable(available));
 126
 127     return m_Part.Read(buffer, bytes_to_read, bytes_read);
 128 }