vpPylonGrabberUsb.cpp

/****************************************************************************
 *
 * This file is part of the ViSP software.
 * Copyright (C) 2005 - 2017 by Inria. All rights reserved.
 *
 * This software is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 * See the file LICENSE.txt at the root directory of this source
 * distribution for additional information about the GNU GPL.
 *
 * For using ViSP with software that can not be combined with the GNU
 * GPL, please contact Inria about acquiring a ViSP Professional
 * Edition License.
 *
 * See http://visp.inria.fr for more information.
 *
 * This software was developed at:
 * Inria Rennes - Bretagne Atlantique
 * Campus Universitaire de Beaulieu
 * 35042 Rennes Cedex
 * France
 *
 * If you have questions regarding the use of this file, please contact
 * Inria at visp@inria.fr
 *
 * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 *
 * Description: Implementation of vpPylonGrabberUsb class.
 *
 * Authors:
 * Wenfeng CAI
 *
 *****************************************************************************/

#include "vpPylonGrabberUsb.h"

#ifdef VISP_HAVE_PYLON

#include <visp3/core/vpException.h>
#include <visp3/core/vpTime.h>

vpPylonGrabberUsb::vpPylonGrabberUsb() : m_camera(), m_index(0), m_numCameras(0), m_connected(false)
{
  getNumCameras();
}

vpPylonGrabberUsb::~vpPylonGrabberUsb() { close(); }

unsigned int vpPylonGrabberUsb::getNumCameras()
{
  Pylon::CTlFactory &TlFactory = Pylon::CTlFactory::GetInstance();
  Pylon::DeviceInfoList_t lstDevices;
  Pylon::DeviceInfoList_t filter; // Filter for USB cameras.
  Pylon::CBaslerUsbDeviceInfo usb_devinfo;
  filter.push_back(usb_devinfo);
  TlFactory.EnumerateDevices(lstDevices, filter);

  m_numCameras = lstDevices.size();
  return m_numCameras;
}

std::ostream &vpPylonGrabberUsb::getCameraInfo(std::ostream &os)
{
  connect();

  Pylon::CDeviceInfo deviceInfo = m_camera.GetDeviceInfo();
  // Get the camera control object.
  GenApi::INodeMap &control = m_camera.GetNodeMap();

  GenApi::CIntegerPtr widthMax = control.GetNode("WidthMax");
  GenApi::CIntegerPtr heightMax = control.GetNode("HeightMax");

  os << "Camera information:   " << std::endl;
  os << " Serial number      : " << deviceInfo.GetSerialNumber() << std::endl;
  os << " Camera model       : " << deviceInfo.GetModelName() << std::endl;
  os << " Camera vendor      : " << deviceInfo.GetVendorName() << std::endl;
  os << " Resolution         : " << widthMax->GetValue() << "x" << heightMax->GetValue() << std::endl;
  os << " Firmware version   : " << deviceInfo.GetDeviceVersion() << std::endl;

  return os;
}

Pylon::CInstantCamera *vpPylonGrabberUsb::getCameraHandler()
{
  connect();

  if (m_connected == true) {
    return &m_camera;
  } else {
    return NULL;
  }
}

float vpPylonGrabberUsb::getFrameRate()
{
  connect();

  float frame_rate = m_camera.AcquisitionFrameRate.GetValue();
  return frame_rate;
}

float vpPylonGrabberUsb::getGain()
{
  connect();

  if (GenApi::IsReadable(m_camera.Gain))
    return m_camera.Gain.GetValue();
  else
    throw vpException(vpException::notImplementedError, "Don't know how to get gain.");
}

float vpPylonGrabberUsb::getBlackLevel()
{
  connect();

  if (GenApi::IsReadable(m_camera.BlackLevel))
    return m_camera.BlackLevel.GetValue();
  else
    throw vpException(vpException::notImplementedError, "Don't know how to get blacklevel.");
}

float vpPylonGrabberUsb::getExposure()
{
  connect();

  if (GenApi::IsReadable(m_camera.ExposureTime))
    return m_camera.ExposureTime.GetValue() * 0.001;
  else
    throw vpException(vpException::notImplementedError, "Don't know how to get exposure.");
}

