1 // Copyright (C) 2014-2017 CEA/DEN, EDF R&D
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 // Lesser General Public License for more details.
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 // See http://www.salome-platform.org/ or
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
21 #include <GDMLPlugin_Ellipsoid.h>
23 #include <GeomAPI_ShapeExplorer.h>
25 #include <ModelAPI_Data.h>
26 #include <ModelAPI_ResultBody.h>
27 #include <ModelAPI_AttributeDouble.h>
28 #include <ModelAPI_AttributeString.h>
32 //=================================================================================================
33 GDMLPlugin_Ellipsoid::GDMLPlugin_Ellipsoid() // Nothing to do during instantiation
37 //=================================================================================================
38 void GDMLPlugin_Ellipsoid::initAttributes()
40 data()->addAttribute(GDMLPlugin_Ellipsoid::AX_ID(), ModelAPI_AttributeDouble::typeId());
41 data()->addAttribute(GDMLPlugin_Ellipsoid::BY_ID(), ModelAPI_AttributeDouble::typeId());
42 data()->addAttribute(GDMLPlugin_Ellipsoid::CZ_ID(), ModelAPI_AttributeDouble::typeId());
43 data()->addAttribute(GDMLPlugin_Ellipsoid::USE_ZCUT1_ID(), ModelAPI_AttributeString::typeId());
44 data()->addAttribute(GDMLPlugin_Ellipsoid::ZCUT1_ID(), ModelAPI_AttributeDouble::typeId());
45 data()->addAttribute(GDMLPlugin_Ellipsoid::USE_ZCUT2_ID(), ModelAPI_AttributeString::typeId());
46 data()->addAttribute(GDMLPlugin_Ellipsoid::ZCUT2_ID(), ModelAPI_AttributeDouble::typeId());
49 //=================================================================================================
50 void GDMLPlugin_Ellipsoid::execute()
52 std::shared_ptr<GeomAlgoAPI_Ellipsoid> anEllipsoidAlgo;
54 double aAx = real(AX_ID())->value();
55 double aBy = real(BY_ID())->value();
56 double aCz = real(CZ_ID())->value();
58 std::string useZCut1 = string(USE_ZCUT1_ID())->value();
59 std::string useZCut2 = string(USE_ZCUT2_ID())->value();
62 if (useZCut1.empty()) {
65 aZCut1 = real(ZCUT1_ID())->value();
68 if (useZCut2.empty()) {
71 aZCut2 = real(ZCUT2_ID())->value();
74 anEllipsoidAlgo = std::shared_ptr<GeomAlgoAPI_Ellipsoid>(
75 new GeomAlgoAPI_Ellipsoid(aAx, aBy, aCz, aZCut1, aZCut2));
77 // Check with that the arguments for anEllipsoidAlgo are correct
78 if (!anEllipsoidAlgo->check()) {
79 setError(anEllipsoidAlgo->getError(), false);
83 anEllipsoidAlgo->build();
85 // Check if the creation of the ellipsoid is correct
86 if (!anEllipsoidAlgo->isDone()) {
87 setError(anEllipsoidAlgo->getError(), false);
91 // Check if the created ellipsoid is valid
92 if (!anEllipsoidAlgo->checkValid("Ellipsoid builder")) {
93 setError(anEllipsoidAlgo->getError(), false);
98 ResultBodyPtr aResultEllipsoid = document()->createBody(data(), aResultIndex);
99 loadNamingDS(anEllipsoidAlgo, aResultEllipsoid);
100 setResult(aResultEllipsoid, aResultIndex);
104 //=================================================================================================
105 void GDMLPlugin_Ellipsoid::loadNamingDS(std::shared_ptr<GeomAlgoAPI_Ellipsoid> theEllipsoidAlgo,
106 std::shared_ptr<ModelAPI_ResultBody> theResultEllipsoid)
109 theResultEllipsoid->store(theEllipsoidAlgo->shape());
111 // Prepare the naming
112 theEllipsoidAlgo->prepareNamingFaces();
115 // Naming for faces and edges
117 std::map< std::string, std::shared_ptr<GeomAPI_Shape> > listOfFaces =
118 theEllipsoidAlgo->getCreatedFaces();
119 for (std::map< std::string, std::shared_ptr<GeomAPI_Shape> >::iterator
120 it=listOfFaces.begin(); it!=listOfFaces.end(); ++it) {
121 std::shared_ptr<GeomAPI_Shape> aFace = (*it).second;
122 theResultEllipsoid->generated(aFace, (*it).first, num++);
126 GeomAPI_DataMapOfShapeShape aVertices;
127 GeomAPI_ShapeExplorer aVertExp(theEllipsoidAlgo->shape(), GeomAPI_Shape::VERTEX);
128 for(int anIndex = 1; aVertExp.more(); aVertExp.next()) {
129 if (!aVertices.isBound(aVertExp.current())) {
130 std::ostringstream aStream;
131 aStream<<"Vertex_"<<anIndex++;
132 theResultEllipsoid->generated(aVertExp.current(), aStream.str(), num++);
133 aVertices.bind(aVertExp.current(), aVertExp.current());