From d7706b6d4cb0e4eae0f7b87184a7c9a09616eea0 Mon Sep 17 00:00:00 2001
From: mpv <mpv@opencascade.com>
Date: Mon, 5 Dec 2016 16:41:57 +0300
Subject: [PATCH] Issue #1865 : finalization of export of fields to GEOM

---
 .../ConnectorPlugin_ExportFeature.py          | 73 +++++++++++++++++++
 src/Model/Model_AttributeTables.cpp           | 20 +++++
 src/Model/Model_AttributeTables.h             |  4 +
 src/ModelAPI/ModelAPI_AttributeTables.h       |  4 +
 4 files changed, 101 insertions(+)

diff --git a/src/ConnectorPlugin/ConnectorPlugin_ExportFeature.py b/src/ConnectorPlugin/ConnectorPlugin_ExportFeature.py
index a67c73633..5c9ec2bb4 100644
--- a/src/ConnectorPlugin/ConnectorPlugin_ExportFeature.py
+++ b/src/ConnectorPlugin/ConnectorPlugin_ExportFeature.py
@@ -155,6 +155,78 @@ class ExportFeature(ModelAPI.ModelAPI_Feature):
           self.geompy.UnionIDs(aGroup,Ids)
           self.geompy.addToStudyInFather(self.brep, aGroup, theGroupName)
 
+    ## Exports all fields
+    def exportFields(self):
+        # iterate all features to find fields
+        aFeaturesNum = self.Part.size("Features")
+        fieldIndex = 0
+        for anIndex in range(0, aFeaturesNum):
+            aFeature = self.Part.object("Features", anIndex)
+            aSelectionList = aFeature.data().selectionList("selected")
+            # if a field has been found
+            if aSelectionList:
+                aFeature = ModelAPI.objectToFeature(aFeature)
+                if aFeature.firstResult() is not None:
+                  aName = aFeature.firstResult().data().name()
+                fieldIndex = fieldIndex + 1
+                self.createFieldFromFeature(aFeature, aName)
+
+    ## Returns a type of the shape type in the old GEOM representation
+    def selectionDim(self, theSelectionType):
+        if theSelectionType == "vertex":
+            return 0
+        if theSelectionType == "edge":
+            return 1
+        if theSelectionType == "face":
+            return 2
+        if theSelectionType == "solid":
+            return 3
+        return -1
+
+    ## Creates a field by the field feature and the name
+    #  @param theField: the field feature
+    #  @param theFieldName: name of the field to create
+    def createFieldFromFeature(self, theField, theFieldName):
+        # iterate on all selected entities of the field
+        # and get the corresponding ID
+        aTables = theField.tables("values")
+        aSelection = theField.selectionList("selected")
+
+        # set component names
+        aComps = theField.stringArray("components_names")
+        aCompNames = []
+        aCompNum = aComps.size()
+        for aCompIndex in range(0, aCompNum):
+          aCompNames.append(aComps.value(aCompIndex))
+
+        #if len(Ids) <> 0:
+        aDim = self.selectionDim(aSelection.selectionType())
+        aResField = self.geompy.CreateField(self.brep, theFieldName, aTables.type(), aDim, aCompNames)
+        #self.geompy.UnionIDs(theField,Ids)
+        self.geompy.addToStudyInFather(self.brep, aResField, theFieldName)
+
+        # set steps
+        aStepsNum = aTables.tables()
+        for aStepIndex in range(0, aStepsNum):
+          aStamp = theField.intArray("stamps").value(aStepIndex)
+          aValues = []
+          aRows = aTables.rows()
+          aCols = aTables.columns()
+          for aRow in range(1, aRows): # no defaults
+            for aCol in range(0, aCols):
+              aVal = aTables.valueStr(aRow, aCol, aStepIndex)
+              if aTables.type() == 0: # bool
+                if aVal == "True":
+                  aVal = True
+                else:
+                  aVal = False
+              elif aTables.type() == 1: # int
+                aVal = int(aVal)
+              elif aTables.type() == 2: # double
+                aVal = float(aVal)
+              aValues.append(aVal)
+          aResField.addStep(aStepIndex + 1, aStamp, aValues)
+
     ## Exports all shapes and groups into the GEOM module.
     def execute(self):
         aSession = ModelAPI.ModelAPI_Session.get()
@@ -169,4 +241,5 @@ class ExportFeature(ModelAPI.ModelAPI_Feature):
         # Export bodies and groups
         self.exportBodies()
         self.exportGroups()
+        self.exportFields()
         pass
diff --git a/src/Model/Model_AttributeTables.cpp b/src/Model/Model_AttributeTables.cpp
index 6bfe82d07..b89e7446d 100644
--- a/src/Model/Model_AttributeTables.cpp
+++ b/src/Model/Model_AttributeTables.cpp
@@ -225,6 +225,26 @@ ModelAPI_AttributeTables::Value Model_AttributeTables::value(
   return aResult;
 }
 
+std::string Model_AttributeTables::valueStr(
+    const int theRow, const int theColumn, const int theTable)
+{
+  std::ostringstream aStr;
+  switch(myType) {
+  case ModelAPI_AttributeTables::DOUBLE:
+    aStr<<value(theRow, theColumn, theTable).myDouble;
+    break;
+  case ModelAPI_AttributeTables::BOOLEAN:
+    aStr<<(value(theRow, theColumn, theTable).myBool ? "True" :"False");
+    break;
+  case ModelAPI_AttributeTables::INTEGER:
+    aStr<<value(theRow, theColumn, theTable).myInt;
+    break;
+  case ModelAPI_AttributeTables::STRING:
+    aStr<<value(theRow, theColumn, theTable).myStr;
+    break;
+  }
+  return aStr.str();
+}
 
 //==================================================================================================
 Model_AttributeTables::Model_AttributeTables(TDF_Label& theLabel)
diff --git a/src/Model/Model_AttributeTables.h b/src/Model/Model_AttributeTables.h
index 62ac72254..3a036c648 100644
--- a/src/Model/Model_AttributeTables.h
+++ b/src/Model/Model_AttributeTables.h
@@ -52,6 +52,10 @@ public:
   MODEL_EXPORT virtual Value value(
     const int theRow, const int theColumn, const int theTable = 0);
 
+  /// Returns the value in the format of string (usefull for the python connection)
+  MODEL_EXPORT virtual std::string valueStr(
+    const int theRow, const int theColumn, const int theTable = 0);
+
 protected:
   /// Objects are created for features automatically
   MODEL_EXPORT Model_AttributeTables(TDF_Label& theLabel);
diff --git a/src/ModelAPI/ModelAPI_AttributeTables.h b/src/ModelAPI/ModelAPI_AttributeTables.h
index 072dc2fd0..789e192b9 100644
--- a/src/ModelAPI/ModelAPI_AttributeTables.h
+++ b/src/ModelAPI/ModelAPI_AttributeTables.h
@@ -63,6 +63,10 @@ public:
   MODELAPI_EXPORT virtual Value value(
     const int theRow, const int theColumn, const int theTable = 0) = 0;
 
+  /// Returns the value in the format of string (usefull for the python connection)
+  MODELAPI_EXPORT virtual std::string valueStr(
+    const int theRow, const int theColumn, const int theTable = 0) = 0;
+
   /// Returns the type of this class of attributes
   MODELAPI_EXPORT static std::string typeId()
   {
-- 
2.39.2