float vpPylonGrabberUsb::getGamma()
{
  connect();

  float gamma = m_camera.Gamma.GetValue();
  return gamma;
}

std::string vpPylonGrabberUsb::getCameraSerial(unsigned int index)
{
  getNumCameras();

  if (index >= m_numCameras) {
    throw(vpException(vpException::badValue, "The camera with index %u is not present. Only %d cameras connected.",
                      index, m_numCameras));
  }

  Pylon::CTlFactory &TlFactory = Pylon::CTlFactory::GetInstance();
  Pylon::DeviceInfoList_t lstDevices; // List of connected cameras
  Pylon::DeviceInfoList_t filter;     // Filter for USB cameras.
  Pylon::CBaslerUsbDeviceInfo usb_devinfo;
  filter.push_back(usb_devinfo);
  TlFactory.EnumerateDevices(lstDevices, filter);

  std::ostringstream os;
  os << lstDevices[index].GetSerialNumber();
  return os.str();
}

bool vpPylonGrabberUsb::loadUserSet(UserSetName user_set)
{
  connect();

  bool success = selectUserSet(user_set);

  if (success) {
    m_camera.UserSetLoad.Execute();
    vpTime::wait(200); // How long you have to wait?
    success = m_camera.UserSetLoad.IsDone();
  }

  return success;
}

vpPylonGrabber::UserSetName vpPylonGrabberUsb::getUserSetDefault()
{
  connect();

  Basler_UsbCameraParams::UserSetDefaultEnums user_set = m_camera.UserSetDefault.GetValue();

  switch (user_set) {
  case Basler_UsbCameraParams::UserSetDefault_Default:
    return USERSET_DEFAULT;
    break;
  case Basler_UsbCameraParams::UserSetDefault_UserSet1:
    return USERSET_USERSET1;
    break;
  case Basler_UsbCameraParams::UserSetDefault_UserSet2:
    return USERSET_USERSET2;
    break;
  case Basler_UsbCameraParams::UserSetDefault_UserSet3:
    return USERSET_USERSET3;
    break;
  default:
    return USERSET_UNKNOWN;
  }
}

void vpPylonGrabberUsb::setCameraIndex(unsigned int index)
{
  if (index >= m_numCameras) {
    throw(vpException(vpException::badValue, "The camera with index %u is not present. Only %d cameras connected.",
                      index, m_numCameras));
  }

  m_index = index;
}

void vpPylonGrabberUsb::setCameraSerial(const std::string &serial)
{
  m_numCameras = getNumCameras();
  for (unsigned int i = 0; i < m_numCameras; i++) {
    if (getCameraSerial(i) == serial) {
      m_index = i;
      return;
    }
  }
  throw(vpException(vpException::badValue, "The camera with serial id %s is not present.", serial.c_str()));
}

float vpPylonGrabberUsb::setFrameRate(float frame_rate)
{
  connect();

  m_camera.AcquisitionFrameRate.SetValue(frame_rate);

  return m_camera.AcquisitionFrameRate.GetValue();
}

float vpPylonGrabberUsb::setGain(bool gain_auto, float gain_value)
{
  connect();

  if (gain_auto)
    m_camera.GainAuto.SetValue(Basler_UsbCameraParams::GainAuto_Continuous);
  else
    m_camera.GainAuto.SetValue(Basler_UsbCameraParams::GainAuto_Off);

  if (GenApi::IsWritable(m_camera.Gain)) {
    m_camera.Gain.SetValue(gain_value);
    return m_camera.Gain.GetValue();
  } else
    throw vpException(vpException::notImplementedError, "Don't know how to set gain.");
}

float vpPylonGrabberUsb::setBlackLevel(float blacklevel_value)
{
  connect();

  if (GenApi::IsWritable(m_camera.BlackLevel)) {
    m_camera.BlackLevel.SetValue(blacklevel_value);
    return m_camera.BlackLevel.GetValue();
  } else
    throw vpException(vpException::notImplementedError, "Don't know how to set blacklevel.");
}

