1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122
|
/*=========================================================================
Program: Visualization Toolkit
Module: vtkStructuredGridWriter.cxx
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
All rights reserved.
See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notice for more information.
=========================================================================*/
#include "vtkStructuredGridWriter.h"
#include "vtkInformation.h"
#include "vtkObjectFactory.h"
#include "vtkStructuredGrid.h"
#include "vtkUnsignedCharArray.h"
#if !defined(_WIN32) || defined(__CYGWIN__)
#include <unistd.h> /* unlink */
#else
#include <io.h> /* unlink */
#endif
VTK_ABI_NAMESPACE_BEGIN
vtkStandardNewMacro(vtkStructuredGridWriter);
void vtkStructuredGridWriter::WriteData()
{
ostream* fp;
vtkStructuredGrid* input = vtkStructuredGrid::SafeDownCast(this->GetInput());
int dim[3];
vtkDebugMacro(<< "Writing vtk structured grid...");
if (!(fp = this->OpenVTKFile()) || !this->WriteHeader(fp))
{
if (fp)
{
vtkErrorMacro("Ran out of disk space; deleting file: " << this->FileName);
this->CloseVTKFile(fp);
unlink(this->FileName);
}
return;
}
// Write structured grid specific stuff
//
*fp << "DATASET STRUCTURED_GRID\n";
// Write data owned by the dataset
if (!this->WriteDataSetData(fp, input))
{
vtkErrorMacro("Ran out of disk space; deleting file: " << this->FileName);
this->CloseVTKFile(fp);
unlink(this->FileName);
return;
}
if (this->WriteExtent)
{
int extent[6];
input->GetExtent(extent);
*fp << "EXTENT " << extent[0] << " " << extent[1] << " " << extent[2] << " " << extent[3] << " "
<< extent[4] << " " << extent[5] << "\n";
}
else
{
input->GetDimensions(dim);
*fp << "DIMENSIONS " << dim[0] << " " << dim[1] << " " << dim[2] << "\n";
}
if (!this->WritePoints(fp, input->GetPoints()))
{
vtkErrorMacro("Ran out of disk space; deleting file: " << this->FileName);
this->CloseVTKFile(fp);
unlink(this->FileName);
return;
}
if (!this->WriteCellData(fp, input))
{
vtkErrorMacro("Ran out of disk space; deleting file: " << this->FileName);
this->CloseVTKFile(fp);
unlink(this->FileName);
return;
}
if (!this->WritePointData(fp, input))
{
vtkErrorMacro("Ran out of disk space; deleting file: " << this->FileName);
this->CloseVTKFile(fp);
unlink(this->FileName);
return;
}
this->CloseVTKFile(fp);
}
int vtkStructuredGridWriter::FillInputPortInformation(int, vtkInformation* info)
{
info->Set(vtkAlgorithm::INPUT_REQUIRED_DATA_TYPE(), "vtkStructuredGrid");
return 1;
}
vtkStructuredGrid* vtkStructuredGridWriter::GetInput()
{
return vtkStructuredGrid::SafeDownCast(this->Superclass::GetInput());
}
vtkStructuredGrid* vtkStructuredGridWriter::GetInput(int port)
{
return vtkStructuredGrid::SafeDownCast(this->Superclass::GetInput(port));
}
void vtkStructuredGridWriter::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
}
VTK_ABI_NAMESPACE_END
|