Fix for the issue #2246 : Naming incorrect after translation.

[modules/shaper.git] / src / BuildPlugin / BuildPlugin_Edge.cpp

diff --git a/src/BuildPlugin/BuildPlugin_Edge.cpp b/src/BuildPlugin/BuildPlugin_Edge.cpp
index df60a8715655e72dbc2d27724f62285fff212a63..62a036a4b83c5fe99503bb395864c7cb68033cd4 100644 (file)
--- a/src/BuildPlugin/BuildPlugin_Edge.cpp
+++ b/src/BuildPlugin/BuildPlugin_Edge.cpp
@@ -1,14 +1,30 @@
-// Copyright (C) 2014-20xx CEA/DEN, EDF R&D
-
-// File:        BuildPlugin_Edge.cpp
-// Created:     18 April 2016
-// Author:      Dmitry Bobylev
+// Copyright (C) 2014-2017  CEA/DEN, EDF R&D
+//
+// 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<mailto:webmaster.salome@opencascade.com>
+//
 
 #include "BuildPlugin_Edge.h"
 
 #include <ModelAPI_AttributeSelectionList.h>
 #include <ModelAPI_ResultBody.h>
 
+#include <GeomAlgoAPI_Copy.h>
+
 //=================================================================================================
 BuildPlugin_Edge::BuildPlugin_Edge()
 {
@@ -59,9 +75,36 @@ void BuildPlugin_Edge::execute()
       return;
     }
 
+    // Copy shape.
+    GeomAlgoAPI_Copy aCopyAlgo(aShape);
+
+    // Check that algo is done.
+    if(!aCopyAlgo.isDone()) {
+      setError("Error: " + getKind() + " algorithm failed.");
+      return;
+    }
+
+    // Check if shape is not null.
+    if(!aCopyAlgo.shape().get() || aCopyAlgo.shape()->isNull()) {
+      setError("Error: Resulting shape is null.");
+      return;
+    }
+
+    // Check that resulting shape is valid.
+    if(!aCopyAlgo.isValid()) {
+      setError("Error: Resulting shape is not valid.");
+      return;
+    }
+
     // Store result.
     ResultBodyPtr aResultBody = document()->createBody(data(), aResultIndex);
-    aResultBody->store(aShape);
+    aResultBody->storeModified(aShape, aCopyAlgo.shape());
+    std::shared_ptr<GeomAPI_DataMapOfShapeShape> aSubShapes = aCopyAlgo.mapOfSubShapes();
+    int aModVertexTag = 1;
+    std::string aModVertexName = "Modified_Vertex";
+    aResultBody->loadAndOrientModifiedShapes(&aCopyAlgo, aShape, GeomAPI_Shape::VERTEX,
+                                             aModVertexTag, aModVertexName, *aSubShapes.get());
+
     setResult(aResultBody, aResultIndex);
     ++aResultIndex;
   }