Integrate with acceleration of publishToStudy

wip: export publishToStudy feature into shaper model, test it

diff --git a/src/ConnectorPlugin/ConnectorPlugin_PublishToStudyFeature.py b/src/ConnectorPlugin/ConnectorPlugin_PublishToStudyFeature.py
index 15f3d7db71fdfd858343749b5238da253555f45b..efc9a9230caece4aa8bc924cfd8c95103e829f09 100644 (file)
--- a/src/ConnectorPlugin/ConnectorPlugin_PublishToStudyFeature.py
+++ b/src/ConnectorPlugin/ConnectorPlugin_PublishToStudyFeature.py
@@ -29,7 +29,6 @@ from salome.shaper import model
 import EventsAPI
 import ModelAPI
 import GeomAPI
-import GeomAlgoAPI
 
 
 # @ingroup Plugins
@@ -82,10 +81,7 @@ class PublishToStudyFeature(ModelAPI.ModelAPI_Feature):
                 )
             )
             currentFeature = self.partSet.currentFeature(True)
-            if (
-                lastFeature.data().featureId()
-                != currentFeature.data().featureId()
-            ):
+            if lastFeature.data().featureId() != currentFeature.data().featureId():
                 EventsAPI.Events_InfoMessage(
                     "PublishToStudy",
                     "Not all PartSet parts are up-to-date, nothing is "
@@ -108,15 +104,9 @@ class PublishToStudyFeature(ModelAPI.ModelAPI_Feature):
         Iterate all parts and all results to publish them in SHAPERSTUDY.
         """
         # TODO: iterate only on changed parts.
-        for partId in range(
-            self.partSet.size(model.ModelAPI_ResultPart.group())
-        ):
-            partObject = self.partSet.object(
-                model.ModelAPI_ResultPart.group(), partId
-            )
-            partRes = ModelAPI.modelAPI_ResultPart(
-                ModelAPI.modelAPI_Result(partObject)
-            )
+        for partId in range(self.partSet.size(model.ModelAPI_ResultPart.group())):
+            partObject = self.partSet.object(model.ModelAPI_ResultPart.group(), partId)
+            partRes = ModelAPI.modelAPI_ResultPart(ModelAPI.modelAPI_Result(partObject))
             partDoc = partRes.partDoc()
             if partDoc is None and partObject is not None:
                 EventsAPI.Events_InfoMessage(
@@ -197,9 +187,7 @@ class PublishToStudyFeature(ModelAPI.ModelAPI_Feature):
         processed = set()
         featuresIndices = {}
         for resId in range(partDoc.size(model.ModelAPI_ResultBody.group())):
-            resObject = partDoc.object(
-                model.ModelAPI_ResultBody.group(), resId
-            )
+            resObject = partDoc.object(model.ModelAPI_ResultBody.group(), resId)
             res = model.objectToResult(resObject)
             # do not export images
             if res.hasTexture() is True:
@@ -222,15 +210,9 @@ class PublishToStudyFeature(ModelAPI.ModelAPI_Feature):
             featuresIndices[resFeatureId] = 0
         sSEntry = str(self.currentPartFeatureId) + ":" + resFeatureId
         self.currentShaperShape = self.engine.FindOrCreateShape(sSEntry)
-        self.currentShaperShape.SetShapeByStream(
-            res.shape().getShapeStream(False)
-        )
-        if (
-            not self.currentShaperShape.GetSO()
-        ):  # publish in case it is a new shape
-            self.engine.AddInStudy(
-                self.currentShaperShape, res.data().name(), None
-            )
+        self.currentShaperShape.SetShapeByStream(res.shape().getShapeStream(False))
+        if not self.currentShaperShape.GetSO():  # publish in case it is a new shape
+            self.engine.AddInStudy(self.currentShaperShape, res.data().name(), None)
         else:  # restore a red reference if it was deleted
             done, sO2 = self.currentShaperShape.GetSO().FindSubObject(1)
             if done:
@@ -248,11 +230,9 @@ class PublishToStudyFeature(ModelAPI.ModelAPI_Feature):
         allGroupsProcessed = set()
         allRefGroups = []
         allRefGroups.append(ModelAPI.referencedFeatures(res, "Group", True))
-        allRefGroups.append(
-            ModelAPI.referencedFeatures(res, "Shared_faces", True)
-        )
+        allRefGroups.append(ModelAPI.referencedFeatures(res, "Shared_faces", True))
         resShape = res.shape()
-        mapOfShape = GeomAlgoAPI.GeomAPI_IndexedMapOfShape(resShape)
+        mapOfShape = GeomAPI.GeomAPI_IndexedMapOfShape(resShape)
         for refGroups in allRefGroups:
             for ref in refGroups:
                 groupsProcessed = self._processARefGroup(res.shape(), ref, mapOfShape)
@@ -272,7 +252,7 @@ class PublishToStudyFeature(ModelAPI.ModelAPI_Feature):
         allRefFields = []
         allRefFields.append(ModelAPI.referencedFeatures(res, "Field", True))
         resShape = res.shape()
-        mapOfShape = GeomAlgoAPI.GeomAPI_IndexedMapOfShape(resShape)
+        mapOfShape = GeomAPI.GeomAPI_IndexedMapOfShape(resShape)
         for refFields in allRefFields:
             for ref in refFields:
                 fieldsProcessed = self._processARefField(res.shape(), ref, mapOfShape)
@@ -341,16 +321,11 @@ class PublishToStudyFeature(ModelAPI.ModelAPI_Feature):
             fieldFeatureId = ref.data().featureId()
             fieldOp = self.engine.GetIFieldOperations()
             fieldEntry = (
-                "field"
-                + str(self.currentPartFeatureId)
-                + ":"
-                + str(fieldFeatureId)
+                "field" + str(self.currentPartFeatureId) + ":" + str(fieldFeatureId)
             )
             field = fieldOp.FindField(self.currentShaperShape, fieldEntry)
             if not field:  # create a new
-                field = fieldOp.CreateFieldByType(
-                    self.currentShaperShape, selType
-                )
+                field = fieldOp.CreateFieldByType(self.currentShaperShape, selType)
                 field.SetEntry(fieldEntry)
                 self.engine.AddInStudy(
                     field,
@@ -420,9 +395,7 @@ class PublishToStudyFeature(ModelAPI.ModelAPI_Feature):
                         vals.append(int(strVal))
                     elif valType == 2:  # double
                         vals.append(float(strVal))
-            field.AddFieldStep(
-                feature.intArray("stamps").value(val), val + 1, vals
-            )
+            field.AddFieldStep(feature.intArray("stamps").value(val), val + 1, vals)
 
     @staticmethod
     def _manageDeadGroups(sO):
@@ -443,9 +416,7 @@ class PublishToStudyFeature(ModelAPI.ModelAPI_Feature):
                 ),
             ):
                 if not group.GetEntry().startswith("dead"):
-                    deadGroupEntry = (
-                        "dead0" + str(index) + "_" + group.GetEntry()
-                    )
+                    deadGroupEntry = "dead0" + str(index) + "_" + group.GetEntry()
                     group.SetEntry(deadGroupEntry)
 
     def _removeNotProcessedType(self, allGroupsProcessed, Type):
@@ -468,3 +439,8 @@ class PublishToStudyFeature(ModelAPI.ModelAPI_Feature):
                         builder = SHAPERSTUDY_utils.getStudy().NewBuilder()
                         builder.RemoveObject(obj.GetSO())
             sOIter.Next()
+
+
+def publishToStudy():
+    feat = PublishToStudyFeature()
+    feat.execute()

diff --git a/src/ConnectorPlugin/Test/TestPublishToStudyAllGroupsAndFields.py b/src/ConnectorPlugin/Test/TestPublishToStudyAllGroupsAndFields.py
new file mode 100644 (file)
index 0000000..887ce40
--- /dev/null
+++ b/src/ConnectorPlugin/Test/TestPublishToStudyAllGroupsAndFields.py
@@ -0,0 +1,412 @@
+# Copyright (C) 2014-2023  CEA, EDF
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+#
+# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+#
+
+import salome
+
+salome.standalone()
+salome.salome_init()
+theStudy = salome.myStudy
+
+from salome.shaper import model
+
+from salome.geom import geomBuilder
+
+
+geompy = geomBuilder.New()
+
+import shaperBuilder
+shaperpy = shaperBuilder.New()
+
+## Get the last object published in the GEOM section of the object browser
+def getLastSHAPERSTUDYShape():
+    salome.myStudy.NewBuilder()
+    comp = salome.myStudy.FindComponent("SHAPERSTUDY")
+    obj = None
+    if comp:
+        iterator = salome.myStudy.NewChildIterator(comp)
+        sobj = None
+        while iterator.More():
+            sobj = iterator.Value()
+            iterator.Next()
+        if sobj:
+            obj = sobj.GetObject()
+    else:
+        raise Exception("SHAPERSTUDY component not found.")
+    return obj
+
+
+## Get the sub-object i of an object in the object browser
+# Numerotation starts at 1
+def getSubObject(obj, i):
+    ok, sub_sobj = salome.ObjectToSObject(obj).FindSubObject(i)
+    if not ok:
+        raise Exception("No child found at %i for %s" % (i, obj.GetName()))
+    sub_obj = sub_sobj.GetObject()
+    return sub_obj
+
+
+def testGroupsAndFieldsExportToStudy():
+    model.begin()
+    partSet = model.moduleDocument()
+    Part_1 = model.addPart(partSet)
+    Part_1_doc = Part_1.document()
+    model.addBox(Part_1_doc, 10, 10, 10)
+    model.addBox(Part_1_doc, 20, 20, 20)
+    Translation_1 = model.addTranslation(
+        Part_1_doc,
+        [model.selection("SOLID", "Box_1_1")],
+        model.selection("EDGE", "PartSet/OX"),
+        -10,
+    )
+    Translation_1.result().setName("Translation_1_1")
+    Partition_1 = model.addPartition(
+        Part_1_doc,
+        [
+            model.selection("SOLID", "Translation_1_1"),
+            model.selection("SOLID", "Box_2_1"),
+        ],
+    )
+    Partition_1.result().setName("Partition_1_1")
+    model.addBox(Part_1_doc, 10, 10, 10)
+    Translation_2 = model.addTranslation(
+        Part_1_doc,
+        [model.selection("SOLID", "Box_3_1")],
+        model.selection("EDGE", "PartSet/OX"),
+        20,
+    )
+    Translation_2.result().setName("Translation_2_1")
+    Group_1_objects = [
+        model.selection("FACE", "Translation_1_1/MF:Translated&Box_1_1/Top"),
+        model.selection("FACE", "Box_2_1/Top"),
+        model.selection("FACE", "Translation_2_1/MF:Translated&Box_3_1/Top"),
+    ]
+    Group_1 = model.addGroup(Part_1_doc, Group_1_objects)
+    Group_1.result().setName("faces_top")
+    Group_2_objects = [
+        model.selection(
+            "EDGE",
+            "[Partition_1_1_1/Modified_Face&Box_1_1/Left][Partition_1_1_1/Modified_Face&Box_1_1/Bottom]",
+        ),
+        model.selection(
+            "EDGE",
+            "[Partition_1_1_2/Modified_Face&Box_2_1/Left][Partition_1_1_2/Modified_Face&Box_2_1/Bottom]",
+        ),
+        model.selection(
+            "EDGE",
+            "[Translation_2_1/MF:Translated&Box_3_1/Left][Translation_2_1/MF:Translated&Box_3_1/Bottom]",
+        ),
+    ]
+    Group_2 = model.addGroup(Part_1_doc, Group_2_objects)
+    Group_2.result().setName("edges_x")
+    Group_3_objects = [
+        model.selection(
+            "VERTEX",
+            "[Translation_2_1/MF:Translated&Box_3_1/Front][Translation_2_1/MF:Translated&Box_3_1/Left][Translation_2_1/MF:Translated&Box_3_1/Bottom]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Translation_2_1/MF:Translated&Box_3_1/Front][Translation_2_1/MF:Translated&Box_3_1/Right][Translation_2_1/MF:Translated&Box_3_1/Bottom]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Translation_2_1/MF:Translated&Box_3_1/Back][Translation_2_1/MF:Translated&Box_3_1/Right][Translation_2_1/MF:Translated&Box_3_1/Bottom]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Box_2_1/Front][Box_2_1/Right][Partition_1_1_2/Modified_Face&Box_2_1/Bottom]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[(Partition_1_1_2/Modified_Face&Box_2_1/Back&Box_1_1/Front)(Partition_1_1_2/Modified_Face&Box_2_1/Left)(Box_2_1/Right)(Partition_1_1_2/Modified_Face&Box_2_1/Bottom)(Box_2_1/Top)][Box_2_1/Right][Partition_1_1_2/Modified_Face&Box_2_1/Bottom]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Partition_1_1_1/Modified_Face&Box_2_1/Back&Box_1_1/Front][Translation_1_1/MF:Translated&Box_1_1/Right][Partition_1_1_1/Modified_Face&Box_1_1/Bottom]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Translation_1_1/MF:Translated&Box_1_1/Back][Translation_1_1/MF:Translated&Box_1_1/Right][Partition_1_1_1/Modified_Face&Box_1_1/Bottom]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Translation_1_1/MF:Translated&Box_1_1/Back][Partition_1_1_1/Modified_Face&Box_1_1/Left][Partition_1_1_1/Modified_Face&Box_1_1/Bottom]",
+        ),
+    ]
+    Group_3 = model.addGroup(Part_1_doc, Group_3_objects)
+    Group_3.result().setName("vertices_bottom")
+    Group_4 = model.addGroup(
+        Part_1_doc,
+        [
+            model.selection("SOLID", "Partition_1_1_1"),
+            model.selection("SOLID", "Translation_2_1"),
+        ],
+    )
+    Group_4.result().setName("solids_small")
+    Field_1_objects = [
+        model.selection("SOLID", "Partition_1_1_1"),
+        model.selection("SOLID", "Partition_1_1_2"),
+        model.selection("SOLID", "Translation_2_1"),
+    ]
+    Field_1 = model.addField(Part_1_doc, 1, "DOUBLE", 1, ["Comp 1"], Field_1_objects)
+    Field_1.result().setName("Field_solids")
+    Field_1.addStep(0, 0, [[0], [1], [2], [3]])
+    Field_2_objects = [
+        model.selection("FACE", "Partition_1_1_1/Modified_Face&Box_1_1/Bottom"),
+        model.selection("FACE", "Partition_1_1_2/Modified_Face&Box_2_1/Bottom"),
+        model.selection("FACE", "Translation_2_1/MF:Translated&Box_3_1/Bottom"),
+        model.selection("FACE", "Translation_1_1/MF:Translated&Box_1_1/Top"),
+        model.selection("FACE", "Box_2_1/Top"),
+        model.selection("FACE", "Translation_2_1/MF:Translated&Box_3_1/Top"),
+    ]
+    Field_2 = model.addField(Part_1_doc, 1, "DOUBLE", 1, ["Comp 1"], Field_2_objects)
+    Field_2.result().setName("Field_faces")
+    Field_2.addStep(0, 0, [[0], [1], [1], [1], [2], [2], [2]])
+    Field_3_objects = [
+        model.selection(
+            "EDGE",
+            "[Translation_2_1/MF:Translated&Box_3_1/Front][Translation_2_1/MF:Translated&Box_3_1/Left]",
+        ),
+        model.selection(
+            "EDGE",
+            "[Translation_2_1/MF:Translated&Box_3_1/Left][Translation_2_1/MF:Translated&Box_3_1/Bottom]",
+        ),
+        model.selection(
+            "EDGE",
+            "[Partition_1_1_2/Modified_Face&Box_2_1/Left][Partition_1_1_2/Modified_Face&Box_2_1/Bottom]",
+        ),
+        model.selection(
+            "EDGE",
+            "[Partition_1_1_1/Modified_Face&Box_1_1/Left][Partition_1_1_1/Modified_Face&Box_1_1/Bottom]",
+        ),
+        model.selection(
+            "EDGE",
+            "[Translation_1_1/MF:Translated&Box_1_1/Back][Partition_1_1_1/Modified_Face&Box_1_1/Left]",
+        ),
+        model.selection(
+            "EDGE",
+            "[Translation_2_1/MF:Translated&Box_3_1/Front][Translation_2_1/MF:Translated&Box_3_1/Right]",
+        ),
+        model.selection(
+            "EDGE",
+            "[Translation_1_1/MF:Translated&Box_1_1/Back][Translation_1_1/MF:Translated&Box_1_1/Right]",
+        ),
+        model.selection("EDGE", "[Box_2_1/Front][Box_2_1/Right]"),
+        model.selection("EDGE", "[Box_2_1/Right][Box_2_1/Top]"),
+        model.selection(
+            "EDGE",
+            "[(Partition_1_1_2/Modified_Face&Box_2_1/Back&Box_1_1/Front)(Partition_1_1_2/Modified_Face&Box_2_1/Left)(Box_2_1/Right)(Partition_1_1_2/Modified_Face&Box_2_1/Bottom)(Box_2_1/Top)][Box_2_1/Right]",
+        ),
+        model.selection(
+            "EDGE", "[Box_2_1/Right][Partition_1_1_2/Modified_Face&Box_2_1/Bottom]"
+        ),
+    ]
+    Field_3 = model.addField(Part_1_doc, 1, "DOUBLE", 1, ["Comp 1"], Field_3_objects)
+    Field_3.result().setName("Field_edges")
+    Field_3.addStep(0, 0, [[0], [1], [1], [1], [2], [2], [2], [2], [3], [3], [3], [3]])
+    Field_4_objects = [
+        model.selection(
+            "VERTEX",
+            "[Translation_1_1/MF:Translated&Box_1_1/Back][Partition_1_1_1/Modified_Face&Box_1_1/Left][Partition_1_1_1/Modified_Face&Box_1_1/Bottom]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Translation_1_1/MF:Translated&Box_1_1/Back][Translation_1_1/MF:Translated&Box_1_1/Right][Partition_1_1_1/Modified_Face&Box_1_1/Bottom]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Partition_1_1_1/Modified_Face&Box_2_1/Back&Box_1_1/Front][Translation_1_1/MF:Translated&Box_1_1/Right][Partition_1_1_1/Modified_Face&Box_1_1/Bottom]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[(Partition_1_1_2/Modified_Face&Box_2_1/Back&Box_1_1/Front)(Partition_1_1_2/Modified_Face&Box_2_1/Left)(Box_2_1/Right)(Partition_1_1_2/Modified_Face&Box_2_1/Bottom)(Box_2_1/Top)][Box_2_1/Right][Partition_1_1_2/Modified_Face&Box_2_1/Bottom]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Box_2_1/Front][Box_2_1/Right][Partition_1_1_2/Modified_Face&Box_2_1/Bottom]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Translation_2_1/MF:Translated&Box_3_1/Back][Translation_2_1/MF:Translated&Box_3_1/Right][Translation_2_1/MF:Translated&Box_3_1/Bottom]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Translation_2_1/MF:Translated&Box_3_1/Front][Translation_2_1/MF:Translated&Box_3_1/Right][Translation_2_1/MF:Translated&Box_3_1/Bottom]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Translation_2_1/MF:Translated&Box_3_1/Front][Translation_2_1/MF:Translated&Box_3_1/Left][Translation_2_1/MF:Translated&Box_3_1/Bottom]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Translation_1_1/MF:Translated&Box_1_1/Back][Partition_1_1_1/Modified_Face&Box_1_1/Left][Translation_1_1/MF:Translated&Box_1_1/Top]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Translation_1_1/MF:Translated&Box_1_1/Back][Translation_1_1/MF:Translated&Box_1_1/Right][Translation_1_1/MF:Translated&Box_1_1/Top]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[(Partition_1_1_2/Modified_Face&Box_2_1/Back&Box_1_1/Front)(Partition_1_1_2/Modified_Face&Box_2_1/Left)(Box_2_1/Right)(Partition_1_1_2/Modified_Face&Box_2_1/Bottom)(Box_2_1/Top)][Partition_1_1_2/Modified_Face&Box_2_1/Left][Box_2_1/Top]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[(Partition_1_1_2/Modified_Face&Box_2_1/Back&Box_1_1/Front)(Partition_1_1_2/Modified_Face&Box_2_1/Left)(Box_2_1/Right)(Partition_1_1_2/Modified_Face&Box_2_1/Bottom)(Box_2_1/Top)][Box_2_1/Right][Box_2_1/Top]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Box_2_1/Front][Partition_1_1_2/Modified_Face&Box_2_1/Left][Box_2_1/Top]",
+        ),
+        model.selection("VERTEX", "[Box_2_1/Front][Box_2_1/Right][Box_2_1/Top]"),
+        model.selection(
+            "VERTEX",
+            "[Translation_2_1/MF:Translated&Box_3_1/Front][Translation_2_1/MF:Translated&Box_3_1/Left][Translation_2_1/MF:Translated&Box_3_1/Top]",
+        ),
+        model.selection(
+            "VERTEX",
+            "[Translation_2_1/MF:Translated&Box_3_1/Front][Translation_2_1/MF:Translated&Box_3_1/Right][Translation_2_1/MF:Translated&Box_3_1/Top]",
+        ),
+    ]
+    Field_4 = model.addField(Part_1_doc, 1, "DOUBLE", 1, ["Comp 1"], Field_4_objects)
+    Field_4.result().setName("Field_vertices")
+    Field_4.addStep(
+        0,
+        0,
+        [
+            [0],
+            [1],
+            [1],
+            [1],
+            [1],
+            [1],
+            [1],
+            [1],
+            [1],
+            [2],
+            [2],
+            [2],
+            [2],
+            [3],
+            [3],
+            [3],
+            [3],
+        ],
+    )
+    model.do()
+    model.publishToStudy()
+    model.end()
+
+
+# check the groups content by the coordinates of a point on its sub-shapes
+def checkGroupByCoords(group, coords, tolerance=1e-7):
+    for coord in coords:
+        p = geompy.MakeVertex(*coord)
+        assert geompy.MinDistance(group, p) < tolerance
+    pass
+
+
+## Check the result imported in GEOM
+def checkResultInSHAPERSTUDY():
+    geomObject_1 = getLastSHAPERSTUDYShape()  # the second, translation result
+    assert geomObject_1.GetName() == "Translation_2_1"
+    print(geomObject_1)
+
+    group_1_GEOM = getSubObject(geomObject_1, 1)
+    print(group_1_GEOM)
+    assert group_1_GEOM.GetName() == "faces_top"
+    assert (
+        geompy.NumberOfFaces(group_1_GEOM) == 1
+    )  # 2 faces are in the first result, only one here
+
+    # coordinates of the barycenters of the faces of Group_1
+    coords_1 = [[25, 5, 10]]
+    checkGroupByCoords(group_1_GEOM, coords_1)
+
+    group_2_GEOM = getSubObject(geomObject_1, 2)
+    assert group_2_GEOM.GetName() == "edges_x"
+    assert (
+        geompy.NumberOfEdges(group_2_GEOM) == 1
+    )  # 2 edges are in the first result, only one here
+
+    # coordinates of the barycenters of the edges of Group_2
+    coords_2 = [[25, 0, 0]]
+    checkGroupByCoords(group_2_GEOM, coords_2)
+
+    group_3_GEOM = getSubObject(geomObject_1, 3)
+    assert group_3_GEOM.GetName() == "vertices_bottom"
+    assert (
+        geompy.NumberOfSubShapes(group_3_GEOM, geompy.ShapeType["VERTEX"]) == 3
+    )  # 3 of 8
+
+    # coordinates of the points of of Group_3
+    coords_3 = [[20, 10, 0], [30, 10, 0], [30, 0, 0]]
+    checkGroupByCoords(group_3_GEOM, coords_3)
+
+    group_4_GEOM = getSubObject(geomObject_1, 4)
+    assert group_4_GEOM.GetName() == "solids_small"
+    assert geompy.NumberOfSolids(group_4_GEOM) == 1  # 1 of 2
+
+    # coordinates of the barycenters of the solids of Group_4
+    coords_4 = [[25, 5, 5]]
+    checkGroupByCoords(group_4_GEOM, coords_4)
+
+    field_1_GEOM = getSubObject(geomObject_1, 5)
+    assert field_1_GEOM.GetName() == "Field_solids"
+    assert field_1_GEOM.GetStep(1).GetValues() == [3.0]
+
+    field_2_GEOM = getSubObject(geomObject_1, 6)
+    assert field_2_GEOM.GetName() == "Field_faces"
+    assert field_2_GEOM.GetStep(1).GetValues() == [0.0, 0.0, 0.0, 0.0, 1.0, 2.0]
+
+    field_3_GEOM = getSubObject(geomObject_1, 7)
+    assert field_3_GEOM.GetName() == "Field_edges"
+    assert field_3_GEOM.GetStep(1).GetValues() == [
+        0.0,
+        0.0,
+        0.0,
+        0.0,
+        1.0,
+        0.0,
+        2.0,
+        0.0,
+        1.0,
+        0.0,
+        0.0,
+        0.0,
+    ]
+
+    field_4_GEOM = getSubObject(geomObject_1, 8)
+    assert field_4_GEOM.GetName() == "Field_vertices"
+    assert field_4_GEOM.GetStep(1).GetValues() == [
+        0.0,
+        0.0,
+        0.0,
+        1.0,
+        3.0,
+        1.0,
+        3.0,
+        1.0,
+    ]
+
+    pass
+
+
+if __name__ == "__main__":
+    # create 3 boxes with groups and fields
+    testGroupsAndFieldsExportToStudy()
+    # checkResultInSHAPERSTUDY()

diff --git a/src/PythonAPI/model/connection/__init__.py b/src/PythonAPI/model/connection/__init__.py
index 98039cb4840ba322353e35ad339e5d27c9a50762..7ea385a19b9f0ce2dc89458dd9c46dccba75c2f1 100644 (file)
--- a/src/PythonAPI/model/connection/__init__.py
+++ b/src/PythonAPI/model/connection/__init__.py
@@ -19,4 +19,5 @@
 """Package for Connection plugin for the Parametric Geometry API of the Modeler.
 """
 
-from ConnectorAPI import exportToGEOM
\ No newline at end of file
+from ConnectorAPI import exportToGEOM
+from ConnectorPlugin_PublishToStudyFeature import publishToStudy