From: Jérôme <jerome.lucas@cesgenslab.fr>
Date: Mon, 14 Dec 2020 16:11:29 +0000 (+0100)
Subject: #20442 add new group in connector plugin to plublish feature
X-Git-Url: http://git.salome-platform.org/gitweb/?a=commitdiff_plain;h=827d651da83e8af8be98ee21e9915d9598ac30a0;p=modules%2Fshaper.git

#20442 add new group in connector plugin to plublish feature
---

diff --git a/src/ConnectorPlugin/ConnectorPlugin_PublishToStudyFeature.py b/src/ConnectorPlugin/ConnectorPlugin_PublishToStudyFeature.py
index a2e401aae..2f36d1ed4 100644
--- a/src/ConnectorPlugin/ConnectorPlugin_PublishToStudyFeature.py
+++ b/src/ConnectorPlugin/ConnectorPlugin_PublishToStudyFeature.py
@@ -160,51 +160,55 @@ class PublishToStudyFeature(ModelAPI.ModelAPI_Feature):
     # If theFields is true, the same is performed for Fields.
     def processGroups(self, theRes, theEngine, thePartFeatureId, theStudyShape, theFields):
       allGroupsProcessed = []
+      allRefGroups = []
       if theFields:
-        aRefGroups = ModelAPI.referencedFeatures(theRes, "Field", True)
+        allRefGroups.append(ModelAPI.referencedFeatures(theRes, "Field", True))
       else:
-        aRefGroups = ModelAPI.referencedFeatures(theRes, "Group", True)
-      for aRef in aRefGroups:
-        aGroupIndices = []
-        aGroupHasIndex = {}
-        aResShape = theRes.shape()
-        if theFields:
-          aSelList = aRef.selectionList("selected")
-        else:
-          aSelList = aRef.selectionList("group_list")
-        aSelType = GeomAPI_Shape.shapeTypeByStr(aSelList.selectionType())
-        for aGroupRes in aRef.results():
-          aShape = aGroupRes.shape()
-          anExplorer = GeomAPI_ShapeExplorer(aShape, aSelType)
-          while anExplorer.more():
-            anId = GeomAlgoAPI.GeomAlgoAPI_CompoundBuilder.id(aResShape, anExplorer.current())
-            if anId > 0 and not anId in aGroupHasIndex:
-              aGroupIndices.append(anId)
-              aGroupHasIndex[anId] = 0
-            anExplorer.next()
-        if len(aGroupIndices): # create group
-          aGroupFeatureId = aRef.data().featureId()
+        allRefGroups.append(ModelAPI.referencedFeatures(theRes, "Group", True))
+        allRefGroups.append(ModelAPI.referencedFeatures(theRes, "Shared_faces", True))
+        allRefGroups.append(ModelAPI.referencedFeatures(theRes, "Duplicated_faces", True))
+      for aRefGroups in allRefGroups:
+        for aRef in aRefGroups:
+          aGroupIndices = []
+          aGroupHasIndex = {}
+          aResShape = theRes.shape()
           if theFields:
-            aFieldOp = theEngine.GetIFieldOperations()
-            aGroupEntry = "field" + str(thePartFeatureId) + ":" + str(aGroupFeatureId)
-            aGroup = aFieldOp.FindField(theStudyShape, aGroupEntry)
+            aSelList = aRef.selectionList("selected")
           else:
-            aGroupOp = theEngine.GetIGroupOperations()
-            aGroupEntry = "group" + str(thePartFeatureId) + ":" + str(aGroupFeatureId)
-            aGroup = aGroupOp.FindGroup(theStudyShape, aGroupEntry)
-          if not aGroup: # create a new
+            aSelList = aRef.selectionList("group_list")
+          aSelType = GeomAPI_Shape.shapeTypeByStr(aSelList.selectionType())
+          for aGroupRes in aRef.results():
+            aShape = aGroupRes.shape()
+            anExplorer = GeomAPI_ShapeExplorer(aShape, aSelType)
+            while anExplorer.more():
+              anId = GeomAlgoAPI.GeomAlgoAPI_CompoundBuilder.id(aResShape, anExplorer.current())
+              if anId > 0 and not anId in aGroupHasIndex:
+                aGroupIndices.append(anId)
+                aGroupHasIndex[anId] = 0
+              anExplorer.next()
+          if len(aGroupIndices): # create group
+            aGroupFeatureId = aRef.data().featureId()
             if theFields:
-              aGroup = aFieldOp.CreateFieldByType(theStudyShape, aSelType)
+              aFieldOp = theEngine.GetIFieldOperations()
+              aGroupEntry = "field" + str(thePartFeatureId) + ":" + str(aGroupFeatureId)
+              aGroup = aFieldOp.FindField(theStudyShape, aGroupEntry)
             else:
