1 // Copyright (C) 2014-2019 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 email : webmaster.salome@opencascade.com
20 #include <PrimitivesPlugin_Cylinder.h>
22 #include <GeomAPI_Dir.h>
23 #include <GeomAPI_Edge.h>
24 #include <GeomAPI_Lin.h>
25 #include <GeomAPI_ShapeIterator.h>
27 #include <GeomAlgoAPI_PointBuilder.h>
29 #include <ModelAPI_AttributeDouble.h>
30 #include <ModelAPI_AttributeSelection.h>
31 #include <ModelAPI_AttributeString.h>
32 #include <ModelAPI_ResultBody.h>
33 #include <ModelAPI_ResultConstruction.h>
34 #include <ModelAPI_Session.h>
36 //=================================================================================================
37 PrimitivesPlugin_Cylinder::PrimitivesPlugin_Cylinder()
41 //=================================================================================================
42 void PrimitivesPlugin_Cylinder::initAttributes()
44 data()->addAttribute(PrimitivesPlugin_Cylinder::CREATION_METHOD(),
45 ModelAPI_AttributeString::typeId());
47 data()->addAttribute(PrimitivesPlugin_Cylinder::BASE_POINT_ID(),
48 ModelAPI_AttributeSelection::typeId());
49 data()->addAttribute(PrimitivesPlugin_Cylinder::AXIS_ID(),
50 ModelAPI_AttributeSelection::typeId());
52 data()->addAttribute(PrimitivesPlugin_Cylinder::RADIUS_ID(),
53 ModelAPI_AttributeDouble::typeId());
54 data()->addAttribute(PrimitivesPlugin_Cylinder::HEIGHT_ID(),
55 ModelAPI_AttributeDouble::typeId());
56 data()->addAttribute(PrimitivesPlugin_Cylinder::ANGLE_ID(),
57 ModelAPI_AttributeDouble::typeId());
59 // Initialize the base point of the cylinder at the origin if the base point is not filled.
60 AttributeSelectionPtr aBasePoint = data()->selection(BASE_POINT_ID());
61 if (!aBasePoint->isInitialized()) {
62 ObjectPtr aPointObj = ModelAPI_Session::get()->moduleDocument()
63 ->objectByName(ModelAPI_ResultConstruction::group(), "Origin");
64 if (aPointObj.get()) {
65 ResultPtr aPointRes = std::dynamic_pointer_cast<ModelAPI_Result>(aPointObj);
66 aBasePoint->setValue(aPointRes, std::shared_ptr<GeomAPI_Shape>());
70 // Initialize the axis at the OZ axis if the axis is not filled.
71 AttributeSelectionPtr anAxis = data()->selection(AXIS_ID());
72 if (!anAxis->isInitialized()) {
73 ObjectPtr anAxisObj = ModelAPI_Session::get()->moduleDocument()
74 ->objectByName(ModelAPI_ResultConstruction::group(), "OZ");
75 if (anAxisObj.get()) {
76 ResultPtr anAxisRes = std::dynamic_pointer_cast<ModelAPI_Result>(anAxisObj);
77 anAxis->setValue(anAxisRes, std::shared_ptr<GeomAPI_Shape>());
82 //=================================================================================================
83 void PrimitivesPlugin_Cylinder::execute()
85 AttributeStringPtr aMethodTypeAttr = string(PrimitivesPlugin_Cylinder::CREATION_METHOD());
86 std::string aMethodType = aMethodTypeAttr->value();
88 if (aMethodType == CREATION_METHOD_CYLINDER()) {
89 createCylinder(false);
92 if (aMethodType == CREATION_METHOD_CYLINDER_PORTION()) {
97 //=================================================================================================
98 void PrimitivesPlugin_Cylinder::createCylinder(bool withAngle)
101 std::shared_ptr<GeomAPI_Pnt> aBasePoint;
102 std::shared_ptr<ModelAPI_AttributeSelection> aPointRef =
103 selection(PrimitivesPlugin_Cylinder::BASE_POINT_ID());
104 if (aPointRef.get() != NULL) {
105 GeomShapePtr aShape1 = aPointRef->value();
106 if (!aShape1.get()) {
107 aShape1 = aPointRef->context()->shape();
110 aBasePoint = GeomAlgoAPI_PointBuilder::point(aShape1);
115 static const std::string aSelectionError = "Error: The axis shape selection is bad.";
116 std::shared_ptr<ModelAPI_AttributeSelection> anEdgeRef = selection(AXIS_ID());
117 GeomShapePtr aShape = anEdgeRef->value();
119 if (anEdgeRef->context().get()) {
120 aShape = anEdgeRef->context()->shape();
124 setError(aSelectionError);
127 std::shared_ptr<GeomAPI_Edge> anEdge;
128 if (aShape->isEdge())
130 anEdge = aShape->edge();
132 else if (aShape->isCompound())
134 GeomAPI_ShapeIterator anIt(aShape);
135 anEdge = anIt.current()->edge();
139 setError(aSelectionError);
145 setError(aSelectionError);
149 std::shared_ptr<GeomAPI_Ax2> anAxis(new GeomAPI_Ax2(aBasePoint,
150 anEdge->line()->direction()));
153 // Getting radius and height
154 double aRadius = real(PrimitivesPlugin_Cylinder::RADIUS_ID())->value();
155 double aHeight = real(PrimitivesPlugin_Cylinder::HEIGHT_ID())->value();
157 std::shared_ptr<GeomAlgoAPI_Cylinder> aCylinderAlgo;
160 double anAngle = real(PrimitivesPlugin_Cylinder::ANGLE_ID())->value();
162 std::shared_ptr<GeomAlgoAPI_Cylinder>(new GeomAlgoAPI_Cylinder(anAxis,
167 std::shared_ptr<GeomAlgoAPI_Cylinder>(new GeomAlgoAPI_Cylinder(anAxis,
171 // These checks should be made to the GUI for the feature but
172 // the corresponding validator does not exist yet.
173 if (!aCylinderAlgo->check()) {
174 setError(aCylinderAlgo->getError());
178 // Build the cylinder
179 aCylinderAlgo->build();
181 // Check if the creation of the cylinder
182 if(!aCylinderAlgo->isDone()) {
183 // The error is not displayed in a popup window. It must be in the message console.
184 setError(aCylinderAlgo->getError());
187 if(!aCylinderAlgo->checkValid("Cylinder builder")) {
188 // The error is not displayed in a popup window. It must be in the message console.
189 setError(aCylinderAlgo->getError());
193 int aResultIndex = 0;
194 ResultBodyPtr aResultBox = document()->createBody(data(), aResultIndex);
195 loadNamingDS(aCylinderAlgo, aResultBox);
196 setResult(aResultBox, aResultIndex);
199 //=================================================================================================
200 void PrimitivesPlugin_Cylinder::loadNamingDS(std::shared_ptr<GeomAlgoAPI_Cylinder> theCylinderAlgo,
201 std::shared_ptr<ModelAPI_ResultBody> theResultCylinder)
204 theResultCylinder->store(theCylinderAlgo->shape());
206 // Prepare the naming
207 theCylinderAlgo->prepareNamingFaces();
210 std::map< std::string, std::shared_ptr<GeomAPI_Shape> > listOfFaces =
211 theCylinderAlgo->getCreatedFaces();
212 for (std::map< std::string, std::shared_ptr<GeomAPI_Shape> >::iterator it = listOfFaces.begin();
213 it != listOfFaces.end();
216 theResultCylinder->generated((*it).second, (*it).first);