Salome HOME
30.10.2014. Redesigned boolean feature.
[modules/shaper.git] / src / Model / Model_ResultBody.cpp
1 // File:        Model_ResultBody.cpp
2 // Created:     08 Jul 2014
3 // Author:      Mikhail PONIKAROV
4
5 #include <Model_ResultBody.h>
6 #include <Model_Data.h>
7 #include <TNaming_Builder.hxx>
8 #include <TNaming_NamedShape.hxx>
9 #include <TopoDS_Shape.hxx>
10 #include <TDF_ChildIterator.hxx>
11 #include <TopTools_MapOfShape.hxx>
12 #include <TopExp_Explorer.hxx>
13 #include <GeomAPI_Shape.h>
14 #include <GeomAlgoAPI_MakeShape.h>
15 // DEB
16 //#include <TCollection_AsciiString.hxx>
17 //#include <TDF_Tool.hxx>
18 Model_ResultBody::Model_ResultBody()
19 {
20   setIsConcealed(false);
21 }
22
23 void Model_ResultBody::store(const boost::shared_ptr<GeomAPI_Shape>& theShape)
24 {
25   boost::shared_ptr<Model_Data> aData = boost::dynamic_pointer_cast<Model_Data>(data());
26   if (aData) {
27     TDF_Label& aShapeLab = aData->shapeLab();
28     // clean builders
29     clean();   
30     // store the new shape as primitive
31     TNaming_Builder aBuilder(aShapeLab);
32     if (!theShape)
33       return;  // bad shape
34     TopoDS_Shape aShape = theShape->impl<TopoDS_Shape>();
35     if (aShape.IsNull())
36       return;  // null shape inside
37
38     aBuilder.Generated(aShape);
39   }
40 }
41
42 void Model_ResultBody::storeGenerated(const boost::shared_ptr<GeomAPI_Shape>& theFromShape,
43                                           const boost::shared_ptr<GeomAPI_Shape>& theToShape)
44 {
45   boost::shared_ptr<Model_Data> aData = boost::dynamic_pointer_cast<Model_Data>(data());
46   if (aData) {
47     TDF_Label& aShapeLab = aData->shapeLab();
48     // clean builders
49     clean();   
50     // store the new shape as primitive
51     TNaming_Builder aBuilder(aShapeLab);
52     if (!theFromShape || !theToShape)
53       return;  // bad shape
54     TopoDS_Shape aShapeBasis = theFromShape->impl<TopoDS_Shape>();
55     if (aShapeBasis.IsNull())
56       return;  // null shape inside
57         TopoDS_Shape aShapeNew = theToShape->impl<TopoDS_Shape>();
58     if (aShapeNew.IsNull())
59       return;  // null shape inside
60     aBuilder.Generated(aShapeBasis, aShapeNew);
61   }
62 }
63
64 void Model_ResultBody::storeModified(const boost::shared_ptr<GeomAPI_Shape>& theOldShape,
65                                           const boost::shared_ptr<GeomAPI_Shape>& theNewShape)
66 {
67   boost::shared_ptr<Model_Data> aData = boost::dynamic_pointer_cast<Model_Data>(data());
68   if (aData) {
69     TDF_Label& aShapeLab = aData->shapeLab();
70     // clean builders
71     clean();   
72     // store the new shape as primitive
73     TNaming_Builder aBuilder(aShapeLab);
74     if (!theOldShape || !theNewShape)
75       return;  // bad shape
76     TopoDS_Shape aShapeOld = theOldShape->impl<TopoDS_Shape>();
77     if (aShapeOld.IsNull())
78       return;  // null shape inside
79         TopoDS_Shape aShapeNew = theNewShape->impl<TopoDS_Shape>();
80     if (aShapeNew.IsNull())
81       return;  // null shape inside
82     aBuilder.Generated(aShapeOld, aShapeNew);
83   }
84 }
85
86 boost::shared_ptr<GeomAPI_Shape> Model_ResultBody::shape()
87 {
88   boost::shared_ptr<Model_Data> aData = boost::dynamic_pointer_cast<Model_Data>(data());
89   if (aData) {
90     TDF_Label& aShapeLab = aData->shapeLab();
91     Handle(TNaming_NamedShape) aName;
92     if (aShapeLab.FindAttribute(TNaming_NamedShape::GetID(), aName)) {
93       TopoDS_Shape aShape = aName->Get();
94       if (!aShape.IsNull()) {
95         boost::shared_ptr<GeomAPI_Shape> aRes(new GeomAPI_Shape);
96         aRes->setImpl(new TopoDS_Shape(aShape));
97         return aRes;
98       }
99     }
100   }
101   return boost::shared_ptr<GeomAPI_Shape>();
102 }
103
104 boost::shared_ptr<ModelAPI_Feature> Model_ResultBody::owner()
105 {
106   return myOwner;
107 }
108
109 void Model_ResultBody::clean()
110 {
111   std::vector<TNaming_Builder*>::iterator aBuilder = myBuilders.begin();
112   for(; aBuilder != myBuilders.end(); aBuilder++)
113     delete *aBuilder;
114 }
115
116 Model_ResultBody::~Model_ResultBody()
117 {
118   clean();
119 }
120
121 TNaming_Builder* Model_ResultBody::builder(const int theTag)
122 {
123   if (myBuilders.size() <= (unsigned int)theTag) {
124     myBuilders.insert(myBuilders.end(), theTag - myBuilders.size() + 1, NULL);
125   }
126   if (!myBuilders[theTag]) {
127     boost::shared_ptr<Model_Data> aData = boost::dynamic_pointer_cast<Model_Data>(data());
128     myBuilders[theTag] = new TNaming_Builder(aData->shapeLab().FindChild(theTag));
129         //TCollection_AsciiString entry;//
130         //TDF_Tool::Entry(aData->shapeLab().FindChild(theTag), entry);
131         //cout << "Label = " <<entry.ToCString() <<endl;
132   }
133   return myBuilders[theTag];
134 }
135
136 void Model_ResultBody::generated(
137   const boost::shared_ptr<GeomAPI_Shape>& theNewShape, const int theTag)
138 {
139   TopoDS_Shape aShape = theNewShape->impl<TopoDS_Shape>();
140   builder(theTag)->Generated(aShape);
141 }
142
143 void Model_ResultBody::generated(const boost::shared_ptr<GeomAPI_Shape>& theOldShape,
144     const boost::shared_ptr<GeomAPI_Shape>& theNewShape, const int theTag)
145 {
146   TopoDS_Shape anOldShape = theOldShape->impl<TopoDS_Shape>();
147   TopoDS_Shape aNewShape = theNewShape->impl<TopoDS_Shape>();
148   builder(theTag)->Generated(anOldShape, aNewShape);
149 }
150
151
152 void Model_ResultBody::modified(const boost::shared_ptr<GeomAPI_Shape>& theOldShape,
153     const boost::shared_ptr<GeomAPI_Shape>& theNewShape, const int theTag)
154 {
155   TopoDS_Shape anOldShape = theOldShape->impl<TopoDS_Shape>();
156   TopoDS_Shape aNewShape = theNewShape->impl<TopoDS_Shape>();
157   builder(theTag)->Modify(anOldShape, aNewShape);
158 }
159
160 void Model_ResultBody::deleted(const boost::shared_ptr<GeomAPI_Shape>& theOldShape,
161     const int theTag)
162 {
163   TopoDS_Shape aShape = theOldShape->impl<TopoDS_Shape>();
164   builder(theTag)->Delete(aShape);
165 }
166
167 void Model_ResultBody::loadDeletedShapes (GeomAlgoAPI_MakeShape* theMS,
168                                                boost::shared_ptr<GeomAPI_Shape>  theShapeIn,
169                                                const int  theKindOfShape,
170                                                const int  theTag)
171 {
172   TopoDS_Shape aShapeIn = theShapeIn->impl<TopoDS_Shape>();
173   TopTools_MapOfShape aView;
174   TopExp_Explorer ShapeExplorer (aShapeIn, (TopAbs_ShapeEnum)theKindOfShape);
175   for (; ShapeExplorer.More(); ShapeExplorer.Next ()) {
176     const TopoDS_Shape& aRoot = ShapeExplorer.Current ();
177     if (!aView.Add(aRoot)) continue;
178         boost::shared_ptr<GeomAPI_Shape> aRShape(new GeomAPI_Shape());
179         aRShape->setImpl((new TopoDS_Shape(aRoot)));
180     if (theMS->isDeleted (aRShape)) {
181                 builder(theTag)->Delete(aRoot);
182     }
183   }
184 }
185
186 void Model_ResultBody::loadAndOrientModifiedShapes (
187                                                    GeomAlgoAPI_MakeShape* theMS,
188                                                boost::shared_ptr<GeomAPI_Shape>  theShapeIn,
189                                                const int  theKindOfShape,
190                                                const int  theTag,
191                                                GeomAPI_DataMapOfShapeShape& theSubShapes)
192 {
193   TopoDS_Shape aShapeIn = theShapeIn->impl<TopoDS_Shape>();
194   TopTools_MapOfShape aView;
195   TopExp_Explorer aShapeExplorer (aShapeIn, (TopAbs_ShapeEnum)theKindOfShape);
196   for (; aShapeExplorer.More(); aShapeExplorer.Next ()) {
197     const TopoDS_Shape& aRoot = aShapeExplorer.Current ();
198     if (!aView.Add(aRoot)) continue;
199         ListOfShape aList;
200         boost::shared_ptr<GeomAPI_Shape> aRShape(new GeomAPI_Shape());
201         aRShape->setImpl((new TopoDS_Shape(aRoot)));
202         theMS->generated(aRShape, aList);
203         std::list<boost::shared_ptr<GeomAPI_Shape> >::const_iterator anIt = aList.begin(), aLast = aList.end();
204     for (; anIt != aLast; anIt++) {
205       TopoDS_Shape aNewShape = (*anIt)->impl<TopoDS_Shape>();     
206           if (theSubShapes.isBound(*anIt)) {
207                 boost::shared_ptr<GeomAPI_Shape> aMapShape(theSubShapes.find(*anIt));
208                 aNewShape.Orientation(aMapShape->impl<TopoDS_Shape>().Orientation());
209       }
210       if (!aRoot.IsSame (aNewShape)) 
211                   builder(theTag)->Modify(aRoot,aNewShape);
212     }
213   }
214 }
215
216 void Model_ResultBody::loadAndOrientGeneratedShapes (
217                                                    GeomAlgoAPI_MakeShape* theMS,
218                                                boost::shared_ptr<GeomAPI_Shape>  theShapeIn,
219                                                const int  theKindOfShape,
220                                                const int  theTag,
221                                                GeomAPI_DataMapOfShapeShape& theSubShapes)
222 {
223   TopoDS_Shape aShapeIn = theShapeIn->impl<TopoDS_Shape>();
224   TopTools_MapOfShape aView;
225   TopExp_Explorer aShapeExplorer (aShapeIn, (TopAbs_ShapeEnum)theKindOfShape);
226   for (; aShapeExplorer.More(); aShapeExplorer.Next ()) {
227     const TopoDS_Shape& aRoot = aShapeExplorer.Current ();
228     if (!aView.Add(aRoot)) continue;
229         ListOfShape aList;
230         boost::shared_ptr<GeomAPI_Shape> aRShape(new GeomAPI_Shape());
231         aRShape->setImpl((new TopoDS_Shape(aRoot)));
232         theMS->generated(aRShape, aList);
233         std::list<boost::shared_ptr<GeomAPI_Shape> >::const_iterator anIt = aList.begin(), aLast = aList.end();
234     for (; anIt != aLast; anIt++) {
235       TopoDS_Shape aNewShape = (*anIt)->impl<TopoDS_Shape>();     
236           if (theSubShapes.isBound(*anIt)) {
237                 boost::shared_ptr<GeomAPI_Shape> aMapShape(theSubShapes.find(*anIt));
238                 aNewShape.Orientation(aMapShape->impl<TopoDS_Shape>().Orientation());
239       }
240       if (!aRoot.IsSame (aNewShape)) 
241                   builder(theTag)->Generated(aRoot,aNewShape);
242     }
243   }
244 }