mitkCropOpenCVImageFilter.cpp

Go to the documentation of this file.

/*============================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.

Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.

============================================================================*/

#include "mitkCropOpenCVImageFilter.h"


namespace mitk {
CropOpenCVImageFilter::CropOpenCVImageFilter( )
  : m_NewCropRegionSet(false)
{
}

bool CropOpenCVImageFilter::OnFilterImage( cv::Mat& image )
{
  if (m_CropRegion.width == 0)
  {
    MITK_WARN("AbstractOpenCVImageFilter")("CropOpenCVImageFilter")
        << "Cropping cannot be done without setting a non-empty crop region first.";
    return false;
  }

  cv::Size imageSize = image.size();

  if (m_CropRegion.x >= imageSize.width || m_CropRegion.y >= imageSize.height)
  {
    MITK_WARN("AbstractOpenCVImageFilter")("CropOpenCVImageFilter")
        << "Cannot crop if top left corner of the roi is outside the image boundaries.";
    return false;
  }

  // We can try and correct too large boundaries (do this only once
  // after a new crop region was set.
  if (m_NewCropRegionSet)
  {
    m_NewCropRegionSet = false;

    if ( m_CropRegion.x + m_CropRegion.width > imageSize.width)
    {
      m_CropRegion.width = imageSize.width - m_CropRegion.x;
      MITK_WARN("AbstractOpenCVImageFilter")("CropOpenCVImageFilter")
          << "Changed too large roi in x direction to fit the image size.";
    }
    if ( m_CropRegion.y + m_CropRegion.height > imageSize.height)
    {
      m_CropRegion.height = imageSize.height - m_CropRegion.y;
      MITK_WARN("AbstractOpenCVImageFilter")("CropOpenCVImageFilter")
          << "Changed too large roi in y direction to fit the image size.";
    }
  }

  // crop image and copy cropped region into the input image
  cv::Mat buffer = image(m_CropRegion);
  buffer.copyTo(image);

  return true;
}

void CropOpenCVImageFilter::SetCropRegion( cv::Rect cropRegion )
{
  // First, let's do some basic checks to make sure rectangle is inside of actual image
  if (cropRegion.x < 0)
  {
    MITK_WARN("AbstractOpenCVImageFilter")("CropOpenCVImageFilter")
        << "Changed negative x value in roi to 0.";
    cropRegion.x = 0;
  }
  if (cropRegion.y < 0)
  {
    cropRegion.y = 0;
    MITK_WARN("AbstractOpenCVImageFilter")("CropOpenCVImageFilter")
        << "Changed negative y value in roi to 0.";
  }

  // Nothing to save, throw an exception
  if ( cropRegion.height < 0 || cropRegion.width < 0 )
  {
    mitkThrow() << "Invalid boundaries supplied to USImageVideoSource::SetRegionOfInterest()";
  }

  m_CropRegion = cropRegion;
}

void CropOpenCVImageFilter::SetCropRegion( int topLeftX, int topLeftY, int bottomRightX, int bottomRightY )
{
  this->SetCropRegion( cv::Rect(topLeftX, topLeftY, bottomRightX - topLeftX, bottomRightY - topLeftY) );
}

cv::Rect CropOpenCVImageFilter::GetCropRegion( )
{
  return m_CropRegion;
}

bool CropOpenCVImageFilter::GetIsCropRegionEmpty( )
{
  return m_CropRegion.width == 0;
}
} // namespace mitk