Anatoly Baksheev
11 years ago
parent
80ab6e889f
commit
84d6fe397d
13 changed files with 578 additions and 373 deletions
@ -1,168 +0,0 @@ |
||||
/*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" |
||||
|
||||
namespace cv { namespace viz |
||||
{ |
||||
vtkStandardNewMacro(vtkCloudColorMatSource); |
||||
|
||||
struct IsNotNan |
||||
{ |
||||
template<typename _Tp> |
||||
bool operator()(const _Tp* data) const |
||||
{ |
||||
return !isNan(data[0]) && !isNan(data[1]) && !isNan(data[2]); |
||||
} |
||||
}; |
||||
}} |
||||
|
||||
cv::viz::vtkCloudColorMatSource::vtkCloudColorMatSource() { SetNumberOfInputPorts(0); } |
||||
cv::viz::vtkCloudColorMatSource::~vtkCloudColorMatSource() {} |
||||
|
||||
void cv::viz::vtkCloudColorMatSource::SetCloud(const Mat& cloud) |
||||
{ |
||||
CV_Assert(cloud.depth() == CV_32F || cloud.depth() == CV_64F); |
||||
CV_Assert(cloud.channels() == 3 || cloud.channels() == 4); |
||||
|
||||
int total = cloud.depth() == CV_32F ? filterNanCopy<float >(cloud, VTK_FLOAT) |
||||
: filterNanCopy<double>(cloud, VTK_DOUBLE); |
||||
|
||||
vertices = vtkSmartPointer<vtkCellArray>::New(); |
||||
vertices->Allocate(vertices->EstimateSize(1, total)); |
||||
vertices->InsertNextCell(total); |
||||
for(int i = 0; i < total; ++i) |
||||
vertices->InsertCellPoint(i); |
||||
} |
||||
|
||||
void cv::viz::vtkCloudColorMatSource::SetColors(const Mat &colors, const Mat &cloud_mask) |
||||
{ |
||||
CV_Assert(colors.depth() == CV_8U && colors.channels() <= 4 && colors.channels() != 2); |
||||
CV_Assert(cloud_mask.depth() == CV_32F || cloud_mask.depth() == CV_64F); |
||||
CV_Assert(colors.size() == cloud_mask.size()); |
||||
|
||||
if (cloud_mask.depth() == CV_32F) |
||||
filterNanCopy<float, IsNotNan>(colors, cloud_mask); |
||||
else if (cloud_mask.depth() == CV_64F) |
||||
filterNanCopy<double, IsNotNan>(colors, cloud_mask); |
||||
} |
||||
|
||||
int cv::viz::vtkCloudColorMatSource::RequestData(vtkInformation *vtkNotUsed(request), vtkInformationVector **vtkNotUsed(inputVector), vtkInformationVector *outputVector) |
||||
{ |
||||
vtkInformation *outInfo = outputVector->GetInformationObject(0); |
||||
vtkPolyData *output = vtkPolyData::SafeDownCast(outInfo->Get(vtkDataObject::DATA_OBJECT())); |
||||
|
||||
output->SetPoints(points); |
||||
output->SetVerts(vertices); |
||||
output->GetPointData()->SetScalars(scalars); |
||||
return 1; |
||||
} |
||||
|
||||
template<typename _Tp> |
||||
int cv::viz::vtkCloudColorMatSource::filterNanCopy(const Mat& source, int dataType) |
||||
{ |
||||
CV_DbgAssert(DataType<_Tp>::depth == source.depth()); |
||||
points = vtkSmartPointer<vtkPoints>::New(); |
||||
points->SetDataType(dataType); |
||||
points->Allocate(source.total()); |
||||
points->SetNumberOfPoints(source.total()); |
||||
|
||||
int cn = source.channels(); |
||||
int total = 0; |
||||
for (int y = 0; y < source.rows; ++y) |
||||
{ |
||||
const _Tp* srow = source.ptr<_Tp>(y); |
||||
const _Tp* send = srow + source.cols * cn; |
||||
|
||||
for (; srow != send; srow += cn) |
||||
if (!isNan(srow[0]) && !isNan(srow[1]) && !isNan(srow[2])) |
||||
points->SetPoint(total++, srow); |
||||
} |
||||
points->SetNumberOfPoints(total); |
||||
points->Squeeze(); |
||||
return total; |
||||
} |
||||
|
||||
template<typename _Msk, class _NanPred> |
||||
void cv::viz::vtkCloudColorMatSource::filterNanCopy(const Mat& colors, const Mat& mask) |
||||
{ |
||||
Mat buffer(colors.size(), CV_8UC3); |
||||
Vec3b* pos = buffer.ptr<Vec3b>(); |
||||
|
||||
int s_chs = colors.channels(); |
||||
int m_chs = mask.channels(); |
||||
|
||||
_NanPred pred; |
||||
|
||||
for (int y = 0; y < colors.rows; ++y) |
||||
{ |
||||
const unsigned char* srow = colors.ptr<unsigned char>(y); |
||||
const unsigned char* send = srow + colors.cols * colors.channels(); |
||||
const _Msk* mrow = mask.empty() ? 0 : mask.ptr<_Msk>(y); |
||||
|
||||
if (colors.channels() == 1) |
||||
{ |
||||
for (; srow != send; srow += s_chs, mrow += m_chs) |
||||
if (pred(mrow)) |
||||
*pos++ = Vec3b(srow[0], srow[0], srow[0]); |
||||
} |
||||
else |
||||
for (; srow != send; srow += s_chs, mrow += m_chs) |
||||
if (pred(mrow)) |
||||
*pos++ = Vec3b(srow[2], srow[1], srow[0]); |
||||
|
||||
} |
||||
|
||||
int total = pos - buffer.ptr<Vec3b>(); |
||||
Vec3b* array = new Vec3b[total]; |
||||
std::copy(buffer.ptr<Vec3b>(), pos, array); |
||||
|
||||
scalars = vtkSmartPointer<vtkUnsignedCharArray>::New(); |
||||
scalars->SetName("colors"); |
||||
scalars->SetNumberOfComponents(3); |
||||
scalars->SetNumberOfTuples(total); |
||||
scalars->SetArray(array->val, total * 3, 0); |
||||
} |
||||
|
||||
|
||||
@ -1,132 +0,0 @@ |
||||
/*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" |
||||
|
||||
namespace cv { namespace viz |
||||
{ |
||||
vtkStandardNewMacro(vtkColorMatSource); |
||||
|
||||
struct IsNotNan |
||||
{ |
||||
template<typename _Tp> |
||||
bool operator()(const _Tp* data) const |
||||
{ |
||||
return !isNan(data[0]) && !isNan(data[1]) && !isNan(data[2]); |
||||
} |
||||
}; |
||||
|
||||
struct AllOk |
||||
{ |
||||
template<typename _Tp> |
||||
bool operator()(const _Tp*) const { return true; } |
||||
}; |
||||
}} |
||||
|
||||
cv::viz::vtkColorMatSource::vtkColorMatSource() { SetNumberOfInputPorts(0); } |
||||
cv::viz::vtkColorMatSource::~vtkColorMatSource() {} |
||||
|
||||
void cv::viz::vtkColorMatSource::SetColors(const Mat &colors, const Mat& mask) |
||||
{ |
||||
CV_Assert(colors.depth() == CV_8U && colors.channels() <= 4 && colors.channels() != 2); |
||||
CV_Assert(mask.empty() || mask.depth() == CV_32F || mask.depth() == CV_64F); |
||||
|
||||
if (!mask.empty() && mask.depth() == CV_32F) |
||||
filterNanCopy<float, IsNotNan>(colors, mask); |
||||
else if (!mask.empty() && mask.depth() == CV_64F) |
||||
filterNanCopy<double, IsNotNan>(colors, mask); |
||||
else /* mask.empty() */ |
||||
filterNanCopy<double, AllOk>(colors, mask); |
||||
} |
||||
|
||||
int cv::viz::vtkColorMatSource::RequestData(vtkInformation *vtkNotUsed(request), vtkInformationVector **vtkNotUsed(inputVector), vtkInformationVector *outputVector) |
||||
{ |
||||
vtkInformation *outInfo = outputVector->GetInformationObject(0); |
||||
vtkPolyData *output = vtkPolyData::SafeDownCast(outInfo->Get(vtkDataObject::DATA_OBJECT())); |
||||
|
||||
output->GetPointData()->SetScalars(scalars); |
||||
|
||||
return 1; |
||||
} |
||||
|
||||
template<typename _Msk, class _NanPred> |
||||
void cv::viz::vtkColorMatSource::filterNanCopy(const Mat& colors, const Mat& cloud_mask) |
||||
{ |
||||
Mat buffer(colors.size(), CV_8UC3); |
||||
Vec3b* pos = buffer.ptr<Vec3b>(); |
||||
|
||||
int s_chs = colors.channels(); |
||||
int m_chs = cloud_mask.channels(); |
||||
|
||||
_NanPred pred; |
||||
|
||||
for (int y = 0; y < colors.rows; ++y) |
||||
{ |
||||
const unsigned char* srow = colors.ptr<unsigned char>(y); |
||||
const unsigned char* send = srow + colors.cols * colors.channels(); |
||||
const _Msk* mrow = cloud_mask.empty() ? 0 : cloud_mask.ptr<_Msk>(y); |
||||
|
||||
if (colors.channels() == 1) |
||||
{ |
||||
for (; srow != send; srow += s_chs, mrow += m_chs) |
||||
if (pred(mrow)) |
||||
*pos++ = Vec3b(srow[0], srow[0], srow[0]); |
||||
} |
||||
else |
||||
for (; srow != send; srow += s_chs, mrow += m_chs) |
||||
if (pred(mrow)) |
||||
*pos++ = Vec3b(srow[2], srow[1], srow[0]); |
||||
|
||||
} |
||||
|
||||
int total = pos - buffer.ptr<Vec3b>(); |
||||
Vec3b* array = new Vec3b[total]; |
||||
std::copy(buffer.ptr<Vec3b>(), pos, array); |
||||
|
||||
scalars = vtkSmartPointer<vtkUnsignedCharArray>::New(); |
||||
scalars->SetName("colors"); |
||||
scalars->SetNumberOfComponents(3); |
||||
scalars->SetNumberOfTuples(total); |
||||
scalars->SetArray(array->val, total * 3, 0); |
||||
} |
||||
@ -0,0 +1,241 @@ |
||||
/*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" |
||||
|
||||
namespace cv { namespace viz |
||||
{ |
||||
vtkStandardNewMacro(vtkOBJWriter); |
||||
}} |
||||
|
||||
cv::viz::vtkOBJWriter::vtkOBJWriter() |
||||
{ |
||||
std::ofstream fout; // only used to extract the default precision
|
||||
this->DecimalPrecision = fout.precision(); |
||||
this->FileName = NULL; |
||||
this->FileType = VTK_ASCII; |
||||
} |
||||
|
||||
cv::viz::vtkOBJWriter::~vtkOBJWriter(){} |
||||
|
||||
void cv::viz::vtkOBJWriter::WriteData() |
||||
{ |
||||
vtkPolyData *input = this->GetInput(); |
||||
if (!input) |
||||
return; |
||||
|
||||
std::ostream *outfilep = this->OpenVTKFile(); |
||||
if (!outfilep) |
||||
return; |
||||
|
||||
std::ostream& outfile = *outfilep; |
||||
|
||||
//write header
|
||||
outfile << "# wavefront obj file written by the visualization toolkit" << std::endl << std::endl; |
||||
outfile << "mtllib NONE" << std::endl << std::endl; |
||||
|
||||
// write out the points
|
||||
for (int i = 0; i < input->GetNumberOfPoints(); i++) |
||||
{ |
||||
Vec3d p; |
||||
input->GetPoint(i, p.val); |
||||
outfile << std::setprecision(this->DecimalPrecision) << "v " << p[0] << " " << p[1] << " " << p[2] << std::endl; |
||||
} |
||||
|
||||
const int idStart = 1; |
||||
|
||||
// write out the point data
|
||||
vtkSmartPointer<vtkDataArray> normals = input->GetPointData()->GetNormals(); |
||||
if(normals) |
||||
{ |
||||
for (int i = 0; i < normals->GetNumberOfTuples(); i++) |
||||
{ |
||||
Vec3d p; |
||||
normals->GetTuple(i, p.val); |
||||
outfile << std::setprecision(this->DecimalPrecision) << "vn " << p[0] << " " << p[1] << " " << p[2] << std::endl; |
||||
} |
||||
} |
||||
|
||||
vtkSmartPointer<vtkDataArray> tcoords = input->GetPointData()->GetTCoords(); |
||||
if (tcoords) |
||||
{ |
||||
for (int i = 0; i < tcoords->GetNumberOfTuples(); i++) |
||||
{ |
||||
Vec2d p; |
||||
tcoords->GetTuple(i, p.val); |
||||
outfile << std::setprecision(this->DecimalPrecision) << "vt " << p[0] << " " << p[1] << std::endl; |
||||
} |
||||
} |
||||
|
||||
// write out a group name and material
|
||||
outfile << std::endl << "g grp" << idStart << std::endl; |
||||
outfile << "usemtl mtlNONE" << std::endl; |
||||
|
||||
// write out verts if any
|
||||
if (input->GetNumberOfVerts() > 0) |
||||
{ |
||||
vtkIdType npts = 0, *index = 0; |
||||
vtkCellArray *cells = input->GetVerts(); |
||||
for (cells->InitTraversal(); cells->GetNextCell(npts, index); ) |
||||
{ |
||||
outfile << "p "; |
||||
for (int i = 0; i < npts; i++) |
||||
outfile << index[i] + idStart << " "; |
||||
outfile << std::endl; |
||||
} |
||||
} |
||||
|
||||
// write out lines if any
|
||||
if (input->GetNumberOfLines() > 0) |
||||
{ |
||||
vtkIdType npts = 0, *index = 0; |
||||
vtkCellArray *cells = input->GetLines(); |
||||
for (cells->InitTraversal(); cells->GetNextCell(npts, index); ) |
||||
{ |
||||
outfile << "l "; |
||||
if (tcoords) |
||||
{ |
||||
for (int i = 0; i < npts; i++) |
||||
outfile << index[i] + idStart << "/" << index[i] + idStart << " "; |
||||
} |
||||
else |
||||
for (int i = 0; i < npts; i++) |
||||
outfile << index[i] + idStart << " "; |
||||
|
||||
outfile << std::endl; |
||||
} |
||||
} |
||||
|
||||
// write out polys if any
|
||||
if (input->GetNumberOfPolys() > 0) |
||||
{ |
||||
vtkIdType npts = 0, *index = 0; |
||||
vtkCellArray *cells = input->GetPolys(); |
||||
for (cells->InitTraversal(); cells->GetNextCell(npts, index); ) |
||||
{ |
||||
outfile << "f "; |
||||
for (int i = 0; i < npts; i++) |
||||
{ |
||||
if (normals) |
||||
{ |
||||
if (tcoords) |
||||
outfile << index[i] + idStart << "/" << index[i] + idStart << "/" << index[i] + idStart << " "; |
||||
else |
||||
outfile << index[i] + idStart << "//" << index[i] + idStart << " "; |
||||
} |
||||
else |
||||
{ |
||||
if (tcoords) |
||||
outfile << index[i] + idStart << " " << index[i] + idStart << " "; |
||||
else |
||||
outfile << index[i] + idStart << " "; |
||||
} |
||||
} |
||||
outfile << std::endl; |
||||
} |
||||
} |
||||
|
||||
// write out tstrips if any
|
||||
if (input->GetNumberOfStrips() > 0) |
||||
{ |
||||
vtkIdType npts = 0, *index = 0; |
||||
vtkCellArray *cells = input->GetStrips(); |
||||
for (cells->InitTraversal(); cells->GetNextCell(npts, index); ) |
||||
{ |
||||
for (int i = 2, i1, i2; i < npts; ++i) |
||||
{ |
||||
if (i % 2) |
||||
{ |
||||
i1 = i - 1; |
||||
i2 = i - 2; |
||||
} |
||||
else |
||||
{ |
||||
i1 = i - 1; |
||||
i2 = i - 2; |
||||
} |
||||
|
||||
if(normals) |
||||
{ |
||||
if (tcoords) |
||||
{ |
||||
outfile << "f " << index[i1] + idStart << "/" << index[i1] + idStart << "/" << index[i1] + idStart << " " |
||||
<< index[i2]+ idStart << "/" << index[i2] + idStart << "/" << index[i2] + idStart << " " |
||||
<< index[i] + idStart << "/" << index[i] + idStart << "/" << index[i] + idStart << std::endl; |
||||
} |
||||
else |
||||
{ |
||||
outfile << "f " << index[i1] + idStart << "//" << index[i1] + idStart << " " << index[i2] + idStart |
||||
<< "//" << index[i2] + idStart << " " << index[i] + idStart << "//" << index[i] + idStart << std::endl; |
||||
} |
||||
} |
||||
else |
||||
{ |
||||
if (tcoords) |
||||
{ |
||||
outfile << "f " << index[i1] + idStart << "/" << index[i1] + idStart << " " << index[i2] + idStart |
||||
<< "/" << index[i2] + idStart << " " << index[i] + idStart << "/" << index[i] + idStart << std::endl; |
||||
} |
||||
else |
||||
outfile << "f " << index[i1] + idStart << " " << index[i2] + idStart << " " << index[i] + idStart << std::endl; |
||||
} |
||||
} /* for (int i = 2; i < npts; ++i) */ |
||||
} |
||||
} /* if (input->GetNumberOfStrips() > 0) */ |
||||
|
||||
this->CloseVTKFile(outfilep); |
||||
|
||||
// Delete the file if an error occurred
|
||||
if (this->ErrorCode == vtkErrorCode::OutOfDiskSpaceError) |
||||
{ |
||||
vtkErrorMacro("Ran out of disk space; deleting file: " << this->FileName); |
||||
unlink(this->FileName); |
||||
} |
||||
} |
||||
|
||||
void cv::viz::vtkOBJWriter::PrintSelf(ostream& os, vtkIndent indent) |
||||
{ |
||||
Superclass::PrintSelf(os, indent); |
||||
os << indent << "DecimalPrecision: " << DecimalPrecision << "\n"; |
||||
} |
||||
@ -0,0 +1,93 @@ |
||||
/*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" |
||||
|
||||
namespace cv { namespace viz |
||||
{ |
||||
vtkStandardNewMacro(vtkXYZWriter); |
||||
}} |
||||
|
||||
cv::viz::vtkXYZWriter::vtkXYZWriter() |
||||
{ |
||||
std::ofstream fout; // only used to extract the default precision
|
||||
this->DecimalPrecision = fout.precision(); |
||||
} |
||||
|
||||
void cv::viz::vtkXYZWriter::WriteData() |
||||
{ |
||||
vtkPolyData *input = this->GetInput(); |
||||
if (!input) |
||||
return; |
||||
|
||||
// OpenVTKFile() will report any errors that happen
|
||||
ostream *outfilep = this->OpenVTKFile(); |
||||
if (!outfilep) |
||||
return; |
||||
|
||||
ostream &outfile = *outfilep; |
||||
|
||||
for(vtkIdType i = 0; i < input->GetNumberOfPoints(); ++i) |
||||
{ |
||||
Vec3d p; |
||||
input->GetPoint(i, p.val); |
||||
outfile << std::setprecision(this->DecimalPrecision) << p[0] << " " << p[1] << " " << p[2] << std::endl; |
||||
} |
||||
|
||||
// Close the file
|
||||
this->CloseVTKFile(outfilep); |
||||
|
||||
// Delete the file if an error occurred
|
||||
if (this->ErrorCode == vtkErrorCode::OutOfDiskSpaceError) |
||||
{ |
||||
vtkErrorMacro("Ran out of disk space; deleting file: " << this->FileName); |
||||
unlink(this->FileName); |
||||
} |
||||
} |
||||
|
||||
void cv::viz::vtkXYZWriter::PrintSelf(ostream& os, vtkIndent indent) |
||||
{ |
||||
this->Superclass::PrintSelf(os,indent); |
||||
os << indent << "DecimalPrecision: " << this->DecimalPrecision << "\n"; |
||||
} |
||||
Loading…
Reference in new issue