wip: add load partitioned dataset for one partition

GEOS-DEV · Sep 10, 2024 · 0cc7fcb · 0cc7fcb
1 parent 8d73ee0
commit 0cc7fcb
Show file tree

Hide file tree

Showing 3 changed files with 38 additions and 8 deletions.
diff --git a/src/coreComponents/mesh/generators/VTKMeshGenerator.cpp b/src/coreComponents/mesh/generators/VTKMeshGenerator.cpp
@@ -103,14 +103,16 @@ void VTKMeshGenerator::postInputInitialization()
   GEOS_ERROR_IF ( m_repositoryName.empty() && !m_meshPath.empty(),
                   getName() << ": using a path in the repository for the mesh, you must supply a repository mode." );
 
-  ExternalDataRepositoryManager & externalDataManager = this->getGroupByPath< ExternalDataRepositoryManager >( "/Problem/ExternalDataRepository" );
-  // objectRepository.
-  m_repository = externalDataManager.getGroupPointer< VTKHierarchicalDataRepository >( m_repositoryName );
-
-  GEOS_THROW_IF( m_repository == nullptr,
-                 getName() << ": VTK Data Object Repository not found: " << m_repositoryName,
-                 InputError );
+  if( !m_repositoryName.empty())
+  {
+    ExternalDataRepositoryManager & externalDataManager = this->getGroupByPath< ExternalDataRepositoryManager >( "/Problem/ExternalDataRepository" );
+    // objectRepository.
+    m_repository = externalDataManager.getGroupPointer< VTKHierarchicalDataRepository >( m_repositoryName );
 
+    GEOS_THROW_IF( m_repository == nullptr,
+                   getName() << ": VTK Data Object Repository not found: " << m_repositoryName,
+                   InputError );
+  }
 
 }
 
@@ -136,7 +138,7 @@ void VTKMeshGenerator::fillCellBlockManager( CellBlockManager & cellBlockManager
     else if( !m_repositoryName.empty())
     {
       vtkSmartPointer< vtkPartitionedDataSet > partitions  = m_repository->search( m_meshPath );
-      // allMeshes = vtk::loadPartitions(partitions);
+      allMeshes = vtk::loadPartitions( partitions );
     }
 
 

diff --git a/src/coreComponents/mesh/generators/VTKUtilities.cpp b/src/coreComponents/mesh/generators/VTKUtilities.cpp
@@ -579,6 +579,25 @@ AllMeshes loadAllMeshes( Path const & filePath,
   return AllMeshes( main, faces );
 }
 
+//TODO handle more general cases: multiple partitions
+AllMeshes loadPartitions( vtkSmartPointer< vtkPartitionedDataSet > & partitions )
+{
+
+  if( MpiWrapper::commRank() == 0 )
+  {
+    std::map< string, vtkSmartPointer< vtkDataSet > > faces;
+
+    vtkDataObject * block = partitions->GetPartition( 0 );
+    if( block->IsA( "vtkDataSet" ) )
+    {
+      vtkSmartPointer< vtkDataSet > mesh = vtkDataSet::SafeDownCast( block );
+      return AllMeshes( mesh, faces );
+    }
+  }
+
+  return AllMeshes();
+}
+
 
 /**
  * @brief Redistributes the mesh using cell graphds methods (ParMETIS or PTScotch)

diff --git a/src/coreComponents/mesh/generators/VTKUtilities.hpp b/src/coreComponents/mesh/generators/VTKUtilities.hpp
@@ -24,6 +24,7 @@
 #include "common/MpiWrapper.hpp"
 #include "mesh/generators/CellBlockManager.hpp"
 
+#include <vtkPartitionedDataSet.h>
 #include <vtkDataSet.h>
 #include <vtkMultiProcessController.h>
 #include <vtkSmartPointer.h>
@@ -136,6 +137,14 @@ AllMeshes loadAllMeshes( Path const & filePath,
                          string const & mainBlockName,
                          array1d< string > const & faceBlockNames );
 
+/**
+ * @brief Load VTK partitions into the VTK data structure
+ *
+ * @param[in] partitions The partitioned dataset to be loaded
+ * @return The compound of the main mesh and the face block meshes.
+ */
+AllMeshes loadPartitions( vtkSmartPointer< vtkPartitionedDataSet > & partitions );
+
 /**
  * @brief Compute the rank neighbor candidate list.
  * @param[in] boundingBoxes the bounding boxes used by the VTK partitioner for all ranks