-              aGroup = aGroupOp.CreateGroup(theStudyShape, aSelType)
-            aGroup.SetEntry(aGroupEntry)
-            theEngine.AddInStudy(aGroup, aRef.firstResult().data().name(), theStudyShape)
-          aGroup.SetSelection(aGroupIndices)
-          if theFields:
-            self.fillField(aGroup, aRef, theEngine, aGroupIndices)
-          # a group takes shape from the main result
-          #aGroup.SetShapeByStream(aRef.firstResult().shape().getShapeStream(False)) # group shape
-          allGroupsProcessed.append(aGroupEntry)
+              aGroupOp = theEngine.GetIGroupOperations()
+              aGroupEntry = "group" + str(thePartFeatureId) + ":" + str(aGroupFeatureId)
+              aGroup = aGroupOp.FindGroup(theStudyShape, aGroupEntry)
+            if not aGroup: # create a new
+              if theFields:
+                aGroup = aFieldOp.CreateFieldByType(theStudyShape, aSelType)
+              else:
+                aGroup = aGroupOp.CreateGroup(theStudyShape, aSelType)
+              aGroup.SetEntry(aGroupEntry)
+              theEngine.AddInStudy(aGroup, aRef.firstResult().data().name(), theStudyShape)
+            aGroup.SetSelection(aGroupIndices)
+            if theFields:
+              self.fillField(aGroup, aRef, theEngine, aGroupIndices)
+            # a group takes shape from the main result
+            #aGroup.SetShapeByStream(aRef.firstResult().shape().getShapeStream(False)) # group shape
+            allGroupsProcessed.append(aGroupEntry)
       # check all existing groups: if some does not processed, remove it from the tree
       aSOIter = SHAPERSTUDY_utils.getStudy().NewChildIterator(theStudyShape.GetSO())
       while aSOIter.More():
diff --git a/src/FeaturesPlugin/FeaturesPlugin_DuplicatedFaces.h b/src/FeaturesPlugin/FeaturesPlugin_DuplicatedFaces.h
index 61c753058..b58172879 100644
--- a/src/FeaturesPlugin/FeaturesPlugin_DuplicatedFaces.h
+++ b/src/FeaturesPlugin/FeaturesPlugin_DuplicatedFaces.h
@@ -55,10 +55,10 @@ public:
     return MY_NUMBER_FACES_ID;
   }
 
-  /// Attribute name for z coodinate.
+  /// Attribute name for list of faces.
   inline static const std::string& LIST_FACES_ID()
   {
-    static const std::string MY_LIST_FACES_ID("faces");
+    static const std::string MY_LIST_FACES_ID("group_list");
     return MY_LIST_FACES_ID;
   }
 
diff --git a/src/FeaturesPlugin/FeaturesPlugin_GroupDuplicatedFaces.h b/src/FeaturesPlugin/FeaturesPlugin_GroupDuplicatedFaces.h
index 583b770ec..e8d261cea 100644
--- a/src/FeaturesPlugin/FeaturesPlugin_GroupDuplicatedFaces.h
+++ b/src/FeaturesPlugin/FeaturesPlugin_GroupDuplicatedFaces.h
@@ -55,10 +55,10 @@ public:
     return MY_NUMBER_FACES_ID;
   }
 
-  /// Attribute name for z coodinate.
+  /// Attribute name for list of faces
   inline static const std::string& LIST_FACES_ID()
   {
-    static const std::string MY_LIST_FACES_ID("faces");
+    static const std::string MY_LIST_FACES_ID("group_list");
     return MY_LIST_FACES_ID;
   }
 
diff --git a/src/FeaturesPlugin/Test/TestCheckDuplicatedFaces.py b/src/FeaturesPlugin/Test/TestCheckDuplicatedFaces.py
index b5d18f47f..da07e6e64 100644
--- a/src/FeaturesPlugin/Test/TestCheckDuplicatedFaces.py
+++ b/src/FeaturesPlugin/Test/TestCheckDuplicatedFaces.py
@@ -68,7 +68,7 @@ if __name__ == '__main__':
     assert Duplicated_faces_1_Feature.error() == ''
     assert Duplicated_faces_1_Feature.name() == "Duplicated_faces_1"
 
-    aSelectionList = Duplicated_faces_1_Feature.selectionList("faces")
+    aSelectionList = Duplicated_faces_1_Feature.selectionList("group_list")
     assert aSelectionList.size() == 2
 
     assert(Part_1_doc.size("Groups") == 1)
diff --git a/src/FeaturesPlugin/duplicated_faces_macro_widget.xml b/src/FeaturesPlugin/duplicated_faces_macro_widget.xml
index df02d0131..1e9c24505 100644
--- a/src/FeaturesPlugin/duplicated_faces_macro_widget.xml
+++ b/src/FeaturesPlugin/duplicated_faces_macro_widget.xml
@@ -10,7 +10,7 @@
   </shape_selector>
   <groupbox title="Duplicated faces">
     <label id="number_duplicated_faces"/>
-    <multi_selector id="faces"
+    <multi_selector id="group_list"
                       label="List of faces :"
                       icon=""
                       shape_types="faces"
diff --git a/src/FeaturesPlugin/duplicated_faces_widget.xml b/src/FeaturesPlugin/duplicated_faces_widget.xml
index ccd2c420a..c5dcfa468 100644
--- a/src/FeaturesPlugin/duplicated_faces_widget.xml
+++ b/src/FeaturesPlugin/duplicated_faces_widget.xml
@@ -10,7 +10,7 @@
   </shape_selector>
   <groupbox title="Duplicated faces">
     <label id="number_duplicated_faces"/>
-    <multi_selector id="faces"
+    <multi_selector id="group_list"
                       label="List of faces :"
                       icon=""
                       shape_types="faces"