1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
3 // File: ModelAPI_ResultPart.cpp
4 // Created: 07 Jul 2014
5 // Author: Mikhail PONIKAROV
7 #include <Model_ResultPart.h>
8 #include <ModelAPI_Data.h>
9 #include <Model_Data.h>
10 #include <ModelAPI_AttributeDocRef.h>
11 #include <ModelAPI_Session.h>
12 #include <ModelAPI_Feature.h>
13 #include <ModelAPI_ResultBody.h>
14 #include <ModelAPI_AttributeIntArray.h>
15 #include <ModelAPI_AttributeSelectionList.h>
16 #include <ModelAPI_AttributeReference.h>
17 #include <Model_Document.h>
18 #include <Events_Loop.h>
19 #include <ModelAPI_Events.h>
21 #include <TNaming_Tool.hxx>
22 #include <TNaming_NamedShape.hxx>
23 #include <TNaming_Iterator.hxx>
24 #include <TDataStd_Name.hxx>
25 #include <TopoDS_Compound.hxx>
26 #include <BRep_Builder.hxx>
27 #include <TopExp_Explorer.hxx>
29 #define baseRef() std::dynamic_pointer_cast<Model_ResultPart>(data()->reference(BASE_REF_ID())->value())
31 void Model_ResultPart::initAttributes()
33 // append the color attribute. It is empty, the attribute will be filled by a request
34 data()->addAttribute(DOC_REF(), ModelAPI_AttributeDocRef::typeId());
35 data()->addAttribute(COLOR_ID(), ModelAPI_AttributeIntArray::typeId());
36 data()->addAttribute(BASE_REF_ID(), ModelAPI_AttributeReference::typeId());
39 std::shared_ptr<ModelAPI_Document> Model_ResultPart::partDoc()
42 return baseRef()->partDoc();
44 DocumentPtr aRes = data()->document(DOC_REF())->value();
45 if (!aRes.get() && myIsInLoad) { // trying to get this document from the session
46 aRes = document()->subDocument(data()->name());
51 Model_ResultPart::Model_ResultPart()
53 myIsDisabled = true; // by default it is not initialized and false to be after created
55 setIsConcealed(false);
58 void Model_ResultPart::activate()
61 baseRef()->activate();
65 std::shared_ptr<ModelAPI_AttributeDocRef> aDocRef = data()->document(DOC_REF());
67 if (!aDocRef->value().get()) { // create (or open) a document if it is not yet created
69 std::shared_ptr<ModelAPI_Document> aDoc = document()->subDocument(data()->name());
72 aDoc->synchronizeTransactions();
73 aDocRef->setValue(aDoc);
76 if (aDocRef->value().get()) {
77 SessionPtr aMgr = ModelAPI_Session::get();
78 bool isNewTransaction = !aMgr->isOperation();
79 // activation may cause changes in current features in document, so it must be in transaction
80 if (isNewTransaction) {
81 aMgr->startOperation("Activation");
83 ModelAPI_Session::get()->setActiveDocument(aDocRef->value());
84 if (isNewTransaction) {
85 aMgr->finishOperation();
90 std::shared_ptr<ModelAPI_ResultPart> Model_ResultPart::original()
93 return baseRef()->original();
95 return std::dynamic_pointer_cast<ModelAPI_ResultPart>(data()->owner());
98 bool Model_ResultPart::isActivated()
101 return baseRef()->isActivated();
104 std::shared_ptr<ModelAPI_AttributeDocRef> aDocRef = data()->document(DOC_REF());
105 return aDocRef->value().get() != NULL;
108 bool Model_ResultPart::setDisabled(std::shared_ptr<ModelAPI_Result> theThis,
111 if (ModelAPI_ResultPart::setDisabled(theThis, theFlag)) {
112 if (!myTrsf.get()) { // disable of base result part
113 DocumentPtr aDoc = Model_ResultPart::partDoc();
114 if (aDoc.get() && aDoc->isOpened()) {
115 // make the current feature the last in any case: to update shapes defore deactivation too
116 FeaturePtr aLastFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aDoc->object(
117 ModelAPI_Feature::group(), aDoc->size(ModelAPI_Feature::group()) - 1));
118 aDoc->setCurrentFeature(aLastFeature, false);
119 if (theFlag) { // disable, so make all features disabled too
120 // update the shape just before the deactivation: it will be used outside of part
123 aDoc->setCurrentFeature(FeaturePtr(), false);
132 std::shared_ptr<GeomAPI_Shape> Model_ResultPart::shape()
134 std::shared_ptr<GeomAPI_Shape> aResult(new GeomAPI_Shape);
135 if (myTrsf.get()) { // get shape of the base result and apply the transformation
136 ResultPtr anOrigResult = baseRef();
137 std::shared_ptr<GeomAPI_Shape> anOrigShape = anOrigResult->shape();
138 if (anOrigShape.get()) {
139 TopoDS_Shape aShape = anOrigShape->impl<TopoDS_Shape>();
140 if (!aShape.IsNull()) {
141 aShape.Move(*(myTrsf.get()));
142 aResult->setImpl(new TopoDS_Shape(aShape));
147 if (myShape.IsNull()) { // shape is not produced yet, create it
148 DocumentPtr aDoc = Model_ResultPart::partDoc();
149 if (aDoc.get() && aDoc->isOpened()) {
150 const std::string& aBodyGroup = ModelAPI_ResultBody::group();
151 TopoDS_Compound aResultComp;
152 BRep_Builder aBuilder;
153 aBuilder.MakeCompound(aResultComp);
155 for(int a = aDoc->size(aBodyGroup) - 1; a >= 0; a--) {
156 ResultPtr aBody = std::dynamic_pointer_cast<ModelAPI_Result>(aDoc->object(aBodyGroup, a));
157 // "object" method filters out disabled and concealed anyway, so don't check
158 if (aBody.get() && aBody->shape().get()) {
159 TopoDS_Shape aShape = *(aBody->shape()->implPtr<TopoDS_Shape>());
160 if (!aShape.IsNull()) {
161 aBuilder.Add(aResultComp, aShape);
167 myShape = aResultComp;
171 if (!myShape.IsNull())
172 aResult->setImpl(new TopoDS_Shape(myShape));
176 // Returns true is transformation matrices are equal
177 static bool IsEqualTrsf(gp_Trsf& theT1, gp_Trsf theT2) {
178 for(int aRow = 1; aRow < 4; aRow++) {
179 for(int aCol = 1; aCol < 5; aCol++) {
180 double aDiff = theT1.Value(aRow, aCol) - theT2.Value(aRow, aCol);
181 if (Abs(aDiff) > 1.e-9)
188 std::string Model_ResultPart::nameInPart(const std::shared_ptr<GeomAPI_Shape>& theShape,
191 theIndex = 0; // not initialized
193 if (myTrsf.get()) { // if this is moved copy of part => return the name of original shape
194 ResultPartPtr anOrigResult = baseRef();
195 // searching in the origin the shape equal to the given but with myTrsf
196 TopoDS_Shape aSelection = theShape->impl<TopoDS_Shape>();
197 gp_Trsf aSelTrsf = aSelection.Location().Transformation();
198 TopoDS_Shape anOrigMain = anOrigResult->shape()->impl<TopoDS_Shape>();
199 if (!aSelection.IsNull() && !anOrigMain.IsNull()) {
200 TopExp_Explorer anExp(anOrigMain, aSelection.ShapeType());
201 for(; anExp.More(); anExp.Next()) {
202 if (anExp.Current().IsPartner(aSelection)) {
203 TopoDS_Shape anOrigMoved = anExp.Current().Moved(*(myTrsf.get()));
204 //if (anOrigMoved.IsSame(aSelection)) {
205 if (IsEqualTrsf(aSelTrsf, anOrigMoved.Location().Transformation())) {
206 std::shared_ptr<GeomAPI_Shape> anOrigSel(new GeomAPI_Shape);
207 anOrigSel->setImpl(new TopoDS_Shape(anExp.Current()));
208 return anOrigResult->nameInPart(anOrigSel, theIndex);
213 // something is not right
217 TopoDS_Shape aShape = theShape->impl<TopoDS_Shape>();
221 // getting an access to the document of part
222 std::shared_ptr<Model_Document> aDoc = std::dynamic_pointer_cast<Model_Document>(partDoc());
223 if (!aDoc.get()) // the part document is not presented for the moment
225 TDF_Label anAccessLabel = aDoc->generalLabel();
228 // check if the subShape is already in DF
229 Handle(TNaming_NamedShape) aNS = TNaming_Tool::NamedShape(aShape, anAccessLabel);
230 Handle(TDataStd_Name) anAttr;
231 if(!aNS.IsNull() && !aNS->IsEmpty()) { // in the document
232 if(aNS->Label().FindAttribute(TDataStd_Name::GetID(), anAttr)) {
233 aName = TCollection_AsciiString(anAttr->Get()).ToCString();
235 const TDF_Label& aLabel = aDoc->findNamingName(aName);
237 static const std::string aPostFix("_");
238 TNaming_Iterator anItL(aNS);
239 for(int i = 1; anItL.More(); anItL.Next(), i++) {
240 if(anItL.NewShape() == aShape) {
242 aName += TCollection_AsciiString (i).ToCString();
249 if (aName.empty()) { // not found, so use the selection mechanism
250 // for this the context result is needed
252 const std::string& aBodyGroup = ModelAPI_ResultBody::group();
253 for(int a = aDoc->size(aBodyGroup) - 1; a >= 0; a--) {
254 ResultPtr aBody = std::dynamic_pointer_cast<ModelAPI_Result>(aDoc->object(aBodyGroup, a));
255 if (aBody.get() && aBody->shape().get() && !aBody->isDisabled()) {
256 TopoDS_Shape aBodyShape = *(aBody->shape()->implPtr<TopoDS_Shape>());
257 // check is body contain the selected sub-shape
258 for(TopExp_Explorer anExp(aBodyShape, aShape.ShapeType()); anExp.More(); anExp.Next()) {
259 if (aShape.IsEqual(anExp.Current())) {
266 if (aContext.get()) {
267 AttributeSelectionListPtr aSelAttr = aDoc->selectionInPartFeature();
268 aSelAttr->append(aContext, theShape);
269 theIndex = aSelAttr->size();
270 AttributeSelectionPtr aNewAttr = aSelAttr->value(theIndex - 1);
271 return aNewAttr->namingName();
277 bool Model_ResultPart::updateInPart(const int theIndex)
279 std::shared_ptr<Model_Document> aDoc = std::dynamic_pointer_cast<Model_Document>(partDoc());
281 AttributeSelectionListPtr aSelAttr = aDoc->selectionInPartFeature();
282 AttributeSelectionPtr aThisAttr = aSelAttr->value(theIndex - 1);
283 if (aThisAttr.get()) {
284 return aThisAttr->update();
287 return false; // something is wrong
290 std::shared_ptr<GeomAPI_Shape> Model_ResultPart::shapeInPart(const std::string& theName)
292 /// TODO: not implemented yet
293 return std::shared_ptr<GeomAPI_Shape>();
297 void Model_ResultPart::colorConfigInfo(std::string& theSection, std::string& theName,
298 std::string& theDefault)
300 theSection = "Visualization";
301 theName = "result_part_color";
302 theDefault = DEFAULT_COLOR();
305 void Model_ResultPart::updateShape()
311 void Model_ResultPart::setTrsf(std::shared_ptr<ModelAPI_Result> theThis,
312 const std::shared_ptr<GeomAPI_Trsf>& theTransformation)
315 if (theTransformation.get()) {
316 myTrsf = std::shared_ptr<gp_Trsf>(new gp_Trsf(theTransformation->impl<gp_Trsf>()));
318 // the result must be explicitly updated
319 static Events_Loop* aLoop = Events_Loop::loop();
320 static Events_ID EVENT_DISP = aLoop->eventByName(EVENT_OBJECT_TO_REDISPLAY);
321 ModelAPI_EventCreator::get()->sendUpdated(theThis, EVENT_DISP); // flush is in preview-update