MediaPinType.cpp
上传用户:liguizhu
上传日期:2015-11-01
资源大小:2422k
文件大小:3k
源码类别:

P2P编程

开发平台:

Visual C++

  1. /*
  2.  *  Openmysee
  3.  *
  4.  *  This program is free software; you can redistribute it and/or modify
  5.  *  it under the terms of the GNU General Public License as published by
  6.  *  the Free Software Foundation; either version 2 of the License, or
  7.  *  (at your option) any later version.
  8.  *
  9.  *  This program is distributed in the hope that it will be useful,
  10.  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.  *  GNU General Public License for more details.
  13.  *
  14.  *  You should have received a copy of the GNU General Public License
  15.  *  along with this program; if not, write to the Free Software
  16.  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17.  *
  18.  */
  19. #include "stdafx.h"
  20. #include ".mediapintype.h"
  21. SampleBuffer::SampleBuffer(void):m_buffer(NULL)
  22. {
  23. }
  24. SampleBuffer::~SampleBuffer()
  25. {
  26.     delete[] m_buffer;
  27. }
  28. bool    SampleBuffer::IsAnyDataSaved()
  29. {
  30.     return m_buffer != NULL;
  31. }
  32. void    SampleBuffer::SaveData(const SampleHeader& header, const BYTE* pData, const UINT maxSize)
  33. {
  34.     if(m_buffer != NULL)
  35.         Clean();
  36.     if(header.size == 0)
  37.     {
  38.         m_size = maxSize;
  39.     }
  40.     else
  41.         m_size = header.size;
  42.     
  43.     m_buffer = new BYTE[sizeof(SampleHeader) + m_size];
  44.     memcpy(m_buffer, &header, sizeof(SampleHeader));
  45.     memcpy(m_buffer + sizeof(SampleHeader), pData, m_size);
  46. }
  47. bool    SampleBuffer::PeerData(SampleHeader& header, BYTE* pData, const UINT maxSize) const
  48. {
  49.     if(m_buffer == NULL)
  50.         return false;
  51.     memcpy(&header, m_buffer, sizeof(SampleHeader));
  52.     UINT datalen;
  53.     if(m_size > maxSize)
  54.     {
  55.         datalen = maxSize;
  56.     }
  57.     else
  58.         datalen = m_size;
  59.     memcpy(pData, m_buffer + sizeof(SampleHeader), datalen);
  60.     return true;
  61. }
  62. bool    SampleBuffer::GetData(SampleHeader& header, BYTE* pData, const UINT maxSize)
  63. {
  64.     if(!PeerData(header, pData, maxSize))
  65.         return false;
  66.     Clean();
  67.     return true;
  68. }
  69. bool    SampleBuffer::Alloc(SampleHeader* &pHeaderbuf, BYTE* &pDatabuf, const size)
  70. {
  71.     Clean();
  72.     m_size = size;
  73.     m_buffer = new BYTE[sizeof(SampleHeader) + m_size];
  74.     pHeaderbuf = reinterpret_cast<SampleHeader*>(m_buffer);
  75.     pDatabuf = m_buffer + sizeof(SampleHeader);
  76.     return true;
  77. }
  78. void    SampleBuffer::Clean()
  79. {
  80.     delete[] m_buffer;
  81.     m_buffer = NULL;
  82. }
  83. MediaPinType::MediaPinType(void)
  84. {
  85. }
  86. MediaPinType::~MediaPinType(void)
  87. {
  88. }
  89. void MediaPinType::SaveMediaTypeData(const SampleHeader& header, const BYTE* pData, const UINT maxSize)
  90. {
  91.     UINT len;
  92.     if(maxSize < 256)
  93.         len = maxSize;
  94.     else
  95.         len = 256;
  96.     SaveData(header, pData, len);
  97. }
  98. bool MediaPinType::RestoreMediaTypeData(SampleHeader& header, BYTE* pData, const UINT maxSize) const
  99. {
  100.     UINT len;
  101.     if(maxSize < 256)
  102.         len = maxSize;
  103.     else
  104.         len = 256;
  105.     return PeerData(header, pData, len);
  106. }