float vpPylonGrabberUsb::setExposure(bool exposure_on, bool exposure_auto, float exposure_value)
{
  connect();

  if (exposure_on)
    m_camera.ExposureMode.SetValue(Basler_UsbCameraParams::ExposureMode_Timed);
  else
    m_camera.ExposureMode.SetValue(Basler_UsbCameraParams::ExposureMode_TriggerWidth);

  if (exposure_auto)
    m_camera.ExposureAuto.SetValue(Basler_UsbCameraParams::ExposureAuto_Continuous);
  else
    m_camera.ExposureAuto.SetValue(Basler_UsbCameraParams::ExposureAuto_Off);

  if (GenApi::IsWritable(m_camera.ExposureTime)) {
    m_camera.ExposureTime.SetValue(exposure_value * 1000);
    return m_camera.ExposureTime.GetValue() * 0.001;
  } else
    throw vpException(vpException::notImplementedError, "Don't know how to set exposure.");
}

float vpPylonGrabberUsb::setGamma(bool gamma_on, float gamma_value)
{
  connect();

  if (GenApi::IsWritable(m_camera.Gamma)) {
    if (gamma_on)
      m_camera.Gamma.SetValue(gamma_value);
    else
      m_camera.Gamma.SetValue(1);
    return m_camera.Gamma.GetValue();
  } else
    throw vpException(vpException::notImplementedError, "Don't know how to set gamma.");
}

bool vpPylonGrabberUsb::saveUserSet(UserSetName user_set, bool set_default)
{
  connect();

  bool success = selectUserSet(user_set);

  if (success) {
    m_camera.UserSetSave.Execute();
    vpTime::wait(200); // How long you have to wait?
    success = m_camera.UserSetSave.IsDone();
  }

  if (success && set_default)
    success = setUserSetDefault(user_set);

  return success;
}

bool vpPylonGrabberUsb::setUserSetDefault(UserSetName user_set)
{
  connect();

  switch (user_set) {
  case USERSET_DEFAULT:
    m_camera.UserSetDefault.SetValue(Basler_UsbCameraParams::UserSetDefault_Default);
    return true;
    break;
  case USERSET_USERSET1:
    m_camera.UserSetDefault.SetValue(Basler_UsbCameraParams::UserSetDefault_UserSet1);
    return true;
    break;
  case USERSET_USERSET2:
    m_camera.UserSetDefault.SetValue(Basler_UsbCameraParams::UserSetDefault_UserSet2);
    return true;
    break;
  case USERSET_USERSET3:
    m_camera.UserSetDefault.SetValue(Basler_UsbCameraParams::UserSetDefault_UserSet3);
    return true;
    break;
  default:
    return false;
  }
}

void vpPylonGrabberUsb::startCapture()
{
  connect();

  if (!m_camera.IsGrabbing()) {
    m_camera.StartGrabbing(1);
  }
  if (m_connected && m_camera.IsGrabbing())
    init = true;
  else
    init = false;
}

void vpPylonGrabberUsb::stopCapture()
{
  if (m_camera.IsGrabbing()) {
    m_camera.StopGrabbing();
  }
  if (m_connected && m_camera.IsGrabbing())
    init = true;
  else
    init = false;
}

void vpPylonGrabberUsb::connect()
{
  if (m_connected == false) {
    m_numCameras = getNumCameras();
    if (m_numCameras == 0) {
      throw(vpException(vpException::fatalError, "No camera found"));
    }

    if (!m_camera.IsPylonDeviceAttached()) {
      Pylon::CTlFactory &TlFactory = Pylon::CTlFactory::GetInstance();
      Pylon::DeviceInfoList_t lstDevices;
      Pylon::DeviceInfoList_t filter; // Filter for USB cameras.
      Pylon::CBaslerUsbDeviceInfo usb_devinfo;
      filter.push_back(usb_devinfo);
      TlFactory.EnumerateDevices(lstDevices, filter);

      m_camera.Attach(TlFactory.CreateDevice(lstDevices[m_index]));
    }
    // Connect to a camera
    m_camera.Open();
    m_connected = true;
  }
  if (m_connected && m_camera.IsGrabbing())
    init = true;
  else
    init = false;
}

void vpPylonGrabberUsb::disconnect()
{
  if (m_connected == true) {
    m_camera.Close();
    m_connected = false;
  }
  if (m_connected && m_camera.IsGrabbing())
    init = true;
  else
    init = false;
}

