Home | History | Annotate | Download | only in vtk 
      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 
     52 cv::viz::vtkXYZWriter::vtkXYZWriter()
     53 {
     54     std::ofstream fout; // only used to extract the default precision
     55     this->DecimalPrecision = fout.precision();
     56 }
     57 
     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 
    102 int cv::viz::vtkXYZWriter::FillInputPortInformation(int, vtkInformation *info)
    103 {
    104     info->Set(vtkAlgorithm::INPUT_REQUIRED_DATA_TYPE(), "vtkPolyData");
    105     return 1;
    106 }
    107 
    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 
    114 vtkPolyData* cv::viz::vtkXYZWriter::GetInput()
    115 {
    116     return vtkPolyData::SafeDownCast(this->Superclass::GetInput());
    117 }
    118 
    119 vtkPolyData* cv::viz::vtkXYZWriter::GetInput(int port)
    120 {
    121     return vtkPolyData::SafeDownCast(this->Superclass::GetInput(port));
    122 }
    123