pull/2173/head
Anatoly Baksheev
11 years ago
parent
462d516743
commit
f37c31742a
12 changed files with 286 additions and 97 deletions
@ -0,0 +1,140 @@ |
||||
/*M///////////////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
|
||||
//
|
||||
// By downloading, copying, installing or using the software you agree to this license.
|
||||
// If you do not agree to this license, do not download, install,
|
||||
// copy or use the software.
|
||||
//
|
||||
//
|
||||
// License Agreement
|
||||
// For Open Source Computer Vision Library
|
||||
//
|
||||
// Copyright (C) 2013, OpenCV Foundation, all rights reserved.
|
||||
// Third party copyrights are property of their respective owners.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without modification,
|
||||
// are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistribution's of source code must retain the above copyright notice,
|
||||
// this list of conditions and the following disclaimer.
|
||||
//
|
||||
// * Redistribution's in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
//
|
||||
// * The name of the copyright holders may not be used to endorse or promote products
|
||||
// derived from this software without specific prior written permission.
|
||||
//
|
||||
// This software is provided by the copyright holders and contributors "as is" and
|
||||
// any express or implied warranties, including, but not limited to, the implied
|
||||
// warranties of merchantability and fitness for a particular purpose are disclaimed.
|
||||
// In no event shall the Intel Corporation or contributors be liable for any direct,
|
||||
// indirect, incidental, special, exemplary, or consequential damages
|
||||
// (including, but not limited to, procurement of substitute goods or services;
|
||||
// loss of use, data, or profits; or business interruption) however caused
|
||||
// and on any theory of liability, whether in contract, strict liability,
|
||||
// or tort (including negligence or otherwise) arising in any way out of
|
||||
// the use of this software, even if advised of the possibility of such damage.
|
||||
//
|
||||
// Authors:
|
||||
// * Anatoly Baksheev, Itseez Inc. myname.mysurname <> mycompany.com
|
||||
//
|
||||
//M*/
|
||||
|
||||
#include "precomp.hpp" |
||||
|
||||
|
||||
#include "vtkImageData.h" |
||||
#include "vtkImageProgressIterator.h" |
||||
#include "vtkMath.h" |
||||
#include "vtkInformation.h" |
||||
#include "vtkInformationVector.h" |
||||
#include "vtkObjectFactory.h" |
||||
#include "vtkStreamingDemandDrivenPipeline.h" |
||||
|
||||
namespace cv { namespace viz |
||||
{ |
||||
vtkStandardNewMacro(vtkImageMatSource); |
||||
}} |
||||
|
||||
cv::viz::vtkImageMatSource::vtkImageMatSource() |
||||
{ |
||||
this->SetNumberOfInputPorts(0); |
||||
this->ImageData = vtkImageData::New(); |
||||
} |
||||
|
||||
int cv::viz::vtkImageMatSource::RequestInformation(vtkInformation *, vtkInformationVector**, vtkInformationVector *outputVector) |
||||
{ |
||||
vtkInformation* outInfo = outputVector->GetInformationObject(0); |
||||
|
||||
outInfo->Set(vtkStreamingDemandDrivenPipeline::WHOLE_EXTENT(), this->ImageData->GetExtent(), 6); |
||||
outInfo->Set(vtkDataObject::SPACING(), 1.0, 1.0, 1.0); |
||||
outInfo->Set(vtkDataObject::ORIGIN(), 0.0, 0.0, 0.0); |
||||
|
||||
vtkDataObject::SetPointDataActiveScalarInfo(outInfo, this->ImageData->GetScalarType(), this->ImageData->GetNumberOfScalarComponents()); |
||||
return 1; |
||||
} |
||||
|
||||
int cv::viz::vtkImageMatSource::RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector *outputVector) |
||||
{ |
||||
vtkInformation *outInfo = outputVector->GetInformationObject(0); |
||||
|
||||
vtkImageData *output = vtkImageData::SafeDownCast(outInfo->Get(vtkDataObject::DATA_OBJECT()) ); |
||||
output->ShallowCopy(this->ImageData); |
||||
return 1; |
||||
} |
||||
|
||||
void cv::viz::vtkImageMatSource::SetImage(InputArray _image) |
||||
{ |
||||
CV_Assert(_image.depth() == CV_8U && _image.channels() == 1 || _image.channels() == 3 || _image.channels() == 4); |
||||
|
||||
Mat image = _image.getMat(); |
||||
|
||||
this->ImageData->SetDimensions(image.cols, image.rows, 1); |
||||
#if VTK_MAJOR_VERSION <= 5 |
||||
this->ImageData->SetNumberOfScalarComponents(std::min(3, image.channels())); |
||||
this->ImageData->SetScalarTypeToUnsignedChar(); |
||||
this->ImageData->AllocateScalars(); |
||||
#else |
||||
this->ImageData->AllocateScalars(VTK_UNSIGNED_CHAR, std::min(3, image.channels())); |
||||
#endif |
||||
|
||||
switch(image.channels()) |
||||
{ |
||||
case 1: copyGrayImage(image, this->ImageData); |
||||
case 3: copyRGBImage (image, this->ImageData); |
||||
case 4: copyRGBAImage(image, this->ImageData); |
||||
} |
||||
this->ImageData->Modified(); |
||||
} |
||||
|
||||
void cv::viz::vtkImageMatSource::copyGrayImage(const Mat &source, vtkSmartPointer<vtkImageData> output) |
||||
{ |
||||
for (int y = 0; y < source.rows; ++y) |
||||
{ |
||||
const unsigned char *srow = source.ptr<unsigned char>(y); |
||||
for (int x = 0; x < source.cols; ++x) |
||||
*reinterpret_cast<unsigned char*>(output->GetScalarPointer(x,y,0)) = *srow++; |
||||
} |
||||
} |
||||
|
||||
void cv::viz::vtkImageMatSource::copyRGBImage(const Mat &source, vtkSmartPointer<vtkImageData> output) |
||||
{ |
||||
for (int y = 0; y < source.rows; ++y) |
||||
{ |
||||
const unsigned char *srow = source.ptr<unsigned char>(y); |
||||
for (int x = 0; x < source.cols; ++x, srow += source.channels()) |
||||
*reinterpret_cast<Vec3b*>(output->GetScalarPointer(x,y,0)) = Vec3b(srow[2], srow[1], srow[0]); |
||||
} |
||||
} |
||||
|
||||
void cv::viz::vtkImageMatSource::copyRGBAImage(const Mat &source, vtkSmartPointer<vtkImageData> output) |
||||
{ |
||||
for (int y = 0; y < source.rows; ++y) |
||||
{ |
||||
const unsigned char *srow = source.ptr<unsigned char>(y); |
||||
for (int x = 0; x < source.cols; ++x, srow += source.channels()) |
||||
*reinterpret_cast<Vec4b*>(output->GetScalarPointer(x,y,0)) = Vec4b(srow[2], srow[1], srow[0], srow[3]); |
||||
} |
||||
} |
||||
@ -0,0 +1,84 @@ |
||||
/*M///////////////////////////////////////////////////////////////////////////////////////
|
||||
//
|
||||
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
|
||||
//
|
||||
// By downloading, copying, installing or using the software you agree to this license.
|
||||
// If you do not agree to this license, do not download, install,
|
||||
// copy or use the software.
|
||||
//
|
||||
//
|
||||
// License Agreement
|
||||
// For Open Source Computer Vision Library
|
||||
//
|
||||
// Copyright (C) 2013, OpenCV Foundation, all rights reserved.
|
||||
// Third party copyrights are property of their respective owners.
|
||||
//
|
||||
// Redistribution and use in source and binary forms, with or without modification,
|
||||
// are permitted provided that the following conditions are met:
|
||||
//
|
||||
// * Redistribution's of source code must retain the above copyright notice,
|
||||
// this list of conditions and the following disclaimer.
|
||||
//
|
||||
// * Redistribution's in binary form must reproduce the above copyright notice,
|
||||
// this list of conditions and the following disclaimer in the documentation
|
||||
// and/or other materials provided with the distribution.
|
||||
//
|
||||
// * The name of the copyright holders may not be used to endorse or promote products
|
||||
// derived from this software without specific prior written permission.
|
||||
//
|
||||
// This software is provided by the copyright holders and contributors "as is" and
|
||||
// any express or implied warranties, including, but not limited to, the implied
|
||||
// warranties of merchantability and fitness for a particular purpose are disclaimed.
|
||||
// In no event shall the Intel Corporation or contributors be liable for any direct,
|
||||
// indirect, incidental, special, exemplary, or consequential damages
|
||||
// (including, but not limited to, procurement of substitute goods or services;
|
||||
// loss of use, data, or profits; or business interruption) however caused
|
||||
// and on any theory of liability, whether in contract, strict liability,
|
||||
// or tort (including negligence or otherwise) arising in any way out of
|
||||
// the use of this software, even if advised of the possibility of such damage.
|
||||
//
|
||||
// Authors:
|
||||
// * Anatoly Baksheev, Itseez Inc. myname.mysurname <> mycompany.com
|
||||
//
|
||||
//M*/
|
||||
|
||||
#include "precomp.hpp" |
||||
|
||||
#ifndef __vtkImageMatSource_h |
||||
#define __vtkImageMatSource_h |
||||
|
||||
namespace cv |
||||
{ |
||||
namespace viz |
||||
{ |
||||
class vtkImageMatSource : public vtkImageAlgorithm |
||||
{ |
||||
public: |
||||
static vtkImageMatSource *New(); |
||||
vtkTypeMacro(vtkImageMatSource,vtkImageAlgorithm); |
||||
|
||||
void SetImage(InputArray image); |
||||
|
||||
protected: |
||||
vtkImageMatSource(); |
||||
~vtkImageMatSource() {} |
||||
|
||||
vtkSmartPointer<vtkImageData> ImageData; |
||||
|
||||
int RequestInformation(vtkInformation*, vtkInformationVector**, vtkInformationVector*); |
||||
int RequestData (vtkInformation*, vtkInformationVector**, vtkInformationVector*); |
||||
private: |
||||
vtkImageMatSource(const vtkImageMatSource&); // Not implemented.
|
||||
void operator=(const vtkImageMatSource&); // Not implemented.
|
||||
|
||||
static void copyGrayImage(const Mat &source, vtkSmartPointer<vtkImageData> output); |
||||
static void copyRGBImage (const Mat &source, vtkSmartPointer<vtkImageData> output); |
||||
static void copyRGBAImage(const Mat &source, vtkSmartPointer<vtkImageData> output); |
||||
}; |
||||
} |
||||
} |
||||
|
||||
|
||||
#endif |
||||
|
||||
|
||||
Loading…
Reference in new issue