[EDF26451] : implementation of MEDFileMesh.addGroupsAtLevel

diff --git a/src/MEDLoader/MEDFileMesh.cxx b/src/MEDLoader/MEDFileMesh.cxx
index e61ac9c6dcfec2bb12861f1144abfd3d646029aa..9a2e20afef10a52fb9500d140a75beedbed98e0c 100644 (file)
--- a/src/MEDLoader/MEDFileMesh.cxx
+++ b/src/MEDLoader/MEDFileMesh.cxx
@@ -1989,6 +1989,17 @@ std::string MEDFileMesh::simpleRepr() const
   return oss.str();
 }
 
+/*!
+ * Simple loop over MEDFileMesh::addGroup method.
+ * 
+ * \sa MEDFileMesh::addGroup
+ */
+void MEDFileMesh::addGroupsAtLevel(int meshDimRelToMaxExt, const std::vector<const DataArrayIdType *>& grps)
+{
+  for(auto grp : grps)
+    this->addGroup(meshDimRelToMaxExt,grp);
+}
+
 /*!
  * This method is nearly like getFamilyFieldAtLevel method. Except that if the array does not exist at the specified level \a meshDimRelToMaxExt
  * an empty one is created.

diff --git a/src/MEDLoader/MEDFileMesh.hxx b/src/MEDLoader/MEDFileMesh.hxx
index 9eaa94f934e3204be8f1140d0d93af30fcd096f6..171f6aab2024e1630059d6312d47c181e3e14fb9 100644 (file)
--- a/src/MEDLoader/MEDFileMesh.hxx
+++ b/src/MEDLoader/MEDFileMesh.hxx
@@ -170,6 +170,7 @@ namespace MEDCoupling
     MEDLOADER_EXPORT virtual int getSpaceDimension() const = 0;
     MEDLOADER_EXPORT virtual std::string simpleRepr() const;
     MEDLOADER_EXPORT virtual std::string advancedRepr() const = 0;
+    MEDLOADER_EXPORT void addGroupsAtLevel(int meshDimRelToMaxExt, const std::vector<const DataArrayIdType *>& grps);
     //
     MEDLOADER_EXPORT virtual void setGroupsAtLevel(int meshDimRelToMaxExt, const std::vector<const DataArrayIdType *>& grps, bool renum=false);
     MEDLOADER_EXPORT virtual void setFamilyFieldArr(int meshDimRelToMaxExt, DataArrayIdType *famArr) = 0;

diff --git a/src/MEDLoader/Swig/MEDLoaderCommon.i b/src/MEDLoader/Swig/MEDLoaderCommon.i
index b33167ffcffb87bd8a1cb35a0e54313370965217..4befee3420805532d6c79a9b0fc64fb8fad38d33 100644 (file)
--- a/src/MEDLoader/Swig/MEDLoaderCommon.i
+++ b/src/MEDLoader/Swig/MEDLoaderCommon.i
@@ -1277,6 +1277,13 @@ namespace MEDCoupling
            self->setGroupsAtLevel(meshDimRelToMaxExt,grps,renum);
          }
 
+         void addGroupsAtLevel(int meshDimRelToMaxExt, PyObject *grps)
+         {
+           std::vector<const DataArrayIdType *> grpsCpp;
+           convertFromPyObjVectorOfObj<const MEDCoupling::DataArrayIdType *>(grps,SWIGTITraits<mcIdType>::TI,"DataArrayInt",grpsCpp);
+           self->addGroupsAtLevel(meshDimRelToMaxExt,grpsCpp);
+         }
+
          PyObject *areFamsEqual(const MEDFileMesh *other) const
          {
            std::string what;

diff --git a/src/MEDLoader/Swig/MEDLoaderTest4.py b/src/MEDLoader/Swig/MEDLoaderTest4.py
index c81ec4b0a6c03b3b0e6420fb0f378a4570d97012..c33c092d6c659fb2524364b7a0a623245ea827c6 100644 (file)
--- a/src/MEDLoader/Swig/MEDLoaderTest4.py
+++ b/src/MEDLoader/Swig/MEDLoaderTest4.py
@@ -5676,6 +5676,33 @@ class MEDLoaderTest4(unittest.TestCase):
         self.assertTrue( cas2.getGroupArr(0, "GNODE1").isEqualWithoutConsideringStr( DataArrayInt([]) ) ) # Expected to return empty array on both cases
         self.assertTrue( cas1.getGroupArr(1, "GCELL1").isEqualWithoutConsideringStr( DataArrayInt([]) ) ) # Expected to return empty array on both cases
 
+    def test45(self):
+        """
+        EDF26451 : addGroupsAtLevel
+        """
+        arr = DataArrayDouble([0,1,2,3])
+        cmesh = MEDCouplingCMesh("TEST")
+        cmesh.setCoords(arr, arr, arr)
+        umesh3d = cmesh.buildUnstructured()
+        umesh2d = umesh3d.computeSkin()
+
+        cas1 = MEDFileUMesh()
+
+        cas1[0] = umesh3d
+        cas1[-1] = umesh2d
+
+        gcell1 = DataArrayInt([1,2,4]) ; gcell1.setName("GCELL1")
+        cas1.setGroupsAtLevel(0, [gcell1])
+
+        gcell2 = DataArrayInt([2,5,6,7]) ; gcell2.setName("GCELL2")
+        gcell3 = DataArrayInt([5,16,17,26]) ; gcell3.setName("GCELL3")
+
+        cas1.addGroupsAtLevel(0,[gcell2,gcell3]) # <- the aim of the test is here
+        self.assertEqual( cas1.getGroupsOnSpecifiedLev(0) , ("GCELL1","GCELL2","GCELL3") )
+        self.assertTrue( cas1.getGroupArr(0,"GCELL1").isEqual(gcell1) )
+        self.assertTrue( cas1.getGroupArr(0,"GCELL2").isEqual(gcell2) )
+        self.assertTrue( cas1.getGroupArr(0,"GCELL3").isEqual(gcell3) )
+
     pass
 
 if __name__ == "__main__":