void vpPylonGrabberUsb::close()
{
  stopCapture();
  disconnect();
}

void vpPylonGrabberUsb::acquire(vpImage<unsigned char> &I)
{
  open();

  Pylon::CGrabResultPtr grabResult;
  // Retrieve an image
  if (!m_camera.RetrieveResult(2000, grabResult)) {
    throw(vpException(vpException::fatalError, "Cannot retrieve image from camera with serial %s",
                      getCameraSerial(m_index).c_str()));
  }

  if (grabResult->GrabSucceeded()) {
    height = grabResult->GetHeight();
    width = grabResult->GetWidth();
    I.resize(height, width);

    Pylon::CImageFormatConverter imageConvert;
    imageConvert.OutputPixelFormat = Pylon::PixelType_Mono8;
    imageConvert.OutputPaddingX = 0;
    // Create a converted image
    imageConvert.Convert(I.bitmap, sizeof(unsigned char) * width * height, (Pylon::IImage &)grabResult);
  }
}

void vpPylonGrabberUsb::acquire(vpImage<vpRGBa> &I)
{
  open();

  Pylon::CGrabResultPtr grabResult;
  // Retrieve an image
  if (!m_camera.RetrieveResult(2000, grabResult)) {
    throw(vpException(vpException::fatalError, "Cannot retrieve image from camera with serial %s",
                      getCameraSerial(m_index).c_str()));
  }

  if (grabResult->GrabSucceeded()) {
    height = grabResult->GetHeight();
    width = grabResult->GetWidth();
    I.resize(height, width);

    Pylon::CImageFormatConverter imageConvert;
    imageConvert.OutputPixelFormat = Pylon::PixelType_BGRA8packed;
    imageConvert.OutputPaddingX = 0;
    // Create a converted image
    Pylon::CPylonImage destImage;
    imageConvert.Convert(destImage, (Pylon::IImage &)grabResult);
    Pylon::SBGRA8Pixel *pixel = (Pylon::SBGRA8Pixel *)destImage.GetBuffer();
    for (unsigned int i = 0; i < height; i++) {
      for (unsigned int j = 0; j < width; j++) {
        unsigned int p_index = i * width + j;
        I[i][j].R = pixel[p_index].R;
        I[i][j].G = pixel[p_index].G;
        I[i][j].B = pixel[p_index].B;
        I[i][j].A = pixel[p_index].A;
      }
    }
  }
}

void vpPylonGrabberUsb::open(vpImage<unsigned char> &I)
{
  open();
  acquire(I);
}

void vpPylonGrabberUsb::open(vpImage<vpRGBa> &I)
{
  open();
  acquire(I);
}

void vpPylonGrabberUsb::open()
{
  connect();
  startCapture();
}

bool vpPylonGrabberUsb::selectUserSet(UserSetName user_set)
{
  connect();

  switch (user_set) {
  case USERSET_DEFAULT:
    m_camera.UserSetSelector.SetValue(Basler_UsbCameraParams::UserSetSelector_Default);
    return true;
    break;
  case USERSET_USERSET1:
    m_camera.UserSetSelector.SetValue(Basler_UsbCameraParams::UserSetSelector_UserSet1);
    return true;
    break;
  case USERSET_USERSET2:
    m_camera.UserSetSelector.SetValue(Basler_UsbCameraParams::UserSetSelector_UserSet2);
    return true;
    break;
  case USERSET_USERSET3:
    m_camera.UserSetSelector.SetValue(Basler_UsbCameraParams::UserSetSelector_UserSet3);
    return true;
    break;
  default:
    return false;
  }
}

vpPylonGrabber &vpPylonGrabberUsb::operator>>(vpImage<unsigned char> &I)
{
  acquire(I);
  return *this;
}

vpPylonGrabber &vpPylonGrabberUsb::operator>>(vpImage<vpRGBa> &I)
{
  acquire(I);
  return *this;
}

#else
// Work arround to avoid warning:
// libvisp_pylon.a(vpPylonGrabberUsb.cpp.o) has no symbols
void dummy_vpPylonGrabberUsb(){};
#endif // #ifdef VISP_HAVE_PYLON