1 /*M///////////////////////////////////////////////////////////////////////////////////////
2 //
3 // IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
4 //
5 // By downloading, copying, installing or using the software you agree to this license.
6 // If you do not agree to this license, do not download, install,
7 // copy or use the software.
8 //
9 //
10 // License Agreement
11 // For Open Source Computer Vision Library
12 //
13 // Copyright (C) 2013, OpenCV Foundation, all rights reserved.
14 // Third party copyrights are property of their respective owners.
15 //
16 // Redistribution and use in source and binary forms, with or without modification,
17 // are permitted provided that the following conditions are met:
18 //
19 // * Redistribution's of source code must retain the above copyright notice,
20 // this list of conditions and the following disclaimer.
21 //
22 // * Redistribution's in binary form must reproduce the above copyright notice,
23 // this list of conditions and the following disclaimer in the documentation
24 // and/or other materials provided with the distribution.
25 //
26 // * The name of the copyright holders may not be used to endorse or promote products
27 // derived from this software without specific prior written permission.
28 //
29 // This software is provided by the copyright holders and contributors "as is" and
30 // any express or implied warranties, including, but not limited to, the implied
31 // warranties of merchantability and fitness for a particular purpose are disclaimed.
32 // In no event shall the Intel Corporation or contributors be liable for any direct,
33 // indirect, incidental, special, exemplary, or consequential damages
34 // (including, but not limited to, procurement of substitute goods or services;
35 // loss of use, data, or profits; or business interruption) however caused
36 // and on any theory of liability, whether in contract, strict liability,
37 // or tort (including negligence or otherwise) arising in any way out of
38 // the use of this software, even if advised of the possibility of such damage.
39 //
40 // Authors:
41 // * Anatoly Baksheev, Itseez Inc. myname.mysurname <> mycompany.com
42 //
43 //M*/
44
45 #include "precomp.hpp"
46
47 namespace cv { namespace viz
48 {
49 vtkStandardNewMacro(vtkXYZWriter);
50 }}
51
vtkXYZWriter()52 cv::viz::vtkXYZWriter::vtkXYZWriter()
53 {
54 std::ofstream fout; // only used to extract the default precision
55 this->DecimalPrecision = fout.precision();
56 }
57
WriteData()58 void cv::viz::vtkXYZWriter::WriteData()
59 {
60 vtkPolyData *input = this->GetInput();
61 if (!input)
62 return;
63
64 if (!this->FileName )
65 {
66 vtkErrorMacro(<< "No FileName specified! Can't write!");
67 this->SetErrorCode(vtkErrorCode::NoFileNameError);
68 return;
69 }
70
71 vtkDebugMacro(<<"Opening vtk file for writing...");
72 ostream *outfilep = new ofstream(this->FileName, ios::out);
73 if (outfilep->fail())
74 {
75 vtkErrorMacro(<< "Unable to open file: "<< this->FileName);
76 this->SetErrorCode(vtkErrorCode::CannotOpenFileError);
77 delete outfilep;
78 return;
79 }
80
81 ostream &outfile = *outfilep;
82
83 for(vtkIdType i = 0; i < input->GetNumberOfPoints(); ++i)
84 {
85 Vec3d p;
86 input->GetPoint(i, p.val);
87 outfile << std::setprecision(this->DecimalPrecision) << p[0] << " " << p[1] << " " << p[2] << std::endl;
88 }
89
90 // Close the file
91 vtkDebugMacro(<<"Closing vtk file\n");
92 delete outfilep;
93
94 // Delete the file if an error occurred
95 if (this->ErrorCode == vtkErrorCode::OutOfDiskSpaceError)
96 {
97 vtkErrorMacro("Ran out of disk space; deleting file: " << this->FileName);
98 unlink(this->FileName);
99 }
100 }
101
FillInputPortInformation(int,vtkInformation * info)102 int cv::viz::vtkXYZWriter::FillInputPortInformation(int, vtkInformation *info)
103 {
104 info->Set(vtkAlgorithm::INPUT_REQUIRED_DATA_TYPE(), "vtkPolyData");
105 return 1;
106 }
107
PrintSelf(ostream & os,vtkIndent indent)108 void cv::viz::vtkXYZWriter::PrintSelf(ostream& os, vtkIndent indent)
109 {
110 this->Superclass::PrintSelf(os,indent);
111 os << indent << "DecimalPrecision: " << this->DecimalPrecision << "\n";
112 }
113
GetInput()114 vtkPolyData* cv::viz::vtkXYZWriter::GetInput()
115 {
116 return vtkPolyData::SafeDownCast(this->Superclass::GetInput());
117 }
118
GetInput(int port)119 vtkPolyData* cv::viz::vtkXYZWriter::GetInput(int port)
120 {
121 return vtkPolyData::SafeDownCast(this->Superclass::GetInput(port));
122 }
123