/*========================================================================= Program: Visualization Toolkit Module: vtkEnSightGoldBinaryReader.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 "vtkEnSightGoldBinaryReader.h" #include "vtkByteSwap.h" #include "vtkCellData.h" #include "vtkCharArray.h" #include "vtkFloatArray.h" #include "vtkMultiBlockDataSet.h" #include "vtkIdList.h" #include "vtkImageData.h" #include "vtkObjectFactory.h" #include "vtkPointData.h" #include "vtkPolyData.h" #include "vtkRectilinearGrid.h" #include "vtkStructuredGrid.h" #include "vtkUnstructuredGrid.h" #include #include #include #include #include #if defined(_WIN32) # define VTK_STAT_STRUCT struct _stat64 # define VTK_STAT_FUNC _stat64 #elif defined _DARWIN_FEATURE_64_BIT_INODE && !defined __FreeBSD__ // FreeBSD and OSX now deprecate stat64 # define VTK_STAT_STRUCT struct stat # define VTK_STAT_FUNC stat #else // here, we're relying on _FILE_OFFSET_BITS defined in vtkWin32Header.h to help // us on POSIX without resorting to using stat64. # define VTK_STAT_STRUCT struct stat64 # define VTK_STAT_FUNC stat64 #endif vtkStandardNewMacro(vtkEnSightGoldBinaryReader); class vtkEnSightGoldBinaryReader::FileOffsetMapInternal { typedef std::string MapKey; typedef std::map MapValue; public: typedef std::map::const_iterator const_iterator; typedef std::map::value_type value_type; std::map Map; }; // This is half the precision of an int. #define MAXIMUM_PART_ID 65536 //---------------------------------------------------------------------------- vtkEnSightGoldBinaryReader::vtkEnSightGoldBinaryReader() { this->FileOffsets = new vtkEnSightGoldBinaryReader::FileOffsetMapInternal; this->GoldIFile = nullptr; this->FileSize = 0; this->SizeOfInt = sizeof(int); this->Fortran = 0; this->NodeIdsListed = 0; this->ElementIdsListed = 0; } //---------------------------------------------------------------------------- vtkEnSightGoldBinaryReader::~vtkEnSightGoldBinaryReader() { delete this->FileOffsets; if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::OpenFile(const char* filename) { if (!filename) { vtkErrorMacro(<<"Missing filename."); return 0; } // Close file from any previous image if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } // Open the new file vtkDebugMacro(<< "Opening file " << filename); VTK_STAT_STRUCT fs; if ( !VTK_STAT_FUNC( filename, &fs) ) { // Find out how big the file is. this->FileSize = static_cast(fs.st_size); #ifdef _WIN32 this->GoldIFile = new ifstream(filename, ios::in | ios::binary); #else this->GoldIFile = new ifstream(filename, ios::in); #endif } else { vtkErrorMacro("stat failed."); return 0; } if (! this->GoldIFile || this->GoldIFile->fail()) { vtkErrorMacro(<< "Could not open file " << filename); return 0; } //we now need to check for Fortran and byte ordering //we need to look at the first 4 bytes of the file, and the 84-87 bytes //of the file to correctly determine what it is. If we only check the first //4 bytes we can get incorrect detection if it is a property file named "P" //we check the 84-87 bytes as that is the start of the next line on a fortran file char result[88]; this->GoldIFile->read(result, 88); if ( this->GoldIFile->eof() || this->GoldIFile->fail() ) { vtkErrorMacro(<GoldIFile->seekg (0, ios::beg); //reset the file to the start // if the first 4 bytes is the length, then this data is no doubt // a fortran data write!, copy the last 76 into the beginning char le_len[4] = {0x50, 0x00, 0x00, 0x00}; char be_len[4] = {0x00, 0x00, 0x00, 0x50}; // the fortran test here depends on the byte ordering. But if the user didn't // set any byte ordering then, we have to try both byte orderings. There was a // bug here which was resulting in binary-fortran-big-endian files being read // incorrectly on intel machines (BUG #10593). This dual-check avoids that // bug. bool le_isFortran = true; bool be_isFortran = true; for (int c=0; c<4; c++) { le_isFortran = le_isFortran && (result[c] == le_len[c]) && (result[c+84] == le_len[c]); be_isFortran = be_isFortran && (result[c] == be_len[c]) && (result[c+84] == be_len[c]); } switch (this->ByteOrder) { case FILE_BIG_ENDIAN: this->Fortran = be_isFortran; break; case FILE_LITTLE_ENDIAN: this->Fortran = le_isFortran; break; case FILE_UNKNOWN_ENDIAN: if (le_isFortran) { this->Fortran = true; this->ByteOrder = FILE_LITTLE_ENDIAN; } else if (be_isFortran) { this->Fortran = true; this->ByteOrder = FILE_BIG_ENDIAN; } else { this->Fortran = false; } break; } return 1; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::InitializeFile(const char* fileName) { char line[80], subLine[80]; // Initialize // if (!fileName) { vtkErrorMacro("A GeometryFileName must be specified in the case file."); return 0; } std::string sfilename; if (this->FilePath) { sfilename = this->FilePath; if (sfilename.at(sfilename.length()-1) != '/') { sfilename += "/"; } sfilename += fileName; vtkDebugMacro("full path to geometry file: " << sfilename.c_str()); } else { sfilename = fileName; } if (this->OpenFile(sfilename.c_str()) == 0) { vtkErrorMacro("Unable to open file: " << sfilename.c_str()); return 0; } line[0] = '\0'; subLine[0] = '\0'; if ( this->ReadLine( line ) == 0 ) { vtkErrorMacro( "Error with line reading upon file initialization" ); return 0; } if ( sscanf( line, " %*s %s", subLine ) != 1 ) { vtkErrorMacro( "Error with subline extraction upon file initialization" ); return 0; } if (strncmp(subLine, "Binary", 6) != 0 && strncmp(subLine, "binary", 6) != 0) { vtkErrorMacro("This is not a binary data set. Try " << "vtkEnSightGoldReader."); return 0; } return 1; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::ReadGeometryFile(const char* fileName, int timeStep, vtkMultiBlockDataSet *output) { char line[80], subLine[80], nameline[80]; int partId, realId; int lineRead, i; if (!this->InitializeFile(fileName)) { return 0; } //this will close the file, so we need to reinitialize it int numberOfTimeStepsInFile=this->CountTimeSteps(); if (!this->InitializeFile(fileName)) { return 0; } if (this->UseFileSets) { if (numberOfTimeStepsInFile>1) { this->AddFileIndexToCache(fileName); i = this->SeekToCachedTimeStep(fileName, timeStep-1); // start w/ the number of TS we skipped, not the one we are at // if we are not at the appropriate time step yet, we keep searching for (; i < timeStep - 1; i++) { if (!this->SkipTimeStep()) { return 0; } } } // use do-while here to initialize 'line' before 'strncmp' is appllied // Thanks go to Brancois for care of this issue do { this->ReadLine(line); } while ( strncmp(line, "BEGIN TIME STEP", 15) != 0 ); // found a time step -> cache it this->AddTimeStepToCache(fileName, timeStep-1, this->GoldIFile->tellg()); } // Skip the 2 description lines. this->ReadLine(line); this->ReadLine(line); // Read the node id and element id lines. this->ReadLine(line); sscanf(line, " %*s %*s %s", subLine); if (strncmp(subLine, "given", 5) == 0) { this->NodeIdsListed = 1; } else if (strncmp(subLine, "ignore", 6) == 0) { this->NodeIdsListed = 1; } else { this->NodeIdsListed = 0; } this->ReadLine(line); sscanf(line, " %*s %*s %s", subLine); if (strncmp(subLine, "given", 5) == 0) { this->ElementIdsListed = 1; } else if (strncmp(subLine, "ignore", 6) == 0) { this->ElementIdsListed = 1; } else { this->ElementIdsListed = 0; } lineRead = this->ReadLine(line); // "extents" or "part" if (strncmp(line, "extents", 7) == 0) { // Skipping the extents. this->GoldIFile->seekg(6*sizeof(float), ios::cur); lineRead = this->ReadLine(line); // "part" } while (lineRead > 0 && strncmp(line, "part", 4) == 0) { this->ReadPartId(&partId); partId--; // EnSight starts #ing at 1. if (partId < 0 || partId >= MAXIMUM_PART_ID) { vtkErrorMacro("Invalid part id; check that ByteOrder is set correctly."); return 0; } realId = this->InsertNewPartId(partId); // Increment the number of geoemtry parts such that the measured geomtry, // if any, can be properly combined into a vtkMultiBlockDataSet object. // --- fix to bug #7453 this->NumberOfGeometryParts ++; this->ReadLine(line); // part description line strncpy(nameline, line, 80); // 80 characters in line are allowed nameline[79] = '\0'; // Ensure nullptr character at end of part name char *name = strdup(nameline); // fix to bug #0008237 // The original "return 1" operation upon "strncmp(line, "interface", 9) == 0" // was removed here as 'interface' is NOT a keyword of an EnSight Gold file. this->ReadLine(line); if (strncmp(line, "block", 5) == 0) { if (sscanf(line, " %*s %s", subLine) == 1) { if (strncmp(subLine, "rectilinear", 11) == 0) { // block rectilinear lineRead = this->CreateRectilinearGridOutput(realId, line, name, output); } else if (strncmp(subLine, "uniform", 7) == 0) { // block uniform lineRead = this->CreateImageDataOutput(realId, line, name, output); } else { // block iblanked lineRead = this->CreateStructuredGridOutput(realId, line, name, output); } } else { // block lineRead = this->CreateStructuredGridOutput(realId, line, name, output); } } else { lineRead = this->CreateUnstructuredGridOutput(realId, line, name, output); if (lineRead < 0) { free(name); if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } return 0; } } free(name); } if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } if (lineRead < 0) { return 0; } return 1; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::CountTimeSteps() { int count=0; while(1) { int result=this->SkipTimeStep(); if (result) { count++; } else { break; } } return count; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::SkipTimeStep() { char line[80], subLine[80]; int lineRead; line[0] = '\0'; while (strncmp(line, "BEGIN TIME STEP", 15) != 0) { if (!this->ReadLine(line)) { return 0; } } // Skip the 2 description lines. this->ReadLine(line); this->ReadLine(line); // Read the node id and element id lines. this->ReadLine(line); sscanf(line, " %*s %*s %s", subLine); if (strncmp(subLine, "given", 5) == 0 || strncmp(subLine, "ignore", 6) == 0) { this->NodeIdsListed = 1; } else { this->NodeIdsListed = 0; } this->ReadLine(line); sscanf(line, " %*s %*s %s", subLine); if (strncmp(subLine, "given", 5) == 0) { this->ElementIdsListed = 1; } else if (strncmp(subLine, "ignore", 6) == 0) { this->ElementIdsListed = 1; } else { this->ElementIdsListed = 0; } lineRead = this->ReadLine(line); // "extents" or "part" if (strncmp(line, "extents", 7) == 0) { // Skipping the extents. this->GoldIFile->seekg(6*sizeof(float), ios::cur); lineRead = this->ReadLine(line); // "part" } while (lineRead > 0 && strncmp(line, "part", 4) == 0) { int tmpInt; this->ReadPartId(&tmpInt); if (tmpInt < 0 || tmpInt > MAXIMUM_PART_ID) { vtkErrorMacro("Invalid part id; check that ByteOrder is set correctly."); return 0; } this->ReadLine(line); // part description line this->ReadLine(line); if (strncmp(line, "block", 5) == 0) { if (sscanf(line, " %*s %s", subLine) == 1) { if (strncmp(subLine, "rectilinear", 11) == 0) { // block rectilinear lineRead = this->SkipRectilinearGrid(line); } else if (strncmp(subLine, "uniform,", 7) == 0) { // block uniform lineRead = this->SkipImageData(line); } else { // block iblanked lineRead = this->SkipStructuredGrid(line); } } else { // block lineRead = this->SkipStructuredGrid(line); } } else { lineRead = this->SkipUnstructuredGrid(line); } } if (lineRead < 0) { if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } return 0; } return 1; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::SkipStructuredGrid(char line[256]) { char subLine[80]; int lineRead; int iblanked = 0; int dimensions[3]; int numPts; if (sscanf(line, " %*s %s", subLine) == 1) { if (strncmp(subLine, "iblanked", 8) == 0) { iblanked = 1; } } this->ReadIntArray(dimensions, 3); numPts = dimensions[0] * dimensions[1] * dimensions[2]; if (dimensions[0] < 0 || static_cast(dimensions[0] * this->SizeOfInt) > this->FileSize || dimensions[1] < 0 || static_cast(dimensions[1] * this->SizeOfInt) > this->FileSize || dimensions[2] < 0 || static_cast(dimensions[2] * this->SizeOfInt) > this->FileSize || numPts < 0 || static_cast(numPts*this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid dimensions read; check that ByteOrder is set correctly."); return -1; } // Skip xCoords, yCoords and zCoords. this->GoldIFile->seekg(sizeof(float)*numPts*3, ios::cur); if (iblanked) { // skip iblank array. this->GoldIFile->seekg(numPts*sizeof(int), ios::cur); } // reading next line to check for EOF lineRead = this->ReadLine(line); return lineRead; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::SkipUnstructuredGrid(char line[256]) { int lineRead = 1; int i; int numElements; int cellType; while(lineRead && strncmp(line, "part", 4) != 0) { if (strncmp(line, "coordinates", 11) == 0) { vtkDebugMacro("coordinates"); int numPts; this->ReadInt(&numPts); if (numPts < 0 || static_cast(numPts * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of points; check that ByteOrder is set correctly."); return -1; } vtkDebugMacro("num. points: " << numPts); if (this->NodeIdsListed) { // skip node ids. this->GoldIFile->seekg(sizeof(int)*numPts, ios::cur); } // Skip xCoords, yCoords and zCoords. this->GoldIFile->seekg(sizeof(float)*3*numPts, ios::cur); } else if (strncmp(line, "point", 5) == 0 || strncmp(line, "g_point", 7) == 0) { vtkDebugMacro("point"); this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of point cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { // skip element ids. this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } else if (strncmp(line, "bar2", 4) == 0 || strncmp(line, "g_bar2", 6) == 0) { vtkDebugMacro("bar2"); this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of bar2 cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*2*numElements, ios::cur); } else if (strncmp(line, "bar3", 4) == 0 || strncmp(line, "g_bar3", 6) == 0) { vtkDebugMacro("bar3"); this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of bar3 cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*3*numElements, ios::cur); } else if (strncmp(line, "nsided", 6) == 0 || strncmp(line, "g_nsided", 8) == 0) { vtkDebugMacro("nsided"); int *numNodesPerElement; int numNodes = 0; //cellType = vtkEnSightReader::NSIDED; this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of nsided cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } numNodesPerElement = new int[numElements]; this->ReadIntArray(numNodesPerElement, numElements); for (i = 0; i < numElements; i++) { numNodes += numNodesPerElement[i]; } // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*numNodes, ios::cur); delete [] numNodesPerElement; } else if (strncmp(line, "tria3", 5) == 0 || strncmp(line, "tria6", 5) == 0 || strncmp(line, "g_tria3", 7) == 0 || strncmp(line, "g_tria6", 7) == 0) { if (strncmp(line, "tria6", 5) == 0 || strncmp(line, "g_tria6", 7) == 0) { vtkDebugMacro("tria6"); cellType = vtkEnSightReader::TRIA6; } else { vtkDebugMacro("tria3"); cellType = vtkEnSightReader::TRIA3; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of triangle cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::TRIA6) { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*6*numElements, ios::cur); } else { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*3*numElements, ios::cur); } } else if (strncmp(line, "quad4", 5) == 0 || strncmp(line, "quad8", 5) == 0 || strncmp(line, "g_quad4", 7) == 0 || strncmp(line, "g_quad8", 7) == 0) { if (strncmp(line, "quad8", 5) == 0 || strncmp(line, "g_quad8", 7) == 0) { vtkDebugMacro("quad8"); cellType = vtkEnSightReader::QUAD8; } else { vtkDebugMacro("quad4"); cellType = vtkEnSightReader::QUAD4; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of quad cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::QUAD8) { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*8*numElements, ios::cur); } else { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*4*numElements, ios::cur); } } else if (strncmp(line, "nfaced", 6) == 0) { vtkDebugMacro("nfaced"); int *numFacesPerElement; int *numNodesPerFace; int numFaces = 0; int numNodes = 0; this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of nfaced cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } numFacesPerElement = new int[numElements]; this->ReadIntArray(numFacesPerElement, numElements); for (i = 0; i < numElements; i++) { numFaces += numFacesPerElement[i]; } delete [] numFacesPerElement; numNodesPerFace = new int[numFaces]; this->ReadIntArray(numNodesPerFace, numFaces); for (i = 0; i < numFaces; i++) { numNodes += numNodesPerFace[i]; } // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*numNodes, ios::cur); delete [] numNodesPerFace; } else if (strncmp(line, "tetra4", 6) == 0 || strncmp(line, "tetra10", 7) == 0 || strncmp(line, "g_tetra4", 8) == 0 || strncmp(line, "g_tetra10", 9) == 0) { if (strncmp(line, "tetra10", 7) == 0 || strncmp(line, "g_tetra10", 9) == 0) { vtkDebugMacro("tetra10"); cellType = vtkEnSightReader::TETRA10; } else { vtkDebugMacro("tetra4"); cellType = vtkEnSightReader::TETRA4; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of tetrahedral cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::TETRA10) { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*10*numElements, ios::cur); } else { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*4*numElements, ios::cur); } } else if (strncmp(line, "pyramid5", 8) == 0 || strncmp(line, "pyramid13", 9) == 0 || strncmp(line, "g_pyramid5", 10) == 0 || strncmp(line, "g_pyramid13", 11) == 0) { if (strncmp(line, "pyramid13", 9) == 0 || strncmp(line, "g_pyramid13", 11) == 0) { vtkDebugMacro("pyramid13"); cellType = vtkEnSightReader::PYRAMID13; } else { vtkDebugMacro("pyramid5"); cellType = vtkEnSightReader::PYRAMID5; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of pyramid cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::PYRAMID13) { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*13*numElements, ios::cur); } else { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*5*numElements, ios::cur); } } else if (strncmp(line, "hexa8", 5) == 0 || strncmp(line, "hexa20", 6) == 0 || strncmp(line, "g_hexa8", 7) == 0 || strncmp(line, "g_hexa20", 8) == 0) { if (strncmp(line, "hexa20", 6) == 0 || strncmp(line, "g_hexa20", 8) == 0) { vtkDebugMacro("hexa20"); cellType = vtkEnSightReader::HEXA20; } else { vtkDebugMacro("hexa8"); cellType = vtkEnSightReader::HEXA8; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of hexahedral cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::HEXA20) { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*20*numElements, ios::cur); } else { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*8*numElements, ios::cur); } } else if (strncmp(line, "penta6", 6) == 0 || strncmp(line, "penta15", 7) == 0 || strncmp(line, "g_penta6", 8) == 0 || strncmp(line, "g_penta15", 9) == 0) { if (strncmp(line, "penta15", 7) == 0 || strncmp(line, "g_penta15", 9) == 0) { vtkDebugMacro("penta15"); cellType = vtkEnSightReader::PENTA15; } else { vtkDebugMacro("penta6"); cellType = vtkEnSightReader::PENTA6; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of pentagonal cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::PENTA15) { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*15*numElements, ios::cur); } else { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*6*numElements, ios::cur); } } else if (strncmp(line, "END TIME STEP", 13) == 0) { return 1; } else { vtkErrorMacro("undefined geometry file line"); return -1; } lineRead = this->ReadLine(line); } return lineRead; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::SkipRectilinearGrid(char line[256]) { char subLine[80]; int lineRead; int iblanked = 0; int dimensions[3]; int numPts; if (sscanf(line, " %*s %*s %s", subLine) == 1) { if (strncmp(subLine, "iblanked", 8) == 0) { iblanked = 1; } } this->ReadIntArray(dimensions, 3); if (dimensions[0] < 0 || static_cast(dimensions[0] * this->SizeOfInt) > this->FileSize || dimensions[1] < 0 || static_cast(dimensions[1] * this->SizeOfInt) > this->FileSize || dimensions[2] < 0 || static_cast(dimensions[2] * this->SizeOfInt) > this->FileSize || (dimensions[0]+dimensions[1]+dimensions[2]) < 0 || static_cast((dimensions[0] + dimensions[1] + dimensions[2]) * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid dimensions read; check that BytetOrder is set correctly."); return -1; } numPts = dimensions[0] * dimensions[1] * dimensions[2]; // Skip xCoords this->GoldIFile->seekg(sizeof(float)*dimensions[0], ios::cur); // Skip yCoords this->GoldIFile->seekg(sizeof(float)*dimensions[1], ios::cur); // Skip zCoords this->GoldIFile->seekg(sizeof(float)*dimensions[2], ios::cur); if (iblanked) { vtkWarningMacro("VTK does not handle blanking for rectilinear grids."); this->GoldIFile->seekg(sizeof(int)*numPts, ios::cur); } // reading next line to check for EOF lineRead = this->ReadLine(line); return lineRead; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::SkipImageData(char line[256]) { char subLine[80]; int lineRead; int iblanked = 0; int dimensions[3]; float origin[3], delta[3]; int numPts; if (sscanf(line, " %*s %*s %s", subLine) == 1) { if (strncmp(subLine, "iblanked", 8) == 0) { iblanked = 1; } } this->ReadIntArray(dimensions, 3); this->ReadFloatArray(origin, 3); this->ReadFloatArray(delta, 3); if (iblanked) { vtkWarningMacro("VTK does not handle blanking for image data."); numPts = dimensions[0] * dimensions[1] * dimensions[2]; if (dimensions[0] < 0 || static_cast(dimensions[0] * this->SizeOfInt) > this->FileSize || dimensions[1] < 0 || static_cast(dimensions[1] * this->SizeOfInt) > this->FileSize || dimensions[2] < 0 || static_cast(dimensions[2] * this->SizeOfInt) > this->FileSize || numPts < 0 || static_cast(numPts * this->SizeOfInt) > this->FileSize) { return -1; } this->GoldIFile->seekg(sizeof(int)*numPts, ios::cur); } // reading next line to check for EOF lineRead = this->ReadLine(line); return lineRead; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::ReadMeasuredGeometryFile(const char* fileName, int timeStep, vtkMultiBlockDataSet *output) { char line[80], subLine[80]; vtkIdType i; int *pointIds; float *xCoords, *yCoords, *zCoords; vtkPoints *points = vtkPoints::New(); vtkPolyData *pd = vtkPolyData::New(); this->NumberOfNewOutputs++; // Initialize // if (!fileName) { vtkErrorMacro("A MeasuredFileName must be specified in the case file."); return 0; } std::string sfilename; if (this->FilePath) { sfilename = this->FilePath; if (sfilename.at(sfilename.length()-1) != '/') { sfilename += "/"; } sfilename += fileName; vtkDebugMacro("full path to measured geometry file: " << sfilename.c_str()); } else { sfilename = fileName; } if (this->OpenFile(sfilename.c_str()) == 0) { vtkErrorMacro("Unable to open file: " << sfilename.c_str()); return 0; } this->ReadLine(line); sscanf(line, " %*s %s", subLine); if (strncmp(subLine, "Binary", 6) != 0) { vtkErrorMacro("This is not a binary data set. Try " << "vtkEnSightGoldReader."); return 0; } if (this->UseFileSets) { this->AddFileIndexToCache(fileName); i = this->SeekToCachedTimeStep(fileName, timeStep-1); // start w/ the number of TS we skipped, not the one we are at // if we are not at the appropriate time step yet, we keep searching for (; i < timeStep - 1; i++) { while (strncmp(line, "BEGIN TIME STEP", 15) != 0) { this->ReadLine(line); } // Skip the description line. this->ReadLine(line); this->ReadLine(line); // "particle coordinates" this->ReadInt(&this->NumberOfMeasuredPoints); // Skip pointIds //this->GoldIFile->ignore(sizeof(int)*this->NumberOfMeasuredPoints); // Skip xCoords //this->GoldIFile->ignore(sizeof(float)*this->NumberOfMeasuredPoints); // Skip yCoords //this->GoldIFile->ignore(sizeof(float)*this->NumberOfMeasuredPoints); // Skip zCoords //this->GoldIFile->ignore(sizeof(float)*this->NumberOfMeasuredPoints); this->GoldIFile->seekg( (sizeof(float)*3 + sizeof(int))*this->NumberOfMeasuredPoints, ios::cur); this->ReadLine(line); // END TIME STEP } while (strncmp(line, "BEGIN TIME STEP", 15) != 0) { this->ReadLine(line); } // found a time step -> cache it this->AddTimeStepToCache(fileName, i, this->GoldIFile->tellg()); } // Skip the description line. this->ReadLine(line); this->ReadLine(line); // "particle coordinates" this->ReadInt(&this->NumberOfMeasuredPoints); pointIds = new int[this->NumberOfMeasuredPoints]; xCoords = new float [this->NumberOfMeasuredPoints]; yCoords = new float [this->NumberOfMeasuredPoints]; zCoords = new float [this->NumberOfMeasuredPoints]; points->Allocate(this->NumberOfMeasuredPoints); pd->Allocate(this->NumberOfMeasuredPoints); // Extract the array of point indices. Note EnSight Manual v8.2 (pp. 559, // http://www-vis.lbl.gov/NERSC/Software/ensight/docs82/UserManual.pdf) // is wrong in describing the format of binary measured geometry files. // As opposed to this description, the actual format employs a 'hybrid' // storage scheme. Specifically, point indices are stored in an array, // whereas 3D coordinates follow the array in a tuple-by-tuple manner. // The following code segment (20+ lines) serves as a fix to bug #9245. this->ReadIntArray( pointIds, this->NumberOfMeasuredPoints ); // Read point coordinates tuple by tuple while each tuple contains three // components: (x-cord, y-cord, z-cord) int floatSize = sizeof( float ); for ( i = 0; i < this->NumberOfMeasuredPoints; i ++ ) { this->GoldIFile->read( ( char * )( xCoords + i ), floatSize ); this->GoldIFile->read( ( char * )( yCoords + i ), floatSize ); this->GoldIFile->read( ( char * )( zCoords + i ), floatSize ); } if ( this->ByteOrder == FILE_LITTLE_ENDIAN ) { vtkByteSwap::Swap4LERange( xCoords, this->NumberOfMeasuredPoints ); vtkByteSwap::Swap4LERange( yCoords, this->NumberOfMeasuredPoints ); vtkByteSwap::Swap4LERange( zCoords, this->NumberOfMeasuredPoints ); } else { vtkByteSwap::Swap4BERange( xCoords, this->NumberOfMeasuredPoints ); vtkByteSwap::Swap4BERange( yCoords, this->NumberOfMeasuredPoints ); vtkByteSwap::Swap4BERange( zCoords, this->NumberOfMeasuredPoints ); } // NOTE: EnSight always employs a 1-based indexing scheme and therefore // 'if (this->ParticleCoordinatesByIndex)' was removed here. Otherwise // the measured geometry could not be proeperly interpreted. // This bug was noticed while fixing bug #7453. for (i = 0; i < this->NumberOfMeasuredPoints; i++) { points->InsertNextPoint(xCoords[i], yCoords[i], zCoords[i]); pd->InsertNextCell(VTK_VERTEX, 1, &i); } pd->SetPoints(points); this->AddToBlock(output, this->NumberOfGeometryParts, pd); points->Delete(); pd->Delete(); delete [] pointIds; delete [] xCoords; delete [] yCoords; delete [] zCoords; if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } return 1; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::ReadScalarsPerNode( const char* fileName, const char* description, int timeStep, vtkMultiBlockDataSet *compositeOutput, int measured, int numberOfComponents, int component) { char line[80]; int partId, realId, numPts, i, lineRead; vtkFloatArray *scalars; float* scalarsRead; vtkDataSet *output; // Initialize // if (!fileName) { vtkErrorMacro("nullptr ScalarPerNode variable file name"); return 0; } std::string sfilename; if (this->FilePath) { sfilename = this->FilePath; if (sfilename.at(sfilename.length()-1) != '/') { sfilename += "/"; } sfilename += fileName; vtkDebugMacro("full path to scalar per node file: " << sfilename.c_str()); } else { sfilename = fileName; } if (this->OpenFile(sfilename.c_str()) == 0) { vtkErrorMacro("Unable to open file: " << sfilename.c_str()); return 0; } if (this->UseFileSets) { this->AddFileIndexToCache(fileName); i = this->SeekToCachedTimeStep(fileName, timeStep-1); // start w/ the number of TS we skipped, not the one we are at // if we are not at the appropriate time step yet, we keep searching for (; i < timeStep - 1; i++) { this->ReadLine(line); while (strncmp(line, "BEGIN TIME STEP", 15) != 0) { this->ReadLine(line); } // found a time step -> cache it this->AddTimeStepToCache(fileName, i, this->GoldIFile->tellg()); this->ReadLine(line); // skip the description line if (measured) { output = static_cast( this->GetDataSetFromBlock(compositeOutput, this->NumberOfGeometryParts)); numPts = output->GetNumberOfPoints(); if (numPts) { this->ReadLine(line); // Skip sclalars this->GoldIFile->seekg(sizeof(float)*numPts, ios::cur); } } while (this->ReadLine(line) && strncmp(line, "part", 4) == 0) { this->ReadPartId(&partId); partId--; // EnSight starts #ing with 1. realId = this->InsertNewPartId(partId); output = static_cast( this->GetDataSetFromBlock(compositeOutput, realId)); numPts = output->GetNumberOfPoints(); if (numPts) { this->ReadLine(line); // "coordinates" or "block" // Skip sclalars this->GoldIFile->seekg(sizeof(float)*numPts, ios::cur); } } } this->ReadLine(line); while (strncmp(line, "BEGIN TIME STEP", 15) != 0) { this->ReadLine(line); } } this->ReadLine(line); // skip the description line if (measured) { output = static_cast( this->GetDataSetFromBlock(compositeOutput, this->NumberOfGeometryParts)); numPts = output->GetNumberOfPoints(); if (numPts) { // 'this->ReadLine(line)' was removed here, otherwise there would be a // problem with timestep retrieval of the measured scalars. // This bug was noticed while fixing bug #7453. scalars = vtkFloatArray::New(); scalars->SetNumberOfComponents(numberOfComponents); scalars->SetNumberOfTuples(numPts); scalarsRead = new float [numPts]; this->ReadFloatArray(scalarsRead, numPts); // Why are we setting only one component here? // Only one component is set because scalars are single-component arrays. // For complex scalars, there is a file for the real part and another // file for the imaginary part, but we are storing them as a 2-component // array. for (i = 0; i < numPts; i++) { scalars->SetComponent(i, component, scalarsRead[i]); } scalars->SetName(description); output->GetPointData()->AddArray(scalars); if (!output->GetPointData()->GetScalars()) { output->GetPointData()->SetScalars(scalars); } scalars->Delete(); delete [] scalarsRead; } if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } return 1; } lineRead = this->ReadLine(line); while (lineRead && strncmp(line, "part", 4) == 0) { this->ReadPartId(&partId); partId--; // EnSight starts #ing with 1. realId = this->InsertNewPartId(partId); output = this->GetDataSetFromBlock(compositeOutput, realId); numPts = output->GetNumberOfPoints(); // If the part has no points, then only the part number is listed in // the variable file. if (numPts) { this->ReadLine(line); // "coordinates" or "block" if (component == 0) { scalars = vtkFloatArray::New(); scalars->SetNumberOfComponents(numberOfComponents); scalars->SetNumberOfTuples(numPts); } else { scalars = (vtkFloatArray*)(output->GetPointData()-> GetArray(description)); } scalarsRead = new float[numPts]; this->ReadFloatArray(scalarsRead, numPts); for (i = 0; i < numPts; i++) { scalars->SetComponent(i, component, scalarsRead[i]); } if (component == 0) { scalars->SetName(description); output->GetPointData()->AddArray(scalars); if (!output->GetPointData()->GetScalars()) { output->GetPointData()->SetScalars(scalars); } scalars->Delete(); } else { output->GetPointData()->AddArray(scalars); } delete [] scalarsRead; } this->GoldIFile->peek(); if (this->GoldIFile->eof()) { lineRead = 0; continue; } lineRead = this->ReadLine(line); } if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } return 1; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::ReadVectorsPerNode( const char* fileName, const char* description, int timeStep, vtkMultiBlockDataSet *compositeOutput, int measured) { char line[80]; int partId, realId, numPts, i, lineRead; vtkFloatArray *vectors; float tuple[3]; float *comp1, *comp2, *comp3; float *vectorsRead; vtkDataSet *output; // Initialize // if (!fileName) { vtkErrorMacro("nullptr VectorPerNode variable file name"); return 0; } std::string sfilename; if (this->FilePath) { sfilename = this->FilePath; if (sfilename.at(sfilename.length()-1) != '/') { sfilename += "/"; } sfilename += fileName; vtkDebugMacro("full path to vector per node file: " << sfilename.c_str()); } else { sfilename = fileName; } if (this->OpenFile(sfilename.c_str()) == 0) { vtkErrorMacro("Unable to open file: " << sfilename.c_str()); return 0; } if (this->UseFileSets) { this->AddFileIndexToCache(fileName); i = this->SeekToCachedTimeStep(fileName, timeStep-1); // start w/ the number of TS we skipped, not the one we are at // if we are not at the appropriate time step yet, we keep searching for (; i < timeStep - 1; i++) { this->ReadLine(line); while (strncmp(line, "BEGIN TIME STEP", 15) != 0) { this->ReadLine(line); } // found a time step -> cache it this->AddTimeStepToCache(fileName, i, this->GoldIFile->tellg()); this->ReadLine(line); // skip the description line if (measured) { output = static_cast( this->GetDataSetFromBlock(compositeOutput, this->NumberOfGeometryParts)); numPts = output->GetNumberOfPoints(); if (numPts) { this->ReadLine(line); // Skip vectors. this->GoldIFile->seekg(sizeof(float)*3*numPts, ios::cur); } } while (this->ReadLine(line) && strncmp(line, "part", 4) == 0) { this->ReadPartId(&partId); partId--; // EnSight starts #ing with 1. realId = this->InsertNewPartId(partId); output = static_cast( this->GetDataSetFromBlock(compositeOutput, realId)); numPts = output->GetNumberOfPoints(); if (numPts) { this->ReadLine(line); // "coordinates" or "block" // Skip comp1, comp2 and comp3 this->GoldIFile->seekg(sizeof(float)*3*numPts, ios::cur); } } } this->ReadLine(line); while (strncmp(line, "BEGIN TIME STEP", 15) != 0) { this->ReadLine(line); } } this->ReadLine(line); // skip the description line if (measured) { output = static_cast( this->GetDataSetFromBlock(compositeOutput, this->NumberOfGeometryParts)); numPts = output->GetNumberOfPoints(); if (numPts) { // NOTE: NO ReadLine() here since there is only one description // line (already read above), immediately followed by the actual data. vectors = vtkFloatArray::New(); vectors->SetNumberOfComponents(3); vectors->SetNumberOfTuples(numPts); vectorsRead = vectors->GetPointer(0); this->ReadFloatArray(vectorsRead, numPts*3); vectors->SetName(description); output->GetPointData()->AddArray(vectors); if (!output->GetPointData()->GetVectors()) { output->GetPointData()->SetVectors(vectors); } vectors->Delete(); } if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } return 1; } lineRead = this->ReadLine(line); while (lineRead && strncmp(line, "part", 4) == 0) { vectors = vtkFloatArray::New(); this->ReadPartId(&partId); partId--; // EnSight starts #ing with 1. realId = this->InsertNewPartId(partId); output = this->GetDataSetFromBlock(compositeOutput, realId); numPts = output->GetNumberOfPoints(); if (numPts) { this->ReadLine(line); // "coordinates" or "block" vectors->SetNumberOfComponents(3); vectors->SetNumberOfTuples(numPts); comp1 = new float[numPts]; comp2 = new float[numPts]; comp3 = new float[numPts]; this->ReadFloatArray(comp1, numPts); this->ReadFloatArray(comp2, numPts); this->ReadFloatArray(comp3, numPts); for (i = 0; i < numPts; i++) { tuple[0] = comp1[i]; tuple[1] = comp2[i]; tuple[2] = comp3[i]; vectors->SetTuple(i, tuple); } vectors->SetName(description); output->GetPointData()->AddArray(vectors); if (!output->GetPointData()->GetVectors()) { output->GetPointData()->SetVectors(vectors); } vectors->Delete(); delete [] comp1; delete [] comp2; delete [] comp3; } this->GoldIFile->peek(); if (this->GoldIFile->eof()) { lineRead = 0; continue; } lineRead = this->ReadLine(line); } if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } return 1; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::ReadTensorsPerNode( const char* fileName, const char* description, int timeStep, vtkMultiBlockDataSet *compositeOutput) { char line[80]; int partId, realId, numPts, i, lineRead; vtkFloatArray *tensors; float *comp1, *comp2, *comp3, *comp4, *comp5, *comp6; float tuple[6]; vtkDataSet *output; // Initialize // if (!fileName) { vtkErrorMacro("nullptr TensorPerNode variable file name"); return 0; } std::string sfilename; if (this->FilePath) { sfilename = this->FilePath; if (sfilename.at(sfilename.length()-1) != '/') { sfilename += "/"; } sfilename += fileName; vtkDebugMacro("full path to tensor per node file: " << sfilename.c_str()); } else { sfilename = fileName; } if (this->OpenFile(sfilename.c_str()) == 0) { vtkErrorMacro("Unable to open file: " << sfilename.c_str()); return 0; } if (this->UseFileSets) { this->AddFileIndexToCache(fileName); i = this->SeekToCachedTimeStep(fileName, timeStep-1); // start w/ the number of TS we skipped, not the one we are at // if we are not at the appropriate time step yet, we keep searching for (; i < timeStep - 1; i++) { this->ReadLine(line); while (strncmp(line, "BEGIN TIME STEP", 15) != 0) { this->ReadLine(line); } // found a time step -> cache it this->AddTimeStepToCache(fileName, i, this->GoldIFile->tellg()); this->ReadLine(line); // skip the description line while (this->ReadLine(line) && strncmp(line, "part", 4) == 0) { this->ReadPartId(&partId); partId--; // EnSight starts #ing with 1. realId = this->InsertNewPartId(partId); output = this->GetDataSetFromBlock(compositeOutput, realId); numPts = output->GetNumberOfPoints(); if (numPts) { this->ReadLine(line); // "coordinates" or "block" // Skip over comp1, comp2, ... comp6 this->GoldIFile->seekg(sizeof(float)*6*numPts, ios::cur); } } } this->ReadLine(line); while (strncmp(line, "BEGIN TIME STEP", 15) != 0) { this->ReadLine(line); } } this->ReadLine(line); // skip the description line lineRead = this->ReadLine(line); while (lineRead && strncmp(line, "part", 4) == 0) { this->ReadPartId(&partId); partId--; // EnSight starts #ing with 1. realId = this->InsertNewPartId(partId); output = this->GetDataSetFromBlock(compositeOutput, realId); numPts = output->GetNumberOfPoints(); if (numPts) { tensors = vtkFloatArray::New(); this->ReadLine(line); // "coordinates" or "block" tensors->SetNumberOfComponents(6); tensors->SetNumberOfTuples(numPts); comp1 = new float[numPts]; comp2 = new float[numPts]; comp3 = new float[numPts]; comp4 = new float[numPts]; comp5 = new float[numPts]; comp6 = new float[numPts]; this->ReadFloatArray(comp1, numPts); this->ReadFloatArray(comp2, numPts); this->ReadFloatArray(comp3, numPts); this->ReadFloatArray(comp4, numPts); this->ReadFloatArray(comp6, numPts); this->ReadFloatArray(comp5, numPts); for (i = 0; i < numPts; i++) { tuple[0] = comp1[i]; tuple[1] = comp2[i]; tuple[2] = comp3[i]; tuple[3] = comp4[i]; tuple[4] = comp5[i]; tuple[5] = comp6[i]; tensors->InsertTuple(i, tuple); } tensors->SetName(description); output->GetPointData()->AddArray(tensors); tensors->Delete(); delete [] comp1; delete [] comp2; delete [] comp3; delete [] comp4; delete [] comp5; delete [] comp6; } this->GoldIFile->peek(); if (this->GoldIFile->eof()) { lineRead = 0; continue; } lineRead = this->ReadLine(line); } if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } return 1; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::ReadScalarsPerElement( const char* fileName, const char* description, int timeStep, vtkMultiBlockDataSet *compositeOutput, int numberOfComponents, int component) { char line[80]; int partId, realId, numCells, numCellsPerElement, i, idx; vtkFloatArray *scalars; float *scalarsRead; int lineRead, elementType; vtkDataSet *output; // Initialize // if (!fileName) { vtkErrorMacro("nullptr ScalarPerElement variable file name"); return 0; } std::string sfilename; if (this->FilePath) { sfilename = this->FilePath; if (sfilename.at(sfilename.length()-1) != '/') { sfilename += "/"; } sfilename += fileName; vtkDebugMacro("full path to scalar per element file: " << sfilename.c_str()); } else { sfilename = fileName; } if (this->OpenFile(sfilename.c_str()) == 0) { vtkErrorMacro("Unable to open file: " << sfilename.c_str()); return 0; } if (this->UseFileSets) { this->AddFileIndexToCache(fileName); i = this->SeekToCachedTimeStep(fileName, timeStep-1); // start w/ the number of TS we skipped, not the one we are at // if we are not at the appropriate time step yet, we keep searching for (; i < timeStep - 1; i++) { this->ReadLine(line); while (strncmp(line, "BEGIN TIME STEP", 15) != 0) { this->ReadLine(line); } // found a time step -> cache it this->AddTimeStepToCache(fileName, i, this->GoldIFile->tellg()); this->ReadLine(line); // skip the description line lineRead = this->ReadLine(line); // "part" while (lineRead && strncmp(line, "part", 4) == 0) { this->ReadPartId(&partId); partId--; // EnSight starts #ing with 1. realId = this->InsertNewPartId(partId); output = this->GetDataSetFromBlock(compositeOutput, realId); numCells = output->GetNumberOfCells(); if (numCells) { this->ReadLine(line); // element type or "block" // need to find out from CellIds how many cells we have of this // element type (and what their ids are) -- IF THIS IS NOT A BLOCK // SECTION if (strncmp(line, "block", 5) == 0) { // Skip over float scalars. this->GoldIFile->seekg(sizeof(float)*numCells, ios::cur); lineRead = this->ReadLine(line); } else { while (lineRead && strncmp(line, "part", 4) != 0 && strncmp(line, "END TIME STEP", 13) != 0) { elementType = this->GetElementType(line); if (elementType == -1) { vtkErrorMacro("Unknown element type \"" << line << "\""); if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } return 0; } idx = this->UnstructuredPartIds->IsId(realId); numCellsPerElement = this->GetCellIds(idx, elementType)-> GetNumberOfIds(); this->GoldIFile->seekg(sizeof(float)*numCellsPerElement, ios::cur); lineRead = this->ReadLine(line); } } // end while } // end if (numCells) else { lineRead = this->ReadLine(line); } } // end while } // end for this->ReadLine(line); while (strncmp(line, "BEGIN TIME STEP", 15) != 0) { this->ReadLine(line); } } this->ReadLine(line); // skip the description line lineRead = this->ReadLine(line); // "part" while (lineRead && strncmp(line, "part", 4) == 0) { this->ReadPartId(&partId); partId--; // EnSight starts #ing with 1. realId = this->InsertNewPartId(partId); output = this->GetDataSetFromBlock(compositeOutput, realId); numCells = output->GetNumberOfCells(); if (numCells) { this->ReadLine(line); // element type or "block" if (component == 0) { scalars = vtkFloatArray::New(); scalars->SetNumberOfComponents(numberOfComponents); scalars->SetNumberOfTuples(numCells); } else { scalars = (vtkFloatArray*)(output->GetCellData()->GetArray(description)); } // need to find out from CellIds how many cells we have of this element // type (and what their ids are) -- IF THIS IS NOT A BLOCK SECTION if (strncmp(line, "block", 5) == 0) { scalarsRead = new float[numCells]; this->ReadFloatArray(scalarsRead, numCells); for (i = 0; i < numCells; i++) { scalars->SetComponent(i, component, scalarsRead[i]); } if (this->GoldIFile->eof()) { lineRead = 0; } else { lineRead = this->ReadLine(line); } delete [] scalarsRead; } else { while (lineRead && strncmp(line, "part", 4) != 0 && strncmp(line, "END TIME STEP", 13) != 0) { elementType = this->GetElementType(line); if (elementType == -1) { vtkErrorMacro("Unknown element type \"" << line << "\""); if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } if (component == 0) { scalars->Delete(); } return 0; } idx = this->UnstructuredPartIds->IsId(realId); numCellsPerElement = this->GetCellIds(idx, elementType)->GetNumberOfIds(); scalarsRead = new float[numCellsPerElement]; this->ReadFloatArray(scalarsRead, numCellsPerElement); for (i = 0; i < numCellsPerElement; i++) { scalars->SetComponent(this->GetCellIds(idx, elementType)->GetId(i), component, scalarsRead[i]); } this->GoldIFile->peek(); if (this->GoldIFile->eof()) { lineRead = 0; } else { lineRead = this->ReadLine(line); } delete [] scalarsRead; } // end while } // end else if (component == 0) { scalars->SetName(description); output->GetCellData()->AddArray(scalars); if (!output->GetCellData()->GetScalars()) { output->GetCellData()->SetScalars(scalars); } scalars->Delete(); } else { output->GetCellData()->AddArray(scalars); } } else { this->GoldIFile->peek(); if (this->GoldIFile->eof()) { lineRead = 0; } else { lineRead = this->ReadLine(line); } } } if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } return 1; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::ReadVectorsPerElement( const char* fileName, const char* description, int timeStep, vtkMultiBlockDataSet *compositeOutput) { char line[80]; int partId, realId, numCells, numCellsPerElement, i, idx; vtkFloatArray *vectors; float *comp1, *comp2, *comp3; int lineRead, elementType; float tuple[3]; vtkDataSet *output; // Initialize // if (!fileName) { vtkErrorMacro("nullptr VectorPerElement variable file name"); return 0; } std::string sfilename; if (this->FilePath) { sfilename = this->FilePath; if (sfilename.at(sfilename.length()-1) != '/') { sfilename += "/"; } sfilename += fileName; vtkDebugMacro("full path to vector per element file: " << sfilename.c_str()); } else { sfilename = fileName; } if (this->OpenFile(sfilename.c_str()) == 0) { vtkErrorMacro("Unable to open file: " << sfilename.c_str()); return 0; } if (this->UseFileSets) { this->AddFileIndexToCache(fileName); i = this->SeekToCachedTimeStep(fileName, timeStep-1); // start w/ the number of TS we skipped, not the one we are at // if we are not at the appropriate time step yet, we keep searching for (; i < timeStep - 1; i++) { this->ReadLine(line); while (strncmp(line, "BEGIN TIME STEP", 15) != 0) { this->ReadLine(line); } // found a time step -> cache it this->AddTimeStepToCache(fileName, i, this->GoldIFile->tellg()); this->ReadLine(line); // skip the description line lineRead = this->ReadLine(line); // "part" while (lineRead && strncmp(line, "part", 4) == 0) { this->ReadPartId(&partId); partId--; // EnSight starts #ing with 1. realId = this->InsertNewPartId(partId); output = this->GetDataSetFromBlock(compositeOutput, realId); numCells = output->GetNumberOfCells(); if (numCells) { this->ReadLine(line); // element type or "block" // need to find out from CellIds how many cells we have of this // element type (and what their ids are) -- IF THIS IS NOT A BLOCK // SECTION if (strncmp(line, "block", 5) == 0) { // Skip over comp1, comp2 and comp3 this->GoldIFile->seekg(sizeof(float)*3*numCells, ios::cur); lineRead = this->ReadLine(line); } else { while (lineRead && strncmp(line, "part", 4) != 0 && strncmp(line, "END TIME STEP", 13) != 0) { elementType = this->GetElementType(line); if (elementType == -1) { vtkErrorMacro("Unknown element type \"" << line << "\""); delete this->IS; this->IS = nullptr; return 0; } idx = this->UnstructuredPartIds->IsId(realId); numCellsPerElement = this->GetCellIds(idx, elementType)-> GetNumberOfIds(); // Skip over comp1, comp2 and comp3 this->GoldIFile->seekg(sizeof(float)*3*numCellsPerElement, ios::cur); lineRead = this->ReadLine(line); } // end while } // end else } else { lineRead = this->ReadLine(line); } } } this->ReadLine(line); while (strncmp(line, "BEGIN TIME STEP", 15) != 0) { this->ReadLine(line); } } this->ReadLine(line); // skip the description line lineRead = this->ReadLine(line); // "part" while (lineRead && strncmp(line, "part", 4) == 0) { this->ReadPartId(&partId); partId--; // EnSight starts #ing with 1. realId = this->InsertNewPartId(partId); output = this->GetDataSetFromBlock(compositeOutput, realId); numCells = output->GetNumberOfCells(); if (numCells) { vectors = vtkFloatArray::New(); this->ReadLine(line); // element type or "block" vectors->SetNumberOfComponents(3); vectors->SetNumberOfTuples(numCells); // need to find out from CellIds how many cells we have of this element // type (and what their ids are) -- IF THIS IS NOT A BLOCK SECTION if (strncmp(line, "block", 5) == 0) { comp1 = new float[numCells]; comp2 = new float[numCells]; comp3 = new float[numCells]; this->ReadFloatArray(comp1, numCells); this->ReadFloatArray(comp2, numCells); this->ReadFloatArray(comp3, numCells); for (i = 0; i < numCells; i++) { tuple[0] = comp1[i]; tuple[1] = comp2[i]; tuple[2] = comp3[i]; vectors->SetTuple(i, tuple); } this->GoldIFile->peek(); if (this->GoldIFile->eof()) { lineRead = 0; } else { lineRead = this->ReadLine(line); } delete [] comp1; delete [] comp2; delete [] comp3; } else { while (lineRead && strncmp(line, "part", 4) != 0 && strncmp(line, "END TIME STEP", 13) != 0) { elementType = this->GetElementType(line); if (elementType == -1) { vtkErrorMacro("Unknown element type \"" << line << "\""); delete this->IS; this->IS = nullptr; vectors->Delete(); return 0; } idx = this->UnstructuredPartIds->IsId(realId); numCellsPerElement = this->GetCellIds(idx, elementType)->GetNumberOfIds(); comp1 = new float[numCellsPerElement]; comp2 = new float[numCellsPerElement]; comp3 = new float[numCellsPerElement]; this->ReadFloatArray(comp1, numCellsPerElement); this->ReadFloatArray(comp2, numCellsPerElement); this->ReadFloatArray(comp3, numCellsPerElement); for (i = 0; i < numCellsPerElement; i++) { tuple[0] = comp1[i]; tuple[1] = comp2[i]; tuple[2] = comp3[i]; vectors->SetTuple(this->GetCellIds(idx, elementType)->GetId(i), tuple); } this->GoldIFile->peek(); if (this->GoldIFile->eof()) { lineRead = 0; } else { lineRead = this->ReadLine(line); } delete [] comp1; delete [] comp2; delete [] comp3; } // end while } // end else vectors->SetName(description); output->GetCellData()->AddArray(vectors); if (!output->GetCellData()->GetVectors()) { output->GetCellData()->SetVectors(vectors); } vectors->Delete(); } else { this->GoldIFile->peek(); if (this->GoldIFile->eof()) { lineRead = 0; } else { lineRead = this->ReadLine(line); } } } if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } return 1; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::ReadTensorsPerElement( const char* fileName, const char* description, int timeStep, vtkMultiBlockDataSet *compositeOutput) { char line[80]; int partId, realId, numCells, numCellsPerElement, i, idx; vtkFloatArray *tensors; int lineRead, elementType; float *comp1, *comp2, *comp3, *comp4, *comp5, *comp6; float tuple[6]; vtkDataSet *output; // Initialize // if (!fileName) { vtkErrorMacro("nullptr TensorPerElement variable file name"); return 0; } std::string sfilename; if (this->FilePath) { sfilename = this->FilePath; if (sfilename.at(sfilename.length()-1) != '/') { sfilename += "/"; } sfilename += fileName; vtkDebugMacro("full path to tensor per element file: " << sfilename.c_str()); } else { sfilename = fileName; } if (this->OpenFile(sfilename.c_str()) == 0) { vtkErrorMacro("Unable to open file: " << sfilename.c_str()); return 0; } if (this->UseFileSets) { this->AddFileIndexToCache(fileName); i = this->SeekToCachedTimeStep(fileName, timeStep-1); // start w/ the number of TS we skipped, not the one we are at // if we are not at the appropriate time step yet, we keep searching for (; i < timeStep - 1; i++) { this->ReadLine(line); while (strncmp(line, "BEGIN TIME STEP", 15) != 0) { this->ReadLine(line); } // found a time step -> cache it this->AddTimeStepToCache(fileName, i, this->GoldIFile->tellg()); this->ReadLine(line); // skip the description line lineRead = this->ReadLine(line); // "part" while (lineRead && strncmp(line, "part", 4) == 0) { this->ReadPartId(&partId); partId--; // EnSight starts #ing with 1. realId = this->InsertNewPartId(partId); output = this->GetDataSetFromBlock(compositeOutput, realId); numCells = output->GetNumberOfCells(); if (numCells) { this->ReadLine(line); // element type or "block" // need to find out from CellIds how many cells we have of this // element type (and what their ids are) -- IF THIS IS NOT A BLOCK // SECTION if (strncmp(line, "block", 5) == 0) { // Skip comp1 - comp6 this->GoldIFile->seekg(sizeof(float)*6*numCells, ios::cur); lineRead = this->ReadLine(line); } else { while (lineRead && strncmp(line, "part", 4) != 0 && strncmp(line, "END TIME STEP", 13) != 0) { elementType = this->GetElementType(line); if (elementType == -1) { vtkErrorMacro("Unknown element type \"" << line << "\""); delete this->IS; this->IS = nullptr; return 0; } idx = this->UnstructuredPartIds->IsId(realId); numCellsPerElement = this->GetCellIds(idx, elementType)-> GetNumberOfIds(); // Skip over comp1->comp6 this->GoldIFile->seekg(sizeof(float)*6*numCellsPerElement, ios::cur); lineRead = this->ReadLine(line); } // end while } // end else } // end if (numCells) else { lineRead = this->ReadLine(line); } } } this->ReadLine(line); while (strncmp(line, "BEGIN TIME STEP", 15) != 0) { this->ReadLine(line); } } this->ReadLine(line); // skip the description line lineRead = this->ReadLine(line); // "part" while (lineRead && strncmp(line, "part", 4) == 0) { this->ReadPartId(&partId); partId--; // EnSight starts #ing with 1. realId = this->InsertNewPartId(partId); output = this->GetDataSetFromBlock(compositeOutput, realId); numCells = output->GetNumberOfCells(); if (numCells) { tensors = vtkFloatArray::New(); this->ReadLine(line); // element type or "block" tensors->SetNumberOfComponents(6); tensors->SetNumberOfTuples(numCells); // need to find out from CellIds how many cells we have of this element // type (and what their ids are) -- IF THIS IS NOT A BLOCK SECTION if (strncmp(line, "block", 5) == 0) { comp1 = new float[numCells]; comp2 = new float[numCells]; comp3 = new float[numCells]; comp4 = new float[numCells]; comp5 = new float[numCells]; comp6 = new float[numCells]; this->ReadFloatArray(comp1, numCells); this->ReadFloatArray(comp2, numCells); this->ReadFloatArray(comp3, numCells); this->ReadFloatArray(comp4, numCells); this->ReadFloatArray(comp6, numCells); this->ReadFloatArray(comp5, numCells); for (i = 0; i < numCells; i++) { tuple[0] = comp1[i]; tuple[1] = comp2[i]; tuple[2] = comp3[i]; tuple[3] = comp4[i]; tuple[4] = comp5[i]; tuple[5] = comp6[i]; tensors->InsertTuple(i, tuple); } this->GoldIFile->peek(); if (this->GoldIFile->eof()) { lineRead = 0; } else { lineRead = this->ReadLine(line); } delete [] comp1; delete [] comp2; delete [] comp3; delete [] comp4; delete [] comp5; delete [] comp6; } else { while (lineRead && strncmp(line, "part", 4) != 0 && strncmp(line, "END TIME STEP", 13) != 0) { elementType = this->GetElementType(line); if (elementType == -1) { vtkErrorMacro("Unknown element type \"" << line << "\""); delete this->IS; this->IS = nullptr; tensors->Delete(); return 0; } idx = this->UnstructuredPartIds->IsId(realId); numCellsPerElement = this->GetCellIds(idx, elementType)->GetNumberOfIds(); comp1 = new float[numCellsPerElement]; comp2 = new float[numCellsPerElement]; comp3 = new float[numCellsPerElement]; comp4 = new float[numCellsPerElement]; comp5 = new float[numCellsPerElement]; comp6 = new float[numCellsPerElement]; this->ReadFloatArray(comp1, numCellsPerElement); this->ReadFloatArray(comp2, numCellsPerElement); this->ReadFloatArray(comp3, numCellsPerElement); this->ReadFloatArray(comp4, numCellsPerElement); this->ReadFloatArray(comp6, numCellsPerElement); this->ReadFloatArray(comp5, numCellsPerElement); for (i = 0; i < numCellsPerElement; i++) { tuple[0] = comp1[i]; tuple[1] = comp2[i]; tuple[2] = comp3[i]; tuple[3] = comp4[i]; tuple[4] = comp5[i]; tuple[5] = comp6[i]; tensors->InsertTuple(this->GetCellIds(idx, elementType)->GetId(i), tuple); } this->GoldIFile->peek(); if (this->GoldIFile->eof()) { lineRead = 0; } else { lineRead = this->ReadLine(line); } delete [] comp1; delete [] comp2; delete [] comp3; delete [] comp4; delete [] comp5; delete [] comp6; } // end while } // end else tensors->SetName(description); output->GetCellData()->AddArray(tensors); tensors->Delete(); } else { this->GoldIFile->peek(); if (this->GoldIFile->eof()) { lineRead = 0; } else { lineRead = this->ReadLine(line); } } } if (this->GoldIFile) { this->GoldIFile->close(); delete this->GoldIFile; this->GoldIFile = nullptr; } return 1; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::CreateUnstructuredGridOutput( int partId, char line[80], const char* name, vtkMultiBlockDataSet *compositeOutput) { int lineRead = 1; int i, j; vtkIdType *nodeIds; int *nodeIdList; int numElements; int idx, cellId, cellType; float *xCoords, *yCoords, *zCoords; this->NumberOfNewOutputs++; if (this->GetDataSetFromBlock(compositeOutput, partId) == nullptr || !this->GetDataSetFromBlock(compositeOutput, partId)->IsA("vtkUnstructuredGrid")) { vtkDebugMacro("creating new unstructured output"); vtkUnstructuredGrid* ugrid = vtkUnstructuredGrid::New(); this->AddToBlock(compositeOutput, partId, ugrid); ugrid->Delete(); this->UnstructuredPartIds->InsertNextId(partId); } vtkUnstructuredGrid* output = vtkUnstructuredGrid::SafeDownCast( this->GetDataSetFromBlock(compositeOutput, partId)); this->SetBlockName(compositeOutput, partId, name); // Clear all cell ids from the last execution, if any. idx = this->UnstructuredPartIds->IsId(partId); for (i = 0; i < vtkEnSightReader::NUMBER_OF_ELEMENT_TYPES; i++) { this->GetCellIds(idx, i)->Reset(); } output->Allocate(1000); while(lineRead && strncmp(line, "part", 4) != 0) { if (strncmp(line, "coordinates", 11) == 0) { vtkDebugMacro("coordinates"); int numPts; this->ReadInt(&numPts); if (numPts < 0 || static_cast(numPts * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of unstructured points read; check that ByteOrder is set correctly."); return -1; } vtkPoints *points = vtkPoints::New(); vtkDebugMacro("num. points: " << numPts); points->Allocate(numPts); if (this->NodeIdsListed) { this->GoldIFile->seekg(sizeof(int)*numPts, ios::cur); } xCoords = new float[numPts]; yCoords = new float[numPts]; zCoords = new float[numPts]; this->ReadFloatArray(xCoords, numPts); this->ReadFloatArray(yCoords, numPts); this->ReadFloatArray(zCoords, numPts); for (i = 0; i < numPts; i++) { points->InsertNextPoint(xCoords[i], yCoords[i], zCoords[i]); } output->SetPoints(points); points->Delete(); delete [] xCoords; delete [] yCoords; delete [] zCoords; } else if (strncmp(line, "point", 5) == 0) { vtkDebugMacro("point"); this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of point cells; check that ByteOrder is set correctly."); return -1; } nodeIds = new vtkIdType[1]; if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } nodeIdList = new int[numElements]; this->ReadIntArray(nodeIdList, numElements); for (i = 0; i < numElements; i++) { nodeIds[0] = nodeIdList[i] - 1; cellId = output->InsertNextCell(VTK_VERTEX, 1, nodeIds); this->GetCellIds(idx, vtkEnSightReader::POINT)->InsertNextId(cellId); } delete [] nodeIds; delete [] nodeIdList; } else if (strncmp(line, "g_point", 7) == 0) { // skipping ghost cells vtkDebugMacro("g_point"); this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of g_point cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { // skip element ids. this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } else if (strncmp(line, "bar2", 4) == 0) { vtkDebugMacro("bar2"); this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of bar2 cells; check that ByteOrder is set correctly."); return -1; } nodeIds = new vtkIdType[2]; if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } nodeIdList = new int[numElements * 2]; this->ReadIntArray(nodeIdList, numElements * 2); for (i = 0; i < numElements; i++) { for (j = 0; j < 2; j++) { nodeIds[j] = nodeIdList[2*i+j] - 1; } cellId = output->InsertNextCell(VTK_LINE, 2, nodeIds); this->GetCellIds(idx, vtkEnSightReader::BAR2)->InsertNextId(cellId); } delete [] nodeIds; delete [] nodeIdList; } else if (strncmp(line, "g_bar2", 6) == 0) { // skipping ghost cells vtkDebugMacro("g_bar2"); this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of g_bar2 cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*2*numElements, ios::cur); } else if (strncmp(line, "bar3", 4) == 0) { vtkDebugMacro("bar3"); this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of bar3 cells; check that ByteOrder is set correctly."); return -1; } nodeIds = new vtkIdType[3]; if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } nodeIdList = new int[numElements*3]; this->ReadIntArray(nodeIdList, numElements*3); for (i = 0; i < numElements; i++) { nodeIds[0] = nodeIdList[3*i]-1; nodeIds[1] = nodeIdList[3*i+2]-1; nodeIds[2] = nodeIdList[3*i+1]-1; cellId = output->InsertNextCell(VTK_QUADRATIC_EDGE, 3, nodeIds); this->GetCellIds(idx, vtkEnSightReader::BAR3)->InsertNextId(cellId); } delete [] nodeIds; delete [] nodeIdList; } else if (strncmp(line, "g_bar3", 6) == 0) { // skipping ghost cells vtkDebugMacro("g_bar3"); this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of g_bar3 cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*2*numElements, ios::cur); } else if (strncmp(line, "nsided", 6) == 0) { vtkDebugMacro("nsided"); int *numNodesPerElement; int numNodes = 0; int nodeCount = 0; cellType = vtkEnSightReader::NSIDED; this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of nsided cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } numNodesPerElement = new int[numElements]; this->ReadIntArray(numNodesPerElement, numElements); for (i = 0; i < numElements; i++) { numNodes += numNodesPerElement[i]; } nodeIdList = new int[numNodes]; this->ReadIntArray(nodeIdList, numNodes); for (i = 0; i < numElements; i++) { nodeIds = new vtkIdType[numNodesPerElement[i]]; for (j = 0; j < numNodesPerElement[i]; j++) { nodeIds[j] = nodeIdList[nodeCount] - 1; nodeCount++; } cellId = output->InsertNextCell(VTK_POLYGON, numNodesPerElement[i], nodeIds); this->GetCellIds(idx, cellType)->InsertNextId(cellId); delete [] nodeIds; } delete [] nodeIdList; delete [] numNodesPerElement; } else if (strncmp(line, "g_nsided", 8) == 0) { // skipping ghost cells vtkDebugMacro("g_nsided"); int *numNodesPerElement; int numNodes = 0; //cellType = vtkEnSightReader::NSIDED; this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of g_nsided cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } numNodesPerElement = new int[numElements]; this->ReadIntArray(numNodesPerElement, numElements); for (i = 0; i < numElements; i++) { numNodes += numNodesPerElement[i]; } // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*numNodes, ios::cur); delete [] numNodesPerElement; } else if (strncmp(line, "tria3", 5) == 0 || strncmp(line, "tria6", 5) == 0) { if (strncmp(line, "tria6", 5) == 0) { vtkDebugMacro("tria6"); cellType = vtkEnSightReader::TRIA6; } else { vtkDebugMacro("tria3"); cellType = vtkEnSightReader::TRIA3; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of triangle cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::TRIA6) { nodeIds = new vtkIdType[6]; nodeIdList = new int[numElements*6]; this->ReadIntArray(nodeIdList, numElements*6); } else { nodeIds = new vtkIdType[3]; nodeIdList = new int[numElements*3]; this->ReadIntArray(nodeIdList, numElements*3); } for (i = 0; i < numElements; i++) { if (cellType == vtkEnSightReader::TRIA6) { for (j = 0; j < 6; j++) { nodeIds[j] = nodeIdList[6*i+j] - 1; } cellId = output->InsertNextCell(VTK_QUADRATIC_TRIANGLE, 6, nodeIds); } else { for (j = 0; j < 3; j++) { nodeIds[j] = nodeIdList[3*i+j] - 1; } cellId = output->InsertNextCell(VTK_TRIANGLE, 3, nodeIds); } this->GetCellIds(idx, cellType)->InsertNextId(cellId); } delete [] nodeIds; delete [] nodeIdList; } else if (strncmp(line, "g_tria3", 7) == 0 || strncmp(line, "g_tria6", 7) == 0) { // skipping ghost cells if (strncmp(line, "g_tria6", 7) == 0) { vtkDebugMacro("g_tria6"); cellType = vtkEnSightReader::TRIA6; } else { vtkDebugMacro("g_tria3"); cellType = vtkEnSightReader::TRIA3; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of triangle cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::TRIA6) { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*6*numElements, ios::cur); } else { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*3*numElements, ios::cur); } } else if (strncmp(line, "quad4", 5) == 0 || strncmp(line, "quad8", 5) == 0) { if (strncmp(line, "quad8", 5) == 0) { vtkDebugMacro("quad8"); cellType = vtkEnSightReader::QUAD8; } else { vtkDebugMacro("quad4"); cellType = vtkEnSightReader::QUAD4; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of quad cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::QUAD8) { nodeIds = new vtkIdType[8]; nodeIdList = new int[numElements*8]; this->ReadIntArray(nodeIdList, numElements*8); } else { nodeIds = new vtkIdType[4]; nodeIdList = new int[numElements*4]; this->ReadIntArray(nodeIdList, numElements*4); } for (i = 0; i < numElements; i++) { if (cellType == vtkEnSightReader::QUAD8) { for (j = 0; j < 8; j++) { nodeIds[j] = nodeIdList[8*i+j] - 1; } cellId = output->InsertNextCell(VTK_QUADRATIC_QUAD, 8, nodeIds); } else { for (j = 0; j < 4; j++) { nodeIds[j] = nodeIdList[4*i+j] - 1; } cellId = output->InsertNextCell(VTK_QUAD, 4, nodeIds); } this->GetCellIds(idx, cellType)->InsertNextId(cellId); } delete [] nodeIds; delete [] nodeIdList; } else if (strncmp(line, "g_quad4", 7) == 0 || strncmp(line, "g_quad8", 7) == 0) { // skipping ghost cells if (strncmp(line, "g_quad8", 7) == 0) { vtkDebugMacro("g_quad8"); cellType = vtkEnSightReader::QUAD8; } else { vtkDebugMacro("g_quad4"); cellType = vtkEnSightReader::QUAD4; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of quad cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::QUAD8) { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*8*numElements, ios::cur); } else { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*4*numElements, ios::cur); } } else if (strncmp(line, "nfaced", 6) == 0) { vtkDebugMacro("nfaced"); int *numFacesPerElement; int *numNodesPerFace; int *numNodesPerElement; int *nodeMarker; int numPts = 0; int numFaces = 0; int numNodes = 0; int faceCount = 0; int nodeCount = 0; int elementNodeCount = 0; cellType = vtkEnSightReader::NFACED; this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of nfaced cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } // array: number of faces per element numFacesPerElement = new int[numElements]; this->ReadIntArray(numFacesPerElement, numElements); // array: number of nodes per face for (i = 0; i < numElements; i++) { numFaces += numFacesPerElement[i]; } numNodesPerFace = new int[numFaces]; this->ReadIntArray(numNodesPerFace, numFaces); // array: number of nodes per element // number of faces of all elements numNodesPerElement = new int[numElements]; for (i = 0; i < numElements; i++) { numNodesPerElement[i] = 0; for (j = 0; j < numFacesPerElement[i]; j++) { numNodesPerElement[i] += numNodesPerFace[faceCount + j]; } faceCount += numFacesPerElement[i]; } //xxx begin //delete [] numFacesPerElement; //delete [] numNodesPerFace; //xxx end // number of nodes of all elements for (i = 0; i < numElements; i++) { numNodes += numNodesPerElement[i]; } // allocate and init markers to determine unique points numPts = output->GetNumberOfPoints(); nodeMarker = new int[numPts]; for (i = 0; i < numPts; i++) { nodeMarker[i] = -1; } // array: node Ids of all elements // NOTE: each node Id is usually referenced multiple times in a // polyhedron and therefore nodeIdList is not a set of // UNIQUE point Ids (instead it an RAW list) nodeIdList = new int[numNodes]; this->ReadIntArray(nodeIdList, numNodes); // yyy begin int k; // indexing each node Id of a face int faceIdx = 0; // indexing faces throughout all polyhedra int nodeIdx = 0; // indexing nodes throughout all polyhedra int arayIdx = 0; // indexing the array of Ids (info of faces) vtkIdType * faceAry = nullptr; // array of Ids describing a vtkPolyhedron // yyy end for (i = 0; i < numElements; i++) { elementNodeCount = 0; nodeIds = new vtkIdType[numNodesPerElement[i]]; // yyy begin arayIdx = 0; faceAry = new vtkIdType[ numFacesPerElement[i] + numNodesPerElement[i] ]; for ( j = 0; j < numFacesPerElement[i]; j ++, faceIdx ++ ) { faceAry[ arayIdx ++ ] = numNodesPerFace[ faceIdx ]; for ( k = 0; k < numNodesPerFace[ faceIdx ]; k ++ ) { faceAry[ arayIdx ++ ] = nodeIdList[ nodeIdx ++ ] - 1; } } //yyy end for (j = 0; j < numNodesPerElement[i]; j++) { if (nodeMarker[nodeIdList[nodeCount] - 1] < i) { nodeIds[elementNodeCount] = nodeIdList[nodeCount] - 1; nodeMarker[nodeIdList[nodeCount] - 1] = i; elementNodeCount += 1; } nodeCount++; } // xxx begin //cellId = output->InsertNextCell( VTK_CONVEX_POINT_SET, // elementNodeCount, nodeIds ); // xxx end // yyy begin cellId = output->InsertNextCell( VTK_POLYHEDRON, elementNodeCount, nodeIds, numFacesPerElement[i], faceAry ); delete [] faceAry; faceAry = nullptr; //yyy end this->GetCellIds(idx, cellType)->InsertNextId(cellId); delete [] nodeIds; } // yyy begin delete [] numNodesPerFace; delete [] numFacesPerElement; numNodesPerFace = nullptr; numFacesPerElement = nullptr; // yyy end delete [] nodeMarker; delete [] nodeIdList; delete [] numNodesPerElement; } else if (strncmp(line, "tetra4", 6) == 0 || strncmp(line, "tetra10", 7) == 0) { if (strncmp(line, "tetra10", 7) == 0) { vtkDebugMacro("tetra10"); cellType = vtkEnSightReader::TETRA10; } else { vtkDebugMacro("tetra4"); cellType = vtkEnSightReader::TETRA4; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of tetrahedral cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::TETRA10) { nodeIds = new vtkIdType[10]; nodeIdList = new int[numElements*10]; this->ReadIntArray(nodeIdList, numElements*10); } else { nodeIds = new vtkIdType[4]; nodeIdList = new int[numElements*4]; this->ReadIntArray(nodeIdList, numElements*4); } for (i = 0; i < numElements; i++) { if (cellType == vtkEnSightReader::TETRA10) { for (j = 0; j < 10; j++) { nodeIds[j] = nodeIdList[10*i+j] - 1; } cellId = output->InsertNextCell(VTK_QUADRATIC_TETRA, 10, nodeIds); } else { for (j = 0; j < 4; j++) { nodeIds[j] = nodeIdList[4*i+j] - 1; } cellId = output->InsertNextCell(VTK_TETRA, 4, nodeIds); } this->GetCellIds(idx, cellType)->InsertNextId(cellId); } delete [] nodeIds; delete [] nodeIdList; } else if (strncmp(line, "g_tetra4", 8) == 0 || strncmp(line, "g_tetra10", 9) == 0) { // skipping ghost cells if (strncmp(line, "g_tetra10", 9) == 0) { vtkDebugMacro("g_tetra10"); cellType = vtkEnSightReader::TETRA10; } else { vtkDebugMacro("g_tetra4"); cellType = vtkEnSightReader::TETRA4; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of tetrahedral cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::TETRA10) { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*10*numElements, ios::cur); } else { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*4*numElements, ios::cur); } } else if (strncmp(line, "pyramid5", 8) == 0 || strncmp(line, "pyramid13", 9) == 0) { if (strncmp(line, "pyramid13", 9) == 0) { vtkDebugMacro("pyramid13"); cellType = vtkEnSightReader::PYRAMID13; } else { vtkDebugMacro("pyramid5"); cellType = vtkEnSightReader::PYRAMID5; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of pyramid cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::PYRAMID13) { nodeIds = new vtkIdType[13]; nodeIdList = new int[numElements*13]; this->ReadIntArray(nodeIdList, numElements*13); } else { nodeIds = new vtkIdType[5]; nodeIdList = new int[numElements*5]; this->ReadIntArray(nodeIdList, numElements*5); } for (i = 0; i < numElements; i++) { if (cellType == vtkEnSightReader::PYRAMID13) { for (j = 0; j < 13; j++) { nodeIds[j] = nodeIdList[13*i+j] - 1; } cellId = output->InsertNextCell(VTK_QUADRATIC_PYRAMID, 13, nodeIds); } else { for (j = 0; j < 5; j++) { nodeIds[j] = nodeIdList[5*i+j] - 1; } cellId = output->InsertNextCell(VTK_PYRAMID, 5, nodeIds); } this->GetCellIds(idx, cellType)->InsertNextId(cellId); } delete [] nodeIds; delete [] nodeIdList; } else if (strncmp(line, "g_pyramid5", 10) == 0 || strncmp(line, "g_pyramid13", 11) == 0) { // skipping ghost cells if (strncmp(line, "g_pyramid13", 11) == 0) { vtkDebugMacro("g_pyramid13"); cellType = vtkEnSightReader::PYRAMID13; } else { vtkDebugMacro("g_pyramid5"); cellType = vtkEnSightReader::PYRAMID5; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of pyramid cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::PYRAMID13) { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*13*numElements, ios::cur); } else { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*5*numElements, ios::cur); } } else if (strncmp(line, "hexa8", 5) == 0 || strncmp(line, "hexa20", 6) == 0) { if (strncmp(line, "hexa20", 6) == 0) { vtkDebugMacro("hexa20"); cellType = vtkEnSightReader::HEXA20; } else { vtkDebugMacro("hexa8"); cellType = vtkEnSightReader::HEXA8; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of hexahedral cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::HEXA20) { nodeIds = new vtkIdType[20]; nodeIdList = new int[numElements*20]; this->ReadIntArray(nodeIdList, numElements*20); } else { nodeIds = new vtkIdType[8]; nodeIdList = new int[numElements*8]; this->ReadIntArray(nodeIdList, numElements*8); } for (i = 0; i < numElements; i++) { if (cellType == vtkEnSightReader::HEXA20) { for (j = 0; j < 20; j++) { nodeIds[j] = nodeIdList[20*i+j] - 1; } cellId = output->InsertNextCell(VTK_QUADRATIC_HEXAHEDRON, 20, nodeIds); } else { for (j = 0; j < 8; j++) { nodeIds[j] = nodeIdList[8*i+j] - 1; } cellId = output->InsertNextCell(VTK_HEXAHEDRON, 8, nodeIds); } this->GetCellIds(idx, cellType)->InsertNextId(cellId); } delete [] nodeIds; delete [] nodeIdList; } else if (strncmp(line, "g_hexa8", 7) == 0 || strncmp(line, "g_hexa20", 8) == 0) { // skipping ghost cells if (strncmp(line, "g_hexa20", 8) == 0) { vtkDebugMacro("g_hexa20"); cellType = vtkEnSightReader::HEXA20; } else { vtkDebugMacro("g_hexa8"); cellType = vtkEnSightReader::HEXA8; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of hexahedral cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::HEXA20) { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*20*numElements, ios::cur); } else { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*8*numElements, ios::cur); } } else if (strncmp(line, "penta6", 6) == 0 || strncmp(line, "penta15", 7) == 0) { if (strncmp(line, "penta15", 7) == 0) { vtkDebugMacro("penta15"); cellType = vtkEnSightReader::PENTA15; } else { vtkDebugMacro("penta6"); cellType = vtkEnSightReader::PENTA6; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of pentagonal cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::PENTA15) { nodeIds = new vtkIdType[15]; nodeIdList = new int[numElements*15]; this->ReadIntArray(nodeIdList, numElements*15); } else { nodeIds = new vtkIdType[6]; nodeIdList = new int[numElements*6]; this->ReadIntArray(nodeIdList, numElements*6); } const unsigned char penta6Map[6] = {0, 2, 1, 3, 5, 4}; const unsigned char penta15Map[15] = {0, 2, 1, 3, 5, 4, 8, 7, 6, 11, 10, 9, 12, 14, 13}; for (i = 0; i < numElements; i++) { if (cellType == vtkEnSightReader::PENTA15) { for (j = 0; j < 15; j++) { nodeIds[penta15Map[j]] = nodeIdList[15*i+j] - 1; } cellId = output->InsertNextCell(VTK_QUADRATIC_WEDGE, 15, nodeIds); } else { for (j = 0; j < 6; j++) { nodeIds[penta6Map[j]] = nodeIdList[6*i+j] - 1; } cellId = output->InsertNextCell(VTK_WEDGE, 6, nodeIds); } this->GetCellIds(idx, cellType)->InsertNextId(cellId); } delete [] nodeIds; delete [] nodeIdList; } else if (strncmp(line, "g_penta6", 8) == 0 || strncmp(line, "g_penta15", 9) == 0) { // skipping ghost cells if (strncmp(line, "g_penta15", 9) == 0) { vtkDebugMacro("g_penta15"); cellType = vtkEnSightReader::PENTA15; } else { vtkDebugMacro("g_penta6"); cellType = vtkEnSightReader::PENTA6; } this->ReadInt(&numElements); if (numElements < 0 || static_cast(numElements * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid number of pentagonal cells; check that ByteOrder is set correctly."); return -1; } if (this->ElementIdsListed) { this->GoldIFile->seekg(sizeof(int)*numElements, ios::cur); } if (cellType == vtkEnSightReader::PENTA15) { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*15*numElements, ios::cur); } else { // Skip nodeIdList. this->GoldIFile->seekg(sizeof(int)*6*numElements, ios::cur); } } else if (strncmp(line, "END TIME STEP", 13) == 0) { return 1; } else if (this->IS->fail()) { //May want consistency check here? //vtkWarningMacro("EOF on geometry file"); return 1; } else { vtkErrorMacro("undefined geometry file line"); return -1; } this->GoldIFile->peek(); if (this->GoldIFile->eof()) { lineRead = 0; continue; } lineRead = this->ReadLine(line); } return lineRead; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::CreateStructuredGridOutput( int partId, char line[80], const char* name, vtkMultiBlockDataSet *compositeOutput) { char subLine[80]; int lineRead; int iblanked = 0; int dimensions[3]; int i; vtkPoints *points = vtkPoints::New(); int numPts; float *xCoords, *yCoords, *zCoords; this->NumberOfNewOutputs++; vtkDataSet* ds = this->GetDataSetFromBlock(compositeOutput, partId); if (ds == nullptr || ! ds->IsA("vtkStructuredGrid")) { vtkDebugMacro("creating new structured grid output"); vtkStructuredGrid* sgrid = vtkStructuredGrid::New(); this->AddToBlock(compositeOutput, partId, sgrid); sgrid->Delete(); ds = sgrid; } vtkStructuredGrid* output = vtkStructuredGrid::SafeDownCast(ds); this->SetBlockName(compositeOutput, partId, name); if (sscanf(line, " %*s %s", subLine) == 1) { if (strncmp(subLine, "iblanked", 8) == 0) { iblanked = 1; } } this->ReadIntArray(dimensions, 3); numPts = dimensions[0] * dimensions[1] * dimensions[2]; if (dimensions[0] < 0 || static_cast(dimensions[0] * this->SizeOfInt) > this->FileSize || dimensions[1] < 0 || static_cast(dimensions[1] * this->SizeOfInt) > this->FileSize || dimensions[2] < 0 || static_cast(dimensions[2] * this->SizeOfInt) > this->FileSize || numPts < 0 || static_cast(numPts*this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid dimensions read; check that ByteOrder is set correctly."); points->Delete(); return -1; } output->SetDimensions(dimensions); points->Allocate(numPts); xCoords = new float[numPts]; yCoords = new float[numPts]; zCoords = new float[numPts]; this->ReadFloatArray(xCoords, numPts); this->ReadFloatArray(yCoords, numPts); this->ReadFloatArray(zCoords, numPts); for (i = 0; i < numPts; i++) { points->InsertNextPoint(xCoords[i], yCoords[i], zCoords[i]); } output->SetPoints(points); if (iblanked) { int *iblanks = new int[numPts]; this->ReadIntArray(iblanks, numPts); for (i = 0; i < numPts; i++) { if (!iblanks[i]) { output->BlankPoint(i); } } delete [] iblanks; } points->Delete(); delete [] xCoords; delete [] yCoords; delete [] zCoords; this->GoldIFile->peek(); if (this->GoldIFile->eof()) { lineRead = 0; } else { lineRead = this->ReadLine(line); } if (strncmp(line, "node_ids", 8) == 0) { int *nodeIds = new int[numPts]; this->ReadIntArray(nodeIds, numPts); lineRead = this->ReadLine(line); delete [] nodeIds; } if (strncmp(line, "element_ids", 11) == 0) { int numElements = (dimensions[0] - 1) * (dimensions[1] - 1) * (dimensions[2] - 1); int *elementIds = new int[numElements]; this->ReadIntArray(elementIds, numElements); lineRead = this->ReadLine(line); delete [] elementIds; } return lineRead; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::CreateRectilinearGridOutput( int partId, char line[80], const char* name, vtkMultiBlockDataSet *compositeOutput) { char subLine[80]; int lineRead; int iblanked = 0; int dimensions[3]; int i; vtkFloatArray *xCoords = vtkFloatArray::New(); vtkFloatArray *yCoords = vtkFloatArray::New(); vtkFloatArray *zCoords = vtkFloatArray::New(); float *tempCoords; int numPts; this->NumberOfNewOutputs++; vtkDataSet* ds = this->GetDataSetFromBlock(compositeOutput, partId); if (ds == nullptr || !ds->IsA("vtkRectilinearGrid")) { vtkDebugMacro("creating new rectilinear grid output"); vtkRectilinearGrid* rgrid = vtkRectilinearGrid::New(); this->AddToBlock(compositeOutput, partId, rgrid); rgrid->Delete(); ds = rgrid; } vtkRectilinearGrid* output = vtkRectilinearGrid::SafeDownCast(ds); this->SetBlockName(compositeOutput, partId, name); if (sscanf(line, " %*s %*s %s", subLine) == 1) { if (strncmp(subLine, "iblanked", 8) == 0) { iblanked = 1; } } this->ReadIntArray(dimensions, 3); if (dimensions[0] < 0 || static_cast(dimensions[0] * this->SizeOfInt) > this->FileSize || dimensions[1] < 0 || static_cast(dimensions[1] * this->SizeOfInt) > this->FileSize || dimensions[2] < 0 || static_cast(dimensions[2] * this->SizeOfInt) > this->FileSize || (dimensions[0]+dimensions[1]+dimensions[2]) < 0 || static_cast((dimensions[0] + dimensions[1] + dimensions[2]) * this->SizeOfInt) > this->FileSize) { vtkErrorMacro("Invalid dimensions read; check that BytetOrder is set correctly."); xCoords->Delete(); yCoords->Delete(); zCoords->Delete(); return -1; } output->SetDimensions(dimensions); xCoords->Allocate(dimensions[0]); yCoords->Allocate(dimensions[1]); zCoords->Allocate(dimensions[2]); tempCoords = new float[dimensions[0]]; this->ReadFloatArray(tempCoords, dimensions[0]); for (i = 0; i < dimensions[0]; i++) { xCoords->InsertNextTuple(&tempCoords[i]); } delete [] tempCoords; tempCoords = new float[dimensions[1]]; this->ReadFloatArray(tempCoords, dimensions[1]); for (i = 0; i < dimensions[1]; i++) { yCoords->InsertNextTuple(&tempCoords[i]); } delete [] tempCoords; tempCoords = new float[dimensions[2]]; this->ReadFloatArray(tempCoords, dimensions[2]); for (i = 0; i < dimensions[2]; i++) { zCoords->InsertNextTuple(&tempCoords[i]); } delete [] tempCoords; if (iblanked) { vtkWarningMacro("VTK does not handle blanking for rectilinear grids."); numPts = dimensions[0] * dimensions[1] * dimensions[2]; int *tempArray = new int[numPts]; this->ReadIntArray(tempArray, numPts); delete [] tempArray; } output->SetXCoordinates(xCoords); output->SetYCoordinates(yCoords); output->SetZCoordinates(zCoords); xCoords->Delete(); yCoords->Delete(); zCoords->Delete(); // reading next line to check for EOF lineRead = this->ReadLine(line); return lineRead; } //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::CreateImageDataOutput( int partId, char line[80], const char* name, vtkMultiBlockDataSet *compositeOutput) { char subLine[80]; int lineRead; int iblanked = 0; int dimensions[3]; float origin[3], delta[3]; int numPts; this->NumberOfNewOutputs++; vtkDataSet* ds = this->GetDataSetFromBlock(compositeOutput, partId); if (ds == nullptr || !ds->IsA("vtkImageData")) { vtkDebugMacro("creating new image data output"); vtkImageData* idata = vtkImageData::New(); this->AddToBlock(compositeOutput, partId, idata); idata->Delete(); ds = idata; } vtkImageData* output = vtkImageData::SafeDownCast(ds); this->SetBlockName(compositeOutput, partId, name); if (sscanf(line, " %*s %*s %s", subLine) == 1) { if (strncmp(subLine, "iblanked", 8) == 0) { iblanked = 1; } } this->ReadIntArray(dimensions, 3); output->SetDimensions(dimensions); this->ReadFloatArray(origin, 3); output->SetOrigin(origin[0], origin[1], origin[2]); this->ReadFloatArray(delta, 3); output->SetSpacing(delta[0], delta[1], delta[2]); if (iblanked) { vtkWarningMacro("VTK does not handle blanking for image data."); numPts = dimensions[0]*dimensions[1]*dimensions[2]; if (dimensions[0] < 0 || static_cast(dimensions[0] * this->SizeOfInt) > this->FileSize || dimensions[1] < 0 || static_cast(dimensions[1] * this->SizeOfInt) > this->FileSize || dimensions[2] < 0 || static_cast(dimensions[2] * this->SizeOfInt) > this->FileSize || numPts < 0 || static_cast(numPts*this->SizeOfInt) > this->FileSize) { return -1; } int *tempArray = new int[numPts]; this->ReadIntArray(tempArray, numPts); delete [] tempArray; } // reading next line to check for EOF lineRead = this->ReadLine(line); return lineRead; } // Internal function to read in a line up to 80 characters. // Returns zero if there was an error. int vtkEnSightGoldBinaryReader::ReadLine(char result[80]) { if (!this->GoldIFile->read(result, 80)) { // The read fails when reading the last part/array when there are no points. // I took out the error macro as a tempory fix. // We need to determine what EnSight does when the part with zero point // is not the last, and change the read array method. //int fixme; // I do not a file to test with yet. vtkDebugMacro("Read failed"); return 0; } // fix to the memory leakage problem detected by Valgrind result[79] = '\0'; if (this->Fortran) { strncpy(result, &result[4], 76); result[76] = 0; // better read an extra 8 bytes to prevent error next time char dummy[8]; if (!this->GoldIFile->read(dummy, 8)) { vtkDebugMacro("Read (fortran) failed"); return 0; } } return 1; } // Internal function to read a single integer. // Returns zero if there was an error. // Sets byte order so that part id is reasonable. int vtkEnSightGoldBinaryReader::ReadPartId(int *result) { // first swap like normal. if (this->ReadInt(result) == 0) { vtkErrorMacro("Read failed"); return 0; } // second: try an experimental byte swap. // Only experiment if byte order is not set. if (this->ByteOrder == FILE_UNKNOWN_ENDIAN) { int tmpLE = *result; int tmpBE = *result; vtkByteSwap::Swap4LE(&tmpLE); vtkByteSwap::Swap4BE(&tmpBE); if (tmpLE >= 0 && tmpLE < MAXIMUM_PART_ID) { this->ByteOrder = FILE_LITTLE_ENDIAN; *result = tmpLE; return 1; } if (tmpBE >= 0 && tmpBE < MAXIMUM_PART_ID) { this->ByteOrder = FILE_BIG_ENDIAN; *result = tmpBE; return 1; } vtkErrorMacro("Byte order could not be determined."); return 0; } return 1; } // Internal function to read a single integer. // Returns zero if there was an error. int vtkEnSightGoldBinaryReader::ReadInt(int *result) { char dummy[4]; if (this->Fortran) { if (!this->GoldIFile->read(dummy, 4)) { vtkErrorMacro("Read (fortran) failed."); return 0; } } if (!this->GoldIFile->read((char*)result, sizeof(int))) { vtkErrorMacro("Read failed"); return 0; } if (this->ByteOrder == FILE_LITTLE_ENDIAN) { vtkByteSwap::Swap4LE(result); } else if (this->ByteOrder == FILE_BIG_ENDIAN) { vtkByteSwap::Swap4BE(result); } if (this->Fortran) { if (!this->GoldIFile->read(dummy, 4)) { vtkErrorMacro("Read (fortran) failed."); return 0; } } return 1; } // Internal function to read an integer array. // Returns zero if there was an error. int vtkEnSightGoldBinaryReader::ReadIntArray(int *result, int numInts) { if (numInts <= 0) { return 1; } char dummy[4]; if (this->Fortran) { if (!this->GoldIFile->read(dummy, 4)) { vtkErrorMacro("Read (fortran) failed."); return 0; } } if (!this->GoldIFile->read((char*)result, sizeof(int)*numInts)) { vtkErrorMacro("Read failed."); return 0; } if (this->ByteOrder == FILE_LITTLE_ENDIAN) { vtkByteSwap::Swap4LERange(result, numInts); } else { vtkByteSwap::Swap4BERange(result, numInts); } if (this->Fortran) { if (!this->GoldIFile->read(dummy, 4)) { vtkErrorMacro("Read (fortran) failed."); return 0; } } return 1; } // Internal function to read a single vtkTypeInt64. // Returns zero if there was an error. int vtkEnSightGoldBinaryReader::ReadLong(vtkTypeInt64 *result) { char dummy[4]; if (this->Fortran) { if (!this->GoldIFile->read(dummy, 4)) { vtkErrorMacro("Read (fortran) failed."); return 0; } } if (!this->GoldIFile->read((char*)result, sizeof(vtkTypeInt64))) { vtkErrorMacro("Read failed"); return 0; } if (this->ByteOrder == FILE_LITTLE_ENDIAN) { vtkByteSwap::Swap8LE(result); } else if (this->ByteOrder == FILE_BIG_ENDIAN) { vtkByteSwap::Swap8BE(result); } if (this->Fortran) { if (!this->GoldIFile->read(dummy, 4)) { vtkErrorMacro("Read (fortran) failed."); return 0; } } return 1; } // Internal function to read a float array. // Returns zero if there was an error. int vtkEnSightGoldBinaryReader::ReadFloatArray(float *result, int numFloats) { if (numFloats <= 0) { return 1; } char dummy[4]; if (this->Fortran) { if (!this->GoldIFile->read(dummy, 4)) { vtkErrorMacro("Read (fortran) failed."); return 0; } } if (!this->GoldIFile->read((char*)result, sizeof(float)*numFloats)) { vtkErrorMacro("Read failed"); return 0; } if (this->ByteOrder == FILE_LITTLE_ENDIAN) { vtkByteSwap::Swap4LERange(result, numFloats); } else { vtkByteSwap::Swap4BERange(result, numFloats); } if (this->Fortran) { if (!this->GoldIFile->read(dummy, 4)) { vtkErrorMacro("Read (fortran) failed."); return 0; } } return 1; } //---------------------------------------------------------------------------- void vtkEnSightGoldBinaryReader::PrintSelf(ostream& os, vtkIndent indent) { this->Superclass::PrintSelf(os,indent); } // Seeks the IFile to the cached timestep nearest the target timestep. // Returns the actually sought to timestep //---------------------------------------------------------------------------- int vtkEnSightGoldBinaryReader::SeekToCachedTimeStep(const char* fileName, int realTimeStep) { typedef vtkEnSightGoldBinaryReader::FileOffsetMapInternal MapType; typedef MapType::const_iterator MapNameIterator; typedef MapType::value_type::second_type::const_iterator FileOffsetIterator; int j = 0; MapNameIterator nameIterator = this->FileOffsets->Map.find(fileName); if(nameIterator == this->FileOffsets->Map.end()) { return j; } // Try to find the nearest time step for which we know the offset for (int i = realTimeStep; i >= 0; i--) { FileOffsetIterator fileOffsetIterator = nameIterator->second.find(i); if (fileOffsetIterator != nameIterator->second.end()) { //we need to account for the last 80 characters as where we need to seek, //as we need to be at the BEGIN TIMESTEP keyword and not //the description line this->GoldIFile->seekg(fileOffsetIterator->second - 80l, ios::beg); j = i; break; } } return j; } //---------------------------------------------------------------------------- // Add a cached file offset void vtkEnSightGoldBinaryReader::AddTimeStepToCache(const char* fileName, int realTimeStep, vtkTypeInt64 address) { if (this->FileOffsets->Map.find(fileName) == this->FileOffsets->Map.end()) { std::map tsMap; this->FileOffsets->Map[fileName] = tsMap; } this->FileOffsets->Map[fileName][realTimeStep] = address; } //---------------------------------------------------------------------------- void vtkEnSightGoldBinaryReader::AddFileIndexToCache(const char* fileName) { // only read the file index if we have not searched for the file index before if (this->FileOffsets->Map.find(fileName) == this->FileOffsets->Map.end()) { char line[80]; vtkTypeInt64 addr; int numTS; // We add an empty map to prevent further attempts at reading the file index std::map tsMap; this->FileOffsets->Map[fileName] = tsMap; // Read the last 80 characters (+ a vtkTypeInt64) of the file and check for FILE_INDEX vtkIdType seekOffset = ( vtkIdType(-80) * static_cast(sizeof(char)) ) - static_cast(sizeof(vtkTypeInt64)); this->GoldIFile->seekg(seekOffset, ios::end); // right before the FILE_INDEX entry we might find the address of the index start this->ReadLong(&addr); this->ReadLine(line); if (strncmp(line, "FILE_INDEX", 10) == 0) { // jump to beginning of the index and add all time steps this->GoldIFile->seekg(addr, ios::beg); this->ReadInt(&numTS); for (int i = 0; i < numTS; ++i) { this->ReadLong(&addr); // The file index points at the description line, while VTK points at BEGIN TIMESTEP this->FileOffsets->Map[fileName][i] = addr; } } } this->GoldIFile->seekg(0l, ios::beg); }