Salome HOME
3b4c4b950c4d645d2f844163f730110eb815196f
[modules/shaper.git] / src / BuildPlugin / BuildPlugin_Validators.cpp
1 // Copyright (C) 2014-2017  CEA/DEN, EDF R&D
2 //
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.
7 //
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.
12 //
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
16 //
17 // See http://www.salome-platform.org/ or
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
19 //
20
21 #include "BuildPlugin_Validators.h"
22
23 #include <ModelAPI_AttributeSelectionList.h>
24 #include <ModelAPI_ResultConstruction.h>
25
26 #include <GeomAPI_PlanarEdges.h>
27 #include <GeomAPI_Pln.h>
28 #include <GeomAPI_ShapeExplorer.h>
29 #include <GeomAPI_ShapeIterator.h>
30
31 #include <GeomAlgoAPI_CompoundBuilder.h>
32 #include <GeomAlgoAPI_PaveFiller.h>
33 #include <GeomAlgoAPI_ShapeBuilder.h>
34 #include <GeomAlgoAPI_ShapeTools.h>
35 #include <GeomAlgoAPI_SketchBuilder.h>
36 #include <GeomAlgoAPI_WireBuilder.h>
37 #include <GeomAlgoAPI_MakeVolume.h>
38 #include <GeomAlgoAPI_Tools.h>
39
40 #include <GeomValidators_FeatureKind.h>
41 #include <GeomValidators_ShapeType.h>
42
43 #include <Events_InfoMessage.h>
44
45 //=================================================================================================
46 bool BuildPlugin_ValidatorBaseForBuild::isValid(const AttributePtr& theAttribute,
47                                                 const std::list<std::string>& theArguments,
48                                                 Events_InfoMessage& theError) const
49 {
50   // Get base objects list.
51   if(theAttribute->attributeType() != ModelAPI_AttributeSelectionList::typeId()) {
52     std::string aMsg = "Error: BuildPlugin_ValidatorBaseForBuild does "
53                        "not support attribute type '%1'\nOnly '%2' is supported.";
54     Events_InfoMessage("BuildPlugin_Validators", aMsg).
55       arg(theAttribute->attributeType()).arg(ModelAPI_AttributeSelectionList::typeId()).send();
56     return false;
57   }
58   AttributeSelectionListPtr aSelectionList =
59     std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(theAttribute);
60   if(!aSelectionList.get()) {
61     theError = "Could not get selection list.";
62     return false;
63   }
64   if(aSelectionList->size() == 0) {
65     theError = "Empty selection list.";
66     return false;
67   }
68
69   // Collect base shapes.
70   for(int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
71     AttributeSelectionPtr aSelection = aSelectionList->value(anIndex);
72     if(!aSelection.get()) {
73       theError = "Could not get selection.";
74       return false;
75     }
76     ResultPtr aContext = aSelection->context();
77     if(!aContext.get()) {
78       theError = "Attribute have empty context.";
79       return false;
80     }
81
82     GeomShapePtr aShape = aSelection->value();
83     GeomShapePtr aContextShape = aContext->shape();
84     if(!aShape.get()) {
85       aShape = aContextShape;
86     }
87     if(!aShape.get()) {
88       theError = "Empty shape selected.";
89       return false;
90     }
91
92     // Check that shapes has acceptable type.
93     GeomValidators_ShapeType aValidatorShapeType;
94     if(!aValidatorShapeType.isValid(aSelection, theArguments, theError)) {
95       return false;
96     }
97
98     // Check that it is shape on sketch.
99     ResultConstructionPtr aConstruction =
100       std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(aContext);
101     if(aConstruction.get()) {
102       if(aConstruction->isInfinite()) {
103         theError = "Inifinte objects not acceptable.";
104         return false;
105       }
106
107       std::shared_ptr<GeomAPI_PlanarEdges> anEdges =
108         std::dynamic_pointer_cast<GeomAPI_PlanarEdges>(aContextShape);
109       if(anEdges.get()) {
110         if(aShape->isEqual(aContextShape)) {
111           // It is whole sketch.
112           return false;
113         }
114
115         continue;
116       }
117     }
118   }
119
120   return true;
121 }
122
123 //=================================================================================================
124 bool BuildPlugin_ValidatorBaseForWire::isValid(const std::shared_ptr<ModelAPI_Feature>& theFeature,
125                                                const std::list<std::string>& theArguments,
126                                                Events_InfoMessage& theError) const
127 {
128   // Get attribute.
129   if(theArguments.size() != 1) {
130     std::string aMsg = "Error: BuildPlugin_ValidatorBaseForWire should be used only "
131                        "with 1 parameter (ID of base objects list).";
132     Events_InfoMessage("BuildPlugin_Validators", aMsg).send();
133     return false;
134   }
135   AttributeSelectionListPtr aSelectionList = theFeature->selectionList(theArguments.front());
136   if(!aSelectionList.get()) {
137     theError = "Empty attribute \"%1\".";
138     theError.arg(theArguments.front());
139     return false;
140   }
141
142
143   // Collect base shapes.
144   ListOfShape aListOfShapes;
145   for(int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
146     AttributeSelectionPtr aSelection = aSelectionList->value(anIndex);
147     GeomShapePtr aShape = aSelection->value();
148     if(!aShape.get()) {
149       if (aSelection->context().get())
150         aShape = aSelection->context()->shape();
151     }
152     if (aShape.get())
153       aListOfShapes.push_back(aShape);
154   }
155
156   // Create wire.
157   GeomShapePtr aWire = GeomAlgoAPI_WireBuilder::wire(aListOfShapes);
158   if(!aWire.get()) {
159     theError = "Result wire empty. Probably it has disconnected edges or non-manifold.";
160     return false;
161   }
162
163   return true;
164 }
165
166 //=================================================================================================
167 bool BuildPlugin_ValidatorBaseForFace::isValid(const std::shared_ptr<ModelAPI_Feature>& theFeature,
168                                                const std::list<std::string>& theArguments,
169                                                Events_InfoMessage& theError) const
170 {
171   // Get attribute.
172   if(theArguments.size() != 1) {
173     std::string aMsg = "Error: BuildPlugin_ValidatorBaseForFace should be used only with "
174       "1 parameter (ID of base objects list).";
175     Events_InfoMessage("BuildPlugin_Validators", aMsg).send();
176     return false;
177   }
178   AttributeSelectionListPtr aSelectionList = theFeature->selectionList(theArguments.front());
179   if(!aSelectionList.get()) {
180     theError = "Empty attribute \"%1\".";
181     theError.arg(theArguments.front());
182     return false;
183   }
184
185   bool hasEdgesOrWires = false;
186   bool hasFaces = false;
187
188   // Collect base shapes.
189   ListOfShape anEdges;
190   for(int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
191     AttributeSelectionPtr aSelection = aSelectionList->value(anIndex);
192     GeomShapePtr aShape = aSelection->value();
193     if(!aShape.get()) {
194       if (!aSelection->context()) {
195         theError = "Objects are not selected.";
196         return false;
197       }
198       aShape = aSelection->context()->shape();
199     }
200     if (aShape->shapeType() == GeomAPI_Shape::FACE) {
201       // skip faces exploding
202       hasFaces = true;
203       continue;
204     }
205
206     for(GeomAPI_ShapeExplorer anExp(aShape, GeomAPI_Shape::EDGE); anExp.more(); anExp.next()) {
207       hasEdgesOrWires = true;
208       GeomShapePtr anEdge = anExp.current();
209       anEdges.push_back(anEdge);
210     }
211   }
212
213   if (hasFaces && hasEdgesOrWires) {
214     theError = "Faces and edges/wires should be selected together.";
215     return false;
216   } else if (hasEdgesOrWires && anEdges.empty()) {
217     theError = "Objects are not selected.";
218     return false;
219   }
220
221   // Check that edges does not have intersections.
222   if(anEdges.size() > 1) {
223     GeomAlgoAPI_PaveFiller aPaveFiller(anEdges, false);
224     if(!aPaveFiller.isDone()) {
225       theError = "Error while checking if edges intersects.";
226       return false;
227     }
228     GeomShapePtr aSectedEdges = aPaveFiller.shape();
229
230     int anEdgesNum = 0;
231     for(GeomAPI_ShapeExplorer
232         anExp(aSectedEdges, GeomAPI_Shape::EDGE); anExp.more(); anExp.next()) {
233       anEdgesNum++;
234     }
235     if(anEdgesNum != anEdges.size()) {
236       theError = "Selected objects have intersections.";
237       return false;
238     }
239   }
240
241   if (!anEdges.empty()) {
242     // Check that they are planar.
243     std::shared_ptr<GeomAPI_Pln> aPln = GeomAlgoAPI_ShapeTools::findPlane(anEdges);
244     if(!aPln.get()) {
245       theError = "Selected object(s) should belong to only one plane.";
246       return false;
247     }
248
249     // Check that selected objects have closed contours.
250     ListOfShape aFaces;
251     GeomAlgoAPI_SketchBuilder::createFaces(aPln->location(), aPln->xDirection(),
252                                            aPln->direction(), anEdges, aFaces);
253     if(aFaces.empty()) {
254       theError = "Selected objects do not generate closed contour.";
255       return false;
256     }
257   }
258
259   return true;
260 }
261
262 //=================================================================================================
263 bool BuildPlugin_ValidatorBaseForSolids::isValid(
264   const std::shared_ptr<ModelAPI_Feature>& theFeature, const std::list<std::string>& theArguments,
265   Events_InfoMessage& theError) const
266 {
267   // Get base objects list.
268   AttributeSelectionListPtr aSelectionList = theFeature->selectionList(theArguments.front());
269   if (!aSelectionList.get()) {
270     theError = "Could not get selection list.";
271     return false;
272   }
273   if (aSelectionList->size() == 0) {
274     theError = "Empty selection list.";
275     return false;
276   }
277
278   // Collect base shapes.
279   ListOfShape anOriginalShapes;
280   for (int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
281     AttributeSelectionPtr aSelection = aSelectionList->value(anIndex);
282     if (!aSelection->context().get()) {
283       theError = "Invalid selection.";
284       return false;
285     }
286     GeomShapePtr aShape = aSelection->value();
287     if (!aShape.get())
288       aShape = aSelection->context()->shape();
289     anOriginalShapes.push_back(aShape);
290   }
291
292   std::shared_ptr<GeomAlgoAPI_MakeVolume> anAlgorithm(
293     new GeomAlgoAPI_MakeVolume(anOriginalShapes, false));
294
295   std::string anErr;
296   if (GeomAlgoAPI_Tools::AlgoError::isAlgorithmFailed(anAlgorithm, "MakeVolume", anErr)) {
297     theError = anErr;
298     return false;
299   }
300
301   // set of allowed types of results
302   std::set<GeomAPI_Shape::ShapeType> aResultType;
303   std::string aType = theArguments.back();
304   if (aType == "solid")
305     aResultType.insert(GeomAPI_Shape::SOLID);
306   else if (aType == "compsolid") {
307     aResultType.insert(GeomAPI_Shape::COMPSOLID);
308     aResultType.insert(GeomAPI_Shape::SOLID);
309   }
310
311   GeomShapePtr aCompound = anAlgorithm->shape();
312   if (aCompound->shapeType() == GeomAPI_Shape::COMPOUND) {
313     GeomAPI_ShapeIterator anIt(aCompound);
314     GeomShapePtr aFoundSub;
315     for (; anIt.more() && !aFoundSub; anIt.next()) {
316       aFoundSub = anIt.current();
317       if (aResultType.count(aFoundSub->shapeType()) == 0) {
318         theError = "Unable to build a solid";
319         return false;
320       }
321     }
322     if (anIt.more() || !aFoundSub.get()) {
323       theError = "Unable to build a solid";
324       return false;
325     }
326   } else if (aResultType.count(aCompound->shapeType()) == 0) {
327     theError = "Unable to build a solid";
328     return false;
329   }
330   // check the internal faces presence
331   for(GeomAPI_ShapeExplorer aFaces(aCompound, GeomAPI_Shape::FACE); aFaces.more(); aFaces.next()) {
332     if (aFaces.current()->orientation() == GeomAPI_Shape::INTERNAL) {
333       theError = "Internal faces are not allowed in the resulting solid";
334       return false;
335     }
336   }
337
338   return true;
339 }
340
341
342 //=================================================================================================
343 bool BuildPlugin_ValidatorSubShapesSelection::isValid(const AttributePtr& theAttribute,
344                                                       const std::list<std::string>& theArguments,
345                                                       Events_InfoMessage& theError) const
346 {
347   if(theArguments.size() != 1) {
348     std::string aMsg = "Error: BuildPlugin_ValidatorSubShapesSelection should be used only with "
349       "1 parameter(Sketch feature id).";
350     Events_InfoMessage("BuildPlugin_Validators", aMsg).send();
351     return false;
352   }
353
354   // Get base objects list.
355   if(theAttribute->attributeType() != ModelAPI_AttributeSelectionList::typeId()) {
356     std::string aMsg =
357       "Error: BuildPlugin_ValidatorSubShapesSelection does not support attribute type \""
358       "%1\"\n Only \"%2\" supported.";
359     Events_InfoMessage("BuildPlugin_Validators", aMsg).
360       arg(theAttribute->attributeType()).arg(ModelAPI_AttributeSelectionList::typeId()).send();
361     return false;
362   }
363   AttributeSelectionListPtr aSelectionList =
364     std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(theAttribute);
365   if(!aSelectionList.get()) {
366     theError = "Could not get selection list.";
367     return false;
368   }
369
370   // Get base shape.
371   const std::string aBaseShapeId = "base_shape";
372   FeaturePtr aFeature = ModelAPI_Feature::feature(theAttribute->owner());
373   AttributeSelectionPtr aShapeAttrSelection = aFeature->selection(aBaseShapeId);
374
375   if(!aShapeAttrSelection.get()) {
376     theError = "Base shape is empty.";
377     return false;
378   }
379
380   ResultPtr aBaseContext = aShapeAttrSelection->context();
381
382   GeomShapePtr aBaseShape  = aShapeAttrSelection->value();
383   if(!aBaseShape.get()) {
384     theError = "Base shape is empty.";
385     return false;
386   }
387
388   GeomAlgoAPI_ShapeBuilder aBuilder;
389   aBuilder.removeInternal(aBaseShape);
390   aBaseShape = aBuilder.shape();
391
392   // If selected shape is wire allow to select only vertices. If face - allow vertices and edges.
393   std::set<GeomAPI_Shape::ShapeType> anAllowedTypes;
394   switch(aBaseShape->shapeType()) {
395     case GeomAPI_Shape::FACE: anAllowedTypes.insert(GeomAPI_Shape::EDGE);
396     case GeomAPI_Shape::WIRE: anAllowedTypes.insert(GeomAPI_Shape::VERTEX);
397     default: break;
398   }
399
400   // Check selected shapes.
401   GeomValidators_FeatureKind aFeatureKindValidator;
402   std::list<std::string> anArguments;
403   anArguments.push_back(theArguments.front());
404   for(int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
405     AttributeSelectionPtr aSelectionAttrInList = aSelectionList->value(anIndex);
406     if(!aSelectionAttrInList.get()) {
407       theError = "Empty attribute in list.";
408       return false;
409     }
410
411     // If context of selection same skip.
412     if(aBaseContext == aSelectionAttrInList->context()) {
413       continue;
414     }
415
416     // Check that it is a selection on Sketch.
417     if(!aFeatureKindValidator.isValid(aSelectionAttrInList, anArguments, theError)) {
418       return false;
419     }
420
421     // Check shape type.
422     GeomShapePtr aShapeInList = aSelectionAttrInList->value();
423     if(!aShapeInList.get()) {
424       aShapeInList = aSelectionAttrInList->context()->shape();
425     }
426     if(anAllowedTypes.find(aShapeInList->shapeType()) == anAllowedTypes.cend()) {
427       theError = "Selected shape has unacceptable type.";
428       return false;
429     }
430
431     // Check that shape inside wire or face.
432     if(!GeomAlgoAPI_ShapeTools::isSubShapeInsideShape(aShapeInList, aBaseShape)) {
433       theError = "Selected shape is not inside base face.";
434       return false;
435     }
436   }
437
438   return true;
439 }
440
441
442 //=================================================================================================
443 bool BuildPlugin_ValidatorFillingSelection::isValid(const AttributePtr& theAttribute,
444                                                       const std::list<std::string>& theArguments,
445                                                       Events_InfoMessage& theError) const
446 {
447   // Get base objects list.
448   if (theAttribute->attributeType() != ModelAPI_AttributeSelectionList::typeId()) {
449     std::string aMsg =
450       "Error: BuildPlugin_ValidatorFillingSelection does not support attribute type \""
451       "%1\"\n Only \"%2\" supported.";
452     Events_InfoMessage("BuildPlugin_Validators", aMsg).
453       arg(theAttribute->attributeType()).arg(ModelAPI_AttributeSelectionList::typeId()).send();
454     return false;
455   }
456   AttributeSelectionListPtr aSelectionList =
457     std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(theAttribute);
458   if (!aSelectionList.get()) {
459     theError = "Could not get selection list.";
460     return false;
461   }
462
463   FeaturePtr anOwner = ModelAPI_Feature::feature(theAttribute->owner());
464
465   // Check selected shapes.
466   for (int anIndex = 0; anIndex < aSelectionList->size(); ++anIndex) {
467     AttributeSelectionPtr aSelectionAttrInList = aSelectionList->value(anIndex);
468     if (!aSelectionAttrInList.get()) {
469       theError = "Empty attribute in list.";
470       return false;
471     }
472
473     // Check shape exists.
474     GeomShapePtr aShapeInList = aSelectionAttrInList->value();
475     if (!aShapeInList.get()) {
476       theError = "Object has no shape";
477       return false;
478     }
479
480     // Check shape type.
481     GeomAPI_Shape::ShapeType aType = aShapeInList->shapeType();
482     if (aType != GeomAPI_Shape::EDGE && aType != GeomAPI_Shape::WIRE) {
483       theError = "Incorrect objects selected";
484       return false;
485     }
486   }
487
488   return true;
489 }