00001 #include "SundanceBamgMeshReader.hpp"
00002 #include "SundanceOut.hpp"
00003 #include "SundanceExceptions.hpp"
00004 #include "Teuchos_StrUtils.hpp"
00005
00006 using namespace Teuchos;
00007 using namespace Sundance;
00008
00009
00010 BamgMeshReader::BamgMeshReader(const std::string& fname,
00011 const MeshType& meshType, bool bbAttr,
00012 const MPIComm& comm)
00013 : MeshReaderBase(fname, meshType, comm),
00014 nodeFilename_(filename()),
00015 elemFilename_(filename()),
00016 parFilename_(filename()),
00017 meshFilename_(filename()),
00018 bbFilename_(filename()),
00019 bbAttr_(bbAttr)
00020
00021 {
00022
00023
00024 if (nProc() > 1)
00025 {
00026 nodeFilename_ = nodeFilename_ + Teuchos::toString(myRank());
00027 parFilename_ = parFilename_ + Teuchos::toString(myRank());
00028 elemFilename_ = elemFilename_ + Teuchos::toString(myRank());
00029 }
00030
00031
00032 nodeFilename_ = nodeFilename_ + ".node";
00033 elemFilename_ = elemFilename_ + ".ele";
00034 parFilename_ = parFilename_ + ".par";
00035 meshFilename_ = meshFilename_ + ".mesh";
00036 bbFilename_ = bbFilename_ + ".bb";
00037
00038 setVerbosity( classVerbosity() );
00039 SUNDANCE_OUT(this->verb() > 1,
00040 "node filename = " << nodeFilename_);
00041
00042 SUNDANCE_OUT(this->verb() > 1,
00043 "elem filename = " << elemFilename_);
00044
00045 SUNDANCE_OUT(this->verb() > 1,
00046 "mesh filename = " << meshFilename_);
00047
00048 SUNDANCE_OUT(this->verb() > 1,
00049 "bb filename = " << bbFilename_);
00050
00051 }
00052 BamgMeshReader::BamgMeshReader(const ParameterList& params)
00053 : MeshReaderBase(params),
00054 nodeFilename_(filename()),
00055 elemFilename_(filename()),
00056 parFilename_(filename()),
00057 meshFilename_(filename())
00058 {
00059
00060
00061 if (nProc() > 1)
00062 {
00063 nodeFilename_ = nodeFilename_ + Teuchos::toString(myRank());
00064 parFilename_ = parFilename_ + Teuchos::toString(myRank());
00065 elemFilename_ = elemFilename_ + Teuchos::toString(myRank());
00066 }
00067
00068
00069 nodeFilename_ = nodeFilename_ + ".node";
00070 elemFilename_ = elemFilename_ + ".ele";
00071 parFilename_ = parFilename_ + ".par";
00072 meshFilename_ = meshFilename_ + ".mesh";
00073
00074 setVerbosity( classVerbosity() );
00075 SUNDANCE_OUT(this->verb() > 1,
00076 "node filename = " << nodeFilename_);
00077
00078 SUNDANCE_OUT(this->verb() > 1,
00079 "elem filename = " << elemFilename_);
00080
00081 SUNDANCE_OUT(this->verb() > 1,
00082 "mesh filename = " << meshFilename_);
00083
00084 SUNDANCE_OUT(this->verb() > 1,
00085 "bb filename = " << bbFilename_);
00086
00087 }
00088
00089
00090 Mesh BamgMeshReader::fillMesh() const
00091 {
00092 Mesh mesh;
00093
00094 Array<int> ptGID;
00095 Array<int> ptOwner;
00096 Array<int> cellGID;
00097 Array<int> cellOwner;
00098
00099 readParallelInfo(ptGID, ptOwner, cellGID, cellOwner);
00100
00101
00102
00103
00104
00105 mesh = readMesh(ptGID, ptOwner);
00106
00107
00108
00109 return mesh;
00110 }
00111
00112 void BamgMeshReader::readParallelInfo(Array<int>& ptGID,
00113 Array<int>& ptOwner,
00114 Array<int>& cellGID,
00115 Array<int>& cellOwner) const
00116 {
00117 int nPoints;
00118 int nElems;
00119 std::string line;
00120 Array<string> tokens;
00121 try
00122 {
00123 ptGID.resize(0);
00124 ptOwner.resize(0);
00125 cellGID.resize(0);
00126 cellOwner.resize(0);
00127
00128
00129
00130
00131 if (nProc() > 1)
00132 {
00133 RCP<std::ifstream> parStream
00134 = openFile(parFilename_, "parallel info");
00135
00136
00137
00138
00139 getNextLine(*parStream, line, tokens, '#');
00140
00141 TEST_FOR_EXCEPTION(tokens.length() != 2, RuntimeError,
00142 "TriangleMeshReader::getMesh() expects 2 entries "
00143 "on the first line of .par file. In "
00144 << parFilename_ << " I found \n[" << line << "]\n");
00145
00146 int np = atoi(tokens[1]);
00147 int pid = atoi(tokens[0]);
00148
00149
00150
00151
00152 TEST_FOR_EXCEPTION(np != nProc(), RuntimeError,
00153 "TriangleMeshReader::getMesh() found "
00154 "a mismatch between the current number of processors="
00155 << nProc() <<
00156 "and the number of processors=" << np
00157 << "in the file " << parFilename_);
00158
00159 TEST_FOR_EXCEPTION(pid != myRank(), RuntimeError,
00160 "TriangleMeshReader::getMesh() found "
00161 "a mismatch between the current processor rank="
00162 << myRank() << "and the processor rank="
00163 << pid << " in the file " << parFilename_);
00164
00165
00166 getNextLine(*parStream, line, tokens, '#');
00167
00168 TEST_FOR_EXCEPTION(tokens.length() != 1, RuntimeError,
00169 "TriangleMeshReader::getMesh() requires 1 entry "
00170 "on the second line of .par file. Found line \n["
00171 << line << "]\n in file " << parFilename_);
00172
00173 nPoints = StrUtils::atoi(tokens[0]);
00174
00175
00176 ptGID.resize(nPoints);
00177 ptOwner.resize(nPoints);
00178
00179 for (int i=0; i<nPoints; i++)
00180 {
00181 getNextLine(*parStream, line, tokens, '#');
00182
00183 TEST_FOR_EXCEPTION(tokens.length() != 3, RuntimeError,
00184 "TriangleMeshReader::getMesh() requires 3 "
00185 "entries on each line of the point section in "
00186 "the .par file. Found line \n[" << line
00187 << "]\n in file " << parFilename_);
00188
00189 ptGID[i] = StrUtils::atoi(tokens[1]);
00190 ptOwner[i] = StrUtils::atoi(tokens[2]);
00191 }
00192
00193
00194
00195
00196 getNextLine(*parStream, line, tokens, '#');
00197
00198 TEST_FOR_EXCEPTION(tokens.length() != 1, RuntimeError,
00199 "TriangleMeshReader::getMesh() requires 1 entry "
00200 "on the cell count line of .par file. Found line \n["
00201 << line << "]\n in file " << parFilename_);
00202
00203 nElems = StrUtils::atoi(tokens[0]);
00204
00205 SUNDANCE_OUT(this->verb() > 1,
00206 "read nElems = " << nElems);
00207
00208
00209
00210
00211 cellGID.resize(nElems);
00212 cellOwner.resize(nElems);
00213 for (int i=0; i<nElems; i++)
00214 {
00215 getNextLine(*parStream, line, tokens, '#');
00216
00217 TEST_FOR_EXCEPTION(tokens.length() != 3, RuntimeError,
00218 "TriangleMeshReader::getMesh() requires 3 "
00219 "entries on each line of the element section in "
00220 "the .par file. Found line \n[" << line
00221 << "]\n in file " << parFilename_);
00222
00223 cellGID[i] = StrUtils::atoi(tokens[1]);
00224 cellOwner[i] = StrUtils::atoi(tokens[2]);
00225 }
00226 }
00227
00228
00229 nPoints = ptGID.length();
00230 nElems = cellGID.length();
00231 }
00232 catch(std::exception& e)
00233 {
00234 SUNDANCE_TRACE(e);
00235 }
00236 }
00237
00238
00239
00240
00241 Mesh BamgMeshReader::readMesh(Array<int>& ptGID,
00242 Array<int>& ptOwner) const
00243 {
00244 Mesh mesh;
00245 std::string line;
00246 Array<string> tokens;
00247 int nPoints=0;
00248
00249
00250
00251
00252
00253
00254
00255
00256
00257
00258
00259
00260
00261
00262
00263
00264
00265
00266
00267
00268
00269
00270
00271
00272
00273
00274
00275
00276
00277
00278
00279
00280
00281
00282
00283
00284
00285
00286
00287
00288
00289
00290
00291
00292
00293
00294
00295
00296
00297
00298
00299
00300
00301
00302
00303
00304
00305
00306
00307
00308
00309
00310
00311
00312
00313
00314
00315
00316
00317
00318
00319
00320
00321
00322
00323
00324
00325
00326
00327
00328
00329
00330
00331
00332
00333
00334
00335
00336
00337
00338
00339
00340
00341 std::cerr << "starting to read meshFile" << std::endl;
00342
00343 RCP<std::ifstream> meshStream = openFile(meshFilename_,
00344 "node & elem info");
00345
00346 Array<string> liners = StrUtils::readFile(*meshStream, '#');
00347
00348 std::cerr << "defined liners" << std::endl;
00349
00350
00351 SUNDANCE_OUT(this->verb() > 3,
00352 "reading nodes from " + meshFilename_);
00353
00354
00355 int dimension=0;
00356 int dimensionIndex = 0;
00357 int lineIndex = 0;
00358 int verticesIndex = 0;
00359 int edgesIndex = 0;
00360 int triangleIndex = 0;
00361 int nCells=0;
00362 std::cerr << "DimensionIndex = " << dimensionIndex << std::endl;
00363
00364
00365 int linerssize = liners.length();
00366 std::cerr << "number of lines in liners = " << linerssize << std::endl;
00367
00368 for (int i = lineIndex; i < linerssize; i++)
00369 {
00370 tokens = StrUtils::stringTokenizer(liners[i]);
00371 std::cerr << "test: i = " << i << "tokens[0] = " << tokens[0] << std::endl;
00372 if (i < 20)
00373 {
00374 std::cerr << "i = " << i << ": number of tokens = " << tokens.length()
00375 << std::endl;
00376 for(int j = 0; j < tokens.length();j++)
00377 {
00378 std::cerr << " token " << j << " = " << tokens[j] << std::endl;
00379 std::cerr << " length of token[" << j << "] = "
00380 << tokens[j].length() << std::endl;
00381 }
00382 }
00383 if (tokens[0] == "Dimension")
00384 {
00385 dimensionIndex = i;
00386 std::cerr << "token[0] = Dimension" << std::endl;
00387 break;
00388 }
00389 }
00390 std::cerr << "DimensionIndex = " << dimensionIndex << std::endl;
00391 if (dimensionIndex > 0)
00392 {
00393 tokens = StrUtils::stringTokenizer(liners[dimensionIndex + 1]);
00394 dimension = atoi(tokens[0]);
00395 std::cerr << "dimension = " << dimension << std::endl;
00396 lineIndex = dimensionIndex + 2;
00397 }
00398
00399 for (int i = lineIndex; i < liners.size(); i++)
00400 {
00401 tokens = StrUtils::stringTokenizer(liners[i]);
00402 if (tokens[0] == "Vertices") verticesIndex = i;
00403 if (tokens[0] == "Vertices") break;
00404 }
00405 if (verticesIndex > 0)
00406 {
00407 tokens = StrUtils::stringTokenizer(liners[verticesIndex + 1]);
00408 nPoints = atoi(tokens[0]);
00409 std::cerr << "nPoints = " << nPoints << std::endl;
00410 ptGID.resize(nPoints);
00411 ptOwner.resize(nPoints);
00412 for (int i=0; i<nPoints; i++)
00413 {
00414 ptGID[i] = i;
00415 ptOwner[i] = 0;
00416 }
00417 lineIndex = verticesIndex + 2;
00418
00419 }
00420
00421 Array<int> ptIndices(nPoints);
00422 Array<int> rawIndices(nPoints);
00423
00424 Array<double> velVector(2*nPoints);
00425 int numbbAttr = 0;
00426
00427
00428
00429 if (bbAttr_)
00430 {
00431 RCP<std::ifstream> bbStream = openFile(bbFilename_,
00432 "velocity info");
00433 Array<string> bbliners = StrUtils::readFile(*bbStream, '#');
00434
00435
00436
00437
00438 int bbdimension;
00439 int bbnumSolns;
00440 int bbnumPoints=0;
00441 int bbsolnType;
00442 int bbdimensionIndex = 0;
00443 int bblineIndex = 0;
00444
00445 std::cerr << "bbDimensionIndex = " << bbdimensionIndex << std::endl;
00446
00447 int bblinerssize = bbliners.size();
00448 std::cerr << "number of lines in bbliners = " << bblinerssize << std::endl;
00449
00450 for (int i = bblineIndex; i < bblinerssize; i++)
00451 {
00452 Array<string> bbtokens = StrUtils::stringTokenizer(bbliners[i]);
00453 if(bbtokens.length() > 0)
00454 {
00455 if(bbtokens.length() != 4)
00456 {
00457 std::cerr <<
00458 "warning: .bb header line should have 4 tokens, read "
00459 << bbtokens.length() << std::endl;
00460 }
00461 std::cerr << "bbline " << i << ": read bbtokens " << bbtokens[0]
00462 << " " << bbtokens[1] << " " << bbtokens[2] << " "
00463 << bbtokens[3] << std::endl;
00464 bbdimensionIndex = i;
00465 break;
00466 }
00467 }
00468 bblineIndex = bbdimensionIndex + 1;
00469 std::cerr << "bblineIndex = " << bblineIndex << std::endl;
00470 if (bblineIndex > 0)
00471 {
00472 Array<string> bbtokens =
00473 StrUtils::stringTokenizer(bbliners[bbdimensionIndex]);
00474 bbdimension = StrUtils::atoi(bbtokens[0]);
00475 std::cerr << "bbdimension = " << bbdimension << std::endl;
00476 if (bbdimension != 2)
00477 std::cerr << "Error! bbdimension should be 2" << std::endl;
00478 bbnumSolns = StrUtils::atoi(bbtokens[1]);
00479 numbbAttr = bbnumSolns;
00480 std::cerr << "number of solutions per vertex = " << bbnumSolns << std::endl;
00481
00482
00483 bbnumPoints = StrUtils::atoi(bbtokens[2]);
00484 std::cerr << "number of vertices = " << bbnumPoints << std::endl;
00485 if (bbnumPoints != nPoints)
00486 std::cerr << "Error!! number of bb points != nPoints" << std::endl;
00487 bbsolnType = StrUtils::atoi(bbtokens[3]);
00488 std::cerr << "bbsolution type = " << bbsolnType << std::endl;
00489 }
00490
00491
00492
00493 for (int i = bblineIndex; i < bbnumPoints + bblineIndex; i++)
00494 {
00495 Array<string> bbtokens = StrUtils::stringTokenizer(bbliners[i]);
00496 if (bbtokens.length() == 0)
00497 std::cerr << "warning: encountered a blank soln line" << std::endl;
00498 int ii = i - bblineIndex;
00499
00500 velVector[ii] = StrUtils::atof(bbtokens[0]);
00501 velVector[ii+bbnumPoints] = StrUtils::atof(bbtokens[1]);
00502
00503
00504
00505 if( i < 5 || i > bbnumPoints - 5)
00506 {
00507
00508
00509
00510
00511 std::cerr << "vel0[" << ii << "] = " << velVector[ii] << std::endl;
00512 std::cerr << "vel1[" << ii << "] = " << velVector[ii+bbnumPoints]
00513 << std::endl;
00514 }
00515
00516 }
00517 }
00518
00519
00520
00521 int nAttributes = 0;
00522 if (bbAttr_) nAttributes = numbbAttr;
00523 std::cerr << "nAttributes = " << nAttributes << std::endl;
00524
00525
00526
00527
00528 mesh = createMesh(dimension);
00529 int count=0;
00530 int offset=0;
00531
00532 Array<bool> usedPoint(nPoints);
00533
00534
00535 nodeAttributes()->resize(nPoints);
00536
00537
00538
00539
00540 for (int i = lineIndex; i < liners.size(); i++)
00541
00542 {
00543 tokens = StrUtils::stringTokenizer(liners[i]);
00544 if (tokens.length() > 1)
00545 {
00546 double x = atof(tokens[0]);
00547 double y = atof(tokens[1]);
00548 count = i - lineIndex;
00549 rawIndices[count] = count;
00550 if ((i == lineIndex) || (i == lineIndex + 1) ||
00551 (i > lineIndex + nPoints - 2))
00552 {
00553 std::cerr << "i = " << i << "; node = (" << x << "," << y << ")"
00554 << std::endl;
00555 std::cerr << "count = " << count << std::endl;
00556 }
00557
00558 double z = 0.0;
00559 Point pt;
00560 int ptLabel = 0;
00561
00562 if (dimension==3)
00563 {
00564 z = atof(tokens[3]);
00565 pt = Point(x,y,z);
00566 }
00567 else
00568 {
00569 pt = Point(x,y);
00570 }
00571
00572 ptIndices[count]
00573 = mesh.addVertex(ptGID[count], pt, ptOwner[count], ptLabel);
00574
00575 (*nodeAttributes())[count].resize(nAttributes);
00576 for (int i=0; i<nAttributes; i++)
00577 {
00578
00579 if(i == 0) (*nodeAttributes())[count][i] = velVector[count];
00580 if(i == 1) (*nodeAttributes())[count][i] =
00581 velVector[count + nPoints];
00582 }
00583 }
00584
00585 if (tokens[0] == "Edges")
00586 {
00587 edgesIndex = i;
00588 break;
00589 }
00590 }
00591 std::cerr << "edgesIndex = " << edgesIndex << std::endl;
00592 std::cerr << "count = " << count << "; npoints - 1 = " << nPoints - 1 << std::endl;
00593
00594 if (count != nPoints - 1) std::cout << "error: # of nodes != # of vertices"
00595 << std::endl;
00596 else std::cerr << "successfully read node data" << std::endl;
00597 lineIndex = edgesIndex + 1;
00598
00599
00600
00601 std::cerr << "lineIndex = " << lineIndex << "; triangleIndex = "
00602 << triangleIndex << std::endl;
00603 for (int i = lineIndex; i < liners.size(); i++)
00604 {
00605 tokens = StrUtils::stringTokenizer(liners[i]);
00606 if (tokens[0] == "Triangles") {triangleIndex = i; break;}
00607 if (tokens[0] == "CrackedEdges") {triangleIndex = i+2; break;}
00608
00609 }
00610 std::cerr << "lineIndex = " << lineIndex << "; triangleIndex = "
00611 << triangleIndex << std::endl;
00612 std::cerr << "tokens[0] = " << tokens[0] << std::endl;
00613
00614 lineIndex = triangleIndex + 1;
00615
00616 Array<int> elemGID;
00617 Array<int> elemOwner;
00618
00619 if (triangleIndex > 0)
00620 {
00621 tokens = StrUtils::stringTokenizer(liners[lineIndex]);
00622 std::cerr << "lineIndex = " << lineIndex << "; tokens[0] = " << tokens[0]
00623 << std::endl;
00624 nCells = atoi(tokens[0]);
00625 elemGID.resize(nCells);
00626 elemOwner.resize(nCells);
00627 std::cerr << "nCells = " << nCells << std::endl;
00628 for (int i=0; i<nCells; i++)
00629 {
00630 elemGID[i] = i;
00631 elemOwner[i] = 0;
00632 }
00633 lineIndex = lineIndex + 1;
00634 }
00635
00636
00637
00638
00639
00640
00641
00642
00643
00644
00645
00646
00647
00648
00649
00650
00651
00652
00653
00654
00655
00656
00657
00658
00659
00660
00661
00662
00663
00664
00665
00666
00667
00668
00669
00670
00671
00672
00673
00674
00675
00676
00677
00678
00679
00680
00681
00682
00683
00684
00685
00686
00687
00688
00689
00690
00691
00692
00693
00694
00695
00696
00697
00698
00699
00700
00701
00702
00703
00704
00705
00706
00707
00708
00709
00710
00711
00712
00713
00714
00715
00716
00717
00718
00719
00720
00721
00722
00723
00724
00725
00726
00727
00728
00729
00730
00731
00732
00733
00734 std::cerr << "# of triangles = " << nCells
00735 << "; starting to read triangle data" << std::endl;
00736 SUNDANCE_OUT(this->verb() > 3,
00737 "done reading nodes, ready to read elements from " + meshFilename_);
00738
00739 int nElems = nCells;
00740 int ptsPerElem = dimension + 1;
00741 elemGID.resize(nElems);
00742 elemOwner.resize(nElems);
00743 offset = 1;
00744
00745 for (int i=0; i<nElems; i++)
00746 {
00747 elemGID[i] = i;
00748 elemOwner[i] = 0;
00749 }
00750 nAttributes = 0;
00751
00752 elemAttributes()->resize(nElems);
00753 count = 0;
00754
00755 int dim = mesh.spatialDim();
00756 Array<int> nodes(dim+1);
00757 if (dim != dimension) std::cerr << "ERROR: dim = " << dim << "!= dimension = "
00758 << dimension << std::endl;
00759
00760 std::cerr << "lineIndex = " << lineIndex << std::endl;
00761 std::cerr << "size of liners = " << liners.size() << std::endl;
00762 std::cerr << "lineIndex+nCells-1 = " << lineIndex+nCells-1 << std::endl;
00763 for (int i = lineIndex; i < liners.size();i++)
00764
00765 {
00766 tokens = StrUtils::stringTokenizer(liners[i]);
00767 if (tokens.length() > 1)
00768 {
00769 if (i < lineIndex + 5) std::cerr << "i = " << i << " first triangles: "
00770 << tokens[0] << " " << tokens[1]
00771 << " " << tokens[2] << std::endl;
00772 if (i == lineIndex+nCells-1) std::cerr << "i = " << i
00773 << " last triangle: " << tokens[0]
00774 << " " << tokens[1] << " "
00775 << tokens[2] << std::endl;
00776
00777 for (int d=0; d<=dim; d++)
00778 {
00779
00780
00781
00782 nodes[d] = ptGID[atoi(tokens[d])-offset];
00783
00784
00785 }
00786
00787 count = i - lineIndex;
00788 int elemLabel = 0;
00789 mesh.addElement(elemGID[count], nodes, elemOwner[count], elemLabel);
00790
00791 (*elemAttributes())[count].resize(nAttributes);
00792 for (int i=0; i<nAttributes; i++)
00793 {
00794
00795
00796
00797 (*elemAttributes())[count][i] = atof(tokens[ptsPerElem+i]);
00798
00799
00800 }
00801 }
00802 if (tokens[0] == "SubDomainFromMesh") break;
00803 }
00804
00805 std::cerr << "# of cells read = " << count + 1 << std::endl;
00806 if (count != nCells - 1) std::cerr << "error: # of cells read != nCells" << std::endl;
00807
00808
00809
00810 return mesh;
00811 }
00812
00813
00814
00815
00816
00817
00818
00819
00820
00821
00822
00823
00824
00825
00826
00827
00828
00829
00830
00831
00832
00833
00834
00835
00836
00837
00838
00839
00840
00841
00842
00843
00844
00845
00846
00847
00848
00849
00850
00851
00852
00853
00854
00855
00856
00857
00858
00859
00860
00861
00862
00863
00864
00865
00866
00867
00868
00869
00870
00871
00872
00873
00874
00875
00876
00877
00878
00879
00880
00881
00882
00883
00884
00885
00886
00887
00888
00889
00890
00891
00892
00893
00894
00895
00896
00897
00898
00899
00900
00901
00902
00903
00904
00905
00906
00907
00908