Galerkin FEM for elliptic PDEs
Nelze vybrat více než 25 témat Téma musí začínat písmenem nebo číslem, může obsahovat pomlčky („-“) a může být dlouhé až 35 znaků.

ResampleWithDataset.cpp 3.3KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101
  1. /* This file is part of Lemma, a geophysical modelling and inversion API.
  2. * More information is available at http://lemmasoftware.org
  3. */
  4. /* This Source Code Form is subject to the terms of the Mozilla Public
  5. * License, v. 2.0. If a copy of the MPL was not distributed with this
  6. * file, You can obtain one at http://mozilla.org/MPL/2.0/.
  7. */
  8. /**
  9. * @file
  10. * @date 01/27/2016 01:24:24 PM
  11. * @version $Id$
  12. * @author Trevor Irons (ti)
  13. * @email tirons@egi.utah.edu
  14. * @copyright Copyright (c) 2016, University of Utah
  15. * @copyright Copyright (c) 2016, Trevor Irons & Lemma Software, LLC
  16. */
  17. #include <vtkProbeFilter.h>
  18. #include <vtkSTLReader.h>
  19. #include <vtkUnstructuredGrid.h>
  20. #include <vtkUnstructuredGridReader.h>
  21. #include <vtkUnstructuredGridWriter.h>
  22. #include <vtkGenericDataObjectReader.h>
  23. #include <vtkXMLUnstructuredGridReader.h>
  24. #include <vtkXMLUnstructuredGridWriter.h>
  25. #include <vtkDoubleArray.h>
  26. #include <vtkPointData.h>
  27. #include <cmath>
  28. #include <Eigen/Core>
  29. int main(int argc, char**argv) {
  30. std::cout << "Mesh boundary assigment routine\n";
  31. if (argc<4) {
  32. std::cout << "usage:\n" << "./ResampleWithDataset inMesh.vtu boundaries.stl outG.vtu" << std::endl;
  33. exit(EXIT_SUCCESS);
  34. }
  35. vtkUnstructuredGrid* uGrid;// = vtkUnstructuredGrid::New();
  36. std::string fn = argv[1];
  37. if(fn.substr(fn.find_last_of(".") + 1) == "vtk") {
  38. vtkGenericDataObjectReader* Reader = vtkGenericDataObjectReader::New();
  39. Reader->SetFileName(argv[1]);
  40. Reader->Update();
  41. if(Reader->IsFileUnstructuredGrid()) {
  42. std::cout << "Found Unstructured grid legacy file" << std::endl;
  43. uGrid = Reader->GetUnstructuredGridOutput();
  44. } else {
  45. std::cerr << "Unknown legacy format";
  46. exit(EXIT_FAILURE);
  47. }
  48. } else {
  49. vtkXMLUnstructuredGridReader* Reader = vtkXMLUnstructuredGridReader::New();
  50. std::cout << "Reading" << argv[1] << std::endl;
  51. Reader->SetFileName(argv[1]);
  52. Reader->Update();
  53. uGrid = Reader->GetOutput();
  54. }
  55. int nc = uGrid->GetNumberOfCells() ;
  56. int nn = uGrid->GetNumberOfPoints() ;
  57. std::cout << "Processing grid with nodes=" << nn << "\t cells=" << nc << "\n";
  58. vtkSTLReader* Stencil = vtkSTLReader::New();
  59. Stencil->SetFileName(argv[2]);
  60. Stencil->Update();
  61. vtkProbeFilter* Resample = vtkProbeFilter::New();
  62. //Resample->SetSourceData( uGrid );
  63. //Resample->SetInputData( Stencil->GetOutput() );
  64. Resample->SetInputData( uGrid );
  65. Resample->SetSourceData( Stencil->GetOutput() );
  66. Resample->SpatialMatchOn();
  67. Resample->SetValidPointMaskArrayName("HomogeneousDirichlet");
  68. Resample->Update();
  69. //std::cout << *Resample << std::endl;.
  70. std::cout << "Resampled grid, ncells=" << Resample->GetOutput()->GetNumberOfCells()
  71. << " npoints=" << Resample->GetOutput()->GetNumberOfPoints() << std::endl;
  72. vtkXMLUnstructuredGridWriter* Writer = vtkXMLUnstructuredGridWriter::New();
  73. Writer->SetInputData( Resample->GetOutput() );
  74. Writer->SetFileName( argv[3] );
  75. Writer->Write();
  76. Writer->Delete();
  77. Resample->Delete();
  78. //Reader->Delete();
  79. Stencil->Delete();
  80. //std::cout << *Stencil << std::endl;
  81. }