ICCapturingListener.cxx

Go to the documentation of this file.

/*=Plus=header=begin======================================================
  Program: Plus
  Copyright (c) Laboratory for Percutaneous Surgery. All rights reserved.
  See License.txt for details.
=========================================================Plus=header=end*/

// ICCapturingListener.cpp: implementation of the ICCapturingListener class.
//----------------------------------------------------------------------------

#include "PlusConfigure.h"

#include "ICCapturingListener.h"
#include <iostream>
#include <sstream>

using namespace DShowLib;

//----------------------------------------------------------------------------
ICCapturingListener::ICCapturingListener()
{
  this->ICCapturingSourceNewFrameCallback = NULL; 
}

//----------------------------------------------------------------------------
ICCapturingListener::~ICCapturingListener()
{
}

//----------------------------------------------------------------------------
void ICCapturingListener::overlayCallback( Grabber& caller, smart_ptr<OverlayBitmap> pBitmap, const tsMediaSampleDesc& MediaSampleDesc)
{
  if( pBitmap->getEnable() == true ) // Draw only, if the overlay bitmap is enabled.
  {
    std::ostringstream ss;
    ss << MediaSampleDesc.FrameNumber << " ";
    pBitmap->drawText( RGB(255,0,0), 0, 0, ss.str().c_str() );
  }
}

//----------------------------------------------------------------------------
void ICCapturingListener::frameReady( Grabber& caller, smart_ptr<MemBuffer> pBuffer, DWORD currFrame)
{
  if (this->ICCapturingSourceNewFrameCallback != NULL)
  {
    pBuffer->lock(); 
    (*ICCapturingSourceNewFrameCallback)( pBuffer->getPtr(), pBuffer->getBufferSize(), currFrame );
    pBuffer->unlock(); 
  }
}

//----------------------------------------------------------------------------
void ICCapturingListener::setBufferSize( unsigned long NumBuffers )
{
  m_BufferWritten.resize( NumBuffers, false );
}

//----------------------------------------------------------------------------
void ICCapturingListener::saveImage( smart_ptr<MemBuffer> pBuffer, DWORD currFrame)
{
  if( currFrame < m_BufferWritten.size() )
  {
    std::ostringstream ss;
    ss << "image" << currFrame << ".bmp";
    saveToFileBMP( *pBuffer, ss.str().c_str() );
    m_BufferWritten.at( currFrame ) = true;
  }
}