1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
3 // File: ConstructionPlugin_Plane.cpp
4 // Created: 12 Dec 2014
5 // Author: Vitaly Smetannikov
7 #include "ConstructionPlugin_Plane.h"
9 #include <Config_PropManager.h>
11 #include <ModelAPI_AttributeSelection.h>
12 #include <ModelAPI_AttributeDouble.h>
13 #include <ModelAPI_ResultConstruction.h>
14 #include <ModelAPI_AttributeIntArray.h>
15 #include <GeomAlgoAPI_FaceBuilder.h>
17 #include <GeomAPI_Pnt2d.h>
20 #define PLANE_SIZE 300
22 ConstructionPlugin_Plane::ConstructionPlugin_Plane()
26 void ConstructionPlugin_Plane::initAttributes()
28 data()->addAttribute(ConstructionPlugin_Plane::FACE(), ModelAPI_AttributeSelection::type());
29 data()->addAttribute(ConstructionPlugin_Plane::DISTANCE(), ModelAPI_AttributeDouble::type());
32 void ConstructionPlugin_Plane::execute()
34 AttributeSelectionPtr aFaceAttr = data()->selection(ConstructionPlugin_Plane::FACE());
35 AttributeDoublePtr aDistAttr = data()->real(ConstructionPlugin_Plane::DISTANCE());
36 if ((aFaceAttr.get() != NULL) && (aDistAttr.get() != NULL) &&
37 aFaceAttr->isInitialized() && aDistAttr->isInitialized()) {
39 double aDist = aDistAttr->value();
40 GeomShapePtr aShape = aFaceAttr->value();
41 if (aShape.get() != NULL) {
42 std::shared_ptr<GeomAPI_Pln> aPln = GeomAlgoAPI_FaceBuilder::plane(aShape);
43 std::shared_ptr<GeomAPI_Pnt> aOrig = aPln->location();
44 std::shared_ptr<GeomAPI_Dir> aDir = aPln->direction();
46 aOrig->translate(aDir, aDist);
47 std::shared_ptr<GeomAPI_Pln> aNewPln =
48 std::shared_ptr<GeomAPI_Pln>(new GeomAPI_Pln(aOrig, aDir));
50 // Create rectangular face close to the selected
51 double aXmin, aYmin, Zmin, aXmax, aYmax, Zmax;
52 aShape->computeSize(aXmin, aYmin, Zmin, aXmax, aYmax, Zmax);
54 std::shared_ptr<GeomAPI_Pnt> aPnt1 =
55 std::shared_ptr<GeomAPI_Pnt>(new GeomAPI_Pnt(aXmin, aYmin, Zmin));
56 std::shared_ptr<GeomAPI_Pnt> aPnt2 =
57 std::shared_ptr<GeomAPI_Pnt>(new GeomAPI_Pnt(aXmax, aYmax, Zmax));
59 std::shared_ptr<GeomAPI_Pnt2d> aPnt2d1 = aPnt1->to2D(aNewPln);
60 std::shared_ptr<GeomAPI_Pnt2d> aPnt2d2 = aPnt2->to2D(aNewPln);
62 double aWidth = aPnt2d2->x() - aPnt2d1->x();
63 double aHeight = aPnt2d2->y() - aPnt2d1->y();
64 double aWgap = aWidth * 0.1;
65 double aHgap = aHeight * 0.1;
67 std::shared_ptr<GeomAPI_Shape> aPlane =
68 GeomAlgoAPI_FaceBuilder::planarFace(aNewPln, aPnt2d1->x() - aWgap, aPnt2d1->y() - aHgap,
69 aWidth + 2 * aWgap, aHeight + 2 * aHgap);
70 ResultConstructionPtr aConstr = document()->createConstruction(data());
71 aConstr->setShape(aPlane);
77 bool ConstructionPlugin_Plane::customisePresentation(ResultPtr theResult, AISObjectPtr thePrs,
78 std::shared_ptr<GeomAPI_ICustomPrs> theDefaultPrs)
80 std::vector<int> aColor;
81 // get color from the attribute of the result
82 if (theResult.get() != NULL && theResult->data()->attribute(ModelAPI_Result::COLOR_ID()).get() != NULL) {
83 AttributeIntArrayPtr aColorAttr = theResult->data()->intArray(ModelAPI_Result::COLOR_ID());
84 if (aColorAttr.get() && aColorAttr->size()) {
85 aColor.push_back(aColorAttr->value(0));
86 aColor.push_back(aColorAttr->value(1));
87 aColor.push_back(aColorAttr->value(2));
91 aColor = Config_PropManager::color("Visualization", "construction_plane_color",
92 ConstructionPlugin_Plane::DEFAULT_COLOR());
94 bool isCustomized = false;
95 if (aColor.size() == 3)
96 isCustomized = thePrs->setColor(aColor[0], aColor[1], aColor[2]);
98 isCustomized = thePrs->setTransparensy(0.6) || isCustomized;