CameraInputWindows.cpp

Go to the documentation of this file.

/*
-----------------------------------------------------------------------------
This source file is part of OpenSpace3D
For the latest info, see http://www.openspace3d.com
 
Copyright (c) 2012 I-maginer
 
This program is free software; you can redistribute it and/or modify it under
the terms of the GNU Lesser General Public License as published by the Free Software
Foundation; either version 2 of the License, or (at your option) any later
version.
 
This program is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.
 
You should have received a copy of the GNU Lesser General Public License along with
this program; if not, write to the Free Software Foundation, Inc., 59 Temple
Place - Suite 330, Boston, MA 02111-1307, USA, or go to
http://www.gnu.org/copyleft/lesser.txt
 
-----------------------------------------------------------------------------
*/
 
#include "ICameraInput.h"
#include "CameraInputWindows.h"
#include <dshow.h>
 
CameraInputWindows::CameraInputWindows(int index)
  : ICameraInput(index)
{
  mVI = new videoInput();
}
 
bool CameraInputWindows::Initialize()
{
  bool ret = false;
 
  ret = mVI->setupDevice(mIndex); // crash when problem occured
  if (!ret)
    ret = mVI->setupDevice(mIndex, VI_COMPOSITE); // crash when problem occured
 
  if (ret)
    mBuffer.resize(mVI->getSize(mIndex));
  
  return ret;
}
 
void CameraInputWindows::Close()
{
  mVI->stopDevice(mIndex);
}
 
CameraInputWindows::~CameraInputWindows()
{
  mVI->stopDevice(mIndex);
  SAFE_DELETE(mVI);
}
 
bool CameraInputWindows::IsOpened()
{
  return mVI->isDeviceSetup(mIndex);
}
 
cv::Mat CameraInputWindows::UpdateImage()
{
  if (!IsOpened())
    throw std::logic_error("Device not initialised");
 
  // Capture the picture
  if (!mVI->isFrameNew(mIndex) || !mVI->getPixels(mIndex, &mBuffer[0], false, true))
  {
    throw std::logic_error("Device has bad pixel buffer");
  }
 
  cv::Mat frame(mVI->getHeight(mIndex), mVI->getWidth(mIndex), CV_8UC3, &mBuffer[0]);
 
  //if (mMirrorMode)    cv::flip(frame,frame,1);
  return frame;
 
}
 
int CameraInputWindows::GetWidth()
{
  if (IsOpened())
    return mVI->getWidth(mIndex);
  else
    return 0;
}
 
int CameraInputWindows::GetHeight()
{
  if (IsOpened())
    return mVI->getHeight(mIndex);
  else
    return 0;
}
 
void CameraInputWindows::SetSize(int width, int height)
{
  if (IsOpened())
  {
    mVI->stopDevice(mIndex);
    mVI->setupDevice(mIndex, width, height);
 
    //if the size is not supported we restart with the default size
    if (!IsOpened())
      mVI->setupDevice(mIndex);
 
    mBuffer.resize(mVI->getSize(mIndex));
  }
}
 
bool CameraInputWindows::TakeSnapshot(std::string path)
{
  if (!IsOpened())
    return false;
 
  cv::Mat frame = UpdateImage();
  if (frame.empty())
    return false;
 
  if (!cv::imwrite(path, frame))
    return false;
 
  std::cout << "Frame written to " << path << "." << std::endl;
  return true;
}