1 // File: PartSet_Presentation.h
2 // Created: 02 June 2014
3 // Author: Natalia ERMOLAEVA
5 #include <PartSet_Presentation.h>
6 #include <PartSet_Tools.h>
8 #include <ModelAPI_Feature.h>
9 #include <ModelAPI_Data.h>
11 #include <GeomDataAPI_Point.h>
12 #include <GeomDataAPI_Point2D.h>
13 #include <GeomDataAPI_Dir.h>
15 #include <SketchPlugin_Sketch.h>
16 #include <SketchPlugin_Line.h>
17 #include <SketchPlugin_Constraint.h>
18 #include <SketchPlugin_ConstraintLength.h>
20 #include <AIS_InteractiveObject.hxx>
21 #include <AIS_LengthDimension.hxx>
22 #include <AIS_Shape.hxx>
27 const Quantity_NameOfColor SKETCH_PLANE_COLOR = Quantity_NOC_CHOCOLATE; /// the plane edge color
28 const int SKETCH_WIDTH = 4; /// the plane edge width
30 Handle(AIS_InteractiveObject) PartSet_Presentation::createPresentation(
31 boost::shared_ptr<ModelAPI_Feature> theFeature,
32 boost::shared_ptr<ModelAPI_Feature> theSketch,
33 const TopoDS_Shape& theShape,
34 Handle_AIS_InteractiveObject thePrevPrs)
36 Handle(AIS_InteractiveObject) anAIS;
38 if (theFeature->getKind() == SKETCH_CONSTRAINT_LENGTH_KIND)
39 anAIS = createSketchConstraintLength(theFeature, theSketch, thePrevPrs);
41 anAIS = createFeature(theFeature, theShape, thePrevPrs);
42 if (theFeature->getKind() == SKETCH_KIND)
44 Handle(AIS_Shape) aShapeAIS = Handle(AIS_Shape)::DownCast(anAIS);
45 aShapeAIS->SetColor(Quantity_Color(SKETCH_PLANE_COLOR));
46 aShapeAIS->SetWidth(SKETCH_WIDTH);
47 aShapeAIS->Redisplay();
54 Handle(AIS_InteractiveObject) PartSet_Presentation::createFeature(
55 boost::shared_ptr<ModelAPI_Feature> theFeature,
56 const TopoDS_Shape& theShape,
57 Handle_AIS_InteractiveObject thePrevPrs)
59 Handle(AIS_InteractiveObject) anAIS = thePrevPrs;
62 Handle(AIS_Shape) aShapeAIS = Handle(AIS_Shape)::DownCast(anAIS);
63 if (!aShapeAIS.IsNull()) {
64 // if the AIS object is displayed in the opened local context in some mode, additional
65 // AIS sub objects are created there. They should be rebuild for correct selecting.
66 // It is possible to correct it by closing local context before the shape set and opening
67 // after. Another workaround to thrown down the selection and reselecting the AIS.
68 // If there was a problem here, try the first solution with close/open local context.
69 aShapeAIS->Set(theShape);
70 aShapeAIS->Redisplay(Standard_True);
75 anAIS = new AIS_Shape(theShape);
80 Handle(AIS_InteractiveObject) PartSet_Presentation::createSketchConstraintLength(
81 boost::shared_ptr<ModelAPI_Feature> theFeature,
82 boost::shared_ptr<ModelAPI_Feature> theSketch,
83 Handle(AIS_InteractiveObject) thePrevPrs)
85 if (!theFeature || !theSketch)
88 boost::shared_ptr<ModelAPI_Data> aData = theSketch->data();
89 boost::shared_ptr<GeomDataAPI_Point> anOrigin =
90 boost::dynamic_pointer_cast<GeomDataAPI_Point>(aData->attribute(SKETCH_ATTR_ORIGIN));
91 boost::shared_ptr<GeomDataAPI_Dir> aNormal =
92 boost::dynamic_pointer_cast<GeomDataAPI_Dir>(aData->attribute(SKETCH_ATTR_NORM));
93 gp_Pln aPlane(aNormal->x(), aNormal->y(), aNormal->z(), 0/*D*/);
95 aData = theFeature->data();
96 boost::shared_ptr<ModelAPI_AttributeRefAttr> anAttr =
97 boost::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(aData->attribute(CONSTRAINT_ATTR_ENTITY_A));
100 boost::shared_ptr<ModelAPI_Feature> aFeature = anAttr->feature();
101 if (!aFeature || aFeature->getKind() != SKETCH_LINE_KIND)
104 aData = aFeature->data();
105 if (!aData->isValid())
108 boost::shared_ptr<GeomDataAPI_Point2D> aPointStart =
109 boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(LINE_ATTR_START));
110 boost::shared_ptr<GeomDataAPI_Point2D> aPointEnd =
111 boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(LINE_ATTR_END));
113 gp_Pnt aPoint1, aPoint2;
114 PartSet_Tools::convertTo3D(aPointStart->x(), aPointStart->y(), theSketch, aPoint1);
115 PartSet_Tools::convertTo3D(aPointEnd->x(), aPointEnd->y(), theSketch, aPoint2);
117 //Build dimension here
118 gp_Pnt aP1 = aPoint1;
119 gp_Pnt aP2 = aPoint2;
121 Handle(AIS_InteractiveObject) anAIS = thePrevPrs;
124 Handle(AIS_LengthDimension) aLenDim = new AIS_LengthDimension (aP1, aP2, aPlane);
126 Handle(Prs3d_DimensionAspect) anAspect = new Prs3d_DimensionAspect();
127 anAspect->MakeArrows3d (Standard_False);
128 anAspect->MakeText3d(false/*is text 3d*/);
129 anAspect->TextAspect()->SetHeight(28);
130 anAspect->MakeTextShaded(false/*is test shaded*/);
131 aLenDim->DimensionAspect()->MakeUnitsDisplayed(false/*is units displayed*/);
132 /*if (isUnitsDisplayed)
134 aLenDim->SetDisplayUnits (aDimDlg->GetUnits ());
136 aLenDim->SetDimensionAspect (anAspect);
137 aLenDim->SetFlyout(12);
142 // update presentation
143 Handle(AIS_LengthDimension) aDimAIS = Handle(AIS_LengthDimension)::DownCast(anAIS);
144 if (!aDimAIS.IsNull()) {
145 aDimAIS->SetMeasuredGeometry(aPoint1, aPoint2, aPlane);
146 aDimAIS->Redisplay(Standard_True);