You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
371 lines
9.5 KiB
371 lines
9.5 KiB
/*=========================================================================
|
|
|
|
Program: Visualization Toolkit
|
|
Module: $RCSfile: vtkStructuredGridReader.cxx,v $
|
|
|
|
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 "vtkStructuredGridReader.h"
|
|
|
|
#include "vtkFieldData.h"
|
|
#include "vtkInformation.h"
|
|
#include "vtkInformationVector.h"
|
|
#include "vtkObjectFactory.h"
|
|
#include "vtkStreamingDemandDrivenPipeline.h"
|
|
#include "vtkStructuredGrid.h"
|
|
#include "vtkUnsignedCharArray.h"
|
|
|
|
vtkCxxRevisionMacro(vtkStructuredGridReader, "$Revision: 1.60 $");
|
|
vtkStandardNewMacro(vtkStructuredGridReader);
|
|
|
|
vtkStructuredGridReader::vtkStructuredGridReader()
|
|
{
|
|
vtkStructuredGrid *output = vtkStructuredGrid::New();
|
|
this->SetOutput(output);
|
|
// Releasing data for pipeline parallism.
|
|
// Filters will know it is empty.
|
|
output->ReleaseData();
|
|
output->Delete();
|
|
}
|
|
|
|
vtkStructuredGridReader::~vtkStructuredGridReader()
|
|
{
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
vtkStructuredGrid* vtkStructuredGridReader::GetOutput()
|
|
{
|
|
return this->GetOutput(0);
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
vtkStructuredGrid* vtkStructuredGridReader::GetOutput(int idx)
|
|
{
|
|
return vtkStructuredGrid::SafeDownCast(this->GetOutputDataObject(idx));
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
void vtkStructuredGridReader::SetOutput(vtkStructuredGrid *output)
|
|
{
|
|
this->GetExecutive()->SetOutputData(0, output);
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// We just need to read the dimensions
|
|
int vtkStructuredGridReader::RequestInformation(
|
|
vtkInformation *,
|
|
vtkInformationVector **,
|
|
vtkInformationVector *outputVector)
|
|
{
|
|
vtkInformation *outInfo = outputVector->GetInformationObject(0);
|
|
return this->ReadMetaData(outInfo);
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
int vtkStructuredGridReader::ReadMetaData(vtkInformation *outInfo)
|
|
{
|
|
char line[256];
|
|
int done=0;
|
|
|
|
if (!this->OpenVTKFile() || !this->ReadHeader())
|
|
{
|
|
return 1;
|
|
}
|
|
|
|
// Read structured grid specific stuff
|
|
//
|
|
if (!this->ReadString(line))
|
|
{
|
|
vtkErrorMacro(<<"Data file ends prematurely!");
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
|
|
if ( !strncmp(this->LowerCase(line),"dataset",(unsigned long)7) )
|
|
{
|
|
// Make sure we're reading right type of geometry
|
|
//
|
|
if (!this->ReadString(line))
|
|
{
|
|
vtkErrorMacro(<<"Data file ends prematurely!");
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
|
|
if ( strncmp(this->LowerCase(line),"structured_grid",15) )
|
|
{
|
|
vtkErrorMacro(<< "Cannot read dataset type: " << line);
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
|
|
// Read keyword and dimensions
|
|
//
|
|
while (!done)
|
|
{
|
|
if (!this->ReadString(line))
|
|
{
|
|
break;
|
|
}
|
|
|
|
// Have to read field data because it may be binary.
|
|
if (! strncmp(this->LowerCase(line), "field", 5))
|
|
{
|
|
vtkFieldData* fd = this->ReadFieldData();
|
|
fd->Delete();
|
|
}
|
|
|
|
if ( ! strncmp(this->LowerCase(line),"dimensions",10) )
|
|
{
|
|
int ext[6];
|
|
if (!(this->Read(ext+1) &&
|
|
this->Read(ext+3) &&
|
|
this->Read(ext+5)))
|
|
{
|
|
vtkErrorMacro(<<"Error reading dimensions!");
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
// read dimensions, change to extent;
|
|
ext[0] = ext[2] = ext[4] = 0;
|
|
--ext[1];
|
|
--ext[3];
|
|
--ext[5];
|
|
outInfo->Set(vtkStreamingDemandDrivenPipeline::WHOLE_EXTENT(),
|
|
ext, 6);
|
|
// That is all we wanted !!!!!!!!!!!!!!!
|
|
this->CloseVTKFile();
|
|
return 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
vtkErrorMacro("Could not read dimensions");
|
|
this->CloseVTKFile ();
|
|
|
|
return 1;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
int vtkStructuredGridReader::RequestData(
|
|
vtkInformation *,
|
|
vtkInformationVector **,
|
|
vtkInformationVector *outputVector)
|
|
{
|
|
vtkInformation *outInfo = outputVector->GetInformationObject(0);
|
|
int numPts=0, npts=0, numCells=0, ncells;
|
|
char line[256];
|
|
int dimsRead=0;
|
|
int done=0;
|
|
vtkStructuredGrid *output = vtkStructuredGrid::SafeDownCast(
|
|
outInfo->Get(vtkDataObject::DATA_OBJECT()));
|
|
|
|
vtkDebugMacro(<<"Reading vtk structured grid file...");
|
|
|
|
if (!this->OpenVTKFile() || !this->ReadHeader())
|
|
{
|
|
return 1;
|
|
}
|
|
|
|
// Read structured grid specific stuff
|
|
//
|
|
if (!this->ReadString(line))
|
|
{
|
|
vtkErrorMacro(<<"Data file ends prematurely!");
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
|
|
if ( !strncmp(this->LowerCase(line),"dataset",(unsigned long)7) )
|
|
{
|
|
// Make sure we're reading right type of geometry
|
|
//
|
|
if (!this->ReadString(line))
|
|
{
|
|
vtkErrorMacro(<<"Data file ends prematurely!");
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
|
|
if ( strncmp(this->LowerCase(line),"structured_grid",15) )
|
|
{
|
|
vtkErrorMacro(<< "Cannot read dataset type: " << line);
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
|
|
// Read keyword and number of points
|
|
//
|
|
while (!done)
|
|
{
|
|
if (!this->ReadString(line))
|
|
{
|
|
break;
|
|
}
|
|
|
|
if (! strncmp(this->LowerCase(line), "field", 5))
|
|
{
|
|
vtkFieldData* fd = this->ReadFieldData();
|
|
output->SetFieldData(fd);
|
|
fd->Delete(); // ?
|
|
}
|
|
else if ( ! strncmp(line, "dimensions",10) )
|
|
{
|
|
int dim[3];
|
|
if (!(this->Read(dim) &&
|
|
this->Read(dim+1) &&
|
|
this->Read(dim+2)))
|
|
{
|
|
vtkErrorMacro(<<"Error reading dimensions!");
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
|
|
numPts = dim[0] * dim[1] * dim[2];
|
|
output->SetDimensions(dim);
|
|
numCells = output->GetNumberOfCells();
|
|
dimsRead = 1;
|
|
}
|
|
|
|
else if ( ! strncmp(line,"blanking",8) )
|
|
{
|
|
if (!this->Read(&npts))
|
|
{
|
|
vtkErrorMacro(<<"Error reading blanking!");
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
|
|
if (!this->ReadString(line))
|
|
{
|
|
vtkErrorMacro(<<"Cannot read blank type!" );
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
|
|
vtkUnsignedCharArray *data = vtkUnsignedCharArray::SafeDownCast(
|
|
this->ReadArray(line, numPts, 1));
|
|
|
|
if ( data != NULL )
|
|
{
|
|
output->SetPointVisibilityArray(data);
|
|
data->Delete();
|
|
}
|
|
}
|
|
|
|
else if ( ! strncmp(line,"points",6) )
|
|
{
|
|
if (!this->Read(&npts))
|
|
{
|
|
vtkErrorMacro(<<"Error reading points!");
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
|
|
this->ReadPoints(output, npts);
|
|
}
|
|
|
|
else if ( ! strncmp(line, "cell_data", 9) )
|
|
{
|
|
if (!this->Read(&ncells))
|
|
{
|
|
vtkErrorMacro(<<"Cannot read cell data!");
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
|
|
if ( ncells != numCells )
|
|
{
|
|
vtkErrorMacro(<<"Number of cells don't match!");
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
|
|
this->ReadCellData(output, ncells);
|
|
break; //out of this loop
|
|
}
|
|
|
|
else if ( ! strncmp(line, "point_data", 10) )
|
|
{
|
|
if (!this->Read(&numPts))
|
|
{
|
|
vtkErrorMacro(<<"Cannot read point data!");
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
|
|
if ( npts != numPts )
|
|
{
|
|
vtkErrorMacro(<<"Number of points don't match!");
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
|
|
this->ReadPointData(output, npts);
|
|
break; //out of this loop
|
|
}
|
|
|
|
else
|
|
{
|
|
vtkErrorMacro(<< "Unrecognized keyword: " << line);
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
if ( !dimsRead ) vtkWarningMacro(<<"No dimensions read.");
|
|
if ( !output->GetPoints() ) vtkWarningMacro(<<"No points read.");
|
|
}
|
|
|
|
else if ( !strncmp(line, "cell_data", 9) )
|
|
{
|
|
vtkWarningMacro(<<"No geometry defined in data file!");
|
|
if (!this->Read(&ncells))
|
|
{
|
|
vtkErrorMacro(<<"Cannot read cell data!");
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
this->ReadCellData(output, ncells);
|
|
}
|
|
|
|
else if ( !strncmp(line, "point_data", 10) )
|
|
{
|
|
vtkWarningMacro(<<"No geometry defined in data file!");
|
|
if (!this->Read(&npts))
|
|
{
|
|
vtkErrorMacro(<<"Cannot read point data!");
|
|
this->CloseVTKFile ();
|
|
return 1;
|
|
}
|
|
this->ReadPointData(output, npts);
|
|
}
|
|
|
|
else
|
|
{
|
|
vtkErrorMacro(<< "Unrecognized keyword: " << line);
|
|
}
|
|
this->CloseVTKFile ();
|
|
|
|
return 1;
|
|
}
|
|
|
|
//----------------------------------------------------------------------------
|
|
int vtkStructuredGridReader::FillOutputPortInformation(int,
|
|
vtkInformation* info)
|
|
{
|
|
info->Set(vtkDataObject::DATA_TYPE_NAME(), "vtkStructuredGrid");
|
|
return 1;
|
|
}
|
|
|
|
void vtkStructuredGridReader::PrintSelf(ostream& os, vtkIndent indent)
|
|
{
|
|
this->Superclass::PrintSelf(os,indent);
|
|
}
|